1HEADmaster

1 files changed, 141 insertions, 0 deletions

diff --git a/utils/shadercompile_launcher/shadercompile_launcher.cpp b/utils/shadercompile_launcher/shadercompile_launcher.cpp
new file mode 100644
index 0000000..7686c01
--- /dev/null
+++ b/utils/shadercompile_launcher/shadercompile_launcher.cpp
@@ -0,0 +1,141 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+// $NoKeywords: $
+//
+//=============================================================================//
+// vrad_launcher.cpp : Defines the entry point for the console application.
+//
+
+#include <windows.h>
+#include "interface.h"
+#include <direct.h>
+#include "tier1/strtools.h"
+#include "tier0/icommandline.h"
+#include "ishadercompiledll.h"
+#include <stdio.h>
+#include <conio.h>
+
+char* GetLastErrorString()
+{
+	static char err[2048];
+	
+	LPVOID lpMsgBuf;
+	FormatMessage( 
+		FORMAT_MESSAGE_ALLOCATE_BUFFER | 
+		FORMAT_MESSAGE_FROM_SYSTEM | 
+		FORMAT_MESSAGE_IGNORE_INSERTS,
+		NULL,
+		GetLastError(),
+		MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
+		(LPTSTR) &lpMsgBuf,
+		0,
+		NULL 
+	);
+
+	strncpy( err, (char*)lpMsgBuf, sizeof( err ) );
+	LocalFree( lpMsgBuf );
+
+	err[ sizeof( err ) - 1 ] = 0;
+
+	return err;
+}
+
+
+void MakeFullPath( const char *pIn, char *pOut, int outLen )
+{
+	if ( pIn[0] == '/' || pIn[0] == '\\' || pIn[1] == ':' )
+	{
+		// It's already a full path.
+		Q_strncpy( pOut, pIn, outLen );
+	}
+	else
+	{
+		_getcwd( pOut, outLen );
+		Q_strncat( pOut, "\\", outLen, COPY_ALL_CHARACTERS );
+		Q_strncat( pOut, pIn, outLen, COPY_ALL_CHARACTERS );
+	}
+}
+
+void Pause( void )
+{
+//	printf( "Pause\n" );
+//	fflush( stdout );
+//	getch();
+}
+
+int main(int argc, char* argv[])
+{
+	char dllName[512];
+
+	Pause();	
+	CommandLine()->CreateCmdLine( argc, argv );
+
+	char fullPath[512], redirectFilename[512];
+	MakeFullPath( argv[0], fullPath, sizeof( fullPath ) );
+	Q_StripFilename( fullPath );
+	Q_snprintf( redirectFilename, sizeof( redirectFilename ), "%s\\%s", fullPath, "vrad.redirect" );
+
+	Pause();	
+	// First, look for vrad.redirect and load the dll specified in there if possible.
+	CSysModule *pModule = NULL;
+	FILE *fp = fopen( redirectFilename, "rt" );
+	if ( fp )
+	{
+		if ( fgets( dllName, sizeof( dllName ), fp ) )
+		{
+			char *pEnd = strstr( dllName, "\n" );
+			if ( pEnd )
+				*pEnd = 0;
+
+			pModule = Sys_LoadModule( dllName );
+			if ( pModule )
+				printf( "Loaded alternate VRAD DLL (%s) specified in vrad.redirect.\n", dllName );
+			else
+				printf( "Can't find '%s' specified in vrad.redirect.\n", dllName );
+		}
+		
+		fclose( fp );
+	}
+	Pause();	
+
+	// If it didn't load the module above, then use the 
+	if ( !pModule )
+	{
+		strcpy( dllName, "shadercompile_dll.dll" );
+		pModule = Sys_LoadModule( dllName );
+	}
+
+	Pause();	
+	if( !pModule )
+	{
+		printf( "vrad_launcher error: can't load %s\n%s", dllName, GetLastErrorString() );
+		Pause();	
+		return 1;
+	}
+
+	Pause();	
+	CreateInterfaceFn fn = Sys_GetFactory( pModule );
+	if( !fn )
+	{
+		printf( "vrad_launcher error: can't get factory from vrad_dll.dll\n" );
+		Sys_UnloadModule( pModule );
+		return 2;
+	}
+
+	int retCode = 0;
+	IShaderCompileDLL *pDLL = (IShaderCompileDLL*)fn( SHADER_COMPILE_INTERFACE_VERSION, &retCode );
+	if( !pDLL )
+	{
+		printf( "vrad_launcher error: can't get IVRadDLL interface from vrad_dll.dll\n" );
+		Sys_UnloadModule( pModule );
+		return 3;
+	}
+
+	int returnValue = pDLL->main( argc, argv );
+	Sys_UnloadModule( pModule );
+	
+	return returnValue;
+}
+