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

package magicbees.tileentity;

import java.util.HashMap;
import java.util.Map;
import magicbees.bees.AuraCharge;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.world.World;

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);
            }
        }
    }
}