OilfieldToolsNet/opm-common
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
07a8c5b0158bbdbfcda90679f89459c53dcc066f
opm-common/python/cxx/eclipse_io.cpp

61 lines
1.9 KiB
C++
Raw Normal View History

python binding for EclFile
2019-10-26 17:33:43 +02:00
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include <pybind11/numpy.h>
#include <src/opm/io/eclipse/EclFile.cpp>
#include <opm/io/eclipse/EclIOdata.hpp>
Rewrite EclFile wrapping - Include the eclipse_io.cpp file in existing extension library - do not create a new library. - Remove Python class EclFile - just expose the C++ through Pybind1, include some monkey-patching of the class to ensure proper Python strings for CHAR keywords. - Remove string based keyword lookup. - Use standard Python methods __len__, __getitem__ and __contains__; get() is an equivalent alternative to __getitem__(). - Add property arrays in addition to getListOfArrays()
2019-10-30 22:38:23 +01:00
#include "export.hpp"
#include "converters.hpp"
python binding for EclFile
2019-10-26 17:33:43 +02:00
namespace py = pybind11;
Rewrite EclFile wrapping - Include the eclipse_io.cpp file in existing extension library - do not create a new library. - Remove Python class EclFile - just expose the C++ through Pybind1, include some monkey-patching of the class to ensure proper Python strings for CHAR keywords. - Remove string based keyword lookup. - Use standard Python methods __len__, __getitem__ and __contains__; get() is an equivalent alternative to __getitem__(). - Add property arrays in addition to getListOfArrays()
2019-10-30 22:38:23 +01:00
namespace {
python binding for EclFile
2019-10-26 17:33:43 +02:00
Rewrite EclFile wrapping - Include the eclipse_io.cpp file in existing extension library - do not create a new library. - Remove Python class EclFile - just expose the C++ through Pybind1, include some monkey-patching of the class to ensure proper Python strings for CHAR keywords. - Remove string based keyword lookup. - Use standard Python methods __len__, __getitem__ and __contains__; get() is an equivalent alternative to __getitem__(). - Add property arrays in addition to getListOfArrays()
2019-10-30 22:38:23 +01:00
py::array get_vector(Opm::EclIO::EclFile * file_ptr, std::size_t array_index) {
auto array_type = std::get<1>(file_ptr->getList()[array_index]);
if (array_type == Opm::EclIO::INTE)
return convert::numpy_array( file_ptr->get<int>(array_index) );
python binding for EclFile
2019-10-26 17:33:43 +02:00
Rewrite EclFile wrapping - Include the eclipse_io.cpp file in existing extension library - do not create a new library. - Remove Python class EclFile - just expose the C++ through Pybind1, include some monkey-patching of the class to ensure proper Python strings for CHAR keywords. - Remove string based keyword lookup. - Use standard Python methods __len__, __getitem__ and __contains__; get() is an equivalent alternative to __getitem__(). - Add property arrays in addition to getListOfArrays()
2019-10-30 22:38:23 +01:00
if (array_type == Opm::EclIO::REAL)
return convert::numpy_array( file_ptr->get<float>(array_index) );
python binding for EclFile
2019-10-26 17:33:43 +02:00
Rewrite EclFile wrapping - Include the eclipse_io.cpp file in existing extension library - do not create a new library. - Remove Python class EclFile - just expose the C++ through Pybind1, include some monkey-patching of the class to ensure proper Python strings for CHAR keywords. - Remove string based keyword lookup. - Use standard Python methods __len__, __getitem__ and __contains__; get() is an equivalent alternative to __getitem__(). - Add property arrays in addition to getListOfArrays()
2019-10-30 22:38:23 +01:00
if (array_type == Opm::EclIO::DOUB)
return convert::numpy_array( file_ptr->get<double>(array_index) );
python binding for EclFile
2019-10-26 17:33:43 +02:00
Rewrite EclFile wrapping - Include the eclipse_io.cpp file in existing extension library - do not create a new library. - Remove Python class EclFile - just expose the C++ through Pybind1, include some monkey-patching of the class to ensure proper Python strings for CHAR keywords. - Remove string based keyword lookup. - Use standard Python methods __len__, __getitem__ and __contains__; get() is an equivalent alternative to __getitem__(). - Add property arrays in addition to getListOfArrays()
2019-10-30 22:38:23 +01:00
if (array_type == Opm::EclIO::LOGI)
return convert::numpy_array( file_ptr->get<bool>(array_index) );
if (array_type == Opm::EclIO::CHAR)
return convert::numpy_string_array( file_ptr->get<std::string>(array_index) );
throw std::logic_error("Data type not supported");
}
}
void python::common::export_IO(py::module& m) {
python binding for EclFile
2019-10-26 17:33:43 +02:00
py::enum_<Opm::EclIO::eclArrType>(m, "eclArrType", py::arithmetic())
.value("INTE", Opm::EclIO::INTE)
.value("REAL", Opm::EclIO::REAL)
.value("DOUB", Opm::EclIO::DOUB)
.value("CHAR", Opm::EclIO::CHAR)
.value("LOGI", Opm::EclIO::LOGI)
.value("MESS", Opm::EclIO::MESS)
.export_values();
Rewrite EclFile wrapping - Include the eclipse_io.cpp file in existing extension library - do not create a new library. - Remove Python class EclFile - just expose the C++ through Pybind1, include some monkey-patching of the class to ensure proper Python strings for CHAR keywords. - Remove string based keyword lookup. - Use standard Python methods __len__, __getitem__ and __contains__; get() is an equivalent alternative to __getitem__(). - Add property arrays in addition to getListOfArrays()
2019-10-30 22:38:23 +01:00
py::class_<Opm::EclIO::EclFile>(m, "EclFile")
.def(py::init<const std::string &, bool>(), py::arg("filename"), py::arg("preload") = false)
.def_property_readonly("arrays", &Opm::EclIO::EclFile::getList)
.def("getListOfArrays", &Opm::EclIO::EclFile::getList)
.def("__contains__", &Opm::EclIO::EclFile::hasKey)
.def("__len__", &Opm::EclIO::EclFile::size)
.def("__getitem__", &get_vector)
.def("get", &get_vector);
python binding for EclFile
2019-10-26 17:33:43 +02:00
}
Reference in New Issue Copy Permalink