NGSolve/netgen
1
0
Fork 0
mirror of https://github.com/NGSolve/netgen.git synced 2026-05-31 01:06:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

array iterator to python

Browse Source
This commit is contained in:
Christopher Lackner
2016-11-15 18:12:51 +01:00
parent 8e97d91912
commit 0a9abc1acc
2 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libsrc/meshing/python_mesh.cpp
View File

@@ -20,6 +20,7 @@ namespace netgen
template <typename T, int BASE = 0, typename TIND = int>
void ExportArray (py::module &m)
{
using TA = Array<T,BASE,TIND>;
string name = string("Array_") + typeid(T).name();
py::class_<Array<T,BASE,TIND>>(m, name.c_str())
.def ("__len__", [] ( Array<T,BASE,TIND> &self ) { return self.Size(); } )
@@ -31,6 +32,9 @@ void ExportArray (py::module &m)
return self[i];
}),
py::return_value_policy::reference)
.def("__iter__", [] ( TA & self) {
return py::make_iterator (self.begin(),self.end());
}, py::keep_alive<0,1>()) // keep array alive while iterator is used
;
}