R_option.hpp

Go to the documentation of this file.

#ifndef MLPACK_BINDINGS_R_R_OPTION_HPP
#define MLPACK_BINDINGS_R_R_OPTION_HPP
#include <mlpack/core/util/param_data.hpp>
#include "get_param.hpp"
#include "get_printable_param.hpp"
#include "print_input_param.hpp"
#include "print_input_processing.hpp"
#include "print_output_processing.hpp"
#include "print_doc.hpp"
#include "print_serialize_util.hpp"

namespace mlpack {
namespace bindings {
namespace r {

template<typename T>
class ROption
{
 public:
  ROption(const T defaultValue,
          const std::string& identifier,
          const std::string& description,
          const std::string& alias,
          const std::string& cppName,
          const bool required = false,
          const bool input = true,
          const bool noTranspose = false,
          const std::string& /* testName */ = "")
  {
    // Create the ParamData object to give to IO.
    util::ParamData data;
    data.desc = description;
    data.name = identifier;
    data.tname = TYPENAME(T);
    data.alias = alias[0];
    data.wasPassed = false;
    data.noTranspose = noTranspose;
    data.required = required;
    data.input = input;
    data.loaded = false;

    // Only "verbose" will be persistent.
    if (identifier == "verbose")
      data.persistent = true;
    else
      data.persistent = false;
    data.cppType = cppName;

    // Every parameter we'll get from R will have the correct type.
    data.value = boost::any(defaultValue);

    // Restore the parameters for this program.
    if (identifier != "verbose")
      IO::RestoreSettings(IO::ProgramName(), false);

    // Set the function pointers that we'll need.  All of these function
    // pointers will be used by both the program that generates the R, and
    // also the binding itself.  (The binding itself will only use GetParam,
    // GetPrintableParam, and GetRawParam.)
    IO::GetSingleton().functionMap[data.tname]["GetParam"] = &GetParam;
    IO::GetSingleton().functionMap[data.tname]["GetPrintableParam"] =
        &GetPrintableParam;

    // These are used by the R generator.
    IO::GetSingleton().functionMap[data.tname]["PrintDoc"] = &PrintDoc;
    IO::GetSingleton().functionMap[data.tname]["PrintInputParam"] =
        &PrintInputParam;
    IO::GetSingleton().functionMap[data.tname]["PrintOutputProcessing"] =
        &PrintOutputProcessing;
    IO::GetSingleton().functionMap[data.tname]["PrintInputProcessing"] =
        &PrintInputProcessing;
    IO::GetSingleton().functionMap[data.tname]["PrintSerializeUtil"] =
        &PrintSerializeUtil;

    // Add the ParamData object, then store.  This is necessary because we may
    // import more than one .so or .o that uses IO, so we have to keep the
    // options separate.  programName is a global variable from mlpack_main.hpp.
    IO::Add(std::move(data));
    if (identifier != "verbose")
      IO::StoreSettings(IO::ProgramName());
    IO::ClearSettings();
  }
};

} // namespace r
} // namespace bindings
} // namespace mlpack

#endif