128 lines
4.9 KiB
Java
128 lines
4.9 KiB
Java
package net.minecraft.world.level.biome;
|
|
|
|
import com.google.common.base.Suppliers;
|
|
import com.google.common.collect.ImmutableList;
|
|
import com.google.common.collect.Lists;
|
|
import com.mojang.logging.LogUtils;
|
|
import com.mojang.serialization.MapCodec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Set;
|
|
import java.util.function.Supplier;
|
|
import java.util.stream.Collectors;
|
|
import net.minecraft.Util;
|
|
import net.minecraft.core.Holder;
|
|
import net.minecraft.core.HolderGetter;
|
|
import net.minecraft.core.HolderSet;
|
|
import net.minecraft.resources.ResourceKey;
|
|
import net.minecraft.world.level.levelgen.GenerationStep;
|
|
import net.minecraft.world.level.levelgen.carver.ConfiguredWorldCarver;
|
|
import net.minecraft.world.level.levelgen.feature.ConfiguredFeature;
|
|
import net.minecraft.world.level.levelgen.feature.Feature;
|
|
import net.minecraft.world.level.levelgen.placement.PlacedFeature;
|
|
import org.slf4j.Logger;
|
|
|
|
public class BiomeGenerationSettings {
|
|
private static final Logger LOGGER = LogUtils.getLogger();
|
|
public static final BiomeGenerationSettings EMPTY = new BiomeGenerationSettings(HolderSet.direct(), List.of());
|
|
public static final MapCodec<BiomeGenerationSettings> CODEC = RecordCodecBuilder.mapCodec(
|
|
instance -> instance.group(
|
|
ConfiguredWorldCarver.LIST_CODEC
|
|
.promotePartial(Util.prefix("Carver: ", LOGGER::error))
|
|
.fieldOf("carvers")
|
|
.forGetter(biomeGenerationSettings -> biomeGenerationSettings.carvers),
|
|
PlacedFeature.LIST_OF_LISTS_CODEC
|
|
.promotePartial(Util.prefix("Features: ", LOGGER::error))
|
|
.fieldOf("features")
|
|
.forGetter(biomeGenerationSettings -> biomeGenerationSettings.features)
|
|
)
|
|
.apply(instance, BiomeGenerationSettings::new)
|
|
);
|
|
private final HolderSet<ConfiguredWorldCarver<?>> carvers;
|
|
private final List<HolderSet<PlacedFeature>> features;
|
|
private final Supplier<List<ConfiguredFeature<?, ?>>> flowerFeatures;
|
|
private final Supplier<Set<PlacedFeature>> featureSet;
|
|
|
|
BiomeGenerationSettings(HolderSet<ConfiguredWorldCarver<?>> carvers, List<HolderSet<PlacedFeature>> features) {
|
|
this.carvers = carvers;
|
|
this.features = features;
|
|
this.flowerFeatures = Suppliers.memoize(
|
|
() -> (List<ConfiguredFeature<?, ?>>)features.stream()
|
|
.flatMap(HolderSet::stream)
|
|
.map(Holder::value)
|
|
.flatMap(PlacedFeature::getFeatures)
|
|
.filter(configuredFeature -> configuredFeature.feature() == Feature.FLOWER)
|
|
.collect(ImmutableList.toImmutableList())
|
|
);
|
|
this.featureSet = Suppliers.memoize(() -> (Set<PlacedFeature>)features.stream().flatMap(HolderSet::stream).map(Holder::value).collect(Collectors.toSet()));
|
|
}
|
|
|
|
public Iterable<Holder<ConfiguredWorldCarver<?>>> getCarvers() {
|
|
return this.carvers;
|
|
}
|
|
|
|
public List<ConfiguredFeature<?, ?>> getFlowerFeatures() {
|
|
return (List<ConfiguredFeature<?, ?>>)this.flowerFeatures.get();
|
|
}
|
|
|
|
public List<HolderSet<PlacedFeature>> features() {
|
|
return this.features;
|
|
}
|
|
|
|
public boolean hasFeature(PlacedFeature feature) {
|
|
return ((Set)this.featureSet.get()).contains(feature);
|
|
}
|
|
|
|
public static class Builder extends BiomeGenerationSettings.PlainBuilder {
|
|
private final HolderGetter<PlacedFeature> placedFeatures;
|
|
private final HolderGetter<ConfiguredWorldCarver<?>> worldCarvers;
|
|
|
|
public Builder(HolderGetter<PlacedFeature> placedFeatures, HolderGetter<ConfiguredWorldCarver<?>> worldCarvers) {
|
|
this.placedFeatures = placedFeatures;
|
|
this.worldCarvers = worldCarvers;
|
|
}
|
|
|
|
public BiomeGenerationSettings.Builder addFeature(GenerationStep.Decoration decoration, ResourceKey<PlacedFeature> feature) {
|
|
this.addFeature(decoration.ordinal(), this.placedFeatures.getOrThrow(feature));
|
|
return this;
|
|
}
|
|
|
|
public BiomeGenerationSettings.Builder addCarver(ResourceKey<ConfiguredWorldCarver<?>> carver) {
|
|
this.addCarver(this.worldCarvers.getOrThrow(carver));
|
|
return this;
|
|
}
|
|
}
|
|
|
|
public static class PlainBuilder {
|
|
private final List<Holder<ConfiguredWorldCarver<?>>> carvers = new ArrayList();
|
|
private final List<List<Holder<PlacedFeature>>> features = new ArrayList();
|
|
|
|
public BiomeGenerationSettings.PlainBuilder addFeature(GenerationStep.Decoration decoration, Holder<PlacedFeature> feature) {
|
|
return this.addFeature(decoration.ordinal(), feature);
|
|
}
|
|
|
|
public BiomeGenerationSettings.PlainBuilder addFeature(int step, Holder<PlacedFeature> feature) {
|
|
this.addFeatureStepsUpTo(step);
|
|
((List)this.features.get(step)).add(feature);
|
|
return this;
|
|
}
|
|
|
|
public BiomeGenerationSettings.PlainBuilder addCarver(Holder<ConfiguredWorldCarver<?>> carver) {
|
|
this.carvers.add(carver);
|
|
return this;
|
|
}
|
|
|
|
private void addFeatureStepsUpTo(int step) {
|
|
while (this.features.size() <= step) {
|
|
this.features.add(Lists.newArrayList());
|
|
}
|
|
}
|
|
|
|
public BiomeGenerationSettings build() {
|
|
return new BiomeGenerationSettings(
|
|
HolderSet.direct(this.carvers), (List<HolderSet<PlacedFeature>>)this.features.stream().map(HolderSet::direct).collect(ImmutableList.toImmutableList())
|
|
);
|
|
}
|
|
}
|
|
}
|