Implement GeometryTrees to contain and handle Regions

src/ansys/motorcad/core/geometry.py

@@ -61,6 +61,7 @@ class RegionType(Enum):
     rotor_pocket = "Rotor Pocket"
     pole_spacer = "Pole Spacer"
     rotor_slot = "Rotor Slot"
+    rotor_bar_end_ring = "Rotor Bar End Ring"
     coil_separator = "Coil Separator"
     damper_bar = "Damper Bar"
     wedge_rotor = "Rotor Wedge"
@@ -75,6 +76,7 @@ class RegionType(Enum):
     barrier = "Barrier"
     mounting_base = "Base Mount"
     mounting_plate = "Plate Mount"
+    endcap = "Endcap"
     banding = "Banding"
     sleeve = "Sleeve"
     rotor_cover = "Rotor Cover"
@@ -84,7 +86,9 @@ class RegionType(Enum):
     slot_wj_duct_no_detail = "Slot Water Jacket Duct (no detail)"
     cowling = "Cowling"
     cowling_gril = "Cowling Grill"
+    cowling_grill_hole = "Cowling Grill Hole"
     brush = "Brush"
+    bearings = "Bearings"
     commutator = "Commutator"
     airgap = "Airgap"
     dxf_import = "DXF Import"
@@ -1589,6 +1593,17 @@ def get_arc_intersection(self, arc):
         """
         return arc.get_line_intersection(self)
 
+    def get_coordinate_distance(self, coordinate):
+        """Get distance of line with another coordinate."""
+        normal_angle = self.angle - 90
+        defining_point = Coordinate.from_polar_coords(1, normal_angle)
+        normal = Line(Coordinate(0, 0), defining_point)
+        normal.translate(coordinate.x, coordinate.y)
+        nearest_point = self.get_line_intersection(normal)
+        if nearest_point is None:
+            return None
+        return sqrt((coordinate.x - nearest_point.x) ** 2 + (coordinate.y - nearest_point.y) ** 2)
+
 
 class _BaseArc(Entity):
     """Internal class to allow creation of Arcs."""

src/ansys/motorcad/core/methods/adaptive_geometry.py

@@ -24,6 +24,7 @@
 from warnings import warn
 
 from ansys.motorcad.core.geometry import Region, RegionMagnet
+from ansys.motorcad.core.methods.geometry_tree import GeometryTree
 from ansys.motorcad.core.rpc_client_core import MotorCADError, is_running_in_internal_scripting
 
 
@@ -304,3 +305,15 @@ def reset_adaptive_geometry(self):
         # No need to do this if running internally
         if not is_running_in_internal_scripting():
             return self.connection.send_and_receive(method)
+
+    def get_geometry_tree(self):
+        """Fetch a GeometryTree object containing all the defining geometry of the loaded motor."""
+        method = "GetGeometryTree"
+        json = self.connection.send_and_receive(method)
+        return GeometryTree._from_json(json, self)
+
+    def set_geometry_tree(self, tree: GeometryTree):
+        """Use a GeometryTree object to set the defining geometry of the loaded motor."""
+        params = [tree._to_json()]
+        method = "SetGeometryTree"
+        return self.connection.send_and_receive(method, params)