diff --git a/resource/tasks.json b/resource/tasks.json index cee0580c39..5054de8a59 100644 --- a/resource/tasks.json +++ b/resource/tasks.json @@ -12752,6 +12752,12 @@ 335, 150, 60 + ], + "ocrReplace": [ + [ + ">", + "" + ] ] }, "Roguelike@StageTraderInvest-Count-Error": { diff --git a/src/MaaCore/Task/Roguelike/RoguelikeInvestTaskPlugin.cpp b/src/MaaCore/Task/Roguelike/RoguelikeInvestTaskPlugin.cpp index 65a7413551..e15dc2d2fe 100644 --- a/src/MaaCore/Task/Roguelike/RoguelikeInvestTaskPlugin.cpp +++ b/src/MaaCore/Task/Roguelike/RoguelikeInvestTaskPlugin.cpp @@ -113,6 +113,13 @@ std::optional asst::RoguelikeInvestTaskPlugin::ocr_current_count(const auto { RegionOCRer ocr(img); ocr.set_task_info(task_name); + + const auto& number_replace = Task.get("NumberOcrReplace")->replace_map; + auto task_replace = Task.get(task_name)->replace_map; + + auto merge_map = std::vector(number_replace); + ranges::copy(task_replace, std::back_inserter(merge_map)); + ocr.set_replace(merge_map); if (!ocr.analyze()) { Log.error(__FUNCTION__, "unable to analyze current investment count."); return std::nullopt;