unskipped design tests

adding more beams support

Author: jacobrkerstetter

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

@@ -21,6 +21,7 @@
 # SOFTWARE.
 """Provides for creating and managing a beam."""
 
+from enum import Enum
 from typing import TYPE_CHECKING
 
 from ansys.geometry.core.math.point import Point3D
@@ -30,6 +31,31 @@
 
 if TYPE_CHECKING:  # pragma: no cover
     from ansys.geometry.core.designer.component import Component
+    from ansys.geometry.core.designer.part import Part
+    from ansys.geometry.core.materials.material import Material
+    from ansys.geometry.core.math.frame import Frame
+    from ansys.geometry.core.shapes.curves.trimmed_curve import TrimmedCurve
+    from ansys.geometry.core.shapes.parameterization import ParamUV
+
+
+class BeamType(Enum):
+    """Provides values for the beam types supported."""
+
+    BEAM = 0
+    SPRING = 1
+    LINK_TRUSS = 2
+    CABLE = 3
+    PIPE = 4
+    THERMALFLUID = 5
+    UNKNOWN = 6 
+
+
+class SectionAnchorType(Enum):
+    """Provides values for the section anchor types supported."""
+
+    CENTROID = 0
+    SHEAR_CENTER = 1
+    ANCHOR_LOCATION = 2
 
 
 class BeamProfile:
@@ -66,7 +92,6 @@ def name(self) -> str:
         """Name of the beam profile."""
         return self._name
 
-
 class BeamCircularProfile(BeamProfile):
     """Represents a single circular beam profile.
 
@@ -145,6 +170,162 @@ def __repr__(self) -> str:
         return "\n".join(lines)
 
 
+class BeamCrossSectionInfo:
+    """Represents the cross-section information for a beam.
+
+    Parameters
+    ----------
+    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 
+        beam path.
+    section_frame : Frame
+        The section frame at the start of the beam.
+    section_profile : BeamProfile
+        The section profile in the XY plane.
+    """
+
+    def __init__(
+        self,
+        section_anchor: SectionAnchorType,
+        section_angle: float,
+        section_frame: Frame,
+        section_profile: BeamProfile,
+    ):
+        """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
+        self._section_profile = section_profile
+
+    @property
+    def section_anchor(self) -> SectionAnchorType:
+        """Specifies how the beam section is anchored to the beam path."""
+        return self._section_anchor
+
+    @property
+    def section_angle(self) -> float:
+        """The rotation angle of the cross section clockwise from the default perpendicular of the beam path."""
+        return self._section_angle
+
+    @property
+    def section_frame(self) -> Frame:
+        """The section frame at the start of the beam."""
+        return self._section_frame
+
+    @property
+    def section_profile(self) -> BeamProfile:
+        """The section profile in the XY plane."""
+        return self._section_profile
+
+    def __repr__(self) -> str:
+        """Represent the ``BeamCrossSectionInfo`` as a string."""
+        lines = [f"ansys.geometry.core.designer.BeamCrossSectionInfo {hex(id(self))}"]
+        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)])
+        return "\n".join(lines)
+
+class BeamProperties:
+    """Represents the properties of a beam.
+    
+    Parameters
+    ----------
+    area : float
+        The cross-sectional area of the beam.
+    centroid : ParamUV
+        The centroid of the beam section.
+    warping_constant : float
+        The warping constant of the beam.
+    ixx : float
+        The moment of inertia about the x-axis.
+    ixy : float
+        The product of inertia.
+    iyy : float
+        The moment of inertia about the y-axis.
+    shear_center : ParamUV
+        The shear center of the beam.
+    torsion_constant : float
+        The torsion constant of the beam.
+    """
+
+    def __init__(
+        self,
+        area: float,
+        centroid: ParamUV,
+        warping_constant: float,
+        ixx: float,
+        ixy: float,
+        iyy: float,
+        shear_center: ParamUV,
+        torsion_constant: float,
+    ):
+        """Initialize ``BeamProperties`` class."""
+        check_type(area, float)
+        check_type(centroid, ParamUV)
+        check_type(warping_constant, float)
+        check_type(ixx, float)
+        check_type(ixy, float)
+        check_type(iyy, float)
+        check_type(shear_center, ParamUV)
+        check_type(torsion_constant, float)
+
+        self._area = area
+        self._centroid = centroid
+        self._warping_constant = warping_constant
+        self._ixx = ixx
+        self._ixy = ixy
+        self._iyy = iyy
+        self._shear_center = shear_center
+        self._torsion_constant = torsion_constant
+
+    @property
+    def area(self) -> float:
+        """The cross-sectional area of the beam."""
+        return self._area
+
+    @property
+    def centroid(self) -> ParamUV:
+        """The centroid of the beam section."""
+        return self._centroid
+
+    @property
+    def warping_constant(self) -> float:
+        """The warping constant of the beam."""
+        return self._warping_constant
+
+    @property
+    def ixx(self) -> float:
+        """The moment of inertia about the x-axis."""
+        return self._ixx
+
+    @property
+    def ixy(self) -> float:
+        """The product of inertia."""
+        return self._ixy
+
+    @property
+    def iyy(self) -> float:
+        """The moment of inertia about the y-axis."""
+        return self._iyy
+
+    @property
+    def shear_center(self) -> ParamUV:
+        """The shear center of the beam."""
+        return self._shear_center
+
+    @property
+    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.
 
