mirror of
https://github.com/MaaAssistantArknights/MaaAssistantArknights.git
synced 2026-07-16 09:50:40 +08:00
128 lines
3.9 KiB
C++
128 lines
3.9 KiB
C++
#include "ProcessTaskImageAnalyzer.h"
|
||
|
||
#include <regex>
|
||
|
||
#include "AsstUtils.hpp"
|
||
#include "Logger.hpp"
|
||
#include "MatchImageAnalyzer.h"
|
||
#include "OcrImageAnalyzer.h"
|
||
#include "Resource.h"
|
||
|
||
asst::ProcessTaskImageAnalyzer::ProcessTaskImageAnalyzer(const cv::Mat image, std::vector<std::string> tasks_name)
|
||
: AbstractImageAnalyzer(image),
|
||
m_tasks_name(std::move(tasks_name))
|
||
{
|
||
;
|
||
}
|
||
|
||
asst::ProcessTaskImageAnalyzer::~ProcessTaskImageAnalyzer() = default;
|
||
|
||
bool asst::ProcessTaskImageAnalyzer::match_analyze(std::shared_ptr<TaskInfo> task_ptr)
|
||
{
|
||
if (!m_match_analyzer) {
|
||
m_match_analyzer = std::make_unique<MatchImageAnalyzer>(m_image);
|
||
}
|
||
const auto match_task_ptr = std::dynamic_pointer_cast<MatchTaskInfo>(task_ptr);
|
||
m_match_analyzer->set_task_info(match_task_ptr);
|
||
|
||
if (m_match_analyzer->analyze()) {
|
||
m_result = match_task_ptr;
|
||
m_result_rect = m_match_analyzer->get_result().rect;
|
||
task_ptr->region_of_appeared = m_result_rect;
|
||
return true;
|
||
}
|
||
return false;
|
||
}
|
||
|
||
bool asst::ProcessTaskImageAnalyzer::ocr_analyze(std::shared_ptr<TaskInfo> task_ptr)
|
||
{
|
||
std::shared_ptr<OcrTaskInfo> ocr_task_ptr = std::dynamic_pointer_cast<OcrTaskInfo>(task_ptr);
|
||
|
||
// 先尝试从缓存的结果里找
|
||
//for (const TextRect& tr : m_ocr_cache) {
|
||
// TextRect temp = tr;
|
||
// for (const auto& [regex, new_str] : ocr_task_ptr->replace_map) {
|
||
// temp.text = std::regex_replace(temp.text, std::regex(regex), new_str);
|
||
// }
|
||
// for (const auto& text : ocr_task_ptr->text) {
|
||
// bool flag = false;
|
||
// if (ocr_task_ptr->need_full_match) {
|
||
// if (temp.text == text) {
|
||
// flag = true;
|
||
// }
|
||
// }
|
||
// else if (temp.text.find(text) != std::string::npos) {
|
||
// flag = true;
|
||
// }
|
||
// // 即使找到了，也得在ROI里才行，不然过滤掉
|
||
// if (flag && ocr_task_ptr->roi.include(tr.rect)) {
|
||
// m_result = ocr_task_ptr;
|
||
// m_result_rect = tr.rect;
|
||
// ocr_task_ptr->region_of_appeared = m_result_rect;
|
||
// Log.trace("ProcessTaskImageAnalyzer::ocr_analyze | found in cache", tr.to_string());
|
||
// return true;
|
||
// }
|
||
// }
|
||
//}
|
||
if (!m_ocr_analyzer) {
|
||
m_ocr_analyzer = std::make_unique<OcrImageAnalyzer>(m_image);
|
||
}
|
||
m_ocr_analyzer->set_task_info(ocr_task_ptr);
|
||
|
||
bool ret = m_ocr_analyzer->analyze();
|
||
|
||
if (ret) {
|
||
m_ocr_analyzer->sort_result_by_required();
|
||
const auto& ocr_result = m_ocr_analyzer->get_result();
|
||
auto& res = ocr_result.front();
|
||
m_result = ocr_task_ptr;
|
||
m_result_rect = res.rect;
|
||
ocr_task_ptr->region_of_appeared = res.rect;
|
||
//m_ocr_cache.insert(m_ocr_cache.end(), ocr_result.begin(), ocr_result.end());
|
||
Log.trace("ProcessTaskImageAnalyzer::ocr_analyze | found", res.to_string());
|
||
}
|
||
return ret;
|
||
}
|
||
|
||
void asst::ProcessTaskImageAnalyzer::reset() noexcept
|
||
{
|
||
//m_ocr_cache.clear();
|
||
m_ocr_analyzer = nullptr;
|
||
m_match_analyzer = nullptr;
|
||
}
|
||
|
||
bool asst::ProcessTaskImageAnalyzer::analyze()
|
||
{
|
||
m_result = nullptr;
|
||
m_result_rect = Rect();
|
||
|
||
for (const std::string& task_name : m_tasks_name) {
|
||
auto task_ptr = Task.get(task_name);
|
||
|
||
switch (task_ptr->algorithm) {
|
||
case AlgorithmType::JustReturn:
|
||
m_result = task_ptr;
|
||
return true;
|
||
case AlgorithmType::MatchTemplate:
|
||
if (match_analyze(task_ptr)) {
|
||
return true;
|
||
}
|
||
break;
|
||
case AlgorithmType::OcrDetect:
|
||
if (ocr_analyze(task_ptr)) {
|
||
return true;
|
||
}
|
||
break;
|
||
default:
|
||
break;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
void asst::ProcessTaskImageAnalyzer::set_image(const cv::Mat image)
|
||
{
|
||
AbstractImageAnalyzer::set_image(image);
|
||
reset();
|
||
}
|