public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add Mobject.looks_identical

Browse source
This commit is contained in:
Grant Sanderson 2022-04-23 18:50:45 -07:00
parent d9475a6860
commit c3c5717dde
1 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
manimlib/mobject/mobject.py
View file

@ -590,6 +590,23 @@ class Mobject(object):
self.refresh_bounding_box(recurse_down=True) self.refresh_bounding_box(recurse_down=True)
return self return self
def looks_identical(self, mobject: Mobject):
fam1 = self.get_family()
fam2 = mobject.get_family()
if len(fam1) != len(fam2):
return False
for m1, m2 in zip(fam1, fam2):
for d1, d2 in [(m1.data, m2.data), (m1.uniforms, m2.uniforms)]:
if set(d1).difference(d2):
return False
for key in d1:
if isinstance(d1[key], np.ndarray):
if not np.all(d1[key] == d2[key]):
return False
elif d1[key] != d2[key]:
return False
return True
# Creating new Mobjects from this one # Creating new Mobjects from this one
def replicate(self, n: int) -> Group: def replicate(self, n: int) -> Group: