Allow Model Loading API to inject and load extra models

Author: embeddedt

fabric/src/main/java/org/embeddedt/modernfix/fabric/mixin/perf/dynamic_resources/ModelBakeryMixin.java

@@ -87,7 +87,6 @@ private void replaceTopLevelBakedModels(ModelBakery bakery, BlockColors val) {
         // we can handle recursion in getModel without issues
         fabric_enableGetOrLoadModelGuard = false;
         this.blockColors = val;
-        this.ignoreModelLoad = true;
         this.loadedBakedModels = CacheBuilder.newBuilder()
                 .expireAfterAccess(ModelBakeryHelpers.MAX_MODEL_LIFETIME_SECS, TimeUnit.SECONDS)
                 .maximumSize(ModelBakeryHelpers.MAX_BAKED_MODEL_COUNT)
@@ -119,6 +118,11 @@ public UnbakedModel put(ResourceLocation key, UnbakedModel value) {
         this.bakedTopLevelModels = new DynamicBakedModelProvider((ModelBakery)(Object)this, bakedCache);
     }
 
+    @Inject(method = "<init>", at = @At(value = "INVOKE", target = "Lnet/minecraft/util/profiling/ProfilerFiller;popPush(Ljava/lang/String;)V", ordinal = 0))
+    private void ignoreFutureModelLoads(CallbackInfo ci) {
+        this.ignoreModelLoad = true;
+    }
+
     private <K, V> void onModelRemoved(RemovalNotification<K, V> notification) {
         if(!debugDynamicModelLoading)
             return;
@@ -193,7 +197,10 @@ private void captureGetter(BiFunction<ResourceLocation, Material, TextureAtlasSp
 
     @Redirect(method = "bakeModels", at = @At(value = "INVOKE", target = "Ljava/util/Map;keySet()Ljava/util/Set;"))
     private Set<ResourceLocation> skipBake(Map<ResourceLocation, UnbakedModel> instance) {
-        return Collections.emptySet();
+        Set<ResourceLocation> modelSet = new HashSet<>(instance.keySet());
+        if(modelSet.size() > 0)
+            ModernFix.LOGGER.info("Early baking {} models", modelSet.size());
+        return modelSet;
     }
 
     /**