MethodPy.cpp

/*
 * Distributed under the OSI-approved Apache License, Version 2.0.  See
 * accompanying file Copyright.txt for details.
 *
 * MethodPy.cpp
 *
 *  Created on: Mar 14, 2017
 *      Author: wfg
 */

#include <iostream>

#include "MethodPy.h"
#include "adiosPyFunctions.h"

namespace adios
{

#ifdef HAVE_BOOSTPYTHON
namespace py = boost::python;
#endif

#ifdef HAVE_PYBIND11
namespace py = pybind11;
#endif

MethodPy::MethodPy(const std::string type, const bool debugMode)
    : Method(type, debugMode)
{
}

MethodPy::~MethodPy() {}

#ifdef HAVE_BOOSTPYTHON
pyObject MethodPy::SetParametersPy(pyTuple args, pyDict kwargs)
{
  if (py::len(args) > 1)
    throw std::invalid_argument("ERROR: syntax of Method SetParameters "
                                "function is incorrect, only use dictionary\n");

  MethodPy &self = PyCast<MethodPy &>(args[0]);
  self.m_Parameters = DictToMap(kwargs);
  return args[0];
}

pyObject MethodPy::AddTransportPy(pyTuple args, pyDict kwargs)
{
  if (py::len(args) != 2)
    throw std::invalid_argument(
        "ERROR: syntax of Method AddTransport function is incorrect, only use "
        "one string for transport followed by a dictionary for parameters\n");

  MethodPy &self = PyCast<MethodPy &>(args[0]);
  const std::string type = PyCast<std::string>(args[1]);

  auto parameters = DictToMap(kwargs);
  parameters.insert(std::make_pair("transport", type));
  self.m_TransportParameters.push_back(parameters);
  return args[0];
}
#endif

#ifdef HAVE_PYBIND11
void MethodPy::SetParametersPyBind11(pybind11::kwargs kwargs)
{
  this->m_Parameters = KwargsToMap(kwargs);
}

void MethodPy::AddTransportPyBind11(const std::string type,
                                    pybind11::kwargs kwargs)
{
  auto parameters = KwargsToMap(kwargs);
  parameters.insert(std::make_pair("transport", type));
  this->m_TransportParameters.push_back(parameters);
}
#endif

void MethodPy::PrintAll() const
{
  std::cout << "Method parameters\n";
  for (const auto &param : m_Parameters)
    std::cout << "Parameter: " << param.first << "\t Value: " << param.second
              << "\n";

  std::cout << "\n";
  std::cout << "Transport Parameters\n";

  for (const auto &transportParameters : m_TransportParameters)
  {
    std::cout << "Transport:\n";
    for (const auto &param : transportParameters)
      std::cout << "Parameter: " << param.first << "\t Value: " << param.second
                << "\n";

    std::cout << "\n";
  }
}

} // end namespace