Hbm-s-Nuclear-Tech-GIT/com/hbm/items/special/ItemBattery.java

package com.hbm.items.special;

import java.util.List;

import com.hbm.items.ModItems;
import com.hbm.lib.Library;

import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.item.EnumRarity;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;

public class ItemBattery extends Item {

	private long maxCharge;
	private long chargeRate;
	private long dischargeRate;

	public ItemBattery(long dura, long chargeRate, long dischargeRate) {
		this.maxCharge = dura;
		this.chargeRate = chargeRate;
		this.dischargeRate = dischargeRate;
		this.setMaxDamage(100);
		this.canRepair = false;
	}
	
	@Override
	public void addInformation(ItemStack itemstack, EntityPlayer player, List list, boolean bool)
	{
		long charge = maxCharge;
		if(itemstack.hasTagCompound())
			charge = ItemBattery.getCharge(itemstack);
		
		if(itemstack.getItem() != ModItems.fusion_core && 
				itemstack.getItem() != ModItems.factory_core_titanium && 
				itemstack.getItem() != ModItems.factory_core_advanced && 
				itemstack.getItem() != ModItems.energy_core && 
				itemstack.getItem() != ModItems.dynosphere_desh && 
				itemstack.getItem() != ModItems.dynosphere_schrabidium && 
				itemstack.getItem() != ModItems.dynosphere_euphemium && 
				itemstack.getItem() != ModItems.dynosphere_dineutronium)
		{
			list.add("Energy stored: " + Library.getShortNumber(charge * 100) + "/" + Library.getShortNumber(maxCharge * 100) + "HE");
		} else {
			String charge1 = Library.getShortNumber((charge  * 100) / this.maxCharge);
			list.add("Charge: " + charge1 + "%");
			list.add("(" + Library.getShortNumber(charge * 100) + "/" + Library.getShortNumber(maxCharge * 100) + "HE)");
		}
	}

    @Override
	public EnumRarity getRarity(ItemStack p_77613_1_) {
    	
    	if(this == ModItems.battery_schrabidium)
    	{
        	return EnumRarity.rare;
    	}

    	if(this == ModItems.fusion_core || 
    			this == ModItems.factory_core_titanium || 
    			this == ModItems.factory_core_advanced || 
    			this == ModItems.energy_core || 
    			this == ModItems.dynosphere_desh || 
    			this == ModItems.dynosphere_schrabidium || 
    			this == ModItems.dynosphere_euphemium || 
    			this == ModItems.dynosphere_dineutronium)
    	{
        	return EnumRarity.uncommon;
    	}
    	
    	return EnumRarity.common;
    }
    
    public void chargeBattery(ItemStack stack, int i) {
    	if(stack.getItem() instanceof ItemBattery) {
    		if(stack.hasTagCompound()) {
    			stack.stackTagCompound.setLong("charge", stack.stackTagCompound.getLong("charge") + i);
    		} else {
    			stack.stackTagCompound = new NBTTagCompound();
    			stack.stackTagCompound.setLong("charge", i);
    		}
    	}
    }
    
    public void setCharge(ItemStack stack, int i) {
    	if(stack.getItem() instanceof ItemBattery) {
    		if(stack.hasTagCompound()) {
    			stack.stackTagCompound.setLong("charge", i);
    		} else {
    			stack.stackTagCompound = new NBTTagCompound();
    			stack.stackTagCompound.setLong("charge", i);
    		}
    	}
    }
    
    public void dischargeBattery(ItemStack stack, int i) {
    	if(stack.getItem() instanceof ItemBattery) {
    		if(stack.hasTagCompound()) {
    			stack.stackTagCompound.setLong("charge", stack.stackTagCompound.getLong("charge") - i);
    		} else {
    			stack.stackTagCompound = new NBTTagCompound();
    			stack.stackTagCompound.setLong("charge", this.maxCharge - i);
    		}
    	}
    }
    
    public static long getCharge(ItemStack stack) {
    	if(stack.getItem() instanceof ItemBattery) {
    		if(stack.hasTagCompound()) {
    			return stack.stackTagCompound.getLong("charge");
    		} else {
    			stack.stackTagCompound = new NBTTagCompound();
    			stack.stackTagCompound.setLong("charge", ((ItemBattery)stack.getItem()).maxCharge);
    			return stack.stackTagCompound.getLong("charge");
    		}
    	}
    	
    	return 0;
    }
    
    public long getMaxCharge() {
    	return maxCharge;
    }
    
    public long getChargeRate() {
    	return chargeRate;
    }
    
    public long getDischargeRate() {
    	return dischargeRate;
    }
    
    public static long getMaxChargeStatic(ItemStack stack) {
    	return ((ItemBattery)stack.getItem()).maxCharge;
    }
    
    public static ItemStack getEmptyBattery(Item item) {
    	
    	if(item instanceof ItemBattery) {
    		ItemStack stack = new ItemStack(item);
    		stack.stackTagCompound = new NBTTagCompound();
    		stack.stackTagCompound.setLong("charge", 0);
    		stack.setItemDamage(100);
    		return stack.copy();
    	}
    	
    	return null;
    }
    
    public static ItemStack getFullBattery(Item item) {
    	
    	if(item instanceof ItemBattery) {
    		ItemStack stack = new ItemStack(item);
    		stack.stackTagCompound = new NBTTagCompound();
    		stack.stackTagCompound.setLong("charge", getMaxChargeStatic(stack));
    		return stack.copy();
    	}
    	
    	return null;
    }
	
	public static void updateDamage(ItemStack stack) {
		
		if(!stack.hasTagCompound()) {
			stack = getFullBattery(stack.getItem()).copy();
		}

		stack.setItemDamage(100 - (int)((double)getCharge(stack) / (double)getMaxChargeStatic(stack) * 100D));
	}
	
}