Aura Battery level mode #4827

Author: seerge

app/AnimeMatrix/SlashDevice.cs

@@ -154,25 +154,6 @@ public void SetMode(SlashMode mode)
             Set(CreatePacket([0xD3, 0x04, 0x00, 0x0C, 0x01, modeByte, 0x02, 0x42, 0x03, 0x13, 0x04, 0x11, 0x05, 0x12, 0x06, 0x13]), "SlashMode");
         }
 
-        public static double GetBatteryChargePercentage()
-        {
-            double batteryCharge = 0;
-            try
-            {
-                ManagementObjectSearcher searcher = new ManagementObjectSearcher("SELECT * FROM Win32_Battery");
-                foreach (ManagementObject battery in searcher.Get())
-                {
-                    batteryCharge = Convert.ToDouble(battery["EstimatedChargeRemaining"]);
-                    break; // Assuming only one battery
-                }
-            }
-            catch (ManagementException e)
-            {
-                Console.WriteLine("An error occurred while querying for WMI data: " + e.Message);
-            }
-            return batteryCharge;
-        }
-
         private byte[] GetPercentagePattern(int brightness, double percentage)
         {
             // because 7 segments, within each led segment represents a percentage bracket of (100/7 = 14.2857%)
@@ -195,7 +176,7 @@ private byte[] GetPercentagePattern(int brightness, double percentage)
 
         public void SetBatteryPattern(int brightness)
         {
-            SetCustom(GetPercentagePattern(brightness, 100 * (GetBatteryChargePercentage() / AppConfig.Get("charge_limit", 100))), null);
+            SetCustom(GetPercentagePattern(brightness, 100 * (HardwareControl.GetBatteryChargePercentage() / AppConfig.Get("charge_limit", 100))), null);
         }
 
         public void SetEmpty()

app/HardwareControl.cs

@@ -278,6 +278,25 @@ public static void ReadSensors(bool log = false)
         }
     }
 
+    public static double GetBatteryChargePercentage()
+    {
+        double batteryCharge = 0;
+        try
+        {
+            ManagementObjectSearcher searcher = new ManagementObjectSearcher("SELECT * FROM Win32_Battery");
+            foreach (ManagementObject battery in searcher.Get())
+            {
+                batteryCharge = Convert.ToDouble(battery["EstimatedChargeRemaining"]);
+                break; // Assuming only one battery
+            }
+        }
+        catch (ManagementException e)
+        {
+            Console.WriteLine("An error occurred while querying for WMI data: " + e.Message);
+        }
+        return batteryCharge;
+    }
+
     public static bool IsUsedGPU(int threshold = 10)
     {
         if (GetGpuUse() > threshold)

app/USB/Aura.cs

@@ -53,6 +53,7 @@ public enum AuraMode : int
         HEATMAP = 20,
         GPUMODE = 21,
         AMBIENT = 22,
+        BATTERY = 23,
     }
 
     public enum AuraSpeed : int
@@ -107,6 +108,7 @@ public static class Aura
             { AuraMode.HEATMAP, "Heatmap"},
             { AuraMode.GPUMODE, "GPU Mode" },
             { AuraMode.AMBIENT, "Ambient"},
+            { AuraMode.BATTERY, "Battery"},
         };
 
         private static Dictionary<AuraMode, string> _modesDynamicLighting = new Dictionary<AuraMode, string>
@@ -124,6 +126,7 @@ public static class Aura
             { AuraMode.AuraColorCycle, Properties.Strings.AuraColorCycle },
             { AuraMode.AuraRainbow, Properties.Strings.AuraRainbow },
             { AuraMode.AuraStrobe, Properties.Strings.AuraStrobe },
+            { AuraMode.BATTERY, "Battery"},
         };
 
         private static Dictionary<AuraMode, string> _modesStrix = new Dictionary<AuraMode, string>
@@ -142,6 +145,7 @@ public static class Aura
             { AuraMode.Flash, "Flash" },
             { AuraMode.HEATMAP, "Heatmap"},
             { AuraMode.AMBIENT, "Ambient"},
+            { AuraMode.BATTERY, "Battery"},
         };
 
         static Aura()
@@ -248,6 +252,10 @@ private static void Timer_Elapsed(object? sender, System.Timers.ElapsedEventArgs
             {
                 CustomRGB.ApplyHeatmap();
             }
+            else if (Mode == AuraMode.BATTERY)
+            {
+                CustomRGB.ApplyBattery();
+            }
             else if (Mode == AuraMode.AMBIENT)
             {
                 CustomRGB.ApplyAmbient();
@@ -697,6 +705,14 @@ public static void ApplyAura(double colorDim = 1)
                 return;
             }
 
+            if (Mode == AuraMode.BATTERY)
+            {
+                CustomRGB.ApplyBattery(true);
+                timer.Enabled = true;
+                timer.Interval = 30000;
+                return;
+            }
+
             if (Mode == AuraMode.AMBIENT)
             {
                 CustomRGB.ApplyAmbient(true);
@@ -770,6 +786,14 @@ public static class CustomRGB
             static Color colorWarm = ColorTranslator.FromHtml(AppConfig.GetString("color_warm", "#FFFF00"));
             static Color colorHot = ColorTranslator.FromHtml(AppConfig.GetString("color_hot", "#FF0000"));
 
+            static float battLow = 20f;
+            static float battMid = 60f;
+            static float battHigh = 100f;
+
+            static Color colorLow = Color.Red;
+            static Color colorMid = Color.Yellow;
+            static Color colorHigh = Color.Lime;
+
             static Color colorUltimate = ColorTranslator.FromHtml(AppConfig.GetString("color_ultimate", "#FF0000"));
             static Color colorStandard = ColorTranslator.FromHtml(AppConfig.GetString("color_standard", "#FFFF00"));
             static Color colorEco = ColorTranslator.FromHtml(AppConfig.GetString("color_eco", "#008000"));
@@ -812,7 +836,34 @@ public static void ApplyHeatmap(bool init = false)
                 ApplyDirect(color, init);
             }
 
+            public static void ApplyBattery(bool init = false)
+            {
+                float battery = (float)HardwareControl.GetBatteryChargePercentage();
+                Color color = colorLow;
+
+                if (battery < battLow)
+                {
+                    color = colorLow;
+                }
+                else if (battery < battMid)
+                {
+                    // Low → Mid
+                    float t = (battery - battLow) / (battMid - battLow);
+                    color = ColorUtils.GetWeightedAverage(colorLow, colorMid, t);
+                }
+                else if (battery < battHigh)
+                {
+                    // Mid → High
+                    float t = (battery - battMid) / (battHigh - battMid);
+                    color = ColorUtils.GetWeightedAverage(colorMid, colorHigh, t);
+                }
+                else
+                {
+                    color = colorHigh;
+                }
 
+                ApplyDirect(color, init);
+            }
 
             public static void ApplyAmbient(bool init = false)
             {