add RAS toggle

FAForever · May 31, 2018 · 610b90c · 610b90c
1 parent 6945125
commit 610b90c
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 10 deletions.
diff --git a/units/INU0001/INU0001_script.lua b/units/INU0001/INU0001_script.lua
@@ -124,6 +124,11 @@ INU0001 = Class(ACUUnit) {
 
         self.Sync.Abilities = self:GetBlueprint().Abilities
         self:HasCapacitorAbility(false)
+
+        --TODO:set these to actual bp values like bp.Enhancements.ResourceAllocation.ProductionPerSecondEnergy
+        self.MassProduction = 30
+        self.EnergyProduction = 3000
+
     end,
 
     OnStopBeingBuilt = function(self, builder, layer)
@@ -514,17 +519,15 @@ INU0001 = Class(ACUUnit) {
         -- RESOURCE ALLOCATION
         -- ---------------------------------------------------------------------------------------
 
-        elseif enh =='ResourceAllocation' then
-
-            local bpEcon = self:GetBlueprint().Economy
-            self:SetProductionPerSecondEnergy(bp.ProductionPerSecondEnergy + bpEcon.ProductionPerSecondEnergy or 0)
-            self:SetProductionPerSecondMass(bp.ProductionPerSecondMass + bpEcon.ProductionPerSecondMass or 0)
+        elseif enh =='ResourceAllocation' then            
+            self:AddToggleCap('RULEUTC_ProductionToggle')
+            self:SetProductionPerSecondEnergy(self.EnergyProduction)
+
 
         elseif enh == 'ResourceAllocationRemove' then
-
-            local bpEcon = self:GetBlueprint().Economy
-            self:SetProductionPerSecondEnergy(bpEcon.ProductionPerSecondEnergy or 0)
-            self:SetProductionPerSecondMass(bpEcon.ProductionPerSecondMass or 0)
+            self:RemoveToggleCap('RULEUTC_ProductionToggle')
+            self:SetProductionPerSecondEnergy(20)
+            self:SetProductionPerSecondmass(1)
 
         -- ---------------------------------------------------------------------------------------
         -- RAPID REPAIR
@@ -766,6 +769,24 @@ INU0001 = Class(ACUUnit) {
     OnAmmoCountIncreased = function(self, amount)
         self:GetAIBrain():EnableSpecialAbility( 'NomadsAreaBombardment', true)
     end,
+
+    OnScriptBitSet = function(self, bit)
+        if bit == 4 then -- Production toggle
+            self:SetProductionPerSecondMass(self.MassProduction)
+            self:SetProductionPerSecondEnergy(20)
+        else
+            ACUUnit.OnScriptBitSet(self, bit)
+        end
+    end,
+
+    OnScriptBitClear = function(self, bit)
+        if bit == 4 then -- Production toggle
+            self:SetProductionPerSecondMass(1)
+            self:SetProductionPerSecondEnergy(self.EnergyProduction)
+        else
+            ACUUnit.OnScriptBitClear(self, bit)
+        end
+    end,
 
 }
 

diff --git a/units/INU0001/INU0001_unit.bp b/units/INU0001/INU0001_unit.bp
@@ -672,7 +672,7 @@ UnitBlueprint {
             BuildTime = 2800,
             Icon = 'isb',
             Name = '<LOC enhancements_0045>Resource Allocation System',
-            ProductionPerSecondEnergy = 1400,
+            ProductionPerSecondEnergy = 3000, --these stats don't change anything try edit script
             ProductionPerSecondMass = 20,
             ShowBones = {
                 'upgrade_back',