89 lines
2.7 KiB
Java
89 lines
2.7 KiB
Java
package techreborn.util;
|
|
|
|
import net.minecraft.inventory.IInventory;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraftforge.common.util.ForgeDirection;
|
|
import net.minecraftforge.fluids.Fluid;
|
|
import net.minecraftforge.fluids.FluidContainerRegistry;
|
|
import net.minecraftforge.fluids.FluidStack;
|
|
import net.minecraftforge.fluids.IFluidHandler;
|
|
|
|
public class FluidUtils {
|
|
|
|
public static boolean drainContainers(IFluidHandler fluidHandler,
|
|
IInventory inv, int inputSlot, int outputSlot)
|
|
{
|
|
ItemStack input = inv.getStackInSlot(inputSlot);
|
|
ItemStack output = inv.getStackInSlot(outputSlot);
|
|
|
|
if (input != null)
|
|
{
|
|
FluidStack fluidInContainer = getFluidStackInContainer(input);
|
|
ItemStack emptyItem = input.getItem().getContainerItem(input);
|
|
if (fluidInContainer != null
|
|
&& (emptyItem == null || output == null || (output.stackSize < output
|
|
.getMaxStackSize() && ItemUtils.isItemEqual(output,
|
|
emptyItem, true, true))))
|
|
{
|
|
int used = fluidHandler.fill(ForgeDirection.UNKNOWN,
|
|
fluidInContainer, false);
|
|
if (used >= fluidInContainer.amount)
|
|
{
|
|
fluidHandler.fill(ForgeDirection.UNKNOWN, fluidInContainer,
|
|
true);
|
|
if (emptyItem != null)
|
|
if (output == null)
|
|
inv.setInventorySlotContents(outputSlot, emptyItem);
|
|
else
|
|
output.stackSize++;
|
|
inv.decrStackSize(inputSlot, 1);
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public static boolean fillContainers(IFluidHandler fluidHandler,
|
|
IInventory inv, int inputSlot, int outputSlot, Fluid fluidToFill)
|
|
{
|
|
ItemStack input = inv.getStackInSlot(inputSlot);
|
|
ItemStack output = inv.getStackInSlot(outputSlot);
|
|
ItemStack filled = getFilledContainer(fluidToFill, input);
|
|
if (filled != null
|
|
&& (output == null || (output.stackSize < output
|
|
.getMaxStackSize() && ItemUtils.isItemEqual(filled,
|
|
output, true, true))))
|
|
{
|
|
FluidStack fluidInContainer = getFluidStackInContainer(filled);
|
|
FluidStack drain = fluidHandler.drain(ForgeDirection.UNKNOWN,
|
|
fluidInContainer, false);
|
|
if (drain != null && drain.amount == fluidInContainer.amount)
|
|
{
|
|
fluidHandler.drain(ForgeDirection.UNKNOWN, fluidInContainer,
|
|
true);
|
|
if (output == null)
|
|
inv.setInventorySlotContents(outputSlot, filled);
|
|
else
|
|
output.stackSize++;
|
|
inv.decrStackSize(inputSlot, 1);
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public static FluidStack getFluidStackInContainer(ItemStack stack)
|
|
{
|
|
return FluidContainerRegistry.getFluidForFilledItem(stack);
|
|
}
|
|
|
|
public static ItemStack getFilledContainer(Fluid fluid, ItemStack empty)
|
|
{
|
|
if (fluid == null || empty == null)
|
|
return null;
|
|
return FluidContainerRegistry.fillFluidContainer(new FluidStack(fluid,
|
|
Integer.MAX_VALUE), empty);
|
|
}
|
|
|
|
}
|