91 lines
2.6 KiB
Java
91 lines
2.6 KiB
Java
package ru.bclib.mixin.common;
|
|
|
|
import net.minecraft.world.level.biome.BiomeSource;
|
|
import net.minecraft.world.level.levelgen.NoiseGeneratorSettings;
|
|
import net.minecraft.world.level.levelgen.SurfaceRules;
|
|
import net.minecraft.world.level.levelgen.SurfaceRules.RuleSource;
|
|
import org.spongepowered.asm.mixin.Final;
|
|
import org.spongepowered.asm.mixin.Mixin;
|
|
import org.spongepowered.asm.mixin.Mutable;
|
|
import org.spongepowered.asm.mixin.Shadow;
|
|
import ru.bclib.api.biomes.BiomeAPI;
|
|
import ru.bclib.interfaces.SurfaceRuleProvider;
|
|
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.Set;
|
|
import java.util.stream.Collectors;
|
|
|
|
@Mixin(NoiseGeneratorSettings.class)
|
|
public class NoiseGeneratorSettingsMixin implements SurfaceRuleProvider {
|
|
@Mutable
|
|
@Final
|
|
@Shadow
|
|
private SurfaceRules.RuleSource surfaceRule;
|
|
|
|
private SurfaceRules.RuleSource bclib_originalSurfaceRule;
|
|
private Set<BiomeSource> bclib_biomeSources = new HashSet<>();
|
|
|
|
private void bclib_updateCutomRules(){
|
|
bclib_setCustomRules(BiomeAPI.getRuleSources(bclib_biomeSources));
|
|
}
|
|
|
|
@Override
|
|
public void bclib_addBiomeSource(BiomeSource source) {
|
|
bclib_biomeSources.add(source);
|
|
bclib_updateCutomRules();
|
|
}
|
|
|
|
@Override
|
|
public void bclib_clearBiomeSources(){
|
|
bclib_biomeSources.clear();
|
|
bclib_clearCustomRules();
|
|
}
|
|
|
|
private void bclib_clearCustomRules() {
|
|
if (bclib_originalSurfaceRule != null){
|
|
this.surfaceRule = bclib_originalSurfaceRule;
|
|
bclib_originalSurfaceRule = null;
|
|
}
|
|
}
|
|
|
|
private void bclib_setCustomRules(List<RuleSource> rules) {
|
|
if (rules.size()==0){
|
|
bclib_clearCustomRules();
|
|
return;
|
|
}
|
|
|
|
RuleSource org = bclib_getOriginalSurfaceRule();
|
|
if (org instanceof SurfaceRules.SequenceRuleSource sequenceRule){
|
|
List<RuleSource> currentSequence = sequenceRule.sequence();
|
|
rules = rules.stream().filter(r -> currentSequence.indexOf(r)<0).collect(Collectors.toList());
|
|
rules.addAll(sequenceRule.sequence());
|
|
} else {
|
|
rules.add(org);
|
|
}
|
|
|
|
bclib_setSurfaceRule(SurfaceRules.sequence(rules.toArray(new RuleSource[rules.size()])));
|
|
}
|
|
|
|
void bclib_setSurfaceRule(SurfaceRules.RuleSource surfaceRule) {
|
|
if (bclib_originalSurfaceRule == null){
|
|
bclib_originalSurfaceRule = this.surfaceRule;
|
|
}
|
|
this.surfaceRule = surfaceRule;
|
|
}
|
|
|
|
RuleSource bclib_getOriginalSurfaceRule() {
|
|
if (bclib_originalSurfaceRule ==null) {
|
|
return surfaceRule;
|
|
}
|
|
|
|
return bclib_originalSurfaceRule;
|
|
}
|
|
|
|
// @Inject(method = "surfaceRule", at = @At("HEAD"), cancellable = true)
|
|
// private void bclib_surfaceRule(CallbackInfoReturnable<SurfaceRules.RuleSource> info) {
|
|
// if (bclib_surfaceRule != null) {
|
|
// info.setReturnValue(bclib_surfaceRule);
|
|
// }
|
|
// }
|
|
}
|