getopt.hxx

#ifndef HXX_GETOPT
#define HXX_GETOPT

#include <vector>
#include <map>
#include <sstream>
#include <getopt.h>

class Getopt
{
public:
  typedef std::vector<std::string> OptionValues;
  typedef std::map<std::string, OptionValues> OptionsMap;

  Getopt (int argc, char *const * argv, const char *documentation = "Usage: %c [options]");


  void add (const char *longOpt, char shortOpt, int argument,
            const char *description = "", const char *argDescription = "");


  void get ();


  inline std::string usage ();


  inline std::string operator[] (std::string opt) const;

  inline int getCount (std::string opt) const;

  inline const OptionValues & getAll (std::string opt) const;


  inline std::string shift ();

  inline int argc () const;

  inline char const *const * argv () const;

  inline std::string argv (int i) const;

private:
  void addOpt   (const char *longOpt, char shortOpt, int argument,
                 const char *description, const char *argDescription);
  void addUsage (const char *longOpt, char shortOpt, int argument,
                 const char *description, const char *argDescription);


private:
  typedef struct option Option;
  int                 argc_;
  char *const *       argv_;
  std::string         shortOpts_;
  std::ostringstream  usage_;
  std::vector<Option> options_;
  std::map<char,int>  shortIndex_;
  OptionsMap          cl_;
  static const OptionValues None_;
};

inline std::string Getopt::usage ()
{
  return usage_.str();
}

inline std::string Getopt::operator[] (std::string opt) const
{
  OptionsMap::const_iterator it = cl_.find(opt);
  if (it == cl_.end()) {
    return "";
  }

  return it->second.back();
}

inline int Getopt::getCount (std::string opt) const
{
  OptionsMap::const_iterator it = cl_.find(opt);
  if (it == cl_.end()) {
    return 0;
  }

  return it->second.size();
}

inline const Getopt::OptionValues & Getopt::getAll (std::string opt) const
{
  OptionsMap::const_iterator it = cl_.find(opt);
  if (it == cl_.end()) {
    return None_;
  }

  return it->second;
}

inline std::string Getopt::shift ()
{
  if (optind == argc_) {
    return "";
  }

  std::string res = argv_[optind];
  optind++;
  return res;
}

inline int Getopt::argc () const
{
  return argc_-optind;
}

inline char const *const * Getopt::argv () const
{
  return argv_+optind;
}

inline std::string Getopt::argv (int i) const
{
  return argv_[optind+i];
}

template <class T>
bool fromString(T& t, const std::string& s)
{
  std::istringstream iss(s);
  return !(iss >> t).fail();
}

#endif // HXX_GETOPT