ActuallyAdditions/src/main/java/de/ellpeck/actuallyadditions/mod/entity/EntityRFMinecart.java

package de.ellpeck.actuallyadditions.mod.entity;

import cofh.api.energy.EnergyStorage;
import net.minecraft.entity.item.EntityMinecart;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.world.World;

public class EntityRFMinecart extends EntityMinecart{

    public EnergyStorage storage;

    public EntityRFMinecart(World world, int rfCap, int maxTransfer){
        super(world);

        this.storage = new EnergyStorage(rfCap, maxTransfer);
    }

    @Override
    protected void writeEntityToNBT(NBTTagCompound compound){
        super.writeEntityToNBT(compound);

        this.storage.writeToNBT(compound);
    }

    @Override
    protected void readEntityFromNBT(NBTTagCompound compound){
        super.readEntityFromNBT(compound);

        this.storage.readFromNBT(compound);
    }

    @Override
    public Type getType(){
        return Type.CHEST;
    }
}