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

Bug fix to Mobject.align_family and Mobject.add_n_more_submobjects

Browse source
This commit is contained in:
Grant Sanderson 2021-01-18 08:06:12 -10:00
parent 1b1b138860
commit b3c8940a18
1 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
manimlib/mobject/mobject.py
View file

@ -1145,8 +1145,14 @@ class Mobject(object):
def align_family(self, mobject): def align_family(self, mobject):
mob1 = self mob1 = self
mob2 = mobject mob2 = mobject
n1 = len(mob1.submobjects) n1 = len(mob1)
n2 = len(mob2.submobjects) n2 = len(mob2)
while n1 == 1 and n2 > 1:
mob1.set_submobjects(mob1[0].submobjects)
n1 = len(mob1)
while n2 == 1 and n1 > 1:
mob2.set_submobjects(mob2[0].submobjects)
n2 = len(mob2)
if n1 != n2: if n1 != n2:
mob1.add_n_more_submobjects(max(0, n2 - n1)) mob1.add_n_more_submobjects(max(0, n2 - n1))
mob2.add_n_more_submobjects(max(0, n1 - n2)) mob2.add_n_more_submobjects(max(0, n1 - n2))
@ -1169,9 +1175,10 @@ class Mobject(object):
curr = len(self.submobjects) curr = len(self.submobjects)
if curr == 0: if curr == 0:
# If empty, simply add n point mobjects # If empty, simply add n point mobjects
center = self.get_center() null_mob = self.copy()
null_mob.set_points([self.get_center()])
self.set_submobjects([ self.set_submobjects([
self.copy().set_points([center]) null_mob.copy()
for k in range(n) for k in range(n)
]) ])
return self return self