Make emulated registry easier to debug

Author: embeddedt

common/src/main/java/org/embeddedt/modernfix/common/mixin/perf/dynamic_resources/ModelManagerMixin.java

@@ -1,6 +1,5 @@
 package org.embeddedt.modernfix.common.mixin.perf.dynamic_resources;
 
-import com.google.common.collect.ForwardingMap;
 import com.google.common.collect.Maps;
 import com.llamalad7.mixinextras.sugar.Local;
 import net.minecraft.client.Minecraft;
@@ -65,30 +64,7 @@ private CompletableFuture<BlockStateModelLoader.LoadedModels> deferBlockStateLoa
     private static Map<BlockState, BlockStateModel> createBlockStateToModelDispatch(Map<BlockState, BlockStateModel> map, BlockStateModel missingModel) {
         var dynamicProvider = Objects.requireNonNull(DynamicModelProvider.currentReloadingModelProvider.get());
 
-        var dynamicRegistry = dynamicProvider.getTopLevelEmulatedRegistry();
-
-        return new ForwardingMap<>() {
-            @Override
-            protected Map<BlockState, BlockStateModel> delegate() {
-                return dynamicRegistry;
-            }
-
-            @Override
-            public BlockStateModel get(Object key) {
-                BlockStateModel result;
-                if (key instanceof IModelHoldingBlockState state) {
-                    result = state.mfix$getModel();
-                    if (result != null) {
-                        return result;
-                    }
-                }
-                result = dynamicRegistry.getOrDefault(key, dynamicProvider.getMissingBakedModel());
-                if (key instanceof IModelHoldingBlockState state) {
-                    state.mfix$setModel(result);
-                }
-                return result;
-            }
-        };
+        return dynamicProvider.getFastTopLevelEmulatedRegistry();
     }
 
     @Redirect(method = "reload", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/resources/model/ClientItemInfoLoader;scheduleLoad(Lnet/minecraft/server/packs/resources/ResourceManager;Ljava/util/concurrent/Executor;)Ljava/util/concurrent/CompletableFuture;"))

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

@@ -3,6 +3,7 @@
 import com.google.common.cache.CacheBuilder;
 import com.google.common.cache.CacheLoader;
 import com.google.common.cache.LoadingCache;
+import com.google.common.collect.ForwardingMap;
 import com.google.common.collect.Iterators;
 import com.google.common.collect.Maps;
 import com.google.gson.JsonObject;
@@ -40,6 +41,7 @@
 import net.minecraft.world.level.block.state.BlockState;
 import net.minecraft.world.level.block.state.StateDefinition;
 import org.embeddedt.modernfix.ModernFix;
+import org.embeddedt.modernfix.duck.IModelHoldingBlockState;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
@@ -202,6 +204,33 @@ public Map<BlockState, BlockStateModel> getTopLevelEmulatedRegistry() {
         return new EmulatedRegistry<>(BlockState.class, this.loadedBakedModels, BlockStateSet::instance, this.mrlModelOverrides);
     }
 
+    public Map<BlockState, BlockStateModel> getFastTopLevelEmulatedRegistry() {
+        var dynamicRegistry = getTopLevelEmulatedRegistry();
+
+        return new ForwardingMap<>() {
+            @Override
+            protected Map<BlockState, BlockStateModel> delegate() {
+                return dynamicRegistry;
+            }
+
+            @Override
+            public BlockStateModel get(Object key) {
+                BlockStateModel result;
+                if (key instanceof IModelHoldingBlockState state) {
+                    result = state.mfix$getModel();
+                    if (result != null) {
+                        return result;
+                    }
+                }
+                result = dynamicRegistry.getOrDefault(key, getMissingBakedModel());
+                if (key instanceof IModelHoldingBlockState state) {
+                    state.mfix$setModel(result);
+                }
+                return result;
+            }
+        };
+    }
+
     /*
     public Map<ResourceLocation, BakedModel> getStandaloneEmulatedRegistry() {
         return new EmulatedRegistry<>(ResourceLocation.class, this.loadedStandaloneModels, Set::of, this.standaloneModelOverrides);