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() self.assemble_family()
return self return self
def remove(self, *mobjects: Mobject, reassemble: bool = True): def remove(self, *to_remove: Mobject, reassemble: bool = True):
for mobject in mobjects: for parent in self.get_family():
if mobject in self.submobjects: for child in to_remove:
self.submobjects.remove(mobject) if child in parent.submobjects:
if self in mobject.parents: parent.submobjects.remove(child)
mobject.parents.remove(self) if parent in child.parents:
if reassemble: child.parents.remove(parent)
self.assemble_family() if reassemble:
parent.assemble_family()
return self return self
def add_to_back(self, *mobjects: Mobject): def add_to_back(self, *mobjects: Mobject):