Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-22 05:00:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Option to yield progress from iterate() and consume_iterator()

Browse Source
This commit is contained in:
Thomas Krijnen
2022-02-26 14:57:21 +01:00
parent d16f92a445
commit e81dd9c7df
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/geom/main.py
+7 -4
View File
@@ -211,17 +211,20 @@ def create_shape(settings, inst, repr=None):
) )
def consume_iterator(it): def consume_iterator(it, with_progress=False):
if it.initialize(): if it.initialize():
while True: while True:
yield it.get() if with_progress:
yield it.progress(), it.get()
else:
yield it.get()
if not it.next(): if not it.next():
break break
def iterate(settings, file_or_filename, num_threads=1, include=None, exclude=None): def iterate(settings, file_or_filename, num_threads=1, include=None, exclude=None, with_progress=False):
it = iterator(settings, file_or_filename, num_threads, include, exclude) it = iterator(settings, file_or_filename, num_threads, include, exclude)
yield from consume_iterator(it) yield from consume_iterator(it, with_progress=with_progress)
def make_shape_function(fn): def make_shape_function(fn):