BloodMagic/src-backup/main/java/WayofTime/alchemicalWizardry/api/alchemy/energy/ReagentContainer.java

package WayofTime.alchemicalWizardry.api.alchemy.energy;

import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.tileentity.TileEntity;
import net.minecraftforge.fluids.Fluid;
import net.minecraftforge.fluids.FluidStack;
import net.minecraftforge.fluids.FluidTank;

public class ReagentContainer implements IReagentContainer
{
	protected ReagentStack reagentStack;
	protected int capacity;

	public ReagentContainer(int capacity)
    {
        this(null, capacity);
    }

    public ReagentContainer(ReagentStack stack, int capacity)
    {
        this.reagentStack = stack;
        this.capacity = capacity;
    }

    public ReagentContainer(Reagent reagent, int amount, int capacity)
    {
        this(new ReagentStack(reagent, amount), capacity);
    }

    public static ReagentContainer readFromNBT(NBTTagCompound nbt)
    {
        ReagentStack reagent = ReagentStack.loadReagentStackFromNBT(nbt);
        int capacity = nbt.getInteger("capacity");
        
        if (reagent != null)
        {
            return new ReagentContainer(reagent, capacity);
        }else
        {
        	return new ReagentContainer(null, capacity);
        }
    }

    public NBTTagCompound writeToNBT(NBTTagCompound nbt)
    {
        if (reagentStack != null)
        {
            reagentStack.writeToNBT(nbt);
        }
        
        nbt.setInteger("capacity", capacity);

        return nbt;
    }
	
    @Override
    public ReagentStack getReagent()
    {
        return reagentStack;
    }

    @Override
    public int getReagentStackAmount()
    {
        if (reagentStack == null)
        {
            return 0;
        }
        return reagentStack.amount;
    }

    @Override
    public int getCapacity()
    {
        return capacity;
    }

    @Override
    public int fill(ReagentStack resource, boolean doFill)
    {
        if (resource == null)
        {
            return 0;
        }

        if (!doFill)
        {
            if (reagentStack == null)
            {
                return Math.min(capacity, resource.amount);
            }

            if (!reagentStack.isReagentEqual(resource))
            {
                return 0;
            }

            return Math.min(capacity - reagentStack.amount, resource.amount);
        }

        if (reagentStack == null)
        {
            reagentStack = new ReagentStack(resource, Math.min(capacity, resource.amount));

            return reagentStack.amount;
        }

        if (!reagentStack.isReagentEqual(resource))
        {
            return 0;
        }
        int filled = capacity - reagentStack.amount;

        if (resource.amount < filled)
        {
            reagentStack.amount += resource.amount;
            filled = resource.amount;
        }
        else
        {
            reagentStack.amount = capacity;
        }

        return filled;
    }

    @Override
    public ReagentStack drain(int maxDrain, boolean doDrain)
    {
        if (reagentStack == null)
        {
            return null;
        }

        int drained = maxDrain;
        if (reagentStack.amount < drained)
        {
            drained = reagentStack.amount;
        }

        ReagentStack stack = new ReagentStack(reagentStack, drained);
        if (doDrain)
        {
            reagentStack.amount -= drained;
            if (reagentStack.amount <= 0)
            {
                reagentStack = null;
            }
        }
        return stack;
    }

	@Override
	public ReagentContainerInfo getInfo() 
	{
		return new ReagentContainerInfo(this);
	}
}