Rewrite biome coloring to not require rendering mod patches

GregTechCEu · Jan 20, 2025 · d38adba · d38adba
1 parent 3dfd0cb
commit d38adba
Show file tree

Hide file tree

Showing 7 changed files with 56 additions and 158 deletions.
diff --git a/src/main/java/com/gregtechceu/gtceu/client/EnvironmentalHazardClientHandler.java b/src/main/java/com/gregtechceu/gtceu/client/EnvironmentalHazardClientHandler.java
@@ -19,8 +19,8 @@
 import it.unimi.dsi.fastutil.floats.FloatIntPair;
 import lombok.Getter;
 
-import java.util.HashMap;
 import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
 
 @OnlyIn(Dist.CLIENT)
 public class EnvironmentalHazardClientHandler {
@@ -44,8 +44,8 @@ private EnvironmentalHazardClientHandler() {
      * Map of source position to a triple of (trigger, material).
      */
     @Getter
-    private final Map<ChunkPos, EnvironmentalHazardSavedData.HazardZone> hazardZones = new HashMap<>();
-    private final Map<ChunkPos, FloatIntPair> chunkColorCache = new HashMap<>();
+    private final Map<ChunkPos, EnvironmentalHazardSavedData.HazardZone> hazardZones = new ConcurrentHashMap<>();
+    private final Map<ChunkPos, FloatIntPair> chunkColorCache = new ConcurrentHashMap<>();
 
     public void onClientTick() {
         if (!ConfigHolder.INSTANCE.gameplay.environmentalHazards) {

diff --git a/src/main/java/com/gregtechceu/gtceu/core/mixins/BiomeColorsMixin.java b/src/main/java/com/gregtechceu/gtceu/core/mixins/BiomeColorsMixin.java
@@ -0,0 +1,52 @@
+package com.gregtechceu.gtceu.core.mixins;
+
+import com.gregtechceu.gtceu.client.EnvironmentalHazardClientHandler;
+import com.gregtechceu.gtceu.config.ConfigHolder;
+
+import net.minecraft.client.renderer.BiomeColors;
+import net.minecraft.core.SectionPos;
+import net.minecraft.world.level.ChunkPos;
+import net.minecraft.world.level.ColorResolver;
+
+import org.spongepowered.asm.mixin.*;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+@Mixin(BiomeColors.class)
+public class BiomeColorsMixin {
+
+    @Shadow
+    @Final
+    @Mutable
+    public static ColorResolver WATER_COLOR_RESOLVER;
+    @Shadow
+    @Final
+    @Mutable
+    public static ColorResolver GRASS_COLOR_RESOLVER;
+    @Shadow
+    @Final
+    @Mutable
+    public static ColorResolver FOLIAGE_COLOR_RESOLVER;
+
+    @Unique
+    private static ColorResolver gtceu$wrapResolver(ColorResolver resolver) {
+        return (biome, x, z) -> {
+            var originalColor = resolver.getColor(biome, x, z);
+            if (!ConfigHolder.INSTANCE.gameplay.environmentalHazards) {
+                return originalColor;
+            }
+
+            var clientHandler = EnvironmentalHazardClientHandler.INSTANCE;
+            var chunkPos = new ChunkPos(SectionPos.posToSectionCoord(x), SectionPos.posToSectionCoord(z));
+            return clientHandler.colorZone(originalColor, chunkPos);
+        };
+    }
+
+    @Inject(method = "<clinit>", at = @At("RETURN"))
+    private static void addPollutionWrapper(CallbackInfo ci) {
+        WATER_COLOR_RESOLVER = gtceu$wrapResolver(WATER_COLOR_RESOLVER);
+        GRASS_COLOR_RESOLVER = gtceu$wrapResolver(GRASS_COLOR_RESOLVER);
+        FOLIAGE_COLOR_RESOLVER = gtceu$wrapResolver(FOLIAGE_COLOR_RESOLVER);
+    }
+}
diff --git a/src/main/java/com/gregtechceu/gtceu/core/mixins/BlockColorsMixin.java b/src/main/java/com/gregtechceu/gtceu/core/mixins/BlockColorsMixin.java
diff --git a/src/main/java/com/gregtechceu/gtceu/core/mixins/GregTechMixinPlugin.java b/src/main/java/com/gregtechceu/gtceu/core/mixins/GregTechMixinPlugin.java
@@ -31,8 +31,6 @@ public boolean shouldApplyMixin(String targetClassName, String mixinClassName) {
             return MixinPluginShared.isClassFound("mezz.jei.api.IModPlugin");
         } else if (mixinClassName.contains("com.gregtechceu.gtceu.core.mixins.emi")) {
             return MixinPluginShared.isClassFound("dev.emi.emi.api.EmiPlugin");
-        } else if (mixinClassName.contains("com.gregtechceu.gtceu.core.mixins.embeddium")) {
-            return MixinPluginShared.isClassFound("me.jellysquid.mods.sodium.client.SodiumClientMod");
         }
         return true;
     }

diff --git a/src/main/java/com/gregtechceu/gtceu/core/mixins/embeddium/BiomeColorCacheMixin.java b/src/main/java/com/gregtechceu/gtceu/core/mixins/embeddium/BiomeColorCacheMixin.java
diff --git a/src/main/java/com/gregtechceu/gtceu/core/mixins/embeddium/VanillaColorAdapterMixin.java b/src/main/java/com/gregtechceu/gtceu/core/mixins/embeddium/VanillaColorAdapterMixin.java
diff --git a/src/main/resources/gtceu.mixins.json b/src/main/resources/gtceu.mixins.json
@@ -6,7 +6,7 @@
     "compatibilityLevel": "JAVA_17",
     "plugin": "com.gregtechceu.gtceu.core.mixins.GregTechMixinPlugin",
     "client": [
-        "BlockColorsMixin",
+        "BiomeColorsMixin",
         "BlockModelMixin",
         "ClientLevelAccessor",
         "GuiGraphicsAccessor",
@@ -64,8 +64,6 @@
         "TagManagerMixin",
         "TagValueAccessor",
         "ae2.GenericStackInvAccessor",
-        "embeddium.BiomeColorCacheMixin",
-        "embeddium.VanillaColorAdapterMixin",
         "emi.FluidEmiStackMixin",
         "jei.FluidHelperMixin",
         "ldlib.SyncUtilsMixin",