#include "precompiled.h"

#include "TechnologyCollection.h"
#include "ps/CLogger.h"
#include "ps/VFSUtil.h"
#include "ps/Player.h"

#define LOG_CATEGORY "tech"

void CTechnologyCollection::LoadFile( const char* path )
{
    //Make tech file reading
    CStrW tag = CStr(path).AfterLast("/").BeforeLast(".xml");
    m_techFilenames[tag] = path;
}

static void LoadTechThunk( const char* path, const DirEnt* UNUSED(ent), void* context )
{
    CTechnologyCollection* this_ = (CTechnologyCollection*)context;
    this_->LoadFile(path);
}

int CTechnologyCollection::LoadTechnologies()
{
    // Load all files in techs/ and subdirectories.
    THROW_ERR( vfs_dir_enum( "technologies/", VFS_DIR_RECURSIVE, "*.xml",
        LoadTechThunk, this ) );
    return 0;
}

CTechnology* CTechnologyCollection::GetTechnology( const CStrW& name, CPlayer* player )
{
    // Find player ID
    debug_assert( player != 0 );
    int id = player->GetPlayerID();

    // Check whether this template has already been loaded.
    //If previously loaded, all slots will be found, so any entry works.
    TechMap::iterator it = m_techs[id].find( name );
    if( it != m_techs[id].end() )
        return( it->second );

    // Find the filename corresponding to this template
    TechFilenameMap::iterator filename_it = m_techFilenames.find( name );
    if( filename_it == m_techFilenames.end() )
        return( NULL );

    CStr path( filename_it->second );

    //Try to load to the tech
    CTechnology* newTemplate = new CTechnology( name, player );
    if( !newTemplate->LoadXml( path ) )
    {
        LOG(ERROR, LOG_CATEGORY, "CTechnologyCollection::GetTechnology(): Couldn't load tech \"%s\"", path.c_str());
        delete newTemplate;
        return( NULL );

    }
    m_techs[id][name] = newTemplate;
    
    LOG(NORMAL, LOG_CATEGORY, "CTechnologyCollection::GetTechnology(): Loaded tech \"%s\"", path.c_str());
    return newTemplate;
}

CTechnologyCollection::~CTechnologyCollection()
{
    for( uint id=0; id<PS_MAX_PLAYERS+1; id++ )
        for( TechMap::iterator it = m_techs[id].begin(); it != m_techs[id].end(); ++it )
            delete( it->second );
}