Mesa gen now uses the useStoneTag check

Signed-off-by: CorgiTaco <>

src/main/java/net/oriondevcorgitaco/unearthed/world/feature/naturalgenerators/NaturalGenerator.java

@@ -86,7 +86,6 @@ public boolean generate(StructureWorldAccess world, ChunkGenerator chunkGenerato
 
                         ConfigBlockReader reader = ConfigBlockReader.blocksFromConfig.get((int) ((clampedValue / 2.0 + 0.5) * ConfigBlockReader.blocksFromConfig.size()));
 
-
                         BlockState mutableState = world.getBlockState(mutable);
 
                         if (useStoneTag(world, mutable))

src/main/java/net/oriondevcorgitaco/unearthed/world/feature/naturalgenerators/TrueMesaGenerator.java

@@ -11,6 +11,7 @@
 import net.minecraft.world.gen.chunk.ChunkGenerator;
 import net.minecraft.world.gen.feature.DefaultFeatureConfig;
 import net.minecraft.world.gen.feature.Feature;
+import net.oriondevcorgitaco.unearthed.Unearthed;
 import net.oriondevcorgitaco.unearthed.util.RegistrationHelper;
 
 import java.util.Random;
@@ -35,7 +36,7 @@ public boolean generate(StructureWorldAccess world, ChunkGenerator chunkGenerato
 
                 for (int y = 0; y < topY; y++) {
                     setStrataLayerBlock(y);
-                    if (world.getBlockState(mutable).getBlock().isIn(BlockTags.BASE_STONE_OVERWORLD))
+                    if (useStoneTag(world, mutable))
                         world.setBlockState(mutable, mesaBlockState, 2);
 
                     mutable.move(Direction.UP);
@@ -45,6 +46,15 @@ public boolean generate(StructureWorldAccess world, ChunkGenerator chunkGenerato
         return true;
     }
 
+    public static boolean useStoneTag(StructureWorldAccess world, BlockPos mutable) {
+        boolean stoneTag = Unearthed.UE_CONFIG.generation.stoneTag;
+        if (stoneTag)
+            return world.getBlockState(mutable).isIn(BlockTags.BASE_STONE_OVERWORLD);
+        else
+            return world.getBlockState(mutable) == Blocks.STONE.getDefaultState();
+
+    }
+
     public void setStrataLayerBlock(int yPos) {
         if (yPos % 4 == 0)
             mesaBlockState = Blocks.TERRACOTTA.getDefaultState();