1HEADmaster

1 files changed, 171 insertions, 0 deletions

diff --git a/game/server/tf/tf_wartracker.cpp b/game/server/tf/tf_wartracker.cpp
new file mode 100644
index 0000000..260a1b4
--- /dev/null
+++ b/game/server/tf/tf_wartracker.cpp
@@ -0,0 +1,171 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: Tracker for War Data on a player
+//
+// $NoKeywords: $
+//=============================================================================//
+#include "cbase.h"
+#include "tf_gc_server.h"
+#include "tf_wartracker.h"
+#include "tf_gcmessages.pb.h"
+#include "tf_player.h"
+#include "gcsdk/gcconstants.h"
+#include "tf_gamerules.h"
+
+// memdbgon must be the last include file in a .cpp file!!!
+#include "tier0/memdbgon.h"
+
+#define WAR_ASSIST_POINTS 1
+#define WAR_KILL_POINTS 2
+#define WAR_WIN_POINTS 20
+
+#ifdef STAGING_ONLY
+ConVar tf_commit_every_war_point( "tf_commit_every_war_point", "0" );
+#endif
+
+
+CTFWarTrackerManager* GetWarTrackerManager()
+{
+	static CTFWarTrackerManager trackerManager;
+	return &trackerManager;
+}
+
+
+CTFWarTrackerManager::CTFWarTrackerManager()
+{}
+
+CSOTrackerManager::SOTrackerMap_t::KeyType_t CTFWarTrackerManager::GetKeyForObjectTracker( const CSharedObject* pItem, CSteamID steamIDOwner )
+{
+	return steamIDOwner.ConvertToUint64();
+}
+
+#ifdef GAME_DLL
+void CTFWarTrackerManager::SendMessageForCommit( const ::google::protobuf::Message* pProtoMessage ) const
+{
+	GCSDK::CProtoBufMsg< CGCMsgGC_War_IndividualUpdate > msg( k_EMsgGC_War_IndividualUpdate );
+	msg.Body() = *assert_cast< const CGCMsgGC_War_IndividualUpdate* >( pProtoMessage );
+	GCClientSystem()->BSendMessage( msg );
+}
+#endif
+
+CFmtStr CTFWarTrackerManager::GetDebugObjectDescription( const CSharedObject* pItem ) const
+{
+	const CWarData* pWarData = assert_cast< const CWarData* >( pItem );
+	const CWarDefinition* pWarDef = GetItemSchema()->GetWarDefinitionByIndex( pWarData->Obj().war_id() );
+	const CWarDefinition::CWarSideDefinition_t* pSide = pWarDef->GetSide( pWarData->Obj().affiliation() );
+	
+	return CFmtStr( "%s: Side: %s Points: %d", pWarDef->GetDefName(), pSide->m_pszLocalizedName, pWarData->Obj().points_scored() );
+}
+
+CBaseSOTracker* CTFWarTrackerManager::AllocateNewTracker( const CSharedObject* pItem, CSteamID steamIDOwner, CSOTrackerManager* pManager ) const
+{
+	return new CTFWarTracker( pItem, steamIDOwner, pManager );
+}
+
+::google::protobuf::Message* CTFWarTrackerManager::AllocateNewProtoMessage() const
+{
+	return new CGCMsgGC_War_IndividualUpdate();
+}
+
+void CTFWarTrackerManager::OnCommitRecieved( const ::google::protobuf::Message* pProtoMsg )
+{
+	// ...
+}
+
+bool CTFWarTrackerManager::ShouldTrackObject( const CSteamID & steamIDOwner, const CSharedObject *pObject ) const
+{
+	// We only want war data!
+	if ( pObject->GetTypeID() != CWarData::k_nTypeID )
+		return false;
+
+#ifdef CLIENT_DLL
+	if ( steamapicontext()->SteamUser()->GetSteamID() != steamIDOwner )
+		return false;
+#endif
+
+	const CWarData* pWarData = assert_cast< const CWarData* >( pObject );
+	const CWarDefinition* pWarDef = GetItemSchema()->GetWarDefinitionByIndex( pWarData->Obj().war_id() );
+	// Needs to be for a real war!
+	if ( !pWarDef )
+		return false;
+
+	if ( !pWarDef->IsActive() )
+		return false;
+
+	// Needs to be for a valid side!
+	if ( !pWarDef->IsValidSide( pWarData->Obj().affiliation() ) )
+		return false;
+
+	return true;
+}
+
+int CTFWarTrackerManager::CompareRecords( const ::google::protobuf::Message* pNewProtoMsg, const ::google::protobuf::Message* pExistingProtoMsg ) const
+{
+	const CGCMsgGC_War_IndividualUpdate* pNew = assert_cast< const CGCMsgGC_War_IndividualUpdate* >( pNewProtoMsg );
+	const CGCMsgGC_War_IndividualUpdate* pExisting = assert_cast< const CGCMsgGC_War_IndividualUpdate* >( pExistingProtoMsg );
+
+	return pNew->score() - pExisting->score();
+}
+
+CTFWarTracker::CTFWarTracker( const CSharedObject* pItem, CSteamID SteamIDOwner, CSOTrackerManager* pManager )
+	: CBaseSOTracker( pItem, SteamIDOwner, pManager )
+{
+	m_ProtoData.Clear();
+	m_ProtoData.set_steam_id( SteamIDOwner.ConvertToUint64() );
+	ListenForGameEvent( "player_score_changed" );
+}
+
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Send to the GC our points scored for Team Spy or Team Engy
+//-----------------------------------------------------------------------------
+void CTFWarTracker::CommitChangesToDB()
+{
+	// Nothing to update!
+ 	if ( m_ProtoData.score() == 0 )
+		return;
+
+	m_pManager->AddCommitRecord( &m_ProtoData, GetOwnerSteamID().ConvertToUint64(), false );
+
+	// Reset!  We're adding to our score on the GC, not setting it
+	m_ProtoData.set_score( 0 );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: Listen for player deaths and give points if the player we're tracking
+//			kills someone on the other team
+//-----------------------------------------------------------------------------
+void CTFWarTracker::FireGameEvent( IGameEvent *pEvent )
+{
+	// Only score points in competitive
+	if ( !TFGameRules()->IsCompetitiveMode() )
+		return;
+
+	const char *pszName = pEvent->GetName();
+
+	// You get a point when you kill, or assist in a kill, while in a comp game
+	if( FStrEq( pszName, "player_score_changed" ) )
+	{
+		const CTFPlayer *pScorer = ToTFPlayer( UTIL_PlayerByIndex( pEvent->GetInt( "player" ) ) );
+		const CTFPlayer *pOwner = GetTrackedPlayer();
+
+		if ( pScorer != pOwner )
+			return;
+
+		int nDelta = pEvent->GetInt( "delta", 0 );
+
+		// Only take positive changes
+		if ( nDelta <= 0 )
+			return;
+
+		m_ProtoData.set_score( m_ProtoData.score() + nDelta );
+	}
+
+#ifdef STAGING_ONLY
+	if ( tf_commit_every_war_point.GetBool() )
+	{
+		CommitChangesToDB();
+	}
+#endif
+}