Implement CheckE3Mode::doQuery

1 files changed, 14 insertions, 2 deletions

diff --git a/src/Game/AI/Query/queryCheckE3Mode.cpp b/src/Game/AI/Query/queryCheckE3Mode.cpp
index 45617654..88bb4056 100644
--- a/src/Game/AI/Query/queryCheckE3Mode.cpp
+++ b/src/Game/AI/Query/queryCheckE3Mode.cpp
@@ -1,5 +1,6 @@
 #include "Game/AI/Query/queryCheckE3Mode.h"
 #include <evfl/Query.h>
+#include "Game/E3Mgr.h"
 
 namespace uking::query {
 
@@ -7,9 +8,20 @@ CheckE3Mode::CheckE3Mode(const InitArg& arg) : ksys::act::ai::Query(arg) {}
 
 CheckE3Mode::~CheckE3Mode() = default;
 
-// FIXME: implement
 int CheckE3Mode::doQuery() {
-    return -1;
+    auto* e3m = E3Mgr::instance();
+    if (e3m != nullptr) {
+        if (e3m->isDemoMode0AndNotStageSelect()) {
+            return 1;
+        }
+        if (e3m->isDemoMode1AndNotStageSelect()) {
+            return 2;
+        }
+        if (e3m->isDemoMode2AndNotStageSelect()) {
+            return 3;
+        }
+    }
+    return 0;
 }
 
 void CheckE3Mode::loadParams(const evfl::QueryArg& arg) {}