Add duplicate element method

Author: josemmo

src/compas_cadwork/elements/element.py

@@ -4,6 +4,7 @@
 from typing import TYPE_CHECKING
 from typing import ClassVar
 from typing import Final
+from typing import Self
 from typing import final
 from uuid import UUID
 
@@ -269,6 +270,24 @@ def translate(self, vector: Vector) -> None:
         """
         ec.move_element([self.id], compas_to_cwpoint(vector))
 
+    def duplicate(self, vector: Vector) -> Self:
+        """Duplicate element by the given vector.
+
+        Parameters
+        ----------
+        vector : Vector
+            The vector by which to duplicate the element.
+
+        Returns
+        -------
+        Self
+            New element.
+        """
+        new_element_ids = ec.copy_elements([self.id], compas_to_cwpoint(vector))
+        if len(new_element_ids) != 1:
+            raise RuntimeError(f"Failed to copy Cadwork element with ID {self.id}")
+        return self.__class__(new_element_ids[0])
+
     def delete(self) -> None:
         """Delete element.
 

tests/compas_cadwork/elements/test_element.py

@@ -363,6 +363,15 @@ def test_translates_element(cadwork) -> None:
     cadwork.ec.move_element.assert_called_once_with([123], cadwork.cadwork.point_3d(100.0, 200.0, 300.0))
 
 
+def test_duplicates_element(cadwork) -> None:
+    cadwork.ec.copy_elements.return_value = [124]
+    element = Element(123)
+    new_element = element.duplicate(Vector(100.0, 200.0, 300.0))
+    assert new_element is not element
+    assert new_element.id == 124
+    cadwork.ec.copy_elements.assert_called_once_with([123], cadwork.cadwork.point_3d(100.0, 200.0, 300.0))
+
+
 def test_deletes_element(cadwork) -> None:
     element = Element(123)
     element.delete()