forked from 0ad/0ad
72 lines
2.1 KiB
C++
72 lines
2.1 KiB
C++
|
#include "precompiled.h"
|
||
|
|
|
||
|
|
#include "BaseTechCollection.h"
|
||
|
|
#include "ps/CLogger.h"
|
||
|
|
#include "ps/VFSUtil.h"
|
||
|
|
|
||
|
|
#define LOG_CATEGORY "tech"
|
||
|
|
|
||
|
|
void CBaseTechCollection::LoadFile( const char* path )
|
||
|
|
{
|
||
|
|
//Make tech file reading
|
||
|
|
CStrW tag = CStr(path).AfterLast("/").BeforeLast(".xml");
|
||
|
|
m_templateFilenames[tag] = path;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void LoadTechThunk( const char* path, const DirEnt* UNUSED(ent), void* context )
|
||
|
|
{
|
||
|
|
CBaseTechCollection* this_ = (CBaseTechCollection*)context;
|
||
|
|
this_->LoadFile(path);
|
||
|
|
}
|
||
|
|
|
||
|
|
int CBaseTechCollection::loadTemplates()
|
||
|
|
{
|
||
|
|
// Load all files in techs/ and subdirectories.
|
||
|
|
THROW_ERR( vfs_dir_enum( "techs/", VFS_DIR_RECURSIVE, "*.xml",
|
||
|
|
LoadTechThunk, this ) );
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
CBaseTech* CBaseTechCollection::getTemplate( CStrW name )
|
||
|
|
{
|
||
|
|
// Check whether this template has already been loaded.
|
||
|
|
//If previously loaded, all slots will be found, so any entry works.
|
||
|
|
templateMap::iterator it = m_templates.find( name );
|
||
|
|
if( it != m_templates.end() )
|
||
|
|
return( it->second );
|
||
|
|
|
||
|
|
// Find the filename corresponding to this template
|
||
|
|
templateFilenameMap::iterator filename_it = m_templateFilenames.find( name );
|
||
|
|
if( filename_it == m_templateFilenames.end() )
|
||
|
|
return( NULL );
|
||
|
|
|
||
|
|
CStr path( filename_it->second );
|
||
|
|
|
||
|
|
//Try to load to the tech
|
||
|
|
CBaseTech* newTemplate = new CBaseTech();
|
||
|
|
if( !newTemplate->loadXML( path ) )
|
||
|
|
{
|
||
|
|
LOG(ERROR, LOG_CATEGORY, "CBaseTechCollection::loadTemplates(): Couldn't load template \"%s\"", path.c_str());
|
||
|
|
delete newTemplate;
|
||
|
|
return( NULL );
|
||
|
|
|
||
|
|
}
|
||
|
|
m_templates[name] = newTemplate;
|
||
|
|
|
||
|
|
LOG(NORMAL, LOG_CATEGORY, "CBaseTechCollection::loadTemplates(): Loaded template \"%s\"", path.c_str());
|
||
|
|
return newTemplate;
|
||
|
|
}
|
||
|
|
|
||
|
|
void CBaseTechCollection::getBaseTechNames( std::vector<CStrW>& names )
|
||
|
|
{
|
||
|
|
for( templateFilenameMap::iterator it = m_templateFilenames.begin(); it != m_templateFilenames.end(); ++it )
|
||
|
|
if( ! (it->first.Length() > 8 && it->first.Left(8) == L"template"))
|
||
|
|
names.push_back( it->first );
|
||
|
|
}
|
||
|
|
|
||
|
|
CBaseTechCollection::~CBaseTechCollection()
|
||
|
|
{
|
||
|
|
for( templateMap::iterator it = m_templates.begin(); it != m_templates.end(); ++it )
|
||
|
|
delete( it->second );
|
||
|
|
}
|