Add GameState.TeamMembers() utility func

Author: markus-wa

demoinfocs_test.go

@@ -103,6 +103,14 @@ func TestDemoInfoCs(t *testing.T) {
 			// We know there should be 10 players at match start in the default demo
 			t.Error("Expected 10 players; got", nPlayers)
 		}
+		if nTerrorists := len(gs.TeamMembers(common.TeamTerrorists)); nTerrorists != 5 {
+			// We know there should be 5 terrorists at match start in the default demo
+			t.Error("Expected 5 terrorists; got", nTerrorists)
+		}
+		if nCTs := len(gs.TeamMembers(common.TeamCounterTerrorists)); nCTs != 5 {
+			// We know there should be 5 CTs at match start in the default demo
+			t.Error("Expected 5 CTs; got", nCTs)
+		}
 	})
 
 	// Net-message stuff

game_state.go

@@ -58,6 +58,17 @@ func (gs GameState) PlayingParticipants() []*common.Player {
 	return r
 }
 
+// TeamMembers returns all players belonging to the requested team.
+func (gs GameState) TeamMembers(team common.Team) []*common.Player {
+	r := make([]*common.Player, 0, len(gs.players))
+	for _, ptcp := range gs.players {
+		if ptcp.Team == team {
+			r = append(r, ptcp)
+		}
+	}
+	return r
+}
+
 func newGameState() GameState {
 	return GameState{players: make(map[int]*common.Player)}
 }