diff options
Diffstat (limited to 'mp/src/game/client/bsp_utils.cpp')
| -rw-r--r-- | mp/src/game/client/bsp_utils.cpp | 170 |
1 files changed, 170 insertions, 0 deletions
diff --git a/mp/src/game/client/bsp_utils.cpp b/mp/src/game/client/bsp_utils.cpp new file mode 100644 index 00000000..c21c2ada --- /dev/null +++ b/mp/src/game/client/bsp_utils.cpp @@ -0,0 +1,170 @@ +//========= Copyright Valve Corporation, All rights reserved. ============// +// +// Purpose: Exposes bsp tools to game for e.g. workshop use +// +// $NoKeywords: $ +//===========================================================================// + +#include "cbase.h" +#include <tier2/tier2.h> +#include "filesystem.h" +#include "bsp_utils.h" +#include "utlbuffer.h" +#include "igamesystem.h" + +// memdbgon must be the last include file in a .cpp file!!! +#include "tier0/memdbgon.h" + +bool BSP_SyncRepack( const char *pszInputMapFile, + const char *pszOutputMapFile, + IBSPPack::eRepackBSPFlags eRepackFlags ) +{ + // load the bsppack dll + IBSPPack *libBSPPack = NULL; + CSysModule *pModule = g_pFullFileSystem->LoadModule( "bsppack" ); + if ( pModule ) + { + CreateInterfaceFn BSPPackFactory = Sys_GetFactory( pModule ); + if ( BSPPackFactory ) + { + libBSPPack = ( IBSPPack * )BSPPackFactory( IBSPPACK_VERSION_STRING, NULL ); + } + } + if( !libBSPPack ) + { + Warning( "Can't load bsppack library - unable to compress bsp\n" ); + return false; + } + + Msg( "Repacking %s -> %s\n", pszInputMapFile, pszOutputMapFile ); + + if ( !g_pFullFileSystem->FileExists( pszInputMapFile ) ) + { + Warning( "Couldn't open input file %s - BSP recompress failed\n", pszInputMapFile ); + return false; + } + + CUtlBuffer inputBuffer; + if ( !g_pFullFileSystem->ReadFile( pszInputMapFile, NULL, inputBuffer ) ) + { + Warning( "Couldn't read file %s - BSP compression failed\n", pszInputMapFile ); + return false; + } + + CUtlBuffer outputBuffer; + + if ( !libBSPPack->RepackBSP( inputBuffer, outputBuffer, eRepackFlags ) ) + { + Warning( "Internal error compressing BSP\n" ); + return false; + } + + g_pFullFileSystem->WriteFile( pszOutputMapFile, NULL, outputBuffer ); + + Msg( "Successfully repacked %s as %s -- %u -> %u bytes\n", + pszInputMapFile, pszOutputMapFile, inputBuffer.TellPut(), outputBuffer.TellPut() ); + + return true; +} + +// Helper to create a thread that calls SyncCompressMap, and clean it up when it exists +void BSP_BackgroundRepack( const char *pszInputMapFile, + const char *pszOutputMapFile, + IBSPPack::eRepackBSPFlags eRepackFlags ) +{ + // Make this a gamesystem and thread, so it can check for completion each frame and clean itself up. Run() is the + // background thread, Update() is the main thread tick. + class BackgroundBSPRepackThread : public CThread, public CAutoGameSystemPerFrame + { + public: + BackgroundBSPRepackThread( const char *pszInputFile, const char *pszOutputFile, IBSPPack::eRepackBSPFlags eRepackFlags ) + : m_strInput( pszInputFile ) + , m_strOutput( pszOutputFile ) + , m_eRepackFlags( eRepackFlags ) + { + Start(); + } + + // CThread job - returns 0 for success + virtual int Run() OVERRIDE + { + return BSP_SyncRepack( m_strInput.Get(), m_strOutput.Get(), m_eRepackFlags ) ? 0 : 1; + } + + // GameSystem + virtual const char* Name( void ) OVERRIDE { return "BackgroundBSPRepackThread"; } + + // Runs on main thread + void CheckFinished() + { + if ( !IsAlive() ) + { + // Thread finished + if ( GetResult() != 0 ) + { + Warning( "Map compression thread failed :(\n" ); + } + + // AutoGameSystem deregisters itself on destruction, we're done + delete this; + } + } + + #ifdef CLIENT_DLL + virtual void Update( float frametime ) OVERRIDE { CheckFinished(); } + #else // GAME DLL + virtual void FrameUpdatePostEntityThink() OVERRIDE { CheckFinished(); } + #endif + private: + CUtlString m_strInput; + CUtlString m_strOutput; + IBSPPack::eRepackBSPFlags m_eRepackFlags; + }; + + Msg( "Starting BSP repack job %s -> %s\n", pszInputMapFile, pszOutputMapFile ); + + // Deletes itself up when done + new BackgroundBSPRepackThread( pszInputMapFile, pszOutputMapFile, eRepackFlags ); +} + +CON_COMMAND( bsp_repack, "Repack and output a (re)compressed version of a bsp file" ) +{ +#ifdef GAME_DLL + if ( !UTIL_IsCommandIssuedByServerAdmin() ) + return; +#endif + + // Handle -nocompress + bool bCompress = true; + const char *szInFilename = NULL; + const char *szOutFilename = NULL; + + if ( args.ArgC() == 4 && V_strcasecmp( args.Arg( 1 ), "-nocompress" ) == 0 ) + { + bCompress = false; + szInFilename = args.Arg( 2 ); + szOutFilename = args.Arg( 3 ); + } + else if ( args.ArgC() == 3 ) + { + szInFilename = args.Arg( 1 ); + szOutFilename = args.Arg( 2 ); + } + + if ( !szInFilename || !szOutFilename || !strlen( szInFilename ) || !strlen( szOutFilename ) ) + { + Msg( "Usage: bsp_repack [-nocompress] map.bsp output_map.bsp\n" ); + return; + } + + if ( bCompress ) + { + // Use default compress flags + BSP_BackgroundRepack( szInFilename, szOutFilename ); + } + else + { + // No compression + BSP_BackgroundRepack( szInFilename, szOutFilename, (IBSPPack::eRepackBSPFlags)0 ); + } +} |