MaaAssistantArknights/src/MeoAssistant/OcrImageAnalyzer.h

#pragma once
#include "AbstractImageAnalyzer.h"

#include <functional>
#include <unordered_map>
#include <vector>

#include "AsstDef.h"

namespace asst
{
    class OcrImageAnalyzer : public AbstractImageAnalyzer
    {
    public:
        using AbstractImageAnalyzer::AbstractImageAnalyzer;
        virtual ~OcrImageAnalyzer() = default;

        virtual bool analyze() override;

        void sort_result();             // 按位置排序，左上角的排在前面
        void sort_result_by_required(); // 按传入的需求数组排序，传入的在前面结果接在前面

        void set_required(std::vector<std::string> required, bool full_match = false) noexcept;
        void set_replace(std::unordered_map<std::string, std::string> replace) noexcept;
        void set_task_info(std::shared_ptr<TaskInfo> task_ptr);
        void set_task_info(const std::string& task_name);

        void set_pred(const TextRectProc& pred);
        const std::vector<TextRect>& get_result() const noexcept;

    protected:
        void set_task_info(OcrTaskInfo task_info) noexcept;

        std::vector<TextRect> m_ocr_result;
        std::vector<std::string> m_required;
        bool m_full_match = false;
        std::unordered_map<std::string, std::string> m_replace;
        TextRectProc m_pred = nullptr;
        bool m_without_det = false;
    };
}