resonant-induction/archive/java/resonantinduction/electrical/wire/PartConductor.java

package resonantinduction.electrical.wire;

import ic2.api.energy.event.EnergyTileLoadEvent;
import ic2.api.energy.event.EnergyTileUnloadEvent;
import ic2.api.energy.tile.IEnergyTile;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.tileentity.TileEntity;
import net.minecraftforge.common.ForgeDirection;
import net.minecraftforge.common.MinecraftForge;
import resonantinduction.core.prefab.part.PartAdvanced;
import universalelectricity.api.CompatibilityModule;
import universalelectricity.api.UniversalClass;
import universalelectricity.api.energy.EnergyNetworkLoader;
import universalelectricity.api.energy.IConductor;
import universalelectricity.api.energy.IEnergyNetwork;
import universalelectricity.api.vector.Vector3;
import universalelectricity.api.vector.VectorHelper;
import codechicken.multipart.TMultiPart;

@UniversalClass
public abstract class PartConductor extends PartAdvanced implements IConductor
{
	private IEnergyNetwork network;

	protected Object[] connections = new Object[6];

	/**
	 * Universal Electricity conductor functions.
	 */
	@Override
	public long onReceiveEnergy(ForgeDirection from, long receive, boolean doReceive)
	{
		return this.getNetwork().produce(this, from.getOpposite(), receive, doReceive);
	}

	@Override
	public long onExtractEnergy(ForgeDirection from, long request, boolean doExtract)
	{
		return 0;
	}

	@Override
	public IEnergyNetwork getNetwork()
	{
		if (this.network == null)
		{
			this.setNetwork(EnergyNetworkLoader.getNewNetwork(this));
		}

		return this.network;
	}

	@Override
	public void setNetwork(IEnergyNetwork network)
	{
		this.network = network;
	}

	@Override
	public boolean doesTick()
	{
		return false;
	}

	@Override
	public Object[] getConnections()
	{
		return this.connections;
	}

	/**
	 * EXTERNAL USE Can this wire be connected by another block?
	 */
	@Override
	public boolean canConnect(ForgeDirection direction)
	{
		Vector3 connectPos = new Vector3(tile()).modifyPositionFromSide(direction);
		TileEntity connectTile = connectPos.getTileEntity(world());

		if (connectTile instanceof IConductor)
		{
			return false;
		}

		return CompatibilityModule.isHandler(connectTile);
	}

	public abstract boolean canConnectTo(Object obj);

	/**
	 * Recalculates all the network connections
	 */
	protected void recalculateConnections()
	{
		this.connections = new Object[6];
		/**
		 * Calculate all external connections with this conductor.
		 */
		for (byte i = 0; i < 6; i++)
		{
			ForgeDirection side = ForgeDirection.getOrientation(i);

			if (this.canConnect(side))
			{
				TileEntity tileEntity = VectorHelper.getTileEntityFromSide(world(), new Vector3(tile()), side);
				connections[i] = tileEntity;
			}
		}
	}

	/**
	 * IC2 Functions
	 */
	@Override
	public void onWorldJoin()
	{
		if (tile() instanceof IEnergyTile && !world().isRemote)
		{
			// Check if there's another part that's an IEnergyTile
			boolean foundAnotherPart = false;

			for (int i = 0; i < tile().partList().size(); i++)
			{
				TMultiPart part = tile().partMap(i);

				if (part instanceof IEnergyTile)
				{
					foundAnotherPart = true;
					break;
				}
			}

			if (!foundAnotherPart)
			{
				MinecraftForge.EVENT_BUS.post(new EnergyTileLoadEvent((IEnergyTile) tile()));
			}
		}
	}

	@Override
	public void preRemove()
	{
		if (!world().isRemote)
		{
			this.getNetwork().split(this);

			if (tile() instanceof IEnergyTile)
			{
				// Check if there's another part that's an IEnergyTile
				boolean foundAnotherPart = false;

				for (int i = 0; i < tile().partList().size(); i++)
				{
					TMultiPart part = tile().partMap(i);

					if (part instanceof IEnergyTile)
					{
						foundAnotherPart = true;
						break;
					}
				}

				if (!foundAnotherPart)
				{
					MinecraftForge.EVENT_BUS.post(new EnergyTileUnloadEvent((IEnergyTile) tile()));
				}
			}
		}

		super.preRemove();
	}

	@Override
	public void save(NBTTagCompound nbt)
	{
		super.save(nbt);
		nbt.setLong("savedBuffer", this.getNetwork().getBufferOf(this));
	}

	@Override
	public void load(NBTTagCompound nbt)
	{
		super.load(nbt);
		this.getNetwork().setBufferFor(this, nbt.getLong("savedBuffer"));
	}
}