MaaAssistantArknights/src/MeoAssistance/InfrastDormTask.cpp

#include "InfrastDormTask.h"

#include "MatchImageAnalyzer.h"
#include "OcrImageAnalyzer.h"
#include "InfrastMoodImageAnalyzer.h"
#include "Resource.h"
#include "Controller.h"
#include "Logger.hpp"

const std::string asst::InfrastDormTask::FacilityName = "Dorm";

bool asst::InfrastDormTask::run()
{
    json::value task_start_json = json::object{
        { "task_type",  "InfrastDormTask" },
        { "task_chain", m_task_chain}
    };
    m_callback(AsstMsg::TaskStart, task_start_json, m_callback_arg);

    for (; m_cur_dorm_index < m_max_num_of_dorm; ++m_cur_dorm_index) {
        if (need_exit()) {
            return false;
        }
        enter_facility(FacilityName, m_cur_dorm_index);
        if (!enter_oper_list_page()) {
            return false;
        }
        swipe_to_the_left_of_operlist();
        click_clear_button();

        int quantity_selected = 0;
        while (quantity_selected < MaxNumOfOpers) {
            if (need_exit()) {
                return false;
            }
            const auto& image = ctrler.get_image();
            InfrastMoodImageAnalyzer mood_analyzer(image);
            if (!mood_analyzer.analyze()) {
                log.error("mood analyze faild!");
                return false;
            }
            mood_analyzer.sort_result();
            const auto& mood_result = mood_analyzer.get_result();

            int quantity_resting = 0;
            for (const auto& mood_info : mood_result) {
                if (need_exit()) {
                    return false;
                }
                if (quantity_selected >= MaxNumOfOpers) {
                    log.trace("quantity_selected:", quantity_selected, ", just break");
                    break;
                }
                switch (mood_info.smiley.type)
                {
                case InfrastSmileyType::Rest:
                    // 如果当前页面休息完成的人数超过5个，说明已经已经把所有心情不满的滑过一遍、没有更多的了，直接退出即可
                    if (++quantity_resting > MaxNumOfOpers) {
                        log.trace("quantity_resting:", quantity_resting, ", confirm");
                        click_confirm_button();
                        return true;
                    }
                    break;
                case InfrastSmileyType::Work:
                case InfrastSmileyType::Distract:
                    // 干员没有被选择的情况下，心情小于30%，或不在工作，就进驻宿舍
                    if (mood_info.selected == false
                        && (mood_info.percentage < m_mood_threshold
                            || mood_info.working == false))
                    {
                        ctrler.click(mood_info.rect);
                        if (++quantity_selected >= MaxNumOfOpers) {
                            log.trace("quantity_selected:", quantity_selected, ", just break");
                            break;
                        }
                    }
                    break;
                default:
                    break;
                }
            }
            if (quantity_selected >= MaxNumOfOpers) {
                log.trace("quantity_selected:", quantity_selected, ", just break");
                break;
            }
            sync_swipe_of_operlist();
        }
        click_confirm_button();
        click_return_button();
    }
    return true;
}

bool asst::InfrastDormTask::click_confirm_button()
{
    LogTraceFunction;

    const auto task_ptr = std::dynamic_pointer_cast<OcrTaskInfo>(
        resource.task().task_ptr("InfrastConfirmButton"));
    ctrler.click(task_ptr->specific_rect);
    sleep(task_ptr->rear_delay);

    // 宿舍在把正在工作的干员换下来的时候，会有个二次确认的按钮
    const auto& image = ctrler.get_image();
    MatchImageAnalyzer cfm_analyzer(image);
    const auto sec_cfm_task_ptr = std::dynamic_pointer_cast<MatchTaskInfo>(
        resource.task().task_ptr("InfrastDormConfirmButton"));
    cfm_analyzer.set_task_info(*sec_cfm_task_ptr);
    if (cfm_analyzer.analyze()) {
        ctrler.click(cfm_analyzer.get_result().rect);
        sleep(sec_cfm_task_ptr->rear_delay);
    }

    // 识别“正在提交反馈至神经”，如果网不好一直确认不了，就多等一会
    OcrImageAnalyzer analyzer;
    analyzer.set_task_info(*task_ptr);
    for (int i = 0; i != m_retry_times; ++i) {
        if (need_exit()) {
            return false;
        }
        const auto& image = ctrler.get_image();
        analyzer.set_image(image);
        if (!analyzer.analyze()) {
            sleep(sec_cfm_task_ptr->rear_delay);
            return true;
        }
        sleep(sec_cfm_task_ptr->rear_delay);
    }
    return false;
}