k j!AdZddlZddlmZmZmZmZddlZddlZddl m Z dZ dZ dZ dZd Zd ZGd d ZGd dZGddeZGddZdS)z# Helper classes for loading shader N)AnyCallableOptionalUnion)ProgramDescription VERTEX_SHADERGEOMETRY_SHADERFRAGMENT_SHADERTESS_CONTROL_SHADERTESS_EVALUATION_SHADERCOMPUTE_SHADERc|eZdZdZdefdZedejfdZ e de ddede ddfdZ e dde dded e d ee d ee d ee dee ddfdZe dde ddede ddfdZdejfdZdejfdZdeegefdd fdZd S)ProgramShadersz:Helper class preparing shader source strings for a programmetach||_d|_d|_d|_d|_d|_d|_dSN)r vertex_sourcegeometry_sourcefragment_sourcetess_control_sourcetess_evaluation_sourcecompute_shader_source)selfrs Y/home/agentuser/manim-venv/lib/python3.11/site-packages/moderngl_window/opengl/program.py__init__zProgramShaders.__init__s? 597;7;;? >B#=A"""returnc(tjS)zThe moderngl context)moderngl_windowctxrs rr zProgramShaders.ctx!s"$$$rclssourcec0||}tt|jp|j||j|_t |vr.tt |jp|j||j|_t|vr.tt|jp|j ||j|_ t|vr.tt|jp|j ||j|_t|vr.tt|jp|j||j|_|S)z6Initialize a single glsl string containing all shadersdefines) ShaderSourcerpath vertex_shaderr&rr geometry_shaderrr fragment_shaderrr tess_control_shaderrr tess_evaluation_shaderr)r"rr#instances r from_singlezProgramShaders.from_single&s7 3t99!-  I ++ L " " "  f $ $'3 1T1 (((H $ f $ $'3 1T1 (((H $ & ( (+7# 5T5 ,,,H ( "V + +.:& 8T8 ///H +rNrrrrrc||}tt|jp|j||j|_|.tt |jp|j||j|_|.tt|jp|j ||j|_ |.tt|jp|j ||j|_|.tt|jp|j ||j|_|S)z"Initialize multiple shader stringsr%)r'rr(r)r&rr r*rr r+rr r,rr r)r"rrrrrrr.s r from_separatezProgramShaders.from_separateUs#3t99!-  I ++ L " " "   &'3 1T1 (((H $  &'3 1T1 (((H $  *+7# 5T5# ,,,H ( " -.:& 5T5& ///H +rrc|||}tt|jdn|j||j|_|S)Nr2r%)r'r compute_shaderr&r)r"rrr.s rr4zProgramShaders.compute_shadersL3t99)5 %-BB43F !L * * * & rcl|j Jd|j|jjS)Nz%There is not compute_shader to create)rr r4r#r!s rcreate_compute_shaderz$ProgramShaders.create_compute_shaders6)557^555x&&t'A'HIIIrc g}|j Jd|jsR|jr&|jjp|j}n%|jjp|j}|j|jj|jr |jjnd|jr |jjnd|j r |j jnd|j r |j jndt|}d|ji|_ |S)zc Creates a shader program. Returns: ModernGL Program instance Nz There is no vertex_source to use)r)r*r+r,r-varyingsr) rrrrr8find_out_attribsr programr#rrtupleextra)r out_attribsr:s rcreatezProgramShaders.creates !--/Q---# Z# Z"i0[D4H4Y4Y4[4[ #i0YD4F4W4W4Y4Y ("",3<@rrrArrrrsDDB/BBBB%X%%%%X%, " #,+=,GJ, ,,,[,\ *.)--10422 " #2 22"# 2 "# 2 &c] 2!) 2 222[2h\^   " # += VY     [ Jx'=JJJJ"(""""HI#0DIIIIIIIrrc eZdZdZ ddeedeededeeeefd ed ef d Z e d efd Z e d efdZ e d e dfdZe d eefdZe d e efdZe d efdZe d eeeffdZ ddeegefdeded dfdZdeeefd dfdZd e efdZddZd efdZdS)r'a Helper class representing a single shader type. It ensures the source has the right format, injects ``#define`` pre-processors, resolves ``#include`` pre-processors etc. A ``ShaderSource`` can be the base/root shader or a source referenced in an ``#include``. NrT shader_typenamer#r&idrootc.||_||_|g|_||_||_|in||_|r|}|d|_|jrL|jd ds,| td|jd| |j|jrA|j dd|jd |j d d dSdS) a:Create shader source. Args: shader_type (str): A preprocessor name for setting the shader type name (str): A string (usually the path) so we can give useful error messages to the user source (str): The raw source for the shader Keyword Args: id (int): The source number. Used when shader consists of multiple sources through includes root (bool): If this shader source is the root shader (Not an include) N rz#versionzMissing #version in z-. A version must be defined in the first line#define z 1z#line 2)_id_root _source_list_type_name_definesstripsplit_lines startswithprint ShaderError apply_definesinsert)rrPrQr#r&rRrSs rrzShaderSource.__init__s.0  !  %o7  $\\^^Fll4((  : dk!n77 CC  JJLLL`tz```  4=))) : - K  q";TZ";";"; < < < K  q) , , , , , - -rrc|jS)zint: The shader number/id)rYr!s rrRzShaderSource.ids xrc6d|jS)z!str: The source lines as a stringrU)joinrar!s rr#zShaderSource.sourcesyy%%%rc|jS)z.list[ShaderSource]: List of all shader sources)r[r!s r source_listzShaderSource.source_lists   rc|jS)z#str: a path or name for this shader)r]r!s rrQzShaderSource.names zrc|jS)z#list[str]: The lines in this shader)rar!s rlineszShaderSource.lines#s {rc*t|jS)z.int: Number of lines in this source (stripped))lenrar!s r line_countzShaderSource.line_count(s4;rc|jS)z(dict: Defines configured for this shader)r^r!s rr&zShaderSource.defines-s }rr?depth source_idc f|dkrtd|} t|jD]\}}|}|drt jd|}|td||d}|dz }||\} } td|| |j|d } | ||dz| |j d|| j z|j |dzdz|_|xj | j z c_ |j d j }ndS) aInject includes into the shader source. This happens recursively up to a max level in case the users has circular includes. We also build up a list of all the included sources in the root shader. Args: load_source_func (func): A function for finding and loading a source depth (int): The current include depth (increase by 1 for every call) dzEReaching an include depth of 100. You probably have circular includesTz#includez#include\s+"?([^"]+)Nz/Could not match '#include\s+"?([^"]+)' in line rVF)r&rRrS)rsrt)rd enumeraterar_rbresearchr'r^rArnr[rkrR) rr?rsrt current_idnrlinematchr(_r#s rrAzShaderSource.handle_includes2s~ 3;;W   %dk22  Dzz||??:..I&=tDDE})WQUWW!8D!OJ 0 0 6 6IAv) $ %" F**+;519Xb*ccc"&*SbS/FL"@4:bSTfhhCW"WDK%%);;%%!%!22!6!9JE+.5 rc<|sdSt|jD]\}}|}|drT |d}||}|sad|d||j|<r#t$rY~wxYwdS)z"Apply the configured define valuesNz#definerVrW )rxrar_rbr`getrn IndexError)rr&r|r}rQvalues rrezShaderSource.apply_defines`s  F!$+..  HB::<%>%>u%>%>DJrNN!D   s1B :B  BBcg}|jD]X}tjd|}|r-||dY|S)z Get all out attributes in the shader source. Returns: list[str]: List of out attribute names z.(layout(.+)\))?(\s+)?(out)(\s+)(\w+)(\s+)(\w+)rw)rnryr~r_appendgroups)rnamesr}ress rr9zShaderSource.find_out_attribsrsaJ / /D(Ldjjll[[C / SZZ\\"-... rctd|jdt|jD]9\}}tt |dd|:tddS)z&Print the shader lines (for debugging)z ---[ START z ]---z: z---[ END {self.name} ]---N)rcrQrxrnrKzfill)rir}s rrczShaderSource.prints ,DI,,,--- ,, 0 0GAt SVV\\!__.... / / / / )*****rc(d|jd|jdS)Nz)rQrRr!s r__repr__zShaderSource.__repr__s:::::::r)NrT)rr)rN)rBrCrDrErrKdictintboolrrFrRr#listrkrQrnrqr&rrrArer9rcrrNrrr'r's|-10-0-c]0-sm0- 0- $sCx.) 0-  0-0-0-0-0-dCX&&&&X&!T.1!!!X!hsmXtCyX C   X c3hX XY,, (# 4,=@,QT, ,,,,\T#s(^$ $s)    ++++;#;;;;;;rr'ceZdZdZdS)rdzGeneric shader related errorN)rBrCrDrErNrrrdrds&&&&rrdc eZdZdZdedejfdZede e fdZ ede e e ffdZedejfdZd e deejejejejejffd Zd e d e de fd Zede fd ZedejfdZedefdZedee dffdZedefdZedefdZ edefdZ!de fdZ"dS)ReloadableProgramzN Programs we want to be reloadable must be created with this wrapper. rr:c"||_||_dS)z Create a shader using either a file path or a name. Args: meta: The program meta program: The program instance N)r:r)rrr:s rrzReloadableProgram.__init__s  rrc2|jjp |jjSr)rr(r)r!s rrQzReloadableProgram.namesy~8!88rc|jjSr)r:_membersr!s rrzReloadableProgram._memberss |$$rc|jjSr)r:r r!s rr zReloadableProgram.ctxs |rkeyc|j|Sr)r:)rrs r __getitem__zReloadableProgram.__getitem__s|C  rdefaultc8|j||Sr)r:r)rrrs rrzReloadableProgram.gets|W---rc|jjSr)r:r<r!s rr<zReloadableProgram.extras |!!rc|jjS)zThe ModernGL Program object)r:mglor!s rrzReloadableProgram.mglos|  rc|jjS)zk int: The internal OpenGL object. This values is provided for debug purposes only. )r:glor!s rrzReloadableProgram.glos |r.c|jjS)z1 tuple: The subroutine uniforms. )r: subroutinesr!s rrzReloadableProgram.subroutiness |''rc|jjS)z int: The geometry input primitive. The GeometryShader's input primitive if the GeometryShader exists. The geometry input primitive will be used for validation. )r:geometry_inputr!s rrz ReloadableProgram.geometry_inputs|**rc|jjS)z int: The geometry output primitive. The GeometryShader's output primitive if the GeometryShader exists. )r:geometry_outputr!s rrz!ReloadableProgram.geometry_outputs |++rc|jjS)zc int: The maximum number of vertices that the geometry shader will output. )r:geometry_verticesr!s rrz#ReloadableProgram.geometry_verticess |--rc(d|jd|jdS)Nzrs_ 111111111111IIIIII ##+1!{I{I{I{I{I{I{I{I|x;x;x;x;x;x;x;x;v''''')'''\A\A\A\A\A\A\A\A\A\Ar