arg_parser.h

Go to the documentation of this file.

 
#ifndef HERMES_COMMON_ARG_PARSER_H
#define HERMES_COMMON_ARG_PARSER_H
 
#include <hermes/common/defs.h>
#include <hermes/common/str.h>
#include <string>
#include <vector>
#include <map>
#include <sstream>
 
namespace hermes {
 
// *********************************************************************************************************************
//                                                                                                          ArgParser
// *********************************************************************************************************************
class ArgParser {
public:
  // *******************************************************************************************************************
  //                                                                                                     CONSTRUCTORS
  // *******************************************************************************************************************
  explicit ArgParser(std::string bin = "", std::string description = "");
  // *******************************************************************************************************************
  //                                                                                                          METHODS
  // *******************************************************************************************************************
  //                                                                                                          parsing
  bool parse(int argc, const char **argv, bool verbose_parsing = false);
  //                                                                                                        arguments
  void addArgument(const std::string &name,
                   const std::string &description = "",
                   bool is_required = false);
  template<typename T>
  T get(const std::string &name, const T &default_value = {}) const {
    const auto &it = m_.find(name);
    if (it == m_.end() || !arguments_[it->second].given || arguments_[it->second].values.empty())
      return default_value;
    std::istringstream in(arguments_[it->second].values[0]);
    T t{};
    in >> t;
    return t;
  }
  template<typename T>
  std::vector<T> getList(const std::string &name, const std::vector<T> &default_value = {}) const {
    const auto &it = m_.find(name);
    if (it == m_.end() || !arguments_[it->second].given || arguments_[it->second].values.empty())
      return default_value;
    // expect a list separated by ',' or ' '
    // expect string: "v1 ... vn" or "v1,...,vn" with no enclosing characters
    auto s_values = hermes::Str::split(arguments_[it->second].values[0], " ,");
    std::vector<T> values;
    for (auto &value : s_values) {
      std::istringstream in(value);
      T t{};
      in >> t;
      values.template emplace_back(t);
    }
    return values;
  }
  [[nodiscard]] bool check(const std::string &name) const;
  void printHelp() const;
 
private:
  struct Argument {
    std::string description{};
    std::vector<std::string> names{};
    std::vector<std::string> values{};
    bool required{false};
    bool given{false};
  };
  std::map<std::string, u64> m_;
  std::vector<Argument> arguments_;
  std::string bin_;
  std::string description_;
};
 
}
 
#endif //HERMES_COMMON_ARG_PARSER_H