package gtPlusPlus.api.thermal.sample;

import gtPlusPlus.api.thermal.energy.IThermalContainerItem;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;

public class ItemThermalContainer extends Item implements IThermalContainerItem {
    protected int capacity;
    protected int maxReceive;
    protected int maxExtract;

    public ItemThermalContainer() {}

    public ItemThermalContainer(int arg0) {
        this(arg0, arg0, arg0);
    }

    public ItemThermalContainer(int arg0, int arg1) {
        this(arg0, arg1, arg1);
    }

    public ItemThermalContainer(int arg0, int arg1, int arg2) {
        this.capacity = arg0;
        this.maxReceive = arg1;
        this.maxExtract = arg2;
    }

    public ItemThermalContainer setCapacity(int arg0) {
        this.capacity = arg0;
        return this;
    }

    public void setMaxTransfer(int arg0) {
        this.setMaxReceive(arg0);
        this.setMaxExtract(arg0);
    }

    public void setMaxReceive(int arg0) {
        this.maxReceive = arg0;
    }

    public void setMaxExtract(int arg0) {
        this.maxExtract = arg0;
    }

    public int receiveThermalEnergy(ItemStack arg0, int arg1, boolean arg2) {
        if (arg0.getTagCompound() == null) {
            arg0.stackTagCompound = new NBTTagCompound();
        }
        int arg3 = arg0.stackTagCompound.getInteger("ThermalEnergy");
        int arg4 = Math.min(this.capacity - arg3, Math.min(this.maxReceive, arg1));
        if (!arg2) {
            arg3 += arg4;
            arg0.stackTagCompound.setInteger("ThermalEnergy", arg3);
        }
        return arg4;
    }

    public int extractThermalEnergy(ItemStack arg0, int arg1, boolean arg2) {
        if (arg0.stackTagCompound != null && arg0.stackTagCompound.hasKey("ThermalEnergy")) {
            int arg3 = arg0.stackTagCompound.getInteger("ThermalEnergy");
            int arg4 = Math.min(arg3, Math.min(this.maxExtract, arg1));
            if (!arg2) {
                arg3 -= arg4;
                arg0.stackTagCompound.setInteger("ThermalEnergy", arg3);
            }
            return arg4;
        } else {
            return 0;
        }
    }

    public int getThermalEnergyStored(ItemStack arg0) {
        return arg0.stackTagCompound != null && arg0.stackTagCompound.hasKey("ThermalEnergy")
                ? arg0.stackTagCompound.getInteger("ThermalEnergy")
                : 0;
    }

    public int getMaxThermalEnergyStored(ItemStack arg0) {
        return this.capacity;
    }
}