From 21ba19bf10d3a8e6bc955bcc4038a40da85ce112 Mon Sep 17 00:00:00 2001 From: Christopher Lackner Date: Thu, 24 Jan 2019 16:13:22 +0100 Subject: [PATCH] export __getitem__ and __len__ for Vec --- libsrc/meshing/python_mesh.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/libsrc/meshing/python_mesh.cpp b/libsrc/meshing/python_mesh.cpp index 436835ef..6d7760a4 100644 --- a/libsrc/meshing/python_mesh.cpp +++ b/libsrc/meshing/python_mesh.cpp @@ -113,6 +113,8 @@ DLL_HEADER void ExportNetgenMeshing(py::module &m) .def(-py::self) .def(double()*py::self) .def("Norm", &Vec<2>::Length) + .def("__getitem__", [](Vec<2>& vec, int index) { return vec[index]; }) + .def("__len__", [](Vec<2>& /*unused*/) { return 2; }) ; py::class_> (m, "Vec3d") @@ -123,6 +125,8 @@ DLL_HEADER void ExportNetgenMeshing(py::module &m) .def(-py::self) .def(double()*py::self) .def("Norm", &Vec<3>::Length) + .def("__getitem__", [](Vec<3>& vec, int index) { return vec[index]; }) + .def("__len__", [](Vec<3>& /*unused*/) { return 3; }) ; m.def ("Vec", FunctionPointer