add workermenu to be able to toggle unit builds

Author: ipax77

src/dsstats.maui/dsstats.builder/dsstats.builder/BuildArea.cs

@@ -55,24 +55,15 @@ public List<InputEvent> GetBuildEvents(ScreenArea screenArea, CmdrBuild build)
 
         foreach (var centerUnit in centerUnits)
         {
-            var unitChar = build.GetUnitChar(centerUnit.UnitName);
-            if (unitChar is null)
-            {
-                continue;
-            }
-            events.AddRange(DsBuilder.BuildUnit(unitChar.Value, centerUnit.Pos.X, centerUnit.Pos.Y));
+            events.AddRange(build.GetBuildEvents(centerUnit.UnitName, centerUnit.Pos, screenArea));
         }
         if (topUnits.Count > 0)
         {
             events.AddRange(DsBuilder.ScrollY(Convert.ToInt32(250 * screenArea._scaleY), screenArea.GetCenter()));
             foreach (var topUnit in topUnits)
             {
-                var unitChar = build.GetUnitChar(topUnit.UnitName);
-                if (unitChar is null)
-                {
-                    continue;
-                }
-                events.AddRange(DsBuilder.BuildUnit(unitChar.Value, topUnit.Pos.X, topUnit.Pos.Y + Convert.ToInt32(125 * screenArea._scaleY)));
+                events.AddRange(build.GetBuildEvents(topUnit.UnitName,
+                    topUnit.Pos with { Y = topUnit.Pos.Y + Convert.ToInt32(125 * screenArea._scaleY) }, screenArea));
             }
         }
         if (bottomUnits.Count > 0)
@@ -83,12 +74,8 @@ public List<InputEvent> GetBuildEvents(ScreenArea screenArea, CmdrBuild build)
             events.AddRange(DsBuilder.ScrollY(Convert.ToInt32(-500 * screenArea._scaleY), screenArea.GetCenter()));
             foreach (var bottomUnit in bottomUnits)
             {
-                var unitChar = build.GetUnitChar(bottomUnit.UnitName);
-                if (unitChar is null)
-                {
-                    continue;
-                }
-                events.AddRange(DsBuilder.BuildUnit(unitChar.Value, bottomUnit.Pos.X, bottomUnit.Pos.Y - Convert.ToInt32(300 * screenArea._scaleY)));
+                events.AddRange(build.GetBuildEvents(bottomUnit.UnitName,
+                    bottomUnit.Pos with { Y = bottomUnit.Pos.Y - Convert.ToInt32(300 * screenArea._scaleY) }, screenArea));
             }
         }
 

src/dsstats.maui/dsstats.builder/dsstats.builder/BuildPlayer.cs

@@ -85,6 +85,12 @@ public static void ReplayInput(List<InputEvent> events)
             {
                 User32Wrapper.SimulateRelativeMouseMove(e.X, e.Y);
             }
+            else if (e.Type == InputType.MouseRightClick)
+            {
+                User32Wrapper.SetCursorPos(e.X, e.Y);
+                User32Wrapper.mouse_event(User32Wrapper.MOUSEEVENTF_RIGHTDOWN, 0, 0, 0, UIntPtr.Zero);
+                User32Wrapper.mouse_event(User32Wrapper.MOUSEEVENTF_RIGHTUP, 0, 0, 0, UIntPtr.Zero);
+            }
         }
         if (shiftCurrentlyDown)
             User32Wrapper.keybd_event(User32Wrapper.VK_SHIFT, 0, User32Wrapper.KEYEVENTF_KEYUP, UIntPtr.Zero);

src/dsstats.maui/dsstats.builder/dsstats.builder/Builds/CmdrBuild.cs

