d66852d8de
- 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()
61 lines
1.9 KiB
C++
61 lines
1.9 KiB
C++
#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>
|
|
|
|
#include "export.hpp"
|
|
#include "converters.hpp"
|
|
|
|
namespace py = pybind11;
|
|
|
|
|
|
namespace {
|
|
|
|
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) );
|
|
|
|
if (array_type == Opm::EclIO::REAL)
|
|
return convert::numpy_array( file_ptr->get<float>(array_index) );
|
|
|
|
if (array_type == Opm::EclIO::DOUB)
|
|
return convert::numpy_array( file_ptr->get<double>(array_index) );
|
|
|
|
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) {
|
|
|
|
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();
|
|
|
|
|
|
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);
|
|
}
|