From ab32553c2d5ab4e6ccaa9d1df765ba2062f07db3 Mon Sep 17 00:00:00 2001
From: Calclavia <admin@calclavia.com>
Date: Sun, 1 Dec 2013 15:14:17 +0800
Subject: [PATCH] Added TE support for universal wires

---
 .../induction/common/MultipartMI.java         |  1 +
 .../common/wire/PartUniversalConductor.java   | 59 ++++++++++++++++++-
 2 files changed, 57 insertions(+), 3 deletions(-)

diff --git a/common/mekanism/induction/common/MultipartMI.java b/common/mekanism/induction/common/MultipartMI.java
index 2493735e5..e6a1b1297 100644
--- a/common/mekanism/induction/common/MultipartMI.java
+++ b/common/mekanism/induction/common/MultipartMI.java
@@ -28,6 +28,7 @@ public class MultipartMI implements IPartFactory
 		MultiPartRegistry.registerParts(this, new String[] { "resonant_induction_wire" });
 		MultipartGenerator.registerPassThroughInterface("universalelectricity.core.block.IConductor");
 		MultipartGenerator.registerPassThroughInterface("buildcraft.api.power.IPowerReceptor");
+		MultipartGenerator.registerPassThroughInterface("cofh.api.energy.IEnergyHandler");
 		MultipartGenerator.registerPassThroughInterface("mekanism.induction.common.wire.IInsulatedMaterial");
 		MultipartGenerator.registerPassThroughInterface("mekanism.induction.common.wire.IBlockableConnection");
 		MultipartGenerator.registerTrait("ic2.api.energy.tile.IEnergySink", "mekanism.induction.common.wire.TEnergySink");
diff --git a/common/mekanism/induction/common/wire/PartUniversalConductor.java b/common/mekanism/induction/common/wire/PartUniversalConductor.java
index 33d480ad1..068bd6f97 100644
--- a/common/mekanism/induction/common/wire/PartUniversalConductor.java
+++ b/common/mekanism/induction/common/wire/PartUniversalConductor.java
@@ -20,8 +20,9 @@ import buildcraft.api.power.IPowerReceptor;
 import buildcraft.api.power.PowerHandler;
 import buildcraft.api.power.PowerHandler.PowerReceiver;
 import buildcraft.api.power.PowerHandler.Type;
+import cofh.api.energy.IEnergyHandler;
 
-public abstract class PartUniversalConductor extends PartConductor implements IEnergySink, IPowerReceptor
+public abstract class PartUniversalConductor extends PartConductor implements IEnergySink, IPowerReceptor, IEnergyHandler
 {
 
 	protected boolean isAddedToEnergyNet;
@@ -38,14 +39,19 @@ public abstract class PartUniversalConductor extends PartConductor implements IE
 	@Override
 	public boolean isValidAcceptor(TileEntity tile)
 	{
-		if (Compatibility.isIndustrialCraft2Loaded() && tile instanceof IEnergyTile)
+		if (tile instanceof IEnergyTile)
 		{
 			return true;
 		}
-		else if (Compatibility.isBuildcraftLoaded() && tile instanceof IPowerReceptor)
+		else if (tile instanceof IPowerReceptor)
 		{
 			return true;
 		}
+		else if (tile instanceof IEnergyHandler)
+		{
+			return true;
+		}
+
 		return super.isValidAcceptor(tile);
 	}
 
@@ -198,4 +204,51 @@ public abstract class PartUniversalConductor extends PartConductor implements IE
 	{
 		return world();
 	}
+
+	/**
+	 * Thermal Expansion Functions
+	 */
+
+	@Override
+	public int receiveEnergy(ForgeDirection from, int maxReceive, boolean simulate)
+	{
+		ElectricityPack pack = ElectricityPack.getFromWatts(maxReceive * Compatibility.TE_RATIO, 1);
+		float request = this.getMaxEnergyStored(from);
+
+		if (!simulate)
+		{
+			if (request > 0)
+			{
+				return (int) (maxReceive - (this.getNetwork().produce(pack, new Vector3(tile()).modifyPositionFromSide(from).getTileEntity(this.world())) * Compatibility.TO_TE_RATIO));
+			}
+
+			return 0;
+		}
+
+		return (int) Math.min(maxReceive, request * Compatibility.TO_TE_RATIO);
+	}
+
+	@Override
+	public int extractEnergy(ForgeDirection from, int maxExtract, boolean simulate)
+	{
+		return 0;
+	}
+
+	@Override
+	public boolean canInterface(ForgeDirection from)
+	{
+		return true;
+	}
+
+	@Override
+	public int getEnergyStored(ForgeDirection from)
+	{
+		return 0;
+	}
+
+	@Override
+	public int getMaxEnergyStored(ForgeDirection from)
+	{
+		return (int) (this.getNetwork().getRequest(new Vector3(tile()).modifyPositionFromSide(from).getTileEntity(this.world())).getWatts() * Compatibility.TO_TE_RATIO);
+	}
 }