src/Utils/Args.h

Go to the documentation of this file.

/*
* Copyright (C) 2005 Norman Casagrande
* For informations write to nova77@gmail.com
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*
*/

#ifndef __ARGS_H
#define __ARGS_H

#include <map>
#include <vector>
#include <set>
#include <string>
#include <iostream>
#include <sstream>

using namespace std;


namespace nor_utils {

enum ArgsOutType
{
   AOT_OK, 
   AOT_NO_ARGUMENTS, 
   AOT_UNKOWN_ARGUMENT, 
   AOT_TOO_FEW_VALUES 
};

class Args
{
public:

   Args() : _currentGroup("general"), _maxColumns(60) {}

   void setGroup( const string& groupName ) { _currentGroup = groupName; }

   void declareArgument( const string& name );

   void declareArgument( const string& name, const string& description, const int numOptions = 0,
                         const string& options = "");

   void printGroup( const string& groupName, ostream& out = cout, const int indSpaces = 3 );

   ArgsOutType   readArguments( const int argc, char* argv[] );

   bool             hasArgument(const string& argument) const
                            { return _resArgs.find(argument) != _resArgs.end(); }

   vector<string>&  getValuesVector(const string& argument);

   template <typename T>
   void getValue(const string& argument, int index, T& valueToFill)
   { 
      if ( _resArgs.find(argument) == _resArgs.end() )
      {
         cerr << "ERROR: Looking for an argument (" << argument << ") that does not exists!" << endl;
         exit(1);
      }

      stringstream ss(_resArgs[argument][index]);    
      ss >> valueToFill;
   }


private:

   string getWrappedString(const string& str, const int leftSpace = 0, 
                           const bool spacesInFirstLine = true) const;

   struct Argument
   {
      Argument() {}

      Argument(const string& name, const string& description, 
               const int numOptions = 0, const string& options = "")
         : name(name), description(description), numOptions(numOptions), options(options) {}

      string name; 
      string description; 
      int numOptions; 
      string options; 
   };

   string  _currentGroup;

   int     _maxColumns;

   map< string, vector<Argument> >       _groupedList; 

   map< string, int >            _argsList; 
   map< string, vector<string> > _resArgs; 
};

}

#endif // __ARGS_H

---