resonant-induction/minecraft/liquidmechanics/common/tileentity/TileEntityPipe.java

package liquidmechanics.common.tileentity;

import liquidmechanics.api.IReadOut;
import liquidmechanics.api.ITankOutputer;
import liquidmechanics.api.helpers.Liquid;
import liquidmechanics.api.helpers.MHelper;
import liquidmechanics.common.LiquidMechanics;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.network.INetworkManager;
import net.minecraft.network.packet.Packet;
import net.minecraft.network.packet.Packet250CustomPayload;
import net.minecraft.tileentity.TileEntity;
import net.minecraftforge.common.ForgeDirection;
import net.minecraftforge.liquids.ILiquidTank;
import net.minecraftforge.liquids.ITankContainer;
import net.minecraftforge.liquids.LiquidContainerRegistry;
import net.minecraftforge.liquids.LiquidStack;
import net.minecraftforge.liquids.LiquidTank;
import universalelectricity.core.vector.Vector3;
import universalelectricity.prefab.network.IPacketReceiver;
import universalelectricity.prefab.network.PacketManager;

import com.google.common.io.ByteArrayDataInput;

public class TileEntityPipe extends TileEntity implements ITankContainer, IPacketReceiver, IReadOut
{
    public Liquid type = Liquid.DEFUALT;
    private int count = 20;
    private int count2, presure = 0;

    protected boolean firstUpdate = true;

    public TileEntity[] connectedBlocks = { null, null, null, null, null, null };
    public LiquidTank stored = new LiquidTank(LiquidContainerRegistry.BUCKET_VOLUME * 3);

    @Override
    public void updateEntity()
    {
        if (++count >= 40)
        {
            count = 0;
            this.connectedBlocks = MHelper.getSourounding(worldObj, xCoord, yCoord, zCoord);
            for (int e = 0; e < 6; e++)
            {
                if (connectedBlocks[e] instanceof ITankContainer)
                {
                    if (connectedBlocks[e] instanceof TileEntityPipe && ((TileEntityPipe) connectedBlocks[e]).type != this.type)
                    {
                        connectedBlocks[e] = null;
                    }
                }
                else
                {
                    connectedBlocks[e] = null;
                }

            }

            if (!worldObj.isRemote)
            {
                this.updatePressure();
                if (count2-- <= 0)
                {
                    count2 = 5;
                    firstUpdate = false;
                    Packet packet = PacketManager.getPacket(LiquidMechanics.CHANNEL, this, this.type.ordinal());
                    PacketManager.sendPacketToClients(packet, worldObj, new Vector3(this), 60);
                }

                LiquidStack stack = stored.getLiquid();
                if (stack != null && stack.amount >= 0)
                {

                    for (int i = 0; i < 6; i++)
                    {
                        ForgeDirection dir = ForgeDirection.getOrientation(i);
                        int moved = 0;
                        if (connectedBlocks[i] instanceof ITankContainer)
                        {
                            if (connectedBlocks[i] instanceof TileEntityPipe)
                            {
                                if (((TileEntityPipe) connectedBlocks[i]).presure < this.presure)
                                {
                                    moved = ((TileEntityPipe) connectedBlocks[i]).stored.fill(stack, true);
                                }

                            }
                            else
                            {
                                moved = ((ITankContainer) connectedBlocks[i]).fill(dir.getOpposite(), stack, true);
                            }
                        }
                        stored.drain(moved, true);
                        // FMLLog.warning("Moved "+moved+ " "+ i);
                        if (stack.amount <= 0)
                        {
                            break;
                        }
                    }
                }
            }
        }

    }

    // returns liquid type
    public Liquid getType()
    {
        return this.type;
    }

    // used by the item to set the liquid type on spawn
    public void setType(Liquid rType)
    {
        this.type = rType;
    }

    // ---------------------
    // data
    // --------------------
    @Override
    public void handlePacketData(INetworkManager network, int packetType, Packet250CustomPayload packet, EntityPlayer player, ByteArrayDataInput data)
    {
        try
        {
            this.setType(Liquid.getLiquid(data.readInt()));
        }
        catch (Exception e)
        {
            e.printStackTrace();
        }

    }

