OilfieldToolsNet/opm-common
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use pybind11 as binding framework

Browse Source 
This commit changes the api for the Schedule class, the various time related
methods now return datetime.datetime instances instead of datetime.data.
This commit is contained in:
Joakim Hove
2018-01-30 07:46:03 +01:00
parent 1566cd4e37
commit 67b30fe715
20 changed files with 268 additions and 346 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
sunbeam/deck_keyword.cpp
View File

@@ -14,6 +14,8 @@ namespace {
/* DeckKeyword */
const DeckRecord& (DeckKeyword::*getRecord)(size_t index) const = &DeckKeyword::getRecord;
py::list item_to_pylist( const DeckItem& item )
{
switch (item.getType())
@@ -39,12 +41,14 @@ struct DeckRecordIterator
this->record = record;
this->it = this->record->begin();
}
const DeckRecord* record;
DeckRecord::const_iterator it;
py::list next() {
if (it == record->end()) {
PyErr_SetString(PyExc_StopIteration, "At end.");
py::throw_error_already_set();
throw py::error_already_set();
}
return item_to_pylist(*(it++));
}
@@ -52,17 +56,18 @@ struct DeckRecordIterator
}
void sunbeam::export_DeckKeyword() {
py::class_< DeckKeyword >( "DeckKeyword", py::no_init )
.def( "__repr__", &DeckKeyword::name, copy() )
void sunbeam::export_DeckKeyword(py::module& module) {
py::class_< DeckKeyword >( module, "DeckKeyword")
.def( "__repr__", &DeckKeyword::name )
.def( "__str__", &str<DeckKeyword> )
.def( "__iter__", py::range< ref >( &DeckKeyword::begin, &DeckKeyword::end ))
.def( "__getitem__", getRecord, ref() )
.def("__iter__", [] (const DeckKeyword &keyword) { return py::make_iterator(keyword.begin(), keyword.end()); }, py::keep_alive<0,1>())
.def( "__getitem__", getRecord, ref_internal)
.def( "__len__", &DeckKeyword::size )
.add_property("name", py::make_function( &DeckKeyword::name, copy() ))
.def_property_readonly("name", &DeckKeyword::name )
;
py::class_< DeckRecord >( "DeckRecord", py::no_init )
py::class_< DeckRecord >( module, "DeckRecord")
.def( "__repr__", &str<DeckRecord> )
.def( "__iter__", +[](const DeckRecord& record){
return DeckRecordIterator(&record);
@@ -76,8 +81,10 @@ void sunbeam::export_DeckKeyword() {
.def( "__len__", &DeckRecord::size )
;
py::class_< DeckRecordIterator >( "DeckRecordIterator", py::no_init )
py::class_< DeckRecordIterator >( module, "DeckRecordIterator")
.def( "__next__", &DeckRecordIterator::next )
.def( "next", &DeckRecordIterator::next )
;
}