]> git.tdb.fi Git - libs/core.git/commitdiff
Fix GetOpt to get strings correctly
authorMikko Rasa <tdb@tdb.fi>
Fri, 11 Sep 2009 17:34:36 +0000 (17:34 +0000)
committerMikko Rasa <tdb@tdb.fi>
Fri, 11 Sep 2009 17:34:36 +0000 (17:34 +0000)
Handle missing/extraneous arguments in OptBase

source/core/getopt.cpp
source/core/getopt.h

index 8fd13a0c0593bc28dab3f276597b8de8a2cf6dd7..be1d151a3ee8e704af0b10b79ed42c7a8b787572 100644 (file)
@@ -247,4 +247,22 @@ GetOpt::OptBase &GetOpt::OptBase::set_help(const string &h, const string &m)
        return *this;
 }
 
+void GetOpt::OptBase::process()
+{
+       if(arg_type==REQUIRED_ARG)
+               throw UsageError("--"+lng+" requires an argument");
+       ++seen_count;
+
+       store();
+}
+
+void GetOpt::OptBase::process(const string &arg)
+{
+       if(arg_type==NO_ARG)
+               throw UsageError("--"+lng+" takes no argument");
+       ++seen_count;
+
+       store(arg);
+}
+
 } // namespace Msp
index f6c48597c5220f73998f1ced0ccc6faf5411a445..94e8b1e4af35e598a6e1f99600f837aef916cfbb 100644 (file)
@@ -35,8 +35,8 @@ public:
                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;
@@ -47,6 +47,8 @@ public:
                std::string metavar;
 
                OptBase(char, const std::string &, ArgType);
+               virtual void store()=0;
+               virtual void store(const std::string &)=0;
        };
 
 private:
@@ -56,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;
@@ -76,12 +69,9 @@ private:
                                throw UsageError("Invalid argument for --"+lng);
 
                        data=tmp;
-                       ++seen_count;
                }
        private:
                T &data;
-
-               void process_() { }
        };
 
        template<typename T>
@@ -91,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);
@@ -105,7 +92,6 @@ private:
                                throw UsageError("Invalid argument for --"+lng);
 
                        data.push_back(tmp);
-                       ++seen_count;
                }
        private:
                T &data;
@@ -142,8 +128,14 @@ private:
        unsigned process_short(const char *const *);
 };
 
-template<> inline void GetOpt::Option<bool>::process_()     { data=true; }
-template<> inline void GetOpt::Option<unsigned>::process_() { ++data; }
+template<> inline void GetOpt::Option<bool>::store()     { data=true; }
+template<> inline void GetOpt::Option<unsigned>::store() { ++data; }
+
+template<> inline void GetOpt::Option<std::string>::store(const std::string &a)
+{ data=a; }
+
+template<> inline void GetOpt::ListOption<std::list<std::string> >::store(const std::string &a)
+{ data.push_back(a); }
 
 } // namespace Msp