wip: adding tesellation endpoints

Author: RobPasMue

src/ansys/geometry/core/_grpc/_services/base/bodies.py

@@ -194,6 +194,11 @@ def get_tesellation(self, **kwargs) -> dict:
         """Get the tessellation of a body."""
         pass
 
+    @abstractmethod
+    def get_tesellation_with_options(self, **kwargs) -> dict:
+        """Get the tessellation of a body with options."""
+        pass
+
     @abstractmethod
     def boolean(self, **kwargs) -> dict:
         """Boolean operation."""

src/ansys/geometry/core/_grpc/_services/v0/bodies.py

@@ -34,8 +34,10 @@
     from_frame_to_grpc_frame,
     from_grpc_material_to_material,
     from_grpc_point_to_point3d,
+    from_grpc_tess_to_pd,
     from_plane_to_grpc_plane,
     from_point3d_to_grpc_point,
+    from_tess_options_to_grpc_tess_options,
     from_unit_vector_to_grpc_direction,
 )
 
@@ -390,7 +392,48 @@ def copy(self, **kwargs) -> dict:  # noqa: D102
 
     @protect_grpc
     def get_tesellation(self, **kwargs) -> dict:  # noqa: D102
-        raise NotImplementedError
+        from ansys.api.geometry.v0.bodies_pb2 import GetTessellationRequest
+
+        tess_map = {}
+        resp = []  # For compatibility with stream response
+        try:
+            resp_single = self.stub.GetTessellation(request=build_grpc_id(kwargs["id"]))
+            resp.append(resp_single)
+        except grpc.RpcError as err:
+            if kwargs["backend_version"] < (25, 2, 0):
+                raise err
+            request = GetTessellationRequest(id=build_grpc_id(kwargs["id"]))
+            resp = self.stub.GetTessellationStream(request=request)
+
+        for elem in resp:
+            for face_id, face_tess in elem.face_tessellation.items():
+                tess_map[face_id] = from_grpc_tess_to_pd(face_tess)
+
+        return {"tessellation": tess_map}
+
+    @protect_grpc
+    def get_tesellation_with_options(self, **kwargs) -> dict:  # noqa: D102
+        from ansys.api.geometry.v0.bodies_pb2 import GetTessellationRequest
+
+        # Create the request - assumes all inputs are valid and of the proper type
+        request = GetTessellationRequest(
+            id=build_grpc_id(kwargs["id"]),
+            options=from_tess_options_to_grpc_tess_options(kwargs["options"]),
+        )
+
+        tess_map = {}
+        resp = []  # For compatibility with stream response
+        try:
+            resp_single = self.stub.GetTessellationWithOptions(request)
+            resp.append(resp_single)
+        except grpc.RpcError:
+            resp = self.stub.GetTessellationStream(request)
+
+        for elem in resp:
+            for face_id, face_tess in elem.face_tessellation.items():
+                tess_map[face_id] = from_grpc_tess_to_pd(face_tess)
+
+        return {"tessellation": tess_map}
 
     @protect_grpc
     def boolean(self, **kwargs) -> dict:  # noqa: D102

src/ansys/geometry/core/_grpc/_services/v0/conversions.py

@@ -21,6 +21,8 @@
 # SOFTWARE.
 """Module containing v0 related conversions from PyAnsys Geometry objects to gRPC messages."""
 
+from typing import TYPE_CHECKING
+
 import pint
 
 from ansys.api.dbu.v0.dbumodels_pb2 import EntityIdentifier
@@ -31,16 +33,23 @@
     MaterialProperty as GRPCMaterialProperty,
     Plane as GRPCPlane,
     Point as GRPCPoint,
+    Tessellation,
+    TessellationOptions as GRPCTessellationOptions,
 )
 from ansys.geometry.core.materials.material import Material
 from ansys.geometry.core.materials.property import MaterialProperty, MaterialPropertyType
 from ansys.geometry.core.math.frame import Frame
 from ansys.geometry.core.math.plane import Plane
 from ansys.geometry.core.math.point import Point3D
 from ansys.geometry.core.math.vector import UnitVector3D
+from ansys.geometry.core.misc.checks import graphics_required
 from ansys.geometry.core.misc.measurements import DEFAULT_UNITS
+from ansys.geometry.core.misc.options import TessellationOptions
 from ansys.geometry.core.misc.units import UNITS
 
