SwitchArg.h

Go to the documentation of this file.

// -*- Mode: c++; c-basic-offset: 4; tab-width: 4; -*-
 
/******************************************************************************
 *
 *  file:  SwitchArg.h
 *
 *  Copyright (c) 2003, Michael E. Smoot .
 *  Copyright (c) 2004, Michael E. Smoot, Daniel Aarno.
 *  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_SWITCH_ARG_H
#define TCLAP_SWITCH_ARG_H
 
#include <tclap/Arg.h>
 
#include <string>
#include <vector>
 
namespace TCLAP {
 
class SwitchArg : public Arg {
protected:
    bool _value;
 
    bool _default;
 
public:
    SwitchArg(const std::string &flag, const std::string &name,
              const std::string &desc, bool def = false, Visitor *v = NULL);
 
    SwitchArg(const std::string &flag, const std::string &name,
              const std::string &desc, ArgContainer &parser, bool def = false,
              Visitor *v = NULL);
 
    virtual bool processArg(int *i, std::vector<std::string> &args);
 
    bool combinedSwitchesMatch(std::string &combined);
 
    bool getValue() const { return _value; }
 
    operator bool() const { return _value; }
 
    virtual void reset();
 
private:
    bool isCombinedSwitchToken(const std::string &token) const;
 
    bool lastCombined(std::string &combined);
 
    void commonProcessing();
};
 
// BEGIN SwitchArg.cpp
inline SwitchArg::SwitchArg(const std::string &flag, const std::string &name,
                            const std::string &desc, bool default_val,
                            Visitor *v)
    : Arg(flag, name, desc, false, false, v),
      _value(default_val),
      _default(default_val) {}
 
inline SwitchArg::SwitchArg(const std::string &flag, const std::string &name,
                            const std::string &desc, ArgContainer &parser,
                            bool default_val, Visitor *v)
    : Arg(flag, name, desc, false, false, v),
      _value(default_val),
      _default(default_val) {
    parser.add(this);
}
 
inline bool SwitchArg::lastCombined(std::string &combinedSwitches) {
    for (unsigned int i = 1; i < combinedSwitches.length(); i++)
        if (combinedSwitches[i] != Arg::blankChar()) return false;
 
    return true;
}
 
inline bool SwitchArg::combinedSwitchesMatch(std::string &combinedSwitches) {
    if (!isCombinedSwitchToken(combinedSwitches)) return false;
 
    for (unsigned int i = 1; i < combinedSwitches.length(); i++) {
        if (combinedSwitches[i] != Arg::flagStartString()[0] &&
            matchesShortFlag(std::string(1, combinedSwitches[i]))) {
            // update the combined switches so this one is no longer
            // present this is necessary so that no unlabeled args are
            // matched later in the processing.
            _setBy = Arg::flagStartString() + combinedSwitches[i];
            combinedSwitches[i] = Arg::blankChar();
            return true;
        }
    }
 
    // none of the switches passed in the list match.
    return false;
}
 
inline bool SwitchArg::isCombinedSwitchToken(const std::string &token) const {
    if (token.length() > 0 &&
        token[0] != Arg::flagStartString()[0])
        return false;
 
    if (token.substr(0, Arg::nameStartString().length()) ==
        Arg::nameStartString())
        return false;
 
    if (token.find_first_of(this->getDelimiter()) != std::string::npos)
        return false;
 
    return true;
}
 
inline void SwitchArg::commonProcessing() {
    if (_alreadySet)
        throw(CmdLineParseException("Argument already set!", toString()));
 
    _alreadySet = true;
 
    _value = !_value;
 
    _checkWithVisitor();
}
 
inline bool SwitchArg::processArg(int *i, std::vector<std::string> &args) {
    if (argMatches(args[*i])) {
        // The whole string matches the flag or name string
        _setBy = args[*i];
        commonProcessing();
 
        return true;
    }
 
    if (combinedSwitchesMatch(args[*i])) {
        // A substring matches the flag as part of a combination
        // check again to ensure we don't misinterpret
        // this as a MultiSwitchArg
        if (combinedSwitchesMatch(args[*i]))
            throw(CmdLineParseException("Argument already set!", toString()));
 
        commonProcessing();
 
        // We only want to return true if we've found the last combined
        // match in the string, otherwise return false so that other
        // switches in the combination will have a chance to match.
        return lastCombined(args[*i]);
    }
 
    return false;
}
 
inline void SwitchArg::reset() {
    Arg::reset();
    _value = _default;
}
 
}  // namespace TCLAP
 
#endif  // TCLAP_SWITCH_ARG_H