Merge pull request #41 from notyourav/queries

match some ai::Query

1 files changed, 10 insertions, 2 deletions

diff --git a/src/Game/AI/Query/queryComparePlayerMaxStamina.cpp b/src/Game/AI/Query/queryComparePlayerMaxStamina.cpp
index 44706d71..fa2c8c54 100644
--- a/src/Game/AI/Query/queryComparePlayerMaxStamina.cpp
+++ b/src/Game/AI/Query/queryComparePlayerMaxStamina.cpp
@@ -1,5 +1,6 @@
 #include "Game/AI/Query/queryComparePlayerMaxStamina.h"
 #include <evfl/Query.h>
+#include "KingSystem/ActorSystem/actPlayerInfo.h"
 
 namespace uking::query {
 
@@ -7,9 +8,16 @@ ComparePlayerMaxStamina::ComparePlayerMaxStamina(const InitArg& arg) : ksys::act
 
 ComparePlayerMaxStamina::~ComparePlayerMaxStamina() = default;
 
-// FIXME: implement
 int ComparePlayerMaxStamina::doQuery() {
-    return -1;
+    auto* pi = ksys::act::PlayerInfo::instance();
+    if (pi == nullptr)
+        return 0;
+
+    s32 stamina = pi->getMaxStamina();
+    if (stamina % 200 <= 0)
+        return stamina / 200 >= *mThreshold;
+    else
+        return stamina / 200 + 1 >= *mThreshold;
 }
 
 void ComparePlayerMaxStamina::loadParams(const evfl::QueryArg& arg) {