C++ | getopt配置传参

getopt() --> 命令行参数解析函数

getopt() 函数声明如下：

#include <unistd.h>

int getopt(int argc, char * const argv[], const char *optstring);

extern char *optarg;
extern int optind, opterr, optopt;

该函数的argc和argv参数通常直接从main()的参数直接传递而来。optstring是选项字母组成的字串。如果该字串里的任一字符后面有冒号，那么这个选项就要求有选项参数

当给定getopt()命令参数的数量 (argc)、指向这些参数的数组 (argv) 和选项字串 (optstring) 后，getopt() 将返回第一个选项，并设置一些全局变量。使用相同的参数再次调用该函数时，它将返回下一个选项，并设置相应的全局变量。如果不再有可识别的选项，将返回 -1，此任务就完成了

getopt() 所设置的全局变量包括：

• char *optarg：当前选项参数字串（如果有）
• int optind：argv的当前索引值。当getopt()在while循环中使用时，循环结束后，剩下的字串视为操作数，在argv[optind]至argv[argc-1]中可以找到
• int opterr：这个变量非零时，getopt()函数为“无效选项”和“缺少参数选项，并输出其错误信息
• int optopt：当发现无效选项字符之时，getopt()函数或返回'?'字符，或返回':'字符，并且optopt包含了所发现的无效选项字符

void showUsage()
{
    DlLogI << "Usage: ./testserialize [m] [b] [f] [i] [l] [h]";
    DlLogI << "    m: 模型文件路径，默认../resource/resnet18.onnx";
    DlLogI << "    b: 模型最大batch，默认值为16";
    DlLogI << "    f: 输出的序列化engine文件路径， 默认./test.dl";
    DlLogI << "    i: 动态shape模型需要指定模型输入的shape信息，以：分隔，例如 data:1:3:960:512";
    DlLogI << "    l: 日志级别，合法值0-4，分别对应ebug,info,warning,error,fatal， 默认值为1";
    DlLogI << "    h: 显示此帮助信息";
    DlLogI << "    eg: ./testserialize -m ../../../resource/resnet18.onnx -b 16 -f ./test.dl";
    exit(0);
}

void getCustomOpt(int argc, char *argv[], std::string &model_path, int &max_batch,
                  std::string &engine_path, std::map<std::string, Dims> &input_dims_map, int &log_level)
{
    int opt = 0;
    const char* opt_string = "m:b:f:l:i:h";
    while(-1 != (opt = getopt(argc, argv, opt_string))) {
        switch (opt) {
        case 'm':
            model_path = optarg;
            break;
        case 'b':
            max_batch = atoi(optarg);
            break;
        case 'f':
            engine_path = optarg;
            break;
        case 'i':
            {
                char* p = strtok(optarg, ":");
                std::string name = p;
                Dims dim;
                dim.nbDims = 0;
                p = strtok(NULL, ":");
                while(nullptr != p) {
                    dim.d[dim.nbDims++] = atoi(p);
                    p = strtok(NULL, ":");
                }
                input_dims_map[name] = dim;
            }
            break;
        case 'l':
            log_level = atoi(optarg);
            break;
        default:
            showUsage();
            break;
        }
    }
}