MultiSwitchArg.h

Go to the documentation of this file.

// -*- Mode: c++; c-basic-offset: 4; tab-width: 4; -*-
 
 
/****************************************************************************** 
*
*  file:  MultiSwitchArg.h
*
*  Copyright (c) 2003, Michael E. Smoot .
*  Copyright (c) 2004, Michael E. Smoot, Daniel Aarno.
*  Copyright (c) 2005, Michael E. Smoot, Daniel Aarno, Erik Zeek.
*  Copyright (c) 2017, Google LLC
*  All rights reserved.
*
*  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 TCLAP_MULTI_SWITCH_ARG_H
#define TCLAP_MULTI_SWITCH_ARG_H
 
#include <string>
#include <vector>
 
#include <tclap/SwitchArg.h>
 
namespace TCLAP {
 
class MultiSwitchArg : public SwitchArg
{
    protected:
 
        int _value;
 
        int _default;
 
    public:
 
        MultiSwitchArg(const std::string& flag, 
                const std::string& name,
                const std::string& desc,
                int init = 0,
                Visitor* v = NULL);
 
 
        MultiSwitchArg(const std::string& flag, 
                const std::string& name,
                const std::string& desc,
                CmdLineInterface& parser,
                int init = 0,
                Visitor* v = NULL);
 
 
        virtual bool processArg(int* i, std::vector<std::string>& args); 
 
        int getValue() const { return _value; }
 
    operator bool() const { return _value != 0; }
 
        std::string shortID(const std::string& val) const;
 
        std::string longID(const std::string& val) const;
        
        void reset();
 
};
 
//BEGIN MultiSwitchArg.cpp
inline MultiSwitchArg::MultiSwitchArg(const std::string& flag,
                    const std::string& name,
                    const std::string& desc,
                    int init,
                    Visitor* v )
: SwitchArg(flag, name, desc, false, v),
_value( init ),
_default( init )
{ }
 
inline MultiSwitchArg::MultiSwitchArg(const std::string& flag,
                    const std::string& name, 
                    const std::string& desc, 
                    CmdLineInterface& parser,
                    int init,
                    Visitor* v )
: SwitchArg(flag, name, desc, false, v),
_value( init ),
_default( init )
{ 
    parser.add( this );
}
 
inline bool MultiSwitchArg::processArg(int *i, std::vector<std::string>& args)
{
    if ( _ignoreable && Arg::ignoreRest() )
        return false;
 
    if ( argMatches( args[*i] ))
    {
        // so the isSet() method will work
        _alreadySet = true;
 
        // Matched argument: increment value.
        ++_value;
 
        _checkWithVisitor();
 
        return true;
    }
    else if ( combinedSwitchesMatch( args[*i] ) )
    {
        // so the isSet() method will work
        _alreadySet = true;
 
        // Matched argument: increment value.
        ++_value;
 
        // Check for more in argument and increment value.
        while ( combinedSwitchesMatch( args[*i] ) ) 
            ++_value;
 
        _checkWithVisitor();
 
        return false;
    }
    else
        return false;
}
 
inline std::string 
MultiSwitchArg::shortID(const std::string& val) const
{
    return Arg::shortID(val) + " ...";
}
 
inline std::string 
MultiSwitchArg::longID(const std::string& val) const
{
    return Arg::longID(val) + "  (accepted multiple times)";
}
 
inline void
MultiSwitchArg::reset()
{
    MultiSwitchArg::_value = MultiSwitchArg::_default;
}
 
//END MultiSwitchArg.cpp
 
} //namespace TCLAP
 
#endif