@@ -172,7 +353,17 @@ def __init__(
         start: Point3D,
         end: Point3D,
         profile: BeamProfile,
-        parent_component: "Component",
+        parent_component: Component,
+        name: str = None,
+        is_deleted: bool = False,
+        is_reversed: bool = False,
+        is_rigid: bool = False,
+        material: Material = None,
+        cross_section: BeamCrossSectionInfo = None,
+        properties: BeamProperties = None,
+        shape: TrimmedCurve = None,
+        type: BeamType = None,
+        
     ):
         """Initialize ``Beam`` class."""
         from ansys.geometry.core.designer.component import Component
@@ -189,6 +380,16 @@ def __init__(
         self._profile = profile
         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
 
     @property
     def id(self) -> str:

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

@@ -59,6 +59,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_beam_to_beam,
+    grpc_material_to_material,
     grpc_matrix_to_matrix,
     plane_to_grpc_plane,
     point3d_to_grpc_point,
@@ -89,6 +91,7 @@
 from ansys.geometry.core.shapes.curves.trimmed_curve import TrimmedCurve
 from ansys.geometry.core.shapes.parameterization import Interval
 from ansys.geometry.core.shapes.surfaces import TrimmedSurface
+from ansys.geometry.core.sketch.arc import Arc
 from ansys.geometry.core.sketch.sketch import Sketch
 from ansys.geometry.core.typing import Real
 
@@ -1121,7 +1124,11 @@ def translate_bodies(
     @check_input_types
     @ensure_design_is_active
     def create_beams(
-        self, segments: list[tuple[Point3D, Point3D]], profile: BeamProfile
+        self,
+        segments: list[tuple[Point3D, Point3D]],
+        profile: BeamProfile,
+        arcs: list[Arc] = None,
+        circles: list[Circle] = None,
     ) -> list[Beam]:
         """Create beams under the component.
 
@@ -1131,6 +1138,10 @@ def create_beams(
             list of start and end pairs, each specifying a single line segment.
         profile : BeamProfile
             Beam profile to use to create the beams.
+        arcs : list[Curve], default: None
+            list of arcs to create beams from.
+        circles : list[Curve], default: None
+            list of circles to create beams from.
 
         Returns
         -------
@@ -1145,7 +1156,7 @@ def create_beams(
         if self._grpc_client.backend_version < (25, 2, 0):
             return self.__create_beams_legacy(segments, profile)
         else:
-            return self.__create_beams(segments, profile)
+            return self.__create_beams(segments, profile, arcs, circles)
 
     def __create_beams_legacy(self, segments: list[tuple[Point3D, Point3D]], profile: BeamProfile
     ) -> list[Beam]:
@@ -1191,8 +1202,12 @@ def __create_beams_legacy(self, segments: list[tuple[Point3D, Point3D]], profile
         self._beams.extend(new_beams)
         return self._beams[-n_beams:]
 
-    beams = []
-    def __create_beams(self, segments: list[tuple[Point3D, Point3D]], profile: BeamProfile
+    def __create_beams(
+            self,
+            segments: list[tuple[Point3D, Point3D]],
+            profile: BeamProfile,
+            arcs: list[Arc] = None,
+            circles: list[Circle] = None,
     ) -> list[Beam]:
         """Create beams under the component.
 
@@ -1208,31 +1223,39 @@ def __create_beams(self, segments: list[tuple[Point3D, Point3D]], profile: BeamP
         list[Beam]
             A list of the created Beams. 
         """
+        request = CreateBeamSegmentRequest(
+            profile=profile.id,
+            parent=self.id,
+        )
+
         for segment in segments:
-            request = CreateBeamSegmentRequest(
-                profile=profile.id,
-                parent=self.id,
+            request.lines.append(
+                Line(start=point3d_to_grpc_point(segment[0]), end=point3d_to_grpc_point(segment[1]))
             )
 
-            for segment in segments:
-                request.lines.append(
-                    Line(start=point3d_to_grpc_point(segment[0]), end=point3d_to_grpc_point(segment[1]))
-                )
+        self._grpc_client.log.debug(f"Creating beams on {self.id}...")
+        response = self._commands_stub.CreateDescriptiveBeamSegments(request)
+        self._grpc_client.log.debug("Beams successfully created.")
 
-            self._grpc_client.log.debug(f"Creating beams on {self.id}...")
-            response = self._commands_stub.CreateBeamSegments(request)
-            self._grpc_client.log.debug("Beams successfully created.")
-
-            for beam in response.created_beams:
-                beams.append(
-                    Beam(
-                        id = beam.id,
-                        start = ,
-                        end = ,
-                        profile=profile,
-                        parent_component=self,
-                    )
+        beams = []
+        for beam in response.created_beams:
+            beams.append( 
+                Beam(
+                    beam.name,
+                    beam.can_suppress,
+                    beam.is_deleted,
+                    beam.is_reversed,
+                    beam.is_rigid,
+                    grpc_material_to_material(beam.material),
+                    beam.id,
+                    self,
+                    grpc_cross_section_to_cross_section(beam.cross_section),
+                    grpc_beam_properties_to_beam_properties(beam.properties),
+                    beam.shape,
+                    beam.type,
+                    beams.append(grpc_beam_to_beam(beam))
                 )
+            )
 
         return beams