initial

game/client/cstrike15/cs_custom_texture_saver.cpp

@@ -0,0 +1,95 @@
+//====== Copyright  Valve Corporation, All rights reserved. =================
+//
+// Purpose:
+//
+//=============================================================================
+
+#include "cbase.h"
+#include "cs_custom_texture_saver.h"
+#include "tier2/p4helpers.h"
+#include "p4lib/ip4.h"
+
+bool CCSCustomTextureSaver::Init()
+{
+	m_materials.RemoveAll();
+	m_bHasJob = false;
+	return true;
+}
+
+void CCSCustomTextureSaver::AddMaterialToWatch( CustomMaterialGenerationData_t embroiderMaterial )
+{
+	m_materials.AddToTail( embroiderMaterial );
+	m_bHasJob = true;
+}
+
+void CCSCustomTextureSaver::Update( float frameTime )
+{
+	if ( m_bHasJob == false )
+		return;
+
+	int count = m_materials.Count();
+	
+	if ( count == 0 )
+	{
+		m_bHasJob = false;
+		return;
+	}
+	
+	CUtlVector<CustomMaterialGenerationData_t> completedMaterials;
+
+	for ( int i = 0; i < count; ++i )
+	{
+		int completionCount = 0;
+
+		for ( int j = 0; j < m_materials[i].nTex; j++ )
+		{
+			completionCount += ( m_materials[i].pCustomMaterial->GetTexture( j )->GenerationComplete() );
+		}
+		if ( completionCount == 3 )
+		{
+			count--;
+			completedMaterials.AddToTail( m_materials[i] );
+			m_materials.Remove( i );
+			i--;
+		}
+	}
+
+	for ( int i = 0; i < completedMaterials.Count(); i++ )
+	{
+		for ( int j = 0; j < completedMaterials[i].nTex; j++ )
+		{
+			if ( g_pFullFileSystem->FileExists( completedMaterials[i].fileNames[j], "MOD" ) &&  !g_pFullFileSystem->IsFileWritable( completedMaterials[i].fileNames[j], "MOD" ) )
+			{
+				// Skip read only files
+				continue;
+			}
+
+			IVTFTexture *pVTFTexture = completedMaterials[i].pCustomMaterial->GetTexture( j )->GetResultVTF();
+
+			CUtlBuffer buf;
+			pVTFTexture->Serialize(buf);
+					
+			FileHandle_t f = g_pFullFileSystem->Open( completedMaterials[i].fileNames[j], "wb", NULL );
+
+			if ( f != FILESYSTEM_INVALID_HANDLE )
+			{
+				g_pFullFileSystem->Write( buf.Base(), buf.TellMaxPut(), f );
+				g_pFullFileSystem->Close(f);
+			}
+			DevMsg( "Saved VTF %s\n", completedMaterials[i].fileNames[j] );
+
+			char szFullVTFPath[ MAX_PATH ];
+			g_pFullFileSystem->RelativePathToFullPath( completedMaterials[i].fileNames[j], "MOD", szFullVTFPath, sizeof( szFullVTFPath ) );
+
+			g_p4factory->SetDummyMode( p4 == NULL );
+			g_p4factory->SetOpenFileChangeList( completedMaterials[i].pchChangeListName );
+			if ( !( p4->IsFileInPerforce( szFullVTFPath ) ) )
+				CP4AutoAddFile autop4_add( szFullVTFPath );
+			else
+				CP4AutoEditFile autop4_edit( szFullVTFPath );
+		}
+		// TODO: add saved files to perforce
+	}
+}
+
+