Also skip feature removal for FlatLevelSource

Otherwise, entire layers of specific block types will be removed instead of just the parts that are not supposed to be there.
Pixaurora · Sep 27, 2023 · f291395 · f291395
1 parent 315939a
commit f291395
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/net/pixaurora/janerator/mixin/FeatureMixin.java b/src/main/java/net/pixaurora/janerator/mixin/FeatureMixin.java
@@ -10,6 +10,7 @@
 import net.minecraft.world.level.ChunkPos;
 import net.minecraft.world.level.WorldGenLevel;
 import net.minecraft.world.level.chunk.ChunkGenerator;
+import net.minecraft.world.level.levelgen.FlatLevelSource;
 import net.minecraft.world.level.levelgen.feature.Feature;
 import net.minecraft.world.level.levelgen.feature.configurations.FeatureConfiguration;
 
@@ -28,7 +29,7 @@ public class FeatureMixin<FC extends FeatureConfiguration> {
         BlockPos originPos,
         CallbackInfoReturnable<Boolean> cir
     ) {
-		if (generator.janerator$notMultiGenerating()) {
+		if (generator instanceof FlatLevelSource || generator.janerator$notMultiGenerating()) {
             return;
         }