diff options
Diffstat (limited to 'src/server/game/Combat/ThreatManager.cpp')
-rw-r--r-- | src/server/game/Combat/ThreatManager.cpp | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/src/server/game/Combat/ThreatManager.cpp b/src/server/game/Combat/ThreatManager.cpp index 1c18292a295..e85ebd5dac2 100644 --- a/src/server/game/Combat/ThreatManager.cpp +++ b/src/server/game/Combat/ThreatManager.cpp @@ -20,6 +20,7 @@ #include "CombatPackets.h" #include "CreatureAI.h" #include "CreatureGroups.h" +#include "MapUtils.h" #include "MotionMaster.h" #include "ObjectAccessor.h" #include "Player.h" @@ -78,7 +79,7 @@ void ThreatReference::UpdateOffline() { _online = ShouldBeSuppressed() ? ONLINE_STATE_SUPPRESSED : ONLINE_STATE_ONLINE; HeapNotifyIncreased(); - _mgr.RegisterForAIUpdate(this); + _mgr.RegisterForAIUpdate(GetVictim()->GetGUID()); } } @@ -644,11 +645,12 @@ ThreatReference const* ThreatManager::ReselectVictim() void ThreatManager::ProcessAIUpdates() { CreatureAI* ai = ASSERT_NOTNULL(_owner->ToCreature())->AI(); - std::vector<ThreatReference const*> v(std::move(_needsAIUpdate)); // _needsAIUpdate is now empty in case this triggers a recursive call + std::vector<ObjectGuid> v(std::move(_needsAIUpdate)); // _needsAIUpdate is now empty in case this triggers a recursive call if (!ai) return; - for (ThreatReference const* ref : v) - ai->JustStartedThreateningMe(ref->GetVictim()); + for (ObjectGuid const& guid : v) + if (ThreatReference const* ref = Trinity::Containers::MapGetValuePtr(_myThreatListEntries, guid)) + ai->JustStartedThreateningMe(ref->GetVictim()); } // returns true if a is LOWER on the threat list than b |