#include "stdafx.h"
#include "Class.h"
#include "Painting.h"
#include "System.h"
#include "Entity.h"
#include "net.minecraft.world.entity.animal.h"
#include "net.minecraft.world.entity.item.h"
#include "net.minecraft.world.entity.monster.h"
#include "net.minecraft.world.entity.projectile.h"
#include "net.minecraft.world.entity.boss.enderdragon.h"
#include "net.minecraft.world.entity.npc.h"
#include "net.minecraft.world.entity.h"
#include "net.minecraft.world.level.h"
#include "com.mojang.nbt.h"
#include "EntityIO.h"

unordered_map<wstring, entityCreateFn> *EntityIO::idCreateMap = new unordered_map<wstring, entityCreateFn>;
unordered_map<eINSTANCEOF, wstring, eINSTANCEOFKeyHash, eINSTANCEOFKeyEq> *EntityIO::classIdMap = new unordered_map<eINSTANCEOF, wstring, eINSTANCEOFKeyHash, eINSTANCEOFKeyEq>;
unordered_map<int, entityCreateFn> *EntityIO::numCreateMap = new unordered_map<int, entityCreateFn>;
unordered_map<int, eINSTANCEOF> *EntityIO::numClassMap = new unordered_map<int, eINSTANCEOF>;
unordered_map<eINSTANCEOF, int, eINSTANCEOFKeyHash, eINSTANCEOFKeyEq> *EntityIO::classNumMap = new unordered_map<eINSTANCEOF, int, eINSTANCEOFKeyHash, eINSTANCEOFKeyEq>;
unordered_map<wstring, int> *EntityIO::idNumMap = new unordered_map<wstring, int>;
unordered_map<int, EntityIO::SpawnableMobInfo *> EntityIO::idsSpawnableInCreative;

void EntityIO::setId(entityCreateFn createFn, eINSTANCEOF clas, const wstring &id, int idNum)
{
    idCreateMap->insert( unordered_map<wstring, entityCreateFn>::value_type(id, createFn) );
	classIdMap->insert( unordered_map<eINSTANCEOF, wstring, eINSTANCEOFKeyHash, eINSTANCEOFKeyEq>::value_type(clas,id ) );
	numCreateMap->insert( unordered_map<int, entityCreateFn>::value_type(idNum, createFn) );
	numClassMap->insert( unordered_map<int, eINSTANCEOF>::value_type(idNum, clas) );
	classNumMap->insert( unordered_map<eINSTANCEOF, int, eINSTANCEOFKeyHash, eINSTANCEOFKeyEq>::value_type(clas, idNum) );
	idNumMap->insert( unordered_map<wstring, int>::value_type(id, idNum) );
}

void EntityIO::setId(entityCreateFn createFn, eINSTANCEOF clas, const wstring &id, int idNum, eMinecraftColour color1, eMinecraftColour color2, int nameId)
{
	setId(createFn, clas, id, idNum);

	idsSpawnableInCreative.insert( unordered_map<int, SpawnableMobInfo *>::value_type( idNum, new SpawnableMobInfo(idNum, color1, color2, nameId) ) );
}

