Artemis/src/Artemis.Plugins.Modules.General/GeneralModule.cs

using System;
using System.Collections.Generic;
using System.Drawing;
using Artemis.Core;
using Artemis.Core.Plugins.Abstract;
using Artemis.Core.Plugins.Models;
using Artemis.Core.Services.Interfaces;
using Artemis.Plugins.Modules.General.ViewModels;
using QRCoder;
using RGB.NET.Core;
using Stylet;
using Color = System.Drawing.Color;
using Rectangle = System.Drawing.Rectangle;

namespace Artemis.Plugins.Modules.General
{
    public class GeneralModule : Module
    {
        private readonly PluginSettings _settings;
        private readonly RGBSurface _surface;
        private Dictionary<Led, Color> _colors;

        public GeneralModule(PluginInfo pluginInfo, IRgbService rgbService, PluginSettings settings) : base(pluginInfo)
        {
            _settings = settings;
            DisplayName = "General";
            ExpandsMainDataModel = true;

            _surface = rgbService.Surface;
            _colors = new Dictionary<Led, Color>();
            
            rgbService.FinishedLoadedDevices += (sender, args) => PopulateColors();

            var testSetting = _settings.GetSetting("TestSetting", DateTime.Now);
        }

        public override void EnablePlugin()
        {
            var qrGenerator = new QRCodeGenerator();
            PopulateColors();
        }

        public override void DisablePlugin()
        {
        }

        public override void Update(double deltaTime)
        {
            if (_colors == null)
                return;

            foreach (var surfaceLed in _surface.Leds)
                UpdateLedColor(surfaceLed, deltaTime);
        }

        public override void Render(double deltaTime, RGBSurface surface, Graphics graphics)
        {
            foreach (var surfaceLed in _surface.Leds)
            {
                if (!_colors.ContainsKey(surfaceLed))
                    continue;

                var brush = new SolidBrush(_colors[surfaceLed]);
                var rectangle = new Rectangle((int) surfaceLed.LedRectangle.X, (int) surfaceLed.LedRectangle.Y, (int) surfaceLed.LedRectangle.Width, (int) surfaceLed.LedRectangle.Height);
                graphics.FillRectangle(brush, rectangle);
                UpdateLedColor(surfaceLed, deltaTime);
            }
        }

        public override IScreen GetMainViewModel()
        {
            return new GeneralViewModel(this);
        }

        public override void Dispose()
        {
            _colors = null;
        }

        private void UpdateLedColor(Led led, double deltaTime)
        {
            if (_colors.ContainsKey(led))
                _colors[led] = ColorHelpers.ShiftColor(_colors[led], (int) (deltaTime * 1000));
            else
                _colors[led] = ColorHelpers.GetRandomRainbowColor();
        }

        private void PopulateColors()
        {
            _colors = new Dictionary<Led, Color>();
            foreach (var surfaceLed in _surface.Leds)
                _colors.Add(surfaceLed, ColorHelpers.GetRandomRainbowColor());
        }
    }
}