Ticket #1357: farm_supply_tech.patch

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

@@ -33 +33 @@
 
 ResourceSupply.prototype.Init = function()
 {
-    // Current resource amount (non-negative; can be a fractional amount)
-    this.amount = this.GetMaxAmount();
+    this.maxAmount = this.GetMaxAmount();
+    this.amount = this.maxAmount;
 };
 
 ResourceSupply.prototype.GetKillBeforeGather = function()
@@ -42 +42 @@
     return (this.template.KillBeforeGather == "true");
 };
 
+ResourceSupply.prototype.GetDefaultAmount = function()
+{
+    return +this.template.Amount;
+}
+
+ResourceSupply.prototype.GetUpgradedAmount = function()
+{
+    var cmpOwnership = Engine.QueryInterface(this.entity, IID_Ownership);
+    if (!cmpOwnership || cmpOwnership.GetOwner() === -1)
+        warn('No ownership or player is -1');
+        return this.GetDefaultAmount();
+
+    var cmpTechMan = QueryOwnerInterface(this.entity, IID_TechnologyManager);
+    return cmpTechMan.ApplyModifications("ResourceSupply/Amount", this.GetDefaultAmount(), this.entity);
+}
+
 ResourceSupply.prototype.GetMaxAmount = function()
 {
-    return +this.template.Amount;
+    return this.GetUpgradedAmount() || this.GetDefaultAmount();
 };
 
 ResourceSupply.prototype.GetCurrentAmount = function()
@@ -82 +98 @@
     return { "generic": type, "specific": subtype };
 };
 
+ResourceSupply.prototype.OnTechnologyModificationChange = function(msg)
+{
+    var oldAmount = this.maxAmount;
+    this.maxAmount = this.GetMaxAmount();
+
+    var difference = this.maxAmount - oldAmount;
+    this.amount += difference;
+}
+
 Engine.RegisterComponentType(IID_ResourceSupply, "ResourceSupply", ResourceSupply);