X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fcore%2Fgetopt.h;h=94e8b1e4af35e598a6e1f99600f837aef916cfbb;hb=ae48dac6a368dbf68b1ec976254a5f896c5b737b;hp=7e5e25f5a1a9ee9994995cc9c77ca4355db8e8b7;hpb=cfc8e0b7b15ea505bd6a6a9599cbc5ce1e316963;p=libs%2Fcore.git diff --git a/source/core/getopt.h b/source/core/getopt.h index 7e5e25f..94e8b1e 100644 --- a/source/core/getopt.h +++ b/source/core/getopt.h @@ -27,14 +27,16 @@ public: class OptBase { public: - OptBase &set_help(const std::string &h) { help=h; return *this; } + OptBase &set_help(const std::string &); + OptBase &set_help(const std::string &, const std::string &); char get_short() const { return shrt; } const std::string &get_long() const { return lng; } ArgType get_arg_type() const { return arg_type; } const std::string &get_help() const { return help; } + const std::string &get_metavar() const { return metavar; } unsigned get_seen_count() const { return seen_count; } - virtual void process()=0; - virtual void process(const std::string &)=0; + void process(); + void process(const std::string &); virtual ~OptBase() { } protected: char shrt; @@ -42,8 +44,11 @@ public: ArgType arg_type; unsigned seen_count; std::string help; + std::string metavar; - OptBase(char s, const std::string &l, ArgType a): shrt(s), lng(l), arg_type(a), seen_count(0) { } + OptBase(char, const std::string &, ArgType); + virtual void store()=0; + virtual void store(const std::string &)=0; }; private: @@ -53,19 +58,10 @@ private: public: Option(char s, const std::string &l, T &d, ArgType a): OptBase(s, l, a), data(d) { } - virtual void process() - { - if(arg_type==REQUIRED_ARG) - throw UsageError("--"+lng+" requires an argument"); - process_(); - ++seen_count; - } + virtual void store() { } - virtual void process(const std::string &a) + virtual void store(const std::string &a) { - if(arg_type==NO_ARG) - throw UsageError("--"+lng+" takes no argument"); - T tmp; std::istringstream ss(a); ss>>tmp; @@ -73,12 +69,9 @@ private: throw UsageError("Invalid argument for --"+lng); data=tmp; - ++seen_count; } private: T &data; - - void process_() { } }; template @@ -88,12 +81,9 @@ private: ListOption(char s, const std::string &l, T &d, ArgType a): OptBase(s, l, a), data(d) { if(arg_type!=REQUIRED_ARG) throw Exception("ListOption with arg_type!=REQUIRED makes no sense"); } - virtual void process() - { - throw UsageError("--"+lng+" requires an argument"); - } + virtual void store() { } - virtual void process(const std::string &a) + virtual void store(const std::string &a) { typename T::value_type tmp; std::istringstream ss(a); @@ -102,7 +92,6 @@ private: throw UsageError("Invalid argument for --"+lng); data.push_back(tmp); - ++seen_count; } private: T &data; @@ -139,8 +128,14 @@ private: unsigned process_short(const char *const *); }; -template<> inline void GetOpt::Option::process_() { data=true; } -template<> inline void GetOpt::Option::process_() { ++data; } +template<> inline void GetOpt::Option::store() { data=true; } +template<> inline void GetOpt::Option::store() { ++data; } + +template<> inline void GetOpt::Option::store(const std::string &a) +{ data=a; } + +template<> inline void GetOpt::ListOption >::store(const std::string &a) +{ data.push_back(a); } } // namespace Msp