//========= Copyright Valve Corporation, All rights reserved. ============//
//
//=============================================================================
// Standard includes
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <io.h>
// Valve includes
#include "itemtest/itemtest.h"
#include "bitmap/bitmap.h"
#include "bitmap/imageformat.h"
#include "bitmap/psd.h"
#include "bitmap/tgaloader.h"
#include "bitmap/tgawriter.h"
#include "vtf/vtf.h"
#include "datamodel/dmattribute.h"
#include "datamodel/dmelement.h"
#include "datamodel/idatamodel.h"
#include "fbxutils/dmfbxserializer.h"
#include "filesystem.h"
#include "movieobjects/dmefaceset.h"
#include "movieobjects/dmematerial.h"
#include "movieobjects/dmemesh.h"
#include "movieobjects/dmemodel.h"
#include "movieobjects/dmobjserializer.h"
#include "movieobjects/dmsmdserializer.h"
#include "movieobjects/dmeanimationlist.h"
#include "movieobjects/dmeclip.h"
#include "movieobjects/dmechannel.h"
#include "movieobjects/dmelog.h"
#include "steam/steam_api.h"
#include "tier1/fmtstr.h"
#include "tier1/utlsymbol.h"
#include "tier2/fileutils.h"
#include "tier2/p4helpers.h"
#include "../public/zip_utils.h"
// Last include
#include "tier0/memdbgon.h"
#ifdef BEGIN_DEFINE_LOGGING_CHANNEL
BEGIN_DEFINE_LOGGING_CHANNEL( LOG_ITEMTEST, "ItemTest", LCF_CONSOLE_ONLY, LS_MESSAGE );
ADD_LOGGING_CHANNEL_TAG( "ItemTest" );
END_DEFINE_LOGGING_CHANNEL();
#endif
// This isn't available in the TF runtime (yet?)
#ifndef FUNCTION_LINE_STRING
#define FUNCTION_LINE_STRINGIFY(x) #x
#define FUNCTION_LINE_TOSTRING(x) FUNCTION_LINE_STRINGIFY(x)
#define FUNCTION_LINE_STRING __FUNCTION__ "(" FUNCTION_LINE_TOSTRING(__LINE__) "): "
#endif
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
enum
{
k64KB = 65536
};
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
static CSteamAPIContext g_SteamAPIContext;
bool CItemUpload::m_bDev = false;
bool CItemUpload::m_bIgnoreEnvVars = false;
bool CItemUpload::m_bP4 = false;
CUtlString CItemUpload::m_szForcedSteamID = "";
CItemTestManifest *CItemUpload::m_pItemTestManifest = NULL;
static bool g_bCompilePreview = false;
static const char* kVMT = "VMT%d";
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
inline bool UtlStringLessThan( const CUtlString &sLhs, const CUtlString &sRhs )
{
return CaselessStringLessThanIgnoreSlashes( sLhs.String(), sRhs.String() );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CItemLog::Msg( const char *pszFormat, ... ) const
{
va_list args;
va_start( args, pszFormat );
CFmtStrMax str;
str.AppendFormatV( pszFormat, args );
Log( kItemtest_Log_Info, str );
va_end( args );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CItemLog::Warning( const char *pszFormat, ... ) const
{
va_list args;
va_start( args, pszFormat );
CFmtStrMax str;
str.AppendFormatV( pszFormat, args );
Log( kItemtest_Log_Warning, str );
va_end( args );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CItemLog::Error( const char *pszFormat, ... ) const
{
va_list args;
va_start( args, pszFormat );
CFmtStrMax str;
str.AppendFormatV( pszFormat, args );
Log( kItemtest_Log_Error, str );
va_end( args );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CItemLog::Log( ItemtestLogLevel_t nLogLevel, const char *pszMessage ) const
{
if ( m_pItemLog && m_pItemLog != this )
{
m_pItemLog->Log( nLogLevel, pszMessage );
return;
}
switch ( nLogLevel )
{
case kItemtest_Log_Info:
Log_Msg( LOG_ITEMTEST, "%s", pszMessage );
break;
case kItemtest_Log_Warning:
Log_Warning( LOG_ITEMTEST, "%s", pszMessage );
break;
case kItemtest_Log_Error:
Log_Error( LOG_ITEMTEST, "%s", pszMessage );
break;
}
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CItemUpload::InitManifest( void )
{
if ( m_pItemTestManifest )
return true;
m_pItemTestManifest = new CItemTestManifest( "scripts/itemtest_manifest.txt", new CItemLog() );
return m_pItemTestManifest->IsValid();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CItemUpload::SanitizeName( const char *pszName, CUtlString &sCleanName )
{
char pszTemp[MAX_PATH];
V_strcpy_safe( pszTemp, pszName );
// Convert to lowercase, strip punctuation and turn spaces into underscores
char *pszSrc = pszTemp;
char *pszDst = pszTemp;
while ( *pszSrc )
{
char c = *pszSrc++;
if ( c >= 'a' && c <= 'z' )
{
*pszDst++ = c;
}
else if ( c >= 'A' && c <= 'Z' )
{
*pszDst++ = c - 'A' + 'a';
}
else if ( c >= '0' && c <= '9' )
{
*pszDst++ = c;
}
else if ( V_isspace(c) || c == '_' )
{
*pszDst++ = '_';
}
else
{
// Punctuation or non-ASCII characters, skip 'em!
}
}
*pszDst = '\0';
sCleanName = pszTemp;
return !sCleanName.IsEmpty();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CItemTestManifest::CItemTestManifest( const char *pszManifestFile, CItemLog *pItemLog ) :
m_pItemLog( pItemLog ),
m_vecVMTTextureRemaps( UtlStringLessThan )
{
m_pManifestKV = NULL;
m_pItemDirectory = "";
m_pAnimationDirectory = "";
m_pIconDirectory = "";
m_pZipSourceDirectory = "";
m_pZipOutputDirectory = "";
m_pQCTemplate = "";
m_pQCITemplate = "";
m_bTerseMessages = false;
m_bItemPathUsesSteamId = true;
m_pManifestKV = new KeyValues( pszManifestFile );
if ( !m_pManifestKV->LoadFromFile(g_pFullFileSystem, pszManifestFile, "MOD") )
{
m_pManifestKV->deleteThis();
m_pManifestKV = NULL;
m_pItemLog->Warning( "ERROR: Failed to load manifest file: %s\n", pszManifestFile );
return;
}
// Class list
if ( !ParseStringsFromManifest( m_pManifestKV, "classes", m_vecClasses ) )
return;
// MDL Extensions
if ( !ParseStringsFromManifest( m_pManifestKV, "mdl_files", m_vecMDLExtensions ) )
return;
// Animation MDL Extensions
if ( !ParseStringsFromManifest( m_pManifestKV, "animation_mdl_files", m_vecAnimationMDLExtensions ) )
return;
// Material types
KeyValues *pKVMaterialTypes = m_pManifestKV->FindKey("material_types");
if ( !pKVMaterialTypes )
{
m_pItemLog->Warning( "ERROR: Failed to find a 'material_types' section in manifest file: %s\n", pszManifestFile );
return;
}
FOR_EACH_SUBKEY( pKVMaterialTypes, pKVMaterialType )
{
const char *pszString = pKVMaterialType->GetName();
int nIdx = m_vecMaterialTypes.AddToTail();
m_vecMaterialTypes[nIdx].pszMaterialType = pszString;
}
const char *pszDefaultMatTypeString = m_pManifestKV->GetString("default_material_type");
if ( !pszDefaultMatTypeString || !pszDefaultMatTypeString[0] )
{
m_pItemLog->Warning( "ERROR: Failed to find a 'default_material_type' string in manifest file: %s\n", pszManifestFile );
return;
}
m_nDefaultMaterialType = GetMaterialType( pszDefaultMatTypeString );
if ( m_nDefaultMaterialType == kInvalidMaterialType )
{
m_pItemLog->Warning( "ERROR: Default material type '%s' wasn't found in the material type list in manifest file: %s\n", m_nDefaultMaterialType, pszManifestFile );
return;
}
// Material skins
KeyValues *pKVMaterialSkins = m_pManifestKV->FindKey("material_skins");
if ( pKVMaterialSkins )
{
FOR_EACH_SUBKEY( pKVMaterialSkins, pKVMaterialSkin )
{
const char *pszString = pKVMaterialSkin->GetName();
int nIdx = m_vecMaterialSkins.AddToTail();
m_vecMaterialSkins[nIdx].pszMaterialSkin = pszString;
m_vecMaterialSkins[nIdx].pszFilenameAppend = pKVMaterialSkin->GetString("file_append");
}
const char *pszDefaultMatSkinString = m_pManifestKV->GetString("default_material_skin");
if ( !pszDefaultMatSkinString || !pszDefaultMatSkinString[0] )
{
m_pItemLog->Warning( "ERROR: Failed to find a 'default_material_skin' string in manifest file: %s\n", pszManifestFile );
return;
}
m_nDefaultMaterialSkin = GetMaterialSkin( pszDefaultMatSkinString );
if ( m_nDefaultMaterialSkin == kInvalidMaterialSkin )
{
m_pItemLog->Warning( "ERROR: Default material skin '%s' wasn't found in the material skin list in manifest file: %s\n", pszDefaultMatSkinString, pszManifestFile );
m_nDefaultMaterialSkin = 0;
return;
}
}
else
{
m_nDefaultMaterialSkin = 0;
}
// Texture types
KeyValues *pKVTextureTypes = m_pManifestKV->FindKey("texture_types");
if ( !pKVTextureTypes )
{
m_pItemLog->Warning( "ERROR: Failed to find a 'texture_types' section in manifest file: %s\n", pszManifestFile );
return;
}
FOR_EACH_SUBKEY( pKVTextureTypes, pKVTexture )
{
const char *pszString = pKVTexture->GetName();
int nIdx = m_vecTextureTypes.AddToTail();
m_vecTextureTypes[nIdx].pszTextureType = pszString;
m_vecTextureTypes[nIdx].bOptional = pKVTexture->GetBool( "optional" );
m_vecTextureTypes[nIdx].pkvAddToVTEXConfig = pKVTexture->FindKey("add_to_vtex_config");
}
// VMT templates
KeyValues *pKVTemplates = m_pManifestKV->FindKey("vmt_templates");
if ( !pKVTemplates )
{
m_pItemLog->Warning( "ERROR: Failed to find a 'vmt_templates' section in manifest file: %s\n", pszManifestFile );
return;
}
KeyValues *pKVClassTemplates = pKVTemplates->FindKey("classes");
if ( pKVClassTemplates )
{
m_vecClassTemplates.SetCount( m_vecClasses.Count() );
for ( int i = 0; i < m_vecClassTemplates.Count(); ++i )
{
m_vecClassTemplates[ i ] = NULL;
}
FOR_EACH_SUBKEY( pKVClassTemplates, pKVClassTemplate )
{
const char *pszHero = pKVClassTemplate->GetName();
const char *pszTemplate = pKVClassTemplate->GetString();
int iClass = m_vecClasses.Find( pszHero );
if ( iClass == m_vecClasses.InvalidIndex() )
{
m_pItemLog->Warning( "ERROR: Found an invalid class '%s' in the vmt_templates entries in manifest file: %s\n", pszHero, pszManifestFile );
return;
}
m_vecClassTemplates[iClass] = pszTemplate;
}
}
KeyValues *pKVVMTRemaps = pKVTemplates->FindKey("vmt_texture_settings");
if ( pKVVMTRemaps )
{
FOR_EACH_SUBKEY( pKVVMTRemaps, pKVRemap )
{
const char *pszVMTVar = pKVRemap->GetName();
const char *pszTexture = pKVRemap->GetString();
m_vecVMTTextureRemaps.Insert( pszTexture, pszVMTVar );
}
}
// Icon types
KeyValues *pKVIconTypes = m_pManifestKV->FindKey("icon_types");
if ( pKVIconTypes )
{
FOR_EACH_SUBKEY( pKVIconTypes, pKVIcon )
{
const char *pszString = pKVIcon->GetName();
int nIdx = m_vecIconTypes.AddToTail();
m_vecIconTypes[nIdx].pszIconType = pszString;
m_vecIconTypes[nIdx].nWidth = pKVIcon->GetInt( "width" );
m_vecIconTypes[nIdx].nHeight = pKVIcon->GetInt( "height" );
m_vecIconTypes[nIdx].pszFilenameAppend = pKVIcon->GetString("file_append");
m_vecIconTypes[nIdx].pkvAddToVTEXConfig = pKVIcon->FindKey("add_to_vtex_config");
KeyValues *pKVVMT = pKVIcon->FindKey("vmt_template");
if ( pKVVMT )
{
m_vecIconTypes[nIdx].pkvVMTTemplate = pKVVMT->GetFirstSubKey();
}
else
{
m_vecIconTypes[nIdx].pkvVMTTemplate = NULL;
}
}
}
m_pItemDirectory = m_pManifestKV->GetString("item_directory");
m_pAnimationDirectory = m_pManifestKV->GetString("animation_directory");
m_pIconDirectory = m_pManifestKV->GetString("icon_directory");
m_pZipSourceDirectory = m_pManifestKV->GetString("archive_source_path");
m_pZipOutputDirectory = m_pManifestKV->GetString("archive_output_path");
m_pQCTemplate = m_pManifestKV->GetString("qc_template");
if ( V_strlen( m_pQCTemplate ) == 0 )
{
m_pItemLog->Warning( "ERROR: qc_template not defined in manifest file: %s\n", pszManifestFile );
return;
}
m_pQCITemplate = m_pManifestKV->GetString( "qci_template" );
if ( V_strlen( m_pQCITemplate ) == 0 )
{
m_pItemLog->Warning( "ERROR: qci_template not defined in manifest file: %s\n", pszManifestFile );
return;
}
const char *pszQCLODDistances = m_pManifestKV->GetString("qc_lod_distances");
CUtlVector<char*> vecQCLODDistances;
V_SplitString( pszQCLODDistances, ",", vecQCLODDistances );
m_vecQCLODDistances.SetCount( vecQCLODDistances.Count() );
for ( int i = 0; i < vecQCLODDistances.Count(); ++i )
{
m_vecQCLODDistances[i] = V_atoi( vecQCLODDistances[i] );
}
m_bTerseMessages = m_pManifestKV->GetBool( "terse_messages", false );
m_bItemPathUsesSteamId = m_pManifestKV->GetBool( "item_path_has_steamid", true );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CItemTestManifest::ParseStringsFromManifest( KeyValues *pKV, const char *pszKeyName, CUtlVector< CUtlString > &vecList )
{
KeyValues *pKVSub = pKV->FindKey(pszKeyName);
if ( !pKVSub )
{
m_pItemLog->Warning( "ERROR: Failed to find a '%s' section in manifest file.\n", pszKeyName );
return false;
}
FOR_EACH_SUBKEY( pKVSub, pKVSubKey )
{
const char *pszString = pKVSubKey->GetName();
vecList.AddToTail( pszString );
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CItemTestManifest::GetMaterialType( const char *pszMaterialType )
{
FOR_EACH_VEC( m_vecMaterialTypes, i )
{
if ( V_stricmp(m_vecMaterialTypes[i].pszMaterialType, pszMaterialType) == 0 )
return i;
}
return kInvalidMaterialType;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CItemTestManifest::GetMaterialSkin( const char *pszMaterialSkin )
{
FOR_EACH_VEC( m_vecMaterialSkins, i )
{
if ( V_stricmp(m_vecMaterialSkins[i].pszMaterialSkin, pszMaterialSkin) == 0 )
return i;
}
return kInvalidMaterialSkin;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CItemTestManifest::GetTextureType( const char *pszTextureType )
{
FOR_EACH_VEC( m_vecTextureTypes, i )
{
if ( V_stricmp(m_vecTextureTypes[i].pszTextureType, pszTextureType) == 0 )
return i;
}
return -1;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
KeyValues *CItemTestManifest::GetTextureAddToVTEXConfig( const char *pszTextureType )
{
int nIdx = GetTextureType(pszTextureType);
if ( nIdx == -1 )
return NULL;
return m_vecTextureTypes[nIdx].pkvAddToVTEXConfig;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CItemTestManifest::GetIconType( const char *pszIconType )
{
FOR_EACH_VEC( m_vecIconTypes, i )
{
if ( V_stricmp(m_vecIconTypes[i].pszIconType, pszIconType) == 0 )
return i;
}
return -1;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CItemTestManifest::GetIconDimensions( int nIcon, int &nWidth, int &nHeight )
{
if ( nIcon >= 0 && nIcon < m_vecIconTypes.Count() )
{
nWidth = m_vecIconTypes[nIcon].nWidth;
nHeight = m_vecIconTypes[nIcon].nHeight;
return true;
}
else
{
nWidth = 0;
nHeight = 0;
return false;
}
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const char *CItemTestManifest::GetVMTVarForTextureType( const char *pszTexture )
{
int nIndex = m_vecVMTTextureRemaps.Find(pszTexture);
if ( nIndex == m_vecVMTTextureRemaps.InvalidIndex() )
return NULL;
return m_vecVMTTextureRemaps[nIndex];
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int GetClassCount()
{
return CItemUpload::Manifest()->GetNumClasses();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const char *GetClassString( int i )
{
return ( i < 0 || i >= GetClassCount() ) ? NULL : CItemUpload::Manifest()->GetClass(i);
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const char *GetClassString( const char *pszClassString )
{
if ( !pszClassString || V_strlen( pszClassString ) <= 0 )
return NULL;
// Make sure it exists in our manifest file
for ( int i = 0; i < CItemUpload::Manifest()->GetNumClasses(); i++ )
{
const char *pszHero = CItemUpload::Manifest()->GetClass(i);
if ( V_stricmp(pszHero, pszClassString) == 0 )
return pszHero;
}
//Log_Warning( LOG_ITEMTEST, "Invalid class specified: %s\n", pszClassString );
return NULL;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int GetClassIndex( const char *pszClassString )
{
const char *pszCleanClassString = GetClassString( pszClassString );
if ( !pszCleanClassString )
return -1;
for ( int i = 0; i < GetClassCount(); ++i )
{
if ( !V_stricmp( pszCleanClassString, GetClassString( i ) ) )
{
return i;
}
}
return -1;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
template < class T >
const Vector &CItemUploadGame< T >::GetBipHead( int i )
{
static const Vector vOrigin( 0, 0, 0 );
return ( i < 0 || i >= GetClassCount() ) ? vOrigin : CItemUploadGame< T >::s_vBipHead[i];
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
template < class T >
const RadianEuler &CItemUploadGame< T >::GetBipHeadRotation( int i )
{
static const RadianEuler eOrigin( 0, 0, 0 );
return ( i < 0 || i >= GetClassCount() ) ? eOrigin : CItemUploadGame< T >::s_eBipHead[i];
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const Vector CItemUploadGame< CItemUploadTF >::s_vBipHead[] =
{
Vector( 0, 76.142968, -0.39608 ), // demo
Vector( 0, 69.030248, -1.264691 ), // engineer
Vector( -0.000138993, 79.541796, -3.352982 ), // heavy
Vector( -0.000111273, 76.504372, -0.565035 ), // medic
Vector( -0.000102534, 71.788881, 2.145585 ), // pyro
Vector( 0, 73.501752, -1.429994 ), // scout
Vector( 0, 75.982279, -3.858408 ), // sniper
Vector( 0, 75.194376, -1.120618 ), // soldier
Vector( 0, 75.679732, -2.87915 ) // spy
};
COMPILE_TIME_ASSERT( ARRAYSIZE( CItemUploadGame< CItemUploadTF >::s_vBipHead ) == CItemUploadTF::kClassCount );
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const RadianEuler CItemUploadGame< CItemUploadTF >::s_eBipHead[] =
{
RadianEuler( DEG2RAD( -180.0 ), 0, 0 ), // demo
RadianEuler( DEG2RAD( -170.459 ), 0, 0 ), // engineer
RadianEuler( DEG2RAD( -180.0 ), 0, 0 ), // heavy
RadianEuler( DEG2RAD( -180.0 ), 0, 0 ), // medic
RadianEuler( DEG2RAD( -154.175 ), 0, 0 ), // pyro
RadianEuler( DEG2RAD( -173.451 ), 0, 0 ), // scout
RadianEuler( DEG2RAD( -172.722 ), 0, 0 ), // sniper
RadianEuler( DEG2RAD( -179.729 ), 0, 0 ), // soldier
RadianEuler( DEG2RAD( -180.0 ), 0, 0 ) // spy
};
COMPILE_TIME_ASSERT( ARRAYSIZE( CItemUploadGame< CItemUploadTF >::s_eBipHead ) == CItemUploadTF::kClassCount );
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
template class CItemUploadGame< CItemUploadTF >;
//-----------------------------------------------------------------------------
//
// Try and get the Steam Account ID and return it as a 10 character hex
// string prefixed with 0x.
//
//-----------------------------------------------------------------------------
bool CItemUpload::GetSteamId( CUtlString &sSteamId )
{
if ( GetDevMode() )
{
sSteamId = "";
return true;
}
const char *pszForcedSteamID = GetForcedSteamID();
if ( pszForcedSteamID && pszForcedSteamID[0] )
{
sSteamId = pszForcedSteamID;
return true;
}
bool bRetVal = false;
char szBuf[ BUFSIZ ];
szBuf[0] = '\0';
// Try to query steam directly, this will fail if steam isn't running or the
// process calling this function wasn't launched through steam (or through a
// process launched through steam) or there isn't a steam_appid.txt in
// the same directory as the executable running this (use 440 for TF).
bool shutdownSteam = false;
if ( !SteamClient() )
{
SteamAPI_InitSafe();
SteamAPI_SetTryCatchCallbacks( false ); // We don't use exceptions, so tell steam not to use try/catch in callback handlers
shutdownSteam = true;
}
if ( SteamAPI_IsSteamRunning() )
{
g_SteamAPIContext.Init();
ISteamUser *pSteamUser = g_SteamAPIContext.SteamUser();
if ( pSteamUser )
{
CSteamID cSteamID = pSteamUser->GetSteamID();
const uint32 nAccountID = cSteamID.GetAccountID();
V_snprintf( szBuf, ARRAYSIZE( szBuf ), "0x%08x", nAccountID );
bRetVal = true;
}
}
if ( shutdownSteam )
{
SteamAPI_Shutdown();
}
sSteamId = szBuf;
return bRetVal;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CItemUpload::GetVProjectDir( CUtlString &sVProjectDir )
{
char szVProject[ BUFSIZ ] = "";
GetModSubdirectory( "", szVProject, ARRAYSIZE( szVProject ) );
V_StripTrailingSlash( szVProject );
sVProjectDir = szVProject;
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CItemUpload::GetVMod( CUtlString &sVMod )
{
sVMod = "";
CUtlString sVProjectDir;
if ( !GetVProjectDir( sVProjectDir ) )
return false;
char szBuf[ k64KB ];
V_FileBase( sVProjectDir.String(), szBuf, ARRAYSIZE( szBuf ) );
sVMod = szBuf;
return true;
}
//-----------------------------------------------------------------------------
// Guess where the SourceSDK root is based on executable directory...
// If there's a /bin/orangebox/bin/ in the executable path we know
// the SourceSDK is above it, otherwise we don't know anything and false is
// returned
//
// If DevMode (-dev) then always returns false
//
// TODO: This is for TF & SourceSDK only and likely this is the only
// weird configuration this hack needs to be done with. If it's
// a normal game/content tree then none of this hacky stuff
// should be needed
//-----------------------------------------------------------------------------
bool CItemUpload::GetSourceSDKFromExe( CUtlString &sSourceSDK, CUtlString &sSourceSDKBin )
{
if ( GetDevMode() )
return false;
CUtlString sCurrentExecutableFileName;
GetCurrentExecutableFileName( sCurrentExecutableFileName );
// Special hack for executables running out of the orange box SDK
sCurrentExecutableFileName.FixSlashes( '/' );
const char *pszBinOrangeBoxBin = V_strstr( sCurrentExecutableFileName.String(), "/bin/orangebox/bin/" );
if ( pszBinOrangeBoxBin )
{
sSourceSDK.SetDirect( sCurrentExecutableFileName.String(), pszBinOrangeBoxBin - sCurrentExecutableFileName.String() );
char szBinDir[ MAX_PATH ];
V_ExtractFilePath( sCurrentExecutableFileName.String(), szBinDir, ARRAYSIZE( szBinDir ) );
sSourceSDKBin = szBinDir;
return true;
}
// Get Source SDK path
HKEY hKey;
char szSDKPath[ k64KB ];
char szEngineVersion[ k64KB ];
szSDKPath[0] = szEngineVersion[0] = '\0';
GetEnvironmentVariable( "SOURCESDK", szSDKPath, sizeof( szSDKPath ) );
if ( ERROR_SUCCESS == RegOpenKey( HKEY_CURRENT_USER, "Software\\Valve\\Source SDK", &hKey ) )
{
DWORD dwSize = sizeof( szEngineVersion );
RegQueryValueEx( hKey, "EngineVer", NULL, NULL, (LPBYTE)szEngineVersion, &dwSize );
RegCloseKey( hKey );
}
else
{
// Let's assume orange box for now
V_strcpy_safe( szEngineVersion, "orangebox" );
}
if ( *szSDKPath )
{
// Normalize slashes to be consistent with the orange box SDK code above
V_FixSlashes( szSDKPath, '/' );
sSourceSDK = szSDKPath;
sSourceSDKBin.Format( "%s/bin/%s/bin", szSDKPath, szEngineVersion );
return true;
}
return false;
}
//-----------------------------------------------------------------------------
// Returns $SOURCESDK_content/FileName( $VPROJECT ) if it SOURCESDK is set
// otherwise returns VCONTENT if it is set, otherwise returns
// $VPROJECT/../
//-----------------------------------------------------------------------------
static bool CheckContentPath( CUtlString &sContentDir )
{
char szContentDir[ MAX_PATH ];
V_FixupPathName( szContentDir, ARRAYSIZE( szContentDir ), sContentDir );
V_StripTrailingSlash( szContentDir );
sContentDir = szContentDir;
return g_pFullFileSystem->IsDirectory( sContentDir );
}
bool CItemUpload::GetContentDir( CUtlString &sContentDir )
{
// Without VPROJECT set, can't figure anything out
CUtlString sVMod;
if ( !GetVMod( sVMod ) )
return false;
char szBuf[ k64KB ];
// The game includes its own content directory?
if ( IgnoreEnvironmentVariables() )
{
CUtlString sVProject;
if ( !GetVProjectDir( sVProject ) )
return false;
sVProject.FixSlashes( '/' );
// When we run in Steam, we get something like this back:
// "u:/steambeta/steamapps/common/[staging] dota 2/dota"
// We need to trim off the game name, and append content.
V_strcpy_safe( szBuf, sVProject );
if ( !V_StripLastDir( szBuf, ARRAYSIZE( szBuf ) ) )
return false;
sContentDir = szBuf;
sContentDir += "content/";
sContentDir += sVMod;
return true;
}
CUtlString sSourceSDK, sSourceSDKBin;
// Check for the game/content layout in dev builds
CUtlString sVProject;
if ( GetVProjectDir( sVProject ) )
{
sVProject.FixSlashes( '/' );
const char *pszGame = V_stristr( sVProject.String(), "/game/" );
if ( pszGame )
{
sContentDir = sVProject;
sContentDir.SetLength( pszGame - sVProject.String() );
sContentDir += "/content/";
sContentDir += sVMod;
V_FixupPathName( szBuf, ARRAYSIZE( szBuf ), sContentDir.String() );
sContentDir = szBuf;
if ( CheckContentPath( sContentDir ) )
{
return true;
}
}
else
{
// try to look for workshop/content in the mod dir
sContentDir = sVProject;
sContentDir += "/workshop/content";
V_FixupPathName( szBuf, ARRAYSIZE( szBuf ), sContentDir.String() );
sContentDir = szBuf;
if ( CheckContentPath( sContentDir ) )
{
return true;
}
}
}
// Check for the VCONTENT environment variable
if ( GetEnvironmentVariable( "VCONTENT", szBuf, ARRAYSIZE( szBuf ) ) != 0 )
{
sContentDir = szBuf;
sContentDir += "/";
sContentDir += sVMod;
V_FixupPathName( szBuf, ARRAYSIZE( szBuf ), sContentDir.String() );
sContentDir = szBuf;
if ( CheckContentPath( sContentDir ) )
{
return true;
}
}
// Check for the Source SDK in steam builds
if ( GetSourceSDKFromExe( sSourceSDK, sSourceSDKBin ) )
{
sContentDir = sSourceSDK;
sContentDir += "_content\\";
sContentDir += sVMod;
V_FixupPathName( szBuf, ARRAYSIZE( szBuf ), sContentDir.String() );
sContentDir = szBuf;
if ( CheckContentPath( sContentDir ) )
{
return true;
}
}
return false;
}
//-----------------------------------------------------------------------------
//
// Find the directory for the binaries
//
//-----------------------------------------------------------------------------
static bool CheckToolPath( CUtlString &sBinDir )
{
char szBinDir[ MAX_PATH ];
V_FixupPathName( szBinDir, ARRAYSIZE( szBinDir ), sBinDir );
V_StripTrailingSlash( szBinDir );
sBinDir = szBinDir;
char szVtexFileName[ MAX_PATH ];
V_ComposeFileName( sBinDir, "vtex.exe", szVtexFileName, ARRAYSIZE( szVtexFileName ) );
char szStudiomdlFileName[ MAX_PATH ];
V_ComposeFileName( sBinDir, "studiomdl.exe", szStudiomdlFileName, ARRAYSIZE( szStudiomdlFileName ) );
return g_pFullFileSystem->FileExists( szVtexFileName ) && g_pFullFileSystem->FileExists( szStudiomdlFileName );
}
bool CItemUpload::GetBinDirectory( CUtlString &sBinDir )
{
// Get the full path to the executable this code is running in
// this should be the 'bin' directory we want... just to be sure
// make sure vtex.exe and studiomdl.exe exist in that directory
CUtlString sCurrentExecutableFileName;
GetCurrentExecutableFileName( sCurrentExecutableFileName );
char szBinDir[ MAX_PATH ];
V_ExtractFilePath( sCurrentExecutableFileName.String(), szBinDir, ARRAYSIZE( szBinDir ) );
sBinDir = szBinDir;
if ( CheckToolPath( sBinDir ) )
{
return true;
}
// Check for the game/bin directory
CUtlString sVProject;
if ( GetVProjectDir( sVProject ) && !sVProject.IsEmpty() )
{
sVProject.FixSlashes( '/' );
const char *pszGame = V_stristr( sVProject.String(), "/game/" );
if ( pszGame )
{
sBinDir = sVProject;
sBinDir.SetLength( pszGame - sVProject.String() );
sBinDir += "/game/bin";
if ( CheckToolPath( sBinDir ) )
{
return true;
}
}
// When we run in Steam, we get something like this back:
// "u:/steambeta/steamapps/common/[staging] dota 2/dota"
// We need to trim off the game name, and append bin.
V_strcpy_safe( szBinDir, sVProject );
if ( !V_StripLastDir( szBinDir, ARRAYSIZE( szBinDir ) ) )
return false;
sBinDir = szBinDir;
sBinDir += "/bin";
if ( CheckToolPath( sBinDir ) )
{
return true;
}
}
// Check for the Source SDK in steam builds
CUtlString sSourceSDK;
if ( !GetDevMode() && GetSourceSDKFromExe( sSourceSDK, sBinDir ) )
{
if ( CheckToolPath( sBinDir ) )
{
return true;
}
}
return false;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CItemUpload::FileExists( const char *pszFilename )
{
DWORD attribs = ::GetFileAttributesA( pszFilename );
if ( attribs == INVALID_FILE_ATTRIBUTES )
return false;
return ( ( attribs & FILE_ATTRIBUTE_DIRECTORY ) == 0 );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CItemUpload::CopyFiles( const char *pszSourceDir, const char *pszPattern, const char *pszDestDir )
{
char szFindPattern[ k64KB ];
bool bAllSucceeded = true;
V_snprintf( szFindPattern, sizeof( szFindPattern ), "%s%s", pszSourceDir, pszPattern );
WIN32_FIND_DATA findData;
HANDLE hFind = FindFirstFile( szFindPattern, &findData );
if ( hFind == INVALID_HANDLE_VALUE )
{
return false;
}
else
{
do
{
char szSrcPath[ k64KB ];
char szDestPath[ k64KB ];
V_snprintf( szSrcPath, sizeof( szSrcPath ), "%s%s", pszSourceDir, findData.cFileName );
V_snprintf( szDestPath, sizeof( szDestPath ), "%s\\%s", pszDestDir, findData.cFileName );
DeleteFile( szDestPath );
::CopyFile( szSrcPath, szDestPath, false );
bAllSucceeded &= FileExists( szDestPath );
} while ( FindNextFile( hFind, &findData ) );
FindClose( hFind );
return bAllSucceeded;
}
}
static bool DoFileCopy( const char *pszSourceFile, const char *pszDestFile )
{
int remaining, count;
char buf[4096];
FileHandle_t in, out;
in = g_pFullFileSystem->Open( pszSourceFile, "rb" );
AssertMsg( in, "DoFileCopy: Input file failed to open" );
if ( in == FILESYSTEM_INVALID_HANDLE )
return false;
// create directories up to the cache file
char szDestPath[MAX_PATH];
V_ExtractFilePath( pszDestFile, szDestPath, sizeof( szDestPath ) );
g_pFullFileSystem->CreateDirHierarchy( szDestPath );
out = g_pFullFileSystem->Open( pszDestFile, "wb" );
AssertMsg( out, "DoFileCopy: Output file failed to open" );
if ( out == FILESYSTEM_INVALID_HANDLE )
{
g_pFullFileSystem->Close( in );
return false;
}
remaining = g_pFullFileSystem->Size( in );
while ( remaining > 0 )
{
if (remaining < sizeof(buf))
{
count = remaining;
}
else
{
count = sizeof(buf);
}
g_pFullFileSystem->Read( buf, count, in );
g_pFullFileSystem->Write( buf, count, out );
remaining -= count;
}
g_pFullFileSystem->Close( in );
g_pFullFileSystem->Close( out );
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CItemUpload::CopyFile( const char *pszSourceFile, const char *pszDestFile )
{
if ( ::CopyFile( pszSourceFile, pszDestFile, false ) == 0 )
return false;
DWORD nFileAttr = GetFileAttributes( pszDestFile );
if ( nFileAttr == INVALID_FILE_ATTRIBUTES )
return false;
nFileAttr &= ~FILE_ATTRIBUTE_READONLY;
SetFileAttributes( pszDestFile, nFileAttr );
return true;
}
//=============================================================================
//
//=============================================================================
static bool RemoveTextBlock( const char *str, char const *search, char *pszOutBuf, int nSizeofOutBuf )
{
if ( str != pszOutBuf )
{
V_strncpy( pszOutBuf, str, nSizeofOutBuf );
}
bool changed = false;
if ( !V_strstr( str, search ) )
{
return false;
}
int offset = 0;
while ( true )
{
char* pos = V_strstr( str + offset, search );
if ( !pos )
{
break;
}
CUtlString temp = str;
// Found an instance
int left = pos - str;
CUtlString strLeft = temp.Slice( 0, left );
pos = V_strstr( str + left, "}" );
if ( !pos )
{
AssertMsg( pos, "cannot find end of text block\n" );
return false;
}
int right = pos - str + 1;
CUtlString strRight = temp.Slice( right );
temp = strLeft;
temp += strRight;
// Replace entire string
V_strncpy( pszOutBuf, temp.String(), nSizeofOutBuf );
offset = right;
changed = true;
}
return changed;
}
//=============================================================================
//
//=============================================================================
CTargetBase::CTargetBase( CAsset *pAsset, const CTargetBase *pTargetParent )
: CItemLog( pAsset )
, m_pAsset( pAsset )
, m_nRefCount( 0 )
, m_pTargetParent( pTargetParent )
, m_bIgnoreP4( false )
, m_kvCustomKeys( new KeyValues( "custom keys" ) )
{
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetBase::Compile()
{
CUtlString sName;
if ( !GetOutputPath( sName, 0, PATH_FLAG_FILE | PATH_FLAG_EXTENSION ) )
{
Warning( "CTarget%s::Compile - GetOutputPath failed\n", GetTypeString() );
return false;
}
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( "CTarget%s::Compile( %s ) - Not Valid: %s\n", GetTypeString(), sName.String(), sTmp.String() );
return false;
}
// Compile all inputs first
if ( !CreateOutputDirectory() )
{
Warning( "CTarget%s::Compile - CreateOutputDirectory failed\n", GetTypeString() );
return false;
}
CUtlVector< CTargetBase * > inputs;
bool bRet = GetInputs( inputs );
if ( !bRet )
{
Warning( "CTarget%s::Compile - GetInputs failed\n", GetTypeString() );
return bRet;
}
for ( int i = 0; i < inputs.Count(); ++i )
{
CTargetBase *pTargetBase = inputs.Element( i );
if ( !pTargetBase )
{
Warning( "WARNING: CTarget%s::Compile - Target %d NULL\n", GetTypeString(), i );
continue;
}
if ( !pTargetBase->Compile() )
{
Warning( "WARNING: CTarget%s::Compile - Target %d Compile Failed\n", GetTypeString(), i );
bRet = false;
break;
}
}
return bRet;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const char *CTargetBase::GetItemDirectory() const
{
if ( m_pTargetParent )
{
return m_pTargetParent->GetItemDirectory();
}
else
{
return CItemUpload::Manifest()->GetItemDirectory();
}
}
//-----------------------------------------------------------------------------
// Return the number of files that are output from this CTarget
//-----------------------------------------------------------------------------
int CTargetBase::GetOutputCount() const
{
const ExtensionList *pList = GetExtensionsAndCount();
return (pList ? pList->Count() : 0);
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetBase::GetOutputPath(
CUtlString &sOutputPath,
int nIndex /* = 0 */,
uint nPathFlags /* = PATH_FLAG_ALL */ ) const
{
sOutputPath.Clear();
if ( nIndex < 0 )
return false;
CUtlVector< CUtlString > sOutputPaths;
if ( !GetOutputPaths( sOutputPaths, nPathFlags ) )
return false;
if ( nIndex >= sOutputPaths.Count() )
return false;
sOutputPath = sOutputPaths.Element( nIndex );
return ( sOutputPath.Length() > 0 );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetBase::GetOutputPaths(
CUtlVector< CUtlString > &sOutputPaths,
uint nPathFlags /* = PATH_FLAG_ALL */,
bool bRecurse /* = false */ ) const
{
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - Not Valid: %s\n", GetTypeString(), sTmp.String() );
return false;
}
CUtlString sDirName;
if ( nPathFlags & PATH_FLAG_PATH )
{
if ( !GetDirName( sDirName, nPathFlags ) )
return false;
}
const ExtensionList *pvecExtensions = GetExtensionsAndCount();
int nExtCount = pvecExtensions ? pvecExtensions->Count() : 0;
for ( int i = 0; i < nExtCount; ++i )
{
CUtlString &sOutputPath = sOutputPaths.Element( sOutputPaths.AddToTail() );
if ( nPathFlags & PATH_FLAG_PATH )
{
sOutputPath = sDirName;
}
if ( nPathFlags & PATH_FLAG_FILE )
{
if ( sOutputPath.Length() > 0 )
{
sOutputPath += "/";
}
CUtlString sName;
GetName( sName );
sOutputPath += sName;
if ( nPathFlags & PATH_FLAG_EXTENSION )
{
sOutputPath += pvecExtensions->Element(i);
}
}
sOutputPath.FixSlashes();
}
bool bRet = sOutputPaths.Count() > 0;
if ( bRecurse )
{
CUtlVector< CTargetBase * > inputs;
if ( GetInputs( inputs ) )
{
for ( int i = 0; i < inputs.Count(); ++i )
{
CTargetBase *pTargetBase = inputs.Element( i );
if ( !pTargetBase )
continue;
bRet = pTargetBase->GetOutputPaths( sOutputPaths, nPathFlags, bRecurse );
}
}
else
{
bRet = false;
}
}
return bRet;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetBase::GetOutputPaths(
CUtlVector< CUtlString > &sOutputPaths,
bool bRelative /* = true */,
bool bRecurse /* = true */,
bool bExtension /* = true */,
bool bPrefix /* = true */ ) const
{
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - Not Valid: %s\n", GetTypeString(), sTmp.String() );
return false;
}
CUtlString sDirA;
if ( bRelative )
{
if ( !Asset()->GetRelativeDir( sDirA, bPrefix ? GetPrefix() : NULL, this ) )
return false;
}
else
{
if ( !Asset()->GetAbsoluteDir( sDirA, bPrefix ? GetPrefix() : NULL, this ) )
return false;
}
const ExtensionList *pvecExtensions = GetExtensionsAndCount();
int nExtCount = pvecExtensions ? pvecExtensions->Count() : 0;
for ( int i = 0; i < nExtCount; ++i )
{
CUtlString &sRelativePath = sOutputPaths.Element( sOutputPaths.AddToTail() );
CUtlString sName;
GetName( sName );
sRelativePath = sDirA;
sRelativePath += "/";
sRelativePath += sName;
if ( bExtension )
{
sRelativePath += pvecExtensions->Element(i);
}
sRelativePath.FixSlashes();
}
bool bRet = sOutputPaths.Count() > 0;
if ( bRecurse )
{
CUtlVector< CTargetBase * > inputs;
if ( GetInputs( inputs ) )
{
for ( int i = 0; i < inputs.Count(); ++i )
{
CTargetBase *pTargetBase = inputs.Element( i );
if ( !pTargetBase )
continue;
bRet = pTargetBase->GetOutputPaths( sOutputPaths, bRelative, bRecurse, bExtension ) && bRet;
}
}
else
{
bRet = false;
}
}
return bRet;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetBase::GetInputPaths(
CUtlVector< CUtlString > &sInputPaths,
bool bRelative /* = true */,
bool bRecurse /* = true */,
bool bExtension /* = true */ )
{
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - Not Valid: %s\n", GetTypeString(), sTmp.String() );
return false;
}
CUtlVector< CTargetBase * > inputs;
if ( !GetInputs( inputs ) )
return false;
bool bRet = true;
for ( int i = 0; bRet && i < inputs.Count(); ++i )
{
CTargetBase *pTargetBase = inputs.Element( i );
if ( !pTargetBase )
continue;
bRet = pTargetBase->GetOutputPaths( sInputPaths, bRelative, false, bExtension ) && bRet;
if ( bRecurse )
{
bRet = pTargetBase->GetInputPaths( sInputPaths, bRelative, bRecurse, bExtension ) && bRet;
}
}
return bRet;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetBase::GetName( CUtlString &sName ) const
{
if ( !GetCustomOutputName().IsEmpty() )
{
sName = GetCustomOutputName();
sName += GetNameSuffix();
return;
}
// don't take parent's custom output name
if ( m_pTargetParent && m_pTargetParent->GetCustomOutputName().IsEmpty() )
{
m_pTargetParent->GetName( sName );
}
else
{
sName = GetAssetName();
}
sName += GetNameSuffix();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const CUtlString &CTargetBase::GetAssetName() const
{
return Asset()->GetAssetName();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CAsset *CTargetBase::Asset() const
{
return m_pAsset;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetBase::CheckFile( const char *pszFilename ) const
{
if ( _access( pszFilename, 06 ) == 0 )
return true;
Warning( "CTarget%s::Compile NO FILE! - %s\n", GetTypeString(), pszFilename );
return false;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetBase::GetDirName(
CUtlString &sDirName,
uint nPathFlags /* = PATH_FLAG_ALL */ ) const
{
sDirName.Clear();
CAsset *pAsset = Asset();
if ( !pAsset )
{
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - NULL Asset\n", GetTypeString() );
return false;
}
CUtlString sTmp;
if ( !pAsset->IsOk( sTmp ) )
{
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - Not Valid: %s\n", GetTypeString(), sTmp.String() );
return false;
}
if ( nPathFlags & PATH_FLAG_ABSOLUTE )
{
if ( IsContent() )
{
if ( !CItemUpload::GetContentDir( sDirName ) )
return false;
}
else
{
if ( !CItemUpload::GetVProjectDir( sDirName ) )
return false;
}
sDirName += "/";
// ABSOLUTE implies PREFIX & MODELS
nPathFlags |= PATH_FLAG_PREFIX;
nPathFlags |= PATH_FLAG_MODELS;
}
if ( nPathFlags & PATH_FLAG_ZIP )
{
const char *pszZipPrefix;
if ( IsContent() )
{
pszZipPrefix = CItemUpload::Manifest()->GetZipSourceDirectory();
}
else
{
pszZipPrefix = CItemUpload::Manifest()->GetZipOutputDirectory();
}
if ( *pszZipPrefix )
{
sDirName += pszZipPrefix;
sDirName += "/";
}
}
if ( ( nPathFlags & PATH_FLAG_PREFIX ) && ( nPathFlags & PATH_FLAG_MODELS ) )
{
const char *pszPrefix = GetPrefix();
if ( pszPrefix )
{
sDirName += pszPrefix;
sDirName += "/";
}
// PREFIX implies MODELS
nPathFlags |= PATH_FLAG_MODELS;
}
if ( nPathFlags & PATH_FLAG_MODELS && IsModelPath() )
{
// If not starting with prefix, then optionally start with models
sDirName += "models/";
}
if ( GetCustomRelativeDir() )
{
sDirName += GetCustomRelativeDir();
}
else
{
sDirName += GetItemDirectory();
sDirName += pAsset->GetClass();
sDirName += "/";
if ( CItemUpload::Manifest()->GetItemPathUsesSteamId() )
{
const char *pszSteamId = pAsset->GetSteamId();
if ( pszSteamId )
{
sDirName += pszSteamId;
sDirName += "/";
}
}
sDirName += pAsset->GetAssetName();
}
char szBuf[ k64KB ];
V_FixupPathName( szBuf, ARRAYSIZE( szBuf ), sDirName.String() );
sDirName = szBuf;
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetBase::CreateOutputDirectory() const
{
CUtlString sDir;
if ( !GetOutputPath( sDir, 0, PATH_FLAG_PATH | PATH_FLAG_ABSOLUTE ) )
return false;
if ( !CItemUpload::CreateDirectory( sDir.String() ) )
{
Warning( "CTarget%s::CreateDirectory( %s ) - Failed\n", GetTypeString(), sDir.String() );
return false;
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetBase::AddOrEditP4File( const char *pszFilePath )
{
if ( m_bIgnoreP4 )
return;
if ( CItemUpload::GetP4() )
{
char szCorrectCaseFilePath[MAX_PATH];
g_pFullFileSystem->GetCaseCorrectFullPath( pszFilePath, szCorrectCaseFilePath );
CP4AutoEditAddFile a( szCorrectCaseFilePath );
}
}
//=============================================================================
//
//=============================================================================
bool VTFGetInfo( const char *fileName, int *width, int *height, ImageFormat *imageFormat, float *sourceGamma )
{
// Just load the whole file into a memory buffer
CUtlBuffer bufFileContents;
if ( !g_pFullFileSystem->ReadFile( fileName, NULL, bufFileContents ) )
{
return false;
}
IVTFTexture *pVTFTexture = CreateVTFTexture();
if ( !pVTFTexture->Unserialize( bufFileContents, true ) )
{
return false;
}
*width = pVTFTexture->Width();
*height = pVTFTexture->Height();
// It's not actually RGBA, but it will be when we decompress and load it...
*imageFormat = IMAGE_FORMAT_RGBA8888;
*sourceGamma = 0.0f;
DestroyVTFTexture( pVTFTexture );
return true;
}
//=============================================================================
//
//=============================================================================
CTargetTGA::CTargetTGA( CAsset *pAsset, const CTargetVMT *pTargetVMT )
: CTargetBase( pAsset, pTargetVMT )
, m_pTargetVMT( pTargetVMT )
, m_nSrcImageType( IMAGE_FILE_UNKNOWN )
, m_nWidth( 0 )
, m_nHeight( 0 )
, m_nChannelCount( 0 )
, m_bNoNiceFiltering( false )
, m_bAlpha( false )
, m_bPowerOfTwo( false )
{
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CTargetTGA::~CTargetTGA()
{
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetTGA::IsOk( CUtlString &sMsg ) const
{
// No input file specified
if ( m_sInputFile.Length() <= 0 )
{
sMsg = "No input file specified";
return false;
}
// Input file invalid size
if ( m_nWidth <= 0 )
{
sMsg = "Invalid image width (";
sMsg += m_nWidth;
sMsg += ")";
return false;
}
if ( m_nHeight <= 0 )
{
sMsg = "Invalid image height (";
sMsg += m_nHeight;
sMsg += ")";
return false;
}
// TODO: Maximum size?
// Only 3 or 4 channel images ok
if ( m_nChannelCount != 3 && m_nChannelCount != 4 )
{
sMsg = "Invalid number of channels (";
sMsg += m_nChannelCount;
sMsg = ") only 3 (RGB) and 4 (RGBA) channel images allowed";
return false;
}
// Has to be a power of two
if ( !m_bPowerOfTwo )
{
sMsg = "Image dimensions are not a power of two";
return false;
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetTGA::IsModelPath() const
{
return m_pTargetVMT->IsModelPath();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetTGA::GetInputPaths(
CUtlVector< CUtlString > &sInputPaths,
bool bRelative /* = true */,
bool bRecurse /* = true */,
bool bExtension /* = true */ )
{
sInputPaths.AddToTail( m_sInputFile );
return CTargetBase::GetInputPaths( sInputPaths, bRelative, bRecurse, bExtension );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const ExtensionList *CTargetTGA::GetExtensionsAndCount( void ) const
{
static ExtensionList vecExtensions;
if ( !vecExtensions.Count() )
{
vecExtensions.AddToTail( ".tga" );
vecExtensions.AddToTail( ".txt" );
}
return &vecExtensions;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const char *CTargetTGA::GetPrefix() const
{
if ( m_sPrefix.IsEmpty() ) {
CUtlString strParentPrefix = m_pTargetVMT->GetPrefix();
m_sPrefix = strParentPrefix.Replace( "materials", "materialsrc" );
}
return m_sPrefix.String();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetTGA::GetName( CUtlString &sName ) const
{
if ( GetCustomOutputName().IsEmpty() )
{
CTargetBase::GetName( sName );
}
else
{
sName = GetCustomOutputName();
}
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetTGA::Compile()
{
// 'reset' the input file from the current value to force a reload on compile, make a temp copy as it's going to be overwritten
if ( !SetInputFile( CUtlString( GetInputFile() ).String() ) )
return false;
if ( !CTargetBase::Compile() )
return false;
CUtlString sName;
if ( !GetOutputPath( sName, 0, PATH_FLAG_FILE | PATH_FLAG_EXTENSION ) )
{
Warning( "CTarget%s::Compile - GetOutputPath Failed\n", GetTypeString() );
return false;
}
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( "CTarget%s::Compile( %s ) - Not Valid: %s\n", GetTypeString(), sName.String(), sTmp.String() );
return false;
}
CUtlString sAbsPath;
GetOutputPath( sAbsPath, 0 );
CUtlString sRelPath;
GetOutputPath( sRelPath, 0, PATH_FLAG_ALL & ~PATH_FLAG_ABSOLUTE );
if ( sAbsPath.IsEmpty() || sRelPath.IsEmpty() )
{
Warning( "CTarget%s::Compile( %s ) - GetOutputPath failed\n", GetTypeString(), sName.String() );
return false;
}
Msg( "Compiling %s: %s\n", GetTypeString(), sRelPath.String() );
if ( CItemUpload::IsSameFile( m_sInputFile.String(), sAbsPath.String() ) )
{
Warning( "CTarget%s::Compile( %s ) - Same File, No Work To Do!\n", GetTypeString(), sName.String() );
return true;
}
Bitmap_t bitmap;
CUtlMemory< unsigned char > tgaBits;
switch ( m_nSrcImageType )
{
case IMAGE_FILE_TGA:
{
int nWidth = 0, nHeight = 0;
if ( !TGALoader::LoadRGBA8888( m_sInputFile.String(), tgaBits, nWidth, nHeight ) )
{
Warning( "CTarget%s::Compile( %s ) - Couldn't Load TGA\n", GetTypeString(), sName.String() );
return false;
}
bitmap.SetBuffer( nWidth, nHeight, IMAGE_FORMAT_RGBA8888, tgaBits.Base(), false, nWidth*4 );
}
break;
case IMAGE_FILE_PSD:
{
if ( !PSDReadFileRGBA8888( m_sInputFile.String(), NULL, bitmap ) )
{
Warning( "CTarget%s::Compile( %s ) - Couldn't Load PSD\n", GetTypeString(), sName.String() );
return false;
}
}
break;
case IMAGE_FILE_VTF:
{
// Just load the whole file into a memory buffer
CUtlBuffer bufFileContents;
if ( !g_pFullFileSystem->ReadFile( m_sInputFile.String(), NULL, bufFileContents ) )
{
Warning( "CTarget%s::Compile( %s ) - Couldn't Load VTF\n", GetTypeString(), sName.String() );
return false;
}
IVTFTexture *pVTFTexture = CreateVTFTexture();
if ( !pVTFTexture->Unserialize( bufFileContents ) )
{
Warning( "CTarget%s::Compile( %s ) - Couldn't Load VTF\n", GetTypeString(), sName.String() );
return false;
}
int nWidth = pVTFTexture->Width();
int nHeight = pVTFTexture->Height();
pVTFTexture->ConvertImageFormat( IMAGE_FORMAT_RGBA8888, false );
int nMemSize = ImageLoader::GetMemRequired( nWidth, nHeight, 1, IMAGE_FORMAT_RGBA8888, false );
tgaBits.EnsureCapacity( nMemSize );
Q_memcpy( tgaBits.Base(), pVTFTexture->ImageData(), nMemSize );
DestroyVTFTexture( pVTFTexture );
bitmap.SetBuffer( nWidth, nHeight, IMAGE_FORMAT_RGBA8888, tgaBits.Base(), false, nWidth*4 );
}
break;
}
if ( bitmap.Format() != IMAGE_FORMAT_RGBA8888 || bitmap.Width() != m_nWidth || bitmap.Height() != m_nHeight )
{
Warning( "CTarget%s::Compile( %s ) - Invalid Bitmap Size, Expected %d x %d x %d (%d)\n",
GetTypeString(), sName.String(), m_nWidth, m_nHeight, 4, m_nWidth * m_nHeight * 4 );
return false;
}
CUtlBuffer bufVTEXConfig( 0, 0, CUtlBuffer::TEXT_BUFFER );
// If we're supposed to resize these to another resolution, setup the VTEX config file to do so.
int nTargetWidth = m_pTargetVMT->GetTargetWidth();
int nTargetHeight = m_pTargetVMT->GetTargetHeight();
if ( nTargetWidth && nTargetHeight )
{
// We want to use "reduce", not "maxwidth" & "maxheight", so we throw away the higher resolution mips.
// Determine the right amount of reduction based on the texture's width & height (and fail if it's the wrong aspect ratio)
int nFactor = (m_nWidth / nTargetWidth);
if ( nFactor <= 0 )
{
Warning( "CTarget%s::Compile( %s ) - Failed to determine reduction factor (target width is %d, texture is %d)\n", GetTypeString(), sName.String(), nTargetWidth, m_nWidth );
return false;
}
int nHeightFactor = (m_nHeight / nTargetHeight);
if ( nFactor != nHeightFactor )
{
Warning( "CTarget%s::Compile( %s ) - Failed to determine reduction factor (target size aspect ratio (%dx%d) doesn't match texture's aspect ratio (%dx%d))\n", GetTypeString(), sName.String(), nTargetWidth, nTargetHeight, m_nWidth, m_nHeight );
return false;
}
if ( nFactor > 1 )
{
bufVTEXConfig.PutString( CFmtStr("reduce %d\n", nFactor) );
}
}
KeyValues *pKV = m_pTargetVMT->GetTextureAddToVTEXConfigForTGA( this );
if ( pKV )
{
FOR_EACH_SUBKEY( pKV, pKVEntry )
{
bufVTEXConfig.Printf( "%s %s\n", pKVEntry->GetName(), pKVEntry->GetString() );
}
}
// check if we should modify abspath
CUtlString sTGAOutput = Asset()->CheckRedundantOutputFilePath( GetInputFile().String(), bufVTEXConfig.String(), sAbsPath.String() );
AddOrEditP4File( sTGAOutput.String() );
char szVTEXConfigFilename[MAX_PATH];
V_strcpy_safe( szVTEXConfigFilename, sTGAOutput.String() );
V_SetExtension( szVTEXConfigFilename, ".txt", sizeof(szVTEXConfigFilename) );
AddOrEditP4File( szVTEXConfigFilename );
g_pFullFileSystem->WriteFile( szVTEXConfigFilename, NULL, bufVTEXConfig );
// TODO: Don't write alpha if VMT using this doesn't specify the alpha is used for anything
CUtlBuffer outBuf;
const bool bWriteTGA = TGAWriter::WriteToBuffer( bitmap.GetBits(), outBuf, bitmap.Width(), bitmap.Height(), bitmap.Format(), m_bAlpha ? IMAGE_FORMAT_BGRA8888 : IMAGE_FORMAT_BGR888 );
if ( !bWriteTGA )
{
Warning( "CTarget%s::Compile( %s ) - Couldn't write TGA to buffer\n", GetTypeString(), sName.String() );
return false;
}
if ( !g_pFullFileSystem->WriteFile( sTGAOutput.String(), NULL, outBuf ) )
{
Warning( "CTarget%s::Compile( %s ) - Couldn't write TGA to file \"%s\"\n", GetTypeString(), sName.String(), sTGAOutput.String() );
return false;
}
if ( !CheckFile( sTGAOutput.String() ) )
{
Warning( "CTarget%s::Compile( %s ) - File Check Failed - \"%s\"\n", GetTypeString(), sName.String(), sTGAOutput.String() );
return false;
}
if ( CItemUpload::Manifest()->UseTerseMessages() )
{
Msg( " - Compilation successful.\n" );
}
else
{
Msg( "CTarget%s::Compile OK! - %s\n", GetTypeString(), sRelPath.String() );
}
// store the output name to use it to output VTF file
char szFileName[FILENAME_MAX];
V_StripExtension( V_GetFileName( szVTEXConfigFilename ), szFileName, ARRAYSIZE( szFileName ) );
SetCustomOutputName( szFileName );
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetTGA::Clear()
{
m_sInputFile = "";
m_sFileBase = "";
m_sExtension = "";
m_nWidth = 0;
m_nHeight = 0;
m_nChannelCount = 0;
m_bNoNiceFiltering = false;
m_bAlpha = false;
m_bPowerOfTwo = false;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetTGA::SetInputFile( const char *pszFilename )
{
Clear();
if ( !pszFilename || V_strlen( pszFilename ) <= 0 )
{
Warning( "ERROR: Empty filename specified for TGA file\n" );
return false;
}
char szBuf[ k64KB ];
m_sInputFile = pszFilename;
V_FileBase( pszFilename, szBuf, ARRAYSIZE( szBuf ) );
m_sFileBase = szBuf;
// Try to automatically handle the suffixes for properly named assets
static const char *szSuffixes[] =
{
//"_color",
"_normal",
"_height",
"_specmask"
"_specexp",
"_trans",
"_illum",
"_color_red",
"_color_blue"
};
CUtlString sTmp;
for ( int i = 0; i < ARRAYSIZE( szSuffixes ); ++i )
{
const char *pszSuffix = szSuffixes[i];
sTmp = GetAssetName();
sTmp += pszSuffix;
if ( m_sFileBase == sTmp )
{
SetNameSuffix( pszSuffix );
}
}
V_ExtractFileExtension( pszFilename, szBuf, ARRAYSIZE( szBuf ) );
m_sExtension = szBuf;
ImageFormat imageFormat;
float flSourceGamma = 0;
if ( TGALoader::GetInfo( pszFilename, &m_nWidth, &m_nHeight, &imageFormat, &flSourceGamma ) )
{
m_nSrcImageType = IMAGE_FILE_TGA;
}
else if ( PSDGetInfo( pszFilename, NULL, &m_nWidth, &m_nHeight, &imageFormat, &flSourceGamma ) )
{
m_nSrcImageType = IMAGE_FILE_PSD;
}
else if ( VTFGetInfo( pszFilename, &m_nWidth, &m_nHeight, &imageFormat, &flSourceGamma ) )
{
m_nSrcImageType = IMAGE_FILE_VTF;
}
else
{
Warning( "ERROR: Specified file is not a TGA (Targa) or PSD File: \"%s\"\n", szBuf );
Clear();
return false;
}
// ImageFormat can only be one of:
switch ( imageFormat )
{
case IMAGE_FORMAT_I8:
m_nChannelCount = 1;
m_bAlpha = false;
break;
case IMAGE_FORMAT_ABGR8888:
case IMAGE_FORMAT_BGRA8888:
m_nChannelCount = 4;
m_bAlpha = true;
break;
case IMAGE_FORMAT_BGR888:
m_nChannelCount = 3;
m_bAlpha = false;
break;
case IMAGE_FORMAT_RGBA8888:
m_nChannelCount = 4;
m_bAlpha = true;
break;
case IMAGE_FORMAT_RGB888:
m_nChannelCount = 3;
m_bAlpha = false;
break;
default:
break;
}
const int nWidthPow = NearestPowerOfTwo( m_nWidth );
const int nHeightPow = NearestPowerOfTwo( m_nHeight );
if ( nWidthPow == m_nWidth && nHeightPow == m_nHeight )
{
m_bPowerOfTwo = true;
}
else
{
Warning( "ERROR: Specified texture file (%s) Size %dx%d dimensions not powers of two, perhaps resize to %dx%d\n",
m_sInputFile.String(),
m_nWidth, m_nHeight,
nWidthPow, nHeightPow );
return false;
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const CUtlString &CTargetTGA::GetInputFile() const
{
return m_sInputFile;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
template < class T >
T CTargetTGA::NearestPowerOfTwo( T v )
{
if (v == 0)
return static_cast< T >( 1 );
T k;
for ( k = sizeof( T ) * 8 - 1; ( ( static_cast< T >( 1U ) << k) & v ) == 0; --k);
if ( ( ( static_cast< T >( 1U ) << ( k - 1 ) ) & v ) == 0 )
return static_cast< T >( 1U ) << k;
return static_cast< T >( 1U ) << ( k + 1 );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetTGA::UpdateManifest( KeyValues *pKv )
{
pKv->SetString( "filename", m_sInputFile.String() );
CUtlString sOutName;
if ( GetOutputPath( sOutName, 0, PATH_FLAG_PATH | PATH_FLAG_FILE | PATH_FLAG_PREFIX | PATH_FLAG_MODELS | PATH_FLAG_EXTENSION ) )
{
pKv->SetString( "out_filename", sOutName );
}
pKv->SetInt( "width", m_nWidth );
pKv->SetInt( "height", m_nHeight );
pKv->SetInt( "channels", m_nChannelCount );
pKv->SetInt( "nonice", m_bNoNiceFiltering );
pKv->SetBool( "alpha", m_bAlpha );
const char *pszTextureType = m_pTargetVMT->GetTextureTypeForTGA( this );
if ( pszTextureType )
{
KeyValues *pVTEXKV = CItemUpload::Manifest()->GetTextureAddToVTEXConfig( pszTextureType );
if ( pVTEXKV )
{
KeyValues *pTmpKey = new KeyValues( pVTEXKV->GetName() );
pKv->AddSubKey( pTmpKey );
pVTEXKV->CopySubkeys( pTmpKey );
}
}
}
//=============================================================================
//
//=============================================================================
CTargetVTF::CTargetVTF( CAsset *pAsset, const CTargetVMT *pTargetVMT )
: CTargetBase( pAsset, pTargetVMT )
, m_pTargetVMT( pTargetVMT )
{
m_pTargetTGA = Asset()->NewTarget< CTargetTGA >( m_pTargetVMT );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CTargetVTF::~CTargetVTF()
{
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetVTF::IsModelPath() const
{
return m_pTargetVMT->IsModelPath();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetVTF::Compile()
{
if ( !CTargetBase::Compile() )
return false;
CUtlString sName;
if ( !GetOutputPath( sName, 0, PATH_FLAG_FILE | PATH_FLAG_EXTENSION ) )
return false;
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( "CTarget%s::Compile( %s ) - Not Valid: %s\n", GetTypeString(), sName.String(), sTmp.String() );
return false;
}
CUtlString sBinDir;
if ( !CItemUpload::GetBinDirectory( sBinDir ) )
{
Warning( "CTarget%s::Compile( %s ) - GetBinDirectory Failed\n", GetTypeString(), sName.String() );
return false;
}
CUtlString sAbsPath;
GetOutputPath( sAbsPath, 0 );
CUtlString sRelPath;
GetOutputPath( sRelPath, 0, PATH_FLAG_ALL & ~PATH_FLAG_ABSOLUTE );
if ( sAbsPath.IsEmpty() || sRelPath.IsEmpty() )
{
Warning( "CTarget%s::Compile( %s ) - GetOutputPath failed\n", GetTypeString(), sName.String() );
return false;
}
Msg( "Compiling %s: %s\n", GetTypeString(), sRelPath.String() );
AddOrEditP4File( sAbsPath.String() );
CUtlVector< CUtlString > sAbsInputPaths;
if ( !GetInputPaths( sAbsInputPaths, false, false ) )
{
Warning( "CTarget%s::Compile( %s ) - GetInputPaths failed\n", GetTypeString(), sName.String() );
return false;
}
if ( sAbsInputPaths.Count() != 2 )
{
Warning( "CTarget%s::Compile( %s ) - GetPaths returned %d paths, expected 2\n", GetTypeString(), sName.String(), sAbsInputPaths.Count() );
return false;
}
CFmtStrN< k64KB > sCmd;
if ( CItemUpload::IgnoreEnvironmentVariables() )
{
// We can't rely on environment variables in VTEX. So tell it to just built it on the spot, and we'll move it afterwards.
sCmd.sprintf( "\"%s\\vtex.exe\" -nop4 -nopause -dontusegamedir \"%s\"", sBinDir.String(), sAbsInputPaths.Element( 0 ).String() );
}
else
{
CUtlString sVProject;
CItemUpload::GetVProjectDir( sVProject );
sCmd.sprintf( "\"%s\\vtex.exe\" -nop4 -nopause -vproject \"%s\" \"%s\"", sBinDir.String(), sVProject.String(), sAbsInputPaths.Element( 0 ).String() );
}
if ( !CItemUpload::RunCommandLine( sCmd.Access(), sBinDir.String(), this ) )
{
Warning( "CTarget%s::Compile( %s ) - RunCommandLine Failed - \"%s\"\n", GetTypeString(), sName.String(), sCmd.Access() );
return false;
}
if ( CItemUpload::IgnoreEnvironmentVariables() )
{
char sVTFName[MAX_PATH];
V_strcpy_safe( sVTFName, sAbsInputPaths.Element( 0 ).String() );
V_SetExtension( sVTFName, ".vtf", sizeof(sVTFName) );
// We built the VTF in the directory with the content. Now move it to the out dir.
if ( ::MoveFileEx( sVTFName, sAbsPath.String(), MOVEFILE_REPLACE_EXISTING ) == 0 )
{
Warning( "CTarget%s::Compile( %s ) - Failed to move \"%s\" to \"%s\"\n", GetTypeString(), sName.String(), sVTFName, sAbsPath.String() );
return false;
}
}
if ( !CheckFile( sAbsPath.String() ) )
{
Warning( "CTarget%s::Compile( %s ) - File Check Failed - \"%s\"\n", GetTypeString(), sName.String(), sAbsPath.String() );
return false;
}
if ( CItemUpload::Manifest()->UseTerseMessages() )
{
Msg( " - Compilation successful.\n" );
}
else
{
Msg( "CTarget%s::Compile OK! - %s\n", GetTypeString(), sRelPath.String() );
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetVTF::GetInputs( CUtlVector< CTargetBase * > &inputs ) const
{
Assert( m_pTargetTGA.IsValid() );
inputs.AddToTail( m_pTargetTGA.GetObject() );
return inputs.Count() > 0;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const ExtensionList *CTargetVTF::GetExtensionsAndCount( void ) const
{
static ExtensionList vecExtensions;
if ( !vecExtensions.Count() )
{
vecExtensions.AddToTail( ".vtf" );
}
return &vecExtensions;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const char *CTargetVTF::GetPrefix() const
{
return m_pTargetVMT->GetPrefix();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetVTF::GetName( CUtlString &sName ) const
{
Assert( m_pTargetTGA.IsValid() );
m_pTargetTGA->GetName( sName );
}
//=============================================================================
//
//=============================================================================
CTargetVMT::CTargetVMT( CAsset *pAsset, const CTargetBase *pTargetParent )
: CTargetBase( pAsset, pTargetParent )
, m_pVMTKV( NULL )
, m_nColorAlphaType( kNoColorAlpha )
, m_nNormalAlphaType( kNoNormalAlpha )
, m_nMaterialType( kInvalidMaterialType )
, m_bDuplicate( false )
, m_nTargetWidth( 0 )
, m_nTargetHeight( 0 )
{
m_vecTargetVTFs.SetCount( CItemUpload::Manifest()->GetNumMaterialSkins() );
FOR_EACH_VEC( m_vecTargetVTFs, i )
{
m_vecTargetVTFs[i].SetCount( CItemUpload::Manifest()->GetNumTextureTypes() );
}
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CTargetVMT::~CTargetVMT()
{
if ( m_pVMTKV )
{
m_pVMTKV->deleteThis();
}
CUtlString sMaterialId;
GetMaterialId( sMaterialId );
Asset()->RemoveMaterial( sMaterialId );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetVMT::Compile()
{
if ( GetDuplicate() )
return true;
if ( !CTargetBase::Compile() )
return g_bCompilePreview ? true : false;
CAsset *pAsset = Asset();
if ( !pAsset )
return false;
CUtlString sName;
if ( !GetOutputPath( sName, 0, PATH_FLAG_FILE | PATH_FLAG_EXTENSION ) )
return false;
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( "CTarget%s::Compile( %s ) - Not Valid: %s\n", GetTypeString(), sName.String(), sTmp.String() );
return false;
}
int nVmtCount = GetOutputCount();
CUtlVector< CUtlString > sAbsOutputPaths;
CUtlVector< CUtlString > sRelOutputPaths;
if ( !GetOutputPaths( sAbsOutputPaths, false, false, true ) || !GetOutputPaths( sRelOutputPaths, true, false, true ) )
{
Warning( "CTarget%s::Compile( %s ) - GetOutputPaths failed\n", GetTypeString(), sName.String() );
return false;
}
if ( sAbsOutputPaths.Count() != nVmtCount || sAbsOutputPaths.Count() != sRelOutputPaths.Count() )
{
Warning( "CTarget%s::Compile( %s ) - GetOutputPaths returned %d paths, expected %d\n", GetTypeString(), sName.String(), sAbsOutputPaths.Count(), 1 );
return false;
}
Assert( nVmtCount == sAbsOutputPaths.Count() );
Assert( nVmtCount == sRelOutputPaths.Count() );
char szBuf[ k64KB ];
for ( int i = 0; i < nVmtCount; ++i )
{
CUtlVector< CUtlString > sChildRelOutputPaths;
if ( !GetOutputPath( sName, i, PATH_FLAG_FILE | PATH_FLAG_EXTENSION ) )
{
Warning( "CTarget%s::Compile( %s ) - Can't GetOutputPath %d\n", GetTypeString(), sName.String(), i );
return false;
}
Msg( "Compiling %s: %s\n", GetTypeString(), sRelOutputPaths.Element( i ).String() );
AddOrEditP4File( sAbsOutputPaths.Element( i ).String() );
// Load the template in.
KeyValues *pVMTKV = GetVMTKV( i );
if ( !pVMTKV )
{
// We've already printed a warning, so we're done
return false;
}
// Set any remapped VMT vars.
FOR_EACH_VEC( m_vecTargetVTFs[i], iVTF )
{
if ( m_vecTargetVTFs[i][iVTF].IsValid() )
{
const char *pszVMTVar = CItemUpload::Manifest()->GetVMTVarForTextureType( CItemUpload::Manifest()->GetTextureType(iVTF) );
if ( pszVMTVar && pszVMTVar[0] )
{
sChildRelOutputPaths.RemoveAll();
m_vecTargetVTFs[i][iVTF]->GetOutputPaths( sChildRelOutputPaths, true, false, false, false );
if ( sChildRelOutputPaths.Count() > 0 )
{
V_strncpy( szBuf, sChildRelOutputPaths.Element( 0 ).String(), ARRAYSIZE( szBuf ) );
V_FixSlashes( szBuf, '/' );
if ( pVMTKV->FindKey( pszVMTVar ) )
{
pVMTKV->SetString( pszVMTVar, szBuf );
}
}
}
}
}
CUtlBuffer fileBuf( 0, 0, CUtlBuffer::TEXT_BUFFER );
pVMTKV->RecursiveSaveToFile( fileBuf, 0 );
g_pFullFileSystem->WriteFile( sAbsOutputPaths.Element( i ).String(), "MOD", fileBuf );
Msg( "CTarget%s::Compile OK! - %s\n", GetTypeString(), sRelOutputPaths.Element( i ).String() );
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const char *CTargetVMT::MaterialTypeToString( int nMaterialType )
{
if ( nMaterialType == kInvalidMaterialType )
return "InvalidMaterialType";
const char *pszMaterialName = CItemUpload::Manifest()->GetMaterialType( nMaterialType );
if ( pszMaterialName )
return pszMaterialName;
return "Unknown";
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CTargetVMT::StringToMaterialType( const char *pszUserData )
{
return CItemUpload::Manifest()->GetMaterialType( pszUserData );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetVMT::IsOk( CUtlString &sMsg ) const
{
if ( GetMaterialType() == kInvalidMaterialType )
{
sMsg += "\"";
sMsg += m_sMaterialId;
sMsg += "\": ";
sMsg += "Invalid material type, must be one of Primary or Secondary";
return false;
}
if ( GetDuplicate() )
return true;
if ( !g_bCompilePreview )
{
// Make sure we have all the required textures.
int nSkinIndex = CItemUpload::Manifest()->GetDefaultMaterialSkin();
FOR_EACH_VEC( m_vecTargetVTFs[ nSkinIndex ], i )
{
if ( !m_vecTargetVTFs[nSkinIndex][i].IsValid() && CItemUpload::Manifest()->IsTextureTypeRequired( i ) )
{
sMsg += "\"";
sMsg += m_sMaterialId;
sMsg += "\": ";
sMsg += "Missing required texture type";
Warning( "Missing texture of type '%s'\n", CItemUpload::Manifest()->GetTextureType( i ) );
return false;
}
}
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const ExtensionList *CTargetVMT::GetExtensionsAndCount( void ) const
{
if ( GetDuplicate() )
return NULL;
m_vecExtensions.SetCount( 0 );
FOR_EACH_VEC( m_vecTargetVTFs, i )
{
FOR_EACH_VEC( m_vecTargetVTFs[i], j )
{
if ( m_vecTargetVTFs[i][j].IsValid() )
{
CUtlString sExtension = CItemUpload::Manifest()->GetMaterialSkinFilenameAppend( i );
sExtension += ".vmt";
m_vecExtensions.AddToTail( sExtension );
break;
}
}
}
return &m_vecExtensions;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetVMT::GetInputs( CUtlVector< CTargetBase * > &inputs ) const
{
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - Not Valid: %s\n", GetTypeString(), sTmp.String() );
return false;
}
FOR_EACH_VEC( m_vecTargetVTFs, i )
{
FOR_EACH_VEC( m_vecTargetVTFs[i], j )
{
if ( m_vecTargetVTFs[i][j].IsValid() )
{
inputs.AddToTail( m_vecTargetVTFs[i][j].GetObject() );
}
}
}
return inputs.Count() > 0;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetVMT::GetName( CUtlString &sName ) const
{
sName.Clear();
if ( GetDuplicate() )
{
}
else
{
CTargetBase::GetName( sName );
if ( m_nMaterialType == kInvalidMaterialType )
{
sName += "_INVALID";
}
else
{
if ( m_nMaterialType > 0 )
{
sName += "_";
sName += m_nMaterialType;
}
}
}
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetVMT::SetMaterialId( const char *pszMaterialId )
{
m_sMaterialId = pszMaterialId;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetVMT::SetMaterialType( int nMaterialType )
{
CAsset *pAsset = Asset();
if ( !pAsset )
return false;
if ( nMaterialType < 0 || nMaterialType >= CItemUpload::Manifest()->GetNumMaterialTypes() )
return false;
m_bDuplicate = false;
m_nMaterialType = kInvalidMaterialType;
if ( nMaterialType == kInvalidMaterialType )
return true;
// Recursively increment other non-duplicate materials which match this material
// and so on...
for ( int i = 0; i < pAsset->GetTargetVMTCount(); ++i )
{
CTargetVMT *pTargetVMT = pAsset->GetTargetVMT( i );
if ( !pTargetVMT )
continue;
if ( !pTargetVMT->GetDuplicate() && pTargetVMT->GetMaterialType() == nMaterialType )
{
pTargetVMT->SetMaterialType( ( ( nMaterialType + 1 ) % CItemUpload::Manifest()->GetNumMaterialTypes() ) );
}
}
m_nMaterialType = nMaterialType;
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CTargetVMT::GetMaterialType() const
{
return m_nMaterialType;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetVMT::SetDuplicate( int nMaterialType )
{
if ( nMaterialType < 0 || nMaterialType >= CItemUpload::Manifest()->GetNumMaterialTypes() )
return;
m_bDuplicate = true;
m_nMaterialType = nMaterialType;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetVMT::SetTargetVTF( const char *pszTextureType, const char *pszFilename, int nSkinIndex )
{
// Make sure it's a valid texture type
int nTextureType = CItemUpload::Manifest()->GetTextureType( pszTextureType );
if ( nTextureType == -1 )
{
Warning( "Invalid texture type specified: '%s'\n", pszTextureType );
return false;
}
if ( nSkinIndex == kInvalidMaterialSkin )
{
Warning( "Invalid skin type specified: '%d'\n", nSkinIndex );
return false;
}
CUtlString sSuffix;
sSuffix += CItemUpload::Manifest()->GetMaterialSkinFilenameAppend( nSkinIndex );
sSuffix += pszTextureType;
return SetTargetVTF( m_vecTargetVTFs[nSkinIndex][nTextureType], pszFilename, sSuffix.String() );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetVMT::SetTargetVTF(
CSmartPtr< CTargetVTF > &pTargetVTF,
const char *pszFilename,
const char *pszSuffix ) const
{
if ( !pTargetVTF )
{
pTargetVTF = Asset()->NewTarget< CTargetVTF >( this );
}
if ( !pTargetVTF )
return false;
pTargetVTF->SetNameSuffix( pszSuffix );
bool bResult = pTargetVTF->SetInputFile( pszFilename );
return bResult;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetVMT::SetVMTKV( const KeyValues *pKV, int nSkinIndex /*= 0*/ )
{
if ( !m_pVMTKV )
{
m_pVMTKV = new KeyValues( "data" );
}
const char* pszKeyName = CFmtStr( kVMT, nSkinIndex );
if ( KeyValues *pPreviousKey = m_pVMTKV->FindKey( pszKeyName ) )
{
m_pVMTKV->RemoveSubKey( pPreviousKey );
}
KeyValues *pNewKey = new KeyValues( pszKeyName );
pNewKey->AddSubKey( pKV->MakeCopy() );
m_pVMTKV->AddSubKey( pNewKey );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
KeyValues *CTargetVMT::GetVMTKV( int nSkinIndex )
{
if ( !m_pVMTKV )
{
m_pVMTKV = new KeyValues( "data" );
}
const char* pszKeyName = CFmtStr( kVMT, nSkinIndex );
KeyValues* pKey = m_pVMTKV->FindKey( pszKeyName );
if ( pKey )
{
return pKey->GetFirstSubKey();
}
KeyValues *pVMTKV = new KeyValues("VMTTemplate");
if ( CItemUpload::Manifest()->HasClassVMTTemplates() )
{
const int nClassIndex = GetClassIndex( Asset()->GetClass() );
const char *pszVMTTemplate = CItemUpload::Manifest()->GetClassVMTTemplate( nClassIndex );
if ( !pszVMTTemplate )
{
Warning( "Could not find a VMT template for class '%s'\n", Asset()->GetClass() );
pVMTKV->deleteThis();
return NULL;
}
if ( !pVMTKV->LoadFromFile( g_pFullFileSystem, pszVMTTemplate, "MOD" ) )
{
Warning( "Failed to load specified VMT template '%s' for class '%s'.\n", pszVMTTemplate, Asset()->GetClass() );
pVMTKV->deleteThis();
return NULL;
}
}
else
{
CreateLegacyTemplate( pVMTKV );
}
KeyValues *pNewKey = new KeyValues( pszKeyName );
pNewKey->AddSubKey( pVMTKV );
m_pVMTKV->AddSubKey( pNewKey );
return pVMTKV;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetVMT::CreateLegacyTemplate( KeyValues *pVMTKV )
{
pVMTKV->SetName( "VertexlitGeneric" );
for ( int iVTF = 0; iVTF < CItemUpload::Manifest()->GetNumTextureTypes(); ++iVTF )
{
const char *pszVMTVar = CItemUpload::Manifest()->GetVMTVarForTextureType( CItemUpload::Manifest()->GetTextureType( iVTF ) );
if ( pszVMTVar && *pszVMTVar )
{
pVMTKV->SetString( pszVMTVar, "" );
}
}
// Wearables usually use this lightwarp: "models/player/pyro/pyro_lightwarp"
// Weapons usually use this lightwarp: "models/lightwarps/weapon_lightwarp"
// Weapons are the more custom case, so we'll default to a good wearable lightwarp
pVMTKV->SetString( "$lightwarptexture", "models/player/pyro/pyro_lightwarp" );
pVMTKV->SetString( "$phong", "1" );
pVMTKV->SetString( "$phongexponent", "25" );
pVMTKV->SetString( "$phongboost", "0.1" );
pVMTKV->SetString( "$phongfresnelranges", "[.25 .5 1]" );
pVMTKV->SetString( "$rimlight", "1" ); // To enable rim lighting (requires phong)
pVMTKV->SetString( "$rimlightexponent", "4" ); // Exponent for phong component of rim
pVMTKV->SetString( "$rimlightboost", "2" ); // Boost for ambient cube component of rim lighting
switch ( m_nColorAlphaType )
{
case kNoColorAlpha:
break;
case kTransparency:
pVMTKV->SetString( "$translucent", "1" );
break;
case kPaintable:
pVMTKV->SetString( "$blendtintbybasealpha", "1" );
pVMTKV->SetString( "$blendtintcoloroverbase", "0" ); // between 0-1 determines how much blended by tinting vs. replacing the color
pVMTKV->SetString( "$colortint_base", "{ 255 255 255 }" ); // put the RGB value of whats being colored when no paint is present, if $blendtintcoloroverbase is 0 then put [255 255 255] here.
pVMTKV->SetString( "$color2", "{ 255 255 255 }" );
pVMTKV->SetString( "$colortint_tmp", "[0 0 0]" );
break;
case kColorSpecPhong:
pVMTKV->SetString( "$basemapalphaphongmask", "1" );
break;
}
switch ( m_nNormalAlphaType )
{
case kNoNormalAlpha:
break;
case kNormalSpecPhong:
pVMTKV->SetString( "$bumpmapalphaphongmask", "1" );
break;
}
// Variables for the cloak effect
pVMTKV->SetString( "$cloakPassEnabled", "1" );
// Variables for the burning effect
pVMTKV->SetString( "$detail", "effects/tiledfire/fireLayeredSlowTiled512" );
pVMTKV->SetString( "$detailscale", "5" );
pVMTKV->SetString( "$detailblendfactor", "0" );
pVMTKV->SetString( "$detailblendmode", "6" );
// Variables for the jarate effect
pVMTKV->SetString( "$yellow", "0" );
// The order of the proxies is important!
KeyValues *pProxies = pVMTKV->FindKey( "Proxies", true );
// Proxies for the cloak effect
KeyValues *pProxiesWeaponInvis = pProxies->FindKey( "invis", true );
pProxiesWeaponInvis->FindKey( "temp_key_wont_print_dont_worry", true );
// Proxies for the burning effect
KeyValues *pProxiesAnimatedTexture = pProxies->FindKey( "AnimatedTexture", true );
pProxiesAnimatedTexture->SetString( "animatedtexturevar", "$detail" );
pProxiesAnimatedTexture->SetString( "animatedtextureframenumvar", "$detailframe" );
pProxiesAnimatedTexture->SetString( "animatedtextureframerate", "30" );
KeyValues *pProxiesBurnLevel = pProxies->FindKey( "BurnLevel", true );
pProxiesBurnLevel->SetString( "resultVar", "$detailblendfactor" );
// Proxies for paintable items
if ( m_nColorAlphaType == kPaintable )
{
KeyValues *pProxiesItemTintColor = pProxies->FindKey( "ItemTintColor", true );
pProxiesItemTintColor->SetString( "resultVar", "$colortint_tmp" );
KeyValues *pProxiesSelectFirstIfNonZero = pProxies->FindKey( "SelectFirstIfNonZero", true );
pProxiesSelectFirstIfNonZero->SetString( "srcVar1", "$colortint_tmp" );
pProxiesSelectFirstIfNonZero->SetString( "srcVar2", "$colortint_base" );
pProxiesSelectFirstIfNonZero->SetString( "resultVar", "$color2" );
// Proxies for the jarate effect
KeyValues *pProxiesYellowLevel = pProxies->FindKey( "YellowLevel", true );
pProxiesYellowLevel->SetString( "resultVar", "$yellow" );
KeyValues *pProxiesMultiply = pProxies->FindKey( "Multiply", true );
pProxiesMultiply->SetString( "srcVar1", "$color2" );
pProxiesMultiply->SetString( "srcVar2", "$yellow" );
pProxiesMultiply->SetString( "resultVar", "$color2" );
}
else
{
// Proxies for the jarate effect
KeyValues *pProxiesYellowLevel = pProxies->FindKey( "YellowLevel", true );
pProxiesYellowLevel->SetString( "resultVar", "$yellow" );
KeyValues *pProxiesEquals = pProxies->FindKey( "Equals", true );
pProxiesEquals->SetString( "srcVar1", "$yellow" );
pProxiesEquals->SetString( "resultVar", "$color2" );
}
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetVMT::UpdateManifest( KeyValues *pKV )
{
pKV->SetString( "materialId", m_sMaterialId.String() );
pKV->SetString( "materialType", MaterialTypeToString( GetMaterialType() ) );
pKV->SetInt( "duplicate", GetDuplicate() );
pKV->SetInt( "target_width", m_nTargetWidth );
pKV->SetInt( "target_height", m_nTargetHeight );
if ( !GetDuplicate() )
{
KeyValues *pColorSubKey = new KeyValues( "color" );
pKV->AddSubKey( pColorSubKey );
pColorSubKey->SetString( "alphaType", ColorAlphaTypeToString( GetColorAlphaType() ) );
FOR_EACH_VEC( m_vecTargetVTFs, i )
{
KeyValues *pSkinSubKey;
const char *pszSkinType = CItemUpload::Manifest()->GetMaterialSkin(i);
if ( pszSkinType )
pSkinSubKey = new KeyValues( pszSkinType );
else
pSkinSubKey = pColorSubKey;
FOR_EACH_VEC( m_vecTargetVTFs[i], j )
{
if ( m_vecTargetVTFs[i][j].IsValid() )
{
const char *pszTextureType = CItemUpload::Manifest()->GetTextureType(j);
KeyValues *pSubKey = new KeyValues( pszTextureType );
pSkinSubKey->AddSubKey( pSubKey );
m_vecTargetVTFs[i][j]->UpdateManifest( pSubKey );
}
}
}
}
}
//=============================================================================
//
//=============================================================================
CTargetIcon::CTargetIcon( CAsset *pAsset, int nIconType )
: CTargetVMT( pAsset, pAsset )
, m_nIconType( nIconType )
{
int nWidth, nHeight;
CItemUpload::Manifest()->GetIconDimensions( m_nIconType, nWidth, nHeight );
SetTargetResolution( nWidth, nHeight );
const char *pszSuffix = CItemUpload::Manifest()->GetIconFilenameAppend( m_nIconType );
SetNameSuffix( pszSuffix );
KeyValues *pVMTKV = CItemUpload::Manifest()->GetIconVMTTemplate( m_nIconType );
if ( pVMTKV )
{
SetVMTKV( pVMTKV );
}
SetMaterialType( CItemUpload::Manifest()->GetDefaultMaterialType() );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetIcon::SetTargetVTF( const char *pszFilename )
{
return CTargetVMT::SetTargetVTF( m_vecTargetVTFs[0][0], pszFilename, "" );
}
//=============================================================================
//
//=============================================================================
CTargetDMX::CTargetDMX( CAsset *pAsset, const CTargetQC *pTargetQC )
: CTargetBase( pAsset, pTargetQC )
, m_pDmRoot( NULL )
, m_nLod( -1 )
, m_flAnimationLoopStartTime( -1.f )
{
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CTargetDMX::~CTargetDMX()
{
Clear();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetDMX::IsOk( CUtlString &sMsg ) const
{
if ( m_nLod < 0 )
{
sMsg = "Invalid LOD (";
sMsg += m_nLod;
sMsg += ")";
return false;
}
if ( m_sInputFile.IsEmpty() || m_sExtension.IsEmpty() )
{
sMsg = "No input file specified";
return false;
}
if ( !CItemUpload::FileExists( m_sInputFile.String() ) )
{
sMsg = "Input file: \"";
sMsg += m_sInputFile;
sMsg += "\" doesn't exist";
return false;
}
if ( !( IsInputSmd() || IsInputObj() || IsInputDmx() || IsInputFbx() ) )
{
sMsg = "Input file: \"";
sMsg += m_sInputFile;
sMsg += "\" is not a DMX, OBJ or SMD file";
return false;
}
if ( !m_pDmRoot )
{
sMsg = "Input file: \"";
sMsg += m_sInputFile;
sMsg += "\" couldn't be read";
return false;
}
const int nPolyCount = GetPolyCount();
if ( nPolyCount <= 0 && m_strQCITemplate.IsEmpty() )
{
sMsg = "Invalid polygon count (";
sMsg += nPolyCount;
sMsg += ")";
return false;
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetDMX::Compile()
{
if ( !ReloadFile() )
return false;
if ( !CTargetBase::Compile() )
return false;
CUtlString sName;
if ( !GetOutputPath( sName, 0, PATH_FLAG_FILE | PATH_FLAG_EXTENSION ) )
return false;
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( "CTarget%s::Compile( %s ) - Not Valid: %s\n", GetTypeString(), sName.String(), sTmp.String() );
return false;
}
CUtlString sAbsPath;
GetOutputPath( sAbsPath, 0 );
CUtlString sRelPath;
GetOutputPath( sRelPath, 0, PATH_FLAG_ALL & ~PATH_FLAG_ABSOLUTE );
if ( sAbsPath.IsEmpty() || sRelPath.IsEmpty() )
{
Warning( "CTarget%s::Compile( %s ) - GetOutputPath failed\n", GetTypeString(), sName.String() );
return false;
}
Msg( "Compiling %s: %s\n", GetTypeString(), sRelPath.String() );
AddOrEditP4File( sAbsPath.String() );
CUndoScopeGuard undo( "replaceMaterial" );
ReplaceMaterials();
SkinToBipHead();
#ifdef _DEBUG
const bool bRet = g_pDataModel->SaveToFile( sAbsPath.String(), NULL, "keyvalues2", "model", m_pDmRoot );
#else // ifdef _DEBUG
const bool bRet = g_pDataModel->SaveToFile( sAbsPath.String(), NULL, "binary", "model", m_pDmRoot );
#endif // _DEBUG
undo.Abort();
if ( !bRet )
{
Msg( "CTarget%s::Compile( %s ) - SaveToFile failed - \"%s\"\n", GetTypeString(), sName.String(), sAbsPath.String() );
return false;
}
if ( !CheckFile( sAbsPath.String() ) )
{
Warning( "CTarget%s::Compile( %s ) - File Check Failed - \"%s\"\n", GetTypeString(), sName.String(), sAbsPath.String() );
return false;
}
if ( !OutputQCIFile() )
{
return false;
}
if ( CItemUpload::Manifest()->UseTerseMessages() )
{
Msg( " - Compilation successful.\n" );
}
else
{
Msg( "CTarget%s::Compile OK! - %s\n", GetTypeString(), sRelPath.String() );
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetDMX::GetInputPaths(
CUtlVector< CUtlString > &sInputPaths,
bool bRelative /* = true */,
bool bRecurse /* = true */,
bool bExtension /* = true */ )
{
sInputPaths.AddToTail( m_sInputFile );
return CTargetBase::GetInputPaths( sInputPaths, bRelative, bRecurse, bExtension );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const ExtensionList *CTargetDMX::GetExtensionsAndCount( void ) const
{
static ExtensionList vecExtensions;
vecExtensions.RemoveAll();
vecExtensions.AddToTail( ".dmx" );
if ( !m_strQCITemplate.IsEmpty() )
{
vecExtensions.AddToTail( ".qci" );
}
return &vecExtensions;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetDMX::GetName( CUtlString &sName ) const
{
CTargetBase::GetName( sName );
if ( GetLod() > 0 )
{
sName += "_lod";
sName += GetLod();
}
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CTargetDMX::GetPolyCount() const
{
if ( !m_pDmRoot )
return 0;
int nPolyCount = 0;
const DmFileId_t hRootFileId = m_pDmRoot->GetFileId();
for ( DmElementHandle_t hElement = g_pDataModel->FirstAllocatedElement(); hElement != DMELEMENT_HANDLE_INVALID; hElement = g_pDataModel->NextAllocatedElement( hElement ) )
{
if ( hElement == DMELEMENT_HANDLE_INVALID )
continue;
CDmeMesh *pDmeMesh = CastElement< CDmeMesh >( g_pDataModel->GetElement( hElement ) );
if ( !pDmeMesh || pDmeMesh->GetFileId() != hRootFileId )
continue;
for ( int i = 0; i < pDmeMesh->FaceSetCount(); ++i )
{
CDmeFaceSet *pDmeFaceSet = pDmeMesh->GetFaceSet( i );
if ( !pDmeFaceSet )
continue;
nPolyCount += pDmeFaceSet->GetFaceCount();
}
}
return nPolyCount;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CTargetDMX::GetTriangleCount() const
{
if ( !m_pDmRoot )
return 0;
int nTriangleCount = 0;
const DmFileId_t hRootFileId = m_pDmRoot->GetFileId();
for ( DmElementHandle_t hElement = g_pDataModel->FirstAllocatedElement(); hElement != DMELEMENT_HANDLE_INVALID; hElement = g_pDataModel->NextAllocatedElement( hElement ) )
{
if ( hElement == DMELEMENT_HANDLE_INVALID )
continue;
CDmeMesh *pDmeMesh = CastElement< CDmeMesh >( g_pDataModel->GetElement( hElement ) );
if ( !pDmeMesh || pDmeMesh->GetFileId() != hRootFileId )
continue;
for ( int i = 0; i < pDmeMesh->FaceSetCount(); ++i )
{
CDmeFaceSet *pDmeFaceSet = pDmeMesh->GetFaceSet( i );
if ( !pDmeFaceSet )
continue;
nTriangleCount += pDmeFaceSet->GetTriangulatedIndexCount() / 3;
}
}
return nTriangleCount;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CTargetDMX::GetVertexCount() const
{
if ( !m_pDmRoot )
return 0;
int nVertexCount = 0;
const DmFileId_t hRootFileId = m_pDmRoot->GetFileId();
for ( DmElementHandle_t hElement = g_pDataModel->FirstAllocatedElement(); hElement != DMELEMENT_HANDLE_INVALID; hElement = g_pDataModel->NextAllocatedElement( hElement ) )
{
if ( hElement == DMELEMENT_HANDLE_INVALID )
continue;
CDmeMesh *pDmeMesh = CastElement< CDmeMesh >( g_pDataModel->GetElement( hElement ) );
if ( !pDmeMesh || pDmeMesh->GetFileId() != hRootFileId )
continue;
CDmeVertexData *pVertexData = pDmeMesh->GetBindBaseState();
if ( !pVertexData )
continue;
nVertexCount += pVertexData->VertexCount();
}
return nVertexCount;
}
bool CTargetDMX::GetAnimationFrameInfo( float& flFrameRate, int& nFrameCount ) const
{
if ( !m_pDmRoot )
return false;
CDmeAnimationList *pDmeAnimationList = m_pDmRoot->GetValueElement< CDmeAnimationList >( "animationList" );
CDmeChannelsClip *pDmeChannelsClip = NULL;
if ( pDmeAnimationList )
{
if ( pDmeAnimationList->GetAnimationCount() < 0 )
{
Warning( "No DmeChannelsClips found in root.animationList of DMX file: \"%s\"\n", m_sInputFile.String() );
return false;
}
pDmeChannelsClip = pDmeAnimationList->GetAnimation( 0 );
if ( !pDmeChannelsClip )
{
Warning( "No DmeChannelsClip found in root.animationList[0] of DMX file: \"%s\"\n", m_sInputFile.String() );
return false;
}
flFrameRate = pDmeChannelsClip->GetValue< int >( "frameRate", 30 );
nFrameCount = (int)( ( pDmeChannelsClip->GetDuration().GetSeconds() * flFrameRate ) + 0.5f ) + 1;
return true;
}
return false;
}
//-----------------------------------------------------------------------------
// Returns true if all meshes in the DMX data have skinning information
// false if any mesh doesn't have skinning data
//-----------------------------------------------------------------------------
bool CTargetDMX::AreAllMeshesSkinned() const
{
if ( !m_pDmRoot )
return false; // No DMX data
const DmFileId_t hRootFileId = m_pDmRoot->GetFileId();
int nMeshCount = 0;
for ( DmElementHandle_t hElement = g_pDataModel->FirstAllocatedElement(); hElement != DMELEMENT_HANDLE_INVALID; hElement = g_pDataModel->NextAllocatedElement( hElement ) )
{
if ( hElement == DMELEMENT_HANDLE_INVALID )
continue;
CDmeMesh *pDmeMesh = CastElement< CDmeMesh >( g_pDataModel->GetElement( hElement ) );
if ( !pDmeMesh || pDmeMesh->GetFileId() != hRootFileId )
continue;
CDmeVertexData *pDmeVertexData = pDmeMesh->GetBindBaseState();
if ( !pDmeVertexData )
continue;
// If even one mesh doesn't have skinning data, abort
if ( !pDmeVertexData->HasSkinningData() )
return false;
}
// Make sure we saw at least one mesh
return ( nMeshCount > 0 );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const CUtlString &CTargetDMX::GetInputFile() const
{
return m_sInputFile;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetDMX::Clear()
{
if ( m_pDmRoot )
{
CDisableUndoScopeGuard noUndo;
m_targetVmtList.RemoveAll();
g_pDataModel->UnloadFile( m_pDmRoot->GetFileId() );
m_pDmRoot = NULL;
}
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetDMX::SetInputFile( const char *pszFilename )
{
// Because of the way datamodel works, bad things will happen if the same file is loaded twice
Clear();
m_sInputFile = pszFilename;
char szBuf[ k64KB ];
V_ExtractFileExtension( pszFilename, szBuf, ARRAYSIZE( szBuf ) );
m_sExtension = szBuf;
DmFileId_t hRootFileId = g_pDataModel->GetFileId( pszFilename );
DmElementHandle_t hRootElement = g_pDataModel->GetFileRoot( hRootFileId );
CDmElement *pDmRoot = g_pDataModel->GetElement( hRootElement );
if ( !pDmRoot )
{
pDmRoot = LoadDmx();
}
if ( !pDmRoot )
{
pDmRoot = LoadFbx();
}
if ( !pDmRoot )
{
pDmRoot = LoadSmd();
}
if ( !pDmRoot )
{
pDmRoot = LoadObj();
}
if ( !pDmRoot )
{
Warning( "ERROR: Don't Know What Type Of Input File Is\n" );
return false;
}
m_pDmRoot = pDmRoot;
// TODO: Change to walking from root?
hRootFileId = m_pDmRoot->GetFileId();
// we must store the root handle so we don't reload the same file multiple time
g_pDataModel->SetFileRoot( hRootFileId, m_pDmRoot->GetHandle() );
int nMaterialType = 0;
for ( DmElementHandle_t hElement = g_pDataModel->FirstAllocatedElement(); hElement != DMELEMENT_HANDLE_INVALID; hElement = g_pDataModel->NextAllocatedElement( hElement ) )
{
if ( hElement == DMELEMENT_HANDLE_INVALID )
continue;
CDmeMaterial *pDmeMaterial = CastElement< CDmeMaterial >( g_pDataModel->GetElement( hElement ) );
if ( !pDmeMaterial || pDmeMaterial->GetFileId() != hRootFileId )
continue;
CSmartPtr< CTargetVMT > pTargetVMT = Asset()->FindOrAddMaterial( pDmeMaterial->GetMaterialName(), nMaterialType );
if ( pTargetVMT.IsValid() )
{
m_targetVmtList.AddToTail( pTargetVMT );
++nMaterialType;
}
}
CUtlString sTmp;
if ( IsOk( sTmp ) )
return true;
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - Not Valid: %s\n", GetTypeString(), sTmp.String() );
return false;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetDMX::ReloadFile()
{
// Keep references to all target VMT's to prevent them from being destroyed by a reload of the same file
CUtlVector< CSmartPtr< CTargetVMT > > tmpVmtCopy;
tmpVmtCopy = m_targetVmtList;
// 'reset' the input file from the current value to force a reload on compile, make a temp copy as it's going to be overwritten
if ( !SetInputFile( CUtlString( GetInputFile() ).String() ) )
return false;
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetDMX::IsInputObj() const
{
// TODO: Perhaps look at magic in the start of the file
return !V_stricmp( "obj", m_sExtension.String() );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetDMX::IsInputSmd() const
{
// TODO: Perhaps look at magic in the start of the file
return !V_stricmp( "smd", m_sExtension.String() );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetDMX::IsInputDmx() const
{
// TODO: Perhaps look at magic in the start of the file
return !V_stricmp( "dmx", m_sExtension.String() );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetDMX::IsInputFbx() const
{
// TODO: Perhaps look at magic in the start of the file
return !V_stricmp( "fbx", m_sExtension.String() );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CDmElement *CTargetDMX::LoadObj()
{
if ( !IsInputObj() )
return NULL;
CDisableUndoScopeGuard noUndo;
CDmObjSerializer dmObjSerializer;
return dmObjSerializer.ReadOBJ( m_sInputFile.String() );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CDmElement *CTargetDMX::LoadSmd()
{
if ( !IsInputSmd() )
return NULL;
CDisableUndoScopeGuard noUndo;
CDmSmdSerializer dmSmdSerializer;
dmSmdSerializer.SetIsAnimation( !m_strQCITemplate.IsEmpty() );
return dmSmdSerializer.ReadSMD( m_sInputFile.String() );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CDmElement *CTargetDMX::LoadDmx()
{
if ( !IsInputDmx() )
return NULL;
CDisableUndoScopeGuard noUndo;
CDmElement *pDmRoot = NULL;
g_pDataModel->RestoreFromFile( m_sInputFile.String(), NULL, NULL, &pDmRoot );
return pDmRoot;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CDmElement *CTargetDMX::LoadFbx()
{
if ( !IsInputFbx() )
return NULL;
CDisableUndoScopeGuard noUndo;
CDmFbxSerializer dmFbxSerializer;
return dmFbxSerializer.ReadFBX( m_sInputFile.String() );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetDMX::ReplaceMaterials() const
{
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - Not Valid: %s\n", GetTypeString(), sTmp.String() );
return;
}
DmFileId_t hRootFileId = m_pDmRoot->GetFileId();
if ( hRootFileId == DMFILEID_INVALID )
return;
CAsset *pAsset = Asset();
if ( !pAsset )
return;
CUtlMap< CUtlString, CUtlString > materialMap( UtlStringLessThan );
CUtlString sRelativeDir;
pAsset->GetRelativeDir( sRelativeDir, NULL, this );
char szBuf[ k64KB ];
for ( int i = 0; i < pAsset->GetTargetVMTCount(); ++i )
{
CTargetVMT *pTargetVMT = pAsset->GetTargetVMT( i );
if ( !pTargetVMT )
continue;
CUtlString sMaterialId;
pTargetVMT->GetMaterialId( sMaterialId );
if ( sMaterialId.IsEmpty() || materialMap.IsValidIndex( materialMap.Find( sMaterialId ) ) )
continue;
CTargetVMT *pSrcTargetVMT = pTargetVMT;
if ( pTargetVMT->GetDuplicate() )
{
const int nMaterialType = pTargetVMT->GetMaterialType();
if ( nMaterialType < 0 || nMaterialType >= CItemUpload::Manifest()->GetNumMaterialTypes() )
{
Warning( "CTarget%s::ReplaceMaterials - Duplicate VMT (%s) with bad material type (%d)\n", GetTypeString(), sMaterialId.String(), nMaterialType );
continue;
}
for ( int j = 0; j < pAsset->GetTargetVMTCount(); ++j )
{
CTargetVMT *pTmpTargetVMT = pAsset->GetTargetVMT( j );
if ( !pTmpTargetVMT || pTmpTargetVMT->GetDuplicate() )
continue;
if ( pTmpTargetVMT->GetMaterialType() == nMaterialType )
{
pSrcTargetVMT = pTmpTargetVMT;
break;
}
}
if ( pTargetVMT == pSrcTargetVMT )
{
Warning( "CTarget%s::ReplaceMaterials - Duplicate VMT (%s) Cannot Find Source Material Type (%d)\n", GetTypeString(), sMaterialId.String(), nMaterialType );
continue;
}
}
sMaterialId.FixSlashes( '/' );
// The full extension can be _red.vmt or _blue.vmt so we need to keep the _red but not the .vmt
CUtlString sMaterialPath;
pSrcTargetVMT->GetOutputPath( sMaterialPath, 0, PATH_FLAG_FILE | PATH_FLAG_EXTENSION | PATH_FLAG_PATH | PATH_FLAG_MODELS );
V_StripExtension( sMaterialPath, szBuf, ARRAYSIZE( szBuf ) );
sMaterialPath.FixSlashes( '/' );
sMaterialPath = szBuf;
materialMap.InsertOrReplace( sMaterialId, sMaterialPath.String() );
}
for ( DmElementHandle_t hElement = g_pDataModel->FirstAllocatedElement(); hElement != DMELEMENT_HANDLE_INVALID; hElement = g_pDataModel->NextAllocatedElement( hElement ) )
{
if ( hElement == DMELEMENT_HANDLE_INVALID )
continue;
CDmeMesh *pDmeMesh = CastElement< CDmeMesh >( g_pDataModel->GetElement( hElement ) );
if ( !pDmeMesh || pDmeMesh->GetFileId() != hRootFileId )
continue;
FOR_EACH_MAP_FAST( materialMap, mmi )
{
pDmeMesh->ReplaceMaterial( materialMap.Key( mmi ), materialMap.Element( mmi ) );
}
}
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetDMX::SkinToBipHead()
{
CAssetTF *pAssetTF = dynamic_cast< CAssetTF * >( Asset() );
if ( !pAssetTF )
return;
if ( !pAssetTF->SkinToBipHead() )
return;
CDisableUndoScopeGuard undoGuard;
CDmeModel *pSrcModel = m_pDmRoot->GetValueElement< CDmeModel >( "model" );
if ( !pSrcModel )
return;
const DmFileId_t nFileId = m_pDmRoot->GetFileId();
CDmElement *pDstRoot = CreateElement< CDmElement >( m_pDmRoot->GetName(), nFileId );
CDmeModel *pDstModel = CreateElement< CDmeModel >( pSrcModel->GetName(), nFileId );
pDstModel->GetTransform()->SetName( pDstModel->GetName() );
pDstRoot->SetValue( "model", pDstModel );
pDstRoot->SetValue( "skeleton", pDstModel );
CDmeJoint *pBipHead = CreateElement< CDmeJoint >( "bip_head", nFileId );
pBipHead->GetTransform()->SetName( "bip_head" );
pDstModel->AddChild( pBipHead );
pDstModel->AddJoint( pBipHead );
const Vector &vBipHead = pAssetTF->GetBipHead();
const RadianEuler &eBipHead = pAssetTF->GetBipHeadRotation();
const Quaternion qBipHead = eBipHead;
pBipHead->GetTransform()->SetOrientation( qBipHead );
CUtlStack< CDmeDag * > depthFirstStack;
depthFirstStack.Push( pSrcModel );
while ( depthFirstStack.Count() )
{
CDmeDag *pDmeDag = NULL;
depthFirstStack.Pop( pDmeDag );
if ( !pDmeDag )
continue;
SkinToBipHead_R( pDstModel, pDmeDag, vBipHead );
for ( int i = pDmeDag->GetChildCount() - 1; i >= 0; --i )
{
depthFirstStack.Push( pDmeDag->GetChild( i ) );
}
}
pDstModel->CaptureJointsToBaseState( "bind" );
DestroyElement( m_pDmRoot, TD_ALL );
m_pDmRoot = pDstRoot;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetDMX::SkinToBipHead_R( CDmeModel *pDstModel, CDmeDag *pSrcDag, const Vector &vBipHead )
{
if ( !pDstModel || !pSrcDag )
return;
CDmeMesh *pSrcMesh = CastElement< CDmeMesh >( pSrcDag->GetShape() );
if ( !pSrcMesh )
return;
CDmeVertexData *pSrcBind = pSrcMesh->GetBindBaseState();
if ( !pSrcBind )
return;
CDmeDag *pDstDag = CreateElement< CDmeDag >( pSrcDag->GetName(), pDstModel->GetFileId() );
if ( !pDstDag )
return;
pDstDag->GetTransform()->SetName( pDstDag->GetName() );
CDmeMesh *pDstMesh = pSrcMesh->Copy();
if ( !pDstMesh )
return;
// Current state is left empty, not sure why
pDstMesh->SetBindBaseState( pDstMesh->GetCurrentBaseState() );
pDstMesh->Resolve();
pDstDag->SetShape( pDstMesh );
pDstModel->AddChild( pDstDag );
matrix3x4_t mIdentity; // TODO: Is there a static identity matrix like quat_identity?
SetIdentityMatrix( mIdentity );
matrix3x4_t mSrcAbs;
// pSrcDag->GetAbsTransform( mSrcAbs );
{
matrix3x4_t parentToWorld;
pSrcDag->GetParentWorldMatrix( parentToWorld );
matrix3x4_t localMatrix;
pSrcDag->GetLocalMatrix( localMatrix );
ConcatTransforms( parentToWorld, localMatrix, mSrcAbs );
}
matrix3x4_t mNormal;
MatrixInverseTranspose( mSrcAbs, mNormal );
bool bConvertToWorldSpace = !MatricesAreEqual( mIdentity, mSrcAbs );
for ( int i = 0; i < pDstMesh->BaseStateCount(); ++i )
{
CDmeVertexData *pDstVertexData = pDstMesh->GetBaseState( i );
if ( !pDstVertexData )
continue;
pDstVertexData->Resolve();
// Remove all skinning information
// This seems a little odd, creating data to remove it but if it already exists this is
// the only way of setting the joint count to 0 without cheating (i.e. referring to the field by name)
FieldIndex_t nJointWeightField;
FieldIndex_t nJointIndexField;
pDstVertexData->CreateJointWeightsAndIndices( 0, &nJointWeightField, &nJointIndexField );
pDstVertexData->RemoveAttribute( pDstVertexData->FieldName( nJointWeightField ) );
pDstVertexData->RemoveAttribute( pDstVertexData->FieldName( nJointIndexField ) );
// Convert data to world space if required
{
Vector vTmp;
static const int nZero = 0;
static const float flOne = 1;
FieldIndex_t nPositionField = pDstVertexData->FindFieldIndex( CDmeVertexData::FIELD_POSITION );
if ( nPositionField >= 0 )
{
CDmAttribute *pPosAttr = pDstVertexData->GetVertexData( nPositionField );
if ( pPosAttr )
{
CDmrArray< Vector > posData = pPosAttr;
// Set everything up to be skinned to joint 0
pDstVertexData->CreateJointWeightsAndIndices( 1, &nJointWeightField, &nJointIndexField );
pDstVertexData->AddVertexData( nJointIndexField, posData.Count() );
pDstVertexData->AddVertexData( nJointWeightField, posData.Count() );
if ( bConvertToWorldSpace )
{
for ( int iPos = 0; iPos < posData.Count(); ++iPos )
{
pDstVertexData->SetVertexData( nJointIndexField, iPos, 1, AT_INT, &nZero );
pDstVertexData->SetVertexData( nJointWeightField, iPos, 1, AT_FLOAT, &flOne );
}
}
else
{
for ( int iPos = 0; iPos < posData.Count(); ++iPos )
{
VectorTransform( posData[iPos], mSrcAbs, vTmp );
posData.Set( iPos, vTmp );
pDstVertexData->SetVertexData( nJointIndexField, i, 1, AT_INT, &nZero );
pDstVertexData->SetVertexData( nJointWeightField, i, 1, AT_FLOAT, &flOne );
}
}
}
}
FieldIndex_t nNormalField = pDstVertexData->FindFieldIndex( CDmeVertexData::FIELD_NORMAL );
if ( nNormalField >= 0 )
{
CDmAttribute *pNormalAttr = pDstVertexData->GetVertexData( nNormalField );
if ( pNormalAttr )
{
CDmrArray< Vector > normalData = pNormalAttr;
for ( int iData = 0; iData < normalData.Count(); ++iData )
{
VectorRotate( normalData[iData], mNormal, vTmp );
vTmp.NormalizeInPlace();
normalData.Set( iData, vTmp );
}
}
}
}
pDstVertexData->Resolve();
}
Vector vMin;
Vector vMax;
pDstMesh->GetBoundingBox( vMin, vMax );
const Vector vCenter = ( vMax - vMin ) / 2.0 + vMin;
const float flSqDistOrigin = vCenter.DistToSqr( Vector( 0, 0, 0 ) );
const float flSqDistBipHead = vCenter.DistToSqr( vBipHead );
// See where the model was modelled... around the bip_head bone or at the origin?
// There are probably other problems which aren't accounted for this is a really naive
// algorithm
if ( flSqDistBipHead < flSqDistOrigin )
{
// Model centered around bip head, subtract vBipHead from position data
Vector vTmp;
for ( int i = 0; i < pDstMesh->BaseStateCount(); ++i )
{
CDmeVertexData *pDstVertexData = pDstMesh->GetBaseState( i );
if ( !pDstVertexData )
continue;
pDstVertexData->Resolve();
FieldIndex_t nPositionField = pDstVertexData->FindFieldIndex( CDmeVertexData::FIELD_POSITION );
if ( nPositionField >= 0 )
{
CDmAttribute *pPosAttr = pDstVertexData->GetVertexData( nPositionField );
if ( pPosAttr )
{
CDmrArray< Vector > posData = pPosAttr;
for ( int iPos = 0; iPos < posData.Count(); ++iPos )
{
VectorSubtract( posData[iPos], vBipHead, vTmp );
posData.Set( iPos, vTmp );
}
}
}
pDstVertexData->Resolve();
}
}
}
//=============================================================================
//
//=============================================================================
bool CTargetDMX::OutputQCIFile()
{
// Should we output QCI?
if ( !m_strQCITemplate.IsEmpty() )
{
CUtlString sDMXName;
if ( !GetOutputPath( sDMXName, 0, PATH_FLAG_FILE | PATH_FLAG_EXTENSION ) )
return false;
CUtlString sName;
if ( !GetOutputPath( sName, 1, PATH_FLAG_FILE ) )
return false;
CUtlString sAbsPath;
if ( !GetOutputPath( sAbsPath, 1 ) )
return false;
float flFrameRate;
int nFrameCount;
if ( !GetAnimationFrameInfo( flFrameRate, nFrameCount ) )
{
Warning( "Failed to get animation's frame info for %s\n", sAbsPath.String() );
return false;
}
const int nBlendFrameOffset = 5;
const int nLastFrameIndex = nFrameCount - 1;
CUtlString strLayerName = "layer_";
strLayerName += sName;
CUtlString strBuf = m_strQCITemplate.Replace( "<ITEMTEST_SEQUENCE_LAYER_NAME>", strLayerName.String() );
strBuf = strBuf.Replace( "<ITEMTEST_SEQUENCE_NAME>", sName.String() );
strBuf = strBuf.Replace( "<ITEMTEST_DMX_FILE>", sName.String() );
strBuf = strBuf.Replace( "<FPS>", CFmtStr( "fps %d", (int)flFrameRate ) );
strBuf = strBuf.Replace( "<BLEND_LAYER_RANGE>", CFmtStr( "0 %d %d <LAST_FRAME_INDEX>", nBlendFrameOffset, nLastFrameIndex - nBlendFrameOffset ) );
strBuf = strBuf.Replace( "<FRAME_COUNT>", CFmtStr( "%d", nFrameCount ) );
strBuf = strBuf.Replace( "<LAST_FRAME_INDEX>", CFmtStr( "%d", nLastFrameIndex ) );
CUtlBuffer bufQCFile( 0, 0, CUtlBuffer::TEXT_BUFFER );
bufQCFile.PutString( strBuf.String() );
AddOrEditP4File( sAbsPath.String() );
g_pFullFileSystem->WriteFile( sAbsPath.String(), "MOD", bufQCFile );
}
return true;
}
void LeftRightToValueBalance( float flLeft, float flRight, float &flOutValue, float &flOutBalance )
{
if ( flLeft > flRight )
{
flOutValue = flLeft;
flOutBalance = 0.5f * flRight / flLeft;
}
else
{
flOutValue = flRight;
flOutBalance = 1.0f - 0.5f * flRight / flLeft;
}
}
void BufPrintf( CUtlBuffer& buf, int nLevel, const char *fmt, ... )
{
va_list argptr;
va_start( argptr, fmt );
while ( nLevel-- > 0 )
{
buf.Printf( " " );
}
buf.VaPrintf( fmt, argptr );
va_end( argptr );
}
void BufBeginBlock( CUtlBuffer& buf, int &nLevel )
{
BufPrintf( buf, nLevel, "{\n" );
++nLevel;
}
void BufEndBlock( CUtlBuffer& buf, int &nLevel )
{
--nLevel;
BufPrintf( buf, nLevel, "}\n" );
}
template< class T >
CDmeTypedLogLayer< T > *GetTopmostLogLayer( CDmeChannel *pChannel )
{
if ( !pChannel )
return NULL;
CDmeTypedLog< T > *pLog = CastElement< CDmeTypedLog< T > >( pChannel->GetLog() );
if ( !pLog )
return NULL;
return pLog->GetLayer( pLog->GetTopmostLayer() );
}
void CTargetDMX::OutputSounds( CUtlBuffer &buf, int nIndentLevel, CDmElement *pExportedSounds )
{
if ( !pExportedSounds )
return;
CDmrElementArray< CDmElement > soundsArray( pExportedSounds, "sounds" );
if ( !soundsArray.IsValid() )
return;
int nSounds = soundsArray.Count();
if ( nSounds == 0 )
return;
CUtlBuffer soundScriptBuffer( 0, 0, CUtlBuffer::TEXT_BUFFER );
if ( !m_strSoundScriptFile.IsEmpty() )
{
if ( !g_pFullFileSystem->ReadFile( m_strSoundScriptFile.String(), "MOD", soundScriptBuffer ) )
{
Warning( "Failed to load sound script file\n" );
return;
}
}
else
{
Warning( "Sound script file not specified\n" );
return;
}
BufPrintf( buf, nIndentLevel, "channel \"sounds\"\n" );
BufBeginBlock( buf, nIndentLevel );
{
for ( int i = 0; i < nSounds; ++i )
{
CDmElement *pSound = soundsArray[ i ];
if ( !pSound )
continue;
float flStartTime = pSound->GetValue< float >( "startTime" );
CUtlString strWaveName = pSound->GetValueString( "fileName" );
V_FixSlashes( strWaveName.GetForModify(), '/' );
if ( !g_pFullFileSystem->FileExists( CFmtStr( "sound/%s", strWaveName.String() ), "GAME" ) )
{
CUtlString strSoundName = strWaveName.StripExtension();
if ( !g_pFullFileSystem->FileExists( CFmtStr( "sound/%s.mp3", strSoundName.String() ), "GAME" ) )
{
Warning( "OutputSounds to VCD missing sound file: %s.wav/mp3\n", strSoundName.String() );
continue;
}
}
bool bIsVO = V_strstr( strWaveName.String(), "vo/" ) != NULL;
const char *pszSoundScriptName = CFmtStr( "<CLASS_NAME>_<ITEMTEST_SEQUENCE_NAME>_%s", pSound->GetName() );
// write sound script entry to sound script file
int nSoundIndentLevel = 0;
BufPrintf( soundScriptBuffer, nSoundIndentLevel, "\"%s\"\n", pszSoundScriptName );
BufBeginBlock( soundScriptBuffer, nSoundIndentLevel );
{
BufPrintf( soundScriptBuffer, nSoundIndentLevel, "\"channel\" \"%s\"\n", bIsVO ? "CHAN_VOICE" : "CHAN_STATIC" );
BufPrintf( soundScriptBuffer, nSoundIndentLevel, "\"volume\" \"VOL_NORM\"\n" );
BufPrintf( soundScriptBuffer, nSoundIndentLevel, "\"pitch\" \"PITCH_NORM\"\n" );
BufPrintf( soundScriptBuffer, nSoundIndentLevel, "\"soundlevel\" \"SNDLVL_95dB\"\n" );
BufPrintf( soundScriptBuffer, nSoundIndentLevel, "\"wave\" \"%s\"\n", strWaveName.String() );
}
BufEndBlock( soundScriptBuffer, nSoundIndentLevel );
// write vcd speak event
BufPrintf( buf, nIndentLevel, "event speak \"%s\"\n", pszSoundScriptName );
BufBeginBlock( buf, nIndentLevel );
{
BufPrintf( buf, nIndentLevel, "time %f <MAX_SEQUENCE_LENGTH>\n", flStartTime );
BufPrintf( buf, nIndentLevel, "param \"%s\"", pszSoundScriptName );
BufPrintf( buf, nIndentLevel, "fixedlength\n" );
BufPrintf( buf, nIndentLevel, "cctpe \"cc_master\"\n" );
BufPrintf( buf, nIndentLevel, "cctoken \"\"\n" );
}
BufEndBlock( buf, nIndentLevel );
}
}
BufEndBlock( buf, nIndentLevel );
CUtlString strBuf = soundScriptBuffer.String();
FOR_EACH_SUBKEY( GetCustomKeyValues(), pSubKey )
{
strBuf = strBuf.Replace( pSubKey->GetName(), pSubKey->GetString() );
}
soundScriptBuffer.Clear();
soundScriptBuffer.PutString( strBuf.String() );
// write to sound script file
CUtlString sWorkingDir;
CItemUpload::GetVProjectDir( sWorkingDir );
char szSoundScriptFullPath[MAX_PATH];
V_MakeAbsolutePath( szSoundScriptFullPath, sizeof( szSoundScriptFullPath ), m_strSoundScriptFile.String(), sWorkingDir.String() );
if ( !g_pFullFileSystem->WriteFile( szSoundScriptFullPath, NULL, soundScriptBuffer ) )
{
Warning( "Failed to output soundscript: %s\n", szSoundScriptFullPath );
}
if ( GetCustomModPath() )
{
char szCustomPath[MAX_PATH];
V_MakeAbsolutePath( szCustomPath, sizeof( szCustomPath ), CFmtStr( "%s/%s", GetCustomModPath(), m_strSoundScriptFile.String() ), sWorkingDir.String() );
V_FixSlashes( szCustomPath );
if ( DoFileCopy( szSoundScriptFullPath, szCustomPath ) )
{
Asset()->AddModOutput( szCustomPath );
}
}
}
// pChannelsClip is assumed to be an exported channelsclip (vs a live channelsclip)
bool CTargetDMX::WriteVCD( CUtlBuffer &vcdBuf )
{
CDmeAnimationList *pDmeAnimationList = m_pDmRoot->GetValueElement< CDmeAnimationList >( "animationList" );
CDmeChannelsClip *pDmeChannelsClip = NULL;
if ( pDmeAnimationList )
{
if ( pDmeAnimationList->GetAnimationCount() < 0 )
{
Warning( "No DmeChannelsClips found in root.animationList of DMX file: \"%s\"\n", m_sInputFile.String() );
return false;
}
pDmeChannelsClip = pDmeAnimationList->GetAnimation( 0 );
if ( !pDmeChannelsClip )
{
Warning( "No DmeChannelsClip found in root.animationList[0] of DMX file: \"%s\"\n", m_sInputFile.String() );
return false;
}
}
DmeFramerate_t framerate( pDmeChannelsClip->GetValue< int >( "frameRate", 30 ) );
int nLevel = 0;
BufPrintf( vcdBuf, nLevel, "// Choreo version 1\n" );
// loop animation?
if ( m_flAnimationLoopStartTime >= 0.f )
{
BufPrintf( vcdBuf, nLevel, "event loop \"<ITEMTEST_SEQUENCE_NAME>_loop\"\n" );
BufBeginBlock( vcdBuf, nLevel );
{
BufPrintf( vcdBuf, nLevel, "time <MAX_SEQUENCE_LENGTH> -1.000000\n" );
BufPrintf( vcdBuf, nLevel, "param \"%f\"\n", m_flAnimationLoopStartTime );
BufPrintf( vcdBuf, nLevel, "loopcount \"-1\"\n" );
}
BufEndBlock( vcdBuf, nLevel );
}
BufPrintf( vcdBuf, nLevel, "actor \"<CLASS_NAME>\"\n" );
BufBeginBlock( vcdBuf, nLevel );
{
// sequence channel
BufPrintf( vcdBuf, nLevel, "channel \"<CLASS_NAME>\"\n" );
BufBeginBlock( vcdBuf, nLevel );
{
BufPrintf( vcdBuf, nLevel, "event sequence <ITEMTEST_SEQUENCE_NAME>\n" );
BufBeginBlock( vcdBuf, nLevel );
{
BufPrintf( vcdBuf, nLevel, "time 0.000000 <MAX_SEQUENCE_LENGTH>\n" );
BufPrintf( vcdBuf, nLevel, "param <ITEMTEST_SEQUENCE_NAME>\n" );
}
BufEndBlock( vcdBuf, nLevel );
}
BufEndBlock( vcdBuf, nLevel );
// Flex Anim channel
BufPrintf( vcdBuf, nLevel, "channel \"Face\"\n" );
BufBeginBlock( vcdBuf, nLevel );
{
int nChannelCount = pDmeChannelsClip->m_Channels.Count();
for ( int ci = 0; ci < nChannelCount; ++ci )
{
CDmeChannel *pChannel = pDmeChannelsClip->m_Channels[ ci ];
// skip all transform
if ( !pChannel || !pChannel->GetToAttribute() || pChannel->GetToAttribute()->GetType() != AT_FLOAT )
continue;
CDmElement *pToElement = pChannel->GetToElement();
const char *pszToElementName = pToElement->GetName();
// figure out IsStereoControl by checking if prefix is left_ and right_
// in pDmeChannelsClip->m_Channels
CDmeChannel *pLeftChannel = NULL; //pChannel->GetValueElement< CDmeChannel >( "leftvaluechannel" );
CDmeChannel *pRightChannel = NULL; //pChannel->GetValueElement< CDmeChannel >( "rightvaluechannel" );
for ( int iSearch = 0; iSearch < nChannelCount; ++iSearch )
{
CDmeChannel *pSearchChannel = pDmeChannelsClip->m_Channels[ iSearch ];
if ( !pSearchChannel )
continue;
// search for left and right channels
const char *pszSearchChannelName = pSearchChannel->GetName();
if ( V_strstr( pszSearchChannelName, pszToElementName ) )
{
if ( V_strstr( pszSearchChannelName, "left_" ) )
{
Assert( pLeftChannel == NULL );
pLeftChannel = pSearchChannel;
}
else if ( V_strstr( pszSearchChannelName, "right_" ) )
{
Assert( pRightChannel == NULL );
pRightChannel = pSearchChannel;
}
}
}
bool bIsStereo = pLeftChannel && pRightChannel;
if ( bIsStereo )
{
// should we do anything for stereo (eye animation)?
AssertMsg( 0, "Stereo channel is not supported." );
continue;
CDmeFloatLogLayer *pLeftLogLayer = GetTopmostLogLayer< float >( pLeftChannel );
CDmeFloatLogLayer *pRightLogLayer = GetTopmostLogLayer< float >( pRightChannel );
if ( !pLeftLogLayer || !pRightLogLayer )
continue;
CUtlVectorFixedGrowable< DmeTime_t, 1024 > times;
CUtlVectorFixedGrowable< float, 1024 > values;
CUtlVectorFixedGrowable< float, 1024 > balances;
// we should have some key count on both left and right key
if ( pLeftLogLayer->GetKeyCount() == 0 || pRightLogLayer->GetKeyCount() == 0 )
{
// something is wrong here.
Assert( 0 );
continue;
}
DmeTime_t tStartOnFrame = MIN( pLeftLogLayer ->GetKeyTime( 0 ), pRightLogLayer->GetKeyTime( 0 ) );
DmeTime_t tEndOnFrame = MAX( pLeftLogLayer ->GetKeyTime( pLeftLogLayer->GetKeyCount() - 1 ), pRightLogLayer->GetKeyTime( pRightLogLayer->GetKeyCount() - 1 ) );
#if 1 // resample time
for ( DmeTime_t tCurr = tStartOnFrame; tCurr <= tEndOnFrame; tCurr = tCurr.TimeAtNextFrame( framerate ) )
{
float flLeftValue = pLeftLogLayer->GetValue( pDmeChannelsClip->FromChildMediaTime( tCurr ) );
float flRightValue = pLeftLogLayer->GetValue( pDmeChannelsClip->FromChildMediaTime( tCurr ) );
float flValue, flBalance;
LeftRightToValueBalance( flLeftValue, flRightValue, flValue, flBalance );
times.AddToTail( tCurr );
values.AddToTail( flValue );
balances.AddToTail( flBalance );
}
#else // merge time list from left and right channels
int nLeft = pLeftLogLayer->GetKeyCount();
int nRight = pRightLogLayer->GetKeyCount();
int iLeft = pLeftLogLayer->FindKey( pDmeChannelsClip->ToChildMediaTime( tStart ) );
int iRight = pLeftLogLayer->FindKey( pDmeChannelsClip->ToChildMediaTime( tStart ) );
while ( iLeft < nLeft || iRight < nRight )
{
DmeTime_t tLeft = iLeft < nLeft ? pDmeChannelsClip->FromChildMediaTime( pLeftLogLayer ->GetKeyTime( iLeft ) ) : DMETIME_MAXTIME;
DmeTime_t tRight = iRight < nRight ? pDmeChannelsClip->FromChildMediaTime( pRightLogLayer->GetKeyTime( iRight ) ) : DMETIME_MAXTIME;
DmeTime_t t;
float flLeftVal, flRightVal;
if ( tLeft == tRight )
{
t = tLeft;
flLeftVal = pLeftLogLayer->GetKeyValue( iLeft );
flRightVal = pRightLogLayer->GetKeyValue( iRight );
iLeft++;
iRight++;
}
else if ( tLeft < tRight )
{
t = tLeft;
flLeftVal = pLeftLogLayer->GetKeyValue( iLeft );
flRightVal = pRightLogLayer->GetValue( t );
iLeft++;
}
else
{
t = tRight;
flLeftVal = pLeftLogLayer->GetValue( t );
flRightVal = pRightLogLayer->GetKeyValue( iRight );
iRight++;
}
float flValue, flBalance;
LeftRightToValueBalance( flLeftVal, flRightVal, flValue, flBalance );
times.AddToTail( t );
values.AddToTail( flValue );
balances.AddToTail( flBalance );
}
#endif
BufPrintf( vcdBuf, nLevel, "\"%s\" combo\n", pChannel->GetName() );
BufBeginBlock( vcdBuf, nLevel );
int nSamples = times.Count();
for ( int si = 0; si < nSamples; ++si )
{
BufPrintf( vcdBuf, nLevel, "%.4f %.4f\n", times[ si ].GetSeconds(), values[ si ] );
}
BufEndBlock( vcdBuf, nLevel );
BufBeginBlock( vcdBuf, nLevel );
for ( int si = 0; si < nSamples; ++si )
{
BufPrintf( vcdBuf, nLevel, "%.4f %.4f\n", times[ si ].GetSeconds(), balances[ si ] );
}
BufEndBlock( vcdBuf, nLevel );
}
else
{
CDmeFloatLogLayer *pLogLayer = GetTopmostLogLayer< float >( pChannel );
if ( !pLogLayer || pLogLayer->GetKeyCount() == 0 )
continue;
BufPrintf( vcdBuf, nLevel, "event expression \"%s\"\n", pToElement->GetName() );
BufBeginBlock( vcdBuf, nLevel );
{
BufPrintf( vcdBuf, nLevel, "time %.4f %.4f\n", pDmeChannelsClip->GetStartTime().GetSeconds(), pDmeChannelsClip->GetEndTime().GetSeconds() );
BufPrintf( vcdBuf, nLevel, "param \"player\\<CLASS_NAME>\\emotion\\emotion\"\n" );
BufPrintf( vcdBuf, nLevel, "param2 \"%s\"\n", pToElement->GetName() );
BufPrintf( vcdBuf, nLevel, "event_ramp\n" );
BufBeginBlock( vcdBuf, nLevel );
{
#if 0 // resampling
if ( tStart < tStartOnFrame )
{
float flValue = pLogLayer->GetValue( pDmeChannelsClip->FromChildMediaTime( tStart ) );
BufPrintf( vcdBuf, nLevel, "%.4f %.4f\n", tStart.GetSeconds(), flValue );
}
for ( DmeTime_t tCurr = tStartOnFrame; tCurr <= tEndOnFrame; tCurr = tCurr.TimeAtNextFrame( framerate ) )
{
float flValue = pLogLayer->GetValue( pDmeChannelsClip->FromChildMediaTime( tCurr ) );
BufPrintf( vcdBuf, nLevel, "%.4f %.4f\n", tCurr.GetSeconds(), flValue );
}
if ( tEnd > tEndOnFrame )
{
float flValue = pLogLayer->GetValue( pDmeChannelsClip->FromChildMediaTime( tEnd ) );
BufPrintf( vcdBuf, nLevel, "%.4f %.4f\n", tEnd.GetSeconds(), flValue );
}
#else // use sample time that's already in the animation
int nKeys = pLogLayer->GetKeyCount();
for ( int k = 0; k < nKeys; ++k )
{
DmeTime_t t = pDmeChannelsClip->FromChildMediaTime( pLogLayer->GetKeyTime( k ) );
float flValue = pLogLayer->GetKeyValue( k );
BufPrintf( vcdBuf, nLevel, "%.4f %.4f\n", t.GetSeconds(), flValue );
}
#endif
}
BufEndBlock( vcdBuf, nLevel );
}
BufEndBlock( vcdBuf, nLevel );
}
}
// channel sounds
OutputSounds( vcdBuf, nLevel, m_pDmRoot->GetValueElement< CDmElement >( "exportedSounds" ) );
}
BufEndBlock( vcdBuf, nLevel );
}
BufEndBlock( vcdBuf, nLevel );
BufPrintf( vcdBuf, nLevel, "fps %d\n", RoundFloatToInt( framerate.GetFramesPerSecond() ) );
BufPrintf( vcdBuf, nLevel, "snap off\n" );
CUtlString strBuf = vcdBuf.String();
FOR_EACH_SUBKEY( GetCustomKeyValues(), pSubKey )
{
strBuf = strBuf.Replace( pSubKey->GetName(), pSubKey->GetString() );
}
vcdBuf.Clear();
vcdBuf.PutString( strBuf );
return true;
}
//=============================================================================
//
//=============================================================================
CTargetVCD::CTargetVCD( CAsset *pAsset, const CTargetQC *pTargetQC )
: CTargetBase( pAsset, pTargetQC ), m_pTargetQC( pTargetQC )
{
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetVCD::Compile()
{
if ( !CTargetBase::Compile() )
return false;
CUtlString sName;
if ( !GetOutputPath( sName, 0, PATH_FLAG_FILE | PATH_FLAG_EXTENSION ) )
return false;
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( "CTarget%s::Compile( %s ) - Not Valid: %s\n", GetTypeString(), sName.String(), sTmp.String() );
return false;
}
CUtlString sBinDir;
if ( !CItemUpload::GetBinDirectory( sBinDir ) )
{
Warning( "CTarget%s::Compile( %s ) - GetBinDirectory Failed\n", GetTypeString(), sName.String() );
return false;
}
CUtlString sAbsPath;
GetOutputPath( sAbsPath, 0 );
CUtlString sRelPath;
GetOutputPath( sRelPath, 0, PATH_FLAG_ALL & ~PATH_FLAG_ABSOLUTE );
if ( sAbsPath.IsEmpty() || sRelPath.IsEmpty() )
{
Warning( "CTarget%s::Compile( %s ) - GetOutputPath failed\n", GetTypeString(), sName.String() );
return false;
}
Msg( "Compiling %s: %s\n", GetTypeString(), sRelPath.String() );
CUtlBuffer bufVCDFile( 0, 0, CUtlBuffer::TEXT_BUFFER );
if ( m_strVCDPath.IsEmpty() )
{
if ( !m_pTargetQC->GetTargetDMX( 0 )->WriteVCD( bufVCDFile ) )
{
Warning( "CTarget%s::Compile( %s ) - failed to create VCD buffer\n", GetTypeString(), m_strVCDPath.String() );
return false;
}
}
else if ( !g_pFullFileSystem->ReadFile( m_strVCDPath, NULL, bufVCDFile ) )
{
Warning( "CTarget%s::Compile( %s ) - failed to read input file\n", GetTypeString(), m_strVCDPath.String() );
return false;
}
// output to the correct file path
AddOrEditP4File( sAbsPath.String() );
g_pFullFileSystem->WriteFile( sAbsPath.String(), "MOD", bufVCDFile );
Msg( "CTarget%s::Compile OK! - %s\n", GetTypeString(), sRelPath );
if ( GetCustomModPath() )
{
CUtlString sWorkingDir;
if ( CItemUpload::IgnoreEnvironmentVariables() )
{
sWorkingDir = sAbsPath;
sWorkingDir.SetLength( sWorkingDir.Length() - sRelPath.Length() );
V_StripTrailingSlash( sWorkingDir.GetForModify() );
}
else
{
CItemUpload::GetVProjectDir( sWorkingDir );
}
char szCustomPath[MAX_PATH];
V_MakeAbsolutePath( szCustomPath, sizeof( szCustomPath ), CFmtStr( "%s/%s", GetCustomModPath(), sRelPath.String() ), sWorkingDir.String() );
V_FixSlashes( szCustomPath );
if ( DoFileCopy( sAbsPath.String(), szCustomPath ) )
{
Asset()->AddModOutput( szCustomPath );
}
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const ExtensionList *CTargetVCD::GetExtensionsAndCount( void ) const
{
static ExtensionList vecExtensions;
if ( !vecExtensions.Count() )
{
vecExtensions.AddToTail( ".vcd" );
}
return &vecExtensions;
}
//=============================================================================
//
//=============================================================================
CTargetQC::CTargetQC( CAsset *pAsset, const CTargetMDL *pTargetMDL )
: CTargetBase( pAsset, pTargetMDL )
, m_QCTemplate( 0, 0, CUtlBuffer::TEXT_BUFFER )
, m_TargetVCD( NULL )
{
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetQC::IsOk( CUtlString &sMsg ) const
{
if ( m_TargetDMXs.Count() <= 0 )
{
sMsg = "No DMX input files specified";
return false;
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetQC::Compile()
{
if ( !CTargetBase::Compile() )
return false;
CUtlString sName;
if ( !GetOutputPath( sName, 0, PATH_FLAG_FILE | PATH_FLAG_EXTENSION ) )
return false;
CUtlString sAbsPath;
GetOutputPath( sAbsPath, 0 );
CUtlString sRelPath;
GetOutputPath( sRelPath, 0, PATH_FLAG_ALL & ~PATH_FLAG_ABSOLUTE );
if ( sAbsPath.IsEmpty() || sRelPath.IsEmpty() )
{
Warning( "CTarget%s::Compile( %s ) - GetOutputPath failed\n", GetTypeString(), sName.String() );
return false;
}
CUtlString sLOD0;
GetOutputPath( sLOD0, 0, PATH_FLAG_PATH | PATH_FLAG_FILE | PATH_FLAG_PREFIX | PATH_FLAG_MODELS );
sLOD0.FixSlashes( '/' );
const char *pszPostModel = V_strstr( sLOD0.String(), "models/" );
CUtlString sModelName = pszPostModel ? pszPostModel + 7 : sLOD0;
sModelName += ".mdl";
Msg( "Compiling %s: %s\n", GetTypeString(), sRelPath.String() );
AddOrEditP4File( sAbsPath.String() );
CUtlString strBuf = GetQCTemplate();
if ( strBuf.IsEmpty() )
{
return false;
}
strBuf = strBuf.Replace( "<ITEMTEST_REPLACE_MDLABSPATH>", sModelName.String() );
for ( int nLOD = 1; nLOD < m_TargetDMXs.Count(); ++nLOD )
{
strBuf = strBuf.Replace( CFmtStr( "<ITEMTEST_REPLACE_LOD%d_HEADER>", nLOD ), CFmtStr( "$lod %d", CItemUpload::Manifest()->GetQCLODDistance( nLOD ) ) );
}
// Remove REPLACE_LOD block if it's not necessary
RemoveTextBlock( strBuf.String(), "<ITEMTEST_REPLACE_LOD1_HEADER>", strBuf.GetForModify(), strBuf.Length() );
RemoveTextBlock( strBuf.String(), "<ITEMTEST_REPLACE_LOD2_HEADER>", strBuf.GetForModify(), strBuf.Length() );
CUtlString sSearch = "<ITEMTEST_REPLACE_SKIN_OPTIONALBLOCK>";
CUtlString sReplace = "$cdmaterials \"<ITEMTEST_REPLACE_MATERIALS_PATH>\"\n$texturegroup skinfamilies\n{\n";
CUtlString sMaterialPath;
for ( int nSkin = 0; nSkin < CItemUpload::Manifest()->GetNumMaterialSkins(); ++nSkin )
{
int nVMTCount = 0;
CUtlString sSkinMaterials = " { ";
for ( int nVMT = 0; nVMT < Asset()->GetTargetVMTCount(); ++nVMT )
{
CTargetVMT *pTargetVMT = Asset()->GetTargetVMT( nVMT );
if ( nSkin >= pTargetVMT->GetOutputCount() )
continue;
// We grab the extension which includes the skin variant, and then trim the actual file extension
CUtlString sVMTName;
if ( !pTargetVMT->GetOutputPath( sVMTName, nSkin, PATH_FLAG_FILE | PATH_FLAG_EXTENSION ) )
{
Warning( "CTarget%s::Compile( %s ) - VMT %d skin %d GetOutputPath failed\n", GetTypeString(), sName.String(), nVMT, nSkin );
continue;
}
V_StripExtension( sVMTName.String(), sVMTName.GetForModify(), sVMTName.Length() );
if ( sMaterialPath.IsEmpty() )
{
pTargetVMT->GetOutputPath( sMaterialPath, 0, PATH_FLAG_PATH | PATH_FLAG_MODELS );
}
sSkinMaterials += CFmtStr( "\"%s\" ", sVMTName.String() );
++nVMTCount;
}
sSkinMaterials += "}\n";
if ( nVMTCount > 0 )
{
sReplace += sSkinMaterials;
}
}
sReplace += "}\n";
strBuf = strBuf.Replace( sSearch.String(), sReplace.String() );
strBuf = strBuf.Replace( "<ITEMTEST_REPLACE_MATERIALS_PATH>", sMaterialPath.String() );
for ( int nLOD = 0; nLOD < m_TargetDMXs.Count(); ++nLOD )
{
CUtlString sLOD;
m_TargetDMXs[ nLOD ]->GetOutputPath( sLOD, 0, PATH_FLAG_FILE | PATH_FLAG_EXTENSION );
strBuf = strBuf.Replace( CFmtStr( "<ITEMTEST_REPLACE_LOD%d>", nLOD ), sLOD.String() );
}
// Should we include QCI?
if ( !m_strQCITemplate.IsEmpty() )
{
CUtlString strQCIDir;
m_TargetDMXs[ 0 ]->GetOutputPath( strQCIDir, 1, PATH_FLAG_PATH & ~PATH_FLAG_ABSOLUTE );
strQCIDir = strQCIDir.Replace( '\\', '/' );
strBuf = strBuf.Replace( "<QCI_RELATIVE_DIR>", strQCIDir.String() );
CUtlString strQCIRel;
m_TargetDMXs[ 0 ]->GetOutputPath( strQCIRel, 1, PATH_FLAG_ALL & ~PATH_FLAG_ABSOLUTE & ~PATH_FLAG_MODELS );
strQCIRel = strQCIRel.Replace( '\\', '/' );
strBuf = strBuf.Replace( "<QCI_RELATIVE_PATH>", strQCIRel.String() );
}
FOR_EACH_SUBKEY( GetCustomKeyValues(), pSubKey )
{
strBuf = strBuf.Replace( pSubKey->GetName(), pSubKey->GetString() );
}
CUtlBuffer bufQCFile( 0, 0, CUtlBuffer::TEXT_BUFFER );
bufQCFile.PutString( strBuf.String() );
g_pFullFileSystem->WriteFile( sAbsPath.String(), "MOD", bufQCFile );
if ( !CheckFile( sAbsPath.String() ) )
{
Warning( "CTarget%s::Compile( %s ) - File Check Failed - \"%s\"\n", GetTypeString(), sName.String(), sAbsPath.String() );
return false;
}
if ( CItemUpload::Manifest()->UseTerseMessages() )
{
Msg( " - Compilation successful.\n" );
}
else
{
Msg( "CTarget%s::Compile OK! - %s\n", GetTypeString(), sRelPath.String() );
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const ExtensionList *CTargetQC::GetExtensionsAndCount( void ) const
{
static ExtensionList vecExtensions;
if ( !vecExtensions.Count() )
{
vecExtensions.AddToTail( ".qc" );
}
return &vecExtensions;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetQC::GetInputs( CUtlVector< CTargetBase * > &sInputs ) const
{
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - Not Valid: %s\n", GetTypeString(), sTmp.String() );
return false;
}
int nOkCount = 0;
for ( int i = 0; i < m_TargetDMXs.Count(); ++i )
{
CTargetBase *pTargetBase = m_TargetDMXs.Element( i ).GetObject();
if ( !pTargetBase )
continue;
nOkCount += ( sInputs.AddToTail( pTargetBase ) >= 0 ? 1 : 0 );
}
if ( m_TargetVCD.IsValid() )
{
CTargetBase *pTargetBase = m_TargetVCD.GetObject();
if ( pTargetBase )
{
nOkCount += ( sInputs.AddToTail( pTargetBase ) >= 0 ? 1 : 0 );
}
}
return nOkCount > 0;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CTargetQC::TargetDMXCount() const
{
return m_TargetDMXs.Count();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CTargetQC::SetQCTemplate( const char *pszQCTemplate )
{
m_QCTemplate.Clear();
m_QCTemplate.PutString( pszQCTemplate );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const char *CTargetQC::GetQCTemplate()
{
if ( m_QCTemplate.TellMaxPut() == 0 )
{
const char *pszQCTemplate = CItemUpload::Manifest()->GetQCTemplate();
if ( pszQCTemplate )
{
if ( !g_pFullFileSystem->ReadFile( pszQCTemplate, "MOD", m_QCTemplate ) )
{
Warning( "Failed to load specified QC template '%s'.\n", pszQCTemplate );
}
}
}
return (char*)m_QCTemplate.Base();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CTargetQC::AddTargetDMX( const char *pszGeometryFile )
{
CSmartPtr< CTargetDMX > pTargetDMX = Asset()->NewTarget< CTargetDMX >( this );
if ( !pTargetDMX )
return -1;
const int nIndex = m_TargetDMXs.AddToTail( pTargetDMX );
if ( nIndex < 0 )
{
// Failed to add
return -1;
}
if ( !SetTargetDMX( nIndex, pszGeometryFile ) )
{
m_TargetDMXs.RemoveMultipleFromTail( 1 );
return -1;
}
return nIndex;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetQC::SetTargetDMX( int nLOD, const char *pszGeometryFile )
{
if ( nLOD >= TargetDMXCount() )
return false;
CSmartPtr< CTargetDMX > pTargetDMX = m_TargetDMXs.Element( nLOD );
if ( !pTargetDMX )
return false;
pTargetDMX->SetNameSuffix( GetNameSuffix() );
pTargetDMX->SetLod( nLOD );
pTargetDMX->SetQCITemplate( m_strQCITemplate.String() );
return pTargetDMX->SetInputFile( pszGeometryFile );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetQC::RemoveTargetDMX( int nLOD )
{
if ( nLOD != m_TargetDMXs.Count() - 1 )
return false;
m_TargetDMXs.RemoveMultipleFromTail( 1 );
return false;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CSmartPtr< CTargetDMX > CTargetQC::GetTargetDMX( int nLOD ) const
{
if ( nLOD >= TargetDMXCount() )
return NULL;
return m_TargetDMXs.Element( nLOD );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CSmartPtr< CTargetVCD > CTargetQC::GetTargetVCD()
{
if ( m_TargetVCD == NULL )
{
m_TargetVCD = Asset()->NewTarget< CTargetVCD >( this );
}
return m_TargetVCD;
}
//=============================================================================
//
//=============================================================================
CTargetMDL::CTargetMDL( CAsset *pAsset, const CTargetBase *pTargetParent )
: CTargetBase( pAsset, pTargetParent )
, m_pTargetQC( NULL )
{
m_pTargetQC = Asset()->NewTarget< CTargetQC >( this );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetMDL::IsOk( CUtlString &sMsg ) const
{
CUtlString sTmp;
if ( !m_pTargetQC || !m_pTargetQC->IsOk( sTmp ) )
{
sMsg = "Invalid QC: ";
sMsg += sTmp;
return false;
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetMDL::Compile()
{
if ( !CTargetBase::Compile() )
return false;
CUtlString sName;
if ( !GetOutputPath( sName, 0, PATH_FLAG_FILE | PATH_FLAG_EXTENSION ) )
return false;
CUtlString sBinDir;
if ( !CItemUpload::GetBinDirectory( sBinDir ) )
{
Warning( "CTarget%s::Compile( %s ) - GetBinDirectory Failed\n", GetTypeString(), sName.String() );
return false;
}
CUtlVector< CUtlString > sAbsPaths;
GetOutputPaths( sAbsPaths );
if ( sAbsPaths.Count() <= 0 )
{
Warning( "CTarget%s::Compile( %s ) - GetOutputPaths failed\n", GetTypeString(), sName.String() );
return false;
}
CUtlString sAbsPath = sAbsPaths.Element( 0 );
CUtlString sRelPath;
GetOutputPath( sRelPath, 0, PATH_FLAG_ALL & ~PATH_FLAG_ABSOLUTE );
if ( sAbsPath.IsEmpty() || sRelPath.IsEmpty() )
{
Warning( "CTarget%s::Compile( %s ) - GetOutputPath failed\n", GetTypeString(), sName.String() );
return false;
}
CUtlVector< CUtlString > sAbsInputPaths;
if ( !GetInputPaths( sAbsInputPaths, false, false ) )
{
Warning( "CTarget%s::Compile( %s ) - GetInputPaths failed\n", GetTypeString(), sName.String() );
return false;
}
if ( sAbsInputPaths.Count() != 1 )
{
Warning( "CTarget%s::Compile( %s ) - GetPaths returned %d paths, expected 1\n", GetTypeString(), sName.String(), sAbsInputPaths.Count() );
return false;
}
if ( !Asset()->Mkdir( NULL, this ) )
{
Warning( "CTarget%s::Compile - Mkdir failed\n", GetTypeString() );
return false;
}
Msg( "Compiling %s: %s\n", GetTypeString(), sRelPath.String() );
if ( CItemUpload::GetP4() )
{
for ( int i = 0; i < sAbsPaths.Count(); ++i )
{
AddOrEditP4File( sAbsPaths.Element( i ).String() );
}
}
CUtlString sWorkingDir;
CFmtStrN< k64KB > sCmd;
if ( CItemUpload::IgnoreEnvironmentVariables() )
{
sWorkingDir = sAbsPath;
sWorkingDir.SetLength( sWorkingDir.Length() - sRelPath.Length() );
V_StripTrailingSlash( sWorkingDir.GetForModify() );
sCmd.sprintf( "\"%s\\studiomdl.exe\" -nop4 -game \"%s\" \"%s\"", sBinDir.String(), sWorkingDir.String(), sAbsInputPaths.Element( 0 ).String() );
}
else
{
CItemUpload::GetVProjectDir( sWorkingDir );
sCmd.sprintf( "\"%s\\studiomdl.exe\" -nop4 -vproject \"%s\" \"%s\"", sBinDir.String(), sWorkingDir.String(), sAbsInputPaths.Element( 0 ).String() );
}
bool bOk = true;
if ( CItemUpload::RunCommandLine( sCmd.Access(), sBinDir.String(), this ) )
{
for ( int i = 0; i < sAbsPaths.Count(); ++i )
{
if ( !CheckFile( sAbsPaths.Element( i ).String() ) )
{
Warning( "CTarget%s::Compile( %s ) - CheckFile failed - \"%s\"\n", GetTypeString(), sName.String(), sAbsPaths.Element( i ).String() );
bOk = false;
break;
}
}
}
else
{
bOk = false;
}
if ( bOk )
{
if ( CItemUpload::Manifest()->UseTerseMessages() )
{
Msg( " - Compilation successful.\n" );
}
else
{
Msg( "CTarget%s::Compile OK! - %s\n", GetTypeString(), sRelPath.String() );
}
}
else
{
Warning( "CTarget%s::Compile Failed - %s\n", GetTypeString(), sAbsPath.String() );
return false;
}
if ( GetCustomModPath() )
{
char szCustomPath[MAX_PATH];
V_MakeAbsolutePath( szCustomPath, sizeof( szCustomPath ), CFmtStr( "%s/%s", GetCustomModPath(), sRelPath.String() ), sWorkingDir.String() );
V_FixSlashes( szCustomPath );
DoFileCopy( sAbsPath.String(), szCustomPath );
Asset()->AddModOutput( szCustomPath );
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const ExtensionList *CTargetMDL::GetExtensionsAndCount( void ) const
{
// Only output .mdl for animation
CSmartPtr< CTargetQC > pTargetQC = GetTargetQC();
if ( pTargetQC.IsValid() && pTargetQC->GetQCITemplate() )
{
return CItemUpload::Manifest()->GetAnimationMDLExtentions();
}
return CItemUpload::Manifest()->GetMDLExtensions();
/*
static const char *s_szExtensionsDX8[] = {
".mdl",
".dx80.vtx",
".dx90.vtx",
".sw.vtx",
".phy",
".vvd"
};
static const char *s_szExtensions[] = {
".mdl",
".dx90.vtx",
".phy",
".vvd"
};
static const char **s_ppszExtensions = s_szExtensionsDX8;
static bool s_bGameInfoParsed = false;
if ( !s_bGameInfoParsed )
{
KeyValues *pKeyValues = new KeyValues( "gameinfo.txt" );
if ( pKeyValues != NULL )
{
if ( g_pFullFileSystem && pKeyValues->LoadFromFile( g_pFullFileSystem, "gameinfo.txt" ) )
{
if ( pKeyValues->GetInt( "SupportsDX8" ) != 0 )
{
s_ppszExtensions = s_szExtensionsDX8;
}
else
{
s_ppszExtensions = s_szExtensions;
}
s_bGameInfoParsed = true;
}
pKeyValues->deleteThis();
}
}
if ( s_ppszExtensions == s_szExtensions )
{
nExtensionCount = ARRAYSIZE( s_szExtensions );
}
else
{
nExtensionCount = ARRAYSIZE( s_szExtensionsDX8 );
}
return s_ppszExtensions;
*/
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CTargetMDL::GetInputs( CUtlVector< CTargetBase * > &inputs ) const
{
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - Not Valid: %s\n", GetTypeString(), sTmp.String() );
return false;
}
inputs.RemoveAll();
inputs.AddToTail( m_pTargetQC.GetObject() );
return true;
}
//=============================================================================
//
//=============================================================================
CAsset::CAsset()
: CTargetBase( NULL, NULL )
, m_bSkinToBipHead( false )
, m_nCurrentModel( -1 )
, m_vmtMap( UtlStringLessThan )
, m_bShouldBuildScenesImage( false )
{
m_pAsset = this;
CItemUpload::GetSteamId( m_sSteamId );
AddModel();
m_vecTargetIcons.SetCount( CItemUpload::Manifest()->GetNumIconTypes() );
m_sExcludeFileExtensions.AddToTail( "zip" );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CAsset::CAsset( const char *pszName, bool *pbOk /* = NULL */ )
: CTargetBase( NULL, NULL )
, m_nCurrentModel( -1 )
, m_bShouldBuildScenesImage( false )
{
m_pAsset = this;
CItemUpload::GetSteamId( m_sSteamId );
m_sName = pszName;
AddModel();
m_vecTargetIcons.SetCount( CItemUpload::Manifest()->GetNumIconTypes() );
if ( pbOk )
{
CUtlString sTmp;
*pbOk = IsOk( sTmp );
if ( !( *pbOk ) )
{
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - Not Valid: %s\n", GetTypeString(), sTmp.String() );
}
}
m_sExcludeFileExtensions.AddToTail( "zip" );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CAsset::~CAsset()
{
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::IsOk( CUtlString &sMsg ) const
{
if ( !IsNameValid() )
{
sMsg = "Invalid asset name";
return false;
}
if ( !IsSteamIdValid() )
{
sMsg = "Invalid steam id";
return false;
}
if ( !GetTargetMDL().IsValid() )
{
sMsg = "No target MDL";
return false;
}
CUtlString sTmp;
if ( !CItemUpload::GetContentDir( sTmp ) || sTmp.IsEmpty() )
{
sMsg = "Cannot figure out content directory, have you installed the Source SDK and restarted Steam?";
return false;
}
return true;
}
bool CAsset::BuildScenesImage()
{
CUtlString sBinDir;
if ( !CItemUpload::GetBinDirectory( sBinDir ) )
{
Warning( "CTarget%s::Compile( %s ) - GetBinDirectory Failed\n", GetTypeString(), GetAssetName() );
return false;
}
CUtlString sAbsPath;
GetOutputPath( sAbsPath );
CUtlString sRelPath;
GetOutputPath( sRelPath, 0, PATH_FLAG_ALL & ~PATH_FLAG_ABSOLUTE );
CUtlString sWorkingDir;
if ( CItemUpload::IgnoreEnvironmentVariables() )
{
sWorkingDir = sAbsPath;
sWorkingDir.SetLength( sWorkingDir.Length() - sRelPath.Length() );
V_StripTrailingSlash( sWorkingDir.GetForModify() );
}
else
{
CItemUpload::GetVProjectDir( sWorkingDir );
}
char szScenesImage[MAX_PATH];
V_MakeAbsolutePath( szScenesImage, sizeof( szScenesImage ), "scenes\\scenes.image", sWorkingDir.String() );
AddOrEditP4File( szScenesImage );
CFmtStrN< k64KB > sCmd;
sCmd.sprintf( "%s\\makescenesimage.exe", sBinDir.String() );
if ( !CItemUpload::RunCommandLine( sCmd.Access(), sWorkingDir.String(), this ) )
{
Warning( "Failed to build %s\n", szScenesImage );
return false;
}
Msg( "Build scenes.image OK!\n" );
if ( GetCustomModPath() )
{
char szCustomPath[MAX_PATH];
V_MakeAbsolutePath( szCustomPath, sizeof( szCustomPath ), CFmtStr( "%s/%s", GetCustomModPath(), "scenes/scenes.image" ), sWorkingDir.String() );
V_FixSlashes( szCustomPath );
DoFileCopy( szScenesImage, szCustomPath );
Asset()->AddModOutput( szCustomPath );
}
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::CompilePreview()
{
bool bOldP4State = CItemUpload::GetP4();
CItemUpload::SetP4( false );
m_CompileOutputFiles.RemoveAll();
g_bCompilePreview = true;
// Record the files that are created so they can be removed after preview
m_sAbsPaths.RemoveAll();
m_sRelPaths.RemoveAll();
m_sModOutputs.RemoveAll();
CUtlVector< CTargetBase * > inputs;
if ( GetInputs( inputs ) )
{
for ( int i = 0; i < inputs.Count(); ++i )
{
CTargetBase *pTargetBase = inputs.Element( i );
if ( !pTargetBase )
continue;
pTargetBase->GetOutputPaths( m_sAbsPaths, PATH_FLAG_ALL, true );
pTargetBase->GetOutputPaths( m_sRelPaths, (PATH_FLAG_ALL & ~PATH_FLAG_ABSOLUTE) | PATH_FLAG_ZIP, true );
}
}
bool bResult = CTargetBase::Compile();
g_bCompilePreview = false;
CItemUpload::SetP4( bOldP4State );
if ( bResult )
{
PostCompilePreview();
}
return bResult;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::PostCompilePreview()
{
Msg( "CAsset::PostCompilePreview Start:\n");
bool bResult = true;
if ( m_bShouldBuildScenesImage )
{
bResult = BuildScenesImage();
}
if ( bResult )
{
Msg( "CAsset::PostCompilePreview OK!\n" );
}
else
{
Msg( "CAsset::PostCompilePreview FAILED!\n" );
}
return bResult;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::Compile()
{
m_CompileOutputFiles.RemoveAll();
if ( m_sArchivePath.IsEmpty() )
{
GetOutputPath( m_sArchivePath, 0 );
}
CUtlString sRelPath;
GetOutputPath( sRelPath, 0, PATH_FLAG_ALL & ~PATH_FLAG_ABSOLUTE );
if ( m_sArchivePath.IsEmpty() || sRelPath.IsEmpty() )
{
Warning( "CTarget%s::Compile - GetOutputPath failed\n", GetTypeString() );
return false;
}
// this gets filled by Compile()
m_sModOutputs.RemoveAll();
if ( !CTargetBase::Compile() )
return false;
m_sAbsPaths.RemoveAll();
GetOutputPaths( m_sAbsPaths, PATH_FLAG_ALL, true );
m_sRelPaths.RemoveAll();
GetOutputPaths( m_sRelPaths, (PATH_FLAG_ALL & ~PATH_FLAG_ABSOLUTE) | PATH_FLAG_ZIP, true );
Assert( m_sAbsPaths.Count() == m_sRelPaths.Count() );
if ( m_sAbsPaths.Count() <= 0 || m_sAbsPaths.Count() != m_sRelPaths.Count() )
{
Warning( "CTarget%s::Compile - GetOutputPaths failed\n", GetTypeString() );
return false;
}
Msg( "Compiling %s: %s\n", GetTypeString(), sRelPath.String() );
AddOrEditP4File( m_sArchivePath.String() );
if ( CItemUpload::FileExists( m_sArchivePath.String() ) )
{
_unlink( m_sArchivePath.String() );
}
HANDLE m_hOutputZipFile = CreateFile( m_sArchivePath.String(), GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL );
if ( m_hOutputZipFile == INVALID_HANDLE_VALUE )
{
Warning( "CTarget%s::Compile CreateZip Failed - Unable to create ZIP file %s.\n", GetTypeString(), m_sArchivePath.String() );
return false;
}
IZip *pZip = IZip::CreateZip();
if ( pZip == NULL )
{
Warning( "CTarget%s::Compile CreateZip Failed - %s\n", GetTypeString(), sRelPath.String() );
CloseHandle( m_hOutputZipFile );
return false;
}
CUtlBuffer kvBuf( 0, 0, CUtlBuffer::TEXT_BUFFER );
CreateManifest( kvBuf );
pZip->AddBufferToZip( "manifest.txt", kvBuf.Base(), kvBuf.TellPut(), true );
for ( int i = 0; i < m_sAbsPaths.Count(); ++i )
{
const char *pszFileExtension = V_GetFileExtension( m_sAbsPaths.Element( i ).String() );
bool bExcluded = false;
FOR_EACH_VEC( m_sExcludeFileExtensions, e )
{
if ( m_sExcludeFileExtensions[e] == pszFileExtension )
{
bExcluded = true;
break;
}
}
if ( bExcluded )
continue;
if ( CItemUpload::Manifest()->UseTerseMessages() )
{
Msg( " - added %s\n", m_sRelPaths.Element( i ).String() );
}
else
{
Msg( " + Zip Add: %s\n", m_sRelPaths.Element( i ).String() );
}
pZip->AddFileToZip( m_sRelPaths.Element( i ).String(), m_sAbsPaths.Element( i ).String() );
}
pZip->SaveToDisk( m_hOutputZipFile );
IZip::ReleaseZip( pZip );
CloseHandle( m_hOutputZipFile );
if ( !CheckFile( m_sArchivePath.String() ) )
{
if ( CItemUpload::Manifest()->UseTerseMessages() )
{
Warning( "Failed to write .zip file: \"%s\"\n", m_sArchivePath.String() );
}
else
{
Warning( "CTarget%s::Compile - File Check Failed - \"%s\"\n", GetTypeString(), m_sArchivePath.String() );
}
return false;
}
if ( CItemUpload::Manifest()->UseTerseMessages() )
{
Msg( " - Compilation successful.\n" );
}
else
{
Msg( "CTarget%s::Compile OK! - %s\n", GetTypeString(), sRelPath.String() );
}
return PostCompile();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::PostCompile()
{
Msg( "CAsset::PostCompile Start:\n");
bool bResult = true;
if ( m_bShouldBuildScenesImage )
{
bResult = BuildScenesImage();
}
if ( bResult )
{
Msg( "CAsset::PostCompile OK!\n" );
}
else
{
Msg( "CAsset::PostCompile FAILED!\n" );
}
return bResult;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::GetInputs( CUtlVector< CTargetBase * > &inputs ) const
{
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - Not Valid: %s\n", GetTypeString(), sTmp.String() );
return false;
}
for ( int nIcon = 0; nIcon < m_vecTargetIcons.Count(); ++nIcon )
{
if ( m_vecTargetIcons[ nIcon ].IsValid() )
{
inputs.AddToTail( m_vecTargetIcons[ nIcon ].GetObject() );
}
}
for ( int i = 0; i < GetTargetVMTCount(); ++i )
{
inputs.AddToTail( GetTargetVMT( i ) );
}
FOR_EACH_VEC( m_vecModels, nModelIndex )
{
if ( m_vecModels[ nModelIndex ].IsValid() )
{
inputs.AddToTail( m_vecModels[ nModelIndex ].GetObject() );
}
}
return inputs.Count() > 0;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const ExtensionList *CAsset::GetExtensionsAndCount( void ) const
{
static ExtensionList vecExtensions;
if ( !vecExtensions.Count() )
{
vecExtensions.AddToTail( ".zip" );
}
return &vecExtensions;
}
//-----------------------------------------------------------------------------
// Returns items/<steamid>/<name>, false if there's something wrong
//-----------------------------------------------------------------------------
bool CAsset::GetRelativeDir( CUtlString &sRelativeDir, const char *pszPrefix, const CTargetBase *pTarget ) const
{
CUtlString sTmp;
if ( !IsOk( sTmp ) )
{
Warning( FUNCTION_LINE_STRING "Error! CTarget%s - Not Valid: %s\n", GetTypeString(), sTmp.String() );
return false;
}
if ( pszPrefix )
{
sRelativeDir = pszPrefix;
sRelativeDir += "/";
}
else
{
sRelativeDir = "";
}
if ( pTarget->IsModelPath() )
sRelativeDir += "models/";
if ( pTarget->GetCustomRelativeDir() )
{
sRelativeDir += pTarget->GetCustomRelativeDir();
}
else
{
sRelativeDir += pTarget->GetItemDirectory();
const char *pszClass = GetClass();
if ( pszClass )
{
sRelativeDir += pszClass;
sRelativeDir += "/";
}
if ( CItemUpload::Manifest()->GetItemPathUsesSteamId() )
{
if ( !m_sSteamId.IsEmpty() )
{
sRelativeDir += m_sSteamId;
sRelativeDir += "/";
}
}
sRelativeDir += m_sName;
}
sRelativeDir.FixSlashes();
return true;
}
//-----------------------------------------------------------------------------
// Returns GetAbsoluteContentDir()/GetRelativeDir() or GetAbsoluteGameDir()/GetRelativeDir()
//-----------------------------------------------------------------------------
bool CAsset::GetAbsoluteDir( CUtlString &sAbsoluteDir, const char *pszPrefix /* = NULL */, const CTargetBase *pTarget ) const
{
CUtlString sDirA;
if ( pTarget->IsContent() )
{
if ( !CItemUpload::GetContentDir( sDirA ) )
return false;
}
else
{
if ( !CItemUpload::GetVProjectDir( sDirA ) )
return false;
}
CUtlString sDirB;
if ( !GetRelativeDir( sDirB, pszPrefix, pTarget ) )
return false;
sAbsoluteDir = sDirA;
sAbsoluteDir += "/";
sAbsoluteDir += sDirB;
sAbsoluteDir.FixSlashes();
return true;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::SetName( const char *pszName )
{
if ( !CItemUpload::SanitizeName( pszName, m_sName ) )
return false;
return IsNameValid();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::IsNameValid() const
{
return m_sName.Length() > 0;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const char *CAsset::GetSteamId() const
{
return m_sSteamId.String();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::IsSteamIdValid() const
{
return CItemUpload::GetDevMode() || m_sSteamId.Length() > 0;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::SetTargetIcon( int nIcon, const char *pszIconFile )
{
if ( !m_vecTargetIcons[ nIcon ].IsValid() )
{
m_vecTargetIcons[ nIcon ] = new CTargetIcon( this, nIcon );
}
return m_vecTargetIcons[ nIcon ]->SetTargetVTF( pszIconFile );
}
//-----------------------------------------------------------------------------
// Add a new model to the output and make it current
//-----------------------------------------------------------------------------
int CAsset::AddModel()
{
CSmartPtr< CTargetMDL > pModel = NewTarget< CTargetMDL >( this );
m_nCurrentModel = m_vecModels.AddToTail( pModel );
return m_nCurrentModel;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::SetCurrentModel( int nModel )
{
if ( nModel >= 0 && nModel < GetNumModels() )
{
m_nCurrentModel = nModel;
return true;
}
return false;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CAsset::RemoveModels()
{
m_vecModels.RemoveAll();
m_nCurrentModel = -1;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CAsset::TargetDMXCount() const
{
CSmartPtr< CTargetQC > pTargetQC = GetTargetQC();
if ( !pTargetQC )
return false;
return pTargetQC->TargetDMXCount();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CAsset::AddTargetDMX( const char *pszGeometryFile )
{
CSmartPtr< CTargetQC > pTargetQC = GetTargetQC();
if ( !pTargetQC )
return -1;
return pTargetQC->AddTargetDMX( pszGeometryFile );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::SetTargetDMX( int nLOD, const char *pszGeometryFile )
{
CSmartPtr< CTargetQC > pTargetQC = GetTargetQC();
if ( !pTargetQC )
return false;
return pTargetQC->SetTargetDMX( nLOD, pszGeometryFile );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::RemoveTargetDMX( int nLOD )
{
CSmartPtr< CTargetQC > pTargetQC = GetTargetQC();
if ( !pTargetQC )
return false;
return pTargetQC->RemoveTargetDMX( nLOD );
}
//-----------------------------------------------------------------------------
//
//----------------------------------------------------------------------------
CSmartPtr< CTargetDMX > CAsset::GetTargetDMX( int nLOD )
{
CSmartPtr< CTargetQC > pTargetQC = GetTargetQC();
if ( !pTargetQC )
return false;
return pTargetQC->GetTargetDMX( nLOD );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CSmartPtr< CTargetMDL > CAsset::GetTargetMDL() const
{
if ( m_nCurrentModel >= 0 )
{
return m_vecModels[ m_nCurrentModel ];
}
return CSmartPtr< CTargetMDL >();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CSmartPtr< CTargetQC > CAsset::GetTargetQC() const
{
CSmartPtr< CTargetMDL > pTargetMDL = GetTargetMDL();
if ( !pTargetMDL )
return NULL;
return pTargetMDL->GetTargetQC();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
int CAsset::GetTargetVMTCount() const
{
return m_vmtMap.Count();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CTargetVMT *CAsset::GetTargetVMT( int nIndex ) const
{
if ( nIndex < 0 || nIndex >= GetTargetVMTCount() )
return NULL;
int nMapIndex = 0;
FOR_EACH_MAP( m_vmtMap, nMapIt )
{
if ( nIndex == nMapIndex )
return m_vmtMap.Element( nMapIt );
++nMapIndex;
}
return NULL;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CSmartPtr< CTargetVMT > CAsset::FindOrAddMaterial( const char *pszMaterial, int nMaterialType )
{
const CUtlString sMaterial( pszMaterial );
CUtlMap< CUtlString, CTargetVMT * >::IndexType_t nIndex = m_vmtMap.Find( sMaterial );
if ( !m_vmtMap.IsValidIndex( nIndex ) )
{
CSmartPtr< CTargetVMT > pTargetVMT = NewTarget< CTargetVMT >( this );
if ( !pTargetVMT )
return NULL;
pTargetVMT->SetMaterialId( pszMaterial );
m_vmtMap.Insert( sMaterial, pTargetVMT.GetObject() );
// See if this is a duplicate of an existing material
for ( int i = 0; i < GetTargetVMTCount(); ++i )
{
CTargetVMT *pTmpTargetVMT = GetTargetVMT( i );
if ( !pTmpTargetVMT )
continue;
if ( pTmpTargetVMT->GetMaterialType() == nMaterialType )
{
pTargetVMT->SetDuplicate( nMaterialType );
break;
}
}
if ( !pTargetVMT->GetDuplicate() )
{
pTargetVMT->SetMaterialType( nMaterialType );
}
return pTargetVMT;
}
else
{
return m_vmtMap.Element( nIndex );
}
return NULL;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
CSmartPtr< CTargetVMT > CAsset::FindMaterial( const char *pszMaterial )
{
const CUtlString sMaterial( pszMaterial );
CUtlMap< CUtlString, CTargetVMT * >::IndexType_t nIndex = m_vmtMap.Find( sMaterial );
if ( m_vmtMap.IsValidIndex( nIndex ) )
return m_vmtMap.Element( nIndex );
return NULL;
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::Mkdir( const char *pszPrefix, const CTargetBase *pTarget )
{
CUtlString sAbsolute;
if ( !GetAbsoluteDir( sAbsolute, pszPrefix, pTarget ) )
return false;
char szBuf[ k64KB ];
if ( _fullpath( szBuf, sAbsolute.String(), ARRAYSIZE( szBuf ) ) == NULL )
return false;
return CItemUpload::CreateDirectory( szBuf );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CAsset::CreateManifest( CUtlBuffer &manifestBuf )
{
KeyValues *pAssetKV = new KeyValues( "asset" );
UpdateManifest( pAssetKV );
manifestBuf.Clear();
manifestBuf.SetBufferType( true, true );
pAssetKV->RecursiveSaveToFile( manifestBuf, 0 );
pAssetKV->deleteThis();
}
const char* CAsset::CheckRedundantOutputFilePath( const char* pszInputFilePath, const char* pszVTEXConfig, const char* pszOutputFilePath )
{
const char* pszLocalVTEXConfig = pszVTEXConfig ? pszVTEXConfig : "";
// we don't want to output multiple of the same texture file with the same vtex config
for ( int i=0; i<m_CompileOutputFiles.Count(); ++i )
{
const CompileOutputFile_t& tga = m_CompileOutputFiles[i];
if ( !V_stricmp( tga.m_strInputFilePath.String(), pszInputFilePath ) )
{
if ( !V_stricmp( tga.m_strVTEXConfig.String(), pszLocalVTEXConfig ) )
{
return tga.m_strOutputFilePath.String();
}
}
}
int index = m_CompileOutputFiles.AddToTail();
CompileOutputFile_t& newTGA = m_CompileOutputFiles[index];
newTGA.m_strInputFilePath = pszInputFilePath;
newTGA.m_strVTEXConfig = pszLocalVTEXConfig;
newTGA.m_strOutputFilePath = pszOutputFilePath;
return newTGA.m_strOutputFilePath.String();
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
bool CAsset::RemoveMaterial( const char *pszMaterial )
{
const CUtlString sMaterial( pszMaterial );
return m_vmtMap.Remove( sMaterial );
}
//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const CUtlString &CAsset::GetAssetName() const
{
return m_sName;
}