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

Speed-ups to Mobject.copy

Browse source
This commit is contained in:
Grant Sanderson 2022-04-24 13:24:20 -07:00
parent 66f695a1ed
commit e49e4b8373
1 changed files with 16 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
manimlib/mobject/mobject.py
View file

@ -362,13 +362,14 @@ class Mobject(object):
self.assemble_family() self.assemble_family()
return self return self
def remove(self, *mobjects: Mobject): def remove(self, *mobjects: Mobject, reassemble: bool = True):
for mobject in mobjects: for mobject in mobjects:
if mobject in self.submobjects: if mobject in self.submobjects:
self.submobjects.remove(mobject) self.submobjects.remove(mobject)
if self in mobject.parents: if self in mobject.parents:
mobject.parents.remove(self) mobject.parents.remove(self)
self.assemble_family() if reassemble:
self.assemble_family()
return self return self
def add_to_back(self, *mobjects: Mobject): def add_to_back(self, *mobjects: Mobject):
@ -389,7 +390,7 @@ class Mobject(object):
return self return self
def set_submobjects(self, submobject_list: list[Mobject]): def set_submobjects(self, submobject_list: list[Mobject]):
self.remove(*self.submobjects) self.remove(*self.submobjects, reassemble=False)
self.add(*submobject_list) self.add(*submobject_list)
return self return self
@ -534,9 +535,18 @@ class Mobject(object):
for key, value in self.uniforms.items() for key, value in self.uniforms.items()
} }
result.submobjects = [] # Instead of adding using result.add, which does some checks for updating
result.add(*(sm.copy() for sm in self.submobjects)) # updater statues and bounding box, just directly modify the family-related
result.match_updaters(self) # lists
result.submobjects = [sm.copy() for sm in self.submobjects]
for sm in result.submobjects:
sm.parents = [result]
result.family = [result, *it.chain(*(sm.get_family() for sm in result.submobjects))]
# Similarly, instead of calling match_updaters, since we know the status
# won't have changed, just directly match.
result.non_time_updaters = list(self.non_time_updaters)
result.time_based_updaters = list(self.time_based_updaters)
family = self.get_family() family = self.get_family()
for attr, value in list(self.__dict__.items()): for attr, value in list(self.__dict__.items()):