added beam scene object

src/compas_cadwork/scene/__init__.py

@@ -1,16 +1,20 @@
 from compas.plugins import plugin
 from compas.scene import register
 
+from compas_timber.elements import Beam
+
 from .camera import Camera
 from .scene import CadworkSceneObject
 from .instructionobject import Text3dSceneObject
 from .instructionobject import LinearDimensionSceneObject
+from .beamobject import BeamSceneObject
 
 __all__ = [
     "Camera",
     "CadworkSceneObject",
     "Text3dSceneObject",
     "LinearDimensionSceneObject",
+    "BeamSceneObject",
 ]
 
 
@@ -29,6 +33,7 @@ def after_draw(*args, **kwargs):
 
 @plugin(category="factories", requires=[CONTEXT])
 def register_scene_objects():
+    register(Beam, BeamSceneObject, context=CONTEXT)
     try:
         from compas_monosashi.sequencer import Text3d
         from compas_monosashi.sequencer import LinearDimension

src/compas_cadwork/scene/beamobject.py

@@ -0,0 +1,40 @@
+from compas_timber.elements import Beam
+
+from compas_cadwork.conversions import point_to_cadwork
+from compas_cadwork.conversions import vector_to_cadwork
+
+import element_controller as ec
+import attribute_controller as ac
+
+from .scene import CadworkSceneObject
+
+
+class BeamSceneObject(CadworkSceneObject):
+    """Scene object for COMPAS Timber Beam objects.
+
+    Parameters
+    ----------
+    item : :class:`compas_timber.elements.Beam`
+        The beam object.
+    """
+
+    def __init__(self, item: Beam, **kwargs) -> None:
+        super().__init__(item)
+
+    def draw(self):
+        """Draw the beam object in the scene.
+
+        Returns
+        -------
+        list
+            List of drawn element ids.
+        """
+        beam = self._item
+        origin = point_to_cadwork(beam.frame.point)
+        xaxis = vector_to_cadwork(beam.frame.xaxis)
+        zaxis = vector_to_cadwork(beam.frame.normal)
+        element_id = ec.create_rectangular_beam_vectors(beam.width, beam.height, beam.length, origin, xaxis, zaxis)
+        beam.attributes["cadwork_id"] = element_id
+        beam.attributes["name"] = f"beam_{beam.guid}"
+        ac.set_name([element_id], beam.attributes["name"])
+        return [element_id]