Fix getting missing models for overridden Item registry entires
This commit is contained in:
parent
fdfaa5036d
commit
b0edb23dba
|
@ -19,26 +19,25 @@
|
|||
|
||||
package net.minecraftforge.client;
|
||||
|
||||
import java.util.IdentityHashMap;
|
||||
import java.util.Map;
|
||||
|
||||
import com.google.common.collect.Maps;
|
||||
|
||||
import gnu.trove.map.hash.TIntObjectHashMap;
|
||||
import gnu.trove.procedure.TIntObjectProcedure;
|
||||
import net.minecraft.client.renderer.ItemModelMesher;
|
||||
import net.minecraft.client.renderer.block.model.IBakedModel;
|
||||
import net.minecraft.client.renderer.block.model.ModelManager;
|
||||
import net.minecraft.client.renderer.block.model.ModelResourceLocation;
|
||||
import net.minecraft.item.Item;
|
||||
import net.minecraftforge.registries.IRegistryDelegate;
|
||||
|
||||
/**
|
||||
* Wrapper around ItemModeMesher that cleans up the internal maps to respect ID remapping.
|
||||
*/
|
||||
public class ItemModelMesherForge extends ItemModelMesher
|
||||
{
|
||||
IdentityHashMap<Item, TIntObjectHashMap<ModelResourceLocation>> locations = Maps.newIdentityHashMap();
|
||||
IdentityHashMap<Item, TIntObjectHashMap<IBakedModel>> models = Maps.newIdentityHashMap();
|
||||
Map<IRegistryDelegate<Item>, TIntObjectHashMap<ModelResourceLocation>> locations = Maps.newHashMap();
|
||||
Map<IRegistryDelegate<Item>, TIntObjectHashMap<IBakedModel>> models = Maps.newHashMap();
|
||||
|
||||
public ItemModelMesherForge(ModelManager manager)
|
||||
{
|
||||
|
@ -48,24 +47,25 @@ public class ItemModelMesherForge extends ItemModelMesher
|
|||
@Override
|
||||
protected IBakedModel getItemModel(Item item, int meta)
|
||||
{
|
||||
TIntObjectHashMap<IBakedModel> map = models.get(item);
|
||||
TIntObjectHashMap<IBakedModel> map = models.get(item.delegate);
|
||||
return map == null ? null : map.get(meta);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void register(Item item, int meta, ModelResourceLocation location)
|
||||
{
|
||||
TIntObjectHashMap<ModelResourceLocation> locs = locations.get(item);
|
||||
TIntObjectHashMap<IBakedModel> mods = models.get(item);
|
||||
IRegistryDelegate<Item> key = item.delegate;
|
||||
TIntObjectHashMap<ModelResourceLocation> locs = locations.get(key);
|
||||
TIntObjectHashMap<IBakedModel> mods = models.get(key);
|
||||
if (locs == null)
|
||||
{
|
||||
locs = new TIntObjectHashMap<ModelResourceLocation>();
|
||||
locations.put(item, locs);
|
||||
locations.put(key, locs);
|
||||
}
|
||||
if (mods == null)
|
||||
{
|
||||
mods = new TIntObjectHashMap<IBakedModel>();
|
||||
models.put(item, mods);
|
||||
models.put(key, mods);
|
||||
}
|
||||
locs.put(meta, location);
|
||||
mods.put(meta, getModelManager().getModel(location));
|
||||
|
@ -75,7 +75,7 @@ public class ItemModelMesherForge extends ItemModelMesher
|
|||
public void rebuildCache()
|
||||
{
|
||||
final ModelManager manager = this.getModelManager();
|
||||
for (Map.Entry<Item, TIntObjectHashMap<ModelResourceLocation>> e : locations.entrySet())
|
||||
for (Map.Entry<IRegistryDelegate<Item>, TIntObjectHashMap<ModelResourceLocation>> e : locations.entrySet())
|
||||
{
|
||||
TIntObjectHashMap<IBakedModel> mods = models.get(e.getKey());
|
||||
if (mods != null)
|
||||
|
|
Loading…
Reference in New Issue