public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-09-01 00:48:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Small tweaks to Mobject.looks_identical for marginal speed

Browse source
This commit is contained in:
Grant Sanderson 2022-04-27 09:52:27 -07:00
parent 2dcc989bb4
commit c498b88750
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
manimlib/mobject/mobject.py
View file

@ -619,8 +619,8 @@ class Mobject(object):
return self return self
def looks_identical(self, mobject: Mobject): def looks_identical(self, mobject: Mobject):
fam1 = self.get_family() fam1 = self.family_members_with_points()
fam2 = mobject.get_family() fam2 = mobject.family_members_with_points()
if len(fam1) != len(fam2): if len(fam1) != len(fam2):
return False return False
for m1, m2 in zip(fam1, fam2): for m1, m2 in zip(fam1, fam2):
@ -628,10 +628,12 @@ class Mobject(object):
if set(d1).difference(d2): if set(d1).difference(d2):
return False return False
for key in d1: for key in d1:
if isinstance(d1[key], np.ndarray): eq = (d1[key] == d2[key])
if not np.all(d1[key] == d2[key]): if isinstance(eq, bool):
if not eq:
return False return False
elif d1[key] != d2[key]: else:
if not eq.all():
return False return False
return True return True