Ticket #643: gatherer-count-2.patch

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

@@ -406 +406 @@
     if (!playerState)
         return;
 
-    getGUIObjectByName("resourceFood").caption = playerState.resourceCounts.food;
-    getGUIObjectByName("resourceWood").caption = playerState.resourceCounts.wood;
-    getGUIObjectByName("resourceStone").caption = playerState.resourceCounts.stone;
-    getGUIObjectByName("resourceMetal").caption = playerState.resourceCounts.metal;
+    getGUIObjectByName("resourceFood").caption = playerState.resourceCounts.food +
+            " (" + playerState.resourceGatherers.count.food + ")";
+    getGUIObjectByName("resourceWood").caption = playerState.resourceCounts.wood +
+            " (" + playerState.resourceGatherers.count.wood + ")";
+    getGUIObjectByName("resourceStone").caption = playerState.resourceCounts.stone +
+            " (" + playerState.resourceGatherers.count.stone + ")";
+    getGUIObjectByName("resourceMetal").caption = playerState.resourceCounts.metal +
+            " (" + playerState.resourceGatherers.count.metal + ")";
     getGUIObjectByName("resourcePop").caption = playerState.popCount + "/" + playerState.popLimit;
 
     g_IsTrainingBlocked = playerState.trainingBlocked;

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

@@ -310 +310 @@
             <!-- Player resource bar -->
             <!-- ================================  ================================ -->
             <object
-                size="10 0 45% 100%"
+                size="5 0 45%+5 100%"
             >
                 <!-- Food -->
-                <object size="0 0 90 100%" type="image" style="resourceCounter" tooltip="Food" tooltip_style="sessionToolTipBold">
+                <object size="0 0 100 100%" type="image" style="resourceCounter" tooltip="Food (Gatherers)" tooltip_style="sessionToolTipBold">
                     <object size="0 -4 40 36" type="image" sprite="stretched:session/icons/resources/food.png"/>
                     <object size="32 0 100% 100%-2" type="text" style="resourceText" name="resourceFood"/>
                 </object>
 
                 <!-- Wood -->
-                <object size="90 0 180 100%" type="image" style="resourceCounter" tooltip="Wood" tooltip_style="sessionToolTipBold">
+                <object size="90 0 190 100%" type="image" style="resourceCounter" tooltip="Wood (Gatherers)" tooltip_style="sessionToolTipBold">
                     <object size="0 -4 40 36" type="image" sprite="stretched:session/icons/resources/wood.png"/>
                     <object size="32 0 100% 100%-2" type="text" style="resourceText" name="resourceWood"/>
                 </object>
 
                 <!-- Stone -->
-                <object size="180 0 270 100%" type="image" style="resourceCounter" tooltip="Stone" tooltip_style="sessionToolTipBold">
+                <object size="180 0 280 100%" type="image" style="resourceCounter" tooltip="Stone (Gatherers)" tooltip_style="sessionToolTipBold">
                     <object size="0 -4 40 36" type="image" sprite="stretched:session/icons/resources/stone.png"/>
                     <object size="32 0 100% 100%-2" type="text" style="resourceText" name="resourceStone"/>
                 </object>
 
                 <!-- Metal -->
-                <object size="270 0 360 100%"  type="image" style="resourceCounter" tooltip="Metal" tooltip_style="sessionToolTipBold">
+                <object size="270 0 370 100%" type="image" style="resourceCounter" tooltip="Metal (Gatherers)" tooltip_style="sessionToolTipBold">
                     <object size="0 -4 40 36" type="image" sprite="stretched:session/icons/resources/metal.png"/>
                     <object size="32 0 100% 100%-2" type="text" style="resourceText" name="resourceMetal"/>
                 </object>
 
                 <!-- Population -->
-                <object size="360 0 450 100%" type="image" style="resourceCounter" tooltip="Population (current / limit)" tooltip_style="sessionToolTipBold">
+                <object size="365 0 450 100%" type="image" style="resourceCounter" tooltip="Population (current / limit)" tooltip_style="sessionToolTipBold">
                     <object size="0 -4 40 34" type="image" sprite="stretched:session/icons/resources/population.png"/>
                     <object size="32 0 100% 100%-2" type="text" style="resourceText" name="resourcePop"/>
                 </object>

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

@@ -78 +78 @@
             "popLimit": cmpPlayer.GetPopulationLimit(),
             "popMax": cmpPlayer.GetMaxPopulation(),
             "resourceCounts": cmpPlayer.GetResourceCounts(),
+            "resourceGatherers": cmpPlayer.GetResourceGatherers(),
             "trainingBlocked": cmpPlayer.IsTrainingBlocked(),
             "state": cmpPlayer.GetState(),
             "team": cmpPlayer.GetTeam(),

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

@@ -19 +19 @@
         "metal": 300,   
         "stone": 300    
     };
+    this.resourceGatherers = {
+        "food": {},
+        "wood": {},
+        "metal": {},
+        "stone": {},
+        "count": {
+            "food": 0,
+            "wood": 0,
+            "metal": 0,
+            "stone": 0
+        }
+    };
 
     this.team = -1; // team number of the player, players on the same team will always have ally diplomatic status - also this is useful for team emblems, scoring, etc.
     this.state = "active"; // game state - one of "active", "defeated", "won"
@@ -137 +149 @@
     return this.resourceCount;
 };
 
