MaaAssistantArknights/src/MeoAssistant/RoguelikeBattleTaskPlugin.h

#pragma once

#include "AbstractTaskPlugin.h"
#include "AsstTypes.h"
#include "TilePack.h"
#include "BattleImageAnalyzer.h"
#include "AsstBattleDef.h"

namespace asst
{
    class RoguelikeBattleTaskPlugin : public AbstractTaskPlugin
    {
        using Loc = asst::TilePack::BuildableType;
    public:
        using AbstractTaskPlugin::AbstractTaskPlugin;
        virtual ~RoguelikeBattleTaskPlugin() = default;

        virtual bool verify(AsstMsg msg, const json::value& details) const override;

        void set_stage_name(std::string stage);

    protected:
        virtual bool _run() override;

        bool get_stage_info();
        bool auto_battle();
        void all_melee_retreat();
        bool speed_up();
        bool use_skill(const Rect& rect);
        bool retreat(const Point& point);
        void clear();
        bool try_possible_skill(const cv::Mat& image);
        bool wait_start();

        // 通过资源文件离线计算可放置干员的位置，优先使用
        // 返回 可格子的位置
        Point get_placed(Loc buildable_type);

        // 计算摆放干员的朝向
        // 返回滑动的方向
        Point calc_direction(Point loc, BattleRole role);

        bool m_opers_used = false;
        int m_pre_hp = 0;

        std::unordered_map<Point, TilePack::TileInfo> m_side_tile_info;
        std::unordered_map<Point, TilePack::TileInfo> m_normal_tile_info;
        std::vector<Point> m_homes;
        size_t m_cur_home_index = 0;
        std::unordered_map<Point, std::string> m_used_tiles;
        std::unordered_map<std::string, int64_t> m_restore_status;

        std::string m_stage_name;
    };
}