Make input global indices a numpy array (#3069)

FEniCS · Feb 26, 2024 · 2838098 · 2838098
1 parent 476c61c
commit 2838098
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/python/dolfinx/wrappers/mesh.cpp b/python/dolfinx/wrappers/mesh.cpp
@@ -178,8 +178,16 @@ void declare_mesh(nb::module_& m, std::string type)
       .def_prop_ro(
           "cmap", [](dolfinx::mesh::Geometry<T>& self) { return self.cmap(); },
           "The coordinate map")
-      .def_prop_ro("input_global_indices",
-                   &dolfinx::mesh::Geometry<T>::input_global_indices);
+      .def_prop_ro(
+          "input_global_indices",
+          [](const dolfinx::mesh::Geometry<T>& self)
+          {
+            const std::vector<std::int64_t>& id_to_global
+                = self.input_global_indices();
+            return nb::ndarray<const std::int64_t, nb::numpy>(
+                id_to_global.data(), {id_to_global.size()});
+          },
+          nb::rv_policy::reference_internal);
 
   std::string pyclass_mesh_name = std::string("Mesh_") + type;
   nb::class_<dolfinx::mesh::Mesh<T>>(m, pyclass_mesh_name.c_str(),