Added getter for working amperes of a hatch (#1646)

* Added getter for working amperes of a hatch

* Replaced max input amp getter with call to working amp getter

1 files changed, 14 insertions, 0 deletions

diff --git a/src/main/java/gregtech/api/util/GT_ExoticEnergyInputHelper.java b/src/main/java/gregtech/api/util/GT_ExoticEnergyInputHelper.java
index 76cdb13f78..2e34fb3332 100644
--- a/src/main/java/gregtech/api/util/GT_ExoticEnergyInputHelper.java
+++ b/src/main/java/gregtech/api/util/GT_ExoticEnergyInputHelper.java
@@ -4,6 +4,7 @@ import static gregtech.api.metatileentity.implementations.GT_MetaTileEntity_Mult
 
 import gregtech.api.interfaces.metatileentity.IMetaTileEntity;
 import gregtech.api.metatileentity.implementations.GT_MetaTileEntity_Hatch;
+import gregtech.api.metatileentity.implementations.GT_MetaTileEntity_Hatch_Energy;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.Collections;
@@ -73,6 +74,19 @@ public class GT_ExoticEnergyInputHelper {
         return rAmp;
     }
 
+    public static long getMaxWorkingInputAmpsMulti(Collection<? extends GT_MetaTileEntity_Hatch> hatches) {
+        long rAmp = 0;
+        for (GT_MetaTileEntity_Hatch tHatch : hatches)
+            if (isValidMetaTileEntity(tHatch)) {
+                if (tHatch instanceof GT_MetaTileEntity_Hatch_Energy) {
+                    rAmp += ((GT_MetaTileEntity_Hatch_Energy) tHatch).maxWorkingAmperesIn();
+                } else {
+                    rAmp += tHatch.getBaseMetaTileEntity().getInputAmperage();
+                }
+            }
+        return rAmp;
+    }
+
     public static List<Class<? extends GT_MetaTileEntity_Hatch>> getAllClasses() {
         return Collections.unmodifiableList(sExoticEnergyHatchType);
     }