genetic-images/src/parseargs.cc

#include "parseargs.hh"
#include <boost/program_options.hpp>
#include <cstdlib>
#include <iostream>
#include <string>

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;
  po::options_description desc("Allowed options");
  desc.add_options()("help,h", "Display this help message")(
      "input,i", po::value<std::string>(), "Input image")(
      "output,o", po::value<std::string>(),
      "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::store(po::parse_command_line(t_ac, t_av, desc), vm);
  po::notify(vm);
  if (vm.count("help") || !vm.count("input")) {
    std::cout << desc << "\n";
    std::exit(1);
  }
  return std::make_tuple(
      vm["input"].as<std::string>(),
      vm.count("output")
          ? vm["output"].as<std::string>()
          : vm["input"].as<std::string>() + std::string{"_output"},
      vm.count("video") ? true : false,
      vm.count("iterations") ? vm["iterations"].as<int>() : DEFAULT_ITERATIONS);
}