Rewrite dynamic resources once again

Author: embeddedt

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

@@ -1,23 +1,32 @@
 package org.embeddedt.modernfix.common.mixin.perf.dynamic_resources;
 
 import com.google.common.collect.ImmutableList;
+import it.unimi.dsi.fastutil.objects.Object2IntMap;
+import net.minecraft.client.Minecraft;
 import net.minecraft.client.renderer.block.model.BlockModel;
+import net.minecraft.client.resources.model.AtlasSet;
 import net.minecraft.client.resources.model.BakedModel;
 import net.minecraft.client.resources.model.BlockStateModelLoader;
+import net.minecraft.client.resources.model.ModelBakery;
 import net.minecraft.client.resources.model.ModelManager;
+import net.minecraft.client.resources.model.ModelResourceLocation;
 import net.minecraft.resources.ResourceLocation;
 import net.minecraft.server.packs.resources.ResourceManager;
+import net.minecraft.util.profiling.ProfilerFiller;
 import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.block.state.BlockState;
 import net.minecraft.world.level.block.state.StateDefinition;
 import org.embeddedt.modernfix.annotation.ClientOnlyMixin;
-import org.embeddedt.modernfix.duck.IExtendedModelManager;
+import org.embeddedt.modernfix.dynamicresources.DynamicModelProvider;
 import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Overwrite;
 import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.Unique;
 import org.spongepowered.asm.mixin.injection.At;
 import org.spongepowered.asm.mixin.injection.Inject;
 import org.spongepowered.asm.mixin.injection.Redirect;
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
 
 import java.util.HashMap;
 import java.util.Map;
@@ -26,9 +35,16 @@
 
 @Mixin(ModelManager.class)
 @ClientOnlyMixin
-public class ModelManagerMixin implements IExtendedModelManager {
+public class ModelManagerMixin {
     @Shadow private Map<ResourceLocation, BakedModel> bakedRegistry;
 
+    @Shadow private BakedModel missingModel;
+    @Unique
+    private DynamicModelProvider mfix$modelProvider;
+    
+    @Unique
+    private Map<ResourceLocation, AtlasSet.StitchResult> mfix$stitchResults;
+
     @Inject(method = "<init>", at = @At("RETURN"))
     private void injectDummyBakedRegistry(CallbackInfo ci) {
         if(this.bakedRegistry == null) {
@@ -51,8 +67,31 @@ private ImmutableList<BlockState> skipCollection(StateDefinition<Block, BlockSta
         return ImmutableList.of();
     }
 
-    @Override
-    public void mfix$tick() {
+    @Inject(method = "loadModels", at = @At("HEAD"))
+    private void saveStitchResults(ProfilerFiller profilerFiller, Map<ResourceLocation, AtlasSet.StitchResult> map, ModelBakery modelBakery, Object2IntMap<BlockState> object2IntMap, CallbackInfoReturnable<?> cir) {
+        this.mfix$stitchResults = map;
+    }
 
+    @Inject(method = "apply", at = @At("RETURN"))
+    private void createModelProvider(CallbackInfo ci) {
+        this.mfix$modelProvider = new DynamicModelProvider(
+                null, // TODO
+                this.missingModel,
+                Minecraft.getInstance().getResourceManager(),
+                this.mfix$stitchResults
+        );
+    }
+
+    /**
+     * @author embeddedt
+     * @reason use dynamic model system
+     */
+    @Overwrite
+    public BakedModel getModel(ModelResourceLocation modelLocation) {
+        if(this.mfix$modelProvider != null) {
+            return this.mfix$modelProvider.getModel(modelLocation);
+        } else {
+            return this.missingModel;
+        }
     }
 }