298 lines
9.7 KiB
Java
298 lines
9.7 KiB
Java
package net.minecraft.world.level.block;
|
|
|
|
import com.mojang.serialization.MapCodec;
|
|
import java.util.Arrays;
|
|
import java.util.Collection;
|
|
import java.util.EnumSet;
|
|
import java.util.Map;
|
|
import java.util.Objects;
|
|
import java.util.Set;
|
|
import java.util.function.Function;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.core.Direction;
|
|
import net.minecraft.util.RandomSource;
|
|
import net.minecraft.world.item.context.BlockPlaceContext;
|
|
import net.minecraft.world.level.BlockGetter;
|
|
import net.minecraft.world.level.Level;
|
|
import net.minecraft.world.level.LevelReader;
|
|
import net.minecraft.world.level.ScheduledTickAccess;
|
|
import net.minecraft.world.level.block.state.BlockBehaviour;
|
|
import net.minecraft.world.level.block.state.BlockState;
|
|
import net.minecraft.world.level.block.state.StateDefinition;
|
|
import net.minecraft.world.level.block.state.properties.BlockStateProperties;
|
|
import net.minecraft.world.level.block.state.properties.BooleanProperty;
|
|
import net.minecraft.world.level.block.state.properties.Property;
|
|
import net.minecraft.world.level.material.FluidState;
|
|
import net.minecraft.world.level.material.Fluids;
|
|
import net.minecraft.world.phys.shapes.CollisionContext;
|
|
import net.minecraft.world.phys.shapes.Shapes;
|
|
import net.minecraft.world.phys.shapes.VoxelShape;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
public class MultifaceBlock extends Block implements SimpleWaterloggedBlock {
|
|
public static final MapCodec<MultifaceBlock> CODEC = simpleCodec(MultifaceBlock::new);
|
|
public static final BooleanProperty WATERLOGGED = BlockStateProperties.WATERLOGGED;
|
|
private static final Map<Direction, BooleanProperty> PROPERTY_BY_DIRECTION = PipeBlock.PROPERTY_BY_DIRECTION;
|
|
protected static final Direction[] DIRECTIONS = Direction.values();
|
|
private final Function<BlockState, VoxelShape> shapes;
|
|
private final boolean canRotate;
|
|
private final boolean canMirrorX;
|
|
private final boolean canMirrorZ;
|
|
|
|
@Override
|
|
protected MapCodec<? extends MultifaceBlock> codec() {
|
|
return CODEC;
|
|
}
|
|
|
|
public MultifaceBlock(BlockBehaviour.Properties properties) {
|
|
super(properties);
|
|
this.registerDefaultState(getDefaultMultifaceState(this.stateDefinition));
|
|
this.shapes = this.makeShapes();
|
|
this.canRotate = Direction.Plane.HORIZONTAL.stream().allMatch(this::isFaceSupported);
|
|
this.canMirrorX = Direction.Plane.HORIZONTAL.stream().filter(Direction.Axis.X).filter(this::isFaceSupported).count() % 2L == 0L;
|
|
this.canMirrorZ = Direction.Plane.HORIZONTAL.stream().filter(Direction.Axis.Z).filter(this::isFaceSupported).count() % 2L == 0L;
|
|
}
|
|
|
|
private Function<BlockState, VoxelShape> makeShapes() {
|
|
Map<Direction, VoxelShape> map = Shapes.rotateAll(Block.boxZ(16.0, 0.0, 1.0));
|
|
return this.getShapeForEachState(blockState -> {
|
|
VoxelShape voxelShape = Shapes.empty();
|
|
|
|
for (Direction direction : DIRECTIONS) {
|
|
if (hasFace(blockState, direction)) {
|
|
voxelShape = Shapes.or(voxelShape, (VoxelShape)map.get(direction));
|
|
}
|
|
}
|
|
|
|
return voxelShape.isEmpty() ? Shapes.block() : voxelShape;
|
|
}, new Property[]{WATERLOGGED});
|
|
}
|
|
|
|
public static Set<Direction> availableFaces(BlockState state) {
|
|
if (!(state.getBlock() instanceof MultifaceBlock)) {
|
|
return Set.of();
|
|
} else {
|
|
Set<Direction> set = EnumSet.noneOf(Direction.class);
|
|
|
|
for (Direction direction : Direction.values()) {
|
|
if (hasFace(state, direction)) {
|
|
set.add(direction);
|
|
}
|
|
}
|
|
|
|
return set;
|
|
}
|
|
}
|
|
|
|
public static Set<Direction> unpack(byte packedDirections) {
|
|
Set<Direction> set = EnumSet.noneOf(Direction.class);
|
|
|
|
for (Direction direction : Direction.values()) {
|
|
if ((packedDirections & (byte)(1 << direction.ordinal())) > 0) {
|
|
set.add(direction);
|
|
}
|
|
}
|
|
|
|
return set;
|
|
}
|
|
|
|
public static byte pack(Collection<Direction> directions) {
|
|
byte b = 0;
|
|
|
|
for (Direction direction : directions) {
|
|
b = (byte)(b | 1 << direction.ordinal());
|
|
}
|
|
|
|
return b;
|
|
}
|
|
|
|
protected boolean isFaceSupported(Direction face) {
|
|
return true;
|
|
}
|
|
|
|
@Override
|
|
protected void createBlockStateDefinition(StateDefinition.Builder<Block, BlockState> builder) {
|
|
for (Direction direction : DIRECTIONS) {
|
|
if (this.isFaceSupported(direction)) {
|
|
builder.add(getFaceProperty(direction));
|
|
}
|
|
}
|
|
|
|
builder.add(WATERLOGGED);
|
|
}
|
|
|
|
@Override
|
|
protected BlockState updateShape(
|
|
BlockState state,
|
|
LevelReader level,
|
|
ScheduledTickAccess scheduledTickAccess,
|
|
BlockPos pos,
|
|
Direction direction,
|
|
BlockPos neighborPos,
|
|
BlockState neighborState,
|
|
RandomSource random
|
|
) {
|
|
if ((Boolean)state.getValue(WATERLOGGED)) {
|
|
scheduledTickAccess.scheduleTick(pos, Fluids.WATER, Fluids.WATER.getTickDelay(level));
|
|
}
|
|
|
|
if (!hasAnyFace(state)) {
|
|
return Blocks.AIR.defaultBlockState();
|
|
} else {
|
|
return hasFace(state, direction) && !canAttachTo(level, direction, neighborPos, neighborState) ? removeFace(state, getFaceProperty(direction)) : state;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected FluidState getFluidState(BlockState state) {
|
|
return state.getValue(WATERLOGGED) ? Fluids.WATER.getSource(false) : super.getFluidState(state);
|
|
}
|
|
|
|
@Override
|
|
protected VoxelShape getShape(BlockState state, BlockGetter level, BlockPos pos, CollisionContext context) {
|
|
return (VoxelShape)this.shapes.apply(state);
|
|
}
|
|
|
|
@Override
|
|
protected boolean canSurvive(BlockState state, LevelReader level, BlockPos pos) {
|
|
boolean bl = false;
|
|
|
|
for (Direction direction : DIRECTIONS) {
|
|
if (hasFace(state, direction)) {
|
|
if (!canAttachTo(level, pos, direction)) {
|
|
return false;
|
|
}
|
|
|
|
bl = true;
|
|
}
|
|
}
|
|
|
|
return bl;
|
|
}
|
|
|
|
@Override
|
|
protected boolean canBeReplaced(BlockState state, BlockPlaceContext useContext) {
|
|
return !useContext.getItemInHand().is(this.asItem()) || hasAnyVacantFace(state);
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
public BlockState getStateForPlacement(BlockPlaceContext context) {
|
|
Level level = context.getLevel();
|
|
BlockPos blockPos = context.getClickedPos();
|
|
BlockState blockState = level.getBlockState(blockPos);
|
|
return (BlockState)Arrays.stream(context.getNearestLookingDirections())
|
|
.map(direction -> this.getStateForPlacement(blockState, level, blockPos, direction))
|
|
.filter(Objects::nonNull)
|
|
.findFirst()
|
|
.orElse(null);
|
|
}
|
|
|
|
public boolean isValidStateForPlacement(BlockGetter level, BlockState state, BlockPos pos, Direction direction) {
|
|
if (this.isFaceSupported(direction) && (!state.is(this) || !hasFace(state, direction))) {
|
|
BlockPos blockPos = pos.relative(direction);
|
|
return canAttachTo(level, direction, blockPos, level.getBlockState(blockPos));
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
@Nullable
|
|
public BlockState getStateForPlacement(BlockState currentState, BlockGetter level, BlockPos pos, Direction lookingDirection) {
|
|
if (!this.isValidStateForPlacement(level, currentState, pos, lookingDirection)) {
|
|
return null;
|
|
} else {
|
|
BlockState blockState;
|
|
if (currentState.is(this)) {
|
|
blockState = currentState;
|
|
} else if (currentState.getFluidState().isSourceOfType(Fluids.WATER)) {
|
|
blockState = this.defaultBlockState().setValue(BlockStateProperties.WATERLOGGED, true);
|
|
} else {
|
|
blockState = this.defaultBlockState();
|
|
}
|
|
|
|
return blockState.setValue(getFaceProperty(lookingDirection), true);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected BlockState rotate(BlockState state, Rotation rotation) {
|
|
return !this.canRotate ? state : this.mapDirections(state, rotation::rotate);
|
|
}
|
|
|
|
@Override
|
|
protected BlockState mirror(BlockState state, Mirror mirror) {
|
|
if (mirror == Mirror.FRONT_BACK && !this.canMirrorX) {
|
|
return state;
|
|
} else {
|
|
return mirror == Mirror.LEFT_RIGHT && !this.canMirrorZ ? state : this.mapDirections(state, mirror::mirror);
|
|
}
|
|
}
|
|
|
|
private BlockState mapDirections(BlockState state, Function<Direction, Direction> directionalFunction) {
|
|
BlockState blockState = state;
|
|
|
|
for (Direction direction : DIRECTIONS) {
|
|
if (this.isFaceSupported(direction)) {
|
|
blockState = blockState.setValue(getFaceProperty((Direction)directionalFunction.apply(direction)), (Boolean)state.getValue(getFaceProperty(direction)));
|
|
}
|
|
}
|
|
|
|
return blockState;
|
|
}
|
|
|
|
public static boolean hasFace(BlockState state, Direction direction) {
|
|
BooleanProperty booleanProperty = getFaceProperty(direction);
|
|
return (Boolean)state.getValueOrElse(booleanProperty, false);
|
|
}
|
|
|
|
public static boolean canAttachTo(BlockGetter level, BlockPos pos, Direction direction) {
|
|
BlockPos blockPos = pos.relative(direction);
|
|
BlockState blockState = level.getBlockState(blockPos);
|
|
return canAttachTo(level, direction, blockPos, blockState);
|
|
}
|
|
|
|
public static boolean canAttachTo(BlockGetter level, Direction direction, BlockPos pos, BlockState state) {
|
|
return Block.isFaceFull(state.getBlockSupportShape(level, pos), direction.getOpposite())
|
|
|| Block.isFaceFull(state.getCollisionShape(level, pos), direction.getOpposite());
|
|
}
|
|
|
|
private static BlockState removeFace(BlockState state, BooleanProperty faceProp) {
|
|
BlockState blockState = state.setValue(faceProp, false);
|
|
return hasAnyFace(blockState) ? blockState : Blocks.AIR.defaultBlockState();
|
|
}
|
|
|
|
public static BooleanProperty getFaceProperty(Direction direction) {
|
|
return (BooleanProperty)PROPERTY_BY_DIRECTION.get(direction);
|
|
}
|
|
|
|
private static BlockState getDefaultMultifaceState(StateDefinition<Block, BlockState> stateDefinition) {
|
|
BlockState blockState = stateDefinition.any().setValue(WATERLOGGED, false);
|
|
|
|
for (BooleanProperty booleanProperty : PROPERTY_BY_DIRECTION.values()) {
|
|
blockState = blockState.trySetValue(booleanProperty, false);
|
|
}
|
|
|
|
return blockState;
|
|
}
|
|
|
|
protected static boolean hasAnyFace(BlockState state) {
|
|
for (Direction direction : DIRECTIONS) {
|
|
if (hasFace(state, direction)) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
private static boolean hasAnyVacantFace(BlockState state) {
|
|
for (Direction direction : DIRECTIONS) {
|
|
if (!hasFace(state, direction)) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|