Team Fortress 2 Source Code as on 22/4/2020
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.
 
 
 
 
 
 

1978 lines
61 KiB

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Queued Loading of map resources. !!!!Specifically!!! designed for the map loading process.
//
// Not designed for application startup or gameplay time. Layered on top of async i/o.
// Queued loading is allowed during the map load process until full connection only,
// but can complete the remaining low priority jobs during the game render.
// The normal loading path can run in units of seconds if it does not have to do I/O,
// which is why this system runs first and gets all the data in memory unhindered
// by dependency blocking. The I/O delivery process achieves its speed by having all the I/O
// requests at once, performing the I/O, and handing the actual consumption
// of the I/O buffer to another available core/thread (via job pool) for computation work.
// The I/O (should be all unbuffered) is then only throttled by physical transfer rates.
//
// The Load process is broken into three phases. The first phase build up I/O requests.
// The second phase fulfills only the high priority I/O requests. This gets the critical
// data in memory, that has to be there for the normal load path to query, or the renderer
// to run (i.e. models and shaders). The third phase is the normal load process.
// The low priority jobs run concurrently with the normal load process. Low priority jobs
// are those that have been specially built such that the game or loading can operate unblocked
// without the actual data (i.e. d3d texture bits).
//
// Phase 1: The reslist is parsed into seperate lists based on handled extensions. Each list
// call its own loader which in turn generates its own dictionaries and I/O requests through
// "AddJob". A single reslist entry could cause a laoder to request multiple jobs. ( i.e. models )
// A loader marks its jobs as high or low priority.
// Phase 2: The I/O requests are sorted (which achieves seek offset order) and
// async i/o commences. Phase 2 does not end until all the high priority jobs
// are complete. This ensures critical data is resident.
// Phase 3: The !!!NORMAL!!! loading path can commence. The legacy loading path then
// is not expected to do I/O (it can, but that's a hole in the reslist), as all of the data
// that it queries, should be resident.
//
// Late added jobs are non-optimal (should have been in reslist), warned, but handled.
//
//===========================================================================//
#include "basefilesystem.h"
#include "tier0/vprof.h"
#include "tier0/tslist.h"
#include "tier1/utlbuffer.h"
#include "tier1/convar.h"
#include "tier1/KeyValues.h"
#include "tier1/utllinkedlist.h"
#include "tier1/utlstring.h"
#include "tier1/UtlSortVector.h"
#include "tier1/utldict.h"
#include "basefilesystem.h"
#include "tier0/icommandline.h"
#include "vstdlib/jobthread.h"
#include "filesystem/IQueuedLoader.h"
#include "tier2/tier2.h"
#include "characterset.h"
#if !defined( _X360 )
#include "xbox/xboxstubs.h"
#endif
// memdbgon must be the last include file in a .cpp file!!!
#include "tier0/memdbgon.h"
#define PRIORITY_HIGH 1
#define PRIORITY_NORMAL 0
#define PRIORITY_LOW -1
// main thread has reason to block and wait for thread pool to finish jobs
#define MAIN_THREAD_YIELD_TIME 20
// discrete stages in the preload process to tick the progress bar
#define PROGRESS_START 0.10f
#define PROGRESS_GOTRESLIST 0.12f
#define PROGRESS_PARSEDRESLIST 0.15f
#define PROGRESS_CREATEDRESOURCES 0.20f
#define PROGRESS_PREPURGE 0.22f
#define PROGRESS_IO 0.25f // up to 1.0
struct FileJob_t
{
FileJob_t()
{
Q_memset( this, 0, sizeof( FileJob_t ) );
}
FileNameHandle_t m_hFilename;
QueuedLoaderCallback_t m_pCallback;
FSAsyncControl_t m_hAsyncControl;
void *m_pContext;
void *m_pContext2;
void *m_pTargetData;
int m_nBytesToRead;
unsigned int m_nStartOffset;
LoaderPriority_t m_Priority;
unsigned int m_SubmitTime;
unsigned int m_FinishTime;
int m_SubmitTag;
int m_nActualBytesRead;
LoaderError_t m_LoaderError;
unsigned int m_ThreadId;
unsigned int m_bFinished : 1;
unsigned int m_bFreeTargetAfterIO : 1;
unsigned int m_bFileExists : 1;
unsigned int m_bClaimed : 1;
};
// dummy stubbed progress interface
class CDummyProgress : public ILoaderProgress
{
void BeginProgress() {}
void UpdateProgress( float progress ) {}
void EndProgress() {}
};
static CDummyProgress s_DummyProgress;
class CQueuedLoader : public CTier2AppSystem< IQueuedLoader >
{
typedef CTier2AppSystem< IQueuedLoader > BaseClass;
public:
CQueuedLoader();
virtual ~CQueuedLoader();
// Inherited from IAppSystem
virtual InitReturnVal_t Init();
virtual void Shutdown();
// IQueuedLoader
virtual void InstallLoader( ResourcePreload_t type, IResourcePreload *pLoader );
virtual void InstallProgress( ILoaderProgress *pProgress );
// Set bOptimizeReload if you want appropriate data (such as static prop lighting)
// to persist - rather than being purged and reloaded - when going from map A to map A.
virtual bool BeginMapLoading( const char *pMapName, bool bLoadForHDR, bool bOptimizeMapReload );
virtual void EndMapLoading( bool bAbort );
virtual bool AddJob( const LoaderJob_t *pLoaderJob );
virtual void AddMapResource( const char *pFilename );
virtual void DynamicLoadMapResource( const char *pFilename, DynamicResourceCallback_t pCallback, void *pContext, void *pContext2 );
virtual void QueueDynamicLoadFunctor( CFunctor* pFunctor );
virtual bool CompleteDynamicLoad();
virtual void QueueCleanupDynamicLoadFunctor( CFunctor* pFunctor );
virtual bool CleanupDynamicLoad();
virtual bool ClaimAnonymousJob( const char *pFilename, QueuedLoaderCallback_t pCallback, void *pContext, void *pContext2 );
virtual bool ClaimAnonymousJob( const char *pFilename, void **pData, int *pDataSize, LoaderError_t *pError );
virtual bool IsMapLoading() const;
virtual bool IsSameMapLoading() const;
virtual bool IsFinished() const;
virtual bool IsBatching() const;
virtual bool IsDynamic() const;
virtual int GetSpewDetail() const;
char *GetFilename( const FileNameHandle_t hFilename, char *pBuff, int nBuffSize );
FileNameHandle_t FindFilename( const char *pFilename );
void SpewInfo();
// submit any queued jobs to the async loader, called by main or async thread to get more work
void SubmitPendingJobs();
void PurgeAll();
private:
class CFileJobsLessFunc
{
public:
int GetLayoutOrderForFilename( const char *pFilename );
bool Less( FileJob_t* const &pFileJobLHS, FileJob_t* const &pFileJobRHS, void *pCtx );
};
class CResourceNameLessFunc
{
public:
bool Less( const FileNameHandle_t &hFilenameLHS, const FileNameHandle_t &hFilenameRHS, void *pCtx );
};
typedef CUtlSortVector< FileNameHandle_t, CResourceNameLessFunc > ResourceList_t;
static void BuildResources( IResourcePreload *pLoader, ResourceList_t *pList, float *pBuildTime );
static void BuildMaterialResources( IResourcePreload *pLoader, ResourceList_t *pList, float *pBuildTime );
void PurgeQueue();
void CleanQueue();
void SubmitBatchedJobs();
void SubmitBatchedJobsAndWait();
void ParseResourceList( CUtlBuffer &resourceList );
void GetJobRequests();
void PurgeUnreferencedResources();
void AddResourceToTable( const char *pFilename );
bool m_bStarted;
bool m_bActive;
bool m_bBatching;
bool m_bDynamic;
bool m_bCanBatch;
bool m_bLoadForHDR;
bool m_bDoProgress;
bool m_bSameMap;
int m_nSubmitCount;
unsigned int m_StartTime;
unsigned int m_EndTime;
char m_szMapNameToCompareSame[MAX_PATH];
DynamicResourceCallback_t m_pfnDynamicCallback;
CUtlString m_DynamicFileName;
void* m_pDynamicContext;
void* m_pDynamicContext2;
CThreadFastMutex m_FunctorQueueMutex;
CUtlVector< CFunctor* > m_FunctorQueue;
CUtlVector< CFunctor* > m_CleanupFunctorQueue;
CUtlFilenameSymbolTable m_Filenames;
CTSList< FileJob_t* > m_PendingJobs;
CTSList< FileJob_t* > m_BatchedJobs;
CUtlLinkedList< FileJob_t* > m_SubmittedJobs;
CUtlDict< FileJob_t*, int > m_AnonymousJobs;
CUtlSymbolTable m_AdditionalResources;
CUtlSortVector< FileNameHandle_t, CResourceNameLessFunc > m_ResourceNames[RESOURCEPRELOAD_COUNT];
IResourcePreload *m_pLoaders[RESOURCEPRELOAD_COUNT];
float m_LoaderTimes[RESOURCEPRELOAD_COUNT];
ILoaderProgress *m_pProgress;
CThreadFastMutex m_Mutex;
};
static CQueuedLoader g_QueuedLoader;
EXPOSE_SINGLE_INTERFACE_GLOBALVAR( CQueuedLoader, IQueuedLoader, QUEUEDLOADER_INTERFACE_VERSION, g_QueuedLoader );
class CResourcePreloadAnonymous : public IResourcePreload
{
virtual bool CreateResource( const char *pName )
{
// create an anonymous job to get the data in memory, claimed during load, or auto-freed
LoaderJob_t loaderJob;
loaderJob.m_pFilename = pName;
loaderJob.m_pPathID = "GAME";
loaderJob.m_Priority = LOADERPRIORITY_DURINGPRELOAD;
g_QueuedLoader.AddJob( &loaderJob );
return true;
}
virtual void PurgeUnreferencedResources() {}
virtual void OnEndMapLoading( bool bAbort ) {}
virtual void PurgeAll() {}
};
static CResourcePreloadAnonymous s_ResourcePreloadAnonymous;
const char *g_ResourceLoaderNames[RESOURCEPRELOAD_COUNT] =
{
"???", // RESOURCEPRELOAD_UNKNOWN
"Sounds", // RESOURCEPRELOAD_SOUND
"Materials", // RESOURCEPRELOAD_MATERIAL
"Models", // RESOURCEPRELOAD_MODEL
"Cubemaps", // RESOURCEPRELOAD_CUBEMAP
"PropLighting", // RESOURCEPRELOAD_STATICPROPLIGHTING
"Anonymous", // RESOURCEPRELOAD_ANONYMOUS
};
static CInterlockedInt g_nActiveJobs;
static CInterlockedInt g_nQueuedJobs;
static CInterlockedInt g_nHighPriorityJobs; // tracks jobs that must finish during preload
static CInterlockedInt g_nJobsToFinishBeforePlay; // tracks jobs that must finish before gameplay
static CInterlockedInt g_nIOMemory; // tracks I/O data from async delivery until consumed
static CInterlockedInt g_nAnonymousIOMemory; // tracks anonymous I/O data from async delivery until consumed
static CInterlockedInt g_SuspendIO; // used to throttle the I/O
static int g_nIOMemoryPeak;
static int g_nAnonymousIOMemoryPeak;
static int g_nHighIOSuspensionMark;
static int g_nLowIOSuspensionMark;
ConVar loader_spew_info( "loader_spew_info", "0", 0, "0:Off, 1:Timing, 2:Completions, 3:Late Completions, 4:Purges, -1:All " );
// Kyle says: this is here only to change the DLL size to force clients to update! This should be removed
// by whoever sees this comment after we've shipped a DLL using it!
ConVar loader_sped_info_ex( "loader_spew_info_ex", "0", 0, "(internal)" );
CON_COMMAND( loader_dump_table, "" )
{
g_QueuedLoader.SpewInfo();
}
//-----------------------------------------------------------------------------
// Constructor
//-----------------------------------------------------------------------------
CQueuedLoader::CQueuedLoader() : BaseClass( false )
{
m_bStarted = false;
m_bActive = false;
m_bBatching = false;
m_bDynamic = false;
m_bCanBatch = false;
m_bLoadForHDR = false;
m_bDoProgress = false;
m_bSameMap = false;
m_nSubmitCount = 0;
m_pfnDynamicCallback = NULL;
m_pDynamicContext = NULL;
m_pDynamicContext2 = NULL;
m_szMapNameToCompareSame[0] = '\0';
m_pProgress = &s_DummyProgress;
V_memset( m_pLoaders, 0, sizeof( m_pLoaders ) );
// set resource dictionaries sort context
for ( int i = 0; i < RESOURCEPRELOAD_COUNT; i++ )
{
m_ResourceNames[i].SetLessContext( (void *)i );
}
InstallLoader( RESOURCEPRELOAD_ANONYMOUS, &s_ResourcePreloadAnonymous );
}
//-----------------------------------------------------------------------------
// Destructor
//-----------------------------------------------------------------------------
CQueuedLoader::~CQueuedLoader()
{
}
//-----------------------------------------------------------------------------
// Computation job to build out objects
//-----------------------------------------------------------------------------
void CQueuedLoader::BuildResources( IResourcePreload *pLoader, ResourceList_t *pList, float *pBuildTime )
{
float t0 = Plat_FloatTime();
if ( pLoader )
{
pList->RedoSort();
for ( int i = 0; i < pList->Count(); i++ )
{
char szFilename[MAX_PATH];
g_QueuedLoader.GetFilename( pList->Element( i ), szFilename, sizeof( szFilename ) );
if ( szFilename[0] )
{
if ( !pLoader->CreateResource( szFilename ) )
{
Warning( "QueuedLoader: Failed to create resource %s\n", szFilename );
}
}
}
}
// finished with list
pList->Purge();
*pBuildTime = Plat_FloatTime() - t0;
}
//-----------------------------------------------------------------------------
// Computation job to build out material objects
//-----------------------------------------------------------------------------
void CQueuedLoader::BuildMaterialResources( IResourcePreload *pLoader, ResourceList_t *pList, float *pBuildTime )
{
float t0 = Plat_FloatTime();
char szLastFilename[MAX_PATH];
szLastFilename[0] = '\0';
// ensure cubemaps are first
pList->RedoSort();
// run a clean operation to cull the non-patched env_cubemap materials, which are not built directly
for ( int i = 0; i < pList->Count(); i++ )
{
char szFilename[MAX_PATH];
char *pFilename = g_QueuedLoader.GetFilename( pList->Element( i ), szFilename, sizeof( szFilename ) );
if ( !V_stristr( pFilename, "maps\\" ) )
{
// list is sorted, first non-cubemap marks end of relevant list
break;
}
// skip past maps/mapname/
pFilename += 5;
pFilename = strchr( pFilename, '\\' ) + 1;
// back up until end of material name is found, need to strip off _%d_%d_%d.vmt
char *pEndFilename = V_stristr( pFilename, ".vmt" );
if ( !pEndFilename )
{
pEndFilename = pFilename + strlen( pFilename );
}
int numUnderscores = 3;
while ( pEndFilename != pFilename && numUnderscores > 0 )
{
pEndFilename--;
if ( pEndFilename[0] == '_' )
{
numUnderscores--;
}
}
if ( numUnderscores == 0 )
{
*pEndFilename = '\0';
if ( !V_strcmp( szLastFilename, pFilename ) )
{
// same cubemap material base already processed, skip it
continue;
}
V_strncpy( szLastFilename, pFilename, sizeof( szLastFilename ) );
strcat( pFilename, ".vmt" );
FileNameHandle_t hFilename = g_QueuedLoader.FindFilename( pFilename );
if ( hFilename )
{
pList->Remove( hFilename );
}
}
}
// process clean list
BuildResources( pLoader, pList, pBuildTime );
*pBuildTime = Plat_FloatTime() - t0;
}
//-----------------------------------------------------------------------------
// Called by multiple worker threads. Throttle the I/O to ensure too many
// buffers don't flood the work queue. Anonymous I/O is allowed to grow unbounded.
//-----------------------------------------------------------------------------
void AdjustAsyncIOSpeed()
{
if ( g_QueuedLoader.IsDynamic() == true )
{
return;
}
// throttle back the I/O to keep the pending buffers from exhausting memory
if ( g_SuspendIO == 0 )
{
if ( g_nIOMemory >= g_nHighIOSuspensionMark && g_nActiveJobs != 0 )
{
// protect against another worker thread
if ( g_SuspendIO.AssignIf( 0, 1 ) )
{
if ( g_QueuedLoader.GetSpewDetail() )
{
Msg( "QueuedLoader: Suspending I/O at %.2f MB\n", (float)g_nIOMemory / ( 1024.0f * 1024.0f ) );
}
g_pFullFileSystem->AsyncSuspend();
}
}
}
else if ( g_SuspendIO == 1 )
{
if ( g_nIOMemory <= g_nLowIOSuspensionMark )
{
// protect against another worker thread
if ( g_SuspendIO.AssignIf( 1, 0 ) )
{
if ( g_QueuedLoader.GetSpewDetail() )
{
Msg( "QueuedLoader: Resuming I/O at %.2f MB\n", (float)g_nIOMemory / ( 1024.0f * 1024.0f ) );
}
g_pFullFileSystem->AsyncResume();
}
}
}
}
//-----------------------------------------------------------------------------
// Computation job to do work after IO, runs callback
//-----------------------------------------------------------------------------
void IOComputationJob( FileJob_t *pFileJob, void *pData, int nSize, LoaderError_t loaderError )
{
int spewDetail = g_QueuedLoader.GetSpewDetail();
if ( spewDetail & ( LOADER_DETAIL_COMPLETIONS|LOADER_DETAIL_LATECOMPLETIONS ) )
{
const char *pLateString = "";
if ( !g_QueuedLoader.IsMapLoading() )
{
// completed outside of load process
pLateString = "(Late) ";
}
if ( ( spewDetail & LOADER_DETAIL_COMPLETIONS ) || ( ( spewDetail & LOADER_DETAIL_LATECOMPLETIONS ) && pLateString[0] ) )
{
char szFilename[MAX_PATH];
g_QueuedLoader.GetFilename( pFileJob->m_hFilename, szFilename, sizeof( szFilename ) );
Msg( "QueuedLoader: Computation:%8.8x, Size:%7d %s%s\n", ThreadGetCurrentId(), nSize, pLateString, szFilename );
}
}
if ( loaderError != LOADERERROR_NONE && pFileJob->m_bFileExists )
{
char szFilename[MAX_PATH];
g_QueuedLoader.GetFilename( pFileJob->m_hFilename, szFilename, sizeof( szFilename ) );
Warning( "QueuedLoader:: I/O Error on %s\n", szFilename );
}
pFileJob->m_nActualBytesRead = nSize;
pFileJob->m_LoaderError = loaderError;
if ( !pFileJob->m_pCallback )
{
// absent callback means resource loader want this system to delay buffer until ready for it
if ( !pFileJob->m_pTargetData )
{
// track it for later, unclaimed buffers will get freed
pFileJob->m_pTargetData = pData;
}
}
else
{
// regardless of error, call job callback so caller can do cleanup of their context
pFileJob->m_pCallback( pFileJob->m_pContext, pFileJob->m_pContext2, pData, nSize, loaderError );
if ( pFileJob->m_bFreeTargetAfterIO && pData )
{
// free our data only
g_pFullFileSystem->FreeOptimalReadBuffer( pData );
}
// memory has been consumed
g_nIOMemory -= nSize;
}
// mark as completed
pFileJob->m_bFinished = true;
pFileJob->m_FinishTime = Plat_MSTime();
pFileJob->m_ThreadId = ThreadGetCurrentId();
if ( pFileJob->m_Priority == LOADERPRIORITY_DURINGPRELOAD )
{
g_nHighPriorityJobs--;
}
else if ( pFileJob->m_Priority == LOADERPRIORITY_BEFOREPLAY )
{
g_nJobsToFinishBeforePlay--;
}
g_nQueuedJobs--;
if ( g_nQueuedJobs == 0 && ( spewDetail & LOADER_DETAIL_TIMING ) )
{
Msg( "QueuedLoader: Finished I/O of all queued jobs!\n" );
}
AdjustAsyncIOSpeed();
}
//-----------------------------------------------------------------------------
// Computation job to do work after anonymous job was asynchronously claimed, runs callback.
//-----------------------------------------------------------------------------
void FinishAnonymousJob( FileJob_t *pFileJob, QueuedLoaderCallback_t pCallback, void *pContext, void *pContext2 )
{
// regardless of error, call job callback so caller can do cleanup of their context
pCallback( pContext, pContext2, pFileJob->m_pTargetData, pFileJob->m_nActualBytesRead, pFileJob->m_LoaderError );
if ( pFileJob->m_bFreeTargetAfterIO && pFileJob->m_pTargetData )
{
// free our data only
g_pFullFileSystem->FreeOptimalReadBuffer( pFileJob->m_pTargetData );
pFileJob->m_pTargetData = NULL;
}
pFileJob->m_bClaimed = true;
// memory has been consumed
g_nAnonymousIOMemory -= pFileJob->m_nActualBytesRead;
}
//-----------------------------------------------------------------------------
// Callback from I/O job thread. Purposely lightweight as possible to keep i/o from stalling.
//-----------------------------------------------------------------------------
void IOAsyncCallback( const FileAsyncRequest_t &asyncRequest, int numReadBytes, FSAsyncStatus_t asyncStatus )
{
FileJob_t *pFileJob = (FileJob_t *)asyncRequest.pContext;
// interpret the async error
LoaderError_t loaderError;
switch ( asyncStatus )
{
case FSASYNC_OK:
loaderError = LOADERERROR_NONE;
break;
case FSASYNC_ERR_FILEOPEN:
loaderError = LOADERERROR_FILEOPEN;
break;
default:
loaderError = LOADERERROR_READING;
}
// track how much i/o data is in flight, consumption will decrement
if ( !pFileJob->m_pCallback )
{
// anonymous io memory is tracked seperatley
g_nAnonymousIOMemory += numReadBytes;
if ( g_nAnonymousIOMemory > g_nAnonymousIOMemoryPeak )
{
g_nAnonymousIOMemoryPeak = g_nAnonymousIOMemory;
}
}
else
{
g_nIOMemory += numReadBytes;
if ( g_nIOMemory > g_nIOMemoryPeak )
{
g_nIOMemoryPeak = g_nIOMemory;
}
}
// have data or error, do callback as a computation job
if ( !g_QueuedLoader.IsDynamic() )
{
g_pThreadPool->QueueCall( IOComputationJob, pFileJob, asyncRequest.pData, numReadBytes, loaderError )->Release();
}
else
{
g_QueuedLoader.QueueDynamicLoadFunctor( CreateFunctor( IOComputationJob, pFileJob, asyncRequest.pData, numReadBytes, loaderError ) );
}
// don't let the i/o starve, possibly get some more work from the pending queue
g_QueuedLoader.SubmitPendingJobs();
// possibly goes to zero atomically, AFTER submission
// prevents contention between main thread
--g_nActiveJobs;
}
//-----------------------------------------------------------------------------
// Public method to filename dictionary
//-----------------------------------------------------------------------------
char *CQueuedLoader::GetFilename( const FileNameHandle_t hFilename, char *pBuff, int nBuffSize )
{
m_Filenames.String( hFilename, pBuff, nBuffSize );
return pBuff;
}
//-----------------------------------------------------------------------------
// Public method to filename dictionary
//-----------------------------------------------------------------------------
FileNameHandle_t CQueuedLoader::FindFilename( const char *pFilename )
{
return m_Filenames.FindFileName( pFilename );
}
//-----------------------------------------------------------------------------
// Sort function for resource names.
//-----------------------------------------------------------------------------
bool CQueuedLoader::CResourceNameLessFunc::Less( const FileNameHandle_t &hFilenameLHS, const FileNameHandle_t &hFilenameRHS, void *pCtx )
{
switch ( (int)pCtx )
{
case RESOURCEPRELOAD_MATERIAL:
{
// Cubemap materials are expected to be at top of list
char szNameLHS[MAX_PATH];
char szNameRHS[MAX_PATH];
const char *pNameLHS = g_QueuedLoader.GetFilename( hFilenameLHS, szNameLHS, sizeof( szNameLHS ) );
const char *pNameRHS = g_QueuedLoader.GetFilename( hFilenameRHS, szNameRHS, sizeof( szNameRHS ) );
bool bIsCubemapLHS = V_stristr( pNameLHS, "maps\\" ) != NULL;
bool bIsCubemapRHS = V_stristr( pNameRHS, "maps\\" ) != NULL;
if ( bIsCubemapLHS != bIsCubemapRHS )
{
return ( bIsCubemapLHS == true && bIsCubemapRHS == false );
}
return ( V_stricmp( pNameLHS, pNameRHS ) < 0 );
}
break;
default:
// sort not really needed, just use numeric handles
return ( hFilenameLHS < hFilenameRHS );
}
}
//-----------------------------------------------------------------------------
// Resolve filenames to expected disc layout order as...
// bsp, graphs, platform, hl2, episodic, ep2, tf, portal, non-zip
// see XGD layout.
//-----------------------------------------------------------------------------
int CQueuedLoader::CFileJobsLessFunc::GetLayoutOrderForFilename( const char *pFilename )
{
bool bIsLocalizedZip = false;
if ( XBX_IsLocalized() )
{
if ( V_stristr( pFilename, "\\zip" ) && V_stristr( pFilename, XBX_GetLanguageString() ) )
{
bIsLocalizedZip = true;
}
}
int order;
if ( V_stristr( pFilename, "\\maps\\" ) )
{
// bsp's and graphs on the opposite layer, these must be topmost
// the queued loader is expecting to do these first, all at once
// this allows for a single layer switch
if ( V_stristr( pFilename, "\\graphs\\" ) )
{
order = 1;
}
else
{
order = 0;
}
}
else if ( V_stristr( pFilename, "\\platform\\zip" ) )
{
order = 2;
}
else if ( V_stristr( pFilename, "\\hl2\\zip" ) )
{
order = 3;
}
else if ( V_stristr( pFilename, "\\episodic\\zip" ) )
{
order = 4;
}
else if ( V_stristr( pFilename, "\\ep2\\zip" ) )
{
order = 5;
}
else if ( V_stristr( pFilename, "\\tf\\zip" ) )
{
order = 6;
}
else if ( V_stristr( pFilename, "\\portal\\zip" ) )
{
order = 7;
}
else
{
// other
order = 8;
}
// localized zips have same relative sort order, but after all other zips
return bIsLocalizedZip ? 10*order : order;
}
//-----------------------------------------------------------------------------
// Sort function, high priority jobs sort first, then offset, then zip
//-----------------------------------------------------------------------------
bool CQueuedLoader::CFileJobsLessFunc::Less( FileJob_t* const &pFileJobLHS, FileJob_t* const &pFileJobRHS, void *pCtx )
{
if ( pFileJobLHS->m_Priority != pFileJobRHS->m_Priority )
{
// higher priorities sort to top
return ( pFileJobLHS->m_Priority > pFileJobRHS->m_Priority );
}
if ( pFileJobLHS->m_hFilename == pFileJobRHS->m_hFilename )
{
// same file (zip), sort by offset
return pFileJobLHS->m_nStartOffset < pFileJobRHS->m_nStartOffset;
}
char szFilenameLHS[MAX_PATH];
char szFilenameRHS[MAX_PATH];
g_QueuedLoader.GetFilename( pFileJobLHS->m_hFilename, szFilenameLHS, sizeof( szFilenameLHS ) );
g_QueuedLoader.GetFilename( pFileJobRHS->m_hFilename, szFilenameRHS, sizeof( szFilenameRHS ) );
// resolve filename to match disk layout of zips
int layoutLHS = GetLayoutOrderForFilename( szFilenameLHS );
int layoutRHS = GetLayoutOrderForFilename( szFilenameRHS );
if ( layoutLHS != layoutRHS )
{
return layoutLHS < layoutRHS;
}
return CaselessStringLessThan( szFilenameLHS, szFilenameRHS );
}
//-----------------------------------------------------------------------------
// Dump the queue contents to the file system.
//-----------------------------------------------------------------------------
void CQueuedLoader::SubmitPendingJobs()
{
// prevents contention between I/O and main thread attempting to submit
if ( ThreadInMainThread() && g_nActiveJobs != 0 && m_bDynamic == false )
{
// main thread can only kick start work if the I/O is idle
// once the I/O is kicked off, the I/O thread is responsible for continual draining
return;
}
else if ( !ThreadInMainThread() && g_nActiveJobs != 1 && m_bDynamic == false )
{
// I/O thread requests more work, but will only fall through and get some when it expects to go idle
// I/O thread still has jobs and doesn't need any more yet
return;
}
CTSList<FileJob_t *>::Node_t *pNode = m_PendingJobs.Detach();
if ( !pNode )
{
return;
}
// used by spew to indicate submission blocks
m_nSubmitCount++;
// sort entries
CUtlSortVector< FileJob_t*, CFileJobsLessFunc > sortedFiles( 0, 128 );
while ( pNode )
{
FileJob_t *pFileJob = pNode->elem;
sortedFiles.InsertNoSort( pFileJob );
CTSList<FileJob_t *>::Node_t *pNext = (CTSList<FileJob_t *>::Node_t*)pNode->Next;
delete pNode;
pNode = pNext;
}
sortedFiles.RedoSort();
FileAsyncRequest_t asyncRequest;
asyncRequest.pfnCallback = IOAsyncCallback;
char szFilename[MAX_PATH];
for ( int i = 0; i<sortedFiles.Count(); i++ )
{
FileJob_t *pFileJob = sortedFiles[i];
pFileJob->m_SubmitTag = m_nSubmitCount;
pFileJob->m_SubmitTime = Plat_MSTime();
m_SubmittedJobs.AddToTail( pFileJob );
// build an async request
if ( pFileJob->m_Priority == LOADERPRIORITY_DURINGPRELOAD )
{
// must finish during preload
asyncRequest.priority = PRIORITY_HIGH;
g_nHighPriorityJobs++;
}
else if ( pFileJob->m_Priority == LOADERPRIORITY_BEFOREPLAY )
{
// must finish before gameplay
asyncRequest.priority = PRIORITY_NORMAL;
g_nJobsToFinishBeforePlay++;
}
else
{
// can finish during gameplay, normal priority
asyncRequest.priority = PRIORITY_NORMAL;
}
// async will allocate unless caller provided a target
// loader always takes ownership of buffer
asyncRequest.pData = pFileJob->m_pTargetData;
asyncRequest.flags = pFileJob->m_pTargetData ? 0 : FSASYNC_FLAGS_ALLOCNOFREE;
asyncRequest.nOffset = pFileJob->m_nStartOffset;
asyncRequest.nBytes = pFileJob->m_nBytesToRead;
asyncRequest.pszFilename = GetFilename( pFileJob->m_hFilename, szFilename, sizeof( szFilename ) );
asyncRequest.pContext = (void *)pFileJob;
if ( pFileJob->m_bFileExists )
{
// start the valid async request
g_nActiveJobs++;
g_pFullFileSystem->AsyncRead( asyncRequest, &pFileJob->m_hAsyncControl );
}
else
{
// prevent dragging the i/o system down for known failures
// still need to do callback so subsystems can do the right thing based on file absence
if ( IsDynamic() )
QueueDynamicLoadFunctor( CreateFunctor( IOComputationJob, pFileJob, pFileJob->m_pTargetData, 0, LOADERERROR_FILEOPEN ) );
else
g_pThreadPool->QueueCall( IOComputationJob, pFileJob, pFileJob->m_pTargetData, 0, LOADERERROR_FILEOPEN )->Release();
}
}
}
//-----------------------------------------------------------------------------
// Add to queue
//-----------------------------------------------------------------------------
bool CQueuedLoader::AddJob( const LoaderJob_t *pLoaderJob )
{
if ( !m_bActive )
{
return false;
}
Assert( pLoaderJob && pLoaderJob->m_pFilename );
if ( m_bCanBatch && !m_bBatching )
{
// should have been part of pre-load batch
DevWarning( "QueuedLoader: Late Queued Job: %s\n", pLoaderJob->m_pFilename );
}
// anonymous jobs lack callbacks and are heavily restricted to ensure their stability
// the caller is expected to claim these before load ends (which auto-purges them)
if ( !pLoaderJob->m_pCallback && pLoaderJob->m_Priority == LOADERPRIORITY_ANYTIME )
{
Assert( 0 );
DevWarning( "QueuedLoader: Ignoring Anonymous Job: %s\n", pLoaderJob->m_pFilename );
return false;
}
MEM_ALLOC_CREDIT();
// all bsp based files get forced to a higher priority in order to achieve a clustered sort
// the bsp files are not going to be anywhere near the zips, thus we don't want head thrashing
bool bFileIsFromBSP;
bool bExists = false;
char *pFullPath;
char szFullPath[MAX_PATH];
if ( V_IsAbsolutePath( pLoaderJob->m_pFilename ) )
{
// an absolute path is trusted, take as is
pFullPath = (char *)pLoaderJob->m_pFilename;
bFileIsFromBSP = V_stristr( pFullPath, ".bsp" ) != NULL;
bExists = true;
}
else
{
// must resolve now, all submitted paths must be absolute for proper sort which achieves seek linearization
// a resolved absolute file ensures its existence
PathTypeFilter_t pathFilter = FILTER_NONE;
if ( IsX360() && ( g_pFullFileSystem->GetDVDMode() == DVDMODE_STRICT ) )
{
if ( V_stristr( pLoaderJob->m_pFilename, ".bsp" ) || V_stristr( pLoaderJob->m_pFilename, ".ain" ) )
{
// only the bsp/ain are allowed to be external
pathFilter = FILTER_CULLPACK;
}
else
{
// all files are expected to be in zip
pathFilter = FILTER_CULLNONPACK;
}
}
PathTypeQuery_t pathType;
g_pFullFileSystem->RelativePathToFullPath( pLoaderJob->m_pFilename, pLoaderJob->m_pPathID, szFullPath, sizeof( szFullPath ), pathFilter, &pathType );
bExists = V_IsAbsolutePath( szFullPath );
pFullPath = szFullPath;
bFileIsFromBSP = ( (pathType & PATH_IS_MAPPACKFILE) != 0 );
}
// create a file job
FileJob_t *pFileJob = new FileJob_t;
pFileJob->m_hFilename = m_Filenames.FindOrAddFileName( pFullPath );
pFileJob->m_bFileExists = bExists;
pFileJob->m_pCallback = pLoaderJob->m_pCallback;
pFileJob->m_pContext = pLoaderJob->m_pContext;
pFileJob->m_pContext2 = pLoaderJob->m_pContext2;
pFileJob->m_pTargetData = pLoaderJob->m_pTargetData;
pFileJob->m_nBytesToRead = pLoaderJob->m_nBytesToRead;
pFileJob->m_nStartOffset = pLoaderJob->m_nStartOffset;
pFileJob->m_Priority = bFileIsFromBSP ? LOADERPRIORITY_DURINGPRELOAD : pLoaderJob->m_Priority;
if ( pLoaderJob->m_pTargetData )
{
// never free caller's buffer, if they provide, they have to free it
pFileJob->m_bFreeTargetAfterIO = false;
}
else
{
// caller can take over ownership, otherwise it gets freed after I/O
pFileJob->m_bFreeTargetAfterIO = ( pLoaderJob->m_bPersistTargetData == false );
}
if ( !pLoaderJob->m_pCallback )
{
// track anonymous jobs
AUTO_LOCK( m_Mutex );
char szFixedName[MAX_PATH];
V_strncpy( szFixedName, pLoaderJob->m_pFilename, sizeof( szFixedName ) );
V_FixSlashes( szFixedName );
m_AnonymousJobs.Insert( szFixedName, pFileJob );
}
g_nQueuedJobs++;
if ( m_bBatching )
{
m_BatchedJobs.PushItem( pFileJob );
}
else
{
m_PendingJobs.PushItem( pFileJob );
SubmitPendingJobs();
}
return true;
}
//-----------------------------------------------------------------------------
// Allows an external system to append to a map's reslist. The next map load
// will append these specified files. Unhandled resources will just get
// quietly discarded. An external system could use this to patch a hole
// or prevent a purge.
//-----------------------------------------------------------------------------
void CQueuedLoader::AddMapResource( const char *pFilename )
{
if ( !pFilename || !pFilename[0] )
{
// pointless
return;
}
// normalize the provided name as a filename
char szFilename[MAX_PATH];
V_strncpy( szFilename, pFilename, sizeof( szFilename ) );
V_FixSlashes( szFilename );
V_strlower( szFilename );
if ( m_AdditionalResources.Find( szFilename ) != UTL_INVAL_SYMBOL )
{
// already added
return;
}
m_AdditionalResources.AddString( szFilename );
}
//-----------------------------------------------------------------------------
// Asynchronous claim for an anonymous job.
// This allows loaders with deep dependencies to get their data in flight, and then claim it
// when the they are in a state to consume it.
//-----------------------------------------------------------------------------
bool CQueuedLoader::ClaimAnonymousJob( const char *pFilename, QueuedLoaderCallback_t pCallback, void *pContext, void *pContext2 )
{
Assert( ThreadInMainThread() );
Assert( pFilename && pCallback && !m_bBatching );
char szFixedName[MAX_PATH];
V_strncpy( szFixedName, pFilename, sizeof( szFixedName ) );
V_FixSlashes( szFixedName );
pFilename = szFixedName;
int iIndex = m_AnonymousJobs.Find( pFilename );
if ( iIndex == m_AnonymousJobs.InvalidIndex() )
{
// unknown
DevWarning( "QueuedLoader: Anonymous Job '%s' not found\n", pFilename );
return false;
}
// caller is claiming
FileJob_t *pFileJob = m_AnonymousJobs[iIndex];
if ( !pFileJob->m_bFinished )
{
// unfinished shouldn't happen and caller can't have it
// anonymous jobs and their claims are very restrictive in such a way to provide stability
// this dead job will get auto-cleaned at end of map loading
Assert( 0 );
return false;
}
m_AnonymousJobs.RemoveAt( iIndex );
g_pThreadPool->QueueCall( FinishAnonymousJob, pFileJob, pCallback, pContext, pContext2 )->Release();
return true;
}
//-----------------------------------------------------------------------------
// Synchronous claim for an anonymous job. This allows loaders
// with deep dependencies to get their data in flight, and then claim it
// when the they are in a state to consume it.
//-----------------------------------------------------------------------------
bool CQueuedLoader::ClaimAnonymousJob( const char *pFilename, void **pData, int *pDataSize, LoaderError_t *pError )
{
Assert( ThreadInMainThread() );
Assert( pFilename && !m_bBatching );
char szFixedName[MAX_PATH];
V_strncpy( szFixedName, pFilename, sizeof( szFixedName ) );
V_FixSlashes( szFixedName );
pFilename = szFixedName;
int iIndex = m_AnonymousJobs.Find( pFilename );
if ( iIndex == m_AnonymousJobs.InvalidIndex() )
{
// unknown
DevWarning( "QueuedLoader: Anonymous Job '%s' not found\n", pFilename );
return false;
}
// caller is claiming
FileJob_t *pFileJob = m_AnonymousJobs[iIndex];
if ( !pFileJob->m_bFinished )
{
// unfinished shouldn't happen and caller can't have it
// anonymous jobs and their claims are very restrictive in such a way to provide stability
// this dead job will get auto-cleaned at end of map loading
Assert( 0 );
return false;
}
pFileJob->m_bClaimed = true;
m_AnonymousJobs.RemoveAt( iIndex );
*pData = pFileJob->m_pTargetData;
*pDataSize = pFileJob->m_LoaderError == LOADERERROR_NONE ? pFileJob->m_nActualBytesRead : 0;
if ( pError )
{
*pError = pFileJob->m_LoaderError;
}
// caller owns the data, regardless of how the job was setup
pFileJob->m_pTargetData = NULL;
// memory has been consumed
g_nAnonymousIOMemory -= pFileJob->m_nActualBytesRead;
return true;
}
//-----------------------------------------------------------------------------
// End of batching. Moves jobs into pending queue and submits but does not wait
//-----------------------------------------------------------------------------
void CQueuedLoader::SubmitBatchedJobs()
{
// end of batching
m_bBatching = false;
CTSList<FileJob_t *>::Node_t *pNode = m_BatchedJobs.Detach();
if ( !pNode )
{
return;
}
// must wait for any initial i/o to finish
// i/o thread must stop in order to submit all the batched jobs atomically
// and get an accurate accounting of high priority jobs
while ( g_nActiveJobs != 0 )
{
g_pThreadPool->Yield( MAIN_THREAD_YIELD_TIME );
}
// dump batched jobs to pending jobs
while ( pNode )
{
FileJob_t *pFileJob = pNode->elem;
m_PendingJobs.PushItem( pFileJob );
CTSList<FileJob_t *>::Node_t *pNext = (CTSList<FileJob_t *>::Node_t*)pNode->Next;
delete pNode;
pNode = pNext;
}
SubmitPendingJobs();
if ( GetSpewDetail() )
{
Msg( "QueuedLoader: High Priority Jobs: %d\n", (int)g_nHighPriorityJobs );
}
}
//-----------------------------------------------------------------------------
// End of batching. High priority jobs are guaranteed completed before function returns.
//-----------------------------------------------------------------------------
void CQueuedLoader::SubmitBatchedJobsAndWait()
{
SubmitBatchedJobs();
// finish only the high priority jobs
// high priority jobs are expected to be complete at the conclusion of batching
int total = g_nHighPriorityJobs;
while ( g_nHighPriorityJobs != 0 )
{
float t = (float)( total - g_nHighPriorityJobs ) / (float)total;
m_pProgress->UpdateProgress( PROGRESS_IO + t * ( 1.0f - PROGRESS_IO ) );
// yield some time
g_pThreadPool->Yield( MAIN_THREAD_YIELD_TIME );
}
}
//-----------------------------------------------------------------------------
// Clean queue of stale entries. Active entries are skipped.
//-----------------------------------------------------------------------------
void CQueuedLoader::CleanQueue()
{
for ( int i = 0; i<RESOURCEPRELOAD_COUNT; i++ )
{
m_ResourceNames[i].Purge();
}
m_BatchedJobs.Purge();
int iIndex = m_SubmittedJobs.Head();
while ( iIndex != m_SubmittedJobs.InvalidIndex() )
{
int iNext = m_SubmittedJobs.Next( iIndex );
FileJob_t *pFileJob = m_SubmittedJobs[iIndex];
if ( pFileJob->m_bFinished )
{
// job is complete, safe to free
m_SubmittedJobs.Free( iIndex );
g_pFullFileSystem->AsyncRelease( pFileJob->m_hAsyncControl );
delete pFileJob;
}
iIndex = iNext;
}
m_Filenames.RemoveAll();
}
//-----------------------------------------------------------------------------
// Abandon queue
//-----------------------------------------------------------------------------
void CQueuedLoader::PurgeQueue()
{
}
//-----------------------------------------------------------------------------
// Spew info abut queued load
//-----------------------------------------------------------------------------
void CQueuedLoader::SpewInfo()
{
Msg( "Queued Loader:\n\n" );
int totalClaimed = 0;
int totalUnclaimed = 0;
if ( IsFinished() || m_bDynamic == true )
{
// can only access submitted jobs safely when io thread complete
int lastPriority = -1;
int iIndex = m_SubmittedJobs.Head();
while ( iIndex != m_SubmittedJobs.InvalidIndex() )
{
FileJob_t *pFileJob = m_SubmittedJobs[iIndex];
int asyncDuration = -1;
if ( pFileJob->m_FinishTime )
{
asyncDuration = pFileJob->m_FinishTime - pFileJob->m_SubmitTime;
}
if ( pFileJob->m_Priority != lastPriority )
{
switch ( pFileJob->m_Priority )
{
case LOADERPRIORITY_DURINGPRELOAD:
Msg( "---- FINISH DURING PRELOAD ( HIGH PRIORITY )----\n" );
break;
case LOADERPRIORITY_BEFOREPLAY:
Msg( "---- FINISH BEFORE GAMEPLAY ( NORMAL PRIORITY )----\n" );
break;
case LOADERPRIORITY_ANYTIME:
Msg( "---- FINISH ANYTIME ( NORMAL PRIORITY )----\n" );
break;
}
lastPriority = pFileJob->m_Priority;
}
char szAnonymousString[MAX_PATH];
const char *pAnonymousStatus = "";
if ( !pFileJob->m_pCallback )
{
V_snprintf( szAnonymousString, sizeof( szAnonymousString ), "(%s) ", pFileJob->m_bClaimed ? "Claimed" : "Unclaimed" );
pAnonymousStatus = szAnonymousString;
if ( pFileJob->m_bClaimed )
{
totalClaimed += pFileJob->m_nActualBytesRead;
}
else
{
totalUnclaimed += pFileJob->m_nActualBytesRead;
}
}
char szFilename[MAX_PATH];
Msg( "Submit:%5dms AsyncDuration:%5dms Tag:%d Thread:%8.8x Size:%7d %s%s\n",
pFileJob->m_SubmitTime - m_StartTime,
asyncDuration,
pFileJob->m_SubmitTag,
pFileJob->m_ThreadId,
pFileJob->m_nActualBytesRead,
pAnonymousStatus,
GetFilename( pFileJob->m_hFilename, szFilename, sizeof( szFilename ) ) );
iIndex = m_SubmittedJobs.Next( iIndex );
}
Msg( "%d Total Jobs\n", m_SubmittedJobs.Count() );
}
Msg( "%d Queued Jobs\n", (int)g_nQueuedJobs );
Msg( "%d Active Jobs\n", (int)g_nActiveJobs );
Msg( "Peak IO Memory: %.2f MB\n", (float)g_nIOMemoryPeak / ( 1024.0f * 1024.0f ) );
Msg( "Peak Anonymous IO Memory: %.2f MB\n", (float)g_nAnonymousIOMemoryPeak / ( 1024.0f * 1024.0f ) );
Msg( " Total Anonymous Claimed: %d\n", totalClaimed );
Msg( " Total Anonymous Unclaimed: %d\n", totalUnclaimed );
if ( m_EndTime )
{
Msg( "Queuing Duration: %dms\n", m_EndTime - m_StartTime );
}
}
//-----------------------------------------------------------------------------
// Initialization
//-----------------------------------------------------------------------------
InitReturnVal_t CQueuedLoader::Init()
{
InitReturnVal_t nRetVal = BaseClass::Init();
if ( nRetVal != INIT_OK )
{
return nRetVal;
}
return INIT_OK;
}
//-----------------------------------------------------------------------------
// Shutdown
//-----------------------------------------------------------------------------
void CQueuedLoader::Shutdown()
{
BaseClass::Shutdown();
}
//-----------------------------------------------------------------------------
// Install a type specific interface from managing system.
//-----------------------------------------------------------------------------
void CQueuedLoader::InstallLoader( ResourcePreload_t type, IResourcePreload *pLoader )
{
m_pLoaders[type] = pLoader;
}
void CQueuedLoader::InstallProgress( ILoaderProgress *pProgress )
{
m_pProgress = pProgress;
}
//-----------------------------------------------------------------------------
// Invoke the loader systems to purge dead resources
//-----------------------------------------------------------------------------
void CQueuedLoader::PurgeUnreferencedResources()
{
ResourcePreload_t purgeOrder[RESOURCEPRELOAD_COUNT];
// the purge operations require a specific order (models and cubemaps before materials)
int numPurges = 0;
purgeOrder[numPurges++] = RESOURCEPRELOAD_SOUND;
purgeOrder[numPurges++] = RESOURCEPRELOAD_STATICPROPLIGHTING;
purgeOrder[numPurges++] = RESOURCEPRELOAD_MODEL;
purgeOrder[numPurges++] = RESOURCEPRELOAD_CUBEMAP;
purgeOrder[numPurges++] = RESOURCEPRELOAD_MATERIAL;
// iterate according to order
for ( int i = 0; i < numPurges; i++ )
{
ResourcePreload_t loader = purgeOrder[i];
if ( m_pLoaders[loader] )
{
m_pLoaders[loader]->PurgeUnreferencedResources();
}
}
m_pProgress->UpdateProgress( PROGRESS_PREPURGE );
}
//-----------------------------------------------------------------------------
// Invoke the loader systems to purge all resources, if possible
//-----------------------------------------------------------------------------
void CQueuedLoader::PurgeAll()
{
ResourcePreload_t purgeOrder[RESOURCEPRELOAD_COUNT];
// the purge operations require a specific order (models and cubemaps before materials)
int numPurges = 0;
purgeOrder[numPurges++] = RESOURCEPRELOAD_SOUND;
purgeOrder[numPurges++] = RESOURCEPRELOAD_STATICPROPLIGHTING;
purgeOrder[numPurges++] = RESOURCEPRELOAD_MODEL;
purgeOrder[numPurges++] = RESOURCEPRELOAD_CUBEMAP;
purgeOrder[numPurges++] = RESOURCEPRELOAD_MATERIAL;
// iterate according to order
for ( int i = 0; i < numPurges; i++ )
{
ResourcePreload_t loader = purgeOrder[i];
if ( m_pLoaders[loader] )
{
m_pLoaders[loader]->PurgeAll();
}
}
*m_szMapNameToCompareSame = 0;
}
//-----------------------------------------------------------------------------
// Invoke the loader systems to request i/o jobs, which are batched.
//-----------------------------------------------------------------------------
void CQueuedLoader::GetJobRequests()
{
COM_TimestampedLog( "CQueuedLoader::GetJobRequests - Start" );
// causes the batch queue to fill with i/o requests
m_bCanBatch = true;
m_bBatching = true;
float t0 = Plat_FloatTime();
if ( !IsPC() && !m_bDynamic )
{
// cubemap textures must be first to install correctly before their cubemap materials are built (and precache the cubmeap textures)
// cannot be overlapped, must run serially
BuildResources( m_pLoaders[RESOURCEPRELOAD_CUBEMAP], &m_ResourceNames[RESOURCEPRELOAD_CUBEMAP], &m_LoaderTimes[RESOURCEPRELOAD_CUBEMAP] );
// Overlapping these is not critical in any way, total time is currently < 2 seconds.
// These operations flood calls (AddJob) back into the queued loader (which has to mutex its lists),
// so in fact it's slightly slower to queue these at this stage. As these routines age they may become more heavyweight.
CJob *jobs[5];
jobs[0] = g_pThreadPool->QueueCall( BuildResources, m_pLoaders[RESOURCEPRELOAD_SOUND], &m_ResourceNames[RESOURCEPRELOAD_SOUND], &m_LoaderTimes[RESOURCEPRELOAD_SOUND] );
jobs[1] = g_pThreadPool->QueueCall( BuildMaterialResources, m_pLoaders[RESOURCEPRELOAD_MATERIAL], &m_ResourceNames[RESOURCEPRELOAD_MATERIAL], &m_LoaderTimes[RESOURCEPRELOAD_MATERIAL] );
jobs[2] = g_pThreadPool->QueueCall( BuildResources, m_pLoaders[RESOURCEPRELOAD_STATICPROPLIGHTING], &m_ResourceNames[RESOURCEPRELOAD_STATICPROPLIGHTING], &m_LoaderTimes[RESOURCEPRELOAD_STATICPROPLIGHTING] );
jobs[3] = g_pThreadPool->QueueCall( BuildResources, m_pLoaders[RESOURCEPRELOAD_MODEL], &m_ResourceNames[RESOURCEPRELOAD_MODEL], &m_LoaderTimes[RESOURCEPRELOAD_MODEL] );
jobs[4] = g_pThreadPool->QueueCall( BuildResources, m_pLoaders[RESOURCEPRELOAD_ANONYMOUS], &m_ResourceNames[RESOURCEPRELOAD_ANONYMOUS], &m_LoaderTimes[RESOURCEPRELOAD_ANONYMOUS] );
// all jobs must finish
float flLastUpdateT = -1000.0f;
// Update as if this takes 2 seconds
float flDelta = ( PROGRESS_CREATEDRESOURCES - PROGRESS_PARSEDRESLIST ) * 0.03 / 2.0f;
float flProgress = PROGRESS_PARSEDRESLIST;
while( true )
{
bool bIsDone = true;
for ( int i=0; i<ARRAYSIZE( jobs ); i++ )
{
if ( !jobs[i]->IsFinished() )
{
bIsDone = false;
break;
}
}
if ( bIsDone )
break;
// Can't sleep; that will allow this thread to be used by the thread pool
float newt = Plat_FloatTime();
if ( newt - flLastUpdateT > .03 )
{
m_pProgress->UpdateProgress( flProgress );
flProgress = clamp( flProgress + flDelta, PROGRESS_PARSEDRESLIST, PROGRESS_CREATEDRESOURCES );
// Necessary to take into account any waits for vsync
flLastUpdateT = Plat_FloatTime();
}
}
for ( int i=0; i<ARRAYSIZE( jobs ); i++ )
{
jobs[i]->Release();
}
}
else
{
BuildResources( m_pLoaders[RESOURCEPRELOAD_CUBEMAP], &m_ResourceNames[RESOURCEPRELOAD_CUBEMAP], &m_LoaderTimes[RESOURCEPRELOAD_CUBEMAP] );
BuildResources( m_pLoaders[RESOURCEPRELOAD_SOUND], &m_ResourceNames[RESOURCEPRELOAD_SOUND], &m_LoaderTimes[RESOURCEPRELOAD_SOUND] );
BuildMaterialResources( m_pLoaders[RESOURCEPRELOAD_MATERIAL], &m_ResourceNames[RESOURCEPRELOAD_MATERIAL], &m_LoaderTimes[RESOURCEPRELOAD_MATERIAL] );
BuildResources( m_pLoaders[RESOURCEPRELOAD_STATICPROPLIGHTING], &m_ResourceNames[RESOURCEPRELOAD_STATICPROPLIGHTING], &m_LoaderTimes[RESOURCEPRELOAD_STATICPROPLIGHTING] );
BuildResources( m_pLoaders[RESOURCEPRELOAD_MODEL], &m_ResourceNames[RESOURCEPRELOAD_MODEL], &m_LoaderTimes[RESOURCEPRELOAD_MODEL] );
BuildResources( m_pLoaders[RESOURCEPRELOAD_ANONYMOUS], &m_ResourceNames[RESOURCEPRELOAD_ANONYMOUS], &m_LoaderTimes[RESOURCEPRELOAD_ANONYMOUS] );
}
if ( g_QueuedLoader.GetSpewDetail() & LOADER_DETAIL_TIMING )
{
for ( int i = RESOURCEPRELOAD_UNKNOWN+1; i<RESOURCEPRELOAD_COUNT; i++ )
{
Msg( "QueuedLoader: %s Creating: %.2f seconds\n", g_ResourceLoaderNames[i], m_LoaderTimes[i] );
}
Msg( "QueuedLoader: Total Creating: %.2f seconds\n", Plat_FloatTime() - t0 );
}
m_pProgress->UpdateProgress( PROGRESS_CREATEDRESOURCES );
COM_TimestampedLog( "CQueuedLoader::GetJobRequests - End" );
}
void CQueuedLoader::AddResourceToTable( const char *pFilename )
{
const char *pExt = V_GetFileExtension( pFilename );
if ( !pExt )
{
// unknown
// all resources are identified by their extension
return;
}
const char *pTypeDir = NULL;
const char *pName = pFilename;
ResourcePreload_t type = RESOURCEPRELOAD_UNKNOWN;
if ( !V_stricmp( pExt, "wav" ) )
{
type = RESOURCEPRELOAD_SOUND;
pTypeDir = "sound\\";
}
else if ( !V_stricmp( pExt, "vmt" ) )
{
type = RESOURCEPRELOAD_MATERIAL;
pTypeDir = "materials\\";
}
else if ( !V_stricmp( pExt, "vtf" ) )
{
if ( V_stristr( pFilename, "maps\\" ) )
{
// only want cubemap textures
if ( !m_bLoadForHDR && V_stristr( pFilename, ".hdr." ) )
{
return;
}
else if ( m_bLoadForHDR && !V_stristr( pFilename, ".hdr." ) )
{
return;
}
type = RESOURCEPRELOAD_CUBEMAP;
pTypeDir = "materials\\";
}
else
{
return;
}
}
else if ( !V_stricmp( pExt, "mdl" ) )
{
type = RESOURCEPRELOAD_MODEL;
pTypeDir = "models\\";
}
else if ( !V_stricmp( pExt, "vhv" ) )
{
// want static props only
pName = V_stristr( pFilename, "sp_" );
if ( !pName )
{
return;
}
if ( !m_bLoadForHDR && V_stristr( pFilename, "_hdr_" ) )
{
return;
}
else if ( m_bLoadForHDR && !V_stristr( pFilename, "_hdr_" ) )
{
return;
}
type = RESOURCEPRELOAD_STATICPROPLIGHTING;
}
else
{
// unknown, ignored
return;
}
if ( pTypeDir )
{
// want object name only
// skip past game/type directory prefixing
const char *pDir = V_stristr( pName, pTypeDir );
if ( pDir )
{
pName = pDir + strlen( pTypeDir );
}
}
FileNameHandle_t hFilename = m_Filenames.FindOrAddFileName( pName );
m_ResourceNames[type].InsertNoSort( hFilename );
}
//-----------------------------------------------------------------------------
// Parse the raw resource list into resource dictionaries
//-----------------------------------------------------------------------------
void CQueuedLoader::ParseResourceList( CUtlBuffer &resourceList )
{
// parse resource list into known types
characterset_t breakSet;
CharacterSetBuild( &breakSet, "" );
char szToken[MAX_PATH];
for ( ;; )
{
int nTokenSize = resourceList.ParseToken( &breakSet, szToken, sizeof( szToken ) );
if ( nTokenSize <= 0 )
{
break;
}
AddResourceToTable( szToken );
}
// add any additional resources
// duplicates don't need to be culled, loaders are supposed to handle resources that already exist
for ( int i = 0; i < m_AdditionalResources.GetNumStrings(); i++ )
{
if ( g_QueuedLoader.GetSpewDetail() )
{
Msg( "QueuedLoader: Appending: %s\n", m_AdditionalResources.String( i ) );
}
AddResourceToTable( m_AdditionalResources.String( i ) );
}
if ( g_QueuedLoader.GetSpewDetail() )
{
for ( int i = RESOURCEPRELOAD_UNKNOWN+1; i < RESOURCEPRELOAD_COUNT; i++ )
{
Msg( "QueuedLoader: %s: %d Entries\n", g_ResourceLoaderNames[i], m_ResourceNames[i].Count() );
}
}
m_pProgress->UpdateProgress( PROGRESS_PARSEDRESLIST );
}
//-----------------------------------------------------------------------------
// Mark the start of the queued loading process.
//-----------------------------------------------------------------------------
bool CQueuedLoader::BeginMapLoading( const char *pMapName, bool bLoadForHDR, bool bOptimizeMapReload )
{
if ( IsPC() )
{
return false;
}
if ( CommandLine()->FindParm( "-noqueuedload" ) || ( g_pFullFileSystem->GetDVDMode() != DVDMODE_STRICT ) )
{
return false;
}
if ( m_bStarted )
{
// already started, shouldn't be started more than once
Assert( 0 );
return true;
}
COM_TimestampedLog( "CQueuedLoader::BeginMapLoading" );
// set the IO throttle markers based on available memory
// these safety watermarks throttle the i/o from flooding memory, when the cores cannot keep up
// the delta must be larger than any single operation, otherwise deadlock
// markers that are too close will cause excessive suspension
size_t usedMemory, freeMemory;
MemAlloc_GlobalMemoryStatus( &usedMemory, &freeMemory );
if ( freeMemory >= 64*1024*1024 )
{
// lots of available memory, can afford to have let the i/o get ahead
g_nHighIOSuspensionMark = 10*1024*1024;
g_nLowIOSuspensionMark = 2*1024*1024;
}
else
{
// low memory, suspend the i/o more frequently
g_nHighIOSuspensionMark = 5*1024*1024;
g_nLowIOSuspensionMark = 1*1024*1024;
}
if ( GetSpewDetail() )
{
Msg( "QueuedLoader: Suspend I/O at [%.2f,%.2f] MB\n", (float)g_nLowIOSuspensionMark/(1024.0f*1024.0f), (float)g_nHighIOSuspensionMark/(1024.0f*1024.0f) );
}
m_bStarted = true;
m_bDynamic = false;
m_bLoadForHDR = bLoadForHDR;
// map pak will be accessed asynchronously throughout loading and into game frame
g_pFullFileSystem->BeginMapAccess();
// remove any prior stale entries
CleanQueue();
Assert( m_SubmittedJobs.Count() == 0 && g_nActiveJobs == 0 && g_nQueuedJobs == 0 );
m_bActive = true;
m_nSubmitCount = 0;
m_StartTime = Plat_MSTime();
m_EndTime = 0;
m_bCanBatch = false;
m_bBatching = false;
m_bDoProgress = false;
g_nIOMemory = 0;
g_nAnonymousIOMemory = 0;
g_nIOMemoryPeak = 0;
g_nAnonymousIOMemoryPeak = 0;
m_bSameMap = bOptimizeMapReload && ( V_stricmp( pMapName, m_szMapNameToCompareSame ) == 0 );
if ( m_bSameMap )
{
// Data will persist (so reloading a map is v. fast)
}
else
{
// Full load of the new map's data
V_strncpy( m_szMapNameToCompareSame, pMapName, sizeof( m_szMapNameToCompareSame ) );
}
m_pProgress->BeginProgress();
m_pProgress->UpdateProgress( PROGRESS_START );
// load this map's resource list before any other i/o
char szBaseName[MAX_PATH];
char szFilename[MAX_PATH];
V_FileBase( pMapName, szBaseName, sizeof( szBaseName ) );
V_snprintf( szFilename, sizeof( szFilename ), "reslists_xbox/%s%s.lst", szBaseName, GetPlatformExt() );
MEM_ALLOC_CREDIT();
CUtlBuffer resListBuffer( 0, 0, CUtlBuffer::TEXT_BUFFER );
if ( !g_pFullFileSystem->ReadFile( szFilename, "GAME", resListBuffer, 0, 0 ) )
{
// very bad, a valid reslist is critical
DevWarning( "QueuedLoader: Failed to get reslist '%s', Non-Optimal Loading.\n", szFilename );
m_bActive = false;
return false;
}
if ( XBX_IsLocalized() )
{
// find optional localized reslist fixup
V_snprintf( szFilename, sizeof( szFilename ), "reslists_xbox/%s%s.lst", XBX_GetLanguageString(), GetPlatformExt() );
CUtlBuffer localizedBuffer( 0, 0, CUtlBuffer::TEXT_BUFFER );
if ( g_pFullFileSystem->ReadFile( szFilename, "GAME", localizedBuffer, 0, 0 ) )
{
// append it
resListBuffer.EnsureCapacity( resListBuffer.TellPut() + localizedBuffer.TellPut() );
resListBuffer.Put( localizedBuffer.PeekGet(), localizedBuffer.TellPut() );
}
}
m_pProgress->UpdateProgress( PROGRESS_GOTRESLIST );
// due to its size, the bsp load is a lengthy i/o operation
// this causes a non-batched async i/o operation to commence immediately
if ( !m_pLoaders[RESOURCEPRELOAD_MODEL]->CreateResource( pMapName ) )
{
// very bad, a valid bsp is critical
DevWarning( "QueuedLoader: Failed to mount BSP '%s', Non-Optimal Loading.\n", pMapName );
m_bActive = false;
return false;
}
// parse the raw resource list into loader specific dictionaries
ParseResourceList( resListBuffer );
// run the distributed precache loaders, generating a batch of i/o requests
GetJobRequests();
// event each loader to discard dead resources
PurgeUnreferencedResources();
// sort and start async fulfilling the i/o requests
// waits for all "must complete" jobs to finish
SubmitBatchedJobsAndWait();
// progress is only relevant during preload
// normal load process takes over any progress bar
// disable progress tracking to prevent any late queued operation from updating
m_pProgress->EndProgress();
return m_bActive;
}
//-----------------------------------------------------------------------------
// Signal the end of the queued loading process, i/o will still be in progress.
//-----------------------------------------------------------------------------
void CQueuedLoader::EndMapLoading( bool bAbort )
{
if ( !m_bStarted )
{
// already stopped or never started
return;
}
/////////////////////////////////////////////////////
// TBD: Cannot abort!!!! feature has not been done //
/////////////////////////////////////////////////////
bAbort = false;
if ( m_bActive )
{
if ( bAbort )
{
PurgeQueue();
}
else
{
// finish all outstanding priority jobs
SubmitPendingJobs();
while ( g_nHighPriorityJobs != 0 || g_nJobsToFinishBeforePlay != 0 )
{
// yield some time
g_pThreadPool->Yield( MAIN_THREAD_YIELD_TIME );
}
}
m_EndTime = Plat_MSTime();
m_bActive = false;
// transmit the end map event
for ( int i = RESOURCEPRELOAD_UNKNOWN+1; i < RESOURCEPRELOAD_COUNT; i++ )
{
if ( m_pLoaders[i] )
{
m_pLoaders[i]->OnEndMapLoading( bAbort );
}
}
// free any unclaimed anonymous buffers
int iIndex = m_AnonymousJobs.First();
while ( iIndex != m_AnonymousJobs.InvalidIndex() )
{
FileJob_t *pFileJob = m_AnonymousJobs[iIndex];
if ( pFileJob->m_bFreeTargetAfterIO && pFileJob->m_pTargetData )
{
g_pFullFileSystem->FreeOptimalReadBuffer( pFileJob->m_pTargetData );
pFileJob->m_pTargetData = NULL;
}
g_nAnonymousIOMemory -= pFileJob->m_nActualBytesRead;
iIndex = m_AnonymousJobs.Next( iIndex );
}
m_AnonymousJobs.Purge();
if ( g_nIOMemory || g_nAnonymousIOMemory )
{
// expected to be zero, otherwise logic flaw
DevWarning( "CQueuedLoader: Unclaimed I/O memory: total:%d anonymous:%d\n", (int)g_nIOMemory, (int)g_nAnonymousIOMemory );
g_nIOMemory = 0;
g_nAnonymousIOMemory = 0;
}
// no longer needed
m_AdditionalResources.RemoveAll();
}
g_pFullFileSystem->EndMapAccess();
m_bStarted = false;
}
//-----------------------------------------------------------------------------
// Returns true if loader is accepting queue requests.
//-----------------------------------------------------------------------------
bool CQueuedLoader::IsMapLoading() const
{
return m_bActive;
}
//-----------------------------------------------------------------------------
// Returns true if loader is working on same map as last load
//-----------------------------------------------------------------------------
bool CQueuedLoader::IsSameMapLoading() const
{
return m_bActive && m_bSameMap;
}
//-----------------------------------------------------------------------------
// Returns true if the loader is idle, indicates all i/o and work has completed.
//-----------------------------------------------------------------------------
bool CQueuedLoader::IsFinished() const
{
return ( m_bActive == false && g_nActiveJobs == 0 && g_nQueuedJobs == 0 );
}
//-----------------------------------------------------------------------------
// Returns true if loader is batching
//-----------------------------------------------------------------------------
bool CQueuedLoader::IsBatching() const
{
return m_bBatching;
}
//-----------------------------------------------------------------------------
// Returns true if loader is batching
//-----------------------------------------------------------------------------
bool CQueuedLoader::IsDynamic() const
{
return m_bDynamic;
}
int CQueuedLoader::GetSpewDetail() const
{
int spewDetail = loader_spew_info.GetInt();
if ( spewDetail <= 0 )
{
return spewDetail;
}
return 1 << ( spewDetail - 1 );
}
void CQueuedLoader::DynamicLoadMapResource( const char *pFilename, DynamicResourceCallback_t pCallback, void *pContext, void *pContext2 )
{
Assert( m_bActive == false );
m_bActive = true;
m_bDynamic = true;
m_DynamicFileName = pFilename;
m_pfnDynamicCallback = pCallback;
m_pDynamicContext = pContext;
m_pDynamicContext2 = pContext2;
CleanQueue();
AddResourceToTable( m_DynamicFileName );
// run the distributed precache loaders, generating a batch of i/o requests
GetJobRequests();
// sort and start async fulfilling the i/o requests
Assert( m_bBatching && g_nActiveJobs == 0 );
SubmitBatchedJobs();
Assert( !m_bBatching );
}
void CQueuedLoader::QueueDynamicLoadFunctor( CFunctor* pFunctor )
{
AUTO_LOCK( m_FunctorQueueMutex );
m_FunctorQueue.AddToTail( pFunctor );
}
bool CQueuedLoader::CompleteDynamicLoad()
{
Assert( m_bActive && m_bDynamic && !m_bBatching );
bool bDone = true;
if ( m_bDynamic )
{
CUtlVector< CFunctor* > functors;
{
AUTO_LOCK( m_FunctorQueueMutex );
functors.Swap( m_FunctorQueue );
}
FOR_EACH_VEC( functors, i )
{
( *functors[i] )();
functors[i]->Release();
}
{
AUTO_LOCK( m_FunctorQueueMutex );
bDone = m_FunctorQueue.Count() == 0 && g_nQueuedJobs == 0 && g_nActiveJobs == 0;
}
if ( bDone )
{
if ( m_pfnDynamicCallback )
{
( *m_pfnDynamicCallback )( m_DynamicFileName, m_pDynamicContext, m_pDynamicContext2 );
}
m_DynamicFileName.Clear();
m_bActive = false;
m_bDynamic = false;
}
}
return bDone;
}
void CQueuedLoader::QueueCleanupDynamicLoadFunctor( CFunctor* pFunctor )
{
Assert( ThreadInMainThread() );
m_CleanupFunctorQueue.AddToTail( pFunctor );
}
bool CQueuedLoader::CleanupDynamicLoad()
{
Assert( ThreadInMainThread() );
FOR_EACH_VEC( m_CleanupFunctorQueue, i )
{
( *m_CleanupFunctorQueue[i] )();
m_CleanupFunctorQueue[i]->Release();
}
m_CleanupFunctorQueue.Purge();
return true;
}