phundrak/genetic-images
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

added number of iterations to arguments list

Browse Source
This commit is contained in:
Phuntsok Drak-pa 2019-03-19 14:22:44 +01:00
parent 3654e56d2e
commit 069b739174
2 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/parseargs.cc
View File

@ -4,13 +4,18 @@
#include <iostream> #include <iostream>
#include <string> #include <string>
std::tuple<std::string, std::string, bool> parse_args(int t_ac, char **t_av) { constexpr int DEFAULT_ITERATIONS = 5000;
std::tuple<std::string, std::string, bool, int> parse_args(int t_ac,
char **t_av) {
namespace po = boost::program_options; namespace po = boost::program_options;
po::options_description desc("Allowed options"); po::options_description desc("Allowed options");
desc.add_options()("help,h", "Display this help message")( desc.add_options()("help,h", "Display this help message")(
"input,i", po::value<std::string>(), "Input image")( "input,i", po::value<std::string>(), "Input image")(
"output,o", po::value<std::string>(), "output,o", po::value<std::string>(),
"Image or video output path")("video,v", "Enable video output"); "Image or video output path (default: input path + \"_output\")")(
"iterations,n", po::value<int>(),
"Number of iterations (default: 5000)")("video,v", "Enable video output");
po::variables_map vm; po::variables_map vm;
po::store(po::parse_command_line(t_ac, t_av, desc), vm); po::store(po::parse_command_line(t_ac, t_av, desc), vm);
po::notify(vm); po::notify(vm);
@ -18,9 +23,11 @@ std::tuple<std::string, std::string, bool> parse_args(int t_ac, char **t_av) {
std::cout << desc << "\n"; std::cout << desc << "\n";
std::exit(1); std::exit(1);
} }
return std::make_tuple(vm["input"].as<std::string>(), return std::make_tuple(
(vm.count("output")) ? vm["output"].as<std::string>() vm["input"].as<std::string>(),
: vm["input"].as<std::string>() + vm.count("output")
std::string{"_output"}, ? vm["output"].as<std::string>()
(vm.count("video")) ? true : false); : vm["input"].as<std::string>() + std::string{"_output"},
vm.count("video") ? true : false,
vm.count("iterations") ? vm["iterations"].as<int>() : DEFAULT_ITERATIONS);
} }

2
src/parseargs.hh
View File

@ -4,6 +4,6 @@
#include <string> #include <string>
#include <tuple> #include <tuple>
std::tuple<std::string, std::string, bool> parse_args(int, char**); std::tuple<std::string, std::string, bool, int> parse_args(int, char **);
#endif /* GENETIC_IMAGE_INCLUDE_GENIMG_PARSEARGS_HH_ */ #endif /* GENETIC_IMAGE_INCLUDE_GENIMG_PARSEARGS_HH_ */