Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
/*
* gluePyBind11.cpp
*
* Created on: Mar 16, 2017
* Author: wfg
*/
#include <stdexcept>
#include <mpi4py/mpi4py.h>
#include "pybind11/pybind11.h"
#include "ADIOSPy.h"
namespace py = pybind11;
adios::ADIOSPy ADIOSPy( py::object py_comm, const bool debug )
{
MPI_Comm* comm_p = PyMPIComm_Get( py_comm.ptr() );
if( comm_p == NULL ) py::error_already_set();
return adios::ADIOSPy( *comm_p, debug );
}
PYBIND11_PLUGIN( ADIOSPy )
{
if (import_mpi4py() < 0) throw std::runtime_error("ERROR: mpi4py not loaded correctly\n"); /* Python 2.X */
py::module m( "ADIOSPy", "ADIOS Python bindings using pybind11" );
m.def("ADIOSPy", &ADIOSPy, "Function that creates an ADIOS object" );
py::class_<adios::ADIOSPy>( m, "ADIOS" )
.def("HelloMPI", &adios::ADIOSPy::HelloMPI )
.def("DefineVariableDouble", &adios::ADIOSPy::DefineVariablePy<double>, py::return_value_policy::reference_internal )
.def("DefineVariableFloat", &adios::ADIOSPy::DefineVariablePy<float>, py::return_value_policy::reference_internal )
// py::arg("localDimensionsPy") = py::list(),
// py::arg("globalDimensionsPy") = py::list(), py::arg("globalOffsetsPy") = py::list() )
.def("DeclareMethod", &adios::ADIOSPy::DeclareMethodPy, py::return_value_policy::reference_internal )
.def("Open", &adios::ADIOSPy::OpenPy )
;
return m.ptr();
}