Merge 1.16 into 1.18

Author: embeddedt

common/src/main/java/org/embeddedt/modernfix/dynamicresources/DynamicBakedModelProvider.java

@@ -2,6 +2,7 @@
 
 import com.google.common.collect.ImmutableSet;
 import com.mojang.math.Transformation;
+import it.unimi.dsi.fastutil.objects.Object2ObjectMaps;
 import it.unimi.dsi.fastutil.objects.Object2ObjectOpenHashMap;
 import net.minecraft.client.renderer.block.model.BakedQuad;
 import net.minecraft.client.renderer.block.model.ItemOverrides;
@@ -38,7 +39,8 @@ public class DynamicBakedModelProvider implements Map<ResourceLocation, BakedMod
     public static DynamicBakedModelProvider currentInstance = null;
     private final ModelBakery bakery;
     private final Map<Triple<ResourceLocation, Transformation, Boolean>, BakedModel> bakedCache;
-    private final Map<ResourceLocation, BakedModel> permanentOverrides;
+    private volatile Map<ResourceLocation, BakedModel> permanentOverridesView = null;
+    private final Map<ResourceLocation, BakedModel> permanentOverridesMutable;
     private BakedModel missingModel;
     private static final BakedModel SENTINEL = new BakedModel() {
         @Override
@@ -85,7 +87,7 @@ public ItemOverrides getOverrides() {
     public DynamicBakedModelProvider(ModelBakery bakery, Map<Triple<ResourceLocation, Transformation, Boolean>, BakedModel> cache) {
         this.bakery = bakery;
         this.bakedCache = cache;
-        this.permanentOverrides = Collections.synchronizedMap(new Object2ObjectOpenHashMap<>());
+        this.permanentOverridesMutable = new Object2ObjectOpenHashMap<>();
         if(currentInstance == null)
             currentInstance = this;
     }
@@ -107,14 +109,27 @@ public boolean isEmpty() {
         return bakedCache.isEmpty();
     }
 
+    private Map<ResourceLocation, BakedModel> getPermanentOverrides() {
+        Map<ResourceLocation, BakedModel> map = permanentOverridesView;
+        if(map == null) {
+            synchronized (this) {
+                map = permanentOverridesView;
+                if(map == null) {
+                    permanentOverridesView = map = Object2ObjectMaps.unmodifiable(new Object2ObjectOpenHashMap<>(permanentOverridesMutable));
+                }
+            }
+        }
+        return map;
+    }
+
     @Override
     public boolean containsKey(Object o) {
-        return permanentOverrides.getOrDefault(o, SENTINEL) != null;
+        return getPermanentOverrides().getOrDefault(o, SENTINEL) != null;
     }
 
     @Override
     public boolean containsValue(Object o) {
-        return permanentOverrides.containsValue(o) || bakedCache.containsValue(o);
+        return getPermanentOverrides().containsValue(o) || bakedCache.containsValue(o);
     }
 
     private static boolean isVanillaTopLevelModel(ResourceLocation location) {
@@ -140,7 +155,7 @@ private static boolean isVanillaTopLevelModel(ResourceLocation location) {
 
     @Override
     public BakedModel get(Object o) {
-        BakedModel model = permanentOverrides.getOrDefault(o, SENTINEL);
+        BakedModel model = getPermanentOverrides().getOrDefault(o, SENTINEL);
         if(model != SENTINEL)
             return model;
         else {
@@ -156,15 +171,19 @@ public BakedModel get(Object o) {
             if(model == missingModel) {
                 // to correctly emulate the original map, we return null for missing models, unless they are top-level
                 model = isVanillaTopLevelModel((ResourceLocation)o) ? model : null;
-                permanentOverrides.put((ResourceLocation) o, model);
+                this.put((ResourceLocation) o, model);
             }
             return model;
         }
     }
 
     @Override
     public BakedModel put(ResourceLocation resourceLocation, BakedModel bakedModel) {
-        BakedModel m = permanentOverrides.put(resourceLocation, bakedModel);
+        BakedModel m;
+        synchronized (this) {
+            m = permanentOverridesMutable.put(resourceLocation, bakedModel);
+            permanentOverridesView = null;
+        }
         if(m != null)
             return m;
         else
@@ -173,15 +192,22 @@ public BakedModel put(ResourceLocation resourceLocation, BakedModel bakedModel)
 
     @Override
     public BakedModel remove(Object o) {
-        BakedModel m = permanentOverrides.remove(o);
+        BakedModel m;
+        synchronized (this) {
+            m = permanentOverridesMutable.remove(o);
+            permanentOverridesView = null;
+        }
         if(m != null)
             return m;
         return bakedCache.remove(vanillaKey(o));
     }
 
     @Override
     public void putAll(@NotNull Map<? extends ResourceLocation, ? extends BakedModel> map) {
-        permanentOverrides.putAll(map);
+        synchronized (this) {
+            permanentOverridesMutable.putAll(map);
+            permanentOverridesView = null;
+        }
     }
 
     @Override
@@ -210,23 +236,27 @@ public Set<Entry<ResourceLocation, BakedModel>> entrySet() {
     @Nullable
     @Override
     public BakedModel replace(ResourceLocation key, BakedModel value) {
-        BakedModel existingOverride = permanentOverrides.get(key);
-        // as long as no valid override was put in (null can mean unable to load model, so we treat as invalid), replace
-        // the model
-        if(existingOverride == null)
-            return this.put(key, value);
-        else
-            return existingOverride;
+        synchronized (this) {
+            BakedModel existingOverride = permanentOverridesMutable.get(key);
+            // as long as no valid override was put in (null can mean unable to load model, so we treat as invalid), replace
+            // the model
+            if(existingOverride == null)
+                return this.put(key, value);
+            else
+                return existingOverride;
+        }
     }
 
     @Override
     public void replaceAll(BiFunction<? super ResourceLocation, ? super BakedModel, ? extends BakedModel> function) {
-        Set<ResourceLocation> overridenLocations = permanentOverrides.keySet();
-        permanentOverrides.replaceAll(function);
+        synchronized (this) {
+            permanentOverridesMutable.replaceAll(function);
+            permanentOverridesView = null;
+        }
         boolean uvLock = BlockModelRotation.X0_Y0.isUvLocked();
         Transformation rotation = BlockModelRotation.X0_Y0.getRotation();
         bakedCache.replaceAll((loc, oldModel) -> {
-            if(loc.getMiddle() != rotation || loc.getRight() != uvLock || overridenLocations.contains(loc.getLeft()))
+            if(loc.getMiddle() != rotation || loc.getRight() != uvLock || getPermanentOverrides().containsKey(loc.getLeft()))
                 return oldModel;
             else
                 return function.apply(loc.getLeft(), oldModel);