match some ai::Query

1 files changed, 7 insertions, 2 deletions

diff --git a/src/Game/AI/Query/queryCheckHorseTakenOut.cpp b/src/Game/AI/Query/queryCheckHorseTakenOut.cpp
index fbaa8bed..d1b25b35 100644
--- a/src/Game/AI/Query/queryCheckHorseTakenOut.cpp
+++ b/src/Game/AI/Query/queryCheckHorseTakenOut.cpp
@@ -1,5 +1,6 @@
 #include "Game/AI/Query/queryCheckHorseTakenOut.h"
 #include <evfl/Query.h>
+#include "KingSystem/GameData/gdtSpecialFlags.h"
 
 namespace uking::query {
 
@@ -7,9 +8,13 @@ CheckHorseTakenOut::CheckHorseTakenOut(const InitArg& arg) : ksys::act::ai::Quer
 
 CheckHorseTakenOut::~CheckHorseTakenOut() = default;
 
-// FIXME: implement
 int CheckHorseTakenOut::doQuery() {
-    return -1;
+    s32 active = ksys::gdt::getS32ByKey("Horse_ActiveIndex");
+    if (active < 0)
+        return 2;
+
+    s32 selected = ksys::gdt::getS32ByKey("Horse_SelectedIndex");
+    return active == selected;
 }
 
 void CheckHorseTakenOut::loadParams(const evfl::QueryArg& arg) {}