74 lines
3.6 KiB
Java
74 lines
3.6 KiB
Java
|
package net.minecraft.world.level.levelgen.feature;
|
||
|
|
|
||
|
|
import com.mojang.serialization.Codec;
|
||
|
|
import java.util.Optional;
|
||
|
|
import java.util.OptionalInt;
|
||
|
|
import java.util.function.Predicate;
|
||
|
|
import net.minecraft.core.BlockPos;
|
||
|
|
import net.minecraft.core.Direction;
|
||
|
|
import net.minecraft.core.Vec3i;
|
||
|
|
import net.minecraft.util.RandomSource;
|
||
|
|
import net.minecraft.world.level.LevelAccessor;
|
||
|
|
import net.minecraft.world.level.WorldGenLevel;
|
||
|
|
import net.minecraft.world.level.block.Blocks;
|
||
|
|
import net.minecraft.world.level.block.state.BlockState;
|
||
|
|
import net.minecraft.world.level.levelgen.Column;
|
||
|
|
import net.minecraft.world.level.levelgen.feature.configurations.UnderwaterMagmaConfiguration;
|
||
|
|
import net.minecraft.world.level.levelgen.structure.BoundingBox;
|
||
|
|
|
||
|
|
public class UnderwaterMagmaFeature extends Feature<UnderwaterMagmaConfiguration> {
|
||
|
|
public UnderwaterMagmaFeature(Codec<UnderwaterMagmaConfiguration> p_160560_) {
|
||
|
|
super(p_160560_);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public boolean place(FeaturePlaceContext<UnderwaterMagmaConfiguration> p_160569_) {
|
||
|
|
WorldGenLevel worldgenlevel = p_160569_.level();
|
||
|
|
BlockPos blockpos = p_160569_.origin();
|
||
|
|
UnderwaterMagmaConfiguration underwatermagmaconfiguration = p_160569_.config();
|
||
|
|
RandomSource randomsource = p_160569_.random();
|
||
|
|
OptionalInt optionalint = getFloorY(worldgenlevel, blockpos, underwatermagmaconfiguration);
|
||
|
|
if (optionalint.isEmpty()) {
|
||
|
|
return false;
|
||
|
|
} else {
|
||
|
|
BlockPos blockpos1 = blockpos.atY(optionalint.getAsInt());
|
||
|
|
Vec3i vec3i = new Vec3i(underwatermagmaconfiguration.placementRadiusAroundFloor, underwatermagmaconfiguration.placementRadiusAroundFloor, underwatermagmaconfiguration.placementRadiusAroundFloor);
|
||
|
|
BoundingBox boundingbox = BoundingBox.fromCorners(blockpos1.subtract(vec3i), blockpos1.offset(vec3i));
|
||
|
|
return BlockPos.betweenClosedStream(boundingbox)
|
||
|
|
.filter(p_225310_ -> randomsource.nextFloat() < underwatermagmaconfiguration.placementProbabilityPerValidPosition)
|
||
|
|
.filter(p_160584_ -> this.isValidPlacement(worldgenlevel, p_160584_))
|
||
|
|
.mapToInt(p_160579_ -> {
|
||
|
|
worldgenlevel.setBlock(p_160579_, Blocks.MAGMA_BLOCK.defaultBlockState(), 2);
|
||
|
|
return 1;
|
||
|
|
})
|
||
|
|
.sum()
|
||
|
|
> 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private static OptionalInt getFloorY(WorldGenLevel p_160565_, BlockPos p_160566_, UnderwaterMagmaConfiguration p_160567_) {
|
||
|
|
Predicate<BlockState> predicate = p_160586_ -> p_160586_.is(Blocks.WATER);
|
||
|
|
Predicate<BlockState> predicate1 = p_160581_ -> !p_160581_.is(Blocks.WATER);
|
||
|
|
Optional<Column> optional = Column.scan(p_160565_, p_160566_, p_160567_.floorSearchRange, predicate, predicate1);
|
||
|
|
return optional.map(Column::getFloor).orElseGet(OptionalInt::empty);
|
||
|
|
}
|
||
|
|
|
||
|
|
private boolean isValidPlacement(WorldGenLevel p_160575_, BlockPos p_160576_) {
|
||
|
|
if (!this.isWaterOrAir(p_160575_, p_160576_) && !this.isWaterOrAir(p_160575_, p_160576_.below())) {
|
||
|
|
for (Direction direction : Direction.Plane.HORIZONTAL) {
|
||
|
|
if (this.isWaterOrAir(p_160575_, p_160576_.relative(direction))) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return true;
|
||
|
|
} else {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private boolean isWaterOrAir(LevelAccessor p_160562_, BlockPos p_160563_) {
|
||
|
|
BlockState blockstate = p_160562_.getBlockState(p_160563_);
|
||
|
|
return blockstate.is(Blocks.WATER) || blockstate.isAir();
|
||
|
|
}
|
||
|
|
}
|