1 files changed, 103 insertions, 0 deletions
diff --git a/src/main/java/gregtech/api/util/ExoticEnergyInputHelper.java b/src/main/java/gregtech/api/util/ExoticEnergyInputHelper.java
new file mode 100644
index 0000000000..6640f03835
--- /dev/null
+++ b/src/main/java/gregtech/api/util/ExoticEnergyInputHelper.java
@@ -0,0 +1,103 @@
+package gregtech.api.util;
+
+import static gregtech.api.util.GTUtility.filterValidMTEs;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+
+import gregtech.api.interfaces.metatileentity.IMetaTileEntity;
+import gregtech.api.metatileentity.implementations.MTEHatch;
+import tectech.thing.metaTileEntity.hatch.MTEHatchEnergyMulti;
+import tectech.thing.metaTileEntity.hatch.MTEHatchEnergyTunnel;
+
+public class ExoticEnergyInputHelper {
+
+    /**
+     * The Valid Types of TecTech Hatch List.
+     */
+    private static final List<Class<? extends MTEHatch>> sExoticEnergyHatchType = new ArrayList<>();
+
+    static {
+        register(MTEHatchEnergyMulti.class);
+        register(MTEHatchEnergyTunnel.class);
+    }
+
+    public static void register(Class<? extends MTEHatch> clazz) {
+        if (!MTEHatch.class.isAssignableFrom(clazz))
+            throw new IllegalArgumentException(clazz.getName() + " is not a subclass of " + MTEHatch.class.getName());
+        sExoticEnergyHatchType.add(clazz);
+    }
+
+    public static boolean drainEnergy(long aEU, Collection<? extends MTEHatch> hatches) {
+        for (MTEHatch tHatch : hatches) {
+            long tDrain = Math.min(
+                tHatch.getBaseMetaTileEntity()
+                    .getStoredEU(),
+                aEU);
+            tHatch.getBaseMetaTileEntity()
+                .decreaseStoredEnergyUnits(tDrain, false);
+            aEU -= tDrain;
+        }
+        return aEU <= 0;
+    }
+
+    public static boolean isExoticEnergyInput(IMetaTileEntity aHatch) {
+        for (Class<?> clazz : sExoticEnergyHatchType) {
+            if (clazz.isInstance(aHatch)) return true;
+        }
+        return false;
+    }
+
+    public static long getTotalEuMulti(Collection<? extends MTEHatch> hatches) {
+        long rEU = 0L;
+        for (MTEHatch tHatch : filterValidMTEs(hatches)) {
+            rEU += tHatch.getBaseMetaTileEntity()
+                .getInputVoltage() * tHatch.maxWorkingAmperesIn();
+        }
+        return rEU;
+    }
+
+    public static long getMaxInputVoltageMulti(Collection<? extends MTEHatch> hatches) {
+        long rVoltage = 0;
+        for (MTEHatch tHatch : filterValidMTEs(hatches)) {
+            rVoltage += tHatch.getBaseMetaTileEntity()
+                .getInputVoltage();
+        }
+        return rVoltage;
+    }
+
+    public static long getAverageInputVoltageMulti(Collection<? extends MTEHatch> hatches) {
+        long rVoltage = 0;
+        for (MTEHatch tHatch : filterValidMTEs(hatches)) {
+            rVoltage += tHatch.getBaseMetaTileEntity()
+                .getInputVoltage();
+        }
+        if (hatches.isEmpty()) {
+            return 0;
+        }
+        return rVoltage / hatches.size();
+    }
+
+    public static long getMaxInputAmpsMulti(Collection<? extends MTEHatch> hatches) {
+        long rAmp = 0;
+        for (MTEHatch tHatch : filterValidMTEs(hatches)) {
+            rAmp += tHatch.getBaseMetaTileEntity()
+                .getInputAmperage();
+        }
+        return rAmp;
+    }
+
+    public static long getMaxWorkingInputAmpsMulti(Collection<? extends MTEHatch> hatches) {
+        long rAmp = 0;
+        for (MTEHatch tHatch : filterValidMTEs(hatches)) {
+            rAmp += tHatch.maxWorkingAmperesIn();
+        }
+        return rAmp;
+    }
+
+    public static List<Class<? extends MTEHatch>> getAllClasses() {
+        return Collections.unmodifiableList(sExoticEnergyHatchType);
+    }
+}