mirror of
https://github.com/MaaAssistantArknights/MaaAssistantArknights.git
synced 2026-07-19 18:47:55 +08:00
整理目录结构,分离Task
This commit is contained in:
76
MeoAssistance/AbstractTask.h
Normal file
76
MeoAssistance/AbstractTask.h
Normal file
@@ -0,0 +1,76 @@
|
||||
#pragma once
|
||||
|
||||
#include <memory>
|
||||
|
||||
#include "AsstMsg.h"
|
||||
|
||||
namespace cv
|
||||
{
|
||||
class Mat;
|
||||
}
|
||||
|
||||
namespace asst {
|
||||
class WinMacro;
|
||||
class Identify;
|
||||
|
||||
enum TaskType {
|
||||
TaskTypeInvalid = 0,
|
||||
TaskTypeWindowZoom = 1,
|
||||
TaskTypeRecognition = 2,
|
||||
TaskTypeClick = 4
|
||||
};
|
||||
|
||||
// 设施类型
|
||||
enum class FacilityType {
|
||||
Invalid,
|
||||
Manufacturing, // 制造站
|
||||
Trade, // 贸易站
|
||||
PowerStation, // 发电站
|
||||
ControlInterface, // 控制中枢
|
||||
ReceptionRoom, // 会客室
|
||||
Dormitory, // 宿舍
|
||||
Office // 办公室
|
||||
// 训练室和加工站用不上,不加了
|
||||
};
|
||||
|
||||
class AbstractTask
|
||||
{
|
||||
public:
|
||||
AbstractTask(AsstCallback callback, void* callback_arg);
|
||||
virtual ~AbstractTask() = default;
|
||||
AbstractTask(const AbstractTask&) = default;
|
||||
AbstractTask(AbstractTask&&) = default;
|
||||
|
||||
virtual void set_ptr(
|
||||
std::shared_ptr<WinMacro> window_ptr,
|
||||
std::shared_ptr<WinMacro> view_ptr,
|
||||
std::shared_ptr<WinMacro> control_ptr,
|
||||
std::shared_ptr<Identify> identify_ptr);
|
||||
virtual bool run() = 0;
|
||||
|
||||
virtual void set_exit_flag(bool* exit_flag);
|
||||
virtual int get_task_type() { return m_task_type; }
|
||||
virtual void set_retry_times(int times) { m_retry_times = times; }
|
||||
virtual int get_retry_times() { return m_retry_times; }
|
||||
virtual void set_task_chain(std::string name) { m_task_chain = std::move(name); }
|
||||
virtual const std::string& get_task_chain() { return m_task_chain; }
|
||||
protected:
|
||||
virtual cv::Mat get_format_image(bool hd = false); // 参数hd:高清截图,会慢一点
|
||||
virtual bool set_control_scale(double scale);
|
||||
virtual bool set_control_scale(int cur_width, int cur_height);
|
||||
virtual bool sleep(unsigned millisecond);
|
||||
virtual bool print_window(const std::string& dir);
|
||||
|
||||
std::shared_ptr<WinMacro> m_window_ptr = nullptr;
|
||||
std::shared_ptr<WinMacro> m_view_ptr = nullptr;
|
||||
std::shared_ptr<WinMacro> m_control_ptr = nullptr;
|
||||
std::shared_ptr<Identify> m_identify_ptr = nullptr;
|
||||
|
||||
AsstCallback m_callback;
|
||||
void* m_callback_arg = NULL;
|
||||
bool* m_exit_flag = NULL;
|
||||
std::string m_task_chain;
|
||||
int m_task_type = TaskType::TaskTypeInvalid;
|
||||
int m_retry_times = INT_MAX;
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user