mirror of
https://github.com/MaaAssistantArknights/MaaAssistantArknights.git
synced 2026-07-19 02:23:01 +08:00
125 lines
4.5 KiB
C++
125 lines
4.5 KiB
C++
#pragma once
|
||
|
||
#include <optional>
|
||
#include <random>
|
||
#include <string>
|
||
#include <condition_variable>
|
||
#include <memory>
|
||
#include <mutex>
|
||
#include <queue>
|
||
#include <shared_mutex>
|
||
#include <thread>
|
||
#include <atomic>
|
||
|
||
#ifdef _WIN32
|
||
#include <Windows.h>
|
||
#endif
|
||
|
||
#include <opencv2/opencv.hpp>
|
||
|
||
#include "AsstDef.h"
|
||
|
||
namespace asst
|
||
{
|
||
class Controller
|
||
{
|
||
public:
|
||
Controller(const Controller&) = delete;
|
||
Controller(Controller&&) = delete;
|
||
~Controller();
|
||
|
||
static Controller& get_instance()
|
||
{
|
||
static Controller unique_instance;
|
||
return unique_instance;
|
||
}
|
||
|
||
static void set_dirname(std::string dirname) noexcept;
|
||
|
||
bool try_capture(const EmulatorInfo& info, bool without_handle = false);
|
||
cv::Mat get_image();
|
||
std::vector<uchar> get_image_encode();
|
||
|
||
// 点击和滑动都是异步执行，返回该任务的id
|
||
int click(const Point& p, bool block = true);
|
||
int click(const Rect& rect, bool block = true);
|
||
int click_without_scale(const Point& p, bool block = true);
|
||
int click_without_scale(const Rect& rect, bool block = true);
|
||
|
||
constexpr static int SwipeExtraDelayDefault = 1000;
|
||
int swipe(const Point& p1, const Point& p2, int duration = 0, bool block = true, int extra_delay = SwipeExtraDelayDefault, bool extra_swipe = false);
|
||
int swipe(const Rect& r1, const Rect& r2, int duration = 0, bool block = true, int extra_delay = SwipeExtraDelayDefault, bool extra_swipe = false);
|
||
int swipe_without_scale(const Point& p1, const Point& p2, int duration = 0, bool block = true, int extra_delay = SwipeExtraDelayDefault, bool extra_swipe = false);
|
||
int swipe_without_scale(const Rect& r1, const Rect& r2, int duration = 0, bool block = true, int extra_delay = SwipeExtraDelayDefault, bool extra_swipe = false);
|
||
|
||
void wait(unsigned id) const noexcept;
|
||
|
||
// 异形屏矫正
|
||
Rect shaped_correct(const Rect& rect) const;
|
||
std::pair<int, int> get_scale_size() const noexcept;
|
||
|
||
Controller& operator=(const Controller&) = delete;
|
||
Controller& operator=(Controller&&) = delete;
|
||
|
||
private:
|
||
Controller();
|
||
|
||
bool connect_adb(const std::string& address);
|
||
|
||
void pipe_working_proc();
|
||
std::optional<std::vector<unsigned char>> call_command(const std::string& cmd);
|
||
int push_cmd(const std::string& cmd);
|
||
|
||
using DecodeFunc = std::function<bool(const std::vector<uchar>&)>;
|
||
bool screencap();
|
||
bool screencap(const std::string& cmd, DecodeFunc decode_func);
|
||
cv::Mat get_resized_image() const;
|
||
|
||
Point rand_point_in_rect(const Rect& rect);
|
||
|
||
void random_delay() const;
|
||
|
||
// 转换data中所有的crlf为lf：有些模拟器自带的adb，exec-out输出的\n，会被替换成\r\n，导致解码错误，所以这里转一下回来（点名批评mumu）
|
||
static void convert_lf(std::vector<unsigned char>& data);
|
||
|
||
inline static std::string m_dirname;
|
||
|
||
bool m_thread_exit = false;
|
||
//bool m_thread_idle = true;
|
||
std::thread m_cmd_thread;
|
||
std::mutex m_cmd_queue_mutex;
|
||
std::condition_variable m_cmd_condvar;
|
||
std::queue<std::string> m_cmd_queue;
|
||
std::atomic<unsigned> m_completed_id = 0;
|
||
unsigned m_push_id = 0; // push_id的自增总是伴随着queue的push，肯定是要上锁的，所以没必要原子
|
||
|
||
mutable std::shared_mutex m_image_mutex;
|
||
cv::Mat m_cache_image;
|
||
|
||
constexpr static int PipeBuffSize = 4 * 1024 * 1024; // 管道缓冲区大小
|
||
std::unique_ptr<uchar[]> m_pipe_buffer = nullptr;
|
||
#ifdef _WIN32
|
||
HANDLE m_pipe_read = nullptr; // 读管道句柄
|
||
HANDLE m_pipe_write = nullptr; // 写管道句柄
|
||
HANDLE m_pipe_child_read = nullptr; // 子进程的读管道句柄
|
||
HANDLE m_pipe_child_write = nullptr; // 子进程的写管道句柄
|
||
SECURITY_ATTRIBUTES m_pipe_sec_attr = { 0 }; // 管道安全描述符
|
||
STARTUPINFOA m_child_startup_info = { 0 }; // 子进程启动信息
|
||
#else
|
||
constexpr static int PIPE_READ = 0;
|
||
constexpr static int PIPE_WRITE = 1;
|
||
int m_pipe_in[2] = { 0 };
|
||
int m_pipe_out[2] = { 0 };
|
||
int m_child = 0;
|
||
#endif
|
||
|
||
EmulatorInfo m_emulator_info;
|
||
std::minstd_rand m_rand_engine;
|
||
std::pair<int, int> m_scale_size;
|
||
double m_control_scale = 1.0;
|
||
};
|
||
|
||
//static auto& ctrler = Controller::get_instance();
|
||
#define Ctrler Controller::get_instance()
|
||
}
|