52 lines
1.7 KiB
Java
52 lines
1.7 KiB
Java
package net.shadowfacts.funnels;
|
|
|
|
import net.minecraft.block.BlockLiquid;
|
|
import net.minecraft.block.state.IBlockState;
|
|
import net.minecraft.entity.player.EntityPlayer;
|
|
import net.minecraft.entity.player.EntityPlayerMP;
|
|
import net.minecraft.init.Blocks;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraft.util.EnumHand;
|
|
import net.minecraft.util.math.BlockPos;
|
|
import net.minecraft.world.World;
|
|
import net.minecraftforge.fluids.*;
|
|
import net.minecraftforge.fluids.capability.IFluidHandler;
|
|
|
|
/**
|
|
* @author shadowfacts
|
|
*/
|
|
public class FluidUtils {
|
|
|
|
public static boolean isFluidBlock(World world, BlockPos pos) {
|
|
IBlockState state = world.getBlockState(pos);
|
|
if (state.getBlock() instanceof BlockLiquid) {
|
|
return state.getValue(BlockLiquid.LEVEL) == 0;
|
|
}
|
|
if (state.getBlock() instanceof IFluidBlock) {
|
|
return ((IFluidBlock)state.getBlock()).canDrain(world, pos);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public static FluidStack drainFluidBlock(World world, BlockPos pos, boolean doDrain) {
|
|
FluidStack stack = null;
|
|
|
|
IBlockState state = world.getBlockState(pos);
|
|
|
|
if (state.getBlock() instanceof BlockLiquid && state.getValue(BlockLiquid.LEVEL) == 0) {
|
|
if (state.getBlock() == Blocks.WATER || state.getBlock() == Blocks.FLOWING_WATER) {
|
|
stack = new FluidStack(FluidRegistry.WATER, Fluid.BUCKET_VOLUME);
|
|
if (doDrain) world.setBlockToAir(pos);
|
|
} else if (state.getBlock() == Blocks.LAVA|| state.getBlock() == Blocks.FLOWING_LAVA) {
|
|
stack = new FluidStack(FluidRegistry.LAVA, Fluid.BUCKET_VOLUME);
|
|
if (doDrain) world.setBlockToAir(pos);
|
|
}
|
|
} else if (state.getBlock() instanceof IFluidBlock) {
|
|
stack = ((IFluidBlock)state.getBlock()).drain(world, pos, doDrain);
|
|
}
|
|
|
|
return stack;
|
|
}
|
|
|
|
}
|