Ticket #2407: refactor_selection_tweaked.patch

binaries/data/config/default.cfg

@@ -225 +225 @@
 [hotkey.selection]
 add = Shift                  ; Add units to selection
 milonly = Alt                ; Add only military units to selection
+idleonly = "I"               ; Select only idle units
 remove = Ctrl                ; Remove units from selection
 cancel = Esc                 ; Un-select all units and cancel building placement 
 idleworker = Period          ; Select next idle worker

binaries/data/mods/public/gui/session/input.js

@@ -437 +437 @@
     return true;
 }
 
-// Limits bandboxed selections to certain types of entities based on priority
-function getPreferredEntities(ents)
+// Updates the bandbox object with new positions and visibility.
+// The coordinates [x0, y0, x1, y1] are returned for further use.
+function updateBandbox(bandbox, ev, hidden)
 {
-    var entStateList = [];
-    var preferredEnts = [];
+    var x0 = dragStart[0];
+    var y0 = dragStart[1];
+    var x1 = ev.x;
+    var y1 = ev.y;
+    // normalize the orientation of the rectangle
+    if (x0 > x1) { let t = x0; x0 = x1; x1 = t; }
+    if (y0 > y1) { let t = y0; y0 = y1; y1 = t; }
 
-    // Check if there are units in the selection and get a list of entity states
-    for each (var ent in ents)
-    {
-        var entState = GetEntityState(ent);
+    bandbox.size = [x0, y0, x1, y1].join(" ");
+    bandbox.hidden = hidden;
+    
+    return [x0, y0, x1, y1];
+}
+
+// Define some useful unit filters for getPreferredEntities
+var unitFilters = {
+    "isUnit": function (entity) {
+        var entState = GetEntityState(entity);
         if (!entState)
-            continue;
-        if (hasClass(entState, "Unit"))
-            preferredEnts.push(ent);
-
-        entStateList.push(entState);
+            return false;
+        return hasClass(entState, "Unit");
+    },
+    "isDefensive": function (entity) {
+        var entState = GetEntityState(entity);
+        if (!entState)
+            return false;
+        return hasClass(entState, "Defensive");
+    },
+    "isNotSupport": function (entity) {
+        var entState = GetEntityState(entity);
+        if (!entState)
+            return false;
+        return hasClass(entState, "Unit") && !hasClass(entState, "Support");
+    },
+    "isIdle": function (entity) {
+        var entState = GetEntityState(entity);
+        if (!entState)
+            return false;
+        return hasClass(entState, "Unit") && entState.unitAI.isIdle;
+    },
+    "everything": function (entity) {
+    return true;
     }
+};
 
-    // If there are no units, check if there are defensive entities in the selection
-    if (!preferredEnts.length)
-        for (var i = 0; i < ents.length; i++)
-            if (hasClass(entStateList[i], "Defensive"))
-                preferredEnts.push(ents[i]);
-
-    return preferredEnts;
-}
-
-// Removes any support units from the passed list of entities
-function getMilitaryEntities(ents)
+// Selection of preferred entities to select.
+// We may use several entity filters, until one returns at least one element.
+function getPreferredEntities(ents)
 {
-    var militaryEnts = [];
-    for each (var ent in ents)
+    // Default filters
+    var filters = [unitFilters.isUnit, unitFilters.isDefensive, unitFilters.everything];
+    
+    // Handle hotkeys
+    if (Engine.HotkeyIsPressed("selection.milonly"))
+        filters = [unitFilters.isNotSupport];
+    if (Engine.HotkeyIsPressed("selection.idleonly"))
+        filters = [unitFilters.isIdle];
+    
+    var preferredEnts = [];
+    for (var i = 0; i < filters.length; ++i)
     {
-        var entState = GetEntityState(ent);
-        if (!hasClass(entState, "Support"))
-            militaryEnts.push(ent);
+        preferredEnts = ents.filter(filters[i]);
+        if (preferredEnts.length > 0) 
+            break;
     }
-    return militaryEnts;
+    return preferredEnts;
 }
 
 function handleInputBeforeGui(ev, hoveredObject)
@@ -511 +543 @@
     switch (inputState)
     {
     case INPUT_BANDBOXING:
+        var bandbox = Engine.GetGUIObjectByName("bandbox");
         switch (ev.type)
         {
         case "mousemotion":
-            var x0 = dragStart[0];
-            var y0 = dragStart[1];
-            var x1 = ev.x;
-            var y1 = ev.y;
-            if (x0 > x1) { var t = x0; x0 = x1; x1 = t; }
-            if (y0 > y1) { var t = y0; y0 = y1; y1 = t; }
+            var rect = updateBandbox(bandbox, ev, false);
 
-            var bandbox = Engine.GetGUIObjectByName("bandbox");
-            bandbox.size = [x0, y0, x1, y1].join(" ");
-            bandbox.hidden = false;
+            var ents = Engine.PickFriendlyEntitiesInRect(rect[0], rect[1], rect[2], rect[3], Engine.GetPlayerID());
+            var preferredEntities = getPreferredEntities(ents);
+            g_Selection.setHighlightList(preferredEntities);
 
-            // TODO: Should we handle "control all units" here as well?
-            var ents = Engine.PickFriendlyEntitiesInRect(x0, y0, x1, y1, Engine.GetPlayerID());
-            g_Selection.setHighlightList(ents);
-
             return false;
 
         case "mousebuttonup":
             if (ev.button == SDL_BUTTON_LEFT)
             {
-                var x0 = dragStart[0];
-                var y0 = dragStart[1];
-                var x1 = ev.x;
-                var y1 = ev.y;
-                if (x0 > x1) { var t = x0; x0 = x1; x1 = t; }
-                if (y0 > y1) { var t = y0; y0 = y1; y1 = t; }
+                var rect = updateBandbox(bandbox, ev, true);
 
-                var bandbox = Engine.GetGUIObjectByName("bandbox");
-                bandbox.hidden = true;
-
                 // Get list of entities limited to preferred entities
-                // TODO: Should we handle "control all units" here as well?
-                var ents = Engine.PickFriendlyEntitiesInRect(x0, y0, x1, y1, Engine.GetPlayerID());
-                var preferredEntities = getPreferredEntities(ents)
+                var ents = getPreferredEntities(Engine.PickFriendlyEntitiesInRect(rect[0], rect[1], rect[2], rect[3], Engine.GetPlayerID()));
 
-                if (preferredEntities.length)
-                {
-                    ents = preferredEntities;
-
-                    if (Engine.HotkeyIsPressed("selection.milonly"))
-                    {
-                        var militaryEntities = getMilitaryEntities(ents);
-                        if (militaryEntities.length)
-                            ents = militaryEntities;
-                    }
-                }
-
                 // Remove the bandbox hover highlighting
                 g_Selection.setHighlightList([]);
 
@@ -585 +587 @@
             else if (ev.button == SDL_BUTTON_RIGHT)
             {
                 // Cancel selection
-                var bandbox = Engine.GetGUIObjectByName("bandbox");
                 bandbox.hidden = true;
 
                 g_Selection.setHighlightList([]);

binaries/data/mods/public/simulation/components/GuiInterface.js

@@ -333 +333 @@
             "canGuard": cmpUnitAI.CanGuard(),
             "isGuarding": cmpUnitAI.IsGuardOf(),
             "possibleStances": cmpUnitAI.GetPossibleStances(),
+            "isIdle":cmpUnitAI.IsIdle(),
         };
         // Add some information needed for ungarrisoning
         if (cmpUnitAI.IsGarrisoned() && ret.player !== undefined)