fixed ross128b & ross128ba oregen (#3127)

2 files changed, 28 insertions, 0 deletions

diff --git a/src/main/java/bartworks/system/oregen/BWWorldGenRoss128b.java b/src/main/java/bartworks/system/oregen/BWWorldGenRoss128b.java
index 0af1a4a993..2c996c4668 100644
--- a/src/main/java/bartworks/system/oregen/BWWorldGenRoss128b.java
+++ b/src/main/java/bartworks/system/oregen/BWWorldGenRoss128b.java
@@ -27,9 +27,11 @@ import static gregtech.api.enums.Materials.Uraninite;
 import net.minecraft.block.Block;
 import net.minecraft.init.Blocks;
 import net.minecraft.util.StatCollector;
+import net.minecraft.world.World;
 
 import bartworks.common.configs.ConfigHandler;
 import bartworks.system.material.WerkstoffLoader;
+import bwcrossmod.galacticraft.planets.ross128b.WorldProviderRoss128b;
 import gregtech.api.interfaces.ISubTagContainer;
 
 public class BWWorldGenRoss128b extends BWOreLayer {
@@ -49,6 +51,18 @@ public class BWWorldGenRoss128b extends BWOreLayer {
         return StatCollector.translateToLocal("planet.Ross128b");
     }
 
+    @SuppressWarnings("rawtypes")
+    @Override
+    public boolean isGenerationAllowed(World aWorld, Class... aAllowedDimensionTypes) {
+        for (Class<?> clazz : aAllowedDimensionTypes) {
+            if (clazz == WorldProviderRoss128b.class) {
+                return true;
+            }
+        }
+
+        return false;
+    }
+
     public BWWorldGenRoss128b(String aName, boolean aDefault, int aMinY, int aMaxY, int aWeight, int aDensity,
         int aSize, ISubTagContainer top, ISubTagContainer bottom, ISubTagContainer between,
         ISubTagContainer sprinkled) {
diff --git a/src/main/java/bartworks/system/oregen/BWWorldGenRoss128ba.java b/src/main/java/bartworks/system/oregen/BWWorldGenRoss128ba.java
index 3a032af822..d4df2b8340 100644
--- a/src/main/java/bartworks/system/oregen/BWWorldGenRoss128ba.java
+++ b/src/main/java/bartworks/system/oregen/BWWorldGenRoss128ba.java
@@ -24,9 +24,11 @@ import static gregtech.api.enums.Materials.Tetrahedrite;
 
 import net.minecraft.block.Block;
 import net.minecraft.util.StatCollector;
+import net.minecraft.world.World;
 
 import bartworks.common.configs.ConfigHandler;
 import bartworks.system.material.WerkstoffLoader;
+import bwcrossmod.galacticraft.planets.ross128ba.WorldProviderRoss128ba;
 import gregtech.api.interfaces.ISubTagContainer;
 
 public class BWWorldGenRoss128ba extends BWOreLayer {
@@ -56,6 +58,18 @@ public class BWWorldGenRoss128ba extends BWOreLayer {
         return StatCollector.translateToLocal("moon.Ross128ba");
     }
 
+    @SuppressWarnings("rawtypes")
+    @Override
+    public boolean isGenerationAllowed(World aWorld, Class... aAllowedDimensionTypes) {
+        for (Class clazz : aAllowedDimensionTypes) {
+            if (clazz == WorldProviderRoss128ba.class) {
+                return true;
+            }
+        }
+
+        return false;
+    }
+
     public static void init_Ores() {
         new BWWorldGenRoss128ba(
             "ore.mix.ross128ba.tib",