+jddlZddlZddlZddlZddlZddlZddlm Z ddl Z ddl Z ddl Z ddlmZmZmZddlmZddlmZejeZdZededadadadZGd d e Zd Z d Z!d Z"dZ#dZ$dZ%dZ&de'de'fdZ(de'de)fdZ*de'de)fdZ+dZ,dZ-dZ.GddZ/de'de)de)fdZ0de'de'de)fdZ1 d.d e'd!e)d"e)d#e2e2e j3d$e2e4d%e2e)d&e2e5dzd'e2e j3dzde2e2e j3fd(Z6Gd)d*ej7Z8Gd+d,eZ9de9fd-Z:dS)/N) TypedDict)ComfyExtensionioui)override) get_missing_requirements_messagectdg}tdtjd|dtdtjd|d|rt dtd td tj tj d rtj d ptj d }tdt||stdtjd}tdtjd}td|rdndd|rdndtddS)zMEarly check for OpenGL availability. Raises RuntimeError if unlikely to work.z$_check_opengl_availability: startingz5_check_opengl_availability: checking for glfw packageglfwNz7_check_opengl_availability: checking for OpenGL packageOpenGLPyOpenGLz#OpenGL dependencies not available.  z%_check_opengl_availability: platform=linuxDISPLAYWAYLAND_DISPLAYz(_check_opengl_availability: has_display=z4_check_opengl_availability: checking for EGL libraryEGLz7_check_opengl_availability: checking for OSMesa libraryOSMesazHeadless mode: EGL=yesnoz , OSMesa=z%_check_opengl_availability: completed)loggerdebug importlibutil find_specappend RuntimeErrorrsysplatform startswithosenvirongetboolctypes find_library)missing has_displayhas_egl has_osmesas 7/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_glsl.py_check_opengl_availabilityr*s LL7888G LLHIII~''/v LLJKKK~))1z"""  Y3S3U3U Y Y Y    LLGGGHHH |w''ujnnY//T2:>>BS3T3T  S[@Q@QSSTTT u LLO P P Pk..u55G LLR S S S11(;;J LLs/IuuTss]gTqTYTYmqss t t t LL899999z=nodes_glsl: running _check_opengl_availability at import timectV3Rr} U U UF VGV T TF 6 ))r+ctjd|}|sdStd|D}t|dztS)z{Detect how many fragColor outputs are used in the shader. Returns the count of outputs needed (1 to MAX_OUTPUTS). zfragColor(\d+)c34K|]}t|VdSN)r<).0ms r) z'_detect_output_count..s(,,qCFF,,,,,,r+)rGfindallmaxmin MAX_OUTPUTS)rBmatches max_indexs r)_detect_output_countrXsS j*F33G q,,G,,,,,I y1}k * **r+ctjd|}|r0tdt|dSdS)zyDetect multi-pass rendering from #pragma passes N directive. Returns the number of passes (1 if not specified). z#pragma\s+passes\s+(\d+)rL)rGsearchrSr<group)rBmatchs r)_detect_pass_countr]sE I16 : :E +1c%++a..))*** 1r+ctdtjdkr)tdt dtdddl}td|st d  td ||j|j ||j d ||j d ||j |j td |d d ddd}|st dtd||td||fS#t $r0td|wxYw)zOInitialize GLFW. Returns (window, glfw_module). Raises RuntimeError on failure.z_init_glfw: startingdarwinz_init_glfw: skipping on macOSz#GLFW backend not supported on macOSz!_init_glfw: importing glfw modulerNz_init_glfw: calling glfw.init()zglfw.init() failedz _init_glfw: setting window hintsz#_init_glfw: calling create_window()@z ComfyUI GLSLzglfw.create_window() failedz*_init_glfw: calling make_context_current()z"_init_glfw: completed successfullyz$_init_glfw: failed, terminating glfw)rrrrrr init window_hintVISIBLEFALSECONTEXT_VERSION_MAJORCONTEXT_VERSION_MINOROPENGL_PROFILEOPENGL_CORE_PROFILE create_windowmake_context_current Exception terminate)_glfwwindows r) _init_glfwrps LL'((( |x 4555@AAA LL4555 LL2333 ::<<1/000 7888 %-555 %5q999 %5q999 %.0IJJJ :;;;$$R^T4HH ><== = ABBB ""6*** 9:::u}  ;<<<  s 0D G:H c`tdddlm}ddlm}m}m}m}m }m }m }m }m } m} m} m} m} m}m}m}m}m}m}m}m}m}m}m}m}tdd}d}d} td|| }||jkrt?dtd | | }}||||sd}t?d td |j!d |j!|||||d |d |d |d |d| g}|j"dz} | }!|||| d|!r |!j!dkrt?d| d}"td|!j!||st?dtd|j#d|j$d|j%|j&| g}#|||"| |#}|| krt?dtd|d|d| g}$|||"|$}||j'krt?dtd|||||st?dtd||||fS#tP$rEtd| | ||| | ||| ||wxYw)zwInitialize EGL for headless rendering. Returns (display, context, surface, EGL_module). Raises RuntimeError on failure.z_init_egl: startingr)r) eglGetDisplay eglInitializeeglChooseConfigeglCreateContexteglMakeCurrenteglCreatePbufferSurface eglBindAPI eglTerminateeglDestroyContexteglDestroySurfaceEGL_DEFAULT_DISPLAYEGL_NO_CONTEXTEGL_NONEEGL_SURFACE_TYPEEGL_PBUFFER_BITEGL_RENDERABLE_TYPEEGL_OPENGL_BIT EGL_RED_SIZEEGL_GREEN_SIZE EGL_BLUE_SIZEEGL_ALPHA_SIZEEGL_DEPTH_SIZE EGL_WIDTH EGL_HEIGHTEGL_OPENGL_APIz_init_egl: imports completedNz"_init_egl: calling eglGetDisplay()zeglGetDisplay() failedz"_init_egl: calling eglInitialize()zeglInitialize() failedz_init_egl: EGL version .rLzeglChooseConfig() failedz&_init_egl: config chosen, num_configs=zeglBindAPI() failedz%_init_egl: calling eglCreateContext()r`zeglCreateContext() failedz,_init_egl: calling eglCreatePbufferSurface()raz eglCreatePbufferSurface() failedz#_init_egl: calling eglMakeCurrent()zeglMakeCurrent() failedz!_init_egl: completed successfullyz_init_egl: failed, cleaning up))rrr r OpenGL.EGLrrrsrtrurvrwrxryrzr{r|r}r~rrrrrrrrrrrrEGL_NO_DISPLAYrEGLintvalue EGLConfigEGL_CONTEXT_MAJOR_VERSIONEGL_CONTEXT_MINOR_VERSIONEGL_CONTEXT_OPENGL_PROFILE_MASK#EGL_CONTEXT_OPENGL_CORE_PROFILE_BITEGL_NO_SURFACErl)%_EGLrrrsrtrurvrwrxryrzr{r|r}r~rrrrrrrrrrrrdisplaycontextsurfacemajorminorconfig_attribsconfigs num_configsconfigcontext_attribspbuffer_attribss% r) _init_eglrs} LL&'''"""""" LL/000GGG> 9:::- 344 d) ) )788 8 9:::{{}}dkkmmu}WeU33 9G788 8 Ju{JJU[JJKKK o  !^Q q.RS Ax   &4>A%((kkmm wKPP ;T_TeijTjTj9:: : Qk>OQQRRRz.)) 6455 5 <===  *A  *A  0$2Z    #"7FNOTT n $ $:;; ; CDDD$b*b(C))'6?KK d) ) )ABB B :;;;~gwAA :899 9 8999$..     5666    gw / / /    gw / / /   L ! ! !  s8I%KAL-c.ddl}tddtjd<tdddlm}ddlm}m }m }m }td ||d ddd}|std d \}}|j ||zd zz} td||| |j||s||tdtd|| fS)zdInitialize OSMesa for software rendering. Returns (context, buffer). Raises RuntimeError on failure.rNz_init_osmesa: startingosmesaPYOPENGL_PLATFORMz%_init_osmesa: importing OpenGL.osmesa)r/)OSMesaCreateContextExtOSMesaMakeCurrentOSMesaDestroyContext OSMESA_RGBAz_init_osmesa: imports completedzOSMesaCreateContextExt() failed)rararAz)_init_osmesa: calling OSMesaMakeCurrent()zOSMesaMakeCurrent() failedz$_init_osmesa: completed successfully)r#rrrr r r/ OpenGL.osmesarrrrrc_ubyteGL_UNSIGNED_BYTE) r#r0rrrrctxr5r6buffers r) _init_osmesarsdMMM LL)***&.BJ"# LL8999       LL2333 b!Q = =C ><===ME63fn 23 6 6F LL<===  S&#*>v N N9S!!!7888 LL7888 ;r+c6eZdZdZdZdZfdZdZdZxZ S) GLContextuManages OpenGL context and resources for shader execution. Tries backends in order: GLFW (desktop) → EGL (headless GPU) → OSMesa (software). NFcl|j&t||_|jSrN) _instancesuper__new__)cls __class__s r)rzGLContext.__new__Cs* = !GGOOC00CM}r+c  tjrtddStdddl}|}d|_d|_d|_d|_ d|_ d|_ d|_ d|_ g}td t\|_ad|_tdnK#t $r>}td||d|fYd}~nd}~wwxYw|jtd  t%\|_|_ |_ ad |_td nK#t $r>}td ||d |fYd}~nd}~wwxYw|jtd t)\|_ |_ d|_tdnK#t $r>}td||d|fYd}~nd}~wwxYw|j\t*jdkrd}nt*jdkrd}nd}dd|D}t1d|d|tdt3td t4d}t4|||_ tdnd#t $rW}td ||r. t4d|gn#t $rYnwxYwYd}~nd}~wwxYw||z d!z}t4t4j} t4t4j } t4t4j!} | r| "nd"} | r| "nd"} | r| "nd"} d#t_t#d$|d%d&|jd'| d(| d)| dS)*Nz1GLContext.__init__: already initialized, skippingz+GLContext.__init__: starting initializationrz'GLContext.__init__: trying GLFW backendr z*GLContext.__init__: GLFW backend succeededz)GLContext.__init__: GLFW backend failed: GLFWz&GLContext.__init__: trying EGL backendeglz)GLContext.__init__: EGL backend succeededz(GLContext.__init__: EGL backend failed: rz)GLContext.__init__: trying OSMesa backendrz,GLContext.__init__: OSMesa backend succeededz+GLContext.__init__: OSMesa backend failed: rwin32zWindows: Ensure GPU drivers are installed and display is available. CPU-only/headless mode is not supported on Windows.r_zmacOS: GLFW is not supported. Install OSMesa via Homebrew: brew install mesa Then: pip install PyOpenGL PyOpenGL-acceleratezLinux: Install one of these backends: Desktop: sudo apt install libgl1-mesa-glx libglfw3 Headless with GPU: sudo apt install libegl1-mesa libgl1-mesa-dri Headless (CPU): sudo apt install libosmesa6r c3,K|]\}}d|d|VdS)z z: Nr=)rOnameerrs r)rQz%GLContext.__init__..s7%Q%QYT3&84&8&83&8&8%Q%Q%Q%Q%Q%Qr+z2Failed to create OpenGL context. Backend errors: z z'GLContext.__init__: importing OpenGL.GLz GLContext.__init__: creating VAOrLz,GLContext.__init__: VAO created successfullyzFGLContext.__init__: VAO creation failed (may be expected for OSMesa): UnknownTzGLSL context initialized in .1fms (z) - z (z), GL )$r _initializedrrtime perf_counter_backend_window _egl_display _egl_context _egl_surface _osmesa_ctx_osmesa_buffer_vaorpr rlrrrrrrjoinrr1r-glGenVertexArraysglBindVertexArrayglDeleteVertexArrays glGetString GL_RENDERER GL_VENDOR GL_VERSIONdecodeinfo) selfrstarterrorse platform_help error_detailsvaoelapsedrenderervendorversions r)__init__zGLContext.__init__Hs?  !  LLL M M M F BCCC  !!##     "  >??? '!+ DL$"DM LLE F F F F ' ' ' LLHQHH I I I MM61+ & & & & & & & & ' = LLA B B B *OX{{L!4#4d6G %  HIIII * * * KKKLLL uaj)))))))) * = LLD E E E -8D5 $"5 (  KLLLL - - - N1NNOOO xm,,,,,,,, - = |w&&S ))G G!II%Q%Q&%Q%Q%QQQM#$1## ##   >???  7888 &&q))C   % % %DI LLG H H H H    LLebcee f f f ++Au5555 D $$&&.$6>>".11 --..//(0?8??$$$i$*9 &-<'.."""9!%  z7zzzDMzzW_zzcizzqxzz{{{{{s07C(( D024D++D0AF G!#4GG!u>>??? Ns 7&A fragment_coder5r6 image_batchesfloatsintsboolscurvesc%ddl}|} |sgSt} | t |} t |} t |} |g}|g}d}d}g}g}g}g}g}t|d} tt| }n,#t$rt d| wxYwt|td}ttj|g}t%| D]v}td}||ttj|ttjdtj||dtjtjd ttjtjtjttjtjtjttjtj |ztj|d|tj |zxt!| |t"tjtj#krtd| dkrt%dD]}td}||ttj|ttjdtj||dtjtjd ttjtjtjttjtjtjttjtj$tj%ttjtj&tj%td}||ttj|ttjtj tj|dt!dtj gt"tjtj#krtdt%|D]}td}||t'tj(|zttj|ttjtjtjttjtjtjttjtj$tj%ttjtj&tj%t)|d|}|dkrt*||t)|d }|dkr6t+|tY|tY|t[|D]D\}}t)|d |}|dkrt.||Et[|D]D\}}t)|d |}|dkrt*||Et[|D]H\}}t)|d |}|dkrt*||rdndIt[|D] \}}td}||t^|z}t'tj(|zttj|ttjdtj0t|ddtj1tj| ttjtjtjttjtjtjttjtj$tj%ttjtj&tj%t)|d |}|dkrt*|| t)|d} t2dd||t3tj4g}!|D]}"t[|"D]%\}}#t'tj(|zttj|||#j5\}$}%}&|&dkrNtmj7|$|%dftlj8}'|#dddddddf|'ddddddf<d|'dddddf<n$tmj9|#dddddddf}'ttjdtj|%|$dtjtj|' '| dkrttj|| dkrt*| dt:ddddt;tj<t=tj>ddnt%| D]u}(|(| dz k})| dkrt*| |(|)r&ttj|n0||(dz}*ttj|*t'tj(|(dkr,ttj|dn3||(dz dz}+ttj|+t:ddddt;tj<t=tj>ddwg},|D]}ttj|t?tjdtjtj}-tmj@|-tlj8A||d}#|,|#dddddddfBtmjC||dftlj8}.t%| tD]}|,|.|!|,|| z dz}/t|}0| dkrd| dnd}1tEd|/dd|0d|0dkrdndd|d||1d |!ttjdtd|D])}tFt|*|D])}tFt|*|D])}tFt|*|D])}tFt|*|tHd|g|D]}tHd|g|tI|SS#ttjdtd|D])}tFt|*|D])}tFt|*|D])}tFt|*|D])}tFt|*|tHd|g|D]}tHd|g|tI|wwxYw)a Render a fragment shader for multiple batches efficiently. Compiles shader once, reuses framebuffer/textures across batches. Supports multi-pass rendering via #pragma passes N directive. Args: fragment_code: User's fragment shader code width: Output width height: Output height image_batches: List of batches, each batch is a list of input images (H, W, C) float32 [0,1] floats: List of float uniforms ints: List of int uniforms bools: List of bool uniforms (passed as int 0/1 to GLSL bool uniforms) curves: List of 1D LUT arrays (float32) of arbitrary size for u_curve0-N Returns: List of batch outputs, each is a list of output images (H, W, 4) float32 [0,1] rNzFragment shader: rLzFramebuffer is not completez%Ping-pong framebuffer is not completeu_image u_resolutionu_floatu_intu_boolu_curveu_passr`rA)dtypeg?rz, z passesrEzGLSL shader executed in rrz batchesx))JrrrrrJrXr]lenr VERTEX_SHADERrrrr- glUseProgramglGenFramebuffersglBindFramebufferGL_FRAMEBUFFERrange glGenTexturesr glBindTexture GL_TEXTURE_2D glTexImage2D GL_RGBA32FGL_RGBAGL_FLOATglTexParameteriGL_TEXTURE_MIN_FILTER GL_LINEARGL_TEXTURE_MAG_FILTERglFramebufferTexture2DGL_COLOR_ATTACHMENT0 glDrawBuffersglCheckFramebufferStatusGL_FRAMEBUFFER_COMPLETEGL_TEXTURE_WRAP_SGL_CLAMP_TO_EDGEGL_TEXTURE_WRAP_TglActiveTexture GL_TEXTURE0glGetUniformLocation glUniform1i glUniform2ffloat enumerate glUniform1f MAX_IMAGESGL_R32FGL_RED glViewport glDisableGL_BLENDshapenpemptyfloat32ascontiguousarray glClearColorglClearGL_COLOR_BUFFER_BIT glDrawArrays GL_TRIANGLES glGetTexImage frombufferreshapecopyzerosrUrglDeleteTexturesr<glDeleteFramebuffersr)2rr5r6rr r r r r start_timerr num_outputs num_passesrfbooutput_texturesinput_texturescurve_texturesping_pong_texturesping_pong_fbos num_inputs draw_buffersitex_pp_texpp_fbolocvlutunitpass_locall_batch_outputsimagesimghwc img_uploadp is_last_pass target_fbo source_tex batch_outputsdata black_imgr num_batches pass_infos2 r)_render_shader_batchrysL:KKK""$$J  ++C-];;O'}55K$M22J } ~G CONNN]1%&&JP( %m_EEGG    LL?o?? @ @ @      ""1%% R.444 {## = =A""1%%C  " "3 ' ' '   R-s 3 3 3 OOB,avqRTR\^`^iko p p p   r/1I2< X X X   r/1I2< X X X  % %b&79PST9TVXVfhkmn o o o    7! ; < < < < l333 & &r'8 9 9R=W W W<== = >>1XX P P))!,,"))&111  !16::: 0!R]E6STVXV`bdbmosttt""2#3R5Mr|\\\""2#3R5Mr|\\\""2#3R5I2K^___""2#3R5I2K^___--a00%%f---$$R%6???))"*;R=TVXVfhnpqrrr  R%<$=>>>..r/@AARE___&'NOOO`z"" ' 'A""1%%C  ! !# & & &   r~1 2 2 2   R-s 3 3 3   r/1I2< X X X   r/1I2< X X X   r/1ErGZ [ [ [   r/1ErGZ [ [ [))'=Q==AACaxxsA&&&%%g~>> !88 NN3e eFmm < < <f%% ' 'DAq))'=Q==AACaxxsA&&&dOO ' 'DAq))';1;;??CaxxsA&&&e$$ 3 3DAq))'D   r~4 5 5 5   R-s 3 3 3 OOB,aSXXq!RYXZXceh i i i   r/1I2< X X X   r/1I2< X X X   r/1ErGZ [ [ [   r/1ErGZ [ [ [))'=Q==AACaxxsD)))**7H== aE6*** R[!!!#F 4F 4F#F++ r r3""2>A#5666  !1>!3DEEE)1a66!#1a)2:!F!F!FJ+.tttQQQz?Jqqq!!!RaRx(*-Jqqq!!!Qw''!#!5c$$B$111*o!F!FJ 0!R]Aq!RZY[YdfpqqqqQ$$R%6<<<q==NN8Q///1a+++ 21222A6666z**;;A$%a$7L 1}}x333#L,,R->DDDD&4AE%: ,,R-> KKK&&r~666Avv(()9>!;LMMMM&8Q! %D (()9:FFFOOAq!Q///JJr5666OOBOQ::::M& = =  !13777''(8!RZUUmD ;;;CCFESTUU$$S2qqq!!!_%9%9%;%;<<<<&%!32:FFFI; 44 0 0$$Y////  $ $] 3 3 3 3$$&&3t;-(( 0:Q,,,,,B  TwTTT TT[fjk[k[kSWSWqsTTw|TT@FTHQTTT U U U  R.222 ! * *C   C ) ) ) )! * *C   C ) ) ) )" * *C   C ) ) ) )% * *C   C ) ) ) ) ?  # #Au - - -$ 1 1F  # #Ax 0 0 0 0     w ' ' ' '  R.222 ! * *C   C ) ) ) )! * *C   C ) ) ) )" * *C   C ) ) ) )% * *C   C ) ) ) ) ?  # #Au - - -$ 1 1F  # #Ax 0 0 0 0     w ' ' ' ' s$B/.AF/)C};AFFE AK,c8eZdZedejfdZe ddededej j dej j dej j d ej j d ej j dej fd Z ed e ejd ejdeee ffdZdS) GLSLShaderrCcvtjtjdddt }tjtjddddt}tjtjd dd dt}tjtj d d d dt}tjtj dddt}tj ddddddtjdtddtjdtjdgtjdtjdddt$jtjdddt$jggd tjd!|d"t dz d#$tjd%|d&tdz d'$tjd(|d)tdz d'$tjd*|d+tdz d,$tjd-|d.tdz d/$gtjd0d12tjd3d42tjd5d62tjd7d82g9S):NimagerL)prefixrTrSr:)defaultrrr<rr"Frcurverr{z GLSL Shaderz image/shaderzRApply GLSL ES fragment shaders to images. u_resolution (vec2) is always available.TrzFGLSL fragment shader source code (GLSL ES 3.00 / WebGL 2.0 compatible))r multilinetooltipr4 from_inputcustomr5i)rrTrSr6zXOutput size: 'from_input' uses first input image dimensions, 'custom' allows manual size)optionsrrjz!Images are available as u_image0-z (sampler2D) in the shader code)templaterr z!Floats are available as u_float0-z in the shader coder zInts are available as u_int0-r z"Booleans are available as u_bool0-z (bool) in the shader coder z!Curves are available as u_curve0-zV (sampler2D, 1D LUT) in the shader code. Sample with texture(u_curve0, vec2(x, 0.5)).rIMAGE0zIAvailable via layout(location = 0) out vec4 fragColor0 in the shader code) display_namerIMAGE1zIAvailable via layout(location = 1) out vec4 fragColor1 in the shader codeIMAGE2zIAvailable via layout(location = 2) out vec4 fragColor2 in the shader codeIMAGE3zIAvailable via layout(location = 3) out vec4 fragColor3 in the shader code)node_idrcategory descriptionis_experimentalhas_intermediate_outputinputsoutputs)rAutogrowTemplatePrefixImageInputr=Float MAX_UNIFORMSIntBoolean MAX_BOOLSCurve MAX_CURVESSchemaStringDEFAULT_FRAGMENT_SHADER DynamicComboOptionnodesMAX_RESOLUTIONOutput)rimage_templatefloat_template int_template bool_templatecurve_templates r) define_schemazGLSLShader.define_schemas33 HNN7 # # 4  33 HNN7CN 0 0 4  {11 FLLL * * 2   22 J  VU  3 3 3  33 HNN7 # # 4  y &#;!$( %3"d   %%..|R@@..$ " $+,/()(-(< !-!"!" !# $,,/()(-(< !-!"!" (w-&0 !!(^N_q{|}q}N_N_N_!`` !!(^NUq}~qNUNUNU!VV !!&<JMiuvwiwJMJMJM!NN !!'MLXpyz{p{LXLXLX!YY !!(^NVq{|}q}NVNVNV!WWG$LX@KLLX@KLLX@KLLX@KLL _5 5 5 5 r+Nrr4rjr r r r c d|D} |rd|Dng} |rd|Dng} |rd|Dng} |rd|Dng} | std|ddkr|d }|d }n| d jd d \}}| d jd }g}t|D]%fd| D}||&t ||||| | | | }dtt D|D]D}t|D]2\}}|tj |3Efdtt D}tj |d| | |d iS)Ncg|]}||SrNr=rOres r) z&GLSLShader.execute..ksBBBAAMaMMMr+cg|]}||nd S)Nrr=rs r)rz&GLSLShader.execute..ms B B BQ!-QQS B B Br+cg|]}||nd S)Nrr=rs r)rz&GLSLShader.execute..os EEE!AAAEEEr+cg|]}||nd S)NFr=rs r)rz&GLSLShader.execute..ps KKKq!-QQUKKKr+ctg|]5}||tj6SrN)to_lutastyperDrFrs r)rz&GLSLShader.execute..rs6^^^PQP]ahhjj'' 33P]P]P]r+z$At least one input image is requiredr4rr5r6rrLr`cg|]K}|tjLSr=)cpunumpyrrDrF)rO img_tensor batch_idxs r)rz&GLSLShader.execute..sGpppWaJy15577==??FFrzRRpppr+cg|]}gSr=r=)rOras r)rz&GLSLShader.execute..s666ar666r+cHg|]}tj|dS)r)dim)torchstack)rOr_ all_outputss r)rz&GLSLShader.execute..s,YYY%+k!n!<<<YYYr+r) values ValueErrorrCr!rryrUr;r from_numpyr NodeOutput_build_ui_output)rrr4rjr r r r kwargs image_list float_listint_list bool_list curve_luts out_width out_height batch_sizer batch_imagesrirtr_out_imgoutput_tensorsrrs @@r)executezGLSLShader.execute^svCBBBB FL T B B&--// B B B BRT JNUEEt{{}}EEEESUOT\KKELLNNKKKKZ\ bhp^^V]]__^^^^np  ECDD D [ !X - -!'*I"8,JJ$.qM$7!$< !J ](+  z** / /IppppeopppL   . . . .0           765#5#5666 . A AM' 66 A A 7A%%e&6w&?&?@@@@ AZYYYeKFXFXYYY}  ##Jq0ABB   r+r output_batchc  g}|D]H}|tj|dtjjddItj|dtjjdd}||dS)zNBuild UI output with input and output images for client-side shader execution.GLSLShader_inputNrL)filename_prefix folder_typercompress_levelGLSLShader_output) input_imagesrj)extendrImageSaveHelper save_imagesr FolderTypetemp)rrrinput_images_uirkoutput_images_uis r)rzGLSLShader._build_ui_outputs   C  " "2#5#A#A 2M. $B$$    -99 / * :  !0;KLLLr+)NNNN)r7r8r9 classmethodrrrr:r3rTyperrlistrTensordictrr=r+r)r{r{s0Y biY Y Y [Y v $(!%"&#'< < < !<   <   < k < {<   <  < < < [< |Mel+M;@<M c4iMMM[MMMr+r{cLeZdZedeeejfdZdS) GLSLExtensionrCcKtgSrN)r{)rs r) get_node_listzGLSLExtension.get_node_lists|r+N) r7r8r9rrtyper ComfyNoderr=r+r)rrsE T$r|*<%=Xr+rc"KtSrN)rr=r+r)comfy_entrypointrs ??r+)NN);rrrGlogging ctypes.utilr#importlib.utilrtypingrrrDrrcomfy_api.latestrrrtyping_extensionsrutils.install_utilr getLoggerr7rr*rr-r rr1r3r=rrrrUrrr:rJr<rXr]rprrrrrrndarrayr:r"ryrr{rrr=r+r)rs  3333333333&&&&&&??????  8 $ $':':':V LMMM   I        *3*3**** + + + + + + s s    """JQQQh>K,K,K,K,K,K,K,K,\ C c c    3@ $&*O(O(O( O( O(RZ() O( K O( s) O( : O(  t #O( $rz O(O(O(O(btMtMtMtMtMtMtMtMnN  r+