wip - adding beams to reading existing doc

Author: jacobrkerstetter

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

@@ -80,6 +80,7 @@
 from ansys.geometry.core.math.point import Point3D
 from ansys.geometry.core.math.vector import UnitVector3D, Vector3D
 from ansys.geometry.core.misc.auxiliary import (
+    get_beams_from_ids,
     get_bodies_from_ids,
     get_edges_from_ids,
     get_faces_from_ids,
@@ -1145,6 +1146,7 @@ def __read_existing_design(self) -> None:
             bodies = get_bodies_from_ids(self, [body.id for body in result.bodies])
             faces = get_faces_from_ids(self, [face.id for face in result.faces])
             edges = get_edges_from_ids(self, [edge.id for edge in result.edges])
+            beams = get_beams_from_ids(self, [beam.id for beam in result.beams])
 
             design_points = []
             for dp in result.design_points:
@@ -1159,7 +1161,7 @@ def __read_existing_design(self) -> None:
                 bodies=bodies,
                 faces=faces,
                 edges=edges,
-                beams=[],  # BEAM IMPORT NOT SUPPORTED FOR NAMED SELECTIONS
+                beams=beams,
                 design_points=design_points,
             )
             self._named_selections[new_ns.name] = new_ns

src/ansys/geometry/core/misc/auxiliary.py

@@ -24,6 +24,7 @@
 from typing import TYPE_CHECKING, Union
 
 if TYPE_CHECKING:  # pragma: no cover
+    from ansys.geometry.core.designer.beam import Beam
     from ansys.geometry.core.designer.body import Body
     from ansys.geometry.core.designer.component import Component
     from ansys.geometry.core.designer.design import Design
@@ -237,6 +238,30 @@ def get_edges_from_ids(design: "Design", edge_ids: list[str]) -> list["Edge"]:
     ]  # noqa: E501
 
 
+def get_beams_from_ids(design: "Design", beam_ids: list[str]) -> list["Beam"]:
+    """Find the ``Beam`` objects inside a ``Design`` from its ids.
+
+    Parameters
+    ----------
+    design : Design
+        Parent design for the beams.
+    beam_ids : list[str]
+        List of beam ids.
+
+    Returns
+    -------
+    list[Beam]
+        List of Beam objects.
+
+    Notes
+    -----
+    This method takes a design and beam ids, and gets their corresponding ``Beam`` objects.
+    """
+    return [
+        beam for beam in design.beams if beam.id in beam_ids
+    ]  # noqa: E501
+
+
 def convert_color_to_hex(
     color: str | tuple[float, float, float] | tuple[float, float, float, float],
 ) -> str:

tests/integration/test_design.py

@@ -466,12 +466,18 @@ def test_named_selection_contents(modeler: Modeler):
     face = box_2.faces[2]
     edge = box_2.edges[0]
 
+    circle_profile_1 = design.add_beam_circular_profile(
+        "CircleProfile1", Quantity(10, UNITS.mm), Point3D([0, 0, 0]), UNITVECTOR3D_X, UNITVECTOR3D_Y
+    )
+    beam = design.create_beam(
+        Point3D([9, 99, 999], UNITS.mm), Point3D([8, 88, 888], UNITS.mm), circle_profile_1
+    )
+
     # Create the NamedSelection
     ns = design.create_named_selection(
-        "MyNamedSelection", bodies=[box, box_2], faces=[face], edges=[edge]
+        "MyNamedSelection", bodies=[box, box_2], faces=[face], edges=[edge], beams=[beam]
     )
 
-    print(ns.bodies)
     # Check that the named selection has everything
     assert len(ns.bodies) == 2
     assert np.isin([box.id, box_2.id], [body.id for body in ns.bodies]).all()
@@ -482,7 +488,7 @@ def test_named_selection_contents(modeler: Modeler):
     assert len(ns.edges) == 1
     assert ns.edges[0].id == edge.id
 
-    assert len(ns.beams) == 0
+    assert len(ns.beams) == 1
     assert len(ns.design_points) == 0