Modify applied cap drain based on ship's cap resistance

pyfa-org · Mar 8, 2016 · 627dac6 · 627dac6
1 parent 68a6a82
commit 627dac6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/eos/saveddata/fit.py b/eos/saveddata/fit.py
@@ -823,7 +823,8 @@ def calculateShieldRecharge(self, percent = PEAK_RECHARGE):
         return 10 / rechargeRate * sqrt(percent) * (1 - sqrt(percent)) * capacity
 
     def addDrain(self, cycleTime, capNeed, clipSize=0):
-        self.__extraDrains.append((cycleTime, capNeed, clipSize))
+        resistance = self.ship.getModifiedItemAttr("energyWarfareResistance") or 1
+        self.__extraDrains.append((cycleTime, capNeed * resistance, clipSize))
 
     def removeDrain(self, i):
         del self.__extraDrains[i]