StatementParameterItemStackExact.java

package buildcraft.core.statements;

import net.minecraft.client.renderer.texture.IIconRegister;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.util.IIcon;

import buildcraft.api.statements.IStatement;
import buildcraft.api.statements.IStatementContainer;
import buildcraft.api.statements.IStatementParameter;
import buildcraft.api.statements.StatementMouseClick;

public class StatementParameterItemStackExact implements IStatementParameter {

	protected ItemStack stack;

	@Override
	public IIcon getIcon() {
		return null;
	}

	@Override
	public ItemStack getItemStack() {
		return stack;
	}

	@Override
	public void onClick(IStatementContainer source, IStatement stmt, ItemStack stack, StatementMouseClick mouse) {
		if (stack != null) {
			if (areItemsEqual(this.stack, stack)) {
				if (mouse.getButton() == 0) {
					this.stack.stackSize += (mouse.isShift()) ? 16 : 1;
					if (this.stack.stackSize > 64) {
						this.stack.stackSize = 64;
					}
				} else {
					this.stack.stackSize -= (mouse.isShift()) ? 16 : 1;
					if (this.stack.stackSize < 0) {
						this.stack.stackSize = 0;
					}
				}
			} else {
				this.stack = stack.copy();
			}
		} else {
			if (this.stack != null) {
				if (mouse.getButton() == 0) {
					this.stack.stackSize += (mouse.isShift()) ? 16 : 1;
					if (this.stack.stackSize > 64) {
						this.stack.stackSize = 64;
					}
				} else {
					this.stack.stackSize -= (mouse.isShift()) ? 16 : 1;
					if (this.stack.stackSize < 0) {
						this.stack = null;
					}
				}
			}
		}
	}

	@Override
	public void writeToNBT(NBTTagCompound compound) {
		if (stack != null) {
			NBTTagCompound tagCompound = new NBTTagCompound();
			stack.writeToNBT(tagCompound);
			compound.setTag("stack", tagCompound);
		}
	}

	@Override
	public void readFromNBT(NBTTagCompound compound) {
		stack = ItemStack.loadItemStackFromNBT(compound.getCompoundTag("stack"));
	}

	@Override
	public boolean equals(Object object) {
		if (object instanceof StatementParameterItemStackExact) {
			StatementParameterItemStackExact param = (StatementParameterItemStackExact) object;

			return areItemsEqual(stack, param.stack);
		} else {
			return false;
		}
	}

	private boolean areItemsEqual(ItemStack stack1, ItemStack stack2) {
		if (stack1 != null) {
			return stack2 != null && stack1.isItemEqual(stack2) && ItemStack.areItemStackTagsEqual(stack1, stack2);
		} else {
			return stack2 == null;
		}
	}

	@Override
	public String getDescription() {
		if (stack != null) {
			return stack.getDisplayName();
		} else {
			return "";
		}
	}

	@Override
	public String getUniqueTag() {
		return "buildcraft:stackExact";
	}

	@Override
	public void registerIcons(IIconRegister iconRegister) {

	}

	@Override
	public IStatementParameter rotateLeft() {
		return this;
	}
}