1 files changed, 90 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..76a24e8802
--- /dev/null
+++ b/src/main/java/gregtech/api/graphs/GenerateNodeMapPower.java
@@ -0,0 +1,90 @@
+package gregtech.api.graphs;
+
+import cofh.api.energy.IEnergyReceiver;
+import gregtech.api.GregTech_API;
+import gregtech.api.graphs.consumers.*;
+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;
+import net.minecraft.tileentity.TileEntity;
+import net.minecraftforge.common.util.ForgeDirection;
+
+import java.util.ArrayList;
+import java.util.HashSet;
+
+//node map generator for power distribution
+public class GenerateNodeMapPower extends GenerateNodeMap {
+    public GenerateNodeMapPower(BaseMetaPipeEntity aTileEntity) {
+        generateNode(aTileEntity,null,1,null,
+                -1,new ArrayList<>(),new HashSet<>());
+    }
+
+    @Override
+    protected boolean isPipe(TileEntity aTileEntity) {
+        return super.isPipe(aTileEntity) && ((BaseMetaPipeEntity) aTileEntity).getMetaTileEntity() instanceof GT_MetaPipeEntity_Cable;
+    }
+
+    @Override
+    protected NodePath getNewPath(MetaPipeEntity[] aPipes) {
+        return new PowerNodePath(aPipes);
+    }
+
+    //used to apply voltage on death ends
+    @Override
+    protected Node getEmptyNode(int aNodeValue, byte aSide, TileEntity aTileEntity, ArrayList<ConsumerNode> aConsumers) {
+        Node tNode = new EmptyPowerConsumer(aNodeValue, aTileEntity, aSide, aConsumers);
+        aConsumers.add((ConsumerNode) tNode);
+        return tNode;
+    }
+
+    @Override
+    protected Node getPipeNode(int aNodeValue, byte aSide, TileEntity aTileEntity, ArrayList<ConsumerNode> aConsumers) {
+        return new PowerNode(aNodeValue, aTileEntity, aConsumers);
+    }
+
+    @Override
+    protected boolean addConsumer(TileEntity aTileEntity, byte aSide, int aNodeValue, ArrayList<ConsumerNode> aConsumers) {
+        if (aTileEntity instanceof BaseMetaTileEntity) {
+            BaseMetaTileEntity tBaseTileEntity = (BaseMetaTileEntity) aTileEntity;
+            if (tBaseTileEntity.inputEnergyFrom(aSide, false)) {
+                ConsumerNode tConsumerNode = new NodeGTBaseMetaTile(aNodeValue, tBaseTileEntity, aSide, aConsumers);
+                aConsumers.add(tConsumerNode);
+                return true;
+            }
+
+        } else if (aTileEntity instanceof IEnergyConnected) {
+            IEnergyConnected tTileEntity = (IEnergyConnected) aTileEntity;
+            if (tTileEntity.inputEnergyFrom(aSide,false)) {
+                ConsumerNode tConsumerNode = new NodeEnergyConnected(aNodeValue,tTileEntity,aSide,aConsumers);
+                aConsumers.add(tConsumerNode);
+                return true;
+            }
+        } else if (aTileEntity instanceof IEnergySink) {
+            //ic2 wants the tilentitty next to it of that side not going to add a bunch of arguments just for ic2
+            //crossborder checks to not not load chuncks just to make sure
+            int dX = aTileEntity.xCoord + ForgeDirection.getOrientation(aSide).offsetX;
+            int dY = aTileEntity.yCoord + ForgeDirection.getOrientation(aSide).offsetY;
+            int dZ = aTileEntity.zCoord + ForgeDirection.getOrientation(aSide).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 (((IEnergySink) aTileEntity).acceptsEnergyFrom(tNextTo, ForgeDirection.getOrientation(aSide))) {
+                ConsumerNode tConsumerNode = new NodeEnergySink(aNodeValue,(IEnergySink) aTileEntity, aSide, aConsumers);
+                aConsumers.add(tConsumerNode);
+                return true;
+            }
+        } else if (GregTech_API.mOutputRF && aTileEntity instanceof IEnergyReceiver) {
+            ConsumerNode tConsumerNode = new NodeEnergyReceiver(aNodeValue,(IEnergyReceiver) aTileEntity, aSide, aConsumers);
+            aConsumers.add(tConsumerNode);
+            return true;
+        }
+        return false;
+    }
+}