#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;
}