89 lines
2.9 KiB
C++
89 lines
2.9 KiB
C++
#include "arg_parse.hpp"
|
|
|
|
#include <cstring>
|
|
#include <iostream>
|
|
|
|
Args::Args()
|
|
: generate_blue_noise_(false),
|
|
use_opencl_(true),
|
|
overwrite_file_(false),
|
|
use_vulkan_(true),
|
|
blue_noise_size_(32),
|
|
threads_(4),
|
|
output_filename_("output.png") {}
|
|
|
|
void Args::DisplayHelp() {
|
|
std::cout << "[-h | --help] [-b <size> | --blue-noise <size>] [--usecl | "
|
|
"--nousecl]\n"
|
|
" -h | --help\t\t\t\tDisplay this help text\n"
|
|
" -b <size> | --blue-noise <size>\tGenerate blue noise "
|
|
"square with "
|
|
"size\n"
|
|
" --usecl | --nousecl\t\t\tUse/Disable OpenCL (enabled by "
|
|
"default)\n"
|
|
" -t <int> | --threads <int>\t\tUse CPU thread count when "
|
|
"not using "
|
|
"OpenCL\n"
|
|
" -o <filename> | --output <filename>\tOutput filename to "
|
|
"use\n"
|
|
" --overwrite\t\t\t\tEnable overwriting of file (default "
|
|
"disabled)\n"
|
|
" --usevulkan | --nousevulkan\t\t\tUse/Disable Vulkan (enabled "
|
|
"by default)\n";
|
|
}
|
|
|
|
bool Args::ParseArgs(int argc, char **argv) {
|
|
--argc;
|
|
++argv;
|
|
while (argc > 0) {
|
|
if (std::strcmp(argv[0], "-h") == 0 ||
|
|
std::strcmp(argv[0], "--help") == 0) {
|
|
DisplayHelp();
|
|
return true;
|
|
} else if (std::strcmp(argv[0], "--usecl") == 0) {
|
|
use_opencl_ = true;
|
|
} else if (std::strcmp(argv[0], "--nousecl") == 0) {
|
|
use_opencl_ = false;
|
|
} else if (std::strcmp(argv[0], "--overwrite") == 0) {
|
|
overwrite_file_ = true;
|
|
} else if (argc > 1 && (std::strcmp(argv[0], "-b") == 0 ||
|
|
std::strcmp(argv[0], "--blue-noise") == 0)) {
|
|
generate_blue_noise_ = true;
|
|
blue_noise_size_ = std::strtoul(argv[1], nullptr, 10);
|
|
if (blue_noise_size_ == 0) {
|
|
std::cout << "ERROR: Failed to parse size for blue-noise" << std::endl;
|
|
generate_blue_noise_ = false;
|
|
}
|
|
--argc;
|
|
++argv;
|
|
} else if (argc > 1 && (std::strcmp(argv[0], "-t") == 0 ||
|
|
std::strcmp(argv[0], "--threads") == 0)) {
|
|
threads_ = std::strtoul(argv[1], nullptr, 10);
|
|
if (threads_ == 0) {
|
|
std::cout << "ERROR: Failed to parse thread count, using 4 by "
|
|
"default"
|
|
<< std::endl;
|
|
threads_ = 4;
|
|
}
|
|
--argc;
|
|
++argv;
|
|
} else if (argc > 1 && (std::strcmp(argv[0], "-o") == 0 ||
|
|
std::strcmp(argv[0], "--output") == 0)) {
|
|
output_filename_ = std::string(argv[1]);
|
|
--argc;
|
|
++argv;
|
|
} else if (std::strcmp(argv[0], "--usevulkan") == 0) {
|
|
use_vulkan_ = true;
|
|
} else if (std::strcmp(argv[0], "--nousevulkan") == 0) {
|
|
use_vulkan_ = false;
|
|
} else {
|
|
std::cout << "WARNING: Ignoring invalid input \"" << argv[0] << "\""
|
|
<< std::endl;
|
|
}
|
|
--argc;
|
|
++argv;
|
|
}
|
|
|
|
return false;
|
|
}
|