void EntityIO::staticCtor()
{
	setId(ItemEntity::create, eTYPE_ITEMENTITY,  L"Item", 1);
	setId(ExperienceOrb::create, eTYPE_EXPERIENCEORB, L"XPOrb", 2);
	
    setId(Painting::create, eTYPE_PAINTING, L"Painting", 9);
    setId(Arrow::create, eTYPE_ARROW, L"Arrow", 10);
    setId(Snowball::create, eTYPE_SNOWBALL, L"Snowball", 11);
	setId(Fireball::create, eTYPE_FIREBALL, L"Fireball", 12);
	setId(SmallFireball::create, eTYPE_SMALL_FIREBALL, L"SmallFireball", 13);
	setId(ThrownEnderpearl::create, eTYPE_THROWNENDERPEARL, L"ThrownEnderpearl", 14);
	setId(EyeOfEnderSignal::create, eTYPE_EYEOFENDERSIGNAL, L"EyeOfEnderSignal", 15);
	setId(ThrownPotion::create, eTYPE_THROWNPOTION, L"ThrownPotion", 16);
	setId(ThrownExpBottle::create, eTYPE_THROWNEXPBOTTLE, L"ThrownExpBottle", 17);
	setId(ItemFrame::create, eTYPE_ITEM_FRAME, L"ItemFrame", 18);

    setId(PrimedTnt::create, eTYPE_PRIMEDTNT, L"PrimedTnt", 20);
    setId(FallingTile::create, eTYPE_FALLINGTILE, L"FallingSand", 21);

    setId(Minecart::create, eTYPE_MINECART, L"Minecart", 40);
    setId(Boat::create, eTYPE_BOAT, L"Boat", 41);

    setId(Mob::create, eTYPE_MOB, L"Mob", 48);
    setId(Monster::create, eTYPE_MONSTER, L"Monster", 49);

    setId(Creeper::create, eTYPE_CREEPER, L"Creeper", 50, eMinecraftColour_Mob_Creeper_Colour1, eMinecraftColour_Mob_Creeper_Colour2, IDS_CREEPER);
    setId(Skeleton::create, eTYPE_SKELETON, L"Skeleton", 51, eMinecraftColour_Mob_Skeleton_Colour1, eMinecraftColour_Mob_Skeleton_Colour2, IDS_SKELETON);
    setId(Spider::create, eTYPE_SPIDER, L"Spider", 52, eMinecraftColour_Mob_Spider_Colour1, eMinecraftColour_Mob_Spider_Colour2, IDS_SPIDER);
    setId(Giant::create, eTYPE_GIANT, L"Giant", 53);
    setId(Zombie::create, eTYPE_ZOMBIE, L"Zombie", 54, eMinecraftColour_Mob_Zombie_Colour1, eMinecraftColour_Mob_Zombie_Colour2, IDS_ZOMBIE);
    setId(Slime::create, eTYPE_SLIME, L"Slime", 55, eMinecraftColour_Mob_Slime_Colour1, eMinecraftColour_Mob_Slime_Colour2, IDS_SLIME);
    setId(Ghast::create, eTYPE_GHAST, L"Ghast", 56, eMinecraftColour_Mob_Ghast_Colour1, eMinecraftColour_Mob_Ghast_Colour2, IDS_GHAST);
    setId(PigZombie::create, eTYPE_PIGZOMBIE, L"PigZombie", 57, eMinecraftColour_Mob_PigZombie_Colour1, eMinecraftColour_Mob_PigZombie_Colour2, IDS_PIGZOMBIE);
	setId(EnderMan::create, eTYPE_ENDERMAN, L"Enderman", 58, eMinecraftColour_Mob_Enderman_Colour1, eMinecraftColour_Mob_Enderman_Colour2, IDS_ENDERMAN);
	setId(CaveSpider::create, eTYPE_CAVESPIDER, L"CaveSpider", 59, eMinecraftColour_Mob_CaveSpider_Colour1, eMinecraftColour_Mob_CaveSpider_Colour2, IDS_CAVE_SPIDER);
	setId(Silverfish::create, eTYPE_SILVERFISH, L"Silverfish", 60, eMinecraftColour_Mob_Silverfish_Colour1, eMinecraftColour_Mob_Silverfish_Colour2, IDS_SILVERFISH);
	setId(Blaze::create, eTYPE_BLAZE, L"Blaze", 61, eMinecraftColour_Mob_Blaze_Colour1, eMinecraftColour_Mob_Blaze_Colour2, IDS_BLAZE);
	setId(LavaSlime::create, eTYPE_LAVASLIME, L"LavaSlime", 62, eMinecraftColour_Mob_LavaSlime_Colour1, eMinecraftColour_Mob_LavaSlime_Colour2, IDS_LAVA_SLIME);
	setId(EnderDragon::create, eTYPE_ENDERDRAGON, L"EnderDragon", 63);

    setId(Pig::create, eTYPE_PIG, L"Pig", 90, eMinecraftColour_Mob_Pig_Colour1, eMinecraftColour_Mob_Pig_Colour2, IDS_PIG);
    setId(Sheep::create, eTYPE_SHEEP, L"Sheep", 91, eMinecraftColour_Mob_Sheep_Colour1, eMinecraftColour_Mob_Sheep_Colour2, IDS_SHEEP);
    setId(Cow::create, eTYPE_COW, L"Cow", 92, eMinecraftColour_Mob_Cow_Colour1, eMinecraftColour_Mob_Cow_Colour2, IDS_COW);
    setId(Chicken::create, eTYPE_CHICKEN, L"Chicken", 93, eMinecraftColour_Mob_Chicken_Colour1, eMinecraftColour_Mob_Chicken_Colour2, IDS_CHICKEN);
    setId(Squid::create, eTYPE_SQUID, L"Squid", 94, eMinecraftColour_Mob_Squid_Colour1, eMinecraftColour_Mob_Squid_Colour2, IDS_SQUID);
    setId(Wolf::create, eTYPE_WOLF, L"Wolf", 95, eMinecraftColour_Mob_Wolf_Colour1, eMinecraftColour_Mob_Wolf_Colour2, IDS_WOLF);
	setId(MushroomCow::create, eTYPE_MUSHROOMCOW, L"MushroomCow", 96, eMinecraftColour_Mob_MushroomCow_Colour1, eMinecraftColour_Mob_MushroomCow_Colour2, IDS_MUSHROOM_COW);
	setId(SnowMan::create, eTYPE_SNOWMAN, L"SnowMan", 97);
	setId(Ozelot::create, eTYPE_OZELOT, L"Ozelot", 98, eMinecraftColour_Mob_Ocelot_Colour1, eMinecraftColour_Mob_Ocelot_Colour2, IDS_OZELOT);
	setId(VillagerGolem::create, eTYPE_VILLAGERGOLEM, L"VillagerGolem", 99);

	setId(Villager::create, eTYPE_VILLAGER, L"Villager", 120, eMinecraftColour_Mob_Villager_Colour1, eMinecraftColour_Mob_Villager_Colour2, IDS_VILLAGER);

	setId(EnderCrystal::create, eTYPE_ENDER_CRYSTAL, L"EnderCrystal", 200);

	// 4J Added
	setId(DragonFireball::create, eTYPE_DRAGON_FIREBALL, L"DragonFireball", 1000);
}

