buildcraftAdditions.tileEntities.varHelpers.Upgrades.java Source code

Introduction

Here is the source code for buildcraftAdditions.tileEntities.varHelpers.Upgrades.java

Source

package buildcraftAdditions.tileEntities.varHelpers;

import java.util.EnumSet;
import java.util.Iterator;
import java.util.Set;

import com.google.common.collect.ImmutableSet;

import io.netty.buffer.ByteBuf;

import net.minecraft.nbt.NBTTagCompound;

import net.minecraftforge.common.util.Constants;

import buildcraftAdditions.api.nbt.INBTSaveable;
import buildcraftAdditions.api.networking.ISyncObject;
import buildcraftAdditions.reference.enums.EnumMachineUpgrades;

/**
 * Copyright (c) 2014-2015, AEnterprise
 * http://buildcraftadditions.wordpress.com/
 * Buildcraft Additions is distributed under the terms of GNU GPL v3.0
 * Please check the contents of the license located in
 * http://buildcraftadditions.wordpress.com/wiki/licensing-stuff/
 */
public class Upgrades implements ISyncObject, INBTSaveable {

    private final EnumSet<EnumMachineUpgrades> upgrades, whitelist, blacklist;
    private byte maxUpgrades;

    public Upgrades(int maxUpgrades) {
        this.maxUpgrades = (byte) (0xFF & maxUpgrades);
        upgrades = EnumSet.noneOf(EnumMachineUpgrades.class);
        whitelist = EnumSet.noneOf(EnumMachineUpgrades.class);
        blacklist = EnumSet.noneOf(EnumMachineUpgrades.class);
    }

    public Upgrades installUpgrade(EnumMachineUpgrades upgrade) {
        upgrades.add(upgrade);
        return this;
    }

    public boolean canInstallUpgrade(EnumMachineUpgrades upgrade) {
        return upgrades.size() < maxUpgrades
                && (!upgrades.contains(upgrade) || upgrade.canBeInstalledMultipleTimes())
                && !blacklist.contains(upgrade) && (whitelist.isEmpty() || whitelist.contains(upgrade));
    }

    public Upgrades blacklistUpgrade(EnumMachineUpgrades upgrade) {
        blacklist.add(upgrade);
        return this;
    }

    public Upgrades whitelistUpgrade(EnumMachineUpgrades upgrade) {
        whitelist.add(upgrade);
        return this;
    }

    public Set<EnumMachineUpgrades> getUpgrades() {
        return ImmutableSet.copyOf(upgrades);
    }

    public boolean hasUpgrade(EnumMachineUpgrades upgrade) {
        return upgrades.contains(upgrade);
    }

    public EnumMachineUpgrades removeUpgrade() {
        if (upgrades.size() == 0)
            return null;
        EnumMachineUpgrades upgrade = null;
        Iterator<EnumMachineUpgrades> it = upgrades.iterator();
        while (it.hasNext())
            upgrade = it.next();
        it.remove();
        return upgrade;
    }

    @Override
    public void writeToByteBuff(ByteBuf buf) {
        buf.writeByte(maxUpgrades);
        buf.writeByte(upgrades.size());
        for (EnumMachineUpgrades upgrade : upgrades)
            buf.writeByte(upgrade.ordinal());
    }

    @Override
    public void readFromByteBuff(ByteBuf buf) {
        maxUpgrades = buf.readByte();
        upgrades.clear();
        byte installedUpgrades = buf.readByte();
        for (byte b = 0; b < installedUpgrades; b++)
            upgrades.add(EnumMachineUpgrades.values()[buf.readByte()]);
    }

    @Override
    public void writeToNBT(NBTTagCompound tag) {
        tag.setByte("maxUpgrades", maxUpgrades);
        byte[] upgradeIDs = new byte[upgrades.size()];
        EnumMachineUpgrades[] upgradeArray = upgrades.toArray(new EnumMachineUpgrades[upgrades.size()]);
        for (int i = 0; i < upgradeIDs.length; i++)
            upgradeIDs[i] = (byte) (0xFF & upgradeArray[i].ordinal());
        tag.setByteArray("upgrades", upgradeIDs);
    }

    //TODO: Remove Integer reading.
    @Override
    public void readFromNBT(NBTTagCompound tag) {
        if (tag.hasKey("maxUpgrades", Constants.NBT.TAG_INT))
            maxUpgrades = (byte) (0xFF & tag.getInteger("maxUpgrades"));
        if (tag.hasKey("maxUpgrades", Constants.NBT.TAG_BYTE))
            maxUpgrades = tag.getByte("maxUpgrades");
        if (tag.hasKey("upgrades", Constants.NBT.TAG_INT_ARRAY)) {
            int[] upgradeIDs = tag.getIntArray("upgrades");
            for (int i : upgradeIDs)
                upgrades.add(EnumMachineUpgrades.values()[i]);
        }
        if (tag.hasKey("upgrades", Constants.NBT.TAG_BYTE_ARRAY)) {
            byte[] upgradeIDs = tag.getByteArray("upgrades");
            for (byte b : upgradeIDs)
                upgrades.add(EnumMachineUpgrades.values()[b]);
        }
    }

    public void setMaxUpgrades(int maxUpgrades) {
        this.maxUpgrades = (byte) (0xFF & maxUpgrades);
    }

    public void invalidate() {
        upgrades.clear();
    }

    @Override
    public String toString() {
        return "Upgrades: maxUpgrades = [" + maxUpgrades + "], upgrades = " + upgrades + ", whitelist = "
                + whitelist + ", blacklist = " + blacklist;
    }
}