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

Edit set_program_uniform

Browse Source
This commit is contained in:
Grant Sanderson
2023-02-03 11:05:40 -08:00
parent ac3db9b636
commit 7c561d3757
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
manimlib/utils/shaders.py
View File

@ -77,18 +77,22 @@ def set_program_uniform(
Returns True if changed the program, False if it left it as is.
"""
if name not in program:
return False
pid = id(program)
if pid not in PROGRAM_UNIFORM_MIRRORS:
PROGRAM_UNIFORM_MIRRORS[pid] = dict()
uniform_mirror = PROGRAM_UNIFORM_MIRRORS[pid]
if isinstance(value, np.ndarray) and value.ndim > 0:
if type(value) is np.ndarray and value.ndim > 0:
value = tuple(value)
if uniform_mirror.get(name, None) == value:
return False
try:
program[name].value = value
except KeyError:
return False
uniform_mirror[name] = value
program[name].value = value
return True