Fix the entity shader hook

This commit is contained in:
Take Weiland committed 2016-02-14 15:25:11 +01:00
1 parent 4d33fe6b26
commit b875274c4f
2 files changed
+8 -5

No files matched your search

@@ -575,12 +575,15 @@ public class ForgeHooksClient
return Optional.of(new TRSRTransformation(matrix));
}
public static void loadEntityShader(Class<? extends Entity> entityClass, EntityRenderer entityRenderer)
public static void loadEntityShader(Entity entity, EntityRenderer entityRenderer)
{
ResourceLocation shader = ClientRegistry.getEntityShader(entityClass);
if (shader != null)
if (entity != null)
{
entityRenderer.loadShader(shader);
ResourceLocation shader = ClientRegistry.getEntityShader(entity.getClass());
if (shader != null)
{
entityRenderer.loadShader(shader);
}
}
}
}