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

Add Mobject.set_uniform

Browse Source
This commit is contained in:
Grant Sanderson
2023-01-28 11:35:15 -08:00
parent 05a80f53a7
commit ab03a0cfba
1 changed files with 12 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
manimlib/mobject/mobject.py
View File

@ -1809,23 +1809,26 @@ class Mobject(object):
def affects_shader_info_id(func: Callable):
@wraps(func)
def wrapper(self):
for mob in self.get_family():
func(mob)
mob.refresh_shader_wrapper_id()
return self
def wrapper(self, *args, **kwargs):
result = func(self, *args, **kwargs)
self.refresh_shader_wrapper_id()
return result
return wrapper
@affects_shader_info_id
def fix_in_frame(self, recurse: bool = True):
def set_uniform(self, recurse: bool = True, **new_uniforms):
for mob in self.get_family(recurse):
mob.uniforms["is_fixed_in_frame"] = 1.0
mob.uniforms.update(new_uniforms)
return self
@affects_shader_info_id
def fix_in_frame(self, recurse: bool = True):
self.set_uniform(recurse, is_fixed_in_frame=1.0)
return self
@affects_shader_info_id
def unfix_from_frame(self, recurse: bool = True):
for mob in self.get_family(recurse):
mob.uniforms["is_fixed_in_frame"] = 0.0
self.set_uniform(recurse, is_fixed_in_frame=0.0)
return self
def is_fixed_in_frame(self) -> bool: