feat(asw_player): add spectate in order

Author: anf3is

src/game/server/swarm/asw_player.cpp

@@ -57,6 +57,7 @@
 #include "missionchooser/iasw_mission_chooser_source.h"
 #include "rd_vgui_vscript_shared.h"
 #include "rd_crafting_defs.h"
+#include <algorithm>
 
 // memdbgon must be the last include file in a .cpp file!!!
 #include "tier0/memdbgon.h"
@@ -2246,8 +2247,87 @@ void CASW_Player::SetSpectatingOrder( const int* iProfiles, int nProfiles )
 	m_bSpectatingInOrder = true;
 }
 
+bool CASW_Player::CompareMarinesSpectatingPriority( CASW_Marine* pM1, CASW_Marine* pM2 ) const
+{
+	// Assign worst priority if no marine or no profile index
+	const int iProfile1 = pM1 ? pM1->GetMarineProfile()->m_ProfileIndex : INT_MAX;
+	const int iProfile2 = pM2 ? pM2->GetMarineProfile()->m_ProfileIndex : INT_MAX;
+
+	const int iPriority1 = ( iProfile1 >= 0 && iProfile1 < ASW_NUM_MARINE_PROFILES ) ? m_iSpectatingMapping[iProfile1] : INT_MAX;
+	const int iPriority2 = ( iProfile2 >= 0 && iProfile2 < ASW_NUM_MARINE_PROFILES ) ? m_iSpectatingMapping[iProfile2] : INT_MAX;
+
+	return iPriority1 < iPriority2;
+}
+
+void CASW_Player::SpectateNextMarineInOrder()
+{
+	CASW_Game_Resource* pGameResource = ASWGameResource();
+	if ( !pGameResource )
+		return;
+
+	int iSortedCount = 0;
+	CASW_Marine* pMarinesSorted[ASW_MAX_MARINE_RESOURCES] = { NULL };
+
+	// Gather all valid marines
+	for ( int i = 0; i < pGameResource->GetMaxMarineResources(); i++ )
+	{
+		CASW_Marine_Resource* pMR = pGameResource->GetMarineResource( i );
+		CASW_Marine *pMarine = pMR ? pMR->GetMarineEntity() : NULL;
+		if ( pMarine && pMarine->IsAlive() && pMarine->GetHealth() > 0 )
+		{
+			pMarinesSorted[iSortedCount] = pMarine;
+			iSortedCount++;
+		}
+	}
+
+	std::sort(
+		pMarinesSorted,
+		pMarinesSorted + iSortedCount,
+		[this](CASW_Marine* a, CASW_Marine* b) {
+			return CompareMarinesSpectatingPriority(a, b);
+		}
+	);
+
+	//Msg("CASW_Player::SpectateNextMarineInOrder\n");
+
+	//Msg("  set first guy as our first\n");
+	CASW_Marine *pFirst = pMarinesSorted[0];
+
+	// loop through all valid marines
+	for ( int i = 0; i < iSortedCount; i++ )
+	{
+		//Msg("Checking pMR %d\n", i);
+		CASW_Marine *pMarine = pMarinesSorted[i]; // Alive marine
+
+		if (GetSpectatingNPC() == NULL)		// if we're not spectating anything yet, then spectate the first one we find
+		{
+			//Msg("  We're not spectating anyone, so we're gonna spec this dude\n");
+			SetSpectatingNPC(pMarine);
+			break;
+		}
+		if (GetSpectatingNPC() == pMarine)	// if we're spectating this one, then clear it, so the next one we find will get set
+		{
+			//Msg("  we're spectating this dude, so clearing our current spectator\n");
+			SetSpectatingNPC(NULL);
+		}
+	}
+	//Msg("end\n");
+	// if we're still not spectating anything but we found at least marine, then that means we were spectating the last one in the list and need to set this
+	if (GetSpectatingNPC() == NULL && pFirst)
+	{
+		//Msg("  but we're still not speccing anyone and we have a first set, so speccing that dude\n");
+		SetSpectatingNPC(pFirst);
+	}
+}
+
 void CASW_Player::SpectateNextMarine()
 {
+	if ( m_bSpectatingInOrder )
+	{
+		SpectateNextMarineInOrder();
+		return;
+	}
+
 	CASW_Game_Resource* pGameResource = ASWGameResource();
 	if (!pGameResource)
 		return;
@@ -2282,7 +2362,7 @@ void CASW_Player::SpectateNextMarine()
 		{
 			//Msg("  we're spectating this dude, so clearing our current spectator\n");
 			SetSpectatingNPC(NULL);
-		}				
+		}
 	}
 	//Msg("end\n");
 	// if we're still not spectating anything but we found at least marine, then that means we were spectating the last one in the list and need to set this

src/game/server/swarm/asw_player.h

@@ -90,8 +90,10 @@ class CASW_Player : public CBaseMultiplayerPlayer, public IASWPlayerAnimStateHel
 	bool m_bSpectatingInOrder = false;
 	int m_iSpectatingMapping[ASW_NUM_MARINE_PROFILES];
 public:
+	bool CompareMarinesSpectatingPriority( CASW_Marine* pM1, CASW_Marine* pM2 ) const;
 	void UnsetSpectatingOrder() { m_bSpectatingInOrder = false; }
 	void SetSpectatingOrder( const int* iProfiles, int nProfiles );
+	void SpectateNextMarineInOrder();
 	void SpectateNextMarine();
 	void SetSpectatingNPC( CASW_Inhabitable_NPC *pSpectating );
 	CASW_Inhabitable_NPC *GetSpectatingNPC() const;