+if TYPE_CHECKING:
+    import pyvista as pv
+
 
 def from_point3d_to_grpc_point(point: Point3D) -> GRPCPoint:
     """Convert a ``Point3D`` class to a point gRPC message.
@@ -227,3 +236,35 @@ def from_plane_to_grpc_plane(plane: Plane) -> GRPCPlane:
             dir_y=from_unit_vector_to_grpc_direction(plane.direction_y),
         )
     )
+
+
+@graphics_required
+def from_grpc_tess_to_pd(tess: Tessellation) -> "pv.PolyData":
+    """Convert an ``ansys.api.geometry.Tessellation`` to ``pyvista.PolyData``."""
+    # lazy imports here to improve initial load
+    import numpy as np
+    import pyvista as pv
+
+    return pv.PolyData(var_inp=np.array(tess.vertices).reshape(-1, 3), faces=tess.faces)
+
+
+def from_tess_options_to_grpc_tess_options(options: TessellationOptions) -> GRPCTessellationOptions:
+    """Convert a ``TessellationOptions`` class to a tessellation options gRPC message.
+
+    Parameters
+    ----------
+    options : TessellationOptions
+        Source tessellation options data.
+
+    Returns
+    -------
+    GRPCTessellationOptions
+        Geometry service gRPC tessellation options message.
+    """
+    return GRPCTessellationOptions(
+        surface_deviation=options.surface_deviation,
+        angle_deviation=options.angle_deviation,
+        maximum_aspect_ratio=options.max_aspect_ratio,
+        maximum_edge_length=options.max_edge_length,
+        watertight=options.watertight,
+    )

src/ansys/geometry/core/designer/body.py

@@ -34,7 +34,6 @@
 from ansys.api.dbu.v0.dbumodels_pb2 import EntityIdentifier
 from ansys.api.geometry.v0.bodies_pb2 import (
     BooleanRequest,
-    GetTessellationRequest,
 )
 from ansys.api.geometry.v0.bodies_pb2_grpc import BodiesStub
 from ansys.api.geometry.v0.commands_pb2 import (
@@ -48,12 +47,10 @@
     ShellRequest,
 )
 from ansys.api.geometry.v0.commands_pb2_grpc import CommandsStub
-from ansys.api.geometry.v0.models_pb2 import TessellationOptions as GRPCTessellationOptions
 from ansys.geometry.core.connection.client import GrpcClient
 from ansys.geometry.core.connection.conversions import (
     plane_to_grpc_plane,
     sketch_shapes_to_grpc_geometries,
-    tess_to_pd,
     trimmed_curve_to_grpc_trimmed_curve,
     unit_vector_to_grpc_direction,
 )
@@ -1220,48 +1217,17 @@ def tessellate(  # noqa: D102
         # cache tessellation
         if not self._tessellation:
             if tess_options is not None:
-                request = GetTessellationRequest(
-                    id=self._grpc_id,
-                    options=GRPCTessellationOptions(
-                        surface_deviation=tess_options.surface_deviation,
-                        angle_deviation=tess_options.angle_deviation,
-                        maximum_aspect_ratio=tess_options.max_aspect_ratio,
-                        maximum_edge_length=tess_options.max_edge_length,
-                        watertight=tess_options.watertight,
-                    ),
+                resp = self._grpc_client.services.body_service.get_tesellation_with_options(
+                    id=self.id,
+                    options=tess_options,
                 )
-                try:
-                    resp = self._bodies_stub.GetTessellationWithOptions(request)
-                    self._tessellation = {
-                        str(face_id): tess_to_pd(face_tess)
-                        for face_id, face_tess in resp.face_tessellation.items()
-                    }
-                except Exception:
-                    tessellation_map = {}
-                    for response in self._bodies_stub.GetTessellationStream(request):
-                        for key, value in response.face_tessellation.items():
-                            tessellation_map[key] = tess_to_pd(value)
-
-                    self._tessellation = tessellation_map
             else:
-                try:
-                    resp = self._bodies_stub.GetTessellation(self._grpc_id)
-                    self._tessellation = {
-                        str(face_id): tess_to_pd(face_tess)
-                        for face_id, face_tess in resp.face_tessellation.items()
-                    }
-                except Exception as err:
-                    # Streaming is not supported in older versions...
-                    if self._grpc_client.backend_version < (25, 2, 0):
-                        raise err
-
-                    tessellation_map = {}
-                    request = GetTessellationRequest(self._grpc_id)
-                    for response in self._bodies_stub.GetTessellationStream(request):
-                        for key, value in response.face_tessellation.items():
-                            tessellation_map[key] = tess_to_pd(value)
-
-                    self._tessellation = tessellation_map
+                resp = self._grpc_client.services.body_service.get_tesellation(
+                    id=self.id,
+                    backend_version=self._grpc_client.backend_version,
+                )
+
+            self._tessellation = resp.get("tessellation")
 
         pdata = [tess.transform(transform, inplace=False) for tess in self._tessellation.values()]
         comp = pv.MultiBlock(pdata)