query: Implement CompareGameDataFloat

2 files changed, 31 insertions, 3 deletions

diff --git a/src/Game/AI/Query/queryCompareGameDataFloat.cpp b/src/Game/AI/Query/queryCompareGameDataFloat.cpp
index 2cf01e62..55055b46 100644
--- a/src/Game/AI/Query/queryCompareGameDataFloat.cpp
+++ b/src/Game/AI/Query/queryCompareGameDataFloat.cpp
@@ -1,5 +1,7 @@
 #include "Game/AI/Query/queryCompareGameDataFloat.h"
 #include <evfl/query.h>
+#include <functional>
+#include "KingSystem/GameData/gdtManager.h"
 
 namespace uking::query {
 
@@ -7,9 +9,35 @@ CompareGameDataFloat::CompareGameDataFloat(const InitArg& arg) : ksys::act::ai::
 
 CompareGameDataFloat::~CompareGameDataFloat() = default;
 
-// FIXME: implement
 int CompareGameDataFloat::doQuery() {
-    return -1;
+    float flag_value_a = 0.0;
+    float flag_value_b = 0.0;
+
+    auto* gdt = ksys::gdt::Manager::instance();
+    if (!gdt)
+        return 0;
+    if (!gdt->getParamBypassPerm().get().getF32(&flag_value_a, mGameDataFloatName_A))
+        return 0;
+    if (!gdt->getParamBypassPerm().get().getF32(&flag_value_b, mGameDataFloatName_B))
+        return 0;
+
+    const auto invert = std::logical_not<>();
+
+    sead::FixedSafeString<32> op = mOperator;
+    if (op == "Equal")
+        return invert(flag_value_a == flag_value_b);
+    if (op == "NotEqual")
+        return invert(flag_value_a != flag_value_b);
+    if (op == "GreaterThan")
+        return invert(flag_value_a > flag_value_b);
+    if (op == "GreaterThanOrEqualTo")
+        return invert(flag_value_a >= flag_value_b);
+    if (op == "LessThan")
+        return invert(flag_value_a < flag_value_b);
+    if (op == "LessThanOrEqualTo")
+        return invert(flag_value_a <= flag_value_b);
+
+    return 0;
 }
 
 void CompareGameDataFloat::loadParams(const evfl::QueryArg& arg) {
diff --git a/src/Game/AI/Query/queryCompareGameDataFloat.h b/src/Game/AI/Query/queryCompareGameDataFloat.h
index 35f7ed46..3dfb9124 100644
--- a/src/Game/AI/Query/queryCompareGameDataFloat.h
+++ b/src/Game/AI/Query/queryCompareGameDataFloat.h
@@ -16,8 +16,8 @@ public:
 
 protected:
     sead::SafeString mGameDataFloatName_A{};
-    sead::SafeString mGameDataFloatName_B{};
     sead::SafeString mOperator{};
+    sead::SafeString mGameDataFloatName_B{};
 };
 
 }  // namespace uking::query