opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-07-29 04:53:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add replace_submobject

Browse Source
This commit is contained in:
Grant Sanderson
2020-02-22 13:19:51 -08:00
parent 11a4d7696e
commit ed6085d07c
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
manimlib/mobject/mobject.py
View File

@ -137,6 +137,14 @@ class Mobject(Container):
self.set_submobjects(list_update(mobjects, self.sub_mobjects)) self.set_submobjects(list_update(mobjects, self.sub_mobjects))
return self return self
def replace_submobject(self, index, new_submob):
old_submob = self.submobjects[index]
if self in old_submob.parents:
old_submob.parents.remove(self)
self.submobjects[index] = new_submob
self.assemble_family()
return self
def set_submobjects(self, submobject_list): def set_submobjects(self, submobject_list):
self.remove(*self.submobjects) self.remove(*self.submobjects)
self.add(*submobject_list) self.add(*submobject_list)
@ -1037,7 +1045,7 @@ class Mobject(Container):
def push_self_into_submobjects(self): def push_self_into_submobjects(self):
copy = self.deepcopy() copy = self.deepcopy()
copy.submobjects.set_submobjects([]) copy.set_submobjects([])
self.reset_points() self.reset_points()
self.add(copy) self.add(copy)
return self return self