9 files changed, 351 insertions, 0 deletions
diff --git a/src/Java/api/cofh/energy/EnergyStorage.java b/src/Java/api/cofh/energy/EnergyStorage.java
new file mode 100644
index 0000000000..a6ec363709
--- /dev/null
+++ b/src/Java/api/cofh/energy/EnergyStorage.java
@@ -0,0 +1,129 @@
+package api.cofh.energy;
+
+import net.minecraft.nbt.NBTTagCompound;
+
+public class EnergyStorage
+  implements IEnergyStorage
+{
+  protected int energy;
+  protected int capacity;
+  protected int maxReceive;
+  protected int maxExtract;
+  
+  public EnergyStorage(int capacity)
+  {
+    this(capacity, capacity, capacity);
+  }
+  
+  public EnergyStorage(int capacity, int maxTransfer)
+  {
+    this(capacity, maxTransfer, maxTransfer);
+  }
+  
+  public EnergyStorage(int capacity, int maxReceive, int maxExtract)
+  {
+    this.capacity = capacity;
+    this.maxReceive = maxReceive;
+    this.maxExtract = maxExtract;
+  }
+  
+  public EnergyStorage readFromNBT(NBTTagCompound nbt)
+  {
+    this.energy = nbt.getInteger("Energy");
+    if (this.energy > this.capacity) {
+      this.energy = this.capacity;
+    }
+    return this;
+  }
+  
+  public NBTTagCompound writeToNBT(NBTTagCompound nbt)
+  {
+    if (this.energy < 0) {
+      this.energy = 0;
+    }
+    nbt.setInteger("Energy", this.energy);
+    return nbt;
+  }
+  
+  public void setCapacity(int capacity)
+  {
+    this.capacity = capacity;
+    if (this.energy > capacity) {
+      this.energy = capacity;
+    }
+  }
+  
+  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 getMaxReceive()
+  {
+    return this.maxReceive;
+  }
+  
+  public int getMaxExtract()
+  {
+    return this.maxExtract;
+  }
+  
+  public void setEnergyStored(int energy)
+  {
+    this.energy = energy;
+    if (this.energy > this.capacity) {
+      this.energy = this.capacity;
+    } else if (this.energy < 0) {
+      this.energy = 0;
+    }
+  }
+  
+  public void modifyEnergyStored(int energy)
+  {
+    this.energy += energy;
+    if (this.energy > this.capacity) {
+      this.energy = this.capacity;
+    } else if (this.energy < 0) {
+      this.energy = 0;
+    }
+  }
+  
+  public int receiveEnergy(int maxReceive, boolean simulate)
+  {
+    int energyReceived = Math.min(this.capacity - this.energy, Math.min(this.maxReceive, maxReceive));
+    if (!simulate) {
+      this.energy += energyReceived;
+    }
+    return energyReceived;
+  }
+  
+  public int extractEnergy(int maxExtract, boolean simulate)
+  {
+    int energyExtracted = Math.min(this.energy, Math.min(this.maxExtract, maxExtract));
+    if (!simulate) {
+      this.energy -= energyExtracted;
+    }
+    return energyExtracted;
+  }
+  
+  public int getEnergyStored()
+  {
+    return this.energy;
+  }
+  
+  public int getMaxEnergyStored()
+  {
+    return this.capacity;
+  }
+}
diff --git a/src/Java/api/cofh/energy/IEnergyConnection.java b/src/Java/api/cofh/energy/IEnergyConnection.java
new file mode 100644
index 0000000000..6351f7702d
--- /dev/null
+++ b/src/Java/api/cofh/energy/IEnergyConnection.java
@@ -0,0 +1,8 @@
+package api.cofh.energy;
+
+import net.minecraftforge.common.util.ForgeDirection;
+
+public abstract interface IEnergyConnection
+{
+  public abstract boolean canConnectEnergy(ForgeDirection paramForgeDirection);
+}
diff --git a/src/Java/api/cofh/energy/IEnergyContainerItem.java b/src/Java/api/cofh/energy/IEnergyContainerItem.java
new file mode 100644
index 0000000000..1f2b8c3c52
--- /dev/null
+++ b/src/Java/api/cofh/energy/IEnergyContainerItem.java
@@ -0,0 +1,14 @@
+package api.cofh.energy;
+
+import net.minecraft.item.ItemStack;
+
+public abstract interface IEnergyContainerItem
+{
+  public abstract int receiveEnergy(ItemStack paramItemStack, int paramInt, boolean paramBoolean);
+  
+  public abstract int extractEnergy(ItemStack paramItemStack, int paramInt, boolean paramBoolean);
+  
+  public abstract int getEnergyStored(ItemStack paramItemStack);
+  
+  public abstract int getMaxEnergyStored(ItemStack paramItemStack);
+}
diff --git a/src/Java/api/cofh/energy/IEnergyHandler.java b/src/Java/api/cofh/energy/IEnergyHandler.java
new file mode 100644
index 0000000000..9d3385c4c0
--- /dev/null
+++ b/src/Java/api/cofh/energy/IEnergyHandler.java
@@ -0,0 +1,15 @@
+package api.cofh.energy;
+
+import net.minecraftforge.common.util.ForgeDirection;
+
+public abstract interface IEnergyHandler
+  extends IEnergyProvider, IEnergyReceiver
+{
+  public abstract int receiveEnergy(ForgeDirection paramForgeDirection, int paramInt, boolean paramBoolean);
+  
+  public abstract int extractEnergy(ForgeDirection paramForgeDirection, int paramInt, boolean paramBoolean);
+  
+  public abstract int getEnergyStored(ForgeDirection paramForgeDirection);
+  
+  public abstract int getMaxEnergyStored(ForgeDirection paramForgeDirection);
+}
diff --git a/src/Java/api/cofh/energy/IEnergyProvider.java b/src/Java/api/cofh/energy/IEnergyProvider.java
new file mode 100644
index 0000000000..ace09db65e
--- /dev/null
+++ b/src/Java/api/cofh/energy/IEnergyProvider.java
@@ -0,0 +1,13 @@
+package api.cofh.energy;
+
+import net.minecraftforge.common.util.ForgeDirection;
+
+public abstract interface IEnergyProvider
+  extends IEnergyConnection
+{
+  public abstract int extractEnergy(ForgeDirection paramForgeDirection, int paramInt, boolean paramBoolean);
+  
+  public abstract int getEnergyStored(ForgeDirection paramForgeDirection);
+  
+  public abstract int getMaxEnergyStored(ForgeDirection paramForgeDirection);
+}
diff --git a/src/Java/api/cofh/energy/IEnergyReceiver.java b/src/Java/api/cofh/energy/IEnergyReceiver.java
new file mode 100644
index 0000000000..68b071b028
--- /dev/null
+++ b/src/Java/api/cofh/energy/IEnergyReceiver.java
@@ -0,0 +1,13 @@
+package api.cofh.energy;
+
+import net.minecraftforge.common.util.ForgeDirection;
+
+public abstract interface IEnergyReceiver
+  extends IEnergyConnection
+{
+  public abstract int receiveEnergy(ForgeDirection paramForgeDirection, int paramInt, boolean paramBoolean);
+  
+  public abstract int getEnergyStored(ForgeDirection paramForgeDirection);
+  
+  public abstract int getMaxEnergyStored(ForgeDirection paramForgeDirection);
+}
diff --git a/src/Java/api/cofh/energy/IEnergyStorage.java b/src/Java/api/cofh/energy/IEnergyStorage.java
new file mode 100644
index 0000000000..bf999f8670
--- /dev/null
+++ b/src/Java/api/cofh/energy/IEnergyStorage.java
@@ -0,0 +1,12 @@
+package api.cofh.energy;
+
+public abstract interface IEnergyStorage
+{
+  public abstract int receiveEnergy(int paramInt, boolean paramBoolean);
+  
+  public abstract int extractEnergy(int paramInt, boolean paramBoolean);
+  
+  public abstract int getEnergyStored();
+  
+  public abstract int getMaxEnergyStored();
+}
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;
+  }
+}
diff --git a/src/Java/api/cofh/energy/TileEnergyHandler.java b/src/Java/api/cofh/energy/TileEnergyHandler.java
new file mode 100644
index 0000000000..606b0ba44b
--- /dev/null
+++ b/src/Java/api/cofh/energy/TileEnergyHandler.java
@@ -0,0 +1,49 @@
+package api.cofh.energy;
+
+import net.minecraft.nbt.NBTTagCompound;
+import net.minecraft.tileentity.TileEntity;
+import net.minecraftforge.common.util.ForgeDirection;
+
+public class TileEnergyHandler
+  extends TileEntity
+  implements IEnergyHandler
+{
+  protected EnergyStorage storage = new EnergyStorage(32000);
+  
+  public void readFromNBT(NBTTagCompound nbt)
+  {
+    super.readFromNBT(nbt);
+    this.storage.readFromNBT(nbt);
+  }
+  
+  public void writeToNBT(NBTTagCompound nbt)
+  {
+    super.writeToNBT(nbt);
+    this.storage.writeToNBT(nbt);
+  }
+  
+  public boolean canConnectEnergy(ForgeDirection from)
+  {
+    return true;
+  }
+  
+  public int receiveEnergy(ForgeDirection from, int maxReceive, boolean simulate)
+  {
+    return this.storage.receiveEnergy(maxReceive, simulate);
+  }
+  
+  public int extractEnergy(ForgeDirection from, int maxExtract, boolean simulate)
+  {
+    return this.storage.extractEnergy(maxExtract, simulate);
+  }
+  
+  public int getEnergyStored(ForgeDirection from)
+  {
+    return this.storage.getEnergyStored();
+  }
+  
+  public int getMaxEnergyStored(ForgeDirection from)
+  {
+    return this.storage.getMaxEnergyStored();
+  }
+}