Обрабатывать сложные параметры с помощью Boost's program_options - PullRequest
Новая
       52

Обрабатывать сложные параметры с помощью Boost's program_options

16 голосов
/ 29 мая 2010

У меня есть программа, которая генерирует графики с использованием различных многоуровневых моделей. Каждая многоуровневая модель состоит из генерации начального графа меньшего размера (скажем, 50 узлов), который можно создать из нескольких моделей (например, для каждого возможного ребра, включите его с вероятностью p).

После генерации начального графа граф расширяется до более крупного (скажем, 1000 узлов) с использованием одной из других моделей.

На каждом из двух этапов каждая модель требует различного количества параметров.

Я бы хотел, чтобы program_options анализировали различные возможные параметры в соответствии с названиями моделей.

Например, скажем, у меня есть две модели начальных графиков: SA, которая имеет 1 параметр, и SB, которая имеет два. Также для части расширения у меня есть две модели: A и B, опять же с 1 и 2 параметрами, соответственно. Я хотел бы иметь возможность сделать что-то вроде: 

./graph_generator --seed=SA 0.1 --expansion=A 0.2
./graph_generator --seed=SB 0.1 3 --expansion=A 0.2
./graph_generator --seed=SA 0.1 --expansion=B 10 20
./graph_generator --seed=SB 0.1 3 --expansion=B 10 20

и правильно проанализировать параметры. Это вообще возможно?

1 Ответ

23 голосов
/ 30 мая 2010

Используя пользовательский валидатор и boost :: program_options :: value :: multitoken , вы можете достичь желаемого результата: 

#include <iostream>
#include <boost/lexical_cast.hpp>
#include <boost/optional.hpp>
#include <boost/program_options.hpp>

// Holds parameters for seed/expansion model
struct Model
{
    std::string type;
    boost::optional<float> param1;
    boost::optional<float> param2;
};

// Called by program_options to parse a set of Model arguments
void validate(boost::any& v, const std::vector<std::string>& values,
              Model*, int)
{
    Model model;
    // Extract tokens from values string vector and populate Model struct.
    if (values.size() == 0)
    {
        throw boost::program_options::validation_error(
            "Invalid model specification");
    }
    model.type = values.at(0); // Should validate for A/B
    if (values.size() >= 2)
        model.param1 = boost::lexical_cast<float>(values.at(1));
    if (values.size() >= 3)
        model.param2 = boost::lexical_cast<float>(values.at(2));

    v = model;
}

int main(int argc, char* argv[])
{
    Model seedModel, expansionModel;

    namespace po = boost::program_options;
    po::options_description options("Generic options");
    options.add_options()
        ("seed",
             po::value<Model>(&seedModel)->multitoken(),
             "seed graph model")
        ("expansion",
             po::value<Model>(&expansionModel)->multitoken(),
             "expansion model")
        ;

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

    std::cout << "Seed type: " << seedModel.type << "\n";
    if (seedModel.param1)
        std::cout << "Seed param1: " << *(seedModel.param1) << "\n";
    if (seedModel.param2)
        std::cout << "Seed param2: " << *(seedModel.param2) << "\n";

    std::cout << "Expansion type: " << expansionModel.type << "\n";
    if (expansionModel.param1)
        std::cout << "Expansion param1: " << *(expansionModel.param1) << "\n";
    if (expansionModel.param2)
        std::cout << "Expansion param2: " << *(expansionModel.param2) << "\n";

    return 0;
}

Функция validate, вероятно, нуждается в большей строгости, но вы поняли идею.

Это компилируется и работает для меня.

...