Ticket #744: free_worker-28-02-2011.diff

binaries/data/mods/public/gui/session/session.xml

@@ -15 +15 @@
     <script file="gui/session/unit_commands.js"/>
     <script file="gui/session/messages.js"/>
     <script file="gui/session/utility_functions.js"/>
+    <script file="gui/session/freeworker.js"/>
 
     <object name="sn" hotkey="session.gui.toggle">
         <action on="Tick">
@@ -387 +388 @@
     </object> <!-- END OF TOP PANEL -->
 
     <!-- ================================  ================================ -->
+    <!-- Free Worker Button -->
+    <!-- ================================  ================================ -->
+    <object name="freeWorkerPanel"
+        size="50%-544 100%-32 50%-512 100%"
+        type="image"
+        sprite="mapPanel"
+        hidden="false"
+        ghost="false"
+    >
+        <object type="button"
+            name="freeWorkerButton"
+            size="0 0 100% 100%"
+            tooltip_style="sessionToolTipBold"
+            tooltip="Find idle worker"
+        >
+            <object size="0 0 100% 100%" type="image" sprite="freeWorker" name="freeWorkerButtonImage" ghost="true" />
+            <action on="Press">findFreeWorker();</action>
+        </object>
+    </object>
+
+    <!-- ================================  ================================ -->
     <!-- START of BOTTOM PANEL -->
     <!-- ================================  ================================ -->
     <object size="50%-512 100%-180 50%+512 100%">

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

@@ +1 @@
+function findFreeWorker()
+{
+    var ent, pos;
+
+    [ent, pos] = Engine.GuiInterfaceCall("FindFreeWorker");
+
+    if (ent)
+    {
+        g_Selection.reset()
+        g_Selection.addList([ent]);
+        Engine.SetCameraTarget(pos.x, pos.y, pos.z);
+    }
+}
+ No newline at end of file

binaries/data/mods/public/gui/session/sprites.xml

@@ -9 +9 @@
         />
     </sprite>
 
+    <sprite name="freeWorker">
+        <image
+            texture="session/icons/single/repair.png"
+            cell_size="32 32"
+            size="0 0 100% 100%"
+        />
+    </sprite>
+
     <!-- ================================  ================================ -->
     <!-- Unit Command Icons -->
     <!-- ================================  ================================ -->

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

@@ -473 +473 @@
     cmpRangeManager.SetDebugOverlay(enabled);
 };
 
+var _prevWorker = -1;
+
+function isFreeWorker(ent)
+{
+    var cmpUnitAI = Engine.QueryInterface(ent, IID_UnitAI);
+    var cmpIdentity = Engine.QueryInterface(ent, IID_Identity);
+
+    if (cmpIdentity && cmpUnitAI && cmpUnitAI.IsIdle())
+    {
+        var classes = cmpIdentity.GetClassesList();
+
+        if (classes.indexOf("Worker") != -1)
+        {
+            return true;
+        }
+    }
+    return false;
+}
+
+GuiInterface.prototype.FindFreeWorker = function(player)
+{
+    var rangeMan = Engine.QueryInterface(SYSTEM_ENTITY, IID_RangeManager);
+    var playerEntities = rangeMan.GetEntitiesByPlayer(player);
+
+    var worker = null;
+
+    for each (var ent in playerEntities)
+    {
+        if ((worker == null || ent > _prevWorker) && isFreeWorker(ent))
+        {
+            worker = ent;
+
+            if (ent > _prevWorker)
+            {
+                break;
+            }
+        }
+    }
+
+    if (worker == null)
+    {
+        return [null, null]
+    }
+
+    var cmpPosition = Engine.QueryInterface(worker, IID_Position);
+
+    if (cmpPosition)
+    {
+        _prevWorker = worker;
+        return [worker, cmpPosition.GetPosition()];
+    }
+    else
+    {
+        return [null, null];
+    }
+};
+
 // List the GuiInterface functions that can be safely called by GUI scripts.
 // (GUI scripts are non-deterministic and untrusted, so these functions must be
 // appropriately careful. They are called with a first argument "player", which is
@@ -495 +552 @@
     "SetPathfinderDebugOverlay": 1,
     "SetObstructionDebugOverlay": 1,
     "SetMotionDebugOverlay": 1,
-    "SetRangeDebugOverlay": 1
+    "SetRangeDebugOverlay": 1,
+
+    "FindFreeWorker": 1
 };
 
 GuiInterface.prototype.ScriptCall = function(player, name, args)