1HEADmaster

1 files changed, 124 insertions, 0 deletions

diff --git a/game/shared/replay_gamestats_shared.cpp b/game/shared/replay_gamestats_shared.cpp
new file mode 100644
index 0000000..60b6dae
--- /dev/null
+++ b/game/shared/replay_gamestats_shared.cpp
@@ -0,0 +1,124 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+//=============================================================================
+
+#include "cbase.h"
+#include "replay_gamestats_shared.h"
+#include "steamworks_gamestats.h"
+
+#if defined( CLIENT_DLL )
+#include "../client/replay/genericclassbased_replay.h"
+#include "replay/replayvideo.h"
+#include "replay/vgui/replaybrowserrenderdialog.h"
+#endif
+
+#include "replay/rendermovieparams.h"
+#include "replay/performance.h"
+
+// NOTE: This has to be the last file included!
+#include "tier0/memdbgon.h"
+
+
+CReplayGameStatsHelper::CReplayGameStatsHelper()
+{
+}
+
+void CReplayGameStatsHelper::UploadError( KeyValues *pData, bool bIncludeTimeField )
+{
+	if ( bIncludeTimeField )
+	{
+//		pData->SetInt( "Time", GetSteamWorksSGameStatsUploader().GetTimeSinceEpoch() );
+	}
+
+	GetSteamWorksSGameStatsUploader().AddStatsForUpload( pData );
+}
+
+CReplayGameStatsHelper &GetReplayGameStatsHelper()
+{
+	static CReplayGameStatsHelper s_Instance;
+	return s_Instance;
+}
+
+#if defined( CLIENT_DLL )
+void CReplayGameStatsHelper::SW_ReplayStats_WriteRenderDataStart( const RenderMovieParams_t& RenderParams, const CReplayRenderDialog *pDlg )
+{
+	SW_ReplayStats_WriteRenderData( true, RenderParams, pDlg, NULL );
+}
+
+void CReplayGameStatsHelper::SW_ReplayStats_WriteRenderDataEnd( const RenderMovieParams_t& RenderParams, const char *pEndReason )
+{
+	SW_ReplayStats_WriteRenderData( false, RenderParams, NULL, pEndReason );
+}
+
+void CReplayGameStatsHelper::SW_ReplayStats_WriteRenderData( bool bStarting, const RenderMovieParams_t& RenderParams,
+															 const CReplayRenderDialog *pDlg, const char *pEndReason/*=NULL*/ )
+{
+#if !defined( NO_STEAM )
+
+#if defined( REPLAY_ENABLED )
+	static uint32 s_iReplayRenderCounter = 0;
+
+	ReplayHandle_t hReplay = RenderParams.m_hReplay;
+	CGenericClassBasedReplay *pReplay = GetGenericClassBasedReplay( hReplay );
+	if ( !pReplay )
+		return;
+
+	KeyValues* pKVData = new KeyValues( "TF2ReplayRenders" );
+
+	// Base settings/DB key.
+	pKVData->SetInt( "RenderCounter", s_iReplayRenderCounter++ );
+	pKVData->SetInt( "ReplayHandle", hReplay );
+	pKVData->SetInt( "PeformanceIndex", RenderParams.m_iPerformance );
+
+	// Dialog-specific
+	if ( pDlg )
+	{
+		pKVData->SetInt( "ShowAdvancedChecked", pDlg->m_pShowAdvancedOptionsCheck->IsSelected() );
+		pKVData->SetString( "CodecID", ReplayVideo_GetCodec( pDlg->m_pCodecCombo->GetActiveItem() ).m_pName );
+		pKVData->SetInt( "RenderQualityPreset", pDlg->m_iQualityPreset );
+	}
+
+	// Render settings
+	CFmtStr fmtResolution( "%ix%i", RenderParams.m_Settings.m_nWidth, RenderParams.m_Settings.m_nHeight );
+	pKVData->SetInt( "QuitWhenDoneChecked", RenderParams.m_bQuitWhenFinished );
+	pKVData->SetString( "ResolutionID", fmtResolution.Access() );
+	pKVData->SetInt( "AAEnabled", RenderParams.m_Settings.m_bAAEnabled );
+	pKVData->SetInt( "MotionBlurEnabled", RenderParams.m_Settings.m_bMotionBlurEnabled );
+	pKVData->SetInt( "MotionBlurQuality", RenderParams.m_Settings.m_nMotionBlurQuality );
+	pKVData->SetInt( "RenderQuality", RenderParams.m_Settings.m_nEncodingQuality);
+	pKVData->SetInt( "ExportRawChecked", RenderParams.m_bExportRaw );
+	pKVData->SetInt( "FPSUPF", RenderParams.m_Settings.m_FPS.GetUnitsPerFrame() );
+	pKVData->SetInt( "FPSUPS", RenderParams.m_Settings.m_FPS.GetUnitsPerSecond() );
+
+	// Replay content.
+	pKVData->SetString( "MapID", pReplay->GetMapName() );
+	pKVData->SetString( "PlayerClassID", pReplay->GetPlayerClass() );
+	pKVData->SetInt( "ReplayLengthRealtime", pReplay->GetDeathTick() - pReplay->GetSpawnTick() );
+
+	CReplayPerformance *pPerformance = pReplay->GetPerformance( RenderParams.m_iPerformance );
+	if ( pPerformance )
+	{
+		int iPerformanceStartTick = pPerformance->HasInTick() ? pPerformance->GetTickIn() : pReplay->GetSpawnTick(),
+			iPeformanceEndTick	  = pPerformance->HasOutTick() ? pPerformance->GetTickOut() : pReplay->GetDeathTick();
+		pKVData->SetInt( "TakeLengthRealtime", iPeformanceEndTick - iPerformanceStartTick );
+	}
+	else
+	{
+		pKVData->SetInt( "TakeLengthRealtime", pReplay->GetDeathTick() - pReplay->GetSpawnTick() );
+	}
+
+	// Start or end time
+	pKVData->SetInt( bStarting ? "StartRenderTime" : "EndRenderTime", GetSteamWorksSGameStatsUploader().GetTimeSinceEpoch() );
+
+	// Write end reason
+	if ( !bStarting && pEndReason )
+	{
+		pKVData->SetString( "EndRenderReasonID", pEndReason );
+	}
+
+	GetSteamWorksSGameStatsUploader().AddStatsForUpload( pKVData );
+#endif // REPLAY_ENABLED
+
+#endif	// !defined( NO_STEAM )
+}
+#endif // defined( CLIENT_DLL )