From 63c85891d5ddc3cc02519c1acccec1a8bb1abfc3 Mon Sep 17 00:00:00 2001
From: Christian Pichler <christian.pichler.msc@gmail.com>
Date: Sun, 5 Aug 2018 14:50:51 +0200
Subject: [PATCH] Added Battery Status for USB Connection

---
 ScpControl/Usb/Ds4/UsbDs4.cs | 47 ++++++++++++++++++++++++++++++++++--
 1 file changed, 45 insertions(+), 2 deletions(-)

diff --git a/ScpControl/Usb/Ds4/UsbDs4.cs b/ScpControl/Usb/Ds4/UsbDs4.cs
index e4705305..bbc5876f 100644
--- a/ScpControl/Usb/Ds4/UsbDs4.cs
+++ b/ScpControl/Usb/Ds4/UsbDs4.cs
@@ -87,6 +87,23 @@ private void SetLightBarColor(DsPadId value)
             }
         }
 
+        private void SetLightBarColorUInt(uint value)
+        {
+            System.Drawing.Color color = System.Drawing.Color.FromArgb((int)value);
+            double brightness = (double)GlobalConfiguration.Instance.Brightness / (double)255;
+
+            if (GlobalConfiguration.Instance.IsLightBarDisabled)
+            {
+                _hidReport[R] = _hidReport[G] = _hidReport[B] = _hidReport[12] = _hidReport[13] = 0x00;
+            }
+            else
+            {
+                _hidReport[R] = (byte)(color.R * brightness);
+                _hidReport[G] = (byte)(color.G * brightness);
+                _hidReport[B] = (byte)(color.B * brightness);
+            }
+        }
+
         #endregion
 
         #region Actions
@@ -314,9 +331,35 @@ protected override void Process(DateTime now)
                     _brightness = 0x00;
                 }
 
-                // set light bar color reflecting pad ID
-                if (XInputSlot.HasValue)
+                if (GlobalConfiguration.Instance.Ds4ShowBatteryInfo)
+                {
+                    switch (Battery)
+                    {
+                        case DsBattery.Dying:
+                            SetLightBarColorUInt(GlobalConfiguration.Instance.Ds4ColorDying);
+                            break;
+                        case DsBattery.Low:
+                            SetLightBarColorUInt(GlobalConfiguration.Instance.Ds4ColorLow);
+                            break;
+                        case DsBattery.Medium:
+                            SetLightBarColorUInt(GlobalConfiguration.Instance.Ds4ColorMedium);
+                            break;
+                        case DsBattery.High:
+                        case DsBattery.Charging:
+                            SetLightBarColorUInt(GlobalConfiguration.Instance.Ds4ColorHigh);
+                            break;
+                        case DsBattery.Full:
+                        case DsBattery.Charged:
+                            SetLightBarColorUInt(GlobalConfiguration.Instance.Ds4ColorFull);
+                            break;
+                        default:
+                            SetLightBarColorUInt(GlobalConfiguration.Instance.Ds4ColorDying);
+                            break;
+                    }
+                }
+                else if (XInputSlot.HasValue)
                 {
+                    // set light bar color reflecting pad ID
                     SetLightBarColor((DsPadId) XInputSlot);
                 }