Applied size reduction effect to neut/nos

eos/effects/energyneutralizerentity.py

@@ -4,9 +4,26 @@
 # Drones from group: Energy Neutralizer Drone (3 of 3)
 from eos.types import State
 type = "active", "projected"
-def handler(fit, container, context):
+def handler(fit, module, context):
-    if "projected" in context and ((hasattr(container, "state") \
+    if "projected" in context and ((hasattr(module, "state") \
-    and container.state >= State.ACTIVE) or hasattr(container, "amountActive")):
+    and module.state >= State.ACTIVE) or hasattr(container, "amountActive")):
-        amount = container.getModifiedItemAttr("energyNeutralizerAmount")
+        amount = module.getModifiedItemAttr("energyNeutralizerAmount")
-        time = container.getModifiedItemAttr("duration")
+        time = module.getModifiedItemAttr("duration")
+        rigSize = fit.ship.getModifiedItemAttr("rigSize")
+        modifierLarge = module.getModifiedItemAttr("entityCapacitorLevelModifierLarge")
+        modifierMedium = module.getModifiedItemAttr("entityCapacitorLevelModifierMedium")
+        modifierSmall = module.getModifiedItemAttr("entityCapacitorLevelModifierSmall")
+
+        #Small rigged ships
+        if (rigSize == 1) and modifierSmall:
+            amount = amount*modifierSmall
+
+        #Medium rigged ships
+        if (rigSize == 2) and modifierMedium:
+            amount = amount*modifierMedium
+
+        #Large rigged ships
+        if (rigSize == 3) and modifierLarge:
+            amount = amount*modifierLarge
+
         fit.addDrain(time, amount, 0)

eos/effects/energyneutralizerfalloff.py

@@ -4,9 +4,27 @@
 # Modules from group: Energy Neutralizer (51 of 51)
 from eos.types import State
 type = "active", "projected"
-def handler(fit, container, context):
+def handler(fit, module, context):
-    if "projected" in context and ((hasattr(container, "state") \
+    if "projected" in context and ((hasattr(module, "state") \
-    and container.state >= State.ACTIVE) or hasattr(container, "amountActive")):
+    and module.state >= State.ACTIVE) or hasattr(container, "amountActive")):
-        amount = container.getModifiedItemAttr("energyNeutralizerAmount")
+        amount = module.getModifiedItemAttr("energyNeutralizerAmount")
-        time = container.getModifiedItemAttr("duration")
+        time = module.getModifiedItemAttr("duration")
+        rigSize = fit.ship.getModifiedItemAttr("rigSize")
+        modifierLarge = module.getModifiedItemAttr("entityCapacitorLevelModifierLarge")
+        modifierMedium = module.getModifiedItemAttr("entityCapacitorLevelModifierMedium")
+        modifierSmall = module.getModifiedItemAttr("entityCapacitorLevelModifierSmall")
+
+        #Small rigged ships
+        if (rigSize == 1) and modifierSmall:
+            amount = amount*modifierSmall
+
+        #Medium rigged ships
+        if (rigSize == 2) and modifierMedium:
+            amount = amount*modifierMedium
+
+        #Large rigged ships
+        if (rigSize == 3) and modifierLarge:
+            amount = amount*modifierLarge
+
         fit.addDrain(time, amount, 0)
+

eos/effects/energynosferatufalloff.py

@@ -7,7 +7,24 @@ runTime = "late"
 def handler(fit, module, context):
     amount = module.getModifiedItemAttr("powerTransferAmount")
     time = module.getModifiedItemAttr("duration")
+    rigSize = fit.ship.getModifiedItemAttr("rigSize")
+    modifierLarge = module.getModifiedItemAttr("entityCapacitorLevelModifierLarge")
+    modifierMedium = module.getModifiedItemAttr("entityCapacitorLevelModifierMedium")
+    modifierSmall = module.getModifiedItemAttr("entityCapacitorLevelModifierSmall")
+
     if "projected" in context:
+        #Small rigged ships
+        if (rigSize == 1) and modifierSmall:
+            amount = amount*modifierSmall
+
+        #Medium rigged ships
+        if (rigSize == 2) and modifierMedium:
+            amount = amount*modifierMedium
+
+        #Large rigged ships
+        if (rigSize == 3) and modifierLarge:
+            amount = amount*modifierLarge
+
         fit.addDrain(time, amount, 0)
     elif "module" in context:
         module.itemModifiedAttributes.force("capacitorNeed", -amount)

eos/effects/entityenergyneutralizerfalloff.py

@@ -4,9 +4,26 @@
 # Drones from group: Energy Neutralizer Drone (3 of 3)
 from eos.types import State
 type = "active", "projected"
-def handler(fit, container, context):
+def handler(fit, module, context):
-    if "projected" in context and ((hasattr(container, "state") \
+    if "projected" in context and ((hasattr(module, "state") \
-    and container.state >= State.ACTIVE) or hasattr(container, "amountActive")):
+    and module.state >= State.ACTIVE) or hasattr(module, "amountActive")):
-        amount = container.getModifiedItemAttr("energyNeutralizerAmount")
+        amount = module.getModifiedItemAttr("energyNeutralizerAmount")
-        time = container.getModifiedItemAttr("energyNeutralizerDuration")
+        time = module.getModifiedItemAttr("energyNeutralizerDuration")
+        rigSize = fit.ship.getModifiedItemAttr("rigSize")
+        modifierLarge = module.getModifiedItemAttr("entityCapacitorLevelModifierLarge")
+        modifierMedium = module.getModifiedItemAttr("entityCapacitorLevelModifierMedium")
+        modifierSmall = module.getModifiedItemAttr("entityCapacitorLevelModifierSmall")
+
+        #Small rigged ships
+        if (rigSize == 1) and modifierSmall:
+            amount = amount*modifierSmall
+
+        #Medium rigged ships
+        if (rigSize == 2) and modifierMedium:
+            amount = amount*modifierMedium
+
+        #Large rigged ships
+        if (rigSize == 3) and modifierLarge:
+            amount = amount*modifierLarge
+
         fit.addDrain(time, amount, 0)

eos/effects/fighterabilityenergyneutralizer.py

@@ -12,8 +12,25 @@ prefix = "fighterAbilityEnergyNeutralizer"
 
 type = "active", "projected"
 
-def handler(fit, container, context):
+def handler(fit, module, context):
     if "projected" in context:
-        amount = container.getModifiedItemAttr("{}Amount".format(prefix))
+        amount = module.getModifiedItemAttr("{}Amount".format(prefix))
-        time = container.getModifiedItemAttr("{}Duration".format(prefix))
+        time = module.getModifiedItemAttr("{}Duration".format(prefix))
+        rigSize = fit.ship.getModifiedItemAttr("rigSize")
+        modifierLarge = module.getModifiedItemAttr("entityCapacitorLevelModifierLarge")
+        modifierMedium = module.getModifiedItemAttr("entityCapacitorLevelModifierMedium")
+        modifierSmall = module.getModifiedItemAttr("entityCapacitorLevelModifierSmall")
+
+        # Small rigged ships
+        if (rigSize == 1) and modifierSmall:
+            amount = amount * modifierSmall
+
+        # Medium rigged ships
+        if (rigSize == 2) and modifierMedium:
+            amount = amount * modifierMedium
+
+        # Large rigged ships
+        if (rigSize == 3) and modifierLarge:
+            amount = amount * modifierLarge
+
         fit.addDrain(time, amount, 0)