MagicBees/src/main/java/magicbees/tileentity/AuraCharges.java

package magicbees.tileentity;

import java.util.HashMap;
import java.util.Map;

import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.world.World;

import magicbees.bees.AuraCharge;

public class AuraCharges {
	private static final String auraChargesTagName = "AuraCharges";

	private final Map<AuraCharge, Long> charges = new HashMap<AuraCharge, Long>(AuraCharge.values().length);

	public void start(AuraCharge auraChargeType, World worldObj) {
		charges.put(auraChargeType, worldObj.getTotalWorldTime());
	}

	public boolean isActive(AuraCharge auraChargeType) {
		return charges.keySet().contains(auraChargeType);
	}

	public boolean isExpired(AuraCharge auraChargeType, World worldObj) {
		Long chargeStart = charges.get(auraChargeType);
		return chargeStart == null || chargeStart + auraChargeType.duration <= worldObj.getTotalWorldTime();
	}

	public void stop(AuraCharge AuraChargeType) {
		charges.remove(AuraChargeType);
	}

	public void writeToNBT(NBTTagCompound nbtTagCompound) {
		NBTTagCompound auraChargeTags = new NBTTagCompound();
		for (Map.Entry<AuraCharge, Long> chargeEntry : charges.entrySet()) {
			String chargeName = chargeEntry.getKey().toString();
			Long chargeStart = chargeEntry.getValue();
			auraChargeTags.setLong(chargeName, chargeStart);
		}
		nbtTagCompound.setTag(auraChargesTagName, auraChargeTags);
	}

	public void readFromNBT(NBTTagCompound nbtTagCompound) {
		NBTTagCompound auraChargeTags = nbtTagCompound.getCompoundTag(auraChargesTagName);
		for (AuraCharge auraCharge : AuraCharge.values()) {
			String chargeName = auraCharge.toString();
			Long chargeStart = auraChargeTags.getLong(chargeName);
			if (chargeStart > 0) {
				charges.put(auraCharge, chargeStart);
			}
		}
	}

	public int writeToFlags() {
		int flags = 0;
		for (AuraCharge auraChargeType : charges.keySet()) {
			flags |= auraChargeType.flag;
		}
		return flags;
	}

	public void readFromFlags(int flags) {
		charges.clear();
		for (AuraCharge auraCharge : AuraCharge.values()) {
			if ((flags & auraCharge.flag) > 0) {
				charges.put(auraCharge, 0L);
			}
		}
	}
}