You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1036 lines
28 KiB
1036 lines
28 KiB
//===== Copyright © 1996-2005, Valve Corporation, All rights reserved. ======//
|
|
//
|
|
// Purpose:
|
|
//
|
|
//===========================================================================//
|
|
|
|
|
|
#include "MapReslistGenerator.h"
|
|
#include "filesystem.h"
|
|
#include "filesystem_engine.h"
|
|
#include "sys.h"
|
|
#include "cmd.h"
|
|
#include "common.h"
|
|
#include "quakedef.h"
|
|
#include "vengineserver_impl.h"
|
|
#include "tier1/strtools.h"
|
|
#include "tier0/icommandline.h"
|
|
#include <ctype.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <tier0/dbg.h>
|
|
#include "host.h"
|
|
#include "host_state.h"
|
|
#include "utlbuffer.h"
|
|
#include "characterset.h"
|
|
#include "tier1/fmtstr.h"
|
|
|
|
// memdbgon must be the last include file in a .cpp file!!!
|
|
#include "tier0/memdbgon.h"
|
|
|
|
#define PAUSE_FRAMES_BETWEEN_MAPS 300
|
|
#define PAUSE_TIME_BETWEEN_MAPS 2.0f
|
|
|
|
extern engineparms_t host_parms;
|
|
|
|
#define ENGINE_RESLIST_FILE "engine.lst"
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose:
|
|
//-----------------------------------------------------------------------------
|
|
void MapReslistGenerator_Usage()
|
|
{
|
|
Msg( "-makereslists usage:\n" );
|
|
Msg( " [ -makereslists <optionalscriptfile> ] -- script file to control more complex makereslists operations (multiple passes, etc.)\n" );
|
|
Msg( " [ -usereslistfile filename ] -- get map list from specified file, default is to build for maps/*.bsp\n" );
|
|
Msg( " [ -startmap mapname ] -- restart generation at specified map (after crash, implies resume)\n" );
|
|
Msg( " [ -condebug ] -- prepend console.log entries with mapname or engine if not in a map\n" );
|
|
Msg( " [ -reslistmap mapname ] -- generate reslists for specified map and exit after that map\n" );
|
|
Msg( " [ -rebuildaudio ] -- force rebuild of _other_rebuild.cache (metacache) file at exit\n" );
|
|
Msg( " [ -forever ] -- when you get to the end of the maplist, start over from the top\n" );
|
|
Msg( " [ -stringtables ] -- force rebuild of the .bsp's stringtable dictionary\n" );
|
|
Msg( " [ -reslistdir ] -- default is 'reslists', use this to override\n" );
|
|
Msg( " [ -startstage nnn ] -- when running from script file, this starts at specified stage\n" );
|
|
Msg( " [ -collate ] -- skip everything, just merge the reslist from temp folders to the final folder again\n" );
|
|
}
|
|
|
|
void MapReslistGenerator_Init()
|
|
{
|
|
// check for reslist generation
|
|
if ( CommandLine()->FindParm("-makereslists") )
|
|
{
|
|
bool usemaplistfile = false;
|
|
if ( CommandLine()->FindParm("-usereslistfile") )
|
|
{
|
|
usemaplistfile = true;
|
|
}
|
|
MapReslistGenerator().EnableReslistGeneration( usemaplistfile );
|
|
}
|
|
else if ( CommandLine()->FindParm( "-rebuildaudio" ) )
|
|
{
|
|
MapReslistGenerator().SetAutoQuit( true );
|
|
}
|
|
|
|
if ( CommandLine()->FindParm( "-trackdeletions" ) )
|
|
{
|
|
MapReslistGenerator().EnableDeletionsTracking();
|
|
}
|
|
|
|
if ( CommandLine()->FindParm( "-autoquit" ) )
|
|
{
|
|
MapReslistGenerator().SetAutoQuit( true );
|
|
}
|
|
}
|
|
|
|
void MapReslistGenerator_Shutdown()
|
|
{
|
|
MapReslistGenerator().Shutdown();
|
|
}
|
|
|
|
void MapReslistGenerator_BuildMapList()
|
|
{
|
|
MapReslistGenerator().BuildMapList();
|
|
}
|
|
|
|
CMapReslistGenerator g_MapReslistGenerator;
|
|
CMapReslistGenerator &MapReslistGenerator()
|
|
{
|
|
return g_MapReslistGenerator;
|
|
}
|
|
|
|
static bool ReslistLogLessFunc( CUtlString const &pLHS, CUtlString const &pRHS )
|
|
{
|
|
return CaselessStringLessThan( pLHS.Get(), pRHS.Get() );
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: Constructor
|
|
//-----------------------------------------------------------------------------
|
|
CMapReslistGenerator::CMapReslistGenerator() :
|
|
m_AlreadyWrittenFileNames( 0, 0, true ),
|
|
m_DeletionListWarnings( 0, 0, DefLessFunc( CUtlSymbol ) ),
|
|
m_EngineLog( 0, 0, ReslistLogLessFunc ),
|
|
m_MapLog( 0, 0, ReslistLogLessFunc ),
|
|
m_bAutoQuit( false )
|
|
{
|
|
MEM_ALLOC_CREDIT_CLASS();
|
|
|
|
m_bUsingMapList = false;
|
|
m_bTrackingDeletions = false;
|
|
m_bLoggingEnabled = false;
|
|
m_bCreatingForXbox = false;
|
|
m_iCurrentMap = 0;
|
|
m_flNextMapRunTime = 0.0f;
|
|
m_szPrefix[0] = '\0';
|
|
m_szLevelName[0] = '\0';
|
|
m_iPauseTimeBetweenMaps = PAUSE_TIME_BETWEEN_MAPS;
|
|
m_bRestartOnTransition = false;
|
|
m_bLogToEngineList = true;
|
|
m_sResListDir = "reslists";
|
|
}
|
|
|
|
void CMapReslistGenerator::SetAutoQuit( bool bState )
|
|
{
|
|
m_bAutoQuit = bState;
|
|
}
|
|
|
|
void CMapReslistGenerator::BuildMapList()
|
|
{
|
|
if ( !IsEnabled() )
|
|
return;
|
|
|
|
if ( CommandLine()->FindParm( "+map" ) != 0 )
|
|
{
|
|
// This entire module is broken in about N different ways.
|
|
// The scripting features have heavily rotted, plus tie in with stringtables (which gets confused with multiple restarts).
|
|
// The whole thing barely works. I am narrowing it to work in exactly the one remaining way necesary for shipping.
|
|
Error( "CMapReslistGenerator Incompatible with normal map loading. Use -reslistmap" );
|
|
}
|
|
|
|
Msg( "********************\n" );
|
|
Msg( "Building Reslists\n" );
|
|
Msg( "********************\n" );
|
|
|
|
// Get the maplist file, if any
|
|
const char *pMapFile = NULL;
|
|
CommandLine()->CheckParm( "-usereslistfile", &pMapFile );
|
|
|
|
// -reslistmap argument precludes using a maplist file
|
|
bool bUseMap = CommandLine()->FindParm( "-reslistmap" ) != 0;
|
|
bool bUseMapListFile = bUseMap ? false : CommandLine()->FindParm("-usereslistfile") != 0;
|
|
|
|
// Build the map list
|
|
if ( !BuildGeneralMapList( &m_Maps, bUseMapListFile, pMapFile, "reslists", &m_iCurrentMap ) )
|
|
{
|
|
m_bLoggingEnabled = false;
|
|
}
|
|
}
|
|
|
|
bool BuildGeneralMapList( CUtlVector<maplist_map_t> *aMaps, bool bUseMapListFile, const char *pMapFile, char *pSystemMsg, int *iCurrentMap )
|
|
{
|
|
if ( !bUseMapListFile )
|
|
{
|
|
// If the user passed in a -reslistmap parameter, just use that single map
|
|
char const *pMapName = NULL;
|
|
if ( CommandLine()->CheckParm( "-reslistmap", &pMapName ) && pMapName )
|
|
{
|
|
// ensure validity
|
|
if (g_pVEngineServer->IsMapValid(pMapName))
|
|
{
|
|
// add to list
|
|
maplist_map_t newMap;
|
|
Q_strncpy(newMap.name, pMapName, sizeof(newMap.name));
|
|
aMaps->AddToTail( newMap );
|
|
}
|
|
|
|
CommandLine()->RemoveParm( "-reslistmap" );
|
|
}
|
|
else
|
|
{
|
|
// build the list of all the levels to scan
|
|
// Search the directory structure.
|
|
const char *mapwild = "maps/*.bsp";
|
|
char const *findfn = Sys_FindFirst( mapwild, NULL, 0 );
|
|
while ( findfn )
|
|
{
|
|
// make sure that it's in the mod filesystem
|
|
if ( !g_pFileSystem->FileExists( va("maps/%s", findfn), "MOD" ) )
|
|
{
|
|
findfn = Sys_FindNext( NULL, 0 );
|
|
continue;
|
|
}
|
|
|
|
// strip extension
|
|
char sz[ MAX_PATH ];
|
|
Q_strncpy( sz, findfn, sizeof( sz ) );
|
|
char *ext = strchr( sz, '.' );
|
|
if (ext)
|
|
{
|
|
ext[0] = 0;
|
|
}
|
|
|
|
// move to next item
|
|
findfn = Sys_FindNext( NULL, 0 );
|
|
|
|
// ensure validity
|
|
if (!g_pVEngineServer->IsMapValid(sz))
|
|
continue;
|
|
|
|
// add to list
|
|
maplist_map_t newMap;
|
|
Q_strncpy(newMap.name, sz, sizeof(newMap.name));
|
|
aMaps->AddToTail( newMap );
|
|
}
|
|
|
|
Sys_FindClose();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Read from file
|
|
if ( pMapFile )
|
|
{
|
|
// Load them in
|
|
FileHandle_t resfilehandle;
|
|
resfilehandle = g_pFileSystem->Open( pMapFile, "rb" );
|
|
if ( FILESYSTEM_INVALID_HANDLE != resfilehandle )
|
|
{
|
|
// Read in and parse mapcycle.txt
|
|
int length = g_pFileSystem->Size(resfilehandle);
|
|
if ( length > 0 )
|
|
{
|
|
char *pStart = (char *)new char[ length + 1 ];
|
|
if ( pStart && ( length == g_pFileSystem->Read(pStart, length, resfilehandle) ) )
|
|
{
|
|
pStart[ length ] = 0;
|
|
const char *pFileList = pStart;
|
|
|
|
while ( 1 )
|
|
{
|
|
char szMap[ MAX_OSPATH ];
|
|
|
|
pFileList = COM_Parse( pFileList );
|
|
if ( strlen( com_token ) <= 0 )
|
|
break;
|
|
|
|
Q_strncpy(szMap, com_token, sizeof(szMap));
|
|
|
|
// ensure validity
|
|
if (!g_pVEngineServer->IsMapValid(szMap))
|
|
continue;
|
|
|
|
// Any more tokens on this line?
|
|
while ( COM_TokenWaiting( pFileList ) )
|
|
{
|
|
pFileList = COM_Parse( pFileList );
|
|
}
|
|
|
|
maplist_map_t newMap;
|
|
Q_strncpy(newMap.name, szMap, sizeof(newMap.name));
|
|
aMaps->AddToTail( newMap );
|
|
}
|
|
}
|
|
delete[] pStart;
|
|
}
|
|
|
|
g_pFileSystem->Close(resfilehandle);
|
|
}
|
|
else
|
|
{
|
|
Error( "Unable to load %s maplist file: %s\n", pSystemMsg, pMapFile );
|
|
return false;
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
Error( "Unable to find %s maplist filename\n", pSystemMsg );
|
|
return false;
|
|
}
|
|
}
|
|
|
|
int c = aMaps->Count();
|
|
if ( c == 0 )
|
|
{
|
|
Msg( "%s: No maps found\n", pSystemMsg );
|
|
return false;
|
|
}
|
|
|
|
Msg( "%s: Creating for:\n", pSystemMsg );
|
|
|
|
// Determine the current map (-startmap allows starts mid-maplist)
|
|
*iCurrentMap = 0;
|
|
char const *startmap = NULL;
|
|
if ( CommandLine()->CheckParm( "-startmap", &startmap ) && startmap )
|
|
{
|
|
for ( int i = 0 ; i < c; ++i )
|
|
{
|
|
if ( !Q_stricmp( aMaps->Element(i).name, startmap ) )
|
|
{
|
|
*iCurrentMap = i;
|
|
}
|
|
}
|
|
}
|
|
|
|
for ( int i = 0 ; i < c; ++i )
|
|
{
|
|
if ( i < *iCurrentMap )
|
|
{
|
|
Msg( "- %s\n", aMaps->Element(i).name );
|
|
}
|
|
else
|
|
{
|
|
Msg( "+ %s\n", aMaps->Element(i).name );
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: Reconstructs engine log dictionary from existing engine reslist.
|
|
// This is used to restore state after a restart, otherwise the engine log
|
|
// would aggregate duplicate files.
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::BuildEngineLogFromReslist()
|
|
{
|
|
m_EngineLog.RemoveAll();
|
|
|
|
CUtlBuffer buffer( 0, 0, CUtlBuffer::TEXT_BUFFER );
|
|
if ( !g_pFileSystem->ReadFile( CFmtStr( "%s\\%s", m_sResListDir.String(), ENGINE_RESLIST_FILE ), "DEFAULT_WRITE_PATH", buffer ) )
|
|
{
|
|
// does not exist
|
|
return;
|
|
}
|
|
|
|
characterset_t breakSet;
|
|
CharacterSetBuild( &breakSet, "" );
|
|
|
|
// parse reslist
|
|
char szToken[MAX_PATH];
|
|
for ( ;; )
|
|
{
|
|
int nTokenSize = buffer.ParseToken( &breakSet, szToken, sizeof( szToken ) );
|
|
if ( nTokenSize <= 0 )
|
|
{
|
|
break;
|
|
}
|
|
|
|
int idx = m_EngineLog.Find( szToken );
|
|
if ( idx == m_EngineLog.InvalidIndex() )
|
|
{
|
|
m_EngineLog.Insert( szToken );
|
|
}
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: Appends specified line to the engine reslist.
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::LogToEngineReslist( char const *pLine )
|
|
{
|
|
// prevent unecessary duplication due to file appending
|
|
int idx = m_EngineLog.Find( pLine );
|
|
if ( idx != m_EngineLog.InvalidIndex() )
|
|
{
|
|
// already logged
|
|
return;
|
|
}
|
|
|
|
m_EngineLog.Insert( pLine );
|
|
|
|
// Open for append, write data, close.
|
|
FileHandle_t fh = g_pFileSystem->Open( CFmtStr( "%s\\%s", m_sResListDir.String(), ENGINE_RESLIST_FILE ), "at", "DEFAULT_WRITE_PATH" );
|
|
if ( fh != FILESYSTEM_INVALID_HANDLE )
|
|
{
|
|
g_pFileSystem->Write( "\"", 1, fh );
|
|
g_pFileSystem->Write( pLine, Q_strlen( pLine ), fh );
|
|
g_pFileSystem->Write( "\"\n", 2, fh );
|
|
g_pFileSystem->Close( fh );
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: initializes the object to enable reslist generation
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::EnableReslistGeneration( bool usemaplistfile )
|
|
{
|
|
// hackhack !!!! This is a work-around until CS precaches things on level start, not player spawn
|
|
if ( !Q_stricmp( "cstrike", GetCurrentMod() ) || !Q_stricmp( "csgo", GetCurrentMod() ) )
|
|
{
|
|
// the CS UI basically broke this, all sorts of waiting for team selection player input in UI
|
|
// 5 for the loading map screen, 10 for team selection, 15 for loading in general.
|
|
m_iPauseTimeBetweenMaps = 5 + 10 + 15;
|
|
}
|
|
|
|
m_bUsingMapList = usemaplistfile;
|
|
|
|
m_bLoggingEnabled = true;
|
|
|
|
char const *pszDir = NULL;
|
|
if ( CommandLine()->CheckParm( "-reslistdir", &pszDir ) && pszDir )
|
|
{
|
|
char szDir[ MAX_PATH ];
|
|
Q_strncpy( szDir, pszDir, sizeof( szDir ) );
|
|
Q_StripTrailingSlash( szDir );
|
|
Q_strlower( szDir );
|
|
Q_FixSlashes( szDir );
|
|
|
|
if ( Q_strlen( szDir ) > 0 )
|
|
{
|
|
m_sResListDir = szDir;
|
|
}
|
|
}
|
|
|
|
m_bCreatingForXbox = CommandLine()->FindParm( "-xboxreslist" ) != 0;
|
|
|
|
// create file to dump out to
|
|
g_pFileSystem->CreateDirHierarchy( m_sResListDir.String() , "DEFAULT_WRITE_PATH" );
|
|
|
|
// Leave the existing one if resuming from a specific map, otherwise, blow it away
|
|
if ( !CommandLine()->FindParm( "-startmap" ) )
|
|
{
|
|
g_pFileSystem->RemoveFile( CFmtStr( "%s\\%s", m_sResListDir.String(), ENGINE_RESLIST_FILE ), "DEFAULT_WRITE_PATH" );
|
|
m_EngineLog.RemoveAll();
|
|
}
|
|
else
|
|
{
|
|
BuildEngineLogFromReslist();
|
|
}
|
|
|
|
// add logging function
|
|
g_pFileSystem->AddLoggingFunc(&FileSystemLoggingFunc);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: starts the first map
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::StartReslistGeneration()
|
|
{
|
|
// wait for the main menu to stabilize then start the first map loading
|
|
m_iCurrentMap = 0;
|
|
m_flNextMapRunTime = Sys_FloatTime() + 10;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose:
|
|
// Input : *mapname -
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::SetPrefix( char const *mapname )
|
|
{
|
|
Q_snprintf( m_szPrefix, sizeof( m_szPrefix ), "%s: ", mapname );
|
|
}
|
|
|
|
void CMapReslistGenerator::OnLevelShutdown()
|
|
{
|
|
m_bLogToEngineList = true;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose:
|
|
// Output : char const
|
|
//-----------------------------------------------------------------------------
|
|
char const *CMapReslistGenerator::LogPrefix()
|
|
{
|
|
// If not recording stuff to file, then use the "default" prefix.
|
|
if ( m_bLogToEngineList )
|
|
{
|
|
return "engine: ";
|
|
}
|
|
|
|
return m_szPrefix;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: call to mark level load/end
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::OnLevelLoadStart(const char *levelName)
|
|
{
|
|
// prepare for map logging
|
|
m_bLogToEngineList = false;
|
|
V_strncpy( m_szLevelName, levelName, sizeof( m_szLevelName ) );
|
|
|
|
if ( !IsEnabled() )
|
|
{
|
|
char basename[ MAX_PATH ];
|
|
Q_FileBase( levelName, basename, sizeof( basename ) );
|
|
Q_strlower( basename );
|
|
SetPrefix( basename );
|
|
return;
|
|
}
|
|
|
|
// reset the duplication list
|
|
m_AlreadyWrittenFileNames.RemoveAll();
|
|
|
|
m_MapLog.RemoveAll();
|
|
|
|
// add in the bsp file to the list, and its node graph
|
|
char path[MAX_PATH];
|
|
Q_snprintf( path, sizeof( path ), "maps\\%s.bsp", levelName );
|
|
OnResourcePrecached( path );
|
|
|
|
bool useNodeGraph = true;
|
|
KeyValues *modinfo = new KeyValues("ModInfo");
|
|
if ( modinfo->LoadFromFile( g_pFileSystem, "gameinfo.txt" ) )
|
|
{
|
|
useNodeGraph = modinfo->GetInt( "nodegraph", 1 ) != 0;
|
|
}
|
|
modinfo->deleteThis();
|
|
|
|
if ( useNodeGraph )
|
|
{
|
|
Q_snprintf(path, sizeof(path), "maps\\graphs\\%s.ain", levelName);
|
|
OnResourcePrecached(path);
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: call to mark level load/end
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::OnLevelLoadEnd()
|
|
{
|
|
}
|
|
|
|
void CMapReslistGenerator::OnPlayerSpawn()
|
|
{
|
|
}
|
|
|
|
void CMapReslistGenerator::OnFullyConnected()
|
|
{
|
|
if ( !IsEnabled() )
|
|
return;
|
|
|
|
// initiate the next level
|
|
m_flNextMapRunTime = Sys_FloatTime() + m_iPauseTimeBetweenMaps;
|
|
}
|
|
|
|
bool CMapReslistGenerator::ShouldRebuildCaches()
|
|
{
|
|
if ( !IsEnabled() )
|
|
{
|
|
return CommandLine()->FindParm( "-rebuildaudio" ) != 0;
|
|
}
|
|
|
|
if ( !CommandLine()->FindParm( "-norebuildaudio" ) )
|
|
return true;
|
|
return false;
|
|
}
|
|
|
|
char const *CMapReslistGenerator::GetResListDirectory() const
|
|
{
|
|
return m_sResListDir.String();
|
|
}
|
|
|
|
void CMapReslistGenerator::DoQuit()
|
|
{
|
|
Cbuf_AddText( Cbuf_GetCurrentPlayer(), "quit\n" );
|
|
// remove the logging
|
|
g_pFileSystem->RemoveLoggingFunc(&FileSystemLoggingFunc);
|
|
m_bLogToEngineList = true;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: call every frame if we're enabled, just so that the next map can be triggered at the right time
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::RunFrame()
|
|
{
|
|
if ( !IsEnabled() )
|
|
{
|
|
if ( m_bAutoQuit )
|
|
{
|
|
m_bAutoQuit = false;
|
|
DoQuit();
|
|
}
|
|
return;
|
|
}
|
|
|
|
if ( m_flNextMapRunTime && m_flNextMapRunTime < Sys_FloatTime() )
|
|
{
|
|
// about to transition or terminate, emit the current map log
|
|
WriteMapLog();
|
|
|
|
if ( m_Maps.IsValidIndex( m_iCurrentMap ) )
|
|
{
|
|
// will start counting again after the level loads
|
|
m_flNextMapRunTime = 0.0f;
|
|
|
|
if ( !m_bRestartOnTransition )
|
|
{
|
|
Cbuf_AddText( Cbuf_GetCurrentPlayer(), va( "map %s\n", m_Maps[m_iCurrentMap].name ) );
|
|
|
|
SetPrefix( m_Maps[m_iCurrentMap].name );
|
|
|
|
++m_iCurrentMap;
|
|
if ( m_Maps.IsValidIndex( m_iCurrentMap ) )
|
|
{
|
|
// cause a full engine restart on the transition to the next map
|
|
// ensure that one-time init code logs correctly to each map reslist
|
|
m_bRestartOnTransition = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// restart at specified map
|
|
CommandLine()->RemoveParm( "-startmap" );
|
|
CommandLine()->AppendParm( "-startmap", m_Maps[m_iCurrentMap].name );
|
|
HostState_Restart();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// no more levels, just quit
|
|
if ( !CommandLine()->FindParm( "-forever" ) )
|
|
{
|
|
DoQuit();
|
|
}
|
|
else
|
|
{
|
|
StartReslistGeneration();
|
|
m_bRestartOnTransition = true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: logs and handles mdl files being precaches
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::OnModelPrecached(const char *relativePathFileName)
|
|
{
|
|
if ( !IsEnabled() )
|
|
return;
|
|
|
|
if (strstr(relativePathFileName, ".vmt"))
|
|
{
|
|
// it's a materials file, make sure that it starts in the materials directory, and we get the .vtf
|
|
char file[_MAX_PATH];
|
|
|
|
if ( StringHasPrefix( relativePathFileName, "materials" ) )
|
|
{
|
|
Q_strncpy(file, relativePathFileName, sizeof(file));
|
|
}
|
|
else
|
|
{
|
|
// prepend the materials directory
|
|
Q_snprintf(file, sizeof(file), "materials\\%s", relativePathFileName);
|
|
}
|
|
OnResourcePrecached(file);
|
|
|
|
// get the matching vtf file
|
|
char *ext = strstr(file, ".vmt");
|
|
if (ext)
|
|
{
|
|
Q_strncpy(ext, ".vtf", 5);
|
|
OnResourcePrecached(file);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
OnResourcePrecached(relativePathFileName);
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: logs sound file access
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::OnSoundPrecached(const char *relativePathFileName)
|
|
{
|
|
// skip any special characters
|
|
if (!V_isalnum(relativePathFileName[0]))
|
|
{
|
|
++relativePathFileName;
|
|
}
|
|
|
|
// prepend the sound/ directory if necessary
|
|
char file[_MAX_PATH];
|
|
if ( StringHasPrefix( relativePathFileName, "sound" ) )
|
|
{
|
|
Q_strncpy(file, relativePathFileName, sizeof(file));
|
|
}
|
|
else
|
|
{
|
|
// prepend the sound directory
|
|
Q_snprintf(file, sizeof(file), "sound\\%s", relativePathFileName);
|
|
}
|
|
|
|
OnResourcePrecached(file);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: logs the precache as a file access
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::OnResourcePrecached(const char *relativePathFileName)
|
|
{
|
|
if ( !IsEnabled() )
|
|
return;
|
|
|
|
// ignore empty string
|
|
if (relativePathFileName[0] == 0)
|
|
return;
|
|
|
|
// ignore files that start with '*' since they signify special models
|
|
if (relativePathFileName[0] == '*')
|
|
return;
|
|
|
|
char fullPath[_MAX_PATH];
|
|
if (g_pFileSystem->GetLocalPath(relativePathFileName, fullPath, sizeof(fullPath)))
|
|
{
|
|
OnResourcePrecachedFullPath(fullPath);
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: Logs out file access to a file
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::OnResourcePrecachedFullPath(const char *fullPathFileName)
|
|
{
|
|
char fixed[ MAX_PATH ];
|
|
Q_strncpy( fixed, fullPathFileName, sizeof( fixed ) );
|
|
Q_strlower( fixed );
|
|
Q_FixSlashes( fixed );
|
|
|
|
// make sure the filename hasn't already been written
|
|
UtlSymId_t filename = m_AlreadyWrittenFileNames.Find( fixed );
|
|
if ( filename != UTL_INVAL_SYMBOL )
|
|
return;
|
|
|
|
// record in list, so we don't write it again
|
|
m_AlreadyWrittenFileNames.AddString( fixed );
|
|
|
|
// add extras for mdl's
|
|
if (strstr(fixed, ".mdl"))
|
|
{
|
|
// it's a model, get it's other files as well
|
|
char file[_MAX_PATH];
|
|
Q_strncpy(file, fixed, sizeof(file) - 10);
|
|
char *ext = strstr(file, ".mdl");
|
|
|
|
Q_strncpy(ext, ".vvd", 10);
|
|
OnResourcePrecachedFullPath(file);
|
|
Q_strncpy(ext, ".ani", 10);
|
|
OnResourcePrecachedFullPath(file);
|
|
Q_strncpy(ext, ".dx90.vtx", 10);
|
|
OnResourcePrecachedFullPath(file);
|
|
Q_strncpy(ext, ".phy", 10);
|
|
OnResourcePrecachedFullPath(file);
|
|
Q_strncpy(ext, ".jpg", 10);
|
|
OnResourcePrecachedFullPath(file);
|
|
}
|
|
|
|
// strip it down relative to the root directory of the game (for steam)
|
|
char const *relativeFileName = Q_stristr( fixed, GetBaseDirectory() );
|
|
if ( relativeFileName )
|
|
{
|
|
// Skip the basedir and slash
|
|
relativeFileName += ( Q_strlen( GetBaseDirectory() ) + 1 );
|
|
}
|
|
|
|
if ( !relativeFileName )
|
|
{
|
|
return;
|
|
}
|
|
|
|
if ( m_bLogToEngineList )
|
|
{
|
|
LogToEngineReslist( relativeFileName );
|
|
}
|
|
else
|
|
{
|
|
// find or add to sorted tree
|
|
int idx = m_MapLog.Find( relativeFileName );
|
|
if ( idx == m_MapLog.InvalidIndex() )
|
|
{
|
|
m_MapLog.Insert( relativeFileName );
|
|
}
|
|
}
|
|
}
|
|
|
|
void CMapReslistGenerator::WriteMapLog()
|
|
{
|
|
if ( !m_szLevelName[0] )
|
|
{
|
|
// log has not been established yet
|
|
return;
|
|
}
|
|
|
|
// write the sorted map log, allows for easier diffs between revisions
|
|
char path[_MAX_PATH];
|
|
Q_snprintf( path, sizeof( path ), "%s\\%s.lst", m_sResListDir.String(), m_szLevelName );
|
|
FileHandle_t fh = g_pFileSystem->Open( path, "wt", "DEFAULT_WRITE_PATH" );
|
|
if ( FILESYSTEM_INVALID_HANDLE != fh )
|
|
{
|
|
for ( int i = m_MapLog.FirstInorder(); i != m_MapLog.InvalidIndex(); i = m_MapLog.NextInorder( i ) )
|
|
{
|
|
const char *pLine = m_MapLog[i].String();
|
|
g_pFileSystem->Write( "\"", 1, fh );
|
|
g_pFileSystem->Write( pLine, Q_strlen( pLine ), fh );
|
|
g_pFileSystem->Write( "\"\n", 2, fh );
|
|
}
|
|
g_pFileSystem->Close( fh );
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: callback function from filesystem
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::FileSystemLoggingFunc(const char *fullPathFileName, const char *options)
|
|
{
|
|
g_MapReslistGenerator.OnResourcePrecachedFullPath(fullPathFileName);
|
|
}
|
|
|
|
#define DELETIONS_BATCH_FILE "deletions.bat"
|
|
#define DELETIONS_WARNINGS_FILE "undelete.lst"
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose:
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::EnableDeletionsTracking()
|
|
{
|
|
unsigned int deletions = 0;
|
|
unsigned int warnings = 0;
|
|
|
|
// Load deletions file and build dictionary
|
|
m_bTrackingDeletions = true;;
|
|
|
|
// Open up deletions.bat and parse out all filenames
|
|
// Load them in
|
|
FileHandle_t deletionsfile;
|
|
deletionsfile = g_pFileSystem->Open( DELETIONS_BATCH_FILE, "rb" );
|
|
if ( FILESYSTEM_INVALID_HANDLE != deletionsfile )
|
|
{
|
|
// Read in and parse mapcycle.txt
|
|
int length = g_pFileSystem->Size(deletionsfile);
|
|
if ( length > 0 )
|
|
{
|
|
char *pStart = (char *)new char[ length + 1 ];
|
|
if ( pStart && ( length == g_pFileSystem->Read(pStart, length, deletionsfile) ) )
|
|
{
|
|
pStart[ length ] = 0;
|
|
const char *pFileList = pStart;
|
|
|
|
while ( 1 )
|
|
{
|
|
char filename[ MAX_OSPATH ];
|
|
|
|
pFileList = COM_Parse( pFileList );
|
|
if ( strlen( com_token ) <= 0 )
|
|
break;
|
|
|
|
if ( !Q_stricmp( com_token, "del" ) )
|
|
continue;
|
|
|
|
Q_snprintf(filename, sizeof( filename ), "%s/%s", com_gamedir, com_token );
|
|
|
|
// Any more tokens on this line?
|
|
while ( COM_TokenWaiting( pFileList ) )
|
|
{
|
|
pFileList = COM_Parse( pFileList );
|
|
}
|
|
|
|
Q_FixSlashes( filename );
|
|
Q_strlower( filename );
|
|
|
|
m_DeletionList.AddString( filename );
|
|
|
|
++deletions;
|
|
}
|
|
}
|
|
delete[] pStart;
|
|
}
|
|
|
|
g_pFileSystem->Close(deletionsfile);
|
|
}
|
|
else
|
|
{
|
|
Warning( "Unable to load deletions.bat file %s\n", DELETIONS_BATCH_FILE );
|
|
m_bTrackingDeletions = false;
|
|
return;
|
|
}
|
|
|
|
FileHandle_t warningsfile = g_pFileSystem->Open( DELETIONS_WARNINGS_FILE, "rb" );
|
|
if ( FILESYSTEM_INVALID_HANDLE != warningsfile )
|
|
{
|
|
// Read in and parse mapcycle.txt
|
|
int length = g_pFileSystem->Size(warningsfile);
|
|
if ( length > 0 )
|
|
{
|
|
char *pStart = (char *)new char[ length + 1 ];
|
|
if ( pStart && ( length == g_pFileSystem->Read(pStart, length, warningsfile) ) )
|
|
{
|
|
pStart[ length ] = 0;
|
|
const char *pFileList = pStart;
|
|
|
|
while ( 1 )
|
|
{
|
|
pFileList = COM_Parse( pFileList );
|
|
if ( strlen( com_token ) <= 0 )
|
|
break;
|
|
|
|
Q_FixSlashes( com_token );
|
|
Q_strlower( com_token );
|
|
|
|
CUtlSymbol sym = m_DeletionListWarningsSymbols.AddString( com_token );
|
|
int idx = m_DeletionListWarnings.Find( sym );
|
|
if ( idx == m_DeletionListWarnings.InvalidIndex() )
|
|
{
|
|
m_DeletionListWarnings.Insert( sym );
|
|
++warnings;
|
|
}
|
|
}
|
|
}
|
|
delete[] pStart;
|
|
}
|
|
|
|
g_pFileSystem->Close(warningsfile);
|
|
}
|
|
|
|
// Hook up logging function
|
|
g_pFileSystem->AddLoggingFunc( &TrackDeletionsLoggingFunc );
|
|
|
|
Msg( "Tracking deletions (%u files in deletion list in '%s', %u previous warnings loaded from '%s'\n",
|
|
deletions, DELETIONS_BATCH_FILE, warnings, DELETIONS_WARNINGS_FILE );
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose:
|
|
// Input : *fullPathFileName -
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::TrackDeletions( const char *fullPathFileName )
|
|
{
|
|
Assert( m_bTrackingDeletions );
|
|
|
|
char test[ _MAX_PATH ];
|
|
Q_strncpy( test, fullPathFileName, sizeof( test ) );
|
|
Q_FixSlashes( test );
|
|
Q_strlower( test );
|
|
|
|
CUtlSymbol sym = m_DeletionList.Find( test );
|
|
if ( UTL_INVAL_SYMBOL != sym )
|
|
{
|
|
CUtlSymbol warningSymbol = m_DeletionListWarningsSymbols.AddString( test );
|
|
|
|
uint idx = m_DeletionListWarnings.Find( warningSymbol );
|
|
if ( idx == m_DeletionListWarnings.InvalidIndex() )
|
|
{
|
|
Msg( "--> Referenced file marked for deletion \"%s\"\n", test );
|
|
m_DeletionListWarnings.Insert( warningSymbol );
|
|
}
|
|
}
|
|
|
|
// add extras for mdl's
|
|
if (strstr(test, ".mdl"))
|
|
{
|
|
// it's a model, get it's other files as well
|
|
char file[_MAX_PATH];
|
|
Q_strncpy(file, test, sizeof(file) - 10);
|
|
char *ext = strstr(file, ".mdl");
|
|
|
|
Q_strncpy(ext, ".vvd", 10);
|
|
TrackDeletions(file);
|
|
Q_strncpy(ext, ".ani", 10);
|
|
TrackDeletions(file);
|
|
Q_strncpy(ext, ".dx80.vtx", 10);
|
|
TrackDeletions(file);
|
|
Q_strncpy(ext, ".dx90.vtx", 10);
|
|
TrackDeletions(file);
|
|
Q_strncpy(ext, ".sw.vtx", 10);
|
|
TrackDeletions(file);
|
|
Q_strncpy(ext, ".phy", 10);
|
|
TrackDeletions(file);
|
|
Q_strncpy(ext, ".jpg", 10);
|
|
TrackDeletions(file);
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: callback function from filesystem
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::TrackDeletionsLoggingFunc(const char *fullPathFileName, const char *options)
|
|
{
|
|
g_MapReslistGenerator.TrackDeletions(fullPathFileName);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose:
|
|
//-----------------------------------------------------------------------------
|
|
void CMapReslistGenerator::Shutdown()
|
|
{
|
|
if ( m_bTrackingDeletions )
|
|
{
|
|
SpewTrackedDeletionsLog();
|
|
|
|
g_pFileSystem->RemoveLoggingFunc( &TrackDeletionsLoggingFunc );
|
|
m_DeletionList.RemoveAll();
|
|
m_DeletionListWarnings.RemoveAll();
|
|
m_DeletionListWarningsSymbols.RemoveAll();
|
|
m_bTrackingDeletions = NULL;
|
|
}
|
|
}
|
|
|
|
void CMapReslistGenerator::SpewTrackedDeletionsLog()
|
|
{
|
|
if ( !m_bTrackingDeletions )
|
|
return;
|
|
|
|
FileHandle_t hUndeleteFile = g_pFileSystem->Open( DELETIONS_WARNINGS_FILE, "wt", "DEFAULT_WRITE_PATH" );
|
|
if ( FILESYSTEM_INVALID_HANDLE == hUndeleteFile )
|
|
{
|
|
return;
|
|
}
|
|
|
|
for ( int i = m_DeletionListWarnings.FirstInorder(); i != m_DeletionListWarnings.InvalidIndex() ; i = m_DeletionListWarnings.NextInorder( i ) )
|
|
{
|
|
char const *filename = m_DeletionListWarningsSymbols.String( m_DeletionListWarnings[ i ] );
|
|
|
|
g_pFileSystem->Write("\"", 1, hUndeleteFile);
|
|
g_pFileSystem->Write(filename, Q_strlen(filename), hUndeleteFile);
|
|
g_pFileSystem->Write("\"\n", 2, hUndeleteFile);
|
|
}
|
|
|
|
g_pFileSystem->Close( hUndeleteFile );
|
|
}
|
|
|
|
bool CMapReslistGenerator::IsCreatingForXbox()
|
|
{
|
|
return IsEnabled() && m_bCreatingForXbox;
|
|
}
|