public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-04-13 09:47:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Match uniforms in Mobject.become

Browse source
This commit is contained in:
Grant Sanderson 2021-01-13 10:23:42 -10:00
parent b40a3b5710
commit 39cdae0983
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
manimlib/mobject/mobject.py
View file

@ -106,6 +106,10 @@ class Mobject(object):
for key in data: for key in data:
self.data[key] = data[key].copy() self.data[key] = data[key].copy()
def set_uniforms(self, uniforms):
for key in uniforms:
self.uniforms[key] = uniforms[key] # Copy?
def resize_points(self, new_length, resize_func=resize_array): def resize_points(self, new_length, resize_func=resize_array):
if new_length != len(self.data["points"]): if new_length != len(self.data["points"]):
self.data["points"] = resize_func(self.data["points"], new_length) self.data["points"] = resize_func(self.data["points"], new_length)
@ -1236,6 +1240,7 @@ class Mobject(object):
self.align_family(mobject) self.align_family(mobject)
for sm1, sm2 in zip(self.get_family(), mobject.get_family()): for sm1, sm2 in zip(self.get_family(), mobject.get_family()):
sm1.set_data(sm2.data) sm1.set_data(sm2.data)
sm1.set_uniforms(sm2.uniforms)
return self return self
def cleanup_from_animation(self): def cleanup_from_animation(self):
@ -1349,6 +1354,7 @@ class Mobject(object):
# For shader data # For shader data
def init_shader_data(self): def init_shader_data(self):
# TODO, only call this when needed?
self.shader_data = np.zeros(len(self.get_points()), dtype=self.shader_dtype) self.shader_data = np.zeros(len(self.get_points()), dtype=self.shader_dtype)
self.shader_indices = None self.shader_indices = None
self.shader_wrapper = ShaderWrapper( self.shader_wrapper = ShaderWrapper(