diff --git a/revitron/geometry.py b/revitron/geometry.py index 220f41c..8bd4c8a 100644 --- a/revitron/geometry.py +++ b/revitron/geometry.py @@ -70,6 +70,29 @@ def getSolids(self): pass return solids + def getCurves(self, curveType='Line'): + """ + Get a list of all curves by type of a given element. + + Note: + Child classes of Autodesk.DB.Curve can be used. + + Args: + curveType (str, optional): The type of curve to return. + Defaults to 'Line'. + + Returns: + list: A list of curve objects + """ + curves = [] + for geo in self._geometry: + try: + for item in geo.GetInstanceGeometry(): + if item.GetType().Name == curveType: + curves.append(item) + except: + pass + return curves class GeometryUtils: """