add db WIP 2, 80%

refactor, renames
2026-04-22 13:56:30 +00:00 · 2024-12-24 06:07:13 +02:00
parent 5fa18aa514
commit 48c9ccbfda
32 changed files with 499 additions and 459 deletions
@@ -0,0 +1,126 @@
+using System.Windows;
+using System.Windows.Controls;
+using System.Windows.Input;
+using System.Windows.Media;
+using System.Windows.Shapes;
+using Azaion.Common.DTO;
+using Label = System.Windows.Controls.Label;
+
+namespace Azaion.Common.Controls;
+
+public class DetectionControl : Border
+{
+    private readonly Action<object, MouseButtonEventArgs> _resizeStart;
+    private const double RESIZE_RECT_SIZE = 9;
+
+    private readonly Grid _grid;
+    private readonly TextBlock _classNameLabel;
+    private readonly Label _probabilityLabel;
+    public TimeSpan? Time { get; set; }
+        
+    private DetectionClass _detectionClass = null!;
+    public DetectionClass DetectionClass
+    {
+        get => _detectionClass;
+        set
+        {
+            _grid.Background = value.ColorBrush;
+            _probabilityLabel.Background = value.ColorBrush;
+            _classNameLabel.Text = value.Name;
+            _detectionClass = value;
+        }
+    }
+    
+    private readonly Rectangle _selectionFrame;
+
+    private bool _isSelected;
+    public bool IsSelected
+    {
+        get => _isSelected;
+        set
+        {
+            _selectionFrame.Visibility = value ? Visibility.Visible : Visibility.Collapsed;
+            _isSelected = value;
+        }
+    }
+    
+    public DetectionControl(DetectionClass detectionClass, TimeSpan? time, Action<object, MouseButtonEventArgs> resizeStart, double? probability = null)
+    {
+        Time = time;
+        _resizeStart = resizeStart;
+        _classNameLabel = new TextBlock
+        {
+            Text = detectionClass.Name,
+            HorizontalAlignment = HorizontalAlignment.Center,
+            VerticalAlignment = VerticalAlignment.Top,
+            Margin = new Thickness(0, 15, 0, 0),
+            FontSize = 14,
+            Cursor = Cursors.SizeAll
+        };
+        _probabilityLabel = new Label
+        {
+            Content = probability.HasValue ? $"{probability.Value:F0}%" : string.Empty,
+            HorizontalAlignment = HorizontalAlignment.Right,
+            VerticalAlignment = VerticalAlignment.Top,
+            Margin = new Thickness(0, -32, 0, 0),
+            FontSize = 16,
+            Visibility = Visibility.Visible
+        };
+        _selectionFrame = new Rectangle
+        {
+            HorizontalAlignment = HorizontalAlignment.Stretch,
+            VerticalAlignment = VerticalAlignment.Stretch,
+            Stroke = new SolidColorBrush(Colors.Black),
+            StrokeThickness = 2,
+            Visibility = Visibility.Collapsed
+        };
+        
+        _grid = new Grid
+        {
+            HorizontalAlignment = HorizontalAlignment.Stretch,
+            VerticalAlignment = VerticalAlignment.Stretch,
+            Children =
+            {
+                _selectionFrame,
+                _classNameLabel,
+                AddRect("rLT", HorizontalAlignment.Left, VerticalAlignment.Top, Cursors.SizeNWSE),
+                AddRect("rCT", HorizontalAlignment.Center, VerticalAlignment.Top, Cursors.SizeNS),
+                AddRect("rRT", HorizontalAlignment.Right, VerticalAlignment.Top, Cursors.SizeNESW),
+                AddRect("rLC", HorizontalAlignment.Left, VerticalAlignment.Center, Cursors.SizeWE),
+                AddRect("rRC", HorizontalAlignment.Right, VerticalAlignment.Center, Cursors.SizeWE),
+                AddRect("rLB", HorizontalAlignment.Left, VerticalAlignment.Bottom, Cursors.SizeNESW),
+                AddRect("rCB", HorizontalAlignment.Center, VerticalAlignment.Bottom, Cursors.SizeNS),
+                AddRect("rRB", HorizontalAlignment.Right, VerticalAlignment.Bottom, Cursors.SizeNWSE)
+            }
+        };
+        if (probability.HasValue)
+            _grid.Children.Add(_probabilityLabel);
+        Child = _grid;
+        Cursor = Cursors.SizeAll;
+        DetectionClass = detectionClass;
+    }
+    
+    //small corners
+    private Rectangle AddRect(string name, HorizontalAlignment ha, VerticalAlignment va, Cursor crs)
+    {
+        var rect = new Rectangle()   // small rectangles at the corners and sides
+        {
+            HorizontalAlignment = ha,
+            VerticalAlignment = va,
+            Width = RESIZE_RECT_SIZE,
+            Height = RESIZE_RECT_SIZE,
+            Stroke = new SolidColorBrush(Color.FromArgb(230, 40, 40, 40)), // small rectangles color
+            Fill = new SolidColorBrush(Color.FromArgb(1, 255, 255, 255)),
+            Cursor = crs,
+            Name = name,
+        };
+        rect.MouseDown += (sender, args) => _resizeStart(sender, args);
+        return rect;
+    }
+
+    public YoloLabel GetLabel(Size canvasSize, Size? videoSize = null)
+    {
+        var label = new CanvasLabel(DetectionClass.Id, Canvas.GetLeft(this), Canvas.GetTop(this), Width, Height);
+        return new YoloLabel(label, canvasSize, videoSize);
+    }
+}