142 lines
4.0 KiB
C++
142 lines
4.0 KiB
C++
//
|
||
// Created by xinyang on 19-3-27.
|
||
//
|
||
|
||
#include <options.h>
|
||
#include <log.h>
|
||
#include <cstring>
|
||
#include <map>
|
||
#include <string>
|
||
|
||
bool show_armor_box = true;
|
||
bool show_armor_boxes = false;
|
||
bool show_light_blobs = false;
|
||
bool show_origin = false;
|
||
bool run_with_camera = true;
|
||
bool save_video = false;
|
||
bool wait_uart = false;
|
||
bool save_labelled_boxes = false;
|
||
bool show_process = false;
|
||
bool show_energy = false;
|
||
bool save_mark = false;
|
||
bool show_info = true;
|
||
bool run_by_frame = false;
|
||
|
||
// 使用map保存所有选项及其描述和操作，加快查找速度。
|
||
std::map<std::string, std::pair<std::string, void(*)(void)>> options = {
|
||
{"--help",{
|
||
"show the help information.", [](){
|
||
LOG(LOG_MSG, "<HELP>: " STR_CTR(WORD_BLUE, "All options below are for debug use."));
|
||
for(const auto &option : options){
|
||
LOG(LOG_MSG, "<HELP>: " STR_CTR(WORD_GREEN, "%s: %s"), option.first.data(), option.second.first.data());
|
||
}
|
||
}
|
||
}},
|
||
{"--show-armor-box", {
|
||
"show the aim box.", []() {
|
||
show_armor_box = true;
|
||
LOGM("Enable show armor box");
|
||
}
|
||
}},
|
||
{"--show-armor-boxes",{
|
||
"show the candidate aim boxes.", [](){
|
||
show_armor_boxes = true;
|
||
LOGM("Enable show armor boxes");
|
||
}
|
||
}},
|
||
{"--show-light-blobs",{
|
||
"show the candidate light blobs.", [](){
|
||
show_light_blobs = true;
|
||
LOGM("Enable show light blobs");
|
||
}
|
||
}},
|
||
{"--show-origin", {
|
||
"show the origin image.", [](){
|
||
show_origin = true;
|
||
LOGM("Enable show origin");
|
||
}
|
||
}},
|
||
{"--run-with-camera", {
|
||
"start the program with camera directly without asking.", []() {
|
||
run_with_camera = true;
|
||
LOGM("Run with camera!");
|
||
}
|
||
}},
|
||
{"--save-video", {
|
||
"save the video.", [](){
|
||
save_video = true;
|
||
LOGM("Enable save video!");
|
||
}
|
||
}},
|
||
{"--save-labelled-boxes",{
|
||
"save the candidate boxes with their id labels.", [](){
|
||
save_labelled_boxes = true;
|
||
LOGM("labelled armor boxes will be saved!");
|
||
}
|
||
}},
|
||
{"--wait-uart", {
|
||
"wait uart until ready before running.", [](){
|
||
wait_uart = true;
|
||
LOGM("Enable wait uart!");
|
||
}
|
||
}},
|
||
{"--run-by-frame",{
|
||
"run the code frame by frame.(normally used when run video)", [](){
|
||
run_by_frame = true;
|
||
LOGM("Enable run frame by frame");
|
||
}
|
||
}},
|
||
{"--show-process", {
|
||
"", [](){
|
||
show_process = true;
|
||
LOGM("Enable show processed image!");
|
||
}
|
||
}},
|
||
{"--show-energy", {
|
||
"",[](){
|
||
show_energy = true;
|
||
LOGM("Enable show energy part!");
|
||
}
|
||
}},
|
||
{"--save-mark", {
|
||
"", [](){
|
||
save_mark = true;
|
||
LOGM("Write down mark");
|
||
}
|
||
}},
|
||
{"--show-info", {
|
||
"", [](){
|
||
show_info = true;
|
||
LOGM("Show information!");
|
||
}
|
||
}},
|
||
{"--show-all", {
|
||
"show all image windows.", [](){
|
||
show_armor_box = true;
|
||
LOGM("Enable show armor box");
|
||
show_armor_boxes = true;
|
||
LOGM("Enable show armor boxes");
|
||
show_light_blobs = true;
|
||
LOGM("Enable show light blobs");
|
||
show_origin = true;
|
||
LOGM("Enable show origin");
|
||
show_process = true;
|
||
LOGM("Enable show processed image");
|
||
show_energy = true;
|
||
LOGM("Enable show energy part");
|
||
}
|
||
}}
|
||
};
|
||
|
||
void processOptions(int argc, char **argv) {
|
||
if (argc >= 2) {
|
||
for (int i = 1; i < argc; i++) {
|
||
auto key = options.find(std::string(argv[i])); // 寻找对应选项。
|
||
if(key != options.end()){
|
||
key->second.second();
|
||
}else{
|
||
LOGW("Unknown option: %s. Use --help to see options.", argv[i]);
|
||
}
|
||
}
|
||
}
|
||
} |