public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-11-14 08:47:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Unfactor internals of batch_by_property for easier readibility, and slight performnace improvement

Browse source
This commit is contained in:
Grant Sanderson 2020-06-15 12:00:03 -07:00
parent 1e6eef89ec
commit edec91e1e4
1 changed files with 6 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
manimlib/utils/iterables.py
View file

@ -53,22 +53,21 @@ def batch_by_property(items, property_func):
preserved)
"""
batch_prop_pairs = []
def add_batch_prop_pair(batch):
if len(batch) > 0:
prop = property_func(batch[0])
batch_prop_pairs.append((batch, prop))
curr_batch = []
curr_prop = None
for item in items:
prop = property_func(item)
if prop != curr_prop:
add_batch_prop_pair(curr_batch)
# Add current batch
if len(curr_batch) > 0:
batch_prop_pairs.append((curr_batch, curr_prop))
# Redefine curr
curr_prop = prop
curr_batch = [item]
else:
curr_batch.append(item)
add_batch_prop_pair(curr_batch)
if len(curr_batch) > 0:
batch_prop_pairs.append((curr_batch, curr_prop))
return batch_prop_pairs