228 lines
10 KiB
Java
228 lines
10 KiB
Java
|
package net.minecraft.world.level.levelgen.flat;
|
||
|
|
|
||
|
|
import com.google.common.collect.Lists;
|
||
|
|
import com.mojang.logging.LogUtils;
|
||
|
|
import com.mojang.serialization.Codec;
|
||
|
|
import com.mojang.serialization.DataResult;
|
||
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
||
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder.Instance;
|
||
|
|
import java.util.List;
|
||
|
|
import java.util.Optional;
|
||
|
|
import java.util.function.Function;
|
||
|
|
import net.minecraft.core.Holder;
|
||
|
|
import net.minecraft.core.HolderGetter;
|
||
|
|
import net.minecraft.core.HolderSet;
|
||
|
|
import net.minecraft.core.RegistryCodecs;
|
||
|
|
import net.minecraft.core.registries.Registries;
|
||
|
|
import net.minecraft.data.worldgen.placement.MiscOverworldPlacements;
|
||
|
|
import net.minecraft.data.worldgen.placement.PlacementUtils;
|
||
|
|
import net.minecraft.resources.RegistryOps;
|
||
|
|
import net.minecraft.world.level.biome.Biome;
|
||
|
|
import net.minecraft.world.level.biome.BiomeGenerationSettings;
|
||
|
|
import net.minecraft.world.level.biome.Biomes;
|
||
|
|
import net.minecraft.world.level.block.Blocks;
|
||
|
|
import net.minecraft.world.level.block.state.BlockState;
|
||
|
|
import net.minecraft.world.level.dimension.DimensionType;
|
||
|
|
import net.minecraft.world.level.levelgen.GenerationStep;
|
||
|
|
import net.minecraft.world.level.levelgen.Heightmap;
|
||
|
|
import net.minecraft.world.level.levelgen.feature.Feature;
|
||
|
|
import net.minecraft.world.level.levelgen.feature.configurations.LayerConfiguration;
|
||
|
|
import net.minecraft.world.level.levelgen.placement.PlacedFeature;
|
||
|
|
import net.minecraft.world.level.levelgen.structure.BuiltinStructureSets;
|
||
|
|
import net.minecraft.world.level.levelgen.structure.StructureSet;
|
||
|
|
import org.slf4j.Logger;
|
||
|
|
|
||
|
|
public class FlatLevelGeneratorSettings {
|
||
|
|
private static final Logger LOGGER = LogUtils.getLogger();
|
||
|
|
public static final Codec<FlatLevelGeneratorSettings> CODEC = RecordCodecBuilder.<FlatLevelGeneratorSettings>create(
|
||
|
|
p_209800_ -> p_209800_.group(
|
||
|
|
RegistryCodecs.homogeneousList(Registries.STRUCTURE_SET).lenientOptionalFieldOf("structure_overrides").forGetter(p_209812_ -> p_209812_.structureOverrides),
|
||
|
|
FlatLayerInfo.CODEC.listOf().fieldOf("layers").forGetter(FlatLevelGeneratorSettings::getLayersInfo),
|
||
|
|
Codec.BOOL.fieldOf("lakes").orElse(false).forGetter(p_161912_ -> p_161912_.addLakes),
|
||
|
|
Codec.BOOL.fieldOf("features").orElse(false).forGetter(p_209809_ -> p_209809_.decoration),
|
||
|
|
Biome.CODEC.lenientOptionalFieldOf("biome").orElseGet(Optional::empty).forGetter(p_209807_ -> Optional.of(p_209807_.biome)),
|
||
|
|
RegistryOps.retrieveElement(Biomes.PLAINS),
|
||
|
|
RegistryOps.retrieveElement(MiscOverworldPlacements.LAKE_LAVA_UNDERGROUND),
|
||
|
|
RegistryOps.retrieveElement(MiscOverworldPlacements.LAKE_LAVA_SURFACE)
|
||
|
|
)
|
||
|
|
.apply(p_209800_, FlatLevelGeneratorSettings::new)
|
||
|
|
)
|
||
|
|
.comapFlatMap(FlatLevelGeneratorSettings::validateHeight, Function.identity())
|
||
|
|
.stable();
|
||
|
|
private final Optional<HolderSet<StructureSet>> structureOverrides;
|
||
|
|
private final List<FlatLayerInfo> layersInfo = Lists.newArrayList();
|
||
|
|
private final Holder<Biome> biome;
|
||
|
|
private final List<BlockState> layers;
|
||
|
|
private boolean voidGen;
|
||
|
|
private boolean decoration;
|
||
|
|
private boolean addLakes;
|
||
|
|
private final List<Holder<PlacedFeature>> lakes;
|
||
|
|
|
||
|
|
private static DataResult<FlatLevelGeneratorSettings> validateHeight(FlatLevelGeneratorSettings p_161906_) {
|
||
|
|
int i = p_161906_.layersInfo.stream().mapToInt(FlatLayerInfo::getHeight).sum();
|
||
|
|
return i > DimensionType.Y_SIZE
|
||
|
|
? DataResult.error(() -> "Sum of layer heights is > " + DimensionType.Y_SIZE, p_161906_)
|
||
|
|
: DataResult.success(p_161906_);
|
||
|
|
}
|
||
|
|
|
||
|
|
private FlatLevelGeneratorSettings(
|
||
|
|
Optional<HolderSet<StructureSet>> p_256456_,
|
||
|
|
List<FlatLayerInfo> p_255826_,
|
||
|
|
boolean p_255740_,
|
||
|
|
boolean p_255726_,
|
||
|
|
Optional<Holder<Biome>> p_256292_,
|
||
|
|
Holder.Reference<Biome> p_255964_,
|
||
|
|
Holder<PlacedFeature> p_256419_,
|
||
|
|
Holder<PlacedFeature> p_255710_
|
||
|
|
) {
|
||
|
|
this(p_256456_, getBiome(p_256292_, p_255964_), List.of(p_256419_, p_255710_));
|
||
|
|
if (p_255740_) {
|
||
|
|
this.setAddLakes();
|
||
|
|
}
|
||
|
|
|
||
|
|
if (p_255726_) {
|
||
|
|
this.setDecoration();
|
||
|
|
}
|
||
|
|
|
||
|
|
this.layersInfo.addAll(p_255826_);
|
||
|
|
this.updateLayers();
|
||
|
|
}
|
||
|
|
|
||
|
|
private static Holder<Biome> getBiome(Optional<? extends Holder<Biome>> p_256142_, Holder<Biome> p_256475_) {
|
||
|
|
if (p_256142_.isEmpty()) {
|
||
|
|
LOGGER.error("Unknown biome, defaulting to plains");
|
||
|
|
return p_256475_;
|
||
|
|
} else {
|
||
|
|
return (Holder<Biome>)p_256142_.get();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public FlatLevelGeneratorSettings(Optional<HolderSet<StructureSet>> p_256029_, Holder<Biome> p_256190_, List<Holder<PlacedFeature>> p_255960_) {
|
||
|
|
this.structureOverrides = p_256029_;
|
||
|
|
this.biome = p_256190_;
|
||
|
|
this.layers = Lists.newArrayList();
|
||
|
|
this.lakes = p_255960_;
|
||
|
|
}
|
||
|
|
|
||
|
|
public FlatLevelGeneratorSettings withBiomeAndLayers(List<FlatLayerInfo> p_256587_, Optional<HolderSet<StructureSet>> p_256500_, Holder<Biome> p_256598_) {
|
||
|
|
FlatLevelGeneratorSettings flatlevelgeneratorsettings = new FlatLevelGeneratorSettings(p_256500_, p_256598_, this.lakes);
|
||
|
|
|
||
|
|
for (FlatLayerInfo flatlayerinfo : p_256587_) {
|
||
|
|
flatlevelgeneratorsettings.layersInfo.add(new FlatLayerInfo(flatlayerinfo.getHeight(), flatlayerinfo.getBlockState().getBlock()));
|
||
|
|
flatlevelgeneratorsettings.updateLayers();
|
||
|
|
}
|
||
|
|
|
||
|
|
if (this.decoration) {
|
||
|
|
flatlevelgeneratorsettings.setDecoration();
|
||
|
|
}
|
||
|
|
|
||
|
|
if (this.addLakes) {
|
||
|
|
flatlevelgeneratorsettings.setAddLakes();
|
||
|
|
}
|
||
|
|
|
||
|
|
return flatlevelgeneratorsettings;
|
||
|
|
}
|
||
|
|
|
||
|
|
public void setDecoration() {
|
||
|
|
this.decoration = true;
|
||
|
|
}
|
||
|
|
|
||
|
|
public void setAddLakes() {
|
||
|
|
this.addLakes = true;
|
||
|
|
}
|
||
|
|
|
||
|
|
public BiomeGenerationSettings adjustGenerationSettings(Holder<Biome> p_226295_) {
|
||
|
|
if (!p_226295_.equals(this.biome)) {
|
||
|
|
return p_226295_.value().getGenerationSettings();
|
||
|
|
} else {
|
||
|
|
BiomeGenerationSettings biomegenerationsettings = this.getBiome().value().getGenerationSettings();
|
||
|
|
BiomeGenerationSettings.PlainBuilder biomegenerationsettings$plainbuilder = new BiomeGenerationSettings.PlainBuilder();
|
||
|
|
if (this.addLakes) {
|
||
|
|
for (Holder<PlacedFeature> holder : this.lakes) {
|
||
|
|
biomegenerationsettings$plainbuilder.addFeature(GenerationStep.Decoration.LAKES, holder);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
boolean flag = (!this.voidGen || p_226295_.is(Biomes.THE_VOID)) && this.decoration;
|
||
|
|
if (flag) {
|
||
|
|
List<HolderSet<PlacedFeature>> list = biomegenerationsettings.features();
|
||
|
|
|
||
|
|
for (int i = 0; i < list.size(); i++) {
|
||
|
|
if (i != GenerationStep.Decoration.UNDERGROUND_STRUCTURES.ordinal()
|
||
|
|
&& i != GenerationStep.Decoration.SURFACE_STRUCTURES.ordinal()
|
||
|
|
&& (!this.addLakes || i != GenerationStep.Decoration.LAKES.ordinal())) {
|
||
|
|
for (Holder<PlacedFeature> holder1 : list.get(i)) {
|
||
|
|
biomegenerationsettings$plainbuilder.addFeature(i, holder1);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
List<BlockState> list1 = this.getLayers();
|
||
|
|
|
||
|
|
for (int j = 0; j < list1.size(); j++) {
|
||
|
|
BlockState blockstate = list1.get(j);
|
||
|
|
if (!Heightmap.Types.MOTION_BLOCKING.isOpaque().test(blockstate)) {
|
||
|
|
list1.set(j, null);
|
||
|
|
biomegenerationsettings$plainbuilder.addFeature(
|
||
|
|
GenerationStep.Decoration.TOP_LAYER_MODIFICATION, PlacementUtils.inlinePlaced(Feature.FILL_LAYER, new LayerConfiguration(j, blockstate))
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return biomegenerationsettings$plainbuilder.build();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public Optional<HolderSet<StructureSet>> structureOverrides() {
|
||
|
|
return this.structureOverrides;
|
||
|
|
}
|
||
|
|
|
||
|
|
public Holder<Biome> getBiome() {
|
||
|
|
return this.biome;
|
||
|
|
}
|
||
|
|
|
||
|
|
public List<FlatLayerInfo> getLayersInfo() {
|
||
|
|
return this.layersInfo;
|
||
|
|
}
|
||
|
|
|
||
|
|
public List<BlockState> getLayers() {
|
||
|
|
return this.layers;
|
||
|
|
}
|
||
|
|
|
||
|
|
public void updateLayers() {
|
||
|
|
this.layers.clear();
|
||
|
|
|
||
|
|
for (FlatLayerInfo flatlayerinfo : this.layersInfo) {
|
||
|
|
for (int i = 0; i < flatlayerinfo.getHeight(); i++) {
|
||
|
|
this.layers.add(flatlayerinfo.getBlockState());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
this.voidGen = this.layers.stream().allMatch(p_209802_ -> p_209802_.is(Blocks.AIR));
|
||
|
|
}
|
||
|
|
|
||
|
|
public static FlatLevelGeneratorSettings getDefault(
|
||
|
|
HolderGetter<Biome> p_256175_, HolderGetter<StructureSet> p_256081_, HolderGetter<PlacedFeature> p_256484_
|
||
|
|
) {
|
||
|
|
HolderSet<StructureSet> holderset = HolderSet.direct(
|
||
|
|
p_256081_.getOrThrow(BuiltinStructureSets.STRONGHOLDS), p_256081_.getOrThrow(BuiltinStructureSets.VILLAGES)
|
||
|
|
);
|
||
|
|
FlatLevelGeneratorSettings flatlevelgeneratorsettings = new FlatLevelGeneratorSettings(
|
||
|
|
Optional.of(holderset), getDefaultBiome(p_256175_), createLakesList(p_256484_)
|
||
|
|
);
|
||
|
|
flatlevelgeneratorsettings.getLayersInfo().add(new FlatLayerInfo(1, Blocks.BEDROCK));
|
||
|
|
flatlevelgeneratorsettings.getLayersInfo().add(new FlatLayerInfo(2, Blocks.DIRT));
|
||
|
|
flatlevelgeneratorsettings.getLayersInfo().add(new FlatLayerInfo(1, Blocks.GRASS_BLOCK));
|
||
|
|
flatlevelgeneratorsettings.updateLayers();
|
||
|
|
return flatlevelgeneratorsettings;
|
||
|
|
}
|
||
|
|
|
||
|
|
public static Holder<Biome> getDefaultBiome(HolderGetter<Biome> p_256645_) {
|
||
|
|
return p_256645_.getOrThrow(Biomes.PLAINS);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static List<Holder<PlacedFeature>> createLakesList(HolderGetter<PlacedFeature> p_256282_) {
|
||
|
|
return List.of(p_256282_.getOrThrow(MiscOverworldPlacements.LAKE_LAVA_UNDERGROUND), p_256282_.getOrThrow(MiscOverworldPlacements.LAKE_LAVA_SURFACE));
|
||
|
|
}
|
||
|
|
}
|