Core/Combat: add new getter to count Players in the threatlist

1 files changed, 11 insertions, 0 deletions

diff --git a/src/server/game/Combat/ThreatManager.cpp b/src/server/game/Combat/ThreatManager.cpp
index 502031712c4..6852b792422 100644
--- a/src/server/game/Combat/ThreatManager.cpp
+++ b/src/server/game/Combat/ThreatManager.cpp
@@ -281,6 +281,17 @@ size_t ThreatManager::GetThreatListSize() const
     return _sortedThreatList->size();
 }
 
+uint32 ThreatManager::GetThreatListPlayerCount(bool includeOffline/* = false*/) const
+{
+    if (includeOffline)
+        return uint32(_sortedThreatList->size());
+    uint32 returnValue = 0;
+    for (ThreatReference const* ref : *_sortedThreatList)
+        if (ref->IsAvailable() && ref->GetOwner()->GetTypeId() == TYPEID_PLAYER)
+            ++returnValue;
+    return returnValue;
+}
+
 Trinity::IteratorPair<ThreatManager::ThreatListIterator, std::nullptr_t> ThreatManager::GetUnsortedThreatList() const
 {
     auto itr = _myThreatListEntries.begin();