Applied-Energistics-2/src/main/java/appeng/parts/layers/LayerIEnergySink.java

/*
 * This file is part of Applied Energistics 2.
 * Copyright (c) 2013 - 2014, AlgorithmX2, All rights reserved.
 *
 * Applied Energistics 2 is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Applied Energistics 2 is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Applied Energistics 2.  If not, see <http://www.gnu.org/licenses/lgpl>.
 */

package appeng.parts.layers;

import net.minecraft.tileentity.TileEntity;
import net.minecraft.world.World;
import net.minecraftforge.common.MinecraftForge;
import net.minecraftforge.common.util.ForgeDirection;

import ic2.api.energy.tile.IEnergySink;
import ic2.api.energy.tile.IEnergyTile;

import appeng.api.parts.IPart;
import appeng.api.parts.IPartHost;
import appeng.api.parts.LayerBase;
import appeng.api.parts.LayerFlags;
import appeng.util.Platform;

public class LayerIEnergySink extends LayerBase implements IEnergySink
{

	private boolean isInIC2()
	{
		return this.getLayerFlags().contains( LayerFlags.IC2_ENET );
	}

	private TileEntity getEnergySinkTile()
	{
		IPartHost host = (IPartHost) this;
		return host.getTile();
	}

	private World getEnergySinkWorld()
	{
		if ( this.getEnergySinkTile() == null )
			return null;

		return this.getEnergySinkTile().getWorldObj();
	}

	private boolean isTileValid()
	{
		TileEntity te = this.getEnergySinkTile();

		if ( te == null )
			return false;

		return !te.isInvalid() && te.getWorldObj().blockExists( te.xCoord, te.yCoord, te.zCoord );
	}

	private void addToENet()
	{
		if ( this.getEnergySinkWorld() == null )
			return;

		// re-add
		this.removeFromENet();

		if ( !this.isInIC2() && Platform.isServer() && this.isTileValid() )
		{
			this.getLayerFlags().add( LayerFlags.IC2_ENET );
			MinecraftForge.EVENT_BUS.post( new ic2.api.energy.event.EnergyTileLoadEvent( (IEnergySink) this.getEnergySinkTile() ) );
		}
	}

	private void removeFromENet()
	{
		if ( this.getEnergySinkWorld() == null )
			return;

		if ( this.isInIC2() && Platform.isServer() )
		{
			this.getLayerFlags().remove( LayerFlags.IC2_ENET );
			MinecraftForge.EVENT_BUS.post( new ic2.api.energy.event.EnergyTileUnloadEvent( (IEnergySink) this.getEnergySinkTile() ) );
		}
	}

	private boolean interestedInIC2()
	{
		if ( !((IPartHost) this).isInWorld() )
			return false;

		int interested = 0;
		for (ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS)
		{
			IPart part = this.getPart( dir );
			if ( part instanceof IEnergyTile )
			{
				interested++;
			}
		}
		return interested == 1;// if more then one tile is interested we need to abandon...
	}

	@Override
	public void partChanged()
	{
		super.partChanged();

		if ( this.interestedInIC2() )
			this.addToENet();
		else
			this.removeFromENet();
	}

	@Override
	public boolean acceptsEnergyFrom(TileEntity emitter, ForgeDirection direction)
	{
		if ( !this.isInIC2() )
			return false;

		IPart part = this.getPart( direction );
		if ( part instanceof IEnergySink )
			return ((IEnergySink) part).acceptsEnergyFrom( emitter, direction );
		return false;
	}

	@Override
	public double getDemandedEnergy()
	{
		if ( !this.isInIC2() )
			return 0;

		// this is a flawed implementation, that requires a change to the IC2 API.

		for (ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS)
		{
			IPart part = this.getPart( dir );
			if ( part instanceof IEnergySink )
			{
				// use lower number cause ic2 deletes power it sends that isn't received.
				return ((IEnergySink) part).getDemandedEnergy();
			}
		}

		return 0;
	}

	@Override
	public double injectEnergy(ForgeDirection directionFrom, double amount, double voltage)
	{
		if ( !this.isInIC2() )
			return amount;

		for (ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS)
		{
			IPart part = this.getPart( dir );
			if ( part instanceof IEnergySink )
			{
				return ((IEnergySink) part).injectEnergy( directionFrom, amount, voltage );
			}
		}

		return amount;
	}

	@Override
	public int getSinkTier()
	{
		return Integer.MAX_VALUE; // no real options here...
	}

}