l j0UddlmZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z ddlZddlZddlm Ze rddlmZdgZejdZd ZddZejZded<ejeZded<GddZiZ ded<ddZ!ddZ"dS) ) annotationsN)MappingSequence)Path) TYPE_CHECKINGSelf TypeAlias)FloatRGBLike_Array ShaderWrappermanimc:ttjdz S)Nshaders)r__file__parentX/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/renderer/shader_wrapper.pyget_shader_dirrs >> 9 ,,r file_namer directories list[Path]returncrSfd|D}|D]7}|r|cSt|d8td)Nc3"K|] }|z V dSNr).0 directoryrs r zfind_file..$s(II i)+IIIIIIrz does not exist.z not Found)existsloggerdebugOSError)rrpossible_pathspaths` r find_filer% sIIII[IIIN44 ;;== 4KKK LLD222 3 3 3 3 Y*** + ++rr _ShaderDType _ShaderDatac~eZdZddddddejfd&dZdZd'dZd(dZd)dZ dZ d*dZ dZ dZ dZd+d!Zd,d%ZdS)-r NF vert_datar' vert_indicesSequence[int] | None shader_folderPath | str | Noneuniforms+dict[str, float | tuple[float, ...]] | None texture_pathsMapping[str, Path | str] | None depth_testboolrender_primitive int | strc||_||_|jj|_t |pd|_|pi|_|pi|_||_ t||_ | | dS)N)r)r*dtypenamesvert_attributesrr,r.r0r2strr4init_program_code refresh_id)selfr)r*r,r.r0r2r4s r__init__zShaderWrapper.__init__2s'02>7@7L#' (;#<#<>Fn" 7D7J *%()9%:%:     rc4tj|}tj|j|_|jtj|j|_|jrt |j|_|jrt |j|_|Sr)copynparrayr)r*r.dictr0)r>results rrAzShaderWrapper.copyIs48DN33   *"$(4+<"="=F  = 2"4=11FO   <#'(:#;#;F  rrcht|jdu|jddu|jddugS)N vertex_shaderfragment_shader)allr) program_coder>s ris_validzShaderWrapper.is_validTsDd*!/2$>!"34D@    rr;c|jSr)idrKs rget_idzShaderWrapper.get_id]s wrintc|jSr) program_idrKs rget_program_idzShaderWrapper.get_program_id`s rc dtt|j|j|j|j|jgS)N|)joinmapr;rRr.r0r2r4rKs r create_idzShaderWrapper.create_idcsHxx OM&O)      rNonecj||_||_dSr)create_program_idrRrXrNrKs rr=zShaderWrapper.refresh_idrs*#5577~~''rc`tdfddDS)Nr7c3<K|]}j|dpdVdS)_shaderr7NrJ)rnamer>s rrz2ShaderWrapper.create_program_id..xsM!T"2"2"239rr)vertexgeometryfragment)hashrVrKs`rr[zShaderWrapper.create_program_idvsL GG>     rc`d fd }|d|d|dd _dS) Nr`r;r str | Nonec8tj|dz S)Nz.glsl)get_shader_code_from_filer,)r`r>s rget_codez1ShaderWrapper.init_program_code..get_codes&,"^^^3 rvertgeomfrag)rGgeometry_shaderrH)r`r;rrfr_)r>ris` rr<zShaderWrapper.init_program_code~s_       &Xf--'x//'x//4 4 rc|jSrr_rKs rget_program_codezShaderWrapper.get_program_codes   roldnewc|j}|D] \}}|rtj|||||<!|dSr)rJitemsresubr=)r>rprqcode_mapr`codes r replace_codezShaderWrapper.replace_codes]$"..** 8 8JD$ 8!#S$!7!7 rshader_wrappers'ShaderWrapper'rct|dkr|S|jt|j}|jg}|jg}|D]P}||j|z||j|t|jz }Qt j||_t j||_n+t j|jgd|D|_|S)Nrc3$K|] }|jV dSr)r))rsws rrz-ShaderWrapper.combine_with..s$"J"JB2<"J"J"J"J"J"Jr)lenr*r)appendrBhstack)r>ry num_verts indices_list data_listr}s r combine_withzShaderWrapper.combine_withs   1 $ $K   (DN++I -.L(I% / /##BOi$?@@@  ...S... " , 7 7D Yy11DNNYK"J"J/"J"J"JKDN r)r)r'r*r+r,r-r.r/r0r1r2r3r4r5)rr3)rr;)rrP)rrY)rpr;rqr;rrY)ryrzrr)__name__ __module__ __qualname__modernglTRIANGLE_STRIPr?rArLrOrSrXr=r[r<rorxrrrrr r 1s "&-1+/@D9= &.&=.           ((((       !!!rrDfilename_to_code_mapfilenamerfc|tvr t|S t|ttdg}n#t$rYdSwxYw|}t jd|t j}|D]R}ttdz | ddz }|dS| ||}S|t|<|S)N/)rz^#include ../include/.*\.glsl$)flagsincludez#include ../include/r7) rr%rrr" read_textrtfindall MULTILINErhreplace)rfilepathrE insertionsline inserted_codes rrhrhs'''#H-- '))4995    tt   ! !F )lJ 551 FFY .Db!I!I I    44m44%+" Ms,A AArgb_listr r;cldd|D}dt|d|dS)N,c3*K|]}dj|VdS)zvec3({}, {}, {})N)format)rrgbs rrz$get_colormap_code..s.HH-&-s3HHHHHHrzvec3[z]())rVr~)rdatas rget_colormap_codersA 88HHxHHH H HD +3x== + +D + + ++r)rrrrrr)rrrrf)rr rr;)# __future__rrAloggingrtcollections.abcrrpathlibrtypingrrr rnumpyrB numpy.typingnpt manim.typingr __all__ getLoggerr rr%voidr&__annotations__NDArrayr'r rrhrrrrrs"""""""  --------11111111110//////    7 # #--- , , , ,' !!!!\2 2222uuuuuuuur D,,,,,,r