Merge branch 'feat/enhanced_beam_implementation' of https://github.com/ansys/pyansys-geometry into feat/enhanced_beam_implementation

Author: jacobrkerstetter

doc/changelog.d/1828.added.md

@@ -0,0 +1 @@
+enhanced beam implementation

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

@@ -48,7 +48,7 @@ class BeamType(Enum):
     CABLE = 3
     PIPE = 4
     THERMALFLUID = 5
-    UNKNOWN = 6 
+    UNKNOWN = 6
 
 
 class SectionAnchorType(Enum):
@@ -93,6 +93,7 @@ def name(self) -> str:
         """Name of the beam profile."""
         return self._name
 
+
 class BeamCircularProfile(BeamProfile):
     """Represents a single circular beam profile.
 
@@ -179,7 +180,7 @@ class BeamCrossSectionInfo:
     section_anchor : SectionAnchorType
         Specifies how the beam section is anchored to the beam path.
     section_angle : float
-        The rotation angle of the cross section clockwise from the default perpendicular of the 
+        The rotation angle of the cross section clockwise from the default perpendicular of the
         beam path.
     section_frame : Frame
         The section frame at the start of the beam.
@@ -227,12 +228,15 @@ def __repr__(self) -> str:
         lines.append(f"  Section Anchor       : {self.section_anchor.name}")
         lines.append(f"  Section Angle        : {self.section_angle}")
         lines.append(f"  Section Frame        : {self.section_frame}")
-        lines.extend(["\n", "  Section Profile info", "  -------------------", str(self.section_profile)])
+        lines.extend(
+            ["\n", "  Section Profile info", "  -------------------", str(self.section_profile)]
+        )
         return "\n".join(lines)
 
+
 class BeamProperties:
     """Represents the properties of a beam.
-    
+
     Parameters
     ----------
     area : float
@@ -323,6 +327,7 @@ def torsion_constant(self) -> float:
         """The torsion constant of the beam."""
         return self._torsion_constant
 
+
 class Beam:
     """Represents a simplified solid body with an assigned 2D cross-section.
 
@@ -362,7 +367,6 @@ def __init__(
         properties: BeamProperties = None,
         shape: TrimmedCurve = None,
         beam_type: BeamType = None,
-        
     ):
         """Initialize ``Beam`` class."""
         from ansys.geometry.core.designer.component import Component

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

@@ -1164,8 +1164,9 @@ def create_beams(
             return self.__create_beams_legacy(segments, profile)
         else:
             return self.__create_beams(segments, profile, arcs, circles)
-        
-    def __create_beams_legacy(self, segments: list[tuple[Point3D, Point3D]], profile: BeamProfile
+
+    def __create_beams_legacy(
+        self, segments: list[tuple[Point3D, Point3D]], profile: BeamProfile
     ) -> list[Beam]:
         """Create beams under the component.
 
@@ -1183,7 +1184,7 @@ def __create_beams_legacy(self, segments: list[tuple[Point3D, Point3D]], profile
         Returns
         -------
         list[Beam]
-            A list of the created Beams. 
+            A list of the created Beams.
         """
         request = CreateBeamSegmentsRequest(parent=self.id, profile=profile.id)
 
@@ -1208,13 +1209,13 @@ def __create_beams_legacy(self, segments: list[tuple[Point3D, Point3D]], profile
 
         self._beams.extend(new_beams)
         return self._beams[-n_beams:]
-    
+
     def __create_beams(
-            self,
-            segments: list[tuple[Point3D, Point3D]],
-            profile: BeamProfile,
-            arcs: list[Arc] = None,
-            circles: list[Circle] = None,
+        self,
+        segments: list[tuple[Point3D, Point3D]],
+        profile: BeamProfile,
+        arcs: list[Arc] = None,
+        circles: list[Circle] = None,
     ) -> list[Beam]:
         """Create beams under the component.
 
@@ -1228,7 +1229,7 @@ def __create_beams(
         Returns
         -------
         list[Beam]
-            A list of the created Beams. 
+            A list of the created Beams.
         """
         request = CreateBeamSegmentsRequest(
             profile=profile.id,
@@ -1251,14 +1252,19 @@ def __create_beams(
                 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],
-            )  
+                    [
+                        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),
@@ -1270,7 +1276,7 @@ def __create_beams(
                 beam.properties.torsional_constant,
             )
 
-            beams.append( 
+            beams.append(
                 Beam(
                     beam.id.id,
                     grpc_point_to_point3d(beam.shape.start),
@@ -1288,7 +1294,7 @@ def __create_beams(
                     beam.type,
                 )
             )
-            
+
         self._beams.extend(beams)
         return beams
 

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

@@ -25,8 +25,6 @@
 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
@@ -67,7 +65,14 @@
     plane_to_grpc_plane,
     point3d_to_grpc_point,
 )
-from ansys.geometry.core.designer.beam import Beam, BeamCircularProfile, BeamCrossSectionInfo, BeamProfile, BeamProperties, SectionAnchorType
+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
@@ -94,6 +99,8 @@
 from ansys.geometry.core.misc.options import ImportOptions
 from ansys.geometry.core.modeler import Modeler
 from ansys.geometry.core.parameters.parameter import Parameter, ParameterUpdateStatus
+from ansys.geometry.core.shapes.curves.trimmed_curve import TrimmedCurve
+from ansys.geometry.core.shapes.parameterization import Interval, ParamUV
 from ansys.geometry.core.typing import RealSequence
 
 
@@ -1149,14 +1156,19 @@ def __read_existing_design(self) -> None:
                 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],
-                )  
+                    [
+                        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),

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

@@ -257,9 +257,7 @@ def get_beams_from_ids(design: "Design", beam_ids: list[str]) -> list["Beam"]:
     -----
     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
+    return [beam for beam in design.beams if beam.id in beam_ids]  # noqa: E501
 
 
 def convert_color_to_hex(

tests/integration/test_design.py

@@ -2665,6 +2665,7 @@ def test_revolve_sketch_fail_invalid_path(modeler: Modeler):
 
 def test_component_tree_print(modeler: Modeler):
     """Test for verifying the tree print for ``Component`` objects."""
+
     def check_list_equality(lines, expected_lines):
         # By doing "a in b" rather than "a == b", we can check for substrings
         # which, in the case of beam ids, is necessary since they are unique

tests/integration/test_design_import.py

@@ -383,6 +383,6 @@ def test_design_import_with_named_selections(modeler: Modeler):
     beam1 = design._named_selections["Beam1"]
     assert len(beam1.bodies) == 0
     assert len(beam1.beams) == 1
-     
+
     beam = beam1.beams[0]
-    assert beam._name == "B1"
+    assert beam._name == "B1"