1 files changed, 21 insertions, 6 deletions
diff --git a/src/main/java/com/anthonyhilyard/iceberg/registry/RendererRegistrar.java b/src/main/java/com/anthonyhilyard/iceberg/registry/RendererRegistrar.java
index e77fd1f..1b0a053 100644
--- a/src/main/java/com/anthonyhilyard/iceberg/registry/RendererRegistrar.java
+++ b/src/main/java/com/anthonyhilyard/iceberg/registry/RendererRegistrar.java
@@ -1,21 +1,36 @@
 package com.anthonyhilyard.iceberg.registry;
 
-import net.minecraft.entity.Entity;
-import net.minecraftforge.fml.client.registry.IRenderFactory;
-import net.minecraftforge.fml.client.registry.RenderingRegistry;
+import java.util.HashMap;
+import java.util.Map;
+
+import net.minecraft.world.entity.Entity;
+import net.minecraft.world.entity.EntityType;
+import net.minecraft.client.renderer.entity.EntityRendererProvider;
+import net.minecraftforge.client.event.EntityRenderersEvent;
 
 public abstract class RendererRegistrar
 {
-	protected static <T extends Entity> void registerRenderer(String name, IRenderFactory<? super T> renderFactory)
+	private static Map<EntityType<? extends Entity>, EntityRendererProvider<?>> entityRenderers = new HashMap<>();
+
+	protected static <T extends Entity> void registerRenderer(String name, EntityRendererProvider<T> rendererProvider)
 	{
 		if (AutoRegistry.isEntityTypeRegistered(name))
 		{
-			// Register the rendering handler.
-			RenderingRegistry.registerEntityRenderingHandler(AutoRegistry.<T>getEntityType(name), renderFactory);
+			// Store this renderer provider.
+			entityRenderers.put(AutoRegistry.getEntityType(name), rendererProvider);
 		}
 		else
 		{
 			throw new RuntimeException("Tried to register a renderer for an unregistered entity type!  Make sure you register renderers after entities.");
 		}
 	}
+
+	@SuppressWarnings({"unchecked", "unused"})
+	private <T extends Entity> void onEntityCreation(EntityRenderersEvent.RegisterRenderers event)
+	{
+		for (EntityType<? extends Entity> entityType : entityRenderers.keySet())
+		{
+			event.registerEntityRenderer((EntityType<T>)entityType, (EntityRendererProvider<T>)entityRenderers.get(entityType));
+		}
+	}
 }