Artemis/Artemis/Artemis/DeviceProviders/Corsair/CorsairMouse.cs

using System;
using System.Drawing;
using System.Threading;
using CUE.NET;
using CUE.NET.Brushes;
using CUE.NET.Devices.Generic.Enums;
using Ninject.Extensions.Logging;

namespace Artemis.DeviceProviders.Corsair
{
    public class CorsairMouse : DeviceProvider
    {
        private readonly ImageBrush _mouseBrush;

        public CorsairMouse(ILogger logger)
        {
            Logger = logger;
            Type = DeviceType.Mouse;
            _mouseBrush = new ImageBrush();
        }

        public ILogger Logger { get; set; }

        public override bool TryEnable()
        {
            try
            {
                CanUse = CanInitializeSdk();
                if (CanUse && !CueSDK.IsInitialized)
                    CueSDK.Initialize();
            }
            catch (Exception)
            {
                CanUse = false;
            }

            Logger.Debug("Attempted to enable Corsair mice. CanUse: {0}", CanUse);

            if (CanUse)
            {
                CueSDK.UpdateMode = UpdateMode.Manual;
                CueSDK.MouseSDK.Brush = _mouseBrush;
            }

            return CanUse;
        }

        public override void Disable()
        {
            throw new NotSupportedException("Can only disable a keyboard");
        }

        public override void UpdateDevice(Bitmap bitmap)
        {
            if (!CanUse || bitmap == null)
                return;
            if (bitmap.Width != bitmap.Height)
                throw new ArgumentException("Bitmap must be a perfect square");

            _mouseBrush.Image = bitmap;
            CueSDK.MouseSDK.Update();
        }

        private static bool CanInitializeSdk()
        {
            // This will skip the check-loop if the SDK is initialized
            if (CueSDK.IsInitialized)
                return CueSDK.IsSDKAvailable(CorsairDeviceType.Mouse);

            for (var tries = 0; tries < 9; tries++)
            {
                if (CueSDK.IsSDKAvailable(CorsairDeviceType.Mouse))
                    return true;
                Thread.Sleep(2000);
            }
            return false;
        }
    }
}