mirror of
https://github.com/HbmMods/Hbm-s-Nuclear-Tech-GIT.git
synced 2026-01-25 10:32:49 +00:00
116 lines
3.5 KiB
Java
116 lines
3.5 KiB
Java
package com.hbm.inventory;
|
|
|
|
import java.util.Arrays;
|
|
|
|
import com.hbm.handler.FluidTypeHandler.FluidType;
|
|
import com.hbm.items.ModItems;
|
|
import com.hbm.packet.PacketDispatcher;
|
|
import com.hbm.packet.TEAssemblerPacket;
|
|
import com.hbm.packet.TEFluidPacket;
|
|
|
|
import net.minecraft.client.gui.inventory.GuiContainer;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraft.nbt.NBTTagCompound;
|
|
|
|
public class FluidTank {
|
|
|
|
FluidType type;
|
|
int fluid;
|
|
int maxFluid;
|
|
public boolean takeIn = true;
|
|
public boolean letOut = false;
|
|
int index;
|
|
|
|
public FluidTank(FluidType type, int maxFluid, int index) {
|
|
this.type = type;
|
|
this.maxFluid = maxFluid;
|
|
this.index = index;
|
|
}
|
|
|
|
public void setFill(int i) {
|
|
fluid = i;
|
|
}
|
|
|
|
//Called on TE update
|
|
public void updateTank(int x, int y, int z) {
|
|
|
|
PacketDispatcher.wrapper.sendToAll(new TEFluidPacket(x, y, z, fluid, index));
|
|
}
|
|
|
|
//Fills tank from canisters
|
|
public void loadTank(int in, int out, ItemStack[] slots) {
|
|
|
|
FluidType inType = FluidType.NONE;
|
|
if(slots[in] != null) {
|
|
inType = FluidContainerRegistry.getFluidType(slots[in]);
|
|
}
|
|
|
|
if(slots[in] != null && inType.name().equals(type.name()) && fluid + FluidContainerRegistry.getFluidContent(slots[in], type) <= maxFluid) {
|
|
if(slots[out] == null) {
|
|
fluid += FluidContainerRegistry.getFluidContent(slots[in], type);
|
|
slots[out] = FluidContainerRegistry.getEmptyContainer(slots[in]);
|
|
slots[in].stackSize--;
|
|
if(slots[in].stackSize <= 0)
|
|
slots[in] = null;
|
|
} else if(slots[out] != null && slots[out].getItem() == FluidContainerRegistry.getEmptyContainer(slots[in]).getItem() && slots[out].stackSize < slots[out].getMaxStackSize()) {
|
|
fluid += FluidContainerRegistry.getFluidContent(slots[in], type);
|
|
slots[in].stackSize--;
|
|
if(slots[in].stackSize <= 0)
|
|
slots[in] = null;
|
|
slots[out].stackSize++;
|
|
}
|
|
}
|
|
}
|
|
|
|
//Fills canisters from tank
|
|
public void unloadTank(int in, int out, ItemStack[] slots) {
|
|
|
|
ItemStack full = null;
|
|
if(slots[in] != null) {
|
|
full = FluidContainerRegistry.getFullContainer(slots[in], type);
|
|
}
|
|
if(full == null)
|
|
return;
|
|
|
|
if(slots[in] != null && fluid - FluidContainerRegistry.getFluidContent(full, type) >= 0) {
|
|
if(slots[out] == null) {
|
|
fluid -= FluidContainerRegistry.getFluidContent(full, type);
|
|
slots[out] = full.copy();
|
|
slots[in].stackSize--;
|
|
if(slots[in].stackSize <= 0)
|
|
slots[in] = null;
|
|
} else if(slots[out] != null && slots[out].getItem() == FluidContainerRegistry.getFullContainer(slots[in], type).getItem() && slots[out].stackSize < slots[out].getMaxStackSize()) {
|
|
fluid -= FluidContainerRegistry.getFluidContent(full, type);
|
|
slots[in].stackSize--;
|
|
if(slots[in].stackSize <= 0)
|
|
slots[in] = null;
|
|
slots[out].stackSize++;
|
|
}
|
|
}
|
|
}
|
|
|
|
//Used in the GUI rendering, renders correct fluid type in container with progress
|
|
public void renderTank(GuiContainer gui, int x, int y, int tx, int ty, int width, int height) {
|
|
|
|
int i = (fluid * height) / maxFluid;
|
|
gui.drawTexturedModalRect(x, y - i, tx, ty - i, width, i);
|
|
}
|
|
|
|
//Called by TE to save fillstate
|
|
public void writeToNBT(NBTTagCompound nbt, String s) {
|
|
nbt.setInteger(s, fluid);
|
|
nbt.setInteger(s + "_type", Arrays.asList(FluidType.values()).indexOf(type));
|
|
nbt.setBoolean(s + "_in", takeIn);
|
|
nbt.setBoolean(s + "_out", letOut);
|
|
}
|
|
|
|
//Called by TE to load fillstate
|
|
public void readFromNBT(NBTTagCompound nbt, String s) {
|
|
fluid = nbt.getInteger(s);
|
|
type = FluidType.getEnum(nbt.getInteger(s + "_type"));
|
|
takeIn = nbt.getBoolean(s + "_in");
|
|
letOut = nbt.getBoolean(s + "_out");
|
|
}
|
|
|
|
}
|