3 This file is part of libmspcore
4 Copyright © 2006-2007 Mikko Rasa, Mikkosoft Productions
5 Distributed under the LGPL
7 #ifndef MSP_CORE_GETOPT_H_
8 #define MSP_CORE_GETOPT_H_
30 OptBase &set_help(const std::string &h) { help=h; return *this; }
31 char get_short() const { return shrt; }
32 const std::string &get_long() const { return lng; }
33 ArgType get_arg_type() const { return arg_type; }
34 const std::string &get_help() const { return help; }
35 unsigned get_seen_count() const { return seen_count; }
36 virtual void process()=0;
37 virtual void process(const std::string &)=0;
38 virtual ~OptBase() { }
46 OptBase(char s, const std::string &l, ArgType a): shrt(s), lng(l), arg_type(a), seen_count(0) { }
51 class Option: public OptBase
54 Option(char s, const std::string &l, T &d, ArgType a): OptBase(s, l, a), data(d) { }
56 virtual void process()
58 if(arg_type==REQUIRED_ARG)
59 throw UsageError("--"+lng+" requires an argument");
64 virtual void process(const std::string &a)
67 throw UsageError("--"+lng+" takes no argument");
70 std::istringstream ss(a);
73 throw UsageError("Invalid argument for --"+lng);
85 class ListOption: public OptBase
88 ListOption(char s, const std::string &l, T &d, ArgType a): OptBase(s, l, a), data(d)
89 { if(arg_type!=REQUIRED_ARG) throw Exception("ListOption with arg_type!=REQUIRED makes no sense"); }
91 virtual void process()
93 throw UsageError("--"+lng+" requires an argument");
96 virtual void process(const std::string &a)
98 typename T::value_type tmp;
99 std::istringstream ss(a);
102 throw UsageError("Invalid argument for --"+lng);
111 std::list<OptBase *> opts;
112 std::vector<std::string> args;
117 const std::vector<std::string> &get_args() const { return args; }
120 OptBase &add_option(char s, const std::string &l, T &d, ArgType a=NO_ARG)
121 { opts.push_back(new Option<T>(s, l, d, a)); return *opts.back(); }
124 OptBase &add_option(char s, const std::string &l, std::list<T> &d, ArgType a=REQUIRED_ARG)
125 { opts.push_back(new ListOption<std::list<T> >(s, l, d, a)); return *opts.back(); }
128 OptBase &add_option(const std::string &l, T &d, ArgType a)
129 { return add_option(0, l, d, a); }
131 std::string generate_usage(const std::string &) const;
132 std::string generate_help() const;
133 void operator()(unsigned, const char *const *);
136 OptBase &get_option(char);
137 OptBase &get_option(const std::string &);
138 unsigned process_long(const char *const *);
139 unsigned process_short(const char *const *);
142 template<> inline void GetOpt::Option<bool>::process_() { data=true; }
143 template<> inline void GetOpt::Option<unsigned>::process_() { ++data; }