ADIOSPy.cpp

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

#include <iostream>

#include <mpi4py/mpi4py.h>

#include "ADIOSPy.h"
#include "EnginePy.h"

namespace adios
{

ADIOSPy::ADIOSPy(MPI_Comm mpiComm, const bool debug)
    : ADIOS(mpiComm, adios::Verbose::ERROR, debug)
{
}

ADIOSPy::~ADIOSPy() {}

void ADIOSPy::HelloMPI()
{
  std::cout << "Hello ADIOSPy from rank " << m_RankMPI << "/" << m_SizeMPI
            << "\n";
}

MethodPy &ADIOSPy::DeclareMethodPy(const std::string methodName)
{
  Method &method = DeclareMethod(methodName);
  return *reinterpret_cast<MethodPy *>(&method);
}

VariablePy ADIOSPy::DefineVariablePy(const std::string name,
                                     const pyList localDimensionsPy,
                                     const pyList globalDimensionsPy,
                                     const pyList globalOffsetsPy)
{
  if (m_DebugMode == true)
  {
    if (m_VariablesPyNames.count(name) == 1)
      throw std::invalid_argument("ERROR: Variable " + name +
                                  " is already defined\n");
  }

  m_VariablesPyNames.insert(name);
  return VariablePy(name, localDimensionsPy, globalDimensionsPy,
                    globalOffsetsPy);
}

void ADIOSPy::DefineVariableType(VariablePy &variablePy) {}

EnginePy ADIOSPy::OpenPy(const std::string name, const std::string accessMode,
                         const MethodPy &method, pyObject py_comm)
{
  EnginePy enginePy(*this);

  bool isEmpty = IsEmpty(py_comm);

  if (isEmpty == true) // None
  {
    enginePy.m_Engine = Open(name, accessMode, method);
  }
  else
  {
    if (import_mpi4py() < 0)
      throw std::logic_error(
          "ERROR: could not import mpi4py communicator in Open " + name + "\n");
    MPI_Comm *comm_p = PyMPIComm_Get(py_comm.ptr());
    if (comm_p == nullptr)
      throw std::invalid_argument(
          "ERROR: MPI communicator is nullptr in Open " + name + "\n");

    enginePy.m_Engine = Open(name, accessMode, *comm_p, method);
  }

  return enginePy;
}

} // end namespace