Add hessian and hessian energy functions to bindings

Author: rubenwiersma

src/curved_hessian_energy.cpp

@@ -0,0 +1,45 @@
+#include "default_types.h"
+#include <igl/curved_hessian_energy.h>
+#include <nanobind/nanobind.h>
+#include <nanobind/ndarray.h>
+#include <nanobind/eigen/dense.h>
+#include <nanobind/eigen/sparse.h>
+#include <nanobind/stl/tuple.h>
+
+namespace nb = nanobind;
+using namespace nb::literals;
+
+namespace pyigl
+{
+  auto curved_hessian_energy(
+    const nb::DRef<const Eigen::MatrixXN> &V,
+    const nb::DRef<const Eigen::MatrixXI> &F)
+  {
+    Eigen::SparseMatrixN QH;
+    igl::curved_hessian_energy(V,F,QH);
+    return QH;
+  }
+
+}
+
+// Bind the wrapper to the Python module
+void bind_curved_hessian_energy(nb::module_ &m)
+{
+  m.def(
+    "curved_hessian_energy",
+    &pyigl::curved_hessian_energy, 
+    "V"_a, 
+    "F"_a,
+R"(Constructs the matrix for the curved Hessian energy for a given
+mesh (V,F).
+
+  @tparam DerivedV  derived type of eigen matrix for V (e.g. derived from
+    MatrixXd)
+  @tparam DerivedF  derived type of eigen matrix for F (e.g. derived from
+    MatrixXi)
+  @tparam Scalar  scalar type for eigen sparse matrix (e.g. double)
+  @param[in] V  #V by dim list of mesh vertex positions
+  @param[in] F  #F by simplex_size list of mesh elements (triangles or tetrahedra)
+  @param[out] QH  #V by #V curved Hessian energy matrix, each row i corresponding to V(i,:))");
+
+}

src/hessian.cpp

@@ -0,0 +1,45 @@
+#include "default_types.h"
+#include <igl/hessian.h>
+#include <nanobind/nanobind.h>
+#include <nanobind/ndarray.h>
+#include <nanobind/eigen/dense.h>
+#include <nanobind/eigen/sparse.h>
+#include <nanobind/stl/tuple.h>
+
+namespace nb = nanobind;
+using namespace nb::literals;
+
+namespace pyigl
+{
+  auto hessian(
+    const nb::DRef<const Eigen::MatrixXN> &V,
+    const nb::DRef<const Eigen::MatrixXI> &F)
+  {
+    Eigen::SparseMatrixN H;
+    igl::hessian(V,F,H);
+    return H;
+  }
+
+}
+
+// Bind the wrapper to the Python module
+void bind_hessian(nb::module_ &m)
+{
+  m.def(
+    "hessian",
+    &pyigl::hessian, 
+    "V"_a, 
+    "F"_a,
+R"(Constructs the Hessian matrix for a given
+mesh (V,F).
+
+  @tparam DerivedV  derived type of eigen matrix for V (e.g. derived from
+    MatrixXd)
+  @tparam DerivedF  derived type of eigen matrix for F (e.g. derived from
+    MatrixXi)
+  @tparam Scalar  scalar type for eigen sparse matrix (e.g. double)
+  @param[in] V  #V by dim list of mesh vertex positions
+  @param[in] F  #F by simplex_size list of mesh elements (triangles or tetrahedra)
+  @param[out] H  #V by #V Hessian matrix, each row i corresponding to V(i,:))");
+
+}

src/hessian_energy.cpp

@@ -0,0 +1,45 @@
+#include "default_types.h"
+#include <igl/hessian_energy.h>
+#include <nanobind/nanobind.h>
+#include <nanobind/ndarray.h>
+#include <nanobind/eigen/dense.h>
+#include <nanobind/eigen/sparse.h>
+#include <nanobind/stl/tuple.h>
+
+namespace nb = nanobind;
+using namespace nb::literals;
+
+namespace pyigl
+{
+  auto hessian_energy(
+    const nb::DRef<const Eigen::MatrixXN> &V,
+    const nb::DRef<const Eigen::MatrixXI> &F)
+  {
+    Eigen::SparseMatrixN QH;
+    igl::hessian_energy(V,F,QH);
+    return QH;
+  }
+
+}
+
+// Bind the wrapper to the Python module
+void bind_hessian_energy(nb::module_ &m)
+{
+  m.def(
+    "hessian_energy",
+    &pyigl::hessian_energy, 
+    "V"_a, 
+    "F"_a,
+R"(Constructs the matrix for the Hessian energy for a given
+mesh (V,F).
+
+  @tparam DerivedV  derived type of eigen matrix for V (e.g. derived from
+    MatrixXd)
+  @tparam DerivedF  derived type of eigen matrix for F (e.g. derived from
+    MatrixXi)
+  @tparam Scalar  scalar type for eigen sparse matrix (e.g. double)
+  @param[in] V  #V by dim list of mesh vertex positions
+  @param[in] F  #F by simplex_size list of mesh elements (triangles or tetrahedra)
+  @param[out] QH  #V by #V Hessian energy matrix, each row i corresponding to V(i,:))");
+
+}

tests/test_all.py

@@ -109,6 +109,9 @@ def test_operators():
     K = igl.cotmatrix_entries(l=l)
     B1,B2,B3 = igl.local_basis(V,F)
     G = igl.grad(V,F)
+    H = igl.hessian(V, F)
+    QH = igl.hessian_energy(V, F)
+    cQH = igl.curved_hessian_energy(V, F)
 
     M = igl.massmatrix(V,F)
     M = igl.massmatrix(V,F,type=igl.MASSMATRIX_TYPE_BARYCENTRIC)