wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-11 08:47:29 +08:00
Code Issues Actions 26 Packages Projects Releases Wiki Activity

_create_program() -> to GLContext method

Browse Source
This commit is contained in:
Lex Darlog (DRL) 2026-02-23 02:38:48 -03:00
parent 2b63d8c26a
commit c9442a21a7
1 changed files with 28 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
comfy_extras/nodes_glsl.py
View File

@ -326,6 +326,33 @@ class GLContext:
return shader
def create_program(self, vertex_source: str, fragment_source: str) -> int:
"""Create and link a shader program."""
gl = self._gl
compile = self.compile_shader
vertex_shader = compile(vertex_source, gl.GL_VERTEX_SHADER)
try:
fragment_shader = compile(fragment_source, gl.GL_FRAGMENT_SHADER)
except RuntimeError:
gl.glDeleteShader(vertex_shader)
raise
program = gl.glCreateProgram()
gl.glAttachShader(program, vertex_shader)
gl.glAttachShader(program, fragment_shader)
gl.glLinkProgram(program)
gl.glDeleteShader(vertex_shader)
gl.glDeleteShader(fragment_shader)
if gl.glGetProgramiv(program, gl.GL_LINK_STATUS) != gl.GL_TRUE:
error = gl.glGetProgramInfoLog(program).decode()
gl.glDeleteProgram(program)
raise RuntimeError(f"Program linking failed:\n{error}")
return program
##########
class _GLContextGLFW(GLContext):
@ -515,32 +542,6 @@ class _GLContextOSMesa(GLContext):
############################################################
def _create_program(vertex_source: str, fragment_source: str) -> int:
"""Create and link a shader program."""
ctx = GLContext()
vertex_shader = ctx.compile_shader(vertex_source, gl.GL_VERTEX_SHADER)
try:
fragment_shader = ctx.compile_shader(fragment_source, gl.GL_FRAGMENT_SHADER)
except RuntimeError:
gl.glDeleteShader(vertex_shader)
raise
program = gl.glCreateProgram()
gl.glAttachShader(program, vertex_shader)
gl.glAttachShader(program, fragment_shader)
gl.glLinkProgram(program)
gl.glDeleteShader(vertex_shader)
gl.glDeleteShader(fragment_shader)
if gl.glGetProgramiv(program, gl.GL_LINK_STATUS) != gl.GL_TRUE:
error = gl.glGetProgramInfoLog(program).decode()
gl.glDeleteProgram(program)
raise RuntimeError(f"Program linking failed:\n{error}")
return program
def _render_shader_batch(
fragment_code: str,
width: int,
@ -597,7 +598,7 @@ def _render_shader_batch(
try:
# Compile shaders (once for all batches)
try:
program = _create_program(VERTEX_SHADER, fragment_source)
program = ctx.create_program(VERTEX_SHADER, fragment_source)
except RuntimeError:
logger.error(f"Fragment shader:\n{fragment_source}")
raise