mirror of
https://github.com/MaaAssistantArknights/MaaAssistantArknights.git
synced 2026-07-19 18:47:55 +08:00
88 lines
2.2 KiB
C++
88 lines
2.2 KiB
C++
#include "AbstractImageAnalyzer.h"
|
|
|
|
#include "NoWarningCV.h"
|
|
|
|
#include "AsstUtils.hpp"
|
|
#include "Controller.h"
|
|
#include "Logger.hpp"
|
|
|
|
asst::AbstractImageAnalyzer::AbstractImageAnalyzer(const cv::Mat& image)
|
|
: m_image(image), m_roi(empty_rect_to_full(Rect(), image))
|
|
#ifdef ASST_DEBUG
|
|
,
|
|
m_image_draw(image.clone())
|
|
#endif
|
|
{}
|
|
|
|
asst::AbstractImageAnalyzer::AbstractImageAnalyzer(const cv::Mat& image, const Rect& roi)
|
|
: m_image(image),
|
|
m_roi(empty_rect_to_full(roi, image))
|
|
#ifdef ASST_DEBUG
|
|
,
|
|
m_image_draw(image.clone())
|
|
#endif
|
|
{}
|
|
|
|
void asst::AbstractImageAnalyzer::set_image(const cv::Mat image)
|
|
{
|
|
m_image = image;
|
|
#ifdef ASST_DEBUG
|
|
m_image_draw = image.clone();
|
|
#endif
|
|
}
|
|
|
|
void asst::AbstractImageAnalyzer::set_image(const cv::Mat image, const Rect& roi)
|
|
{
|
|
set_image(image);
|
|
m_roi = empty_rect_to_full(roi, image);
|
|
}
|
|
|
|
void asst::AbstractImageAnalyzer::set_roi(const Rect& roi) noexcept
|
|
{
|
|
m_roi = empty_rect_to_full(roi, m_image);
|
|
}
|
|
|
|
asst::Rect asst::AbstractImageAnalyzer::empty_rect_to_full(const Rect& rect, const cv::Mat& image) noexcept
|
|
{
|
|
if (image.empty()) {
|
|
return rect;
|
|
}
|
|
if (rect.empty()) {
|
|
return { 0, 0, image.cols, image.rows };
|
|
}
|
|
|
|
Rect res = rect;
|
|
if (image.cols < res.x) {
|
|
Log.error("roi is out of range", image.cols, image.rows, res.to_string());
|
|
res.x = image.cols - res.width;
|
|
}
|
|
if (image.rows < res.y) {
|
|
Log.error("roi is out of range", image.cols, image.rows, res.to_string());
|
|
res.y = image.rows - res.height;
|
|
}
|
|
|
|
if (image.cols < res.x + res.width) {
|
|
Log.error("roi is out of range", res.to_string());
|
|
res.width = image.cols - res.x;
|
|
}
|
|
if (image.rows < res.y + res.height) {
|
|
Log.error("roi is out of range", res.to_string());
|
|
res.height = image.rows - res.y;
|
|
}
|
|
return res;
|
|
}
|
|
|
|
bool asst::AbstractImageAnalyzer::save_img()
|
|
{
|
|
std::string stem = utils::get_format_time();
|
|
stem = utils::string_replace_all_batch(stem, { {":", "-"}, {" ", "_"} });
|
|
std::filesystem::create_directory("debug");
|
|
bool ret = cv::imwrite("debug/" + stem + "_raw.png", m_image);
|
|
|
|
#ifdef ASST_DEBUG
|
|
cv::imwrite("debug/" + stem + "_draw.png", m_image_draw);
|
|
#endif
|
|
|
|
return ret;
|
|
}
|