mirror of
https://github.com/MaaAssistantArknights/MaaAssistantArknights.git
synced 2026-07-18 18:20:39 +08:00
77 lines
2.3 KiB
C++
77 lines
2.3 KiB
C++
#include "RecruitImageAnalyzer.h"
|
|
|
|
#include "MatchImageAnalyzer.h"
|
|
#include "MultiMatchImageAnalyzer.h"
|
|
#include "OcrImageAnalyzer.h"
|
|
#include "Resource.h"
|
|
|
|
bool asst::RecruitImageAnalyzer::analyze()
|
|
{
|
|
m_tags_result.clear();
|
|
|
|
time_analyze();
|
|
refresh_analyze();
|
|
bool ret = tags_analyze();
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool asst::RecruitImageAnalyzer::tags_analyze()
|
|
{
|
|
static bool analyzer_inited = false;
|
|
static OcrImageAnalyzer tags_analyzer;
|
|
if (!analyzer_inited) {
|
|
const auto tags_task_ptr = Task.get<OcrTaskInfo>("RecruitTags");
|
|
tags_analyzer.set_roi(tags_task_ptr->roi);
|
|
auto& all_tags_set = Resrc.recruit().get_all_tags();
|
|
std::vector<std::string> all_tags_vec;
|
|
all_tags_vec.assign(all_tags_set.begin(), all_tags_set.end());
|
|
// 把因为“资深干员”是“高级资深干员”的子串，把“高级资深干员”放到最前面，免得先被“资深干员”匹配上了
|
|
auto ssr_iter = ranges::find(all_tags_vec, "高级资深干员");
|
|
if (ssr_iter != all_tags_vec.end()) {
|
|
std::swap(*ssr_iter, all_tags_vec.front());
|
|
}
|
|
|
|
tags_analyzer.set_required(std::move(all_tags_vec));
|
|
tags_analyzer.set_replace(tags_task_ptr->replace_map);
|
|
analyzer_inited = true;
|
|
}
|
|
|
|
tags_analyzer.set_image(m_image);
|
|
|
|
if (tags_analyzer.analyze()) {
|
|
m_tags_result = tags_analyzer.get_result();
|
|
return true;
|
|
//if (m_tags_result.size() == Resrc.recruit().CorrectNumberOfTags) {
|
|
// return true;
|
|
//}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool asst::RecruitImageAnalyzer::time_analyze()
|
|
{
|
|
MultiMatchImageAnalyzer decrement_a(m_image);
|
|
decrement_a.set_task_info("RecruitTimerDecrement");
|
|
if (!decrement_a.analyze()) return false;
|
|
if (decrement_a.get_result().size() != 2) return false; // expecting two buttons
|
|
decrement_a.sort_result_horizontal();
|
|
m_hour_decrement = decrement_a.get_result().at(0).rect;
|
|
m_minute_decrement = decrement_a.get_result().at(1).rect;
|
|
return true;
|
|
}
|
|
|
|
bool asst::RecruitImageAnalyzer::refresh_analyze()
|
|
{
|
|
MatchImageAnalyzer refresh_analyzer(m_image);
|
|
|
|
refresh_analyzer.set_task_info("RecruitRefresh");
|
|
|
|
if (refresh_analyzer.analyze()) {
|
|
m_refresh_rect = refresh_analyzer.get_result().rect;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|