Artemis/src/Artemis.UI/Screens/SurfaceEditor/Visualization/SurfaceDeviceViewModel.cs

using System;
using System.Collections.Generic;
using System.Windows;
using System.Windows.Input;
using Artemis.Core.Models.Surface;
using PropertyChanged;
using RGB.NET.Core;
using Stylet;
using Point = System.Windows.Point;

namespace Artemis.UI.Screens.SurfaceEditor.Visualization
{
    public class SurfaceDeviceViewModel : PropertyChangedBase
    {
        private readonly List<SurfaceLedViewModel> _leds;
        private double _dragOffsetX;
        private double _dragOffsetY;

        public SurfaceDeviceViewModel(Device device)
        {
            Device = device;
            _leds = new List<SurfaceLedViewModel>();

            if (Device.RgbDevice != null)
            {
                foreach (var led in Device.RgbDevice)
                    _leds.Add(new SurfaceLedViewModel(led));
            }

            Device.DeviceUpdated += DeviceOnDeviceUpdated;
        }

        private void DeviceOnDeviceUpdated(object sender, EventArgs e)
        {
            NotifyOfPropertyChange(() => RgbDeviceRectangle);        
            NotifyOfPropertyChange(() => RgbDeviceRectangle.Location.X);        
            NotifyOfPropertyChange(() => RgbDeviceRectangle.Location.Y);     
            
        }

        public Device Device { get; set; }
        public SelectionStatus SelectionStatus { get; set; }
        public Cursor Cursor { get; set; }

        public Rectangle RgbDeviceRectangle => Device.RgbDevice.DeviceRectangle;

        public IReadOnlyCollection<SurfaceLedViewModel> Leds => _leds.AsReadOnly();

        public Rect DeviceRectangle => Device.RgbDevice == null
            ? new Rect()
            : new Rect(Device.X, Device.Y, Device.RgbDevice.Size.Width, Device.RgbDevice.Size.Height);

        public void StartMouseDrag(Point mouseStartPosition)
        {
            _dragOffsetX = Device.X - mouseStartPosition.X;
            _dragOffsetY = Device.Y - mouseStartPosition.Y;
        }

        public void UpdateMouseDrag(Point mousePosition)
        {
            var roundedX = Math.Round((mousePosition.X + _dragOffsetX) / 10, 0, MidpointRounding.AwayFromZero) * 10;
            var roundedY = Math.Round((mousePosition.Y + _dragOffsetY) / 10, 0, MidpointRounding.AwayFromZero) * 10;
            Device.X = Math.Max(0, roundedX);
            Device.Y = Math.Max(0, roundedY);
        }

        // ReSharper disable once UnusedMember.Global - Called from view
        public void MouseEnter(object sender, MouseEventArgs e)
        {
            if (SelectionStatus == SelectionStatus.None)
            {
                SelectionStatus = SelectionStatus.Hover;
                Cursor = Cursors.Hand;
            }
        }

        // ReSharper disable once UnusedMember.Global - Called from view
        public void MouseLeave()
        {
            if (SelectionStatus == SelectionStatus.Hover)
            {
                SelectionStatus = SelectionStatus.None;
                Cursor = Cursors.Arrow;
            }
        }

        public MouseDevicePosition GetMouseDevicePosition(Point position)
        {
            if ((new Point(0, 0) - position).LengthSquared < 5)
            {
                return MouseDevicePosition.TopLeft;
            }

            return MouseDevicePosition.Regular;
        }
    }

    public enum MouseDevicePosition
    {
        Regular,
        TopLeft,
        TopRight,
        BottomLeft,
        BottomRight
    }

    public enum SelectionStatus
    {
        None,
        Hover,
        Selected
    }
}