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

Small clean up

Browse source
This commit is contained in:
Grant Sanderson 2023-01-26 23:05:09 -08:00
parent 422c9cebd2
commit 2863672740
1 changed files with 20 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
manimlib/mobject/mobject.py
View file

@ -1768,20 +1768,26 @@ class Mobject(object):
self.locked_data_keys = set(keys)
def lock_matching_data(self, mobject1: Mobject, mobject2: Mobject):
for sm, sm1, sm2 in zip(self.get_family(), mobject1.get_family(), mobject2.get_family()):
if sm.data.dtype == sm1.data.dtype == sm2.data.dtype:
names = sm.data.dtype.names
sm.lock_data(filter(
lambda name: arrays_match(sm1.data[name], sm2.data[name]),
names,
))
sm.const_data_keys = set(filter(
lambda name: all(
array_is_constant(mob.data[name])
for mob in (sm, sm1, sm2)
),
names
))
tuples = zip(
self.get_family(),
mobject1.get_family(),
mobject2.get_family(),
)
for sm, sm1, sm2 in tuples:
if not sm.data.dtype == sm1.data.dtype == sm2.data.dtype:
continue
names = sm.data.dtype.names
sm.lock_data(filter(
lambda name: arrays_match(sm1.data[name], sm2.data[name]),
names,
))
sm.const_data_keys = set(filter(
lambda name: all(
array_is_constant(mob.data[name])
for mob in (sm, sm1, sm2)
),
names
))
return self