@@ -4,28 +4,67 @@ namespace dsstats.builder;
 
 public abstract class CmdrBuild
 {
-    protected Dictionary<string, char> UnitMap = [];
-    protected Dictionary<string, char> UpgradeMap = [];
-    protected Dictionary<string, char> AbilityMap = [];
+    protected Dictionary<string, BuildOption> UnitMap = [];
+    protected Dictionary<string, BuildOption> UpgradeMap = [];
+    protected Dictionary<string, BuildOption> AbilityMap = [];
+    protected WorkerMenu WorkerMenu = new();
+    protected Dictionary<string, bool> activeUnits = [];
+
+    protected void CreateActiveUnits()
+    {
+        activeUnits = UnitMap.Where(x => x.Value.RequiresToggle).ToDictionary(k => k.Key, v => v.Value.IsActive);
+    }
+
+    public virtual List<InputEvent> GetBuildEvents(string unitName, RlPoint pos, ScreenArea screenArea)
+    {
+        if (!UnitMap.TryGetValue(unitName, out var buildOption)
+            || buildOption is null)
+        {
+            return [];
+        }
+        List<InputEvent> events = [];
+        if (buildOption.RequiresToggle)
+        {
+            if (activeUnits.TryGetValue(unitName, out var isActive))
+            {
+                if (!isActive)
+                {
+                    var toggleEvent = WorkerMenu.ToggleBuildMenu(buildOption.Key, screenArea);
+                    if (toggleEvent != null)
+                    {
+                        events.Add(toggleEvent);
+                        var otherToggleUnit = UnitMap.FirstOrDefault(f => f.Value.Key == buildOption.Key && f.Key != unitName);
+                        if (!string.IsNullOrEmpty(otherToggleUnit.Key))
+                        {
+                            activeUnits[otherToggleUnit.Key] = false;
+                        }
+                        activeUnits[unitName] = true;
+                    }
+                }
+            }
+        }
+        events.AddRange(DsBuilder.BuildUnit(buildOption.Key, pos.X, pos.Y));
+        return events;
+    }
 
     public virtual char? GetUnitChar(string unitName)
     {
-        return UnitMap.TryGetValue(unitName, out var c)
-            ? c
+        return UnitMap.TryGetValue(unitName, out var buildOption)
+            ? buildOption.Key
             : null;
     }
 
     public virtual char? GetUpgradeChar(string upgradeName)
     {
-        return UpgradeMap.TryGetValue(upgradeName, out var c)
-            ? c
+        return UpgradeMap.TryGetValue(upgradeName, out var buildOption)
+            ? buildOption.Key
             : null;
     }
 
     public virtual char? GetAbilityChar(string abilityName)
     {
-        return AbilityMap.TryGetValue(abilityName, out var c)
-            ? c
+        return AbilityMap.TryGetValue(abilityName, out var buildOption)
+            ? buildOption.Key
             : null;
     }
 }
@@ -42,4 +81,6 @@ public static class CmdrBuildFactory
             _ => null
         };
     }
-}
+}
+
+public sealed record BuildOption(char Key, bool RequiresToggle = false, bool IsActive = false);

src/dsstats.maui/dsstats.builder/dsstats.builder/Builds/ProtossBuild.cs

