Update entity data code

src/main/java/uk/insrt/minecraft/inslib/Main.java

@@ -2,7 +2,10 @@ package uk.insrt.minecraft.inslib;
 
 import org.apache.logging.log4j.Logger;
 
+import net.minecraft.client.renderer.entity.Render;
+import net.minecraft.client.renderer.entity.RenderManager;
 import net.minecraftforge.common.MinecraftForge;
+import net.minecraftforge.fml.client.registry.IRenderFactory;
 import net.minecraftforge.fml.common.Mod;
 import net.minecraftforge.fml.common.Mod.EventHandler;
 import net.minecraftforge.fml.common.SidedProxy;
@@ -11,6 +14,8 @@ import net.minecraftforge.fml.common.event.FMLPostInitializationEvent;
 import net.minecraftforge.fml.common.event.FMLPreInitializationEvent;
 import net.minecraftforge.fml.common.eventhandler.EventBus;
 import uk.insrt.minecraft.inslib.events.RenderGameOverlayText;
+import uk.insrt.minecraft.inslib.helpers.TheRegistry;
+import uk.insrt.minecraft.inslib.helpers.objects.EntityData;
 import uk.insrt.minecraft.inslib.proxy.CommonProxy;
 
 @Mod(modid = Main.MODID, name = Main.NAME, version = Main.VERSION)

src/main/java/uk/insrt/minecraft/inslib/helpers/TheRegistry.java

@@ -93,8 +93,12 @@ public class TheRegistry {
     	
     	public static void onEntityRegister() {
     		for (EntityData data : entities) {
-    			EntityRegistry.registerModEntity(new ResourceLocation(data.getRegistryName()), data.getEntityClass(),
-    					data.getName(), data.getId(), data.getModId(), 100, 100, true, 5, 3232);
+    			ResourceLocation registry = new ResourceLocation(data.getRegistryName()); 
+    			EntityRegistry.registerModEntity(registry, data.getEntityClass(),
+    					data.getName(), data.getId(), data.getModId(), 64, 3, true);
+    			if (data.hasEgg()) {
+    				EntityRegistry.registerEgg(registry, data.getEggColourPrimary(), data.getEggColourSecondary());
+    			}
     			RenderingRegistry.registerEntityRenderingHandler(data.getEntityClass(), data.getRenderFactory());
     		}
     	}

src/main/java/uk/insrt/minecraft/inslib/helpers/objects/EntityData.java

@@ -2,6 +2,7 @@ package uk.insrt.minecraft.inslib.helpers.objects;
 
 import net.minecraft.entity.Entity;
 import net.minecraftforge.fml.client.registry.IRenderFactory;
+import uk.insrt.minecraft.inslib.helpers.Colour;
 
 public class EntityData {
 	
@@ -11,12 +12,24 @@ public class EntityData {
 	private Class<? extends Entity> entityClass;
 	private IRenderFactory renderFactory;
 	
+	private boolean egg;
+	private int eggA;
+	private int eggB;
+	
 	public EntityData(int id, String modid, String name, Class<? extends Entity> entityClass, IRenderFactory renderFactory) {
 		this.id = id;
 		this.modid = modid;
 		this.name = name;
 		this.entityClass = entityClass;
 		this.renderFactory = renderFactory;
+		this.egg = false;
+	}
+	
+	public EntityData(int id, String modid, String name, Class<? extends Entity> entityClass, IRenderFactory renderFactory, Colour eggA, Colour eggB) {
+		this(id, modid, name, entityClass, renderFactory);
+		this.egg = true;
+		this.eggA = eggA.toHex();
+		this.eggB = eggB.toHex();
 	}
 	
 	public int getId() {
@@ -43,4 +56,16 @@ public class EntityData {
 		return renderFactory;
 	}
 	
+	public boolean hasEgg() {
+		return egg;
+	}
+	
+	public int getEggColourPrimary() {
+		return eggA;
+	}
+	
+	public int getEggColourSecondary() {
+		return eggB;
+	}
+	
 }