1 files changed, 104 insertions, 0 deletions
diff --git a/src/main/java/gregtech/api/graphs/GenerateNodeMapPower.java b/src/main/java/gregtech/api/graphs/GenerateNodeMapPower.java
new file mode 100644
index 0000000000..95f9aee32d
--- /dev/null
+++ b/src/main/java/gregtech/api/graphs/GenerateNodeMapPower.java
@@ -0,0 +1,104 @@
+package gregtech.api.graphs;
+
+import java.util.ArrayList;
+import java.util.HashSet;
+
+import net.minecraft.tileentity.TileEntity;
+import net.minecraftforge.common.util.ForgeDirection;
+
+import cofh.api.energy.IEnergyReceiver;
+import gregtech.api.GregTech_API;
+import gregtech.api.graphs.consumers.ConsumerNode;
+import gregtech.api.graphs.consumers.EmptyPowerConsumer;
+import gregtech.api.graphs.consumers.NodeEnergyConnected;
+import gregtech.api.graphs.consumers.NodeEnergyReceiver;
+import gregtech.api.graphs.consumers.NodeEnergySink;
+import gregtech.api.graphs.consumers.NodeGTBaseMetaTile;
+import gregtech.api.graphs.paths.NodePath;
+import gregtech.api.graphs.paths.PowerNodePath;
+import gregtech.api.interfaces.tileentity.IEnergyConnected;
+import gregtech.api.metatileentity.BaseMetaPipeEntity;
+import gregtech.api.metatileentity.BaseMetaTileEntity;
+import gregtech.api.metatileentity.MetaPipeEntity;
+import gregtech.api.metatileentity.implementations.GT_MetaPipeEntity_Cable;
+import ic2.api.energy.tile.IEnergySink;
+
+// node map generator for power distribution
+public class GenerateNodeMapPower extends GenerateNodeMap {
+
+    public GenerateNodeMapPower(BaseMetaPipeEntity aTileEntity) {
+        generateNode(aTileEntity, null, 1, null, ForgeDirection.UNKNOWN, new ArrayList<>(), new HashSet<>());
+    }
+
+    @Override
+    protected boolean isPipe(TileEntity aTileEntity) {
+        return super.isPipe(aTileEntity)
+            && ((BaseMetaPipeEntity) aTileEntity).getMetaTileEntity() instanceof GT_MetaPipeEntity_Cable;
+    }
+
+    @Override
+    protected boolean addConsumer(TileEntity aTileEntity, ForgeDirection side, int aNodeValue,
+        ArrayList<ConsumerNode> aConsumers) {
+        if (aTileEntity instanceof BaseMetaTileEntity tBaseTileEntity) {
+            if (tBaseTileEntity.inputEnergyFrom(side, false)) {
+                ConsumerNode tConsumerNode = new NodeGTBaseMetaTile(aNodeValue, tBaseTileEntity, side, aConsumers);
+                aConsumers.add(tConsumerNode);
+                return true;
+            }
+
+        } else if (aTileEntity instanceof IEnergyConnected tTileEntity) {
+            if (tTileEntity.inputEnergyFrom(side, false)) {
+                ConsumerNode tConsumerNode = new NodeEnergyConnected(aNodeValue, tTileEntity, side, aConsumers);
+                aConsumers.add(tConsumerNode);
+                return true;
+            }
+        } else if (aTileEntity instanceof IEnergySink sink) {
+            // ic2 wants the tilentity next to it of that side not going to add a bunch of arguments just for ic2
+            // crossborder checks to not load chuncks just to make sure
+            int dX = aTileEntity.xCoord + side.offsetX;
+            int dY = aTileEntity.yCoord + side.offsetY;
+            int dZ = aTileEntity.zCoord + side.offsetZ;
+            boolean crossesChuncks = dX >> 4 != aTileEntity.xCoord >> 4 || dZ >> 4 != aTileEntity.zCoord >> 4;
+            TileEntity tNextTo = null;
+            if (!crossesChuncks || !aTileEntity.getWorldObj()
+                .blockExists(dX, dY, dZ))
+                tNextTo = aTileEntity.getWorldObj()
+                    .getTileEntity(dX, dY, dZ);
+
+            if (sink.acceptsEnergyFrom(tNextTo, side)) {
+                ConsumerNode tConsumerNode = new NodeEnergySink(
+                    aNodeValue,
+                    (IEnergySink) aTileEntity,
+                    side,
+                    aConsumers);
+                aConsumers.add(tConsumerNode);
+                return true;
+            }
+        } else if (GregTech_API.mOutputRF && aTileEntity instanceof IEnergyReceiver receiver) {
+            ConsumerNode tConsumerNode = new NodeEnergyReceiver(aNodeValue, receiver, side, aConsumers);
+            aConsumers.add(tConsumerNode);
+            return true;
+        }
+        return false;
+    }
+
+    @Override
+    protected NodePath getNewPath(MetaPipeEntity[] aPipes) {
+        return new PowerNodePath(aPipes);
+    }
+
+    // used to apply voltage on dead ends
+    @Override
+    protected Node getEmptyNode(int aNodeValue, ForgeDirection side, TileEntity aTileEntity,
+        ArrayList<ConsumerNode> aConsumers) {
+        ConsumerNode tNode = new EmptyPowerConsumer(aNodeValue, aTileEntity, side, aConsumers);
+        aConsumers.add(tNode);
+        return tNode;
+    }
+
+    @Override
+    protected Node getPipeNode(int aNodeValue, ForgeDirection side, TileEntity aTileEntity,
+        ArrayList<ConsumerNode> aConsumers) {
+        return new PowerNode(aNodeValue, aTileEntity, aConsumers);
+    }
+}