add_to_po.hpp

Go to the documentation of this file.

#ifndef MLPACK_BINDINGS_CLI_ADD_TO_PO_HPP
#define MLPACK_BINDINGS_CLI_ADD_TO_PO_HPP

#include <mlpack/core/util/param_data.hpp>
#include <boost/program_options.hpp>
#include <mlpack/core/util/is_std_vector.hpp>
#include "map_parameter_name.hpp"

namespace mlpack {
namespace bindings {
namespace cli {

template<typename T>
void AddToPO(const std::string& boostName,
             const std::string& descr,
             boost::program_options::options_description& desc,
             const typename boost::disable_if<util::IsStdVector<T>>::type* = 0,
             const typename boost::disable_if<std::is_same<T, bool>>::type* = 0)
{
  desc.add_options()(boostName.c_str(), boost::program_options::value<T>(),
      descr.c_str());
}

template<typename T>
void AddToPO(const std::string& boostName,
             const std::string& descr,
             boost::program_options::options_description& desc,
             const typename boost::enable_if<util::IsStdVector<T>>::type* = 0,
             const typename boost::disable_if<std::is_same<T, bool>>::type* = 0)
{
  desc.add_options()(boostName.c_str(),
      boost::program_options::value<T>()->multitoken(), descr.c_str());
}

template<typename T>
void AddToPO(const std::string& boostName,
             const std::string& descr,
             boost::program_options::options_description& desc,
             const typename boost::disable_if<util::IsStdVector<T>>::type* = 0,
             const typename boost::enable_if<std::is_same<T, bool>>::type* = 0)
{
  desc.add_options()(boostName.c_str(), descr.c_str());
}

template<typename T>
void AddToPO(const util::ParamData& d,
             const void* /* input */,
             void* output)
{
  // Cast boost::program_options::options_description object.
  boost::program_options::options_description* desc =
      (boost::program_options::options_description*) output;

  // Generate the name to be given to boost::program_options.
  const std::string mappedName =
      MapParameterName<typename std::remove_pointer<T>::type>(d.name);
  std::string boostName = (d.alias != '\0') ? mappedName + "," +
      std::string(1, d.alias) : mappedName;

  // Note that we have to add the option as type equal to the mapped type, not
  // the true type of the option.
  AddToPO<typename ParameterType<typename std::remove_pointer<T>::type>::type>(
      boostName, d.desc, *desc);
}

} // namespace cli
} // namespace bindings
} // namespace mlpack

#endif