test using map

Author: alecjacobson

src/sparse_noop.cpp

@@ -0,0 +1,27 @@
+#include "default_types.h"
+#include <nanobind/nanobind.h>
+#include <nanobind/eigen/sparse.h>
+#include <nanobind/ndarray.h>
+#include <nanobind/stl/tuple.h>
+
+namespace nb = nanobind;
+using namespace nb::literals;
+
+namespace pyigl
+{
+  auto sparse_noop( const Eigen::Map<const Eigen::SparseMatrix<Integer>> &A)
+  {
+    return A;
+  }
+}
+
+// Bind the wrapper to the Python module
+void bind_sparse_noop(nb::module_ &m)
+{
+  m.def(
+    "sparse_noop",
+    &pyigl::sparse_noop,
+    "A"_a,
+R"("Returns input A)");
+}
+

src/sparse_shape.cpp

@@ -0,0 +1,29 @@
+#include "default_types.h"
+#include <nanobind/nanobind.h>
+#include <nanobind/eigen/sparse.h>
+#include <nanobind/ndarray.h>
+#include <nanobind/stl/tuple.h>
+
+namespace nb = nanobind;
+using namespace nb::literals;
+
+namespace pyigl
+{
+  auto sparse_shape( const Eigen::Map<const Eigen::SparseMatrix<Integer>> &A)
+  {
+    Eigen::Matrix<Integer,2,1> dims(A.rows(), A.cols());
+    return dims;
+  }
+}
+
+// Bind the wrapper to the Python module
+void bind_sparse_shape(nb::module_ &m)
+{
+  m.def(
+    "sparse_shape",
+    &pyigl::sparse_shape,
+    "A"_a,
+R"("Returns shape of A as 2-vector)");
+}
+
+

test_sparse.py

@@ -0,0 +1,22 @@
+import igl
+import time
+
+V, F = igl.icosahedron()
+
+max_iters = 10
+for i in range(max_iters):
+    A = igl.adjacency_matrix(F)
+    n = A.shape[0]
+
+    start = time.time()
+    dims = igl.sparse_shape(A)
+    t_shape = time.time() - start
+
+    start = time.time()
+    A2 = igl.sparse_noop(A)
+    t_noop = time.time() - start
+
+    print(f"{n} {t_shape:.6g} {t_noop:.6g}")
+
+    if i != max_iters-1:
+        V, F = igl.upsample(V, F)