50 lines
2.4 KiB
Java
50 lines
2.4 KiB
Java
|
package net.minecraft.world.level.levelgen.feature.stateproviders;
|
||
|
|
|
||
|
|
import com.mojang.serialization.Codec;
|
||
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
||
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder.Instance;
|
||
|
|
import java.util.List;
|
||
|
|
import net.minecraft.core.BlockPos;
|
||
|
|
import net.minecraft.util.RandomSource;
|
||
|
|
import net.minecraft.world.level.WorldGenLevel;
|
||
|
|
import net.minecraft.world.level.block.Block;
|
||
|
|
import net.minecraft.world.level.block.state.BlockState;
|
||
|
|
import net.minecraft.world.level.levelgen.blockpredicates.BlockPredicate;
|
||
|
|
|
||
|
|
public record RuleBasedBlockStateProvider(BlockStateProvider fallback, List<RuleBasedBlockStateProvider.Rule> rules) {
|
||
|
|
public static final Codec<RuleBasedBlockStateProvider> CODEC = RecordCodecBuilder.create(
|
||
|
|
p_225939_ -> p_225939_.group(
|
||
|
|
BlockStateProvider.CODEC.fieldOf("fallback").forGetter(RuleBasedBlockStateProvider::fallback),
|
||
|
|
RuleBasedBlockStateProvider.Rule.CODEC.listOf().fieldOf("rules").forGetter(RuleBasedBlockStateProvider::rules)
|
||
|
|
)
|
||
|
|
.apply(p_225939_, RuleBasedBlockStateProvider::new)
|
||
|
|
);
|
||
|
|
|
||
|
|
public static RuleBasedBlockStateProvider simple(BlockStateProvider p_225941_) {
|
||
|
|
return new RuleBasedBlockStateProvider(p_225941_, List.of());
|
||
|
|
}
|
||
|
|
|
||
|
|
public static RuleBasedBlockStateProvider simple(Block p_225937_) {
|
||
|
|
return simple(BlockStateProvider.simple(p_225937_));
|
||
|
|
}
|
||
|
|
|
||
|
|
public BlockState getState(WorldGenLevel p_225933_, RandomSource p_225934_, BlockPos p_225935_) {
|
||
|
|
for (RuleBasedBlockStateProvider.Rule rulebasedblockstateprovider$rule : this.rules) {
|
||
|
|
if (rulebasedblockstateprovider$rule.ifTrue().test(p_225933_, p_225935_)) {
|
||
|
|
return rulebasedblockstateprovider$rule.then().getState(p_225934_, p_225935_);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return this.fallback.getState(p_225934_, p_225935_);
|
||
|
|
}
|
||
|
|
|
||
|
|
public record Rule(BlockPredicate ifTrue, BlockStateProvider then) {
|
||
|
|
public static final Codec<RuleBasedBlockStateProvider.Rule> CODEC = RecordCodecBuilder.create(
|
||
|
|
p_225956_ -> p_225956_.group(
|
||
|
|
BlockPredicate.CODEC.fieldOf("if_true").forGetter(RuleBasedBlockStateProvider.Rule::ifTrue),
|
||
|
|
BlockStateProvider.CODEC.fieldOf("then").forGetter(RuleBasedBlockStateProvider.Rule::then)
|
||
|
|
)
|
||
|
|
.apply(p_225956_, RuleBasedBlockStateProvider.Rule::new)
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|