From ee6108b63254cb911a2e207b7fdb5ec53ede3675 Mon Sep 17 00:00:00 2001
From: aozhiwei <aozhiwei@kingsome.cn>
Date: Wed, 29 Jul 2020 14:51:27 +0800
Subject: [PATCH] 1

---
 server/gameserver/aicomponent.cc   | 9 +++++++++
 server/gameserver/aicomponent.h    | 3 +++
 server/gameserver/zombiemode.ai.cc | 2 +-
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/server/gameserver/aicomponent.cc b/server/gameserver/aicomponent.cc
index a8f6c68..4c1d816 100644
--- a/server/gameserver/aicomponent.cc
+++ b/server/gameserver/aicomponent.cc
@@ -17,8 +17,17 @@ void AIComponent::SetAiLevel(int ai_level)
     ai_level_ = ai_level;
 }
 
+void AIComponent::SetAiMode(int ai_mode)
+{
+    ai_mode_ = ai_mode;
+}
+
 int AIComponent::GetAiLevel()
 {
     return ai_level_;
 }
 
+int AIComponent::GetAiMode()
+{
+    return ai_mode_;
+}
diff --git a/server/gameserver/aicomponent.h b/server/gameserver/aicomponent.h
index fba3938..526be0d 100644
--- a/server/gameserver/aicomponent.h
+++ b/server/gameserver/aicomponent.h
@@ -10,9 +10,12 @@ class AIComponent
     virtual void Update(int delta_time);
     virtual float GetAttackRate() { return 0;};
     void SetAiLevel(int ai_level);
+    void SetAiMode(int ai_mode);
     int GetAiLevel();
+    int GetAiMode();
 
 private:
     int ai_level_ = 0;
+    int ai_mode_ = 0;
 
 };
diff --git a/server/gameserver/zombiemode.ai.cc b/server/gameserver/zombiemode.ai.cc
index 4ecb8de..db8afb6 100644
--- a/server/gameserver/zombiemode.ai.cc
+++ b/server/gameserver/zombiemode.ai.cc
@@ -97,7 +97,7 @@ void ZombieModeAI::UpdateAI()
         return;
     }
     if (!node_->ai_meta && GetAiLevel() != 0) {
-        node_->ai_meta = MetaMgr::Instance()->GetAI(GetAiLevel(), 1);
+        node_->ai_meta = MetaMgr::Instance()->GetAI(GetAiLevel(), GetAiMode());
         if (!node_->ai_meta) {
             abort();
         }