IfcOpenShell#1153 update draw.py with two new options

aothms · aothms · commit a17c44c076ee · 2021-11-20T14:05:11.000+01:00
diff --git a/src/ifcopenshell-python/ifcopenshell/draw.py b/src/ifcopenshell-python/ifcopenshell/draw.py
@@ -54,6 +54,8 @@ class draw_settings:
     include_entities: str = ""
     exclude_entities: str = "IfcOpeningElement"
     drawing_guid: str = ""
+    profile_threshold: int = -1
+    cells: bool = True
 
 
 def main(settings, files, iterators=None, merge_projection=True, progress_function=DO_NOTHING):
@@ -115,6 +117,7 @@ def yield_from_iterator(it):
     # sane default?
     sr.setAlwaysProject(True)
 
+    sr.setProfileThreshold(settings.profile_threshold)
     sr.setBoundingRectangle(settings.width, settings.height)
     sr.setScale(settings.scale)
     sr.setAutoElevation(settings.auto_elevation)
@@ -167,6 +170,9 @@ def yield_from_iterator(it):
 
     if not merge_projection:
         return svg_data_1
+    
+    if not settings.cells:
+        return svg_data_1.encode("ascii", "xmlcharrefreplace")
 
     # Parse SVG into vector of line segments
     #
