1HEADmaster

1 files changed, 217 insertions, 0 deletions

diff --git a/game/shared/tf/tf_rating_data.cpp b/game/shared/tf/tf_rating_data.cpp
new file mode 100644
index 0000000..b9c0ce0
--- /dev/null
+++ b/game/shared/tf/tf_rating_data.cpp
@@ -0,0 +1,217 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+
+#include "cbase.h"
+
+#include "tf_rating_data.h"
+#include "gcsdk/enumutils.h"
+
+#ifdef CLIENT_DLL
+#include "tf_matchmaking_shared.h"
+#endif
+
+using namespace GCSDK;
+
+#if defined( GC )
+#include "tf_matchmaking_rating.h"
+
+IMPLEMENT_CLASS_MEMPOOL( CTFRatingData, 1000, UTLMEMORYPOOL_GROW_SLOW );
+// Make sure you visit all these if you add new fields to rating data
+#define ASSERT_LAST_FIELD( foo ) COMPILE_TIME_ASSERT( CSchRatingData::k_iFieldMax == CSchRatingData::k_iField_##foo + 1 )
+#else // defined( GC )
+#define ASSERT_LAST_FIELD( foo ) // No Sch on client. As long as it fails somewhere.
+#endif // defined( GC )
+
+// memdbgon must be the last include file in a .cpp file!!!
+#include "tier0/memdbgon.h"
+
+//-----------------------------------------------------------------------------
+// Purpose: Get player's rating data by steamID.  Returns NULL if it doesn't exist
+//-----------------------------------------------------------------------------
+CTFRatingData *CTFRatingData::YieldingGetPlayerRatingDataBySteamID( const CSteamID &steamID, EMMRating eRatingType )
+{
+#ifdef GC_DLL
+	GCSDK::CGCSharedObjectCache *pSOCache = GGCEcon()->YieldingFindOrLoadSOCache( steamID );
+#else
+	GCSDK::CGCClientSharedObjectCache *pSOCache = GCClientSystem()->GetSOCache( steamID );
+#endif
+	if ( pSOCache )
+	{
+		auto *pTypeCache = pSOCache->FindTypeCache( CTFRatingData::k_nTypeID );
+		if ( pTypeCache )
+		{
+			for ( uint32 i = 0; i < pTypeCache->GetCount(); ++i )
+			{
+				CTFRatingData *pRatingData = (CTFRatingData*)pTypeCache->GetObject( i );
+				if ( eRatingType == (EMMRating)pRatingData->Obj().rating_type() )
+				{
+					return pRatingData;
+				}
+			}
+		}
+	}
+
+	return nullptr;
+}
+
+//-----------------------------------------------------------------------------
+#ifdef GC
+CTFRatingData::CTFRatingData( uint32 unAccountID, EMMRating eRatingType )
+	: CTFRatingData( unAccountID, eRatingType, ITFMMRatingBackend::GetRatingBackend( eRatingType ).GetDefaultRatingData() )
+{}
+#endif
+
+//-----------------------------------------------------------------------------
+CTFRatingData::CTFRatingData()
+{}
+
+//-----------------------------------------------------------------------------
+CTFRatingData::CTFRatingData( uint32 unAccountID, EMMRating eRatingType, const MMRatingData_t &ratingData )
+{
+	MutObj().set_account_id( unAccountID );
+	MutObj().set_rating_type( eRatingType );
+
+	MutObj().set_rating_primary( ratingData.unRatingPrimary );
+	MutObj().set_rating_secondary( ratingData.unRatingSecondary );
+	MutObj().set_rating_tertiary( ratingData.unRatingTertiary );
+	// MMRatingData_t fields
+	ASSERT_LAST_FIELD( unRatingTertiary );
+}
+
+#ifdef GC
+
+//-----------------------------------------------------------------------------
+bool CTFRatingData::BIsKeyLess( const CSharedObject & soRHS ) const
+{
+	Assert( GetTypeID() == soRHS.GetTypeID() );
+	const CSOTFRatingData &obj = Obj();
+	const CSOTFRatingData &rhs = ( static_cast< const CTFRatingData & >( soRHS ) ).Obj();
+
+	if ( obj.account_id() < rhs.account_id() ) return true;
+	if ( obj.account_id() > rhs.account_id() ) return false;
+
+	return obj.rating_type() < rhs.rating_type();
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: This is a database backed object, but, all mutations to these
+//          should go through BYieldingUpdateRatingAndAddToTransaction and its
+//          CTFSharedObjectCache user.
+//
+// Not implementing to prevent mis-use until there's a path that legitimately
+// needs them to work.
+//-----------------------------------------------------------------------------
+bool CTFRatingData::BYieldingAddInsertToTransaction( GCSDK::CSQLAccess & sqlAccess )
+	{ Assert( false ); return false; }
+bool CTFRatingData::BYieldingAddWriteToTransaction( GCSDK::CSQLAccess & sqlAccess, const CUtlVector< int > &fields )
+	{ Assert( false ); return false; }
+bool CTFRatingData::BYieldingAddRemoveToTransaction( GCSDK::CSQLAccess & sqlAccess )
+	{ Assert( false ); return false; }
+
+//-----------------------------------------------------------------------------
+void CTFRatingData::WriteToRecord( CSchRatingData *pRatingData ) const
+{
+	pRatingData->m_unAccountID = Obj().account_id();
+	pRatingData->m_nRatingType = Obj().rating_type();
+	pRatingData->m_unRatingPrimary = Obj().rating_primary();
+	pRatingData->m_unRatingSecondary = Obj().rating_secondary();
+	pRatingData->m_unRatingTertiary = Obj().rating_tertiary();
+	ASSERT_LAST_FIELD( unRatingTertiary );
+}
+
+//-----------------------------------------------------------------------------
+void CTFRatingData::ReadFromRecord( const CSchRatingData &ratingData )
+{
+	MutObj().set_account_id( ratingData.m_unAccountID );
+	MutObj().set_rating_type( ratingData.m_nRatingType );
+	MutObj().set_rating_primary( ratingData.m_unRatingPrimary );
+	MutObj().set_rating_secondary( ratingData.m_unRatingSecondary );
+	MutObj().set_rating_tertiary( ratingData.m_unRatingTertiary );
+	ASSERT_LAST_FIELD( unRatingTertiary );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: Hook for CTFSubscriberMessageFilter to allow us to stop certain
+//          rating types from going to certain subscribers.
+//-----------------------------------------------------------------------------
+uint32_t CTFRatingData::FilterSendFlags( uint32_t unTypeFlags ) const
+{
+	auto &backend = ITFMMRatingBackend::GetRatingBackend( (EMMRating)Obj().rating_type() );
+	ITFMMRatingBackend::ERatingFlags eBackendFlags = backend.GetRatingFlags();
+
+	bool bNetworked = ( eBackendFlags & ITFMMRatingBackend::eRatingFlag_Networked );
+	bool bOwnerOnly = ( eBackendFlags & ITFMMRatingBackend::eRatingFlag_OwnerOnly );
+
+	uint32_t unRemoveFlags = 0u;
+
+	if ( !bNetworked || bOwnerOnly )
+	{
+		unRemoveFlags |= k_ESOFlag_SendToOtherGameservers;
+		unRemoveFlags |= k_ESOFlag_SendToOtherUsers;
+	}
+
+	if ( !bNetworked )
+	{
+		unRemoveFlags |= k_ESOFlag_SendToOwner;
+	}
+
+	return unTypeFlags & ~unRemoveFlags;
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: Updates the rating fields from an MMRatingData_t, add a write or
+//          insert to the given transaction including an audit entry.
+//-----------------------------------------------------------------------------
+bool CTFRatingData::BYieldingUpdateRatingAndAddToTransaction( CSQLAccess &transaction,
+                                                              const MMRatingData_t &newData,
+                                                              EMMRatingSource eAdjustmentSource,
+                                                              uint64_t unSourceEventID,
+                                                              RTime32 rtTimestamp )
+{
+	// MMRatingData_t fields
+	ASSERT_LAST_FIELD( unRatingTertiary );
+	CSchRatingHistory schHistory;
+	schHistory.m_unAccountID          = Obj().account_id();
+	schHistory.m_nRatingType          = Obj().rating_type();
+	schHistory.m_nRatingSource        = eAdjustmentSource;
+	schHistory.m_unEventID            = unSourceEventID;
+	schHistory.m_rtime32Timestamp     = rtTimestamp;
+	schHistory.m_unOldRatingPrimary   = Obj().rating_primary();
+	schHistory.m_unOldRatingSecondary = Obj().rating_secondary();
+	schHistory.m_unOldRatingTertiary  = Obj().rating_tertiary();
+	schHistory.m_unNewRatingPrimary   = newData.unRatingPrimary;
+	schHistory.m_unNewRatingSecondary = newData.unRatingSecondary;
+	schHistory.m_unNewRatingTertiary  = newData.unRatingTertiary;
+
+	// Update ourself
+	// MMRatingData_t fields
+	ASSERT_LAST_FIELD( unRatingTertiary );
+	MutObj().set_rating_primary( newData.unRatingPrimary );
+	MutObj().set_rating_secondary( newData.unRatingSecondary );
+	MutObj().set_rating_tertiary( newData.unRatingTertiary );
+
+	// Create record
+	CSchRatingData schRatingData;
+	WriteToRecord( &schRatingData );
+
+	// Add to transaction
+	return transaction.BYieldingInsertOrUpdateOnPK( &schRatingData ) &&
+	       transaction.BYieldingInsertRecord( &schHistory );
+}
+
+#endif	// GC
+
+//-----------------------------------------------------------------------------
+// Purpose: Reads out the rating fields for rating systems to use
+//-----------------------------------------------------------------------------
+MMRatingData_t CTFRatingData::GetRatingData()
+{
+	MMRatingData_t rvoData;
+	rvoData.unRatingPrimary = Obj().rating_primary();
+	rvoData.unRatingSecondary = Obj().rating_secondary();
+	rvoData.unRatingTertiary = Obj().rating_tertiary();
+
+	// If you expanded MMRatingData_t
+	ASSERT_LAST_FIELD( unRatingTertiary );
+
+	return rvoData;
+}