    /**
     * Reads a tile entity from NBT.
     */
    @Override
    public void readFromNBT(NBTTagCompound par1NBTTagCompound)
    {
        super.readFromNBT(par1NBTTagCompound);
        this.type = Liquid.getLiquid(par1NBTTagCompound.getInteger("type"));
        int vol = par1NBTTagCompound.getInteger("liquid");
        this.stored.setLiquid(Liquid.getStack(type, vol));
    }

    /**
     * Writes a tile entity to NBT.
     */
    @Override
    public void writeToNBT(NBTTagCompound par1NBTTagCompound)
    {
        super.writeToNBT(par1NBTTagCompound);
        int s = 0;
        LiquidStack stack = this.stored.getLiquid();
        if (stack != null)
            s = stack.amount;
        par1NBTTagCompound.setInteger("liquid", s);
        par1NBTTagCompound.setInteger("type", this.type.ordinal());
    }

    @Override
    public String getMeterReading(EntityPlayer user, ForgeDirection side)
    {
        String output = "";
        LiquidStack stack = stored.getLiquid();
        if (stack != null)
            output += (stack.amount / LiquidContainerRegistry.BUCKET_VOLUME) + " " + this.type.displayerName;
        output += " @" + this.presure + "psi";
        if (stack != null)
            return output;

        return "Error";
    }

    @Override
    public int fill(ForgeDirection from, LiquidStack resource, boolean doFill)
    {
        LiquidStack stack = stored.getLiquid();
        if (stack == null) stored.setLiquid(Liquid.getStack(this.type, 1));
        if (stack != null && Liquid.isStackEqual(resource, this.type)) return fill(0, resource, doFill);

        return 0;
    }

    @Override
    public int fill(int tankIndex, LiquidStack resource, boolean doFill)
    {
        if (tankIndex != 0 || resource == null)
            return 0;

        return stored.fill(resource, doFill);
    }

    @Override
    public LiquidStack drain(ForgeDirection from, int maxDrain, boolean doDrain)
    {
        return drain(0, maxDrain, doDrain);
    }

    @Override
    public LiquidStack drain(int tankIndex, int maxDrain, boolean doDrain)
    {
        return stored.drain(maxDrain, doDrain);
    }

    @Override
    public ILiquidTank[] getTanks(ForgeDirection direction)
    {
        return new ILiquidTank[] { this.stored };
    }

    @Override
    public ILiquidTank getTank(ForgeDirection direction, LiquidStack type)
    {
        return null;
    }

    /**
     * Used to determan pipe connection rules
     */
    public boolean canConntect(TileEntity entity)
    {
        if (entity instanceof TileEntityPipe)
        {
            if (((TileEntityPipe) entity).type == this.type && this.type != Liquid.DEFUALT) { return true; }
        }
        return false;
    }

    /**
     * used to cause the pipes pressure to update depending on what is connected
     * to it
     * 
     * @return
     */
    public void updatePressure()
    {
        int highestPressure = 0;
        this.presure = 0;

        for (int i = 0; i < 6; i++)
        {
            ForgeDirection dir = ForgeDirection.getOrientation(i);

            if (connectedBlocks[i] instanceof TileEntityPipe && ((TileEntityPipe) connectedBlocks[i]).canConntect(this))
            {
                if (((TileEntityPipe) connectedBlocks[i]).getPressure() > highestPressure)
                {
                    highestPressure = ((TileEntityPipe) connectedBlocks[i]).getPressure();
                }
            }
            if (connectedBlocks[i] instanceof ITankOutputer && ((ITankOutputer) connectedBlocks[i]).canPressureToo(this.type, dir))
            {

                int p = ((ITankOutputer) connectedBlocks[i]).presureOutput(this.type, dir);
                if (p > highestPressure)
                    highestPressure = p;
            }
        }
        this.presure = highestPressure - 1;
    }

    public int getPressure()
    {
        return this.presure;
    }
}