231 lines
11 KiB
Java
231 lines
11 KiB
Java
package org.betterx.betterend.util;
|
|
|
|
import org.betterx.bclib.blocks.BaseDoublePlantBlock;
|
|
import org.betterx.bclib.blocks.BaseVineBlock;
|
|
import org.betterx.bclib.blocks.StalactiteBlock;
|
|
import org.betterx.bclib.util.BlocksHelper;
|
|
import org.betterx.betterend.blocks.BlueVineBlock;
|
|
import org.betterx.betterend.blocks.basis.FurBlock;
|
|
import org.betterx.betterend.registry.EndBlocks;
|
|
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.core.BlockPos.MutableBlockPos;
|
|
import net.minecraft.core.Direction;
|
|
import net.minecraft.core.Registry;
|
|
import net.minecraft.core.registries.Registries;
|
|
import net.minecraft.resources.ResourceLocation;
|
|
import net.minecraft.world.level.LevelAccessor;
|
|
import net.minecraft.world.level.block.Blocks;
|
|
import net.minecraft.world.level.block.FallingBlock;
|
|
import net.minecraft.world.level.block.state.BlockState;
|
|
import net.minecraft.world.level.dimension.DimensionType;
|
|
|
|
import com.google.common.collect.Sets;
|
|
|
|
import java.util.Set;
|
|
import java.util.stream.IntStream;
|
|
|
|
public class BlockFixer {
|
|
private static final BlockState AIR = Blocks.AIR.defaultBlockState();
|
|
private static final BlockState WATER = Blocks.WATER.defaultBlockState();
|
|
|
|
public static void fixBlocks(LevelAccessor level, BlockPos start, BlockPos end) {
|
|
final Registry<DimensionType> registry = level.registryAccess()
|
|
.registryOrThrow(Registries.DIMENSION_TYPE);
|
|
final ResourceLocation dimKey = registry.getKey(level.dimensionType());
|
|
if (dimKey != null && "world_blender".equals(dimKey.getNamespace())) {
|
|
return;
|
|
}
|
|
final Set<BlockPos> doubleCheck = Sets.newConcurrentHashSet();
|
|
final int dx = end.getX() - start.getX() + 1;
|
|
final int dz = end.getZ() - start.getZ() + 1;
|
|
final int count = dx * dz;
|
|
final int minY = Math.max(start.getY(), level.getMinBuildHeight());
|
|
final int maxY = Math.min(end.getY(), level.getMaxBuildHeight());
|
|
IntStream.range(0, count).forEach(index -> {
|
|
MutableBlockPos POS = new MutableBlockPos();
|
|
POS.setX((index % dx) + start.getX());
|
|
POS.setZ((index / dx) + start.getZ());
|
|
BlockState state;
|
|
for (int y = minY; y <= maxY; y++) {
|
|
POS.setY(y);
|
|
state = level.getBlockState(POS);
|
|
|
|
if (state.getBlock() instanceof FurBlock) {
|
|
doubleCheck.add(POS.immutable());
|
|
}
|
|
// Liquids
|
|
else if (!state.getFluidState().isEmpty()) {
|
|
if (!state.canSurvive(level, POS)) {
|
|
setWithoutUpdate(level, POS, WATER);
|
|
POS.setY(POS.getY() - 1);
|
|
state = level.getBlockState(POS);
|
|
while (!state.canSurvive(level, POS)) {
|
|
state = state.getFluidState().isEmpty() ? AIR : WATER;
|
|
setWithoutUpdate(level, POS, state);
|
|
POS.setY(POS.getY() - 1);
|
|
state = level.getBlockState(POS);
|
|
}
|
|
}
|
|
POS.setY(y - 1);
|
|
if (level.isEmptyBlock(POS)) {
|
|
POS.setY(y);
|
|
while (!level.getFluidState(POS).isEmpty()) {
|
|
setWithoutUpdate(level, POS, AIR);
|
|
POS.setY(POS.getY() + 1);
|
|
}
|
|
continue;
|
|
}
|
|
for (Direction dir : BlocksHelper.HORIZONTAL) {
|
|
if (level.isEmptyBlock(POS.relative(dir))) {
|
|
try {
|
|
level.scheduleTick(POS, state.getFluidState().getType(), 0);
|
|
} catch (Exception e) {
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
} else if (state.is(EndBlocks.SMARAGDANT_CRYSTAL) || state.is(EndBlocks.BUDDING_SMARAGDANT_CRYSTAL)) {
|
|
POS.setY(POS.getY() - 1);
|
|
if (level.isEmptyBlock(POS)) {
|
|
POS.setY(POS.getY() + 1);
|
|
while (state.is(EndBlocks.SMARAGDANT_CRYSTAL) || state.is(EndBlocks.BUDDING_SMARAGDANT_CRYSTAL)) {
|
|
setWithoutUpdate(level, POS, AIR);
|
|
POS.setY(POS.getY() + 1);
|
|
state = level.getBlockState(POS);
|
|
}
|
|
}
|
|
} else if (state.getBlock() instanceof StalactiteBlock) {
|
|
if (!state.canSurvive(level, POS)) {
|
|
if (level.getBlockState(POS.above()).getBlock() instanceof StalactiteBlock) {
|
|
while (state.getBlock() instanceof StalactiteBlock) {
|
|
setWithoutUpdate(level, POS, AIR);
|
|
POS.setY(POS.getY() + 1);
|
|
state = level.getBlockState(POS);
|
|
}
|
|
} else {
|
|
while (state.getBlock() instanceof StalactiteBlock) {
|
|
setWithoutUpdate(level, POS, AIR);
|
|
POS.setY(POS.getY() - 1);
|
|
state = level.getBlockState(POS);
|
|
}
|
|
}
|
|
}
|
|
} else if (state.is(EndBlocks.CAVE_PUMPKIN)) {
|
|
if (!level.getBlockState(POS.above()).is(EndBlocks.CAVE_PUMPKIN_SEED)) {
|
|
setWithoutUpdate(level, POS, AIR);
|
|
}
|
|
} else if (!state.canSurvive(level, POS)) {
|
|
// Chorus
|
|
if (state.is(Blocks.CHORUS_PLANT)) {
|
|
Set<BlockPos> ends = Sets.newHashSet();
|
|
Set<BlockPos> add = Sets.newHashSet();
|
|
ends.add(POS.immutable());
|
|
|
|
for (int i = 0; i < 64 && !ends.isEmpty(); i++) {
|
|
ends.forEach((pos) -> {
|
|
setWithoutUpdate(level, pos, AIR);
|
|
for (Direction dir : BlocksHelper.HORIZONTAL) {
|
|
BlockPos p = pos.relative(dir);
|
|
BlockState st = level.getBlockState(p);
|
|
if ((st.is(Blocks.CHORUS_PLANT) || st.is(Blocks.CHORUS_FLOWER)) && !st.canSurvive(
|
|
level,
|
|
p
|
|
)) {
|
|
add.add(p);
|
|
}
|
|
}
|
|
BlockPos p = pos.above();
|
|
BlockState st = level.getBlockState(p);
|
|
if ((st.is(Blocks.CHORUS_PLANT) || st.is(Blocks.CHORUS_FLOWER)) && !st.canSurvive(
|
|
level,
|
|
p
|
|
)) {
|
|
add.add(p);
|
|
}
|
|
});
|
|
ends.clear();
|
|
ends.addAll(add);
|
|
add.clear();
|
|
}
|
|
}
|
|
// Vines
|
|
else if (state.getBlock() instanceof BaseVineBlock) {
|
|
while (level.getBlockState(POS).getBlock() instanceof BaseVineBlock) {
|
|
setWithoutUpdate(level, POS, AIR);
|
|
POS.setY(POS.getY() - 1);
|
|
}
|
|
}
|
|
// Falling blocks
|
|
else if (state.getBlock() instanceof FallingBlock) {
|
|
BlockState falling = state;
|
|
|
|
POS.setY(POS.getY() - 1);
|
|
state = level.getBlockState(POS);
|
|
|
|
int ray = BlocksHelper.downRayRep(level, POS.immutable(), 64);
|
|
if (ray > 32) {
|
|
setWithoutUpdate(level, POS, Blocks.END_STONE.defaultBlockState());
|
|
if (level.getRandom().nextBoolean()) {
|
|
POS.setY(POS.getY() - 1);
|
|
state = level.getBlockState(POS);
|
|
setWithoutUpdate(level, POS, Blocks.END_STONE.defaultBlockState());
|
|
}
|
|
} else {
|
|
POS.setY(y);
|
|
BlockState replacement = AIR;
|
|
for (Direction dir : BlocksHelper.HORIZONTAL) {
|
|
state = level.getBlockState(POS.relative(dir));
|
|
if (!state.getFluidState().isEmpty()) {
|
|
replacement = state;
|
|
break;
|
|
}
|
|
}
|
|
setWithoutUpdate(level, POS, replacement);
|
|
POS.setY(y - ray);
|
|
setWithoutUpdate(level, POS, falling);
|
|
}
|
|
}
|
|
// Blocks without support
|
|
else {
|
|
// Blue Vine
|
|
if (state.getBlock() instanceof BlueVineBlock) {
|
|
while (state.is(EndBlocks.BLUE_VINE) || state.is(EndBlocks.BLUE_VINE_LANTERN) || state.is(
|
|
EndBlocks.BLUE_VINE_FUR)) {
|
|
setWithoutUpdate(level, POS, AIR);
|
|
POS.setY(POS.getY() + 1);
|
|
state = level.getBlockState(POS);
|
|
}
|
|
}
|
|
// Double plants
|
|
if (state.getBlock() instanceof BaseDoublePlantBlock) {
|
|
setWithoutUpdate(level, POS, AIR);
|
|
POS.setY(POS.getY() + 1);
|
|
setWithoutUpdate(level, POS, AIR);
|
|
}
|
|
// Other blocks
|
|
else {
|
|
setWithoutUpdate(level, POS, getAirOrFluid(state));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
});
|
|
|
|
doubleCheck.forEach((pos) -> {
|
|
if (!level.getBlockState(pos).canSurvive(level, pos)) {
|
|
setWithoutUpdate(level, pos, AIR);
|
|
}
|
|
});
|
|
}
|
|
|
|
private static BlockState getAirOrFluid(BlockState state) {
|
|
return state.getFluidState().isEmpty() ? AIR : state.getFluidState().createLegacyBlock();
|
|
}
|
|
|
|
private static void setWithoutUpdate(LevelAccessor world, BlockPos pos, BlockState state) {
|
|
synchronized (world) {
|
|
BlocksHelper.setWithoutUpdate(world, pos, state);
|
|
}
|
|
}
|
|
}
|