From a2d05d04ec8b195056005e09f77ef83758169d39 Mon Sep 17 00:00:00 2001
From: Ben Spiers <ben.spiers22@gmail.com>
Date: Mon, 30 Sep 2013 02:52:42 +0100
Subject: [PATCH] Of course IC2 would be awkward. Add a prewritten java trait
 as I don't think it's possible to make auto-generated ones not crash.

---
 src/resonantinduction/MultipartRI.java        |  2 +-
 .../multipart/javatraits/TEnergySink.java     | 92 +++++++++++++++++++
 2 files changed, 93 insertions(+), 1 deletion(-)
 create mode 100644 src/resonantinduction/wire/multipart/javatraits/TEnergySink.java

diff --git a/src/resonantinduction/MultipartRI.java b/src/resonantinduction/MultipartRI.java
index 1282f527..389e79b7 100644
--- a/src/resonantinduction/MultipartRI.java
+++ b/src/resonantinduction/MultipartRI.java
@@ -21,10 +21,10 @@ public class MultipartRI implements IPartFactory
 	{
 		MultiPartRegistry.registerParts(this, new String[]{"resonant_induction_wire"});
 		MultipartGenerator.registerPassThroughInterface("universalelectricity.core.block.IConductor");
-		MultipartGenerator.registerPassThroughInterface("ic2.api.energy.tile.IEnergySink");
 		MultipartGenerator.registerPassThroughInterface("buildcraft.api.power.IPowerReceptor");
 		MultipartGenerator.registerPassThroughInterface("resonantinduction.wire.IInsulatedMaterial");
 		MultipartGenerator.registerPassThroughInterface("resonantinduction.wire.multipart.IBlockableConnection");
+		MultipartGenerator.registerTrait("ic2.api.energy.tile.IEnergySink", "resonantinduction.wire.multipart.javatraits.TEnergySink");
 	}
 
 }
diff --git a/src/resonantinduction/wire/multipart/javatraits/TEnergySink.java b/src/resonantinduction/wire/multipart/javatraits/TEnergySink.java
new file mode 100644
index 00000000..ba5c023d
--- /dev/null
+++ b/src/resonantinduction/wire/multipart/javatraits/TEnergySink.java
@@ -0,0 +1,92 @@
+package resonantinduction.wire.multipart.javatraits;
+
+import java.util.Set;
+import java.util.HashSet;
+
+import net.minecraft.tileentity.TileEntity;
+import net.minecraftforge.common.ForgeDirection;
+import ic2.api.energy.tile.IEnergySink;
+import codechicken.multipart.TMultiPart;
+import codechicken.multipart.TileMultipart;
+
+public class TEnergySink extends TileMultipart implements IEnergySink
+{
+	public Set<IEnergySink> ic2Sinks = new HashSet<IEnergySink>();
+
+	@Override
+    public void copyFrom(TileMultipart that)
+    {
+        super.copyFrom(that);
+        if(that instanceof TEnergySink)
+            ic2Sinks = ((TEnergySink)that).ic2Sinks;
+    }
+    
+    @Override
+    public void bindPart(TMultiPart part)
+    {
+        super.bindPart(part);
+        if(part instanceof IEnergySink)
+            ic2Sinks.add((IEnergySink)part);
+    }
+    
+    @Override
+    public void partRemoved(TMultiPart part, int p)
+    {
+        super.partRemoved(part, p);
+        if(part instanceof IEnergySink)
+            ic2Sinks.remove((IEnergySink)part);
+    }
+    
+    @Override
+    public void clearParts()
+    {
+        super.clearParts();
+        ic2Sinks.clear();
+    }
+    
+
+	@Override
+	public boolean acceptsEnergyFrom(TileEntity emitter, ForgeDirection direction)
+	{
+		for (IEnergySink sink: this.ic2Sinks)
+		{
+			if (sink.acceptsEnergyFrom(emitter, direction))
+				return true;
+		}
+		return false;
+	}
+
+	@Override
+	public double demandedEnergyUnits()
+	{
+		double demanded = 0;
+
+		for (IEnergySink sink: this.ic2Sinks)
+		{
+			demanded += sink.demandedEnergyUnits();
+		}
+		return demanded;
+	}
+
+	@Override
+	public double injectEnergyUnits(ForgeDirection directionFrom, double amount)
+	{
+		for (IEnergySink sink: this.ic2Sinks)
+		{
+			amount = sink.injectEnergyUnits(directionFrom, Math.min(amount, sink.demandedEnergyUnits()));
+		}
+		return amount;
+	}
+
+	@Override
+	public int getMaxSafeInput()
+	{
+		int safe = 0;
+		for (IEnergySink sink: this.ic2Sinks)
+		{
+			safe += sink.getMaxSafeInput();
+		}
+		return safe;
+	}
+
+}