refactor: slight tweak

Zepalesque · May 23, 2024 · ea62905 · ea62905
1 parent 5bb5c89
commit ea62905
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/src/main/java/net/zepalesque/zenith/Zenith.java b/src/main/java/net/zepalesque/zenith/Zenith.java
@@ -69,7 +69,7 @@ private void commonSetup(final FMLCommonSetupEvent event) {
     }
 
     private void registerDataMaps(RegisterDataMapTypesEvent event) {
-        BiomeTint.MAPS.forEach(event::register);
+        BiomeTints.TINT_REGISTRY.forEach(tint -> tint.register(event));
     }
 
     @Mod.EventBusSubscriber(modid = MODID, bus = Mod.EventBusSubscriber.Bus.MOD, value = Dist.CLIENT)

diff --git a/src/main/java/net/zepalesque/zenith/api/biometint/BiomeTint.java b/src/main/java/net/zepalesque/zenith/api/biometint/BiomeTint.java
@@ -5,6 +5,7 @@
 import net.minecraft.resources.ResourceLocation;
 import net.minecraft.world.level.biome.Biome;
 import net.neoforged.neoforge.registries.datamaps.DataMapType;
+import net.neoforged.neoforge.registries.datamaps.RegisterDataMapTypesEvent;
 
 import java.util.ArrayList;
 import java.util.Collection;
@@ -16,12 +17,9 @@ public class BiomeTint {
     private final DataMapType<Biome, Integer> dataMap;
     private Map<Biome, Integer> tints = null;
     private final int defaultColor;
-
-    public static Collection<DataMapType<Biome, Integer>> MAPS = new ArrayList<>();
-
+
     public BiomeTint(ResourceLocation loc, int defaultColor) {
         DataMapType<Biome, Integer> dataMap = createTintMap(loc);
-        MAPS.add(dataMap);
         this.dataMap = dataMap;
         this.defaultColor = defaultColor;
     }
@@ -56,4 +54,8 @@ public void addTint(Biome biome, int color) {
     public DataMapType<Biome, Integer> getDataMap() {
         return this.dataMap;
     }
+
+    public void register(RegisterDataMapTypesEvent event) {
+        event.register(this.dataMap);
+    }
 }