From 67c84b1f3d74985fd2521feb9e536e56b9c9f2e9 Mon Sep 17 00:00:00 2001
From: megamage <none@none>
Date: Thu, 21 May 2009 10:48:00 -0500
Subject: Mobs fleeing and getting assistance feature implementaion. Author:
 Neo2003

--HG--
branch : trunk
---
 src/game/IdleMovementGenerator.cpp | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

(limited to 'src/game/IdleMovementGenerator.cpp')

diff --git a/src/game/IdleMovementGenerator.cpp b/src/game/IdleMovementGenerator.cpp
index 2428c49a77a..44575302e75 100644
--- a/src/game/IdleMovementGenerator.cpp
+++ b/src/game/IdleMovementGenerator.cpp
@@ -19,7 +19,8 @@
  */
 
 #include "IdleMovementGenerator.h"
-#include "Unit.h"
+#include "CreatureAI.h"
+#include "Creature.h"
 
 IdleMovementGenerator si_idleMovement;
 
@@ -60,3 +61,16 @@ DistractMovementGenerator::Update(Unit& owner, const uint32& time_diff)
     return true;
 }
 
+void
+AssistanceDistractMovementGenerator::Finalize(Unit &unit)
+{
+    unit.clearUnitState(UNIT_STAT_DISTRACTED);
+    if (Unit* victim = unit.getVictim())
+    {
+        if (unit.isAlive())
+        {
+            unit.AttackStop();
+            ((Creature*)&unit)->AI()->AttackStart(victim);
+        }
+    }
+}
-- 
cgit v1.2.3