mirror of
https://github.com/MaaAssistantArknights/MaaAssistantArknights.git
synced 2026-07-16 09:50:40 +08:00
整理目录结构，分离Task
This commit is contained in:
MistEO
288
MeoAssistance/Configer.cpp
Normal file
288
MeoAssistance/Configer.cpp
Normal file
@@ -0,0 +1,288 @@
|
||||
#include "Configer.h"
|
||||
|
||||
#include <fstream>
|
||||
#include <sstream>
|
||||
#include <algorithm>
|
||||
|
||||
#include <json.h>
|
||||
|
||||
#include "Logger.hpp"
|
||||
|
||||
using namespace asst;
|
||||
|
||||
//Configer::Configer(const Configer& rhs)
|
||||
// : m_version(rhs.m_version),
|
||||
// m_options(rhs.m_options),
|
||||
// m_tasks(rhs.m_tasks),
|
||||
// m_handles(rhs.m_handles)
|
||||
//{
|
||||
//}
|
||||
//
|
||||
//Configer::Configer(Configer&& rhs) noexcept
|
||||
// : m_version(std::move(rhs.m_version)),
|
||||
// m_options(std::move(rhs.m_options)),
|
||||
// m_tasks(std::move(rhs.m_tasks)),
|
||||
// m_handles(std::move(rhs.m_handles))
|
||||
//{
|
||||
//}
|
||||
|
||||
bool Configer::load(const std::string& filename)
|
||||
{
|
||||
DebugTraceFunction;
|
||||
DebugTrace("Configer::load | ", filename);
|
||||
|
||||
Configer temp;
|
||||
if (temp._load(filename)) {
|
||||
*this = std::move(temp);
|
||||
return true;
|
||||
}
|
||||
else {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
bool Configer::set_param(const std::string& type, const std::string& param, const std::string& value)
|
||||
{
|
||||
// 暂时只用到了这些，总的参数太多了，后面要用啥再加上
|
||||
if (type == "task.type") {
|
||||
if (m_all_tasks_info.find(param) == m_all_tasks_info.cend()) {
|
||||
return false;
|
||||
}
|
||||
auto task_info_ptr = m_all_tasks_info[param];
|
||||
std::string type = value;
|
||||
std::transform(type.begin(), type.end(), type.begin(), std::tolower);
|
||||
if (type == "clickself") {
|
||||
task_info_ptr->type = ProcessTaskType::ClickSelf;
|
||||
}
|
||||
else if (type == "clickrand") {
|
||||
task_info_ptr->type = ProcessTaskType::ClickRand;
|
||||
}
|
||||
else if (type == "donothing" || type.empty()) {
|
||||
task_info_ptr->type = ProcessTaskType::DoNothing;
|
||||
}
|
||||
else if (type == "stop") {
|
||||
task_info_ptr->type = ProcessTaskType::Stop;
|
||||
}
|
||||
else if (type == "clickrect") {
|
||||
task_info_ptr->type = ProcessTaskType::ClickRect;
|
||||
}
|
||||
else {
|
||||
DebugTraceError("Task", param, "'s type error:", type);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else if (type == "task.maxTimes") {
|
||||
if (m_all_tasks_info.find(param) == m_all_tasks_info.cend()) {
|
||||
return false;
|
||||
}
|
||||
m_all_tasks_info[param]->max_times = std::stoi(value);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool asst::Configer::_load(const std::string& filename)
|
||||
{
|
||||
std::ifstream ifs(filename, std::ios::in);
|
||||
if (!ifs.is_open()) {
|
||||
return false;
|
||||
}
|
||||
std::stringstream iss;
|
||||
iss << ifs.rdbuf();
|
||||
ifs.close();
|
||||
std::string content(iss.str());
|
||||
|
||||
auto&& ret = json::parser::parse(content);
|
||||
if (!ret) {
|
||||
DebugTrace("parse error", content);
|
||||
return false;
|
||||
}
|
||||
|
||||
json::value root = ret.value();
|
||||
try {
|
||||
m_version = root["version"].as_string();
|
||||
|
||||
json::value& options_json = root["options"];
|
||||
{
|
||||
m_options.task_identify_delay = options_json["taskIdentifyDelay"].as_integer();
|
||||
m_options.task_control_delay = options_json["taskControlDelay"].as_integer();
|
||||
m_options.identify_cache = options_json["identifyCache"].as_boolean();
|
||||
m_options.control_delay_lower = options_json["controlDelayRange"][0].as_integer();
|
||||
m_options.control_delay_upper = options_json["controlDelayRange"][1].as_integer();
|
||||
m_options.print_window = options_json["printWindow"].as_boolean();
|
||||
m_options.print_window_delay = options_json["printWindowDelay"].as_integer();
|
||||
m_options.print_window_crop_offset = options_json["printWindowCropOffset"].as_integer();
|
||||
m_options.ocr_gpu_index = options_json["ocrGpuIndex"].as_integer();
|
||||
m_options.ocr_thread_number = options_json["ocrThreadNumber"].as_integer();
|
||||
}
|
||||
DebugTrace("Options", Utf8ToGbk(options_json.to_string()));
|
||||
|
||||
for (auto&& [name, task_json] : root["tasks"].as_object()) {
|
||||
std::string algorithm_str = task_json.get("algorithm", "matchtemplate");
|
||||
std::transform(algorithm_str.begin(), algorithm_str.end(), algorithm_str.begin(), std::tolower);
|
||||
AlgorithmType algorithm = AlgorithmType::Invaild;
|
||||
if (algorithm_str == "matchtemplate") {
|
||||
algorithm = AlgorithmType::MatchTemplate;
|
||||
}
|
||||
else if (algorithm_str == "justreturn") {
|
||||
algorithm = AlgorithmType::JustReturn;
|
||||
}
|
||||
else if (algorithm_str == "ocrdetect") {
|
||||
algorithm = AlgorithmType::OcrDetect;
|
||||
}
|
||||
//else if (algorithm_str == "comparehist") {} // CompareHist是MatchTemplate的衍生算法，不应作为单独的配置参数出现
|
||||
else {
|
||||
DebugTraceError("Algorithm error:", algorithm_str);
|
||||
return false;
|
||||
}
|
||||
|
||||
std::shared_ptr<TaskInfo> task_info_ptr = nullptr;
|
||||
switch (algorithm) {
|
||||
case AlgorithmType::JustReturn:
|
||||
task_info_ptr = std::make_shared<TaskInfo>();
|
||||
break;
|
||||
case AlgorithmType::MatchTemplate:
|
||||
{
|
||||
auto match_task_info_ptr = std::make_shared<MatchTaskInfo>();
|
||||
match_task_info_ptr->template_filename = task_json["template"].as_string();
|
||||
match_task_info_ptr->templ_threshold = task_json.get("templThreshold", Defaulttempl_threshold);
|
||||
match_task_info_ptr->hist_threshold = task_json.get("histThreshold", DefaultCachetempl_threshold);
|
||||
task_info_ptr = match_task_info_ptr;
|
||||
}
|
||||
break;
|
||||
case AlgorithmType::OcrDetect:
|
||||
{
|
||||
auto ocr_task_info_ptr = std::make_shared<OcrTaskInfo>();
|
||||
for (const json::value& text : task_json["text"].as_array())
|
||||
{
|
||||
ocr_task_info_ptr->text.emplace_back(text.as_string());
|
||||
}
|
||||
ocr_task_info_ptr->need_match = task_json.get("need_match", false);
|
||||
if (task_json.exist("ocrReplace"))
|
||||
{
|
||||
for (const auto& [key, value] : task_json["ocrReplace"].as_object()) {
|
||||
ocr_task_info_ptr->replace_map.emplace(key, value.as_string());
|
||||
}
|
||||
}
|
||||
task_info_ptr = ocr_task_info_ptr;
|
||||
}
|
||||
break;
|
||||
}
|
||||
task_info_ptr->algorithm = algorithm;
|
||||
task_info_ptr->name = name;
|
||||
std::string type = task_json["type"].as_string();
|
||||
std::transform(type.begin(), type.end(), type.begin(), std::tolower);
|
||||
if (type == "clickself") {
|
||||
task_info_ptr->type = ProcessTaskType::ClickSelf;
|
||||
}
|
||||
else if (type == "clickrand") {
|
||||
task_info_ptr->type = ProcessTaskType::ClickRand;
|
||||
}
|
||||
else if (type == "donothing" || type.empty()) {
|
||||
task_info_ptr->type = ProcessTaskType::DoNothing;
|
||||
}
|
||||
else if (type == "stop") {
|
||||
task_info_ptr->type = ProcessTaskType::Stop;
|
||||
}
|
||||
else if (type == "clickrect") {
|
||||
task_info_ptr->type = ProcessTaskType::ClickRect;
|
||||
json::value& area_json = task_json["specificArea"];
|
||||
task_info_ptr->specific_area = Rect(
|
||||
area_json[0].as_integer(),
|
||||
area_json[1].as_integer(),
|
||||
area_json[2].as_integer(),
|
||||
area_json[3].as_integer());
|
||||
}
|
||||
else if (type == "printwindow") {
|
||||
task_info_ptr->type = ProcessTaskType::PrintWindow;
|
||||
}
|
||||
else {
|
||||
DebugTraceError("Task:", name, "error:", type);
|
||||
return false;
|
||||
}
|
||||
|
||||
task_info_ptr->max_times = task_json.get("maxTimes", INT_MAX);
|
||||
if (task_json.exist("exceededNext")) {
|
||||
json::array& excceed_next_arr = task_json["exceededNext"].as_array();
|
||||
for (const json::value& excceed_next : excceed_next_arr) {
|
||||
task_info_ptr->exceeded_next.emplace_back(excceed_next.as_string());
|
||||
}
|
||||
}
|
||||
else {
|
||||
task_info_ptr->exceeded_next.emplace_back("Stop");
|
||||
}
|
||||
task_info_ptr->pre_delay = task_json.get("preDelay", 0);
|
||||
task_info_ptr->rear_delay = task_json.get("rearDelay", 0);
|
||||
if (task_json.exist("reduceOtherTimes")) {
|
||||
json::array& reduce_arr = task_json["reduceOtherTimes"].as_array();
|
||||
for (const json::value& reduce : reduce_arr) {
|
||||
task_info_ptr->reduce_other_times.emplace_back(reduce.as_string());
|
||||
}
|
||||
}
|
||||
|
||||
json::array& next_arr = task_json["next"].as_array();
|
||||
for (const json::value& next : next_arr) {
|
||||
task_info_ptr->next.emplace_back(next.as_string());
|
||||
}
|
||||
|
||||
m_all_tasks_info.emplace(name, task_info_ptr);
|
||||
}
|
||||
|
||||
for (auto&& [name, emulator_json] : root["handle"].as_object()) {
|
||||
EmulatorInfo emulator_info;
|
||||
emulator_info.name = name;
|
||||
|
||||
for (json::value& info : emulator_json["window"].as_array()) {
|
||||
HandleInfo handle_info;
|
||||
handle_info.class_name = info["class"].as_string();
|
||||
handle_info.window_name = info["window"].as_string();
|
||||
emulator_info.window.emplace_back(std::move(handle_info));
|
||||
}
|
||||
for (json::value& info : emulator_json["view"].as_array()) {
|
||||
HandleInfo handle_info;
|
||||
handle_info.class_name = info["class"].as_string();
|
||||
handle_info.window_name = info["window"].as_string();
|
||||
emulator_info.view.emplace_back(std::move(handle_info));
|
||||
}
|
||||
for (json::value& info : emulator_json["control"].as_array()) {
|
||||
HandleInfo handle_info;
|
||||
handle_info.class_name = info["class"].as_string();
|
||||
handle_info.window_name = info["window"].as_string();
|
||||
emulator_info.control.emplace_back(std::move(handle_info));
|
||||
}
|
||||
if (emulator_json.exist("adb")) {
|
||||
emulator_info.is_adb = true;
|
||||
emulator_info.adb.path = emulator_json["adb"]["path"].as_string();
|
||||
emulator_info.adb.connect = emulator_json["adb"]["connect"].as_string();
|
||||
emulator_info.adb.click = emulator_json["adb"]["click"].as_string();
|
||||
emulator_info.adb.swipe = emulator_json["adb"]["swipe"].as_string();
|
||||
emulator_info.adb.display = emulator_json["adb"]["display"].as_string();
|
||||
emulator_info.adb.display_regex = emulator_json["adb"]["displayRegex"].as_string();
|
||||
emulator_info.adb.screencap = emulator_json["adb"]["screencap"].as_string();
|
||||
emulator_info.adb.pullscreen = emulator_json["adb"]["pullscreen"].as_string();
|
||||
}
|
||||
emulator_info.x_offset = emulator_json.get("xOffset", 0);
|
||||
emulator_info.y_offset = emulator_json.get("yOffset", 0);
|
||||
emulator_info.right_offset = emulator_json.get("rightOffset", 0);
|
||||
emulator_info.bottom_offset = emulator_json.get("bottomOffset", 0);
|
||||
|
||||
emulator_info.width = DefaultWindowWidth + emulator_info.x_offset + emulator_info.right_offset;
|
||||
emulator_info.height = DefaultWindowHeight + emulator_info.y_offset + emulator_info.bottom_offset;
|
||||
|
||||
m_handles.emplace(name, std::move(emulator_info));
|
||||
}
|
||||
|
||||
for (json::value& rep : root["infrastOcrReplace"].as_array()) {
|
||||
m_infrast_ocr_replace.emplace(rep.as_array()[0].as_string(), rep.as_array()[1].as_string());
|
||||
}
|
||||
for (json::value& rep : root["recruitOcrReplace"].as_array()) {
|
||||
m_recruit_ocr_replace.emplace(rep.as_array()[0].as_string(), rep.as_array()[1].as_string());
|
||||
}
|
||||
}
|
||||
catch (json::exception& e) {
|
||||
DebugTraceError("Load config json error!", e.what());
|
||||
return false;
|
||||
}
|
||||
DebugTrace("Load config succeed");
|
||||
|
||||
return true;
|
||||
}
|
||||
Reference in New Issue
Block a user