frame from position

Author: tomvanmele

src/compas_occ/interop/__init__.py

@@ -34,6 +34,7 @@
     compas_vector_to_occ_direction
     compas_vector_from_occ_vector
     compas_line_to_occ_line
+    compas_frame_from_occ_position
 
 Meshes
 ======
@@ -66,6 +67,7 @@
 from .primitives import compas_vector_to_occ_direction  # noqa: F401
 from .primitives import compas_vector_from_occ_vector  # noqa: F401
 from .primitives import compas_line_to_occ_line  # noqa: F401
+from .primitives import compas_frame_from_occ_position  # noqa: F401
 
 from .meshes import triangle_to_face  # noqa: F401
 from .meshes import quad_to_face  # noqa: F401

src/compas_occ/interop/primitives.py

@@ -3,7 +3,9 @@
 from compas.geometry import Point
 from compas.geometry import Vector
 from compas.geometry import Line
+from compas.geometry import Frame
 
+from OCC.Core.gp import gp_Ax3
 from OCC.Core.gp import gp_Pnt
 from OCC.Core.gp import gp_Vec
 from OCC.Core.gp import gp_Dir
@@ -35,6 +37,20 @@ def compas_vector_to_occ_direction(self: Vector) -> gp_Dir:
     return gp_Dir(* self)
 
 
+def compas_vector_from_occ_direction(cls: Vector, vector: gp_Dir) -> Vector:
+    """Construct a COMPAS vector from an OCC direction."""
+    return cls(vector.X(), vector.Y(), vector.Z())
+
+
 def compas_line_to_occ_line(self: Line) -> gp_Lin:
     """Convert a COMPAS line to an OCC line."""
     return gp_Lin(compas_point_to_occ_point(self.start), compas_vector_to_occ_direction(self.direction))
+
+
+def compas_frame_from_occ_position(cls: Frame, position: gp_Ax3) -> Frame:
+    """Construct a COMPAS frame from an OCC position."""
+    return cls(
+        compas_point_from_occ_point(Point, position.Location()),
+        compas_vector_from_occ_direction(Vector, position.XDirection()),
+        compas_vector_from_occ_direction(Vector, position.YDirection())
+    )