python binding for EclFile

2019-10-26 17:33:43 +02:00
parent 0b8a42fd1f
commit 7cacc734e8
11 changed files with 41158 additions and 1 deletions
--- a/python/cxx/eclipse_io.cpp
+++ b/python/cxx/eclipse_io.cpp
@@ -0,0 +1,73 @@
+#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 <src/opm/io/eclipse/EclUtil.cpp>
+
+#include <src/opm/common/OpmLog/OpmLog.cpp>
+#include <src/opm/common/OpmLog/LogBackend.cpp>
+#include <src/opm/common/OpmLog/LogUtil.cpp>
+#include <src/opm/common/OpmLog/StreamLog.cpp>
+#include <src/opm/common/OpmLog/Logger.cpp>      
+
+
+namespace py = pybind11;
+
+class EclFileTmp : public Opm::EclIO::EclFile {
+    
+public:
+    EclFileTmp(const std::string& filename): Opm::EclIO::EclFile(filename) {};
+
+    py::array_t<float> getFloatNumpy(int arrIndex) {
+        std::vector<float> tmp=get<float>(arrIndex);
+        return py::array(py::dtype("f"), {tmp.size()}, {}, &tmp[0]);
+    };
+
+    py::array_t<double> getDoubleNumpy(int arrIndex) {
+        std::vector<double> tmp=get<double>(arrIndex);
+        return py::array(py::dtype("d"), {tmp.size()}, {}, &tmp[0]);
+    };
+
+    py::array_t<int> getIntegerNumpy(int arrIndex) {
+        std::vector<int> tmp=get<int>(arrIndex);
+        return py::array(py::dtype("i"), {tmp.size()}, {}, &tmp[0]); 
+    };
+};
+
+
+PYBIND11_MODULE(libopmioecl_python, 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_<std::vector<std::tuple<std::string,Opm::EclIO::eclArrType,int>>>(m, "EclEntry")
+        .def(py::init<>())
+        .def("__len__", [](const std::vector<std::tuple<std::string,Opm::EclIO::eclArrType,int>> &v) { return v.size(); })
+        .def("__getitem__", [](std::vector<std::tuple<std::string,Opm::EclIO::eclArrType,int>> &v, int item) { return v[item];})
+        .def("__iter__", [](std::vector<std::tuple<std::string,Opm::EclIO::eclArrType,int>> &v) {
+            return py::make_iterator(v.begin(), v.end());
+    }, py::keep_alive<0, 1>());
+
+        
+    py::class_<EclFileTmp>(m, "EclFileBind")
+        .def(py::init<const std::string &>())
+        .def("getList", &EclFileTmp::getList)   
+        .def("hasKey", &EclFileTmp::hasKey)   
+
+        .def("loadAllData", (void (EclFileTmp::*)(void)) &EclFileTmp::loadData)   
+        .def("loadDataByIndex", (void (EclFileTmp::*)(int)) &EclFileTmp::loadData)   
+
+        .def("getRealFromIndexNumpy", &EclFileTmp::getFloatNumpy)   
+        .def("getDoubFromIndexNumpy", &EclFileTmp::getDoubleNumpy)   
+        .def("getInteFromIndexNumpy", &EclFileTmp::getIntegerNumpy)   
+        .def("getLogiFromIndex", (const std::vector<bool>& (EclFileTmp::*)(int)) &EclFileTmp::get<bool>)   
+        .def("getCharFromIndex", (const std::vector<std::string>& (EclFileTmp::*)(int)) &EclFileTmp::get<std::string>);   
+}