ADD project points and project mesh.

Author: petrasvestartas

docs/api/compas_cgal.meshing.rst

@@ -10,4 +10,5 @@ compas_cgal.meshing
 
     mesh_remesh
     remesh_dual
-    mesh_project
+    project_mesh_on_mesh
+    project_points_on_mesh

docs/examples/example_projection.py

@@ -1,9 +1,10 @@
 from pathlib import Path
 
+from compas.geometry import Point
 from compas.datastructures import Mesh
 from compas_viewer import Viewer
 
-from compas_cgal.meshing import mesh_project
+from compas_cgal.meshing import project_mesh_on_mesh, project_points_on_mesh
 
 
 # Mesh to project
@@ -14,7 +15,14 @@
 input_file_1 = Path(__file__).parent.parent.parent / "data" / "rhinovault_mesh_1.ply"
 mesh_1 = Mesh.from_ply(input_file_1)
 
-mesh_result = mesh_project(mesh_0, mesh_1, use_normals=False)
+mesh_result = project_mesh_on_mesh(mesh_0, mesh_1, use_normals=False)
+
+# Points with normals to project
+v, f = mesh_0.to_vertices_and_faces()
+normals = []
+for i in range(len(v)):
+    normals.append(mesh_0.vertex_normal(i))
+points_result = project_points_on_mesh(v, mesh_1, normals)
 
 # ==============================================================================
 # Visualize
@@ -27,5 +35,7 @@
 viewer.scene.add(mesh_0, name="source mesh_0", show_faces=False)
 viewer.scene.add(mesh_1, name="target mesh_1", show_faces=False)
 viewer.scene.add(mesh_result, name="projected mesh_result", color=[255, 0, 0])
+for i, point in enumerate(points_result):
+    viewer.scene.add(Point(*point), name=f"point_{i}", color=[0, 255, 0])
 
 viewer.show()

src/compas_cgal/meshing.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 import numpy as np
 from compas.datastructures import Mesh
 from compas.plugins import plugin
@@ -56,7 +58,7 @@ def mesh_remesh(
     return _meshing.remesh(V, F, target_edge_length, number_of_iterations, do_project)
 
 
-def mesh_project(
+def project_mesh_on_mesh(
     mesh_source: VerticesFaces,
     mesh_target: VerticesFaces,
     use_normals: bool = True,
@@ -108,6 +110,56 @@ def mesh_project(
     return Mesh.from_vertices_and_faces(numpy_V_source, F_source)
 
 
+def project_points_on_mesh(
+    points: list[list[float]],
+    mesh_target: VerticesFaces,
+    normals: Optional[list[list[float]]] = None,
+) -> list[list[float]]:
+    """Project points onto mesh_target surface.
+
+    Parameters
+    ----------
+    points : list[list[float]]
+        The points to project.
+    mesh_target : :attr:`compas_cgal.types.VerticesFaces`
+        The mesh whose vertices will be projected (source points).
+    normals : list[list[float]], optional
+        The normals of the points to project.
+
+    Returns
+    -------
+    list[list[float]]
+        The projected points (vertices on mesh_target surface).
+
+    """
+    V_target, F_target = mesh_target.to_vertices_and_faces(triangulated=True)
+
+    # Convert inputs to numpy arrays
+    numpy_V_source = np.asarray(points, dtype=np.float64, order="C")
+
+    numpy_V_target = np.asarray(V_target, dtype=np.float64, order="C")
+    numpy_F_target = np.asarray(F_target, dtype=np.int32, order="C")
+
+    # Handle normals calculation
+    if normals:
+        # Pre-allocate numpy array for normals with correct size
+        numpy_N_source = np.zeros((len(points), 3), dtype=np.float64)
+
+        # Fill the array with vertex normals
+        for i, normal in enumerate(normals):
+            numpy_N_source[i, 0] = normal[0]
+            numpy_N_source[i, 1] = normal[1]
+            numpy_N_source[i, 2] = normal[2]
+    else:
+        # Don't use normals - create empty array
+        numpy_N_source = np.zeros((0, 0), dtype=np.float64)
+
+    # Call the C++ function
+    _meshing.project(numpy_V_target, numpy_F_target, numpy_V_source, numpy_N_source)
+
+    return numpy_V_source
+
+
 def remesh_dual(
     mesh: VerticesFaces,
     length_factor: float = 1.0,
@@ -137,7 +189,7 @@ def remesh_dual(
     -------
     tuple
         A tuple containing:
-        
+
         - Remeshed mesh vertices as an Nx3 numpy array.
         - Remeshed mesh faces as an Mx3 numpy array.
         - Dual mesh vertices as an Nx3 numpy array.

tests/test_projection.py

@@ -1,6 +1,6 @@
 from pathlib import Path
 from compas.datastructures import Mesh
-from compas_cgal.meshing import mesh_project
+from compas_cgal.meshing import project_mesh_on_mesh
 
 
 def mesh_projection():
@@ -12,6 +12,6 @@ def mesh_projection():
     input_file_1 = Path(__file__).parent.parent.parent / "data" / "rhinovault_mesh_1.ply"
     mesh_1 = Mesh.from_ply(input_file_1)
 
-    mesh_result = mesh_project(mesh_0, mesh_1, use_normals=False)
+    mesh_result = project_mesh_on_mesh(mesh_0, mesh_1, use_normals=False)
 
     assert mesh_result.number_of_vertices() == mesh_0.number_of_vertices()