MaaAssistantArknights/MeoAssistance/OcrAbstractTask.h

#pragma once

#include "AbstractTask.h"

#include "AsstDef.h"
#include "AsstAux.h"

namespace asst {
	class OcrAbstractTask : public AbstractTask
	{
	public:
		OcrAbstractTask(AsstCallback callback, void* callback_arg);
		virtual ~OcrAbstractTask() = default;

		virtual bool run() override = 0;

	protected:
		std::vector<TextArea> ocr_detect();
		std::vector<TextArea> ocr_detect(const cv::Mat& image);

		// 文字匹配，要求相等
		template<typename FilterArray, typename ReplaceMap>
		std::vector<TextArea> text_match(const std::vector<TextArea>& src,
			const FilterArray& filter_array, const ReplaceMap& replace_map)
		{
			std::vector<TextArea> dst;
			for (const TextArea& text_area : src) {
				TextArea temp = text_area;
				for (const auto& [old_str, new_str] : replace_map) {
					temp.text = StringReplaceAll(temp.text, old_str, new_str);
				}
				for (const auto& text : filter_array) {
					if (temp.text == text) {
						dst.emplace_back(std::move(temp));
					}
				}
			}
			return dst;
		}

		// 文字搜索，是子串即可
		template<typename FilterArray, typename ReplaceMap>
		std::vector<TextArea> text_search(const std::vector<TextArea>& src,
			const FilterArray& filter_array, const ReplaceMap& replace_map)
		{
			std::vector<TextArea> dst;
			for (const TextArea& text_area : src) {
				TextArea temp = text_area;
				for (const auto& [old_str, new_str] : replace_map) {
					temp.text = StringReplaceAll(temp.text, old_str, new_str);
				}
				for (const auto& text : filter_array) {
					if (temp.text.find(text) != std::string::npos) {
						dst.emplace_back(text, std::move(temp.rect));
					}
				}
			}
			return dst;
		}
	};
}