feat: vertex references (#2083)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: pyansys-ci-bot <[email protected]>
Co-authored-by: Roberto Pastor Muela <[email protected]>

Author: 4 people

doc/changelog.d/2083.added.md

@@ -0,0 +1 @@
+Vertex references

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

@@ -119,6 +119,11 @@ def get_edges(self, **kwargs) -> dict:
         """Get the edges of a body."""
         pass
 
+    @abstractmethod
+    def get_vertices(self, **kwargs) -> dict:
+        """Get the vertices of a body."""
+        pass
+
     @abstractmethod
     def get_volume(self, **kwargs) -> dict:
         """Get the volume of a body."""

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

@@ -69,6 +69,11 @@ def get_faces(self, **kwargs) -> dict:
         """Get the faces that are connected to the edge."""
         pass
 
+    @abstractmethod
+    def get_vertices(self, **kwargs) -> dict:
+        """Get the vertices that are connected to the edge."""
+        pass
+
     @abstractmethod
     def get_bounding_box(self, **kwargs) -> dict:
         """Get the bounding box of the edge."""

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

@@ -59,6 +59,11 @@ def get_edges(self, **kwargs) -> dict:
         """Get the edges of a face."""
         pass
 
+    @abstractmethod
+    def get_vertices(self, **kwargs) -> dict:
+        """Get the vertices of a face."""
+        pass
+
     @abstractmethod
     def get_loops(self, **kwargs) -> dict:
         """Get the loops of a face."""

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

@@ -383,6 +383,22 @@ def get_edges(self, **kwargs) -> dict:  # noqa: D102
             ]
         }
 
+    @protect_grpc
+    def get_vertices(self, **kwargs) -> dict:  # noqa: D102
+        # Call the gRPC service
+        resp = self.stub.GetVertices(request=build_grpc_id(kwargs["id"]))
+
+        # Return the response - formatted as a dictionary
+        return {
+            "vertices": [
+                {
+                    "id": vertex.id.id,
+                    "position": from_grpc_point_to_point3d(vertex.position),
+                }
+                for vertex in resp.vertices
+            ]
+        }
+
     @protect_grpc
     def get_volume(self, **kwargs) -> dict:  # noqa: D102
         # Call the gRPC service

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

@@ -138,6 +138,27 @@ def get_faces(self, **kwargs) -> dict:  # noqa: D102
             ],
         }
 
+    @protect_grpc
+    def get_vertices(self, **kwargs) -> dict:  # noqa: D102
+        from ansys.api.geometry.v0.edges_pb2 import GetVerticesRequest
+
+        # Create the request - assumes all inputs are valid and of proper type
+        request = GetVerticesRequest(edge=build_grpc_id(kwargs["id"]))
+
+        # Call the gRPC service
+        response = self.stub.GetVertices(request=request)
+
+        # Return the response - formatted as a dictionary
+        return {
+            "vertices": [
+                {
+                    "id": vertex.id.id,
+                    "position": from_grpc_point_to_point3d(vertex.position),
+                }
+                for vertex in response.vertices
+            ],
+        }
+
     @protect_grpc
     def get_bounding_box(self, **kwargs) -> dict:  # noqa: D102
         # Create the request - assumes all inputs are valid and of the proper type

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

@@ -114,6 +114,27 @@ def get_edges(self, **kwargs) -> dict:  # noqa: D102
             ]
         }
 
+    @protect_grpc
+    def get_vertices(self, **kwargs) -> dict:  # noqa: D102
+        # Create the request - assumes all inputs are valid and of the proper type
+        from ansys.api.geometry.v0.faces_pb2 import GetVerticesRequest
+
+        request = GetVerticesRequest(face_id=build_grpc_id(kwargs["id"]))
+
+        # Call the gRPC service
+        response = self.stub.GetVertices(request=request)
+
+        # Return the response - formatted as a dictionary
+        return {
+            "vertices": [
+                {
+                    "id": vertex.id.id,
+                    "position": from_grpc_point_to_point3d(vertex.position),
+                }
+                for vertex in response.vertices
+            ]
+        }
+
     @protect_grpc
     def get_loops(self, **kwargs) -> dict:  # noqa: D102
         # Create the request - assumes all inputs are valid and of the proper type

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

@@ -67,6 +67,7 @@ def get_named_selection(self, **kwargs):  # noqa: D102
             "beams": [beam.id.id for beam in response.beams],
             "design_points": [(dp.id, dp.points[0]) for dp in response.design_points],
             "components": [comp.id for comp in response.components],
+            "vertices": [vertex.id.id for vertex in response.vertices],
         }
 
     @protect_grpc
@@ -92,6 +93,7 @@ def create_named_selection(self, **kwargs):  # noqa: D102
             "beams": [beam.id.id for beam in response.beams],
             "design_points": [dp.id for dp in response.design_points],
             "components": [comp.id for comp in response.components],
+            "vertices": [vertex.id.id for vertex in response.vertices],
         }
 
     @protect_grpc

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

@@ -112,6 +112,10 @@ def get_faces(self, **kwargs) -> dict:  # noqa: D102
     def get_edges(self, **kwargs) -> dict:  # noqa: D102
         raise NotImplementedError
 
+    @protect_grpc
+    def get_vertices(self, **kwargs) -> dict:  # noqa: D102
+        raise NotImplementedError
+
     @protect_grpc
     def get_volume(self, **kwargs) -> dict:  # noqa: D102
         raise NotImplementedError

src/ansys/geometry/core/_grpc/_services/v1/edges.py

@@ -71,6 +71,10 @@ def get_interval(self, **kwargs) -> dict:  # noqa: D102
     def get_faces(self, **kwargs) -> dict:  # noqa: D102
         return NotImplementedError
 
+    @protect_grpc
+    def get_vertices(self, **kwargs) -> dict:  # noqa: D102
+        return NotImplementedError
+
     @protect_grpc
     def get_bounding_box(self, **kwargs) -> dict:  # noqa: D102
         return NotImplementedError