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

Allow Mobject.remove to remove any family member, not just immediate submobjects

Browse source
This commit is contained in:
Grant Sanderson 2023-01-27 15:15:16 -08:00
parent 1f6363821b
commit d5b1a1725d
1 changed files with 9 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
manimlib/mobject/mobject.py
View file

@ -408,14 +408,15 @@ class Mobject(object):
self.assemble_family()
return self
def remove(self, *mobjects: Mobject, reassemble: bool = True):
for mobject in mobjects:
if mobject in self.submobjects:
self.submobjects.remove(mobject)
if self in mobject.parents:
mobject.parents.remove(self)
if reassemble:
self.assemble_family()
def remove(self, *to_remove: Mobject, reassemble: bool = True):
for parent in self.get_family():
for child in to_remove:
if child in parent.submobjects:
parent.submobjects.remove(child)
if parent in child.parents:
child.parents.remove(parent)
if reassemble:
parent.assemble_family()
return self
def add_to_back(self, *mobjects: Mobject):