#include "precompiled.h"

#include "BaseEntityCollection.h"
#include "ObjectManager.h"
#include "Model.h"
#include "CLogger.h"

#define LOG_CATEGORY "entity"

void CBaseEntityCollection::loadTemplates()
{
	Handle handle;
	vfsDirEnt type;
	
	CStr basepath = "entities/";
	CStr pathname;

	// Iterate through all subdirectories 

	Handle maindir = vfs_open_dir( basepath );

	if( maindir > 0 )
	{
		LoadDirectory( maindir, basepath );
		while( !vfs_next_dirent( maindir, &type, "/" ) )
		{
			pathname = basepath + type.name;

			handle = vfs_open_dir( pathname );

			pathname += "/";

			if( handle > 0 )
			{
				LoadDirectory( handle, pathname );
				vfs_close_dir( handle );
			}
			else
			{
				LOG(ERROR, LOG_CATEGORY, "CBaseEntityCollection::loadTemplates(): Failed to enumerate entity template directory");
				return;
			}
		}
		vfs_close_dir( maindir );
	}
	else
		LOG(ERROR, LOG_CATEGORY, "CBaseEntityCollection::loadTemplates(): Unable to open directory entities/");

	// Fix up parent links in the templates.
	
	std::vector<CBaseEntity*>::iterator it, it_done;
	std::vector<CBaseEntity*> done;

	// TODO: MT: Circular references check.

	while( done.size() < m_templates.size() )
	{
		for( it = m_templates.begin(); it != m_templates.end(); it++ )
		{
			if( !( (*it)->m_Base_Name.Length() ) )
			{
				done.push_back( *it );
				continue;
			}

			CBaseEntity* Base = getTemplate( (*it)->m_Base_Name );
			if( Base )
			{
				// Check whether it's been loaded yet.
				for( it_done = done.begin(); it_done != done.end(); it_done++ )
				{
					if( *it_done == Base )
					{
						(*it)->m_base = Base;
						(*it)->loadBase();
						Base = NULL;
						break;
					}
				}
				if( !Base )
				{
					// Done
					done.push_back( *it );
					continue;
				}
			}
			else
				LOG( WARNING, LOG_CATEGORY, "Parent template %s does not exist in template %s", CStr8( (*it)->m_Base_Name ).c_str(), CStr8( (*it)->m_Tag ).c_str() );
		}
	}
}

void CBaseEntityCollection::LoadDirectory( Handle directory, CStr pathname )
{
	vfsDirEnt file;
	while( !vfs_next_dirent( directory, &file, "*.xml") )
	{
		CBaseEntity* newTemplate = new CBaseEntity();
		if( newTemplate->loadXML( pathname + file.name ) )
		{
			addTemplate( newTemplate );
			LOG(NORMAL, LOG_CATEGORY, "CBaseEntityCollection::loadTemplates(): Loaded template \"%s%s\"", pathname.c_str(), file.name);
		}
		else
			LOG(ERROR, LOG_CATEGORY, "CBaseEntityCollection::loadTemplates(): Couldn't load template \"%s%s\"", pathname.c_str(), file.name);
	}
}

void CBaseEntityCollection::addTemplate( CBaseEntity* temp )
{
	m_templates.push_back( temp );
}

CBaseEntity* CBaseEntityCollection::getTemplate( CStrW name )
{
	for( u16 t = 0; t < m_templates.size(); t++ )
		if( m_templates[t]->m_Tag == name ) return( m_templates[t] );

	return( NULL );
}

CBaseEntity* CBaseEntityCollection::getTemplateByActor( CObjectEntry* actor )
{
	for( u16 t = 0; t < m_templates.size(); t++ )
		if( m_templates[t]->m_actorObject == actor ) return( m_templates[t] );

	return( NULL );
}

CBaseEntityCollection::~CBaseEntityCollection()
{
	for( u16 t = 0; t < m_templates.size(); t++ )
		delete( m_templates[t] );
}