ItemEntry.java

package crazypants.enderio.machine.invpanel.server;

import crazypants.enderio.machine.invpanel.ItemEntryBase;
import java.util.IdentityHashMap;
import net.minecraft.nbt.NBTTagCompound;

public class ItemEntry extends ItemEntryBase {
  private final IdentityHashMap<SlotKey, SlotKey> slots = new IdentityHashMap<SlotKey, SlotKey>();

  public ItemEntry(int dbID, int hash, int itemID, int meta, NBTTagCompound nbt) {
    super(dbID, hash, itemID, meta, nbt);
  }

  void addSlot(SlotKey slotKey) {
    slots.put(slotKey, slotKey);
  }

  void removeSlot(SlotKey slotKey) {
    slots.remove(slotKey);
  }

  int countItems(InventoryDatabaseServer db) {
    int count = 0;
    for (SlotKey slotKey : slots.values()) {
      count += slotKey.count;
    }
    return count;
  }

  int extractItems(InventoryDatabaseServer db, int count) {
    int extracted = 0;
    SlotKey[] copy = slots.values().toArray(new SlotKey[slots.size()]);
    for (SlotKey slotKey : copy) {
      int amount = slotKey.inv.extractItem(db, this, slotKey.slot, count);
      count -= amount;
      extracted += amount;
      if (count <= 0) {
        break;
      }
    }
    return extracted;
  }

}