From f3f1e585dcd6f625ea33513c5ae3512db3184ba8 Mon Sep 17 00:00:00 2001
From: Ben Staddon <bs2609@users.noreply.github.com>
Date: Sun, 29 Jul 2018 02:55:33 +0100
Subject: [PATCH] Add function to lookup model location for an ItemStack
 (#5017)

---
 .../client/ItemModelMesherForge.java          | 35 +++++++++++++++++++
 src/main/resources/forge_at.cfg               |  7 ++++
 2 files changed, 42 insertions(+)

diff --git a/src/main/java/net/minecraftforge/client/ItemModelMesherForge.java b/src/main/java/net/minecraftforge/client/ItemModelMesherForge.java
index 9e6229af04..5afe4ffb64 100644
--- a/src/main/java/net/minecraftforge/client/ItemModelMesherForge.java
+++ b/src/main/java/net/minecraftforge/client/ItemModelMesherForge.java
@@ -26,13 +26,19 @@ import com.google.common.collect.Maps;
 import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
 import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
 
+import net.minecraft.client.renderer.ItemMeshDefinition;
 import net.minecraft.client.renderer.ItemModelMesher;
 import net.minecraft.client.renderer.block.model.IBakedModel;
+import net.minecraft.client.renderer.block.model.ModelBakery;
 import net.minecraft.client.renderer.block.model.ModelManager;
 import net.minecraft.client.renderer.block.model.ModelResourceLocation;
 import net.minecraft.item.Item;
+import net.minecraft.item.ItemStack;
 import net.minecraftforge.registries.IRegistryDelegate;
 
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+
 /**
  * Wrapper around ItemModeMesher that cleans up the internal maps to respect ID remapping.
  */
@@ -47,6 +53,7 @@ public class ItemModelMesherForge extends ItemModelMesher
     }
 
     @Override
+    @Nullable
     protected IBakedModel getItemModel(Item item, int meta)
     {
         Int2ObjectMap<IBakedModel> map = models.get(item.delegate);
@@ -95,4 +102,32 @@ public class ItemModelMesherForge extends ItemModelMesher
             );
         }
     }
+
+    public ModelResourceLocation getLocation(@Nonnull ItemStack stack)
+    {
+        Item item = stack.getItem();
+        ModelResourceLocation location = null;
+
+        Int2ObjectMap<ModelResourceLocation> map = locations.get(item.delegate);
+        if (map != null)
+        {
+            location = map.get(getMetadata(stack));
+        }
+
+        if (location == null)
+        {
+            ItemMeshDefinition definition = shapers.get(item);
+            if (definition != null)
+            {
+                location = definition.getModelLocation(stack);
+            }
+        }
+
+        if (location == null)
+        {
+            location = ModelBakery.MODEL_MISSING;
+        }
+
+        return location;
+    }
 }
diff --git a/src/main/resources/forge_at.cfg b/src/main/resources/forge_at.cfg
index 9aa55e5e69..6152e46c4c 100644
--- a/src/main/resources/forge_at.cfg
+++ b/src/main/resources/forge_at.cfg
@@ -107,6 +107,9 @@ public net.minecraft.client.renderer.block.model.ModelBlock field_178318_c # tex
 public net.minecraft.client.renderer.block.model.ModelBlock field_178315_d # parent
 public net.minecraft.client.renderer.block.model.ModelBlock field_178322_i # ambientOcclusion
 public net.minecraft.client.renderer.block.model.ModelBlock func_187966_f()Ljava/util/List; # getOverrides
+
+# ModelBakery
+public net.minecraft.client.renderer.block.model.ModelBakery field_177604_a # MODEL_MISSING
 protected net.minecraft.client.renderer.block.model.ModelBakery field_177602_b # LOCATIONS_BUILTIN_TEXTURES
 protected net.minecraft.client.renderer.block.model.ModelBakery field_177598_f # resourceManager
 protected net.minecraft.client.renderer.block.model.ModelBakery field_177599_g # sprites
@@ -136,6 +139,10 @@ protected net.minecraft.client.renderer.block.model.ModelBakery func_177595_c()V
 protected net.minecraft.client.renderer.block.model.ModelBakery func_188637_e()V # loadMultipartVariantModels
 protected net.minecraft.client.renderer.block.model.ModelBakery func_188638_a(Lnet/minecraft/client/renderer/block/model/ModelResourceLocation;Lnet/minecraft/client/renderer/block/model/VariantList;)V # loadVariantList
 #public net.minecraft.client.renderer.block.model.WeightedBakedModel field_177565_b # models
+
+# ItemModelMesher
+protected net.minecraft.client.renderer.ItemModelMesher field_178092_c # shapers
+
 # EnumFacing
 public net.minecraft.util.EnumFacing field_82609_l # VALUES
 public net.minecraft.util.EnumFacing field_176754_o # HORIZONTALS