src/Utils/Args.cpp

/*
* 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
*
*/

#include "Args.h"

namespace nor_utils {


void Args::declareArgument( const string& name )
{
   _argsList[name] = 0;
}

// -----------------------------------------------------------------------

void Args::declareArgument( const string& name, const string& description, const int numOptions,
                        const string& options )
{
   // Add to standard argument list
   _argsList[name] = numOptions;

   // not very efficient but speed is not an issue here
   _groupedList[_currentGroup].push_back(Argument(name, description, numOptions, options));
}

// -----------------------------------------------------------------------

void Args::printGroup(const string& groupName, ostream& out, const int indSpaces)
{
   out << "\n" << groupName << ":" << endl;

   vector<Argument>& args = _groupedList[groupName];

   vector<Argument>::iterator it;
   for (it = args.begin(); it != args.end(); ++it)
   {
      for (int i = 0; i < indSpaces; ++i)
         out << ' ';

      out << it->name;
      
      if (!it->options.empty())
         out << " " << it->options;
      
      out << ":" << endl;
      out << getWrappedString(it->description, indSpaces+3) << endl;

   }

}

// -----------------------------------------------------------------------

ArgsOutType Args::readArguments( const int argc, char* argv[] )
{
  if (argc < 2 )
    return AOT_NO_ARGUMENTS;

  for (int i = 1; i < argc; ++i)
  {
    string aStr = argv[i];

    if ( _argsList.find(aStr) == _argsList.end() )
    {
      cerr << "ERROR: Unknown argument: " << aStr << endl;
      return AOT_UNKOWN_ARGUMENT;
    }
    else
    {
      // arg exists?
      const int numArgs = _argsList[aStr];
      
      if (numArgs == 0)
        _resArgs[aStr].push_back("");
      else
      {      
        if (argc < i + numArgs + 1)
        {
          cerr << "ERROR: The number of values for " << aStr << " is incorrect!" << endl;
          return AOT_TOO_FEW_VALUES; 
        }
      
        for (int j = 0; j < numArgs; ++j)
          _resArgs[aStr].push_back(argv[++i]);
      }
    }
    
  }
  
  return AOT_OK;
}

// -----------------------------------------------------------------------

vector<string>&  Args::getValuesVector(const string& argument)
{ 
  if ( _resArgs.find(argument) == _resArgs.end() )
  {
    cout << "ERROR: Looking for an argument (" << argument << ") that do not exists!" << endl;
    exit(1);
  }
  
  return _resArgs[argument]; 
}  

// -----------------------------------------------------------------------

string Args::getWrappedString(const string& str, const int leftSpace, const bool spacesInFirstLine) const
{
   string result;
   insert_iterator<string> iIt(result, result.begin());
   string::const_iterator sIt;

   int col = leftSpace;

   if (spacesInFirstLine)
   {
      for (int i = 0; i < leftSpace; ++i)
         *iIt++ = ' ';
   }

   for (sIt = str.begin(); sIt != str.end(); ++sIt)
   {
      if ( (*sIt == ' ' && col > _maxColumns) || *sIt == '\n' )
      {
         col = leftSpace;
         *iIt++ = '\n';
         for (int i = 0; i < leftSpace; ++i)
            *iIt++ = ' ';
      }
      else
      {
         ++col;
         *iIt++ = *sIt;
      }
   }

   return result;
}

// -----------------------------------------------------------------------

} // end of namespace nor_utils

---