Function parseArguments

src/cmd_line_parser.h:43–98

};

inline bool parseArguments(int argc, char *argv[], CommandLineArguments &arguments) {
    if (argc == 1) {
        showHelp(argv);
        return false;
    }

    for (int i = 1; i < argc; i++) {
        std::string argument = argv[i];

        if (argument.substr(0, 2) == "--") {
            std::string flag = argument.substr(2);
            std::string nextArgument;

            if (flag == "onnx_model") {
                if (!tryGetNextArgument(argc, argv, i, nextArgument, flag))
                    return false;

                if (!Util::doesFileExist(nextArgument)) {
                    std::cout << "Error: Unable to find model at path '" << nextArgument << "' for flag '" << flag << "'" << std::endl;
                    return false;
                }

                arguments.onnxModelPath = nextArgument;
            }

            else if (flag == "trt_model") {
                if (!tryGetNextArgument(argc, argv, i, nextArgument, flag))
                    return false;

                if (!Util::doesFileExist(nextArgument)) {
                    std::cout << "Error: Unable to find model at path '" << nextArgument << "' for flag '" << flag << "'" << std::endl;
                    return false;
                }

                arguments.trtModelPath = nextArgument;
            }

            else {
                std::cout << "Error: Unknown flag '" << flag << "'" << std::endl;
                showHelp(argv);
                return false;
            }
        } else {
            std::cout << "Error: Unknown argument '" << argument << "'" << std::endl;
            showHelp(argv);
            return false;
        }
    }

    if (arguments.onnxModelPath.empty() && arguments.trtModelPath.empty()) {
        std::cout << "Error: Must specify either 'onnx_model' or 'trt_model'" << std::endl;
        return false;
    }

    return true;
}