@@ -4,65 +4,66 @@ public class ProtossBuild : CmdrBuild
 {
     public ProtossBuild()
     {
-        UnitMap = new Dictionary<string, char>
+        UnitMap = new Dictionary<string, BuildOption>
         {
             // Q-menu units
-            { "Zealot", 'q' },
-            { "Stalker", 'w' },
-            { "Adept", 'e' },
-            { "Sentry", 'r' },
-            { "HighTemplar", 't' },
-            { "DarkTemplar", 'a' },
-            { "Immortal", 's' },
-            { "Colossus", 'd' },
-            { "Disruptor", 'f' },
-            { "Archon", 'g' },
-            { "Phoenix", 'z' },
-            { "VoidRay", 'x' },
-            { "Oracle", 'c' },
-            { "Tempest", 'v' },
-            { "Carrier", 'b' },
-            { "Mothership", 'n' },
+            { "Zealot", new('q') },
+            { "Stalker", new('w') },
+            { "Adept", new('e') },
+            { "Sentry", new('r') },
+            { "HighTemplar", new('t') },
+            { "DarkTemplar", new('a') },
+            { "Immortal", new('s') },
+            { "Colossus", new('d') },
+            { "Disruptor", new('f') },
+            { "Archon", new('g') },
+            { "Phoenix", new('z') },
+            { "VoidRay", new('x') },
+            { "Oracle", new('c') },
+            { "Tempest", new('v') },
+            { "Carrier", new('b') },
+            { "Mothership", new('n') },
         };
 
-        AbilityMap = new Dictionary<string, char>
+        AbilityMap = new Dictionary<string, BuildOption>
         {
             // W-menu: abilities and tech
-            { "Charge", 'q' },
-            { "Blink", 'w' },
-            { "ResonatingGlaives", 'e' },
-            { "GuardianShield", 'r' },
-            { "PsionicStorm", 't' },
-            { "ShadowStride", 'a' },
-            { "Barrier", 's' },
-            { "ExtendedThermalLance", 'd' },
-            { "PurificationNova", 'f' },
-            { "GravitonBeam", 'g' },
-            { "FluxVanes", 'z' },
-            { "Revelation", 'x' },
-            { "TectonicDestabilizers", 'c' },
-            { "InterceptorLaunchSpeed", 'v' },
-            { "MassRecall", 'b' },
+            { "Charge", new('q') },
+            { "Blink", new('w') },
+            { "ResonatingGlaives", new('e') },
+            { "GuardianShield", new('r') },
+            { "PsionicStorm", new('t') },
+            { "ShadowStride", new('a') },
+            { "Barrier", new('s') },
+            { "ExtendedThermalLance", new('d') },
+            { "PurificationNova", new('f') },
+            { "GravitonBeam", new('g') },
+            { "FluxVanes", new('z') },
+            { "Revelation", new('x') },
+            { "TectonicDestabilizers", new('c') },
+            { "InterceptorLaunchSpeed", new('v') },
+            { "MassRecall", new('b') },
         };
 
-        UpgradeMap = new Dictionary<string, char>
+        UpgradeMap = new Dictionary<string, BuildOption>
         {
             // Forge + Cyber Core + Fleet Beacon
-            { "GroundWeaponsLevel1", 'a' },
-            { "GroundWeaponsLevel2", 'a' },
-            { "GroundWeaponsLevel3", 'a' },
-            { "GroundArmorLevel1", 's' },
-            { "GroundArmorLevel2", 's' },
-            { "GroundArmorLevel3", 's' },
-            { "ShieldsLevel1", 'd' },
-            { "ShieldsLevel2", 'd' },
-            { "ShieldsLevel3", 'd' },
-            { "AirWeaponsLevel1", 'f' },
-            { "AirWeaponsLevel2", 'f' },
-            { "AirWeaponsLevel3", 'f' },
-            { "AirArmorLevel1", 'g' },
-            { "AirArmorLevel2", 'g' },
-            { "AirArmorLevel3", 'g' },
+            { "GroundWeaponsLevel1", new('a') },
+            { "GroundWeaponsLevel2", new('a') },
+            { "GroundWeaponsLevel3", new('a') },
+            { "GroundArmorLevel1", new('s') },
+            { "GroundArmorLevel2", new('s') },
+            { "GroundArmorLevel3", new('s') },
+            { "ShieldsLevel1", new('d') },
+            { "ShieldsLevel2", new('d') },
+            { "ShieldsLevel3", new('d') },
+            { "AirWeaponsLevel1", new('f') },
+            { "AirWeaponsLevel2", new('f') },
+            { "AirWeaponsLevel3", new('f') },
+            { "AirArmorLevel1", new('g') },
+            { "AirArmorLevel2", new('g') },
+            { "AirArmorLevel3", new('g') },
         };
+        CreateActiveUnits();
     }
 }

src/dsstats.maui/dsstats.builder/dsstats.builder/Builds/TerranBuild.cs

