match more ai::Query

1 files changed, 6 insertions, 2 deletions

diff --git a/src/Game/AI/Query/queryBranchByDyeColor.cpp b/src/Game/AI/Query/queryBranchByDyeColor.cpp
index 085367f0..30cfe43e 100644
--- a/src/Game/AI/Query/queryBranchByDyeColor.cpp
+++ b/src/Game/AI/Query/queryBranchByDyeColor.cpp
@@ -1,5 +1,6 @@
 #include "Game/AI/Query/queryBranchByDyeColor.h"
 #include <evfl/Query.h>
+#include "KingSystem/GameData/gdtCommonFlagsUtils.h"
 
 namespace uking::query {
 
@@ -7,9 +8,12 @@ BranchByDyeColor::BranchByDyeColor(const InitArg& arg) : ksys::act::ai::Query(ar
 
 BranchByDyeColor::~BranchByDyeColor() = default;
 
-// FIXME: implement
 int BranchByDyeColor::doQuery() {
-    return -1;
+    auto index = ksys::gdt::getFlag_ColorChange_MaterialIndex();
+    if (index >= 1 && index <= 15) {
+        return index;
+    }
+    return 0;
 }
 
 void BranchByDyeColor::loadParams(const evfl::QueryArg& arg) {}