wip importing beams in NS

Author: jacobrkerstetter

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

@@ -24,6 +24,8 @@
 from enum import Enum
 from typing import TYPE_CHECKING
 
+from beartype import beartype as check_input_types
+
 from ansys.geometry.core.materials.material import Material
 from ansys.geometry.core.math.frame import Frame
 from ansys.geometry.core.math.point import Point3D
@@ -185,19 +187,15 @@ class BeamCrossSectionInfo:
         The section profile in the XY plane.
     """
 
+    @check_input_types
     def __init__(
         self,
         section_anchor: SectionAnchorType,
         section_angle: float,
         section_frame: Frame,
-        section_profile: BeamProfile,
+        section_profile: list[list[TrimmedCurve]] | None,
     ):
         """Initialize ``BeamCrossSectionInfo`` class."""
-        check_type(section_anchor, SectionAnchorType)
-        check_type(section_angle, float)
-        check_type(section_frame, Frame)
-        check_type(section_profile, BeamProfile)
-
         self._section_anchor = section_anchor
         self._section_angle = section_angle
         self._section_frame = section_frame

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

@@ -58,6 +58,8 @@
 from ansys.api.geometry.v0.models_pb2 import Direction, Line, TrimmedCurveList
 from ansys.geometry.core.connection.client import GrpcClient
 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,
@@ -68,7 +70,13 @@
     trimmed_surface_to_grpc_trimmed_surface,
     unit_vector_to_grpc_direction,
 )
-from ansys.geometry.core.designer.beam import Beam, BeamProfile
+from ansys.geometry.core.designer.beam import (
+    Beam,
+    BeamCrossSectionInfo,
+    BeamProfile,
+    BeamProperties,
+    SectionAnchorType,
+)
 from ansys.geometry.core.designer.body import Body, CollisionType, MasterBody
 from ansys.geometry.core.designer.coordinate_system import CoordinateSystem
 from ansys.geometry.core.designer.designpoint import DesignPoint
@@ -88,7 +96,7 @@
 from ansys.geometry.core.misc.measurements import DEFAULT_UNITS, Angle, Distance
 from ansys.geometry.core.shapes.curves.circle import Circle
 from ansys.geometry.core.shapes.curves.trimmed_curve import TrimmedCurve
-from ansys.geometry.core.shapes.parameterization import Interval
+from ansys.geometry.core.shapes.parameterization import Interval, ParamUV
 from ansys.geometry.core.shapes.surfaces import TrimmedSurface
 from ansys.geometry.core.sketch.arc import Arc
 from ansys.geometry.core.sketch.sketch import Sketch
@@ -1238,6 +1246,31 @@ 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,
+                grpc_frame_to_frame(beam.cross_section.section_frame),
+                [
+                    [TrimmedCurve(
+                        grpc_curve_to_curve(curve.geometry),
+                        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] 
+                    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,
+            )
+
             beams.append( 
                 Beam(
                     beam.id.id,
@@ -1250,10 +1283,8 @@ def __create_beams(
                     beam.is_reversed,
                     beam.is_rigid,
                     grpc_material_to_material(beam.material),
-                    None,
-                    None,
-                    # grpc_cross_section_to_cross_section(beam.cross_section),
-                    # grpc_beam_properties_to_beam_properties(beam.properties),
+                    cross_section,
+                    properties,
                     beam.shape,
                     beam.type,
                 )

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

@@ -1147,6 +1147,7 @@ def __read_existing_design(self) -> None:
             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)
 
             design_points = []
             for dp in result.design_points:
@@ -1166,6 +1167,15 @@ 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():