ARKEntityRegistry.java

package com.arkcraft.module.creature.common.entity;

import java.util.HashMap;
import java.util.Map;

import net.minecraftforge.fml.common.registry.EntityRegistry;

import com.arkcraft.module.core.ARKCraft;
import com.arkcraft.module.creature.common.entity.creature.Creature;
import com.arkcraft.module.creature.common.entity.creature.CreatureRaptor;

/**
 * @author gegy1000
 */
public class ARKEntityRegistry
{
	private static Map<Class<? extends EntityARKCreature>, Creature> creatureMap = new HashMap<Class<? extends EntityARKCreature>, Creature>();

	public static void register()
	{
		registerCreature(new CreatureRaptor());
	}

	public static Creature getCreature(EntityARKCreature dino)
	{
		return creatureMap.get(dino.getClass());
	}

	public static void registerCreature(Creature creature)
	{
		Class<? extends EntityARKCreature> entityClass = creature.getEntityClass();

		creatureMap.put(entityClass, creature);

		int uniqueId = EntityRegistry.findGlobalUniqueEntityId();

		String entityName = creature.getName().replaceAll(" ", "");
		EntityRegistry.registerGlobalEntityID(entityClass, entityName, uniqueId);
		EntityRegistry.registerModEntity(entityClass, entityName, uniqueId, ARKCraft.instance, 256,
				1, true);
	}
}