Source code for geomfum.wrap.pp3d
"""potpourri3d wrapper.
https://github.com/nmwsharp/potpourri3d
by Nicholas Sharp.
"""
import geomstats.backend as gs
import potpourri3d as pp3d
from geomfum.metric.mesh import FinitePointSetMetric, _SingleDispatchMixins
[docs]
class Pp3dHeatDistanceMetric(_SingleDispatchMixins, FinitePointSetMetric):
"""Heat distance metric between vertices of a mesh.
Parameters
----------
shape : Shape
Shape.
References
----------
.. [CWW2017] Crane, K., Weischedel, C., Wardetzky, M., 2017.
The heat method for distance computation. Commun. ACM 60, 90–99.
https://doi.org/10.1145/3131280
"""
def __init__(self, shape):
super().__init__(shape)
self.solver = pp3d.MeshHeatMethodDistanceSolver(
gs.to_numpy(shape.vertices), gs.to_numpy(shape.faces)
)
[docs]
def dist_matrix(self):
"""Distance between mesh vertices.
Returns
-------
dist_matrix : array-like, shape=[n_vertices, n_vertices]
Distance matrix.
Notes
-----
slow
"""
dist_mat = []
for i in range(self._shape.n_vertices):
dist_mat.append(gs.asarray(self.solver.compute_distance(i)))
return gs.stack(dist_mat, axis=0)
def _dist_from_source_single(self, source_point):
"""Distance between mesh vertices.
Parameters
----------
source_point : array-like, shape=()
Index of source point.
Returns
-------
dist : array-like, shape=[n_vertices]
Distance.
target_point : array-like, shape=[n_vertices,]
Target index.
"""
dist = self.solver.compute_distance(source_point.item())
target_point = gs.arange(self._shape.n_vertices)
return gs.asarray(dist), target_point
def _dist_single(self, point_a, point_b):
"""Distance between mesh vertices.
Parameters
----------
point_a : array-like, shape=()
Index of source point.
point_b : array-like, shape=()
Index of target point.
Returns
-------
dist : numeric
Distance.
"""
dist = self.solver.compute_distance(point_a)[point_b]
return gs.asarray(dist)
