1 files changed, 98 insertions, 0 deletions
diff --git a/src/Java/api/cofh/energy/ItemEnergyContainer.java b/src/Java/api/cofh/energy/ItemEnergyContainer.java
new file mode 100644
index 0000000000..f4da898919
--- /dev/null
+++ b/src/Java/api/cofh/energy/ItemEnergyContainer.java
@@ -0,0 +1,98 @@
+package api.cofh.energy;
+
+import net.minecraft.item.Item;
+import net.minecraft.item.ItemStack;
+import net.minecraft.nbt.NBTTagCompound;
+
+public class ItemEnergyContainer
+  extends Item
+  implements IEnergyContainerItem
+{
+  protected int capacity;
+  protected int maxReceive;
+  protected int maxExtract;
+  
+  public ItemEnergyContainer() {}
+  
+  public ItemEnergyContainer(int capacity)
+  {
+    this(capacity, capacity, capacity);
+  }
+  
+  public ItemEnergyContainer(int capacity, int maxTransfer)
+  {
+    this(capacity, maxTransfer, maxTransfer);
+  }
+  
+  public ItemEnergyContainer(int capacity, int maxReceive, int maxExtract)
+  {
+    this.capacity = capacity;
+    this.maxReceive = maxReceive;
+    this.maxExtract = maxExtract;
+  }
+  
+  public ItemEnergyContainer setCapacity(int capacity)
+  {
+    this.capacity = capacity;
+    return this;
+  }
+  
+  public void setMaxTransfer(int maxTransfer)
+  {
+    setMaxReceive(maxTransfer);
+    setMaxExtract(maxTransfer);
+  }
+  
+  public void setMaxReceive(int maxReceive)
+  {
+    this.maxReceive = maxReceive;
+  }
+  
+  public void setMaxExtract(int maxExtract)
+  {
+    this.maxExtract = maxExtract;
+  }
+  
+  public int receiveEnergy(ItemStack container, int maxReceive, boolean simulate)
+  {
+    if (container.stackTagCompound == null) {
+      container.stackTagCompound = new NBTTagCompound();
+    }
+    int energy = container.stackTagCompound.getInteger("Energy");
+    int energyReceived = Math.min(this.capacity - energy, Math.min(this.maxReceive, maxReceive));
+    if (!simulate)
+    {
+      energy += energyReceived;
+      container.stackTagCompound.setInteger("Energy", energy);
+    }
+    return energyReceived;
+  }
+  
+  public int extractEnergy(ItemStack container, int maxExtract, boolean simulate)
+  {
+    if ((container.stackTagCompound == null) || (!container.stackTagCompound.hasKey("Energy"))) {
+      return 0;
+    }
+    int energy = container.stackTagCompound.getInteger("Energy");
+    int energyExtracted = Math.min(energy, Math.min(this.maxExtract, maxExtract));
+    if (!simulate)
+    {
+      energy -= energyExtracted;
+      container.stackTagCompound.setInteger("Energy", energy);
+    }
+    return energyExtracted;
+  }
+  
+  public int getEnergyStored(ItemStack container)
+  {
+    if ((container.stackTagCompound == null) || (!container.stackTagCompound.hasKey("Energy"))) {
+      return 0;
+    }
+    return container.stackTagCompound.getInteger("Energy");
+  }
+  
+  public int getMaxEnergyStored(ItemStack container)
+  {
+    return this.capacity;
+  }
+}