cmdSwitchArg.h

Go to the documentation of this file.

/****************************************************************************** 
 * 
 *  file:  SwitchArg.h
 * 
 *  Copyright (c) 2003, Michael E. Smoot .
 *  Copyright (c) 2004, Michael E. Smoot, Daniel Aarno.
 *  All rights reverved.
 * 
 *  See the file COPYING in the top directory of this distribution for
 *  more information.
 *  
 *  THE SOFTWARE IS PROVIDED _AS IS_, WITHOUT WARRANTY OF ANY KIND, EXPRESS 
 *  OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 
 *  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 
 *  THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 
 *  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 
 *  FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER 
 *  DEALINGS IN THE SOFTWARE.  
 *  
 *****************************************************************************/ 


#ifndef DtCmdLine_SWITCH_ARG_H
#define DtCmdLine_SWITCH_ARG_H


#include <string>
#include <vector>

#include "cmdArg.h"

#ifdef DtUSE_UTILITIES_NAMESPACE
namespace DtUSE_UTILITIES_NAMESPACE
{
#endif

   namespace DtCmdLine {


      class SwitchArg : public Arg
      {
      protected:

         bool _value;


      public:

         SwitchArg(const std::string& flag, 
                   const std::string& name, 
                   const std::string& desc,
                   bool def,
                   Visitor* v = NULL);


         SwitchArg(const std::string& flag, 
                   const std::string& name, 
                   const std::string& desc,
                   bool def,
                   CmdLineInterface& parser,
                   Visitor* v = NULL);


         virtual bool processArg(int* i, std::vector<std::string>& args); 

         bool combinedSwitchesMatch(std::string& combined);

         bool getValue() const;

         operator bool();

      };

      //BEGIN SwitchArg.cpp
      inline SwitchArg::SwitchArg(const std::string& flag, 
                                  const std::string& name, 
                                  const std::string& desc, 
                                  bool _default,
                                  Visitor* v )
         : Arg(flag, name, desc, false, false, v),
         _value( _default )
      { }

      inline SwitchArg::SwitchArg(const std::string& flag, 
                                  const std::string& name, 
                                  const std::string& desc, 
                                  bool _default,
                                  CmdLineInterface& parser,
                                  Visitor* v )
         : Arg(flag, name, desc, false, false, v),
         _value( _default )
      { 
         parser.add( this );
      }

      inline bool SwitchArg::getValue() const { return _value; }

      inline SwitchArg::operator bool() { return _value; }

      inline bool SwitchArg::combinedSwitchesMatch(std::string& combinedSwitches )
      {
         if ( combinedSwitches[0] != Arg::flagStartString()[0] )
            return false;

         if ( combinedSwitches.substr( 0, Arg::nameStartString().length() ) == 
              Arg::nameStartString() )
            return false;

         for ( unsigned int i = 1; i < combinedSwitches.length(); i++ )
            if ( !_flag.empty() && combinedSwitches[i] == _flag[0] ) 
            {
               //combinedSwitches.erase(i,1);
               combinedSwitches[i] = Arg::blankChar(); 
               return true;
            }

         return false;  
      }


      inline bool SwitchArg::processArg(int *i, std::vector<std::string>& args)
      {
         if ( _ignoreable && Arg::ignoreRest() )
            return false;

         if ( argMatches( args[*i] ) || combinedSwitchesMatch( args[*i] ) )
         {
            bool ret = false;
            if ( argMatches( args[*i] ) )
               ret = true;

            if ( _alreadySet )
               throw(CmdLineParseException("Argument already set!", toString()));   

            _alreadySet = true;

            if ( _value == true )
               _value = false;
            else
               _value = true;

            _checkWithVisitor();

            return ret;
         }
         else
            return false;
      }

      //End SwitchArg.cpp

   } //namespace DtCmdLine

#ifdef DtUSE_UTILITIES_NAMESPACE
}     
#endif

#endif