47 lines
1.4 KiB
C++
47 lines
1.4 KiB
C++
#include <opm/parser/eclipse/EclipseState/Eclipse3DProperties.hpp>
|
|
#include <pybind11/stl.h>
|
|
|
|
#include "sunbeam.hpp"
|
|
#include "converters.hpp"
|
|
|
|
namespace {
|
|
|
|
py::list getitem( const Eclipse3DProperties& p, const std::string& kw) {
|
|
const auto& ip = p.getIntProperties();
|
|
if (ip.supportsKeyword(kw) && ip.hasKeyword(kw))
|
|
return iterable_to_pylist(p.getIntGridProperty(kw).getData());
|
|
|
|
|
|
const auto& dp = p.getDoubleProperties();
|
|
if (dp.supportsKeyword(kw) && dp.hasKeyword(kw))
|
|
return iterable_to_pylist(p.getDoubleGridProperty(kw).getData());
|
|
|
|
throw py::key_error( "no such grid property " + kw );
|
|
}
|
|
|
|
bool contains( const Eclipse3DProperties& p, const std::string& kw) {
|
|
return
|
|
(p.getIntProperties().supportsKeyword(kw) &&
|
|
p.getIntProperties().hasKeyword(kw))
|
|
||
|
|
(p.getDoubleProperties().supportsKeyword(kw) &&
|
|
p.getDoubleProperties().hasKeyword(kw))
|
|
;
|
|
}
|
|
|
|
std::vector<int> regions( const Eclipse3DProperties& p, const std::string& kw) {
|
|
return p.getRegions(kw);
|
|
}
|
|
|
|
}
|
|
|
|
void sunbeam::export_Eclipse3DProperties(py::module& module) {
|
|
|
|
py::class_< Eclipse3DProperties >( module, "Eclipse3DProperties")
|
|
.def( "getRegions", ®ions )
|
|
.def( "__contains__", &contains )
|
|
.def( "__getitem__", &getitem )
|
|
;
|
|
|
|
}
|