Увеличить пользовательский валидатор для Enum - PullRequest
Новая
       24

Увеличить пользовательский валидатор для Enum

20 голосов
/ 06 марта 2011

Я пытаюсь проверить ввод командной строки в Enum, который я определил, но получаю ошибки компилятора. Я использовал Обработка сложных опций с параметрами_программы Boost в качестве примера для работы. 

namespace po = boost::program_options;

namespace Length
{

enum UnitType
{
    METER,
    INCH
};

}

void validate(boost::any& v, const std::vector<std::string>& values, Length::UnitType*, int)
{
    Length::UnitType unit;

    if (values.size() < 1)
    {   
        throw boost::program_options::validation_error("A unit must be specified");
    }   

    // make sure no previous assignment was made
    //po::validators::check_first_occurence(v); // tried this but compiler said it couldn't find it
    std::string input = values.at(0);
    //const std::string& input = po::validators::get_single_string(values); // tried this but compiler said it couldn't find it

    // I'm just trying one for now
    if (input.compare("inch") == 0)
    {
        unit = Length::INCH;
    }   

    v = boost::any(unit);
}

// int main(int argc, char *argv[]) not included

И чтобы сэкономить в том числе больше кода, чем необходимо, я добавляю опцию следующим образом: 

po::options_description config("Configuration");
config.add_options()
    ("to-unit", po::value<std::vector<Length::UnitType> >(), "The unit(s) of length to convert to")
;

Если нужна ошибка компилятора, я могу опубликовать ее, но надеялся, что вопрос будет выглядеть просто. Я пытался искать примеры, но единственный другой пример, который я действительно смог найти, это examples / regex.cpp с сайта Boost .

  1. Есть ли разница между моим сценарием и найденными примерами, кроме того, что мой - это Enum, где другие - это структуры? РЕДАКТИРОВАТЬ: Мой сценарий не требует перегрузки пользовательского валидатора.
  2. Есть ли способ перегрузить validate метод для Enum? РЕДАКТИРОВАТЬ: Не требуется.

1 Ответ

29 голосов
/ 06 марта 2011

В вашем случае вам просто нужно перегрузить operator>>, чтобы извлечь Length::Unit из istream, как показано здесь: 

#include <iostream>
#include <boost/foreach.hpp>
#include <boost/program_options.hpp>

namespace Length
{

enum Unit {METER, INCH};

std::istream& operator>>(std::istream& in, Length::Unit& unit)
{
    std::string token;
    in >> token;
    if (token == "inch")
        unit = Length::INCH;
    else if (token == "meter")
        unit = Length::METER;
    else 
        in.setstate(std::ios_base::failbit);
    return in;
}

};

typedef std::vector<Length::Unit> UnitList;

int main(int argc, char* argv[])
{
    UnitList units;

    namespace po = boost::program_options;
    po::options_description options("Program options");
    options.add_options()
        ("to-unit",
             po::value<UnitList>(&units)->multitoken(),
             "The unit(s) of length to convert to")
        ;

    po::variables_map vm;
    po::store(po::parse_command_line(argc, argv, options), vm);
    po::notify(vm);

    BOOST_FOREACH(Length::Unit unit, units)
    {
        std::cout << unit << " ";
    }
    std::cout << "\n";

    return 0;
}

Пользовательский валидатор не требуется.

...