Geometry enhancements

weldx/geometry.py

@@ -14,6 +14,7 @@
 
 import weldx.transformations as tf
 import weldx.util as ut
+from weldx.constants import Q_
 from weldx.constants import WELDX_UNIT_REGISTRY as UREG
 
 _DEFAULT_LEN_UNIT = UREG.millimeters
@@ -23,6 +24,8 @@
 if TYPE_CHECKING:  # pragma: no cover
     import matplotlib.axes
 
+    import weldx.welding.groove.iso_9692_1 as iso
+
 # helper -------------------------------------------------------------------------------
 
 
@@ -2002,26 +2005,33 @@ class Geometry:
 
     """
 
-    def __init__(self, profile, trace):
+    def __init__(
+        self,
+        profile: Union[Profile, VariableProfile, iso.IsoBaseGroove],
+        trace_or_length: Union[Trace, pint.Quantity],
+    ):
         """Construct a geometry.
 
         Parameters
         ----------
         profile : Profile, VariableProfile
             Constant or variable profile that is used as cross section along the
             specified trace
-        trace : Trace
-            The path that is used to extrude the given profile
+        trace_or_length :
+            The path that is used to extrude the given profile or a quantity that
+            specifies the length of a linear, horizontal extrusion
 
         Returns
         -------
         Geometry :
             A Geometry class instance
 
         """
-        self._check_inputs(profile, trace)
+        if isinstance(trace_or_length, Q_):
+            trace_or_length = Trace(LinearHorizontalTraceSegment(trace_or_length))
+        self._check_inputs(profile, trace_or_length)
         self._profile = profile
-        self._trace = trace
+        self._trace = trace_or_length
 
     def __repr__(self):
         """Output representation of a Geometry class."""

weldx/tests/test_geometry.py

@@ -2849,7 +2849,7 @@ def get_test_geometry_constant_profile() -> geo.Geometry:
     """
     profile = get_test_profile()
     trace = geo.Trace([geo.LinearHorizontalTraceSegment(Q_(1, "cm"))])
-    return geo.Geometry(profile=profile, trace=trace)
+    return geo.Geometry(profile=profile, trace_or_length=trace)
 
 
 def get_test_geometry_variable_profile():
@@ -2866,7 +2866,7 @@ def get_test_geometry_variable_profile():
         [profile, profile], [0, 1], [geo.linear_profile_interpolation_sbs]
     )
     trace = geo.Trace([geo.LinearHorizontalTraceSegment(Q_(1, "cm"))])
-    return geo.Geometry(profile=variable_profile, trace=trace)
+    return geo.Geometry(profile=variable_profile, trace_or_length=trace)
 
 
 class TestGeometry: