108 lines
3.5 KiB
Java
108 lines
3.5 KiB
Java
|
package net.minecraftforge.fluids;
|
||
|
|
|
||
|
|
import net.minecraft.block.BlockDispenser;
|
||
|
|
import net.minecraft.dispenser.BehaviorDefaultDispenseItem;
|
||
|
|
import net.minecraft.dispenser.IBlockSource;
|
||
|
|
import net.minecraft.item.ItemStack;
|
||
|
|
import net.minecraft.tileentity.TileEntityDispenser;
|
||
|
|
import net.minecraft.util.EnumFacing;
|
||
|
|
import net.minecraft.util.math.BlockPos;
|
||
|
|
import net.minecraft.world.World;
|
||
|
|
import net.minecraftforge.fluids.capability.IFluidHandler;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Fills or drains a fluid container item using a Dispenser.
|
||
|
|
*/
|
||
|
|
public class DispenseFluidContainer extends BehaviorDefaultDispenseItem
|
||
|
|
{
|
||
|
|
private static final DispenseFluidContainer INSTANCE = new DispenseFluidContainer();
|
||
|
|
|
||
|
|
public static DispenseFluidContainer getInstance()
|
||
|
|
{
|
||
|
|
return INSTANCE;
|
||
|
|
}
|
||
|
|
|
||
|
|
private DispenseFluidContainer() {}
|
||
|
|
|
||
|
|
private final BehaviorDefaultDispenseItem dispenseBehavior = new BehaviorDefaultDispenseItem();
|
||
|
|
|
||
|
|
public ItemStack dispenseStack(IBlockSource source, ItemStack stack)
|
||
|
|
{
|
||
|
|
if (FluidUtil.getFluidContained(stack) != null)
|
||
|
|
{
|
||
|
|
return dumpContainer(source, stack);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
return fillContainer(source, stack);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Picks up fluid in front of a Dispenser and fills a container with it.
|
||
|
|
*/
|
||
|
|
private ItemStack fillContainer(IBlockSource source, ItemStack stack)
|
||
|
|
{
|
||
|
|
World world = source.getWorld();
|
||
|
|
EnumFacing dispenserFacing = BlockDispenser.getFacing(source.getBlockMetadata());
|
||
|
|
BlockPos blockpos = source.getBlockPos().offset(dispenserFacing);
|
||
|
|
|
||
|
|
ItemStack result = FluidUtil.tryPickUpFluid(stack, null, world, blockpos, dispenserFacing.getOpposite());
|
||
|
|
if (result == null)
|
||
|
|
{
|
||
|
|
return super.dispenseStack(source, stack);
|
||
|
|
}
|
||
|
|
|
||
|
|
world.setBlockToAir(blockpos);
|
||
|
|
|
||
|
|
if (--stack.stackSize == 0)
|
||
|
|
{
|
||
|
|
stack.deserializeNBT(result.serializeNBT());
|
||
|
|
}
|
||
|
|
else if (((TileEntityDispenser)source.getBlockTileEntity()).addItemStack(result) < 0)
|
||
|
|
{
|
||
|
|
this.dispenseBehavior.dispense(source, result);
|
||
|
|
}
|
||
|
|
|
||
|
|
return stack;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Drains a filled container and places the fluid in front of the Dispenser.
|
||
|
|
*/
|
||
|
|
private ItemStack dumpContainer(IBlockSource source, ItemStack stack)
|
||
|
|
{
|
||
|
|
ItemStack dispensedStack = stack.copy();
|
||
|
|
dispensedStack.stackSize = 1;
|
||
|
|
IFluidHandler fluidHandler = FluidUtil.getFluidHandler(dispensedStack);
|
||
|
|
if (fluidHandler == null)
|
||
|
|
{
|
||
|
|
return super.dispenseStack(source, stack);
|
||
|
|
}
|
||
|
|
|
||
|
|
FluidStack fluidStack = fluidHandler.drain(Fluid.BUCKET_VOLUME, false);
|
||
|
|
EnumFacing dispenserFacing = BlockDispenser.getFacing(source.getBlockMetadata());
|
||
|
|
BlockPos blockpos = source.getBlockPos().offset(dispenserFacing);
|
||
|
|
|
||
|
|
if (fluidStack != null && fluidStack.amount == Fluid.BUCKET_VOLUME && FluidUtil.tryPlaceFluid(null, source.getWorld(), fluidStack, blockpos))
|
||
|
|
{
|
||
|
|
fluidHandler.drain(Fluid.BUCKET_VOLUME, true);
|
||
|
|
|
||
|
|
if (--stack.stackSize == 0)
|
||
|
|
{
|
||
|
|
stack.deserializeNBT(dispensedStack.serializeNBT());
|
||
|
|
}
|
||
|
|
else if (((TileEntityDispenser)source.getBlockTileEntity()).addItemStack(dispensedStack) < 0)
|
||
|
|
{
|
||
|
|
this.dispenseBehavior.dispense(source, dispensedStack);
|
||
|
|
}
|
||
|
|
|
||
|
|
return stack;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
return this.dispenseBehavior.dispense(source, stack);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|