79 lines
2.3 KiB
C++
79 lines
2.3 KiB
C++
#include <opm/input/eclipse/Deck/Deck.hpp>
|
|
#include <opm/input/eclipse/Units/UnitSystem.hpp>
|
|
|
|
#include <pybind11/pybind11.h>
|
|
#include "converters.hpp"
|
|
#include "export.hpp"
|
|
|
|
|
|
namespace {
|
|
|
|
size_t size( const Deck& deck ) {
|
|
return deck.size();
|
|
}
|
|
|
|
|
|
size_t count( const Deck& deck, const std::string& kw ) {
|
|
return deck.count(kw);
|
|
}
|
|
|
|
bool hasKeyword( const Deck& deck, const std::string& kw ) {
|
|
return deck.hasKeyword(kw);
|
|
}
|
|
|
|
const DeckKeyword& getKeyword_tuple( const Deck& deck, py::tuple kw_index ) {
|
|
const std::string kw = py::cast<const std::string>(kw_index[0]);
|
|
const size_t index = py::cast<size_t>(kw_index[1]);
|
|
return deck[kw][index];
|
|
}
|
|
|
|
const DeckKeyword& getKeyword_string( const Deck& deck, const std::string& kw ) {
|
|
return deck[kw].back();
|
|
}
|
|
|
|
const DeckKeyword& getKeyword_int( const Deck& deck, size_t index ) {
|
|
return deck[index];
|
|
}
|
|
|
|
//This adds a keyword by copy
|
|
void addKeyword(Deck& deck, const DeckKeyword kw) {
|
|
deck.addKeyword(kw);
|
|
}
|
|
|
|
|
|
}
|
|
|
|
void python::common::export_Deck(py::module &module) {
|
|
|
|
// Note: In the below class we use std::shared_ptr as the holder type, see:
|
|
//
|
|
// https://pybind11.readthedocs.io/en/stable/advanced/smart_ptrs.html
|
|
//
|
|
// this makes it possible to share the returned object with e.g. and
|
|
// opm.simulators.BlackOilSimulator Python object
|
|
//
|
|
py::class_< Deck, std::shared_ptr<Deck> >(module, "Deck")
|
|
.def( "__len__", &size )
|
|
.def( "__contains__", &hasKeyword )
|
|
.def("__iter__",
|
|
[] (const Deck &deck) { return py::make_iterator(deck.begin(), deck.end()); }, py::keep_alive<0, 1>())
|
|
.def( "__getitem__", &getKeyword_int, ref_internal)
|
|
.def( "__getitem__", &getKeyword_string, ref_internal)
|
|
.def( "__getitem__", &getKeyword_tuple, ref_internal)
|
|
.def( "__str__", &str<Deck>)
|
|
|
|
.def("active_unit_system", [](const Deck& deck) -> const UnitSystem& {
|
|
return deck.getActiveUnitSystem();
|
|
} )
|
|
|
|
.def("default_unit_system", [](const Deck& deck) -> const UnitSystem& {
|
|
return deck.getDefaultUnitSystem();
|
|
} )
|
|
|
|
.def( "count", &count )
|
|
.def( "add", &addKeyword)
|
|
;
|
|
}
|
|
|
|
|