1 files changed, 53 insertions, 0 deletions

diff --git a/src/main/java/gregtech/api/metatileentity/BaseTileEntity.java b/src/main/java/gregtech/api/metatileentity/BaseTileEntity.java
index 66d1ade42d..eb18ef0d0c 100644
--- a/src/main/java/gregtech/api/metatileentity/BaseTileEntity.java
+++ b/src/main/java/gregtech/api/metatileentity/BaseTileEntity.java
@@ -1,10 +1,14 @@
 package gregtech.api.metatileentity;
 
+import cpw.mods.fml.common.FMLCommonHandler;
+import gregtech.api.interfaces.metatileentity.IMetaTileEntity;
 import gregtech.api.interfaces.tileentity.IGregTechTileEntity;
 import gregtech.api.interfaces.tileentity.IHasWorldObjectAndCoords;
 import gregtech.api.net.GT_Packet_Block_Event;
 import gregtech.api.util.GT_LanguageManager;
 import gregtech.api.util.GT_Utility;
+import ic2.api.energy.event.EnergyTileLoadEvent;
+import ic2.api.energy.event.EnergyTileUnloadEvent;
 import net.minecraft.block.Block;
 import net.minecraft.entity.player.EntityPlayer;
 import net.minecraft.init.Blocks;
@@ -13,6 +17,7 @@ import net.minecraft.nbt.NBTTagCompound;
 import net.minecraft.tileentity.TileEntity;
 import net.minecraft.world.World;
 import net.minecraft.world.biome.BiomeGenBase;
+import net.minecraftforge.common.MinecraftForge;
 import net.minecraftforge.common.util.ForgeDirection;
 import net.minecraftforge.fluids.IFluidHandler;
 
@@ -398,12 +403,14 @@ public abstract class BaseTileEntity extends TileEntity implements IHasWorldObje
 
     @Override
     public void invalidate() {
+        leaveEnet();
         clearNullMarkersFromTileEntityBuffer();
         super.invalidate();
     }
 
     @Override
     public void onChunkUnload() {
+        leaveEnet();
         clearNullMarkersFromTileEntityBuffer();
         super.onChunkUnload();
         isDead = true;
@@ -442,4 +449,50 @@ public abstract class BaseTileEntity extends TileEntity implements IHasWorldObje
     public String trans(String aKey, String aEnglish){
     	return GT_LanguageManager.addStringLocalization("Interaction_DESCRIPTION_Index_"+aKey, aEnglish, false);
     }
+
+    /*
+     * IC2 Energy Compat
+     */
+    protected TileIC2EnergySink ic2EnergySink = null;
+    protected boolean joinedIc2Enet = false;
+
+    public IMetaTileEntity getMetaTileEntity() { return null; }
+
+    protected void createIc2Sink() {
+        if(ic2EnergySink == null && FMLCommonHandler.instance().getEffectiveSide().isServer() && shouldJoinIc2Enet()) {
+            ic2EnergySink = new TileIC2EnergySink((IGregTechTileEntity)this);
+            ic2EnergySink.xCoord = xCoord;
+            ic2EnergySink.yCoord = yCoord;
+            ic2EnergySink.zCoord = zCoord;
+            ic2EnergySink.setWorldObj(worldObj);
+        }
+    }
+
+    public void doEnetUpdate() {
+        leaveEnet();
+        joinEnet();
+    }
+
+    protected void joinEnet() {
+        if (joinedIc2Enet || !shouldJoinIc2Enet()) return;
+
+        if(ic2EnergySink == null) createIc2Sink();
+
+        if (ic2EnergySink != null) {
+            MinecraftForge.EVENT_BUS.post(new EnergyTileLoadEvent(ic2EnergySink));
+            joinedIc2Enet = true;
+        }
+    }
+
+    protected void leaveEnet() {
+        if (joinedIc2Enet && ic2EnergySink != null && FMLCommonHandler.instance().getEffectiveSide().isServer()) {
+            joinedIc2Enet = false;
+            MinecraftForge.EVENT_BUS.post(new EnergyTileUnloadEvent(ic2EnergySink));
+        }
+    }
+
+    public boolean shouldJoinIc2Enet() {
+        final IMetaTileEntity meta = getMetaTileEntity();
+        return meta != null && meta.shouldJoinIc2Enet();
+    }
 }
\ No newline at end of file