Ticket #3265: [PATCH] 3265_-_Tribute_all_resources_to_active_allies_before_resigning.patch

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

@@ -5 +5 @@
         "<text/>" +
     "</element>";
 
+Player.prototype.resourceTypes = ["food", "wood", "metal", "stone"];
+
 Player.prototype.Init = function()
 {
     this.playerID = undefined;
@@ -15 +17 @@
     this.popBonuses = 0; // sum of population bonuses of player's entities
     this.maxPop = 300; // maximum population
     this.trainingBlocked = false; // indicates whether any training queue is currently blocked
-    this.resourceCount = {
-        "food": 300,
-        "wood": 300,
-        "metal": 300,
-        "stone": 300
-    };
+    this.resourceCount = [];
+    for (let resourceType of Player.prototype.resourceTypes)
+    {
+        this.resourceCount[resourceType] = 300;
+    }
     // goods for next trade-route and its proba in % (the sum of probas must be 100)
     this.tradingGoods = [
         { "goods":  "wood", "proba": 30 },
@@ -176 +177 @@
     this.trainingBlocked = false;
 };
 
+/**
+ * Set the quantities for each resource type
+ * @param resources - an array from resource type to quantity {@see Player.prototype.resourceTypes}
+ */
 Player.prototype.SetResourceCounts = function(resources)
 {
-    if (resources.food !== undefined)
-        this.resourceCount.food = resources.food;
-    if (resources.wood !== undefined)
-        this.resourceCount.wood = resources.wood;
-    if (resources.stone !== undefined)
-        this.resourceCount.stone = resources.stone;
-    if (resources.metal !== undefined)
-        this.resourceCount.metal = resources.metal;
+    for (let resourceType of Player.prototype.resourceTypes)
+    {
+        if (resources[resourceType] !== undefined)
+            this.resourceCount[resourceType] = resources[resourceType];
+    }
 };
 
 Player.prototype.GetResourceCounts = function()
@@ -340 +342 @@
     return this.state;
 };
 
+/**
+ * This function will check if the player is still active
+ * @returns {boolean} - true if player is active, false otherwise (resigned, quit, defeated)
+ */
+Player.prototype.IsActive = function()
+{
+    return this.state == "active";
+};
+
 Player.prototype.SetState = function(newState)
 {
     this.state = newState;
@@ -558 +569 @@
 };
 
 /**
+ * Get all the allies' ids of a given player.
+ * @returns {Array} - All the allies' ids, regardless of state.
+ */
+Player.prototype.GetAllies = function()
+{
+    var allies = [];
+    for (let i = 0; i < this.diplomacy.length; i++)
+        if (this.diplomacy[i] > 0 && i != this.playerID)
+            allies.push(i);
+    return allies;
+};
+
+/**
+ * Get all the active allies ids of a given player.
+ * @returns {Array} - All the active allies as actual player objects, not IDs
+ */
+Player.prototype.GetActiveAlliedPlayers = function()
+{
+    var allies = this.GetAllies();
+    var activeAllies = [];
+
+    var cmpPlayerManager = Engine.QueryInterface(SYSTEM_ENTITY, IID_PlayerManager);
+
+    for (var allyId of allies)
+    {
+        let cmpPlayer = Engine.QueryInterface(cmpPlayerManager.GetPlayerByID(allyId), IID_Player);
+        if (cmpPlayer.IsActive()) {
+            activeAllies.push(cmpPlayer);
+        }
+    }
+    return activeAllies;
+};
+
+/**
  * Check if given player is our enemy
  */
 Player.prototype.IsEnemy = function(id)
@@ -636 +681 @@
 {
     this.state = "defeated";
 
-    // TODO: Tribute all resources to this player's active allies (if any)
+    this.tributeResourcesToAllies();
 
     // Reassign all player's entities to Gaia
     var cmpRangeManager = Engine.QueryInterface(SYSTEM_ENTITY, IID_RangeManager);
@@ -663 +708 @@
     var notification = {"type": "defeat", "players": [this.playerID]};
     var cmpGUIInterface = Engine.QueryInterface(SYSTEM_ENTITY, IID_GuiInterface);
     cmpGUIInterface.PushNotification(notification);
+};
+
+/**
+ * When a player is defeated, his resources will be split among the still active allies.
+ */
+Player.prototype.tributeResourcesToAllies = function()
+{
+    var activeAllies = this.GetActiveAlliedPlayers();
+    var alliesNumber = activeAllies.length;
+
+    if (alliesNumber == 0 || this.team == -1) {
+        return;
+    }
+
+    var resourceForEach = [];
+
+    for (let resourceType of Player.prototype.resourceTypes)
+    {
+        resourceForEach[resourceType] = Math.floor(this.resourceCount[resourceType] / alliesNumber);
+    }
+    for (let ally of activeAllies)
+    {
+        ally.AddResources(resourceForEach);
+    }
+
+    this.SubtractResourcesOrNotify(this.resourceCount);
 };
 
 Player.prototype.OnResearchFinished = function(msg)