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 "adios2/ADIOSPy.h"
#include "adios2/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