shared_ptr<Entity> EntityIO::newEntity(const wstring& id, Level *level)
{
    shared_ptr<Entity> entity;

	AUTO_VAR(it, idCreateMap->find(id));
	if(it != idCreateMap->end() )
	{
		entityCreateFn create = it->second;
		if (create != NULL) entity = shared_ptr<Entity>(create(level));
	}

    return entity;
}

shared_ptr<Entity> EntityIO::loadStatic(CompoundTag *tag, Level *level)
{
    shared_ptr<Entity> entity;

	AUTO_VAR(it, idCreateMap->find(tag->getString(L"id")));
	if(it != idCreateMap->end() )
	{
		entityCreateFn create = it->second;
		if (create != NULL) entity = shared_ptr<Entity>(create(level));
	}

    if (entity != NULL)
	{
        entity->load(tag);
    }
	else
	{
#ifdef _DEBUG
		app.DebugPrintf("Skipping Entity with id %ls\n", tag->getString(L"id").c_str() );
#endif
    }
    return entity;
}

shared_ptr<Entity> EntityIO::newById(int id, Level *level)
{
    shared_ptr<Entity> entity;

	AUTO_VAR(it, numCreateMap->find(id));
	if(it != numCreateMap->end() )
	{
		entityCreateFn create = it->second;
		if (create != NULL) entity = shared_ptr<Entity>(create(level));
	}

    if (entity != NULL)
	{
    }
	else
	{
        //printf("Skipping Entity with id %d\n", id ) ;
    }
    return entity;
}

shared_ptr<Entity> EntityIO::newByEnumType(eINSTANCEOF eType, Level *level)
{
    shared_ptr<Entity> entity;

	unordered_map<eINSTANCEOF, int, eINSTANCEOFKeyHash, eINSTANCEOFKeyEq>::iterator it = classNumMap->find( eType );
	if( it != classNumMap->end() )
	{
		AUTO_VAR(it2, numCreateMap->find(it->second));
		if(it2 != numCreateMap->end() )
		{
			entityCreateFn create = it2->second;
			if (create != NULL) entity = shared_ptr<Entity>(create(level));
		}
	}

    return entity;
}

int EntityIO::getId(shared_ptr<Entity> entity)
{
	unordered_map<eINSTANCEOF, int, eINSTANCEOFKeyHash, eINSTANCEOFKeyEq>::iterator it = classNumMap->find( entity->GetType() );
	return (*it).second;
}

wstring EntityIO::getEncodeId(shared_ptr<Entity> entity)
{
	unordered_map<eINSTANCEOF, wstring, eINSTANCEOFKeyHash, eINSTANCEOFKeyEq>::iterator it = classIdMap->find( entity->GetType() );
	if( it != classIdMap->end() )
		return (*it).second;
	else
		return L"";
}

int EntityIO::getId(const wstring &encodeId)
{
	AUTO_VAR(it, idNumMap->find(encodeId));
	if (it == idNumMap->end())
	{
		// defaults to pig...
		return 90;
	}
	return it->second;
}

wstring EntityIO::getEncodeId(int entityIoValue)
{
	//Class<? extends Entity> class1 = numClassMap.get(entityIoValue);
	//if (class1 != null)
	//{
	//return classIdMap.get(class1);
	//}
	
	AUTO_VAR(it, numClassMap->find(entityIoValue));
	if(it != numClassMap->end() )
	{
		unordered_map<eINSTANCEOF, wstring, eINSTANCEOFKeyHash, eINSTANCEOFKeyEq>::iterator classIdIt = classIdMap->find( it->second );
		if( classIdIt != classIdMap->end() )
			return (*classIdIt).second;
		else
			return L"";
	}

	return L"";
}

int EntityIO::getNameId(int entityIoValue)
{
	int id = -1;

	AUTO_VAR(it, idsSpawnableInCreative.find(entityIoValue));
	if(it != idsSpawnableInCreative.end())
	{
		id = it->second->nameId;
	}

	return id;
}

eINSTANCEOF EntityIO::getType(const wstring &idString)
{
	AUTO_VAR(it, numClassMap->find(getId(idString)));
	if(it != numClassMap->end() )
	{
		return it->second;
	}
	return eTYPE_NOTSET;
}

eINSTANCEOF EntityIO::getClass(int id)
{
	AUTO_VAR(it, numClassMap->find(id));
	if(it != numClassMap->end() )
	{
		return it->second;
	}
	return eTYPE_NOTSET;
}

int EntityIO::eTypeToIoid(eINSTANCEOF eType)
{
	unordered_map<eINSTANCEOF, int, eINSTANCEOFKeyHash, eINSTANCEOFKeyEq>::iterator it = classNumMap->find( eType );
	if( it != classNumMap->end() )
		return it->second;
	return -1;
}