EnergyStorageSerializable.java

package com.rwtema.funkylocomotion.energy;

import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagInt;
import net.minecraftforge.common.util.INBTSerializable;
import net.minecraftforge.energy.EnergyStorage;

public class EnergyStorageSerializable extends EnergyStorage implements INBTSerializable<NBTTagInt> {
	public EnergyStorageSerializable(int capacity) {
		super(capacity);
	}

	public EnergyStorageSerializable(int capacity, int maxTransfer) {
		super(capacity, maxTransfer);
	}

	public EnergyStorageSerializable(int capacity, int maxReceive, int maxExtract) {
		super(capacity, maxReceive, maxExtract);
	}

	@Override
	public NBTTagInt serializeNBT() {
		return new NBTTagInt(energy);
	}

	@Override
	public void deserializeNBT(NBTTagInt nbt) {
		energy = nbt.getInt();
	}

	public void readFromNBT(NBTTagCompound tag) {
		energy = tag.getInteger("energy");
	}

	public NBTTagCompound writeToNBT(NBTTagCompound tag) {
		tag.setInteger("energy", energy);
		return tag;
	}
}