fz-mirrors/MaaAssistantArknights
0
0
Fork 0
mirror of https://github.com/MaaAssistantArknights/MaaAssistantArknights.git synced 2026-07-18 02:10:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: 更新测试代码

Browse Source
This commit is contained in:
MistEO
2023-03-09 21:07:03 +08:00
parent 83cf55bea1
commit 16cf3e26db
2 changed files with 36 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
src/MaaCore/Task/Interface/DebugTask.cpp
View File

@@ -4,21 +4,22 @@
#include "Utils/NoWarningCV.h"
// #include "Plugin/RoguelikeSkillSelectionTaskPlugin.h"
#include "Utils/ImageIo.hpp"
#include "Utils/Logger.hpp"
#include "Vision/Miscellaneous/BattleSkillReadyImageAnalyzer.h"
#include "Vision/Miscellaneous/DepotImageAnalyzer.h"
#include "Vision/Miscellaneous/StageDropsImageAnalyzer.h"
asst::DebugTask::DebugTask(const AsstCallback& callback, Assistant* inst) : InterfaceTask(callback, inst, TaskType)
{
////auto task_ptr = std::make_shared<RoguelikeSkillSelectionTaskPlugin>(callback, inst, TaskType);
// auto task_ptr = std::make_shared<StageDropsTaskPlugin>(callback, inst, TaskType);
// m_subtasks.emplace_back(task_ptr);
}
asst::DebugTask::DebugTask(const AsstCallback& callback, Assistant* inst) : InterfaceTask(callback, inst, TaskType) {}
bool asst::DebugTask::run()
{
test_drops();
return true;
}
void asst::DebugTask::test_drops()
{
size_t total = 0;
size_t success = 0;
@@ -34,5 +35,27 @@ bool asst::DebugTask::run()
success += analyzer.analyze();
}
Log.info(__FUNCTION__, success, "/", total);
return true;
}
void asst::DebugTask::test_skill_ready()
{
int total = 0;
int correct = 0;
for (const auto& entry : std::filesystem::directory_iterator(R"(../../test/skill_ready/y)")) {
cv::Mat image = imread(entry.path().string());
BattleSkillReadyImageAnalyzer analyzer(image);
total++;
if (analyzer.analyze()) {
correct++;
}
}
for (const auto& entry : std::filesystem::directory_iterator(R"(../../test/skill_ready/n)")) {
cv::Mat image = imread(entry.path().string());
BattleSkillReadyImageAnalyzer analyzer(image);
total++;
if (!analyzer.analyze()) {
correct++;
}
}
Log.info(__FUNCTION__, correct, "/", total, ",", double(correct) / total);
}