RGB.NET/RGB.NET.Devices.OpenRGB/Generic/OpenRGBGenericDevice.cs

using OpenRGB.NET.Enums;
using RGB.NET.Core;

namespace RGB.NET.Devices.OpenRGB.Generic
{
    /// <inheritdoc />
    public class OpenRGBGenericDevice : AbstractOpenRGBDevice<OpenRGBGenericDeviceInfo>
    {
        /// <summary>
        /// Initializes a new instance of the <see cref="OpenRGBGenericDevice"/> class.
        /// </summary>
        /// <param name="info">Generic information for the device.</param>
        /// <param name="updateQueue">The queue used to update the device.</param>
        public OpenRGBGenericDevice(OpenRGBGenericDeviceInfo info, IUpdateQueue updateQueue)
            : base(info, updateQueue)
        {
            InitializeLayout();
        }

        /// <summary>
        /// Initializes the LEDs of the device based on the data provided by the SDK.
        /// </summary>
        private void InitializeLayout()
        {
            LedId initial = Helper.GetInitialLedIdForDeviceType(DeviceInfo.DeviceType);

            int y = 0;
            Size ledSize = new Size(19);
            int zoneLedIndex = 0;
            const int ledSpacing = 20;

            foreach (global::OpenRGB.NET.Models.Zone? zone in DeviceInfo.OpenRGBDevice.Zones)
            {
                if (zone.Type == ZoneType.Matrix)
                {
                    for (int row = 0; row < zone.MatrixMap.Height; row++)
                    {
                        for (int column = 0; column < zone.MatrixMap.Width; column++)
                        {
                            uint index = zone.MatrixMap.Matrix[row, column];

                            //will be max value if the position does not have an associated key
                            if (index == uint.MaxValue)
                                continue;

                            LedId ledId = StandardKeyNames.Default.TryGetValue(DeviceInfo.OpenRGBDevice.Leds[zoneLedIndex + index].Name, out LedId l)
                                ? l
                                : initial++;

                            //HACK: doing this because some different Led Names are mapped to the same LedId
                            //for example, "Enter" and "ISO Enter".
                            //this way, at least they'll be controllable as CustomX
                            while (AddLed(ledId, new Point(ledSpacing * column, y + (ledSpacing * row)), ledSize, zoneLedIndex + (int)index) == null)
                                ledId = initial++;
                        }
                    }
                    y += (int)(zone.MatrixMap.Height * ledSpacing);
                }
                else
                {
                    for (int i = 0; i < zone.LedCount; i++)
                    {
                        LedId ledId = initial++;

                        while (AddLed(ledId, new Point(ledSpacing * i, y), ledSize, zoneLedIndex + i) == null)
                            ledId = initial++;
                    }
                }

                //we'll just set each zone in its own row for now,
                //with each led for that zone being horizontally distributed
                y += ledSpacing;
                zoneLedIndex += (int)zone.LedCount;
            }
        }
    }
}