Artemis/src/Artemis.UI.Shared/Controls/DeviceVisualizerLed.cs

using System;
using System.IO;
using System.Windows;
using System.Windows.Media;
using System.Windows.Media.Imaging;
using Artemis.Core.Models.Surface;
using RGB.NET.Core;
using Color = System.Windows.Media.Color;

namespace Artemis.UI.Shared.Controls
{
    internal class DeviceVisualizerLed
    {
        public DeviceVisualizerLed(ArtemisLed led)
        {
            Led = led;
            LedRect = new Rect(
                Led.RgbLed.LedRectangle.Location.X,
                Led.RgbLed.LedRectangle.Location.Y,
                Led.RgbLed.LedRectangle.Size.Width,
                Led.RgbLed.LedRectangle.Size.Height
            );

            if (Led.RgbLed.Image != null && File.Exists(Led.RgbLed.Image.AbsolutePath))
                LedImage = new BitmapImage(Led.RgbLed.Image);

            CreateLedGeometry();
        }


        public ArtemisLed Led { get; }
        public Rect LedRect { get; set; }
        public BitmapImage LedImage { get; set; }

        public Geometry DisplayGeometry { get; private set; }

        public void RenderColor(DrawingContext drawingContext, bool isDimmed)
        {
            if (DisplayGeometry == null)
                return;

            var r = Led.RgbLed.Color.GetR();
            var g = Led.RgbLed.Color.GetG();
            var b = Led.RgbLed.Color.GetB();

            drawingContext.DrawRectangle(isDimmed
                ? new SolidColorBrush(Color.FromArgb(100, r, g, b))
                : new SolidColorBrush(Color.FromRgb(r, g, b)), null, LedRect);
        }

        public void RenderImage(DrawingContext drawingContext)
        {
            if (LedImage == null)
                return;

            drawingContext.DrawImage(LedImage, LedRect);
        }

        public void RenderOpacityMask(DrawingContext drawingContext)
        {
            if (DisplayGeometry == null)
                return;

            var fillBrush = new SolidColorBrush(Color.FromArgb(100, 255, 255, 255));
            fillBrush.Freeze();
            var penBrush = new SolidColorBrush(Color.FromArgb(255, 255, 255, 255));
            penBrush.Freeze();

            drawingContext.DrawGeometry(fillBrush, new Pen(penBrush, 1), DisplayGeometry);
        }

        private void CreateLedGeometry()
        {
            // The minimum required size for geometry to be created
            if (Led.RgbLed.Size.Width < 2 || Led.RgbLed.Size.Height < 2)
                return;

            switch (Led.RgbLed.Shape)
            {
                case Shape.Custom:
                    if (Led.RgbLed.Device.DeviceInfo.DeviceType == RGBDeviceType.Keyboard || Led.RgbLed.Device.DeviceInfo.DeviceType == RGBDeviceType.Keypad)
                        CreateCustomGeometry(2.0);
                    else
                        CreateCustomGeometry(1.0);
                    break;
                case Shape.Rectangle:
                    if (Led.RgbLed.Device.DeviceInfo.DeviceType == RGBDeviceType.Keyboard || Led.RgbLed.Device.DeviceInfo.DeviceType == RGBDeviceType.Keypad)
                        CreateKeyCapGeometry();
                    else
                        CreateRectangleGeometry();
                    break;
                case Shape.Circle:
                    CreateCircleGeometry();
                    break;
                default:
                    throw new ArgumentOutOfRangeException();
            }

            // Stroke geometry is the display geometry excluding the inner geometry
            DisplayGeometry.Transform = new TranslateTransform(Led.RgbLed.LedRectangle.Location.X, Led.RgbLed.LedRectangle.Location.Y);
            // Try to gain some performance
            DisplayGeometry.Freeze();
        }

        private void CreateRectangleGeometry()
        {
            DisplayGeometry = new RectangleGeometry(new Rect(0.5, 0.5, Led.RgbLed.Size.Width - 1, Led.RgbLed.Size.Height - 1));
        }

        private void CreateCircleGeometry()
        {
            DisplayGeometry = new EllipseGeometry(new Rect(0.5, 0.5, Led.RgbLed.Size.Width - 1, Led.RgbLed.Size.Height - 1));
        }

        private void CreateKeyCapGeometry()
        {
            DisplayGeometry = new RectangleGeometry(new Rect(1, 1, Led.RgbLed.Size.Width - 2, Led.RgbLed.Size.Height - 2), 1.6, 1.6);
        }

        private void CreateCustomGeometry(double deflateAmount)
        {
            try
            {
                DisplayGeometry = Geometry.Combine(
                    Geometry.Empty,
                    Geometry.Parse(Led.RgbLed.ShapeData),
                    GeometryCombineMode.Union,
                    new TransformGroup
                    {
                        Children = new TransformCollection
                        {
                            new ScaleTransform(Led.RgbLed.Size.Width - deflateAmount, Led.RgbLed.Size.Height - deflateAmount),
                            new TranslateTransform(deflateAmount / 2, deflateAmount / 2)
                        }
                    }
                );
            }
            catch (Exception)
            {
                CreateRectangleGeometry();
            }
        }
    }
}