CommandLineOptions.hh

Go to the documentation of this file.

#ifndef COMMAND_LINE_OPTIONS_HH_
#define COMMAND_LINE_OPTIONS_HH_
 
#include <boost/program_options.hpp>
#include <string>
#include <vector>
 
#include "Verbosity.hh"
#include "ModelOptions.hh"
#include "Gym.hh"
 
class CommandLineOptions {
 
public:
 
  CommandLineOptions();
 
  int parseCommandLineOptions(int argc, char * argv[]);
 
  boost::program_options::variables_map & options() { return varMap; }
 
  boost::program_options::variables_map const & options() const 
  { return varMap; }
 
  std::string const & inputFile() const { return inputFileName; }
 
  Verbosity::Level verbosity() const { return (Verbosity::Level) verbosityLevel;
}
 
  ModelOptions::ReachType reachSolver() const {
    if (varMap["reach-with"].as<std::string>() == std::string("glop"))
      return ModelOptions::ReachType::glop;
    if (varMap["reach-with"].as<std::string>() == std::string("iter"))
      return ModelOptions::ReachType::iter;
    throw std::out_of_range("Invalid solver type");
  }
 
  ModelOptions::SSPType sspSolver() const {
    if (varMap["ssp-with"].as<std::string>() == std::string("glop"))
      return ModelOptions::SSPType::glop;
    if (varMap["ssp-with"].as<std::string>() == std::string("iter"))
      return ModelOptions::SSPType::iter;
    if (varMap["ssp-with"].as<std::string>() == std::string("poly"))
      return ModelOptions::SSPType::poly;
    throw std::out_of_range("Invalid solver type");
  }
 
  void fillModelOptions(ModelOptions & m) const;
 
  bool learnEnabled() const {
    if (varMap["learn"].as<std::string>() != std::string("none"))
      return true;
    else
      return false;
  }
 
  bool QEnabled() const {
    if (varMap["learn"].as<std::string>() == std::string("Q"))
      return true;
    else
      return false;
  }
 
  bool DQEnabled() const {
    if (varMap["learn"].as<std::string>() == std::string("DQ"))
      return true;
    else
      return false;
  }
 
  bool SLEnabled() const {
    if (varMap["learn"].as<std::string>() == std::string("SL"))
      return true;
    else
      return false;
  }
 
  bool AQEnabled() const {
    if (varMap["learn"].as<std::string>() == std::string("AQ"))
      return true;
    else
      return false;
  }
 
  GymOptions::GymRewardTypes getRewardType() const {
    if (varMap["reward-type"].as<std::string>() == std::string("default-type"))
      return GymOptions::GymRewardTypes::default_type;
    else if (varMap["reward-type"].as<std::string>() == std::string("prism"))
      return GymOptions::GymRewardTypes::prism;
    else if (varMap["reward-type"].as<std::string>() == std::string("zeta-reach"))
      return GymOptions::GymRewardTypes::zeta_reach;
    else if (varMap["reward-type"].as<std::string>() == std::string("zeta-acc"))
      return GymOptions::GymRewardTypes::zeta_acc;
    else if (varMap["reward-type"].as<std::string>() == std::string("zeta-discount"))
      return GymOptions::GymRewardTypes::zeta_discount;
    else if (varMap["reward-type"].as<std::string>() == std::string("reward-on-acc"))
      return GymOptions::GymRewardTypes::reward_on_acc;
    else if (varMap["reward-type"].as<std::string>() == std::string("multi-discount"))
      return GymOptions::GymRewardTypes::multi_discount;
    else if (varMap["reward-type"].as<std::string>() == std::string("parity"))
      return GymOptions::GymRewardTypes::parity;
    else if (varMap["reward-type"].as<std::string>() == std::string("pri-tracker"))
      return GymOptions::GymRewardTypes::pri_tracker;
    else if (varMap["reward-type"].as<std::string>() == std::string("lexo"))
      return GymOptions::GymRewardTypes::lexo;
    else if (varMap["reward-type"].as<std::string>() == std::string("avg"))
      return GymOptions::GymRewardTypes::avg;
    else
      throw(std::runtime_error("Reward type invalid.")); 
  }
 
private:
  boost::program_options::options_description visible;
  boost::program_options::options_description hidden;
  boost::program_options::options_description cmdline_options;
  boost::program_options::positional_options_description posOpt;
  boost::program_options::variables_map varMap;
 
  // Variables that reflect the command line options go here
 
  std::string inputFileName;
 
  int verbosityLevel;
 
};
 
#endif