beams imported from existing designs and existing named selections

Author: jacobrkerstetter

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

@@ -217,7 +217,7 @@ def section_frame(self) -> Frame:
         return self._section_frame
 
     @property
-    def section_profile(self) -> BeamProfile:
+    def section_profile(self) -> list[list[TrimmedCurve]] | None:
         """The section profile in the XY plane."""
         return self._section_profile
 
@@ -349,7 +349,9 @@ def __init__(
         id: str,
         start: Point3D,
         end: Point3D,
-        profile: BeamProfile,
+        profile: BeamProfile | None,
+        # TODO: Beams need BeamProfiles imported from existing design
+        # https://github.com/ansys/pyansys-geometry/issues/1825
         parent_component: "Component",
         name: str = None,
         is_deleted: bool = False,
@@ -359,7 +361,7 @@ def __init__(
         cross_section: BeamCrossSectionInfo = None,
         properties: BeamProperties = None,
         shape: TrimmedCurve = None,
-        type: BeamType = None,
+        beam_type: BeamType = None,
 
     ):
         """Initialize ``Beam`` class."""
@@ -368,7 +370,7 @@ def __init__(
         check_type(id, str)
         check_type(start, Point3D)
         check_type(end, Point3D)
-        check_type(profile, BeamProfile)
+        check_type(profile, (type(None), BeamProfile))
         check_type(parent_component, Component)
 
         self._id = id
@@ -378,15 +380,14 @@ def __init__(
         self._parent_component = parent_component
         self._is_alive = True
         self._name = name
-        self._id = id
         self._is_deleted = is_deleted
         self._is_reversed = is_reversed
         self._is_rigid = is_rigid
         self._material = material
         self._cross_section = cross_section
         self._properties = properties
         self._shape = shape
-        self._type = type
+        self._type = beam_type
 
     @property
     def id(self) -> str:

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

@@ -1246,7 +1246,6 @@ def __create_beams(
 
         beams = []
         for beam in response.created_beams:
-            print(beam.cross_section.section_profile)
             cross_section = BeamCrossSectionInfo(
                 SectionAnchorType(beam.cross_section.section_anchor),
                 beam.cross_section.section_angle,
@@ -1259,7 +1258,7 @@ def __create_beams(
                         Interval(curve.interval_start, curve.interval_end),
                         curve.length) for curve in curve_list] 
                     for curve_list in beam.cross_section.section_profile],
-                )  
+            )  
             properties = BeamProperties(
                 beam.properties.area,
                 ParamUV(beam.properties.centroid_x, beam.properties.centroid_y),

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

@@ -25,6 +25,8 @@
 from pathlib import Path
 from typing import Union
 
+from ansys.geometry.core.shapes.curves.trimmed_curve import TrimmedCurve
+from ansys.geometry.core.shapes.parameterization import Interval, ParamUV
 from beartype import beartype as check_input_types
 from google.protobuf.empty_pb2 import Empty
 import numpy as np
@@ -57,13 +59,15 @@
 from ansys.api.geometry.v0.parts_pb2_grpc import PartsStub
 from ansys.geometry.core.connection.backend import BackendType
 from ansys.geometry.core.connection.conversions import (
+    grpc_curve_to_curve,
     grpc_frame_to_frame,
+    grpc_material_to_material,
     grpc_matrix_to_matrix,
     grpc_point_to_point3d,
     plane_to_grpc_plane,
     point3d_to_grpc_point,
 )
-from ansys.geometry.core.designer.beam import Beam, BeamCircularProfile, BeamProfile
+from ansys.geometry.core.designer.beam import Beam, BeamCircularProfile, BeamCrossSectionInfo, BeamProfile, BeamProperties, SectionAnchorType
 from ansys.geometry.core.designer.body import Body, MasterBody, MidSurfaceOffsetType
 from ansys.geometry.core.designer.component import Component, SharedTopologyType
 from ansys.geometry.core.designer.coordinate_system import CoordinateSystem
@@ -1138,6 +1142,53 @@ def __read_existing_design(self) -> None:
             m = Material(material.name, density, properties)
             self.materials.append(m)
 
+        # Create Beams
+        for beam in response.beams:
+            cross_section = BeamCrossSectionInfo(
+                SectionAnchorType(beam.cross_section.section_anchor),
+                beam.cross_section.section_angle,
+                grpc_frame_to_frame(beam.cross_section.section_frame),
+                [
+                    [TrimmedCurve(
+                        grpc_curve_to_curve(curve.curve),
+                        grpc_point_to_point3d(curve.start),
+                        grpc_point_to_point3d(curve.end),
+                        Interval(curve.interval_start, curve.interval_end),
+                        curve.length) for curve in curve_list.curves] 
+                    for curve_list in beam.cross_section.section_profile],
+                )  
+            properties = BeamProperties(
+                beam.properties.area,
+                ParamUV(beam.properties.centroid_x, beam.properties.centroid_y),
+                beam.properties.warping_constant,
+                beam.properties.ixx,
+                beam.properties.ixy,
+                beam.properties.iyy,
+                ParamUV(beam.properties.shear_center_x, beam.properties.shear_center_y),
+                beam.properties.torsional_constant,
+            )
+
+            new_beam = Beam(
+                beam.id.id,
+                grpc_point_to_point3d(beam.shape.start),
+                grpc_point_to_point3d(beam.shape.end),
+                None,
+                # TODO: Beams need BeamProfiles imported from existing design
+                # https://github.com/ansys/pyansys-geometry/issues/1825
+                self,
+                beam.name,
+                beam.is_deleted,
+                beam.is_reversed,
+                beam.is_rigid,
+                grpc_material_to_material(beam.material),
+                cross_section,
+                properties,
+                beam.shape,
+                beam.type,
+            )
+
+            self._beams.append(new_beam)
+
         # Create NamedSelections
         for ns in response.named_selections:
             result = self._named_selections_stub.Get(EntityIdentifier(id=ns.id))
@@ -1146,8 +1197,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])
-            print(result.beams)
+            beams = get_beams_from_ids(self, [beam.id.id for beam in result.beams])
 
             design_points = []
             for dp in result.design_points:
@@ -1167,15 +1217,6 @@ def __read_existing_design(self) -> None:
             )
             self._named_selections[new_ns.name] = new_ns
 
-        # Create Beams
-        # for beam in response.beams:
-        #     new_beam = Beam(
-        #         beam.id,
-
-        #     )
-
-        #     self._beams.append(new_beam)
-
         # Create CoordinateSystems
         num_created_coord_systems = 0
         for component_id, coordinate_systems in response.component_coord_systems.items():