+Player.prototype.RemoveResourceGatherer = function(gatherer, force)
+{
+    var cmpResourceGatherer = Engine.QueryInterface(gatherer, IID_ResourceGatherer);
+    if (!cmpResourceGatherer)
+        return;
+
+    var resource = cmpResourceGatherer.GetLastGathered();
+    if (!resource)
+        return;
+
+    var cmpUnitAI = Engine.QueryInterface(gatherer, IID_UnitAI);
+    var gatheredResource = undefined;
+    for (var j = 0; j < cmpUnitAI.orderQueue.length; j++ )
+    {
+        var order = cmpUnitAI.orderQueue[j];
+        if (order.type == "Gather" && !order.data.force)
+            gatheredResource = order.data.type.generic;
+    }
+
+    // Prune entities if they're no longer gathering or gathering a different resource.
+    if (!gatheredResource || gatheredResource != resource.generic || force)
+    {
+        cmpResourceGatherer.lastGathered = undefined;
+        var resourceType = (resource.generic == "treasure") ? resource.specific : resource.generic;
+        delete this.resourceGatherers[resourceType][gatherer];
+        this.resourceGatherers.count[resourceType]--;
+    }
+};
+
+Player.prototype.GetResourceGatherers = function()
+{
+    // Define if necessary, for older saves.
+    if (typeof this.resourceGatherers === "undefined")
+    {
+        this.resourceGatherers = {
+            "food": {},
+            "wood": {},
+            "metal": {},
+            "stone": {},
+            "count": {
+                "food": 0,
+                "wood": 0,
+                "metal": 0,
+                "stone": 0
+            }
+        };
+    }
+
+    return this.resourceGatherers;
+};
+
+Player.prototype.AddResourceGatherer = function(gatherer, type)
+{
+    // We want to add units that are gathering a new resource.
+    var cmpResourceGatherer = Engine.QueryInterface(gatherer, IID_ResourceGatherer);
+    if (cmpResourceGatherer)
+    {
+        var lastGathered = cmpResourceGatherer.GetLastGathered();
+        if (!lastGathered || lastGathered.generic != type.generic)
+        {
+            cmpResourceGatherer.lastGathered = type;
+            var resourceType = (type.generic == "treasure") ? type.specific : type.generic;
+            this.resourceGatherers[resourceType][gatherer] = true;
+            this.resourceGatherers.count[resourceType]++;
+        }
+    }
+};
+
 /**
  * Add resource of specified type to player
  * @param type Generic type of resource (string)

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

@@ -65 +65 @@
 
     // The last exact type gathered, so we can render appropriate props
     this.lastCarriedType = undefined; // { generic, specific }
+    this.lastGathered = undefined;
 };
 
 /**
@@ -125 +126 @@
         return undefined;
 };
 
+ResourceGatherer.prototype.GetLastGathered = function()
+{
+    if ( this.lastGathered )
+        return this.lastGathered;
+    else
+        return undefined;
+}
+
 ResourceGatherer.prototype.GetGatherRates = function()
 {
     var ret = {};

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

@@ -1022 +1022 @@
                         return;
                     }
 
+                    var cmpPlayer = this.GetOwner(this.entity);
+                    if (cmpPlayer)
+                        cmpPlayer.AddResourceGatherer(this.entity, this.order.data.type);
+
                     // Scale timing interval based on rate, and start timer
                     // The offset should be at least as long as the repeat time so we use the same value for both.
                     var offset = 1000/rate;
@@ -1041 +1045 @@
                 },
 
                 "leave": function() {
+                    var cmpPlayer = this.GetOwner(this.entity);
+                    if (cmpPlayer)
+                        cmpPlayer.RemoveResourceGatherer(this.entity, false);
+
                     this.StopTimer();
                 },
 
@@ -1812 +1820 @@
     this.SetupRangeQuery();
     if (this.IsHealer())
         this.SetupHealRangeQuery();
+
+    // Ensure the entity is no longer counted as gathering.
+    if (msg.from != -1)
+    {
+        var cmpPlayerMan = Engine.QueryInterface(SYSTEM_ENTITY, IID_PlayerManager);
+        var cmpPlayer = Engine.QueryInterface(cmpPlayerMan.GetPlayerByID(msg.from), IID_Player);
+        if (cmpPlayer)
+            cmpPlayer.RemoveResourceGatherer(this.entity, true);
+    }
 };
 
 UnitAI.prototype.OnDestroy = function()
@@ -1927 +1944 @@
     Engine.PostMessage(this.entity, MT_UnitAIStateChanged, { "to": state });
 };
 
+UnitAI.prototype.GetOwner = function(entity)
+{
+    var cmpOwnership = Engine.QueryInterface(entity, IID_Ownership);
+    if (!cmpOwnership)
+        return undefined;
+
+    var owner = cmpOwnership.GetOwner();
+    if (owner == -1)
+        return undefined;
+
+    var cmpPlayerMan = Engine.QueryInterface(SYSTEM_ENTITY, IID_PlayerManager);
+    var cmpPlayer = Engine.QueryInterface(cmpPlayerMan.GetPlayerByID(owner), IID_Player);
+    if (!cmpPlayer)
+        return undefined;
+
+    return cmpPlayer;
+};
+
 /**
  * Call when the current order has been completed (or failed).
  * Removes the current order from the queue, and processes the