@@ -4,62 +4,63 @@ public class TerranBuild : CmdrBuild
 {
     public TerranBuild()
     {
-        UnitMap = new Dictionary<string, char>
+        UnitMap = new Dictionary<string, BuildOption>
         {
             // Q-menu
-            { "Marine", 'q' },
-            { "Marauder", 'w' },
-            { "Reaper", 'e' },
-            { "Ghost", 'r' },
-            { "Hellion", 't' },
-            { "Hellbat", 'a' },
-            { "SiegeTank", 's' },
-            { "WidowMine", 'd' },
-            { "Cyclone", 'f' },
-            { "Thor", 'g' },
-            { "Viking", 'z' },
-            { "Medivac", 'x' },
-            { "Banshee", 'c' },
-            { "Raven", 'v' },
-            { "Battlecruiser", 'b' },
+            { "Marine", new('q') },
+            { "Marauder", new('w') },
+            { "Reaper", new('e') },
+            { "Ghost", new('r') },
+            { "Hellion", new('t') },
+            { "Hellbat", new('a') },
+            { "SiegeTank", new('s') },
+            { "WidowMine", new('d') },
+            { "Cyclone", new('f') },
+            { "Thor", new('g') },
+            { "Viking", new('z') },
+            { "Medivac", new('x') },
+            { "Banshee", new('c') },
+            { "Raven", new('v') },
+            { "Battlecruiser", new('b') },
         };
 
-        AbilityMap = new Dictionary<string, char>
+        AbilityMap = new Dictionary<string, BuildOption>
         {
             // W-menu (upgrades & abilities)
-            { "Stimpack", 'q' },
-            { "ConcussiveShells", 'w' },
-            { "CloakingField", 'e' },
-            { "DrillingClaws", 'r' },
-            { "InfernalPreigniter", 't' }, // Hellion upgrade
-            { "SmartServos", 'a' },        // Viking/Thor transform speed
-            { "HighImpactPayload", 's' },  // Thor mode toggle
-            { "HyperflightRotors", 'd' },  // Banshee speed
-            { "AdvancedBallistics", 'f' }, // Raven range
-            { "WeaponRefit", 'g' },        // Battlecruiser Yamato
+            { "Stimpack", new('q') },
+            { "ConcussiveShells", new('w') },
+            { "CloakingField", new('e') },
+            { "DrillingClaws", new('r') },
+            { "InfernalPreigniter", new('t') }, // Hellion upgrade
+            { "SmartServos", new('a') },        // Viking/Thor transform speed
+            { "HighImpactPayload", new('s') },  // Thor mode toggle
+            { "HyperflightRotors", new('d') },  // Banshee speed
+            { "AdvancedBallistics", new('f') }, // Raven range
+            { "WeaponRefit", new('g') },        // Battlecruiser Yamato
         };
 
-        UpgradeMap = new Dictionary<string, char>
+        UpgradeMap = new Dictionary<string, BuildOption>
         {
             // Main armory upgrades
-            { "InfantryWeaponsLevel1", 'a' },
-            { "InfantryWeaponsLevel2", 'a' },
-            { "InfantryWeaponsLevel3", 'a' },
-            { "InfantryArmorLevel1", 's' },
-            { "InfantryArmorLevel2", 's' },
-            { "InfantryArmorLevel3", 's' },
-            { "VehicleWeaponsLevel1", 'd' },
-            { "VehicleWeaponsLevel2", 'd' },
-            { "VehicleWeaponsLevel3", 'd' },
-            { "ShipWeaponsLevel1", 'f' },
-            { "ShipWeaponsLevel2", 'f' },
-            { "ShipWeaponsLevel3", 'f' },
-            { "VehiclePlatingLevel1", 'g' },
-            { "VehiclePlatingLevel2", 'g' },
-            { "VehiclePlatingLevel3", 'g' },
-            { "ShipPlatingLevel1", 'h' },
-            { "ShipPlatingLevel2", 'h' },
-            { "ShipPlatingLevel3", 'h' },
+            { "InfantryWeaponsLevel1", new('a') },
+            { "InfantryWeaponsLevel2", new('a') },
+            { "InfantryWeaponsLevel3", new('a') },
+            { "InfantryArmorLevel1", new('s') },
+            { "InfantryArmorLevel2", new('s') },
+            { "InfantryArmorLevel3", new('s') },
+            { "VehicleWeaponsLevel1", new('d') },
+            { "VehicleWeaponsLevel2", new('d') },
+            { "VehicleWeaponsLevel3", new('d') },
+            { "ShipWeaponsLevel1", new('f') },
+            { "ShipWeaponsLevel2", new('f') },
+            { "ShipWeaponsLevel3", new('f') },
+            { "VehiclePlatingLevel1", new('g') },
+            { "VehiclePlatingLevel2", new('g') },
+            { "VehiclePlatingLevel3", new('g') },
+            { "ShipPlatingLevel1", new('h') },
+            { "ShipPlatingLevel2", new('h') },
+            { "ShipPlatingLevel3", new('h') },
         };
+        CreateActiveUnits();
     }
 }