l jBUddlmZddlZddlZddlZddlZddlmZmZm Z ddl m Z ddl m Z mZmZmZddlZddlZddlm Ze r1ddlmZedegdfZded <edgdfZded <eezZded <dd lmZmZd dl m!Z!d dl"m#Z#e e$j%dz Z&ia'ded<iZ(ded<gdZ)d'dZ*d(dZ+Gd dZ,Gd!d"e,Z-Gd#d$Z.Gd%d&e-Z/dS))) annotationsN)CallableIteratorSequence)Path) TYPE_CHECKINGAnySelf TypeAlias)OpenGLRendererObject3Dr MeshTimeBasedUpdaterMeshNonTimeBasedUpdater MeshUpdater)MatrixMNPoint3D)config)openglshaderszdict[str, moderngl.Program]shader_program_cachezdict[Path, str]file_path_to_code_map)r MeshShaderFullScreenQuad file_pathrreturnstrcd|tvr t|S|}tjd|tj}|D]W}|d}t |j|z }||d|}X|t|<|S)Nz%^#include (?P.*\.glsl)$)flags include_pathr) r read_textrefinditer MULTILINEgroupget_shader_code_from_fileparentreplace)rsource include_linesmatchr! included_codes P/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/renderer/shader.pyr'r''s)))$Y//  " "FK0lM ??{{>22 1  | +   A >>'-)$ Munfiltered_attributes npt.NDArray attributes Sequence[str]cg}t|jjD]L\}}||vrC|||j|jdj|j|jfMtj||jjdjd|}|jjD]}||vr ||||<|S)Nrdtype) enumerater6namesappendsubdtypershapenpzeros)r0r2filtered_attributes_dtypei dtype_namefiltered_attributess r.filter_attributesrB:s!#"#8#>#DEE :  # # % , ,)/2;A>B)/28   (39?BCI!L' ,17PP  # #.CJ.O  + r/ceZdZdd Zd>d Zd?dZddd@dZdAdZ dBdZ dCdZ dDdZ dEdZ dFdZdFd ZdGd!ZdHdId$ZdJd&ZdKd'ZdLd)Z dMdNd0Z dOdPd2Z dOdQd4ZdRd5ZdSd6ZdTd8ZdSd9ZdUdVd:ZdSd;Zd*S)Wr childrenctjd|_tjd|_g|_d|_|j||dS)N)r<eye model_matrix normal_matrixrDr(add init_updaters)selfrDs r.__init__zObject3D.__init__VsWF1IIVAYY(* '+ ( r/startendalphafloat_r rNoneczd|z |jz||jzz|_d|z |jz||jzz|_dS)N)rHrI)rLrNrOrPrRs r. interpolatezObject3D.interpolate_sPY%*<zz -2244!/4466 r/c i}|g}|ru|d}||j|}|||<|j$||ur ||j||u||SNr)popextendrDr[r(rJ)rL node_to_copybfsnode node_copys r.rYz Object3D.copyks f 9771::D JJt} % % %((**I!*L {&4t+;+;T[)--i888 9D!!r/c|D]}|jtd|j|ddi|j||D] }||_ dS)Nz=Attempt to add child that's already added to another Object3Dcurrent_children_onlyF)r( ExceptionremoverDr_)rLrDchilds r.rJz Object3D.add{s  E|'S(  X;U;;; X&&&  EELL  r/T)rereboolc|rD]}|j|krtdttfd|j|_D] }d|_ dS)Nz9Attempt to remove child that isn't added to this Object3Dc |vSrX)rhrDs r.z!Object3D.remove..s %x2Gr/)r(rflistfilterrD)rLrerDrhs ` r.rgzObject3D.removes  !  <4''#S(V$G$G$G$GWWXX   EELL  r/rc4|jdddfddSNrHrLs r. get_positionzObject3D.get_positions! A&rr**r/positionr c2||jdddfdd<|Srqrs)rLrvs r. set_positionzObject3D.set_positions%&.!!!Q$# r/Iterator[Mesh]c#K|g}|rK|}t|tr|V||j|IdSdSrX)r^ isinstancerr_rDrLdfsr(s r. get_mesheszObject3D.get_meshessof (WWYYF&$''  JJv ' ' '  ( ( ( ( (r/Iterator[Object3D]c#K|g}|r6|}|V||j|4dSdSrX)r^r_rDr|s r. get_familyzObject3D.get_familys]f (WWYYFLLL JJv ' ' ' ( ( ( ( (r/cdSrXrl)rLrRs r.align_data_and_familyzObject3D.align_data_and_familys r/rc |j|jS|jg}|}|j-||jj|j}|j-tjt t|SrX)r(rHr9r<linalg multi_dotrnreversed)rLmodel_matricescurrent_objects r.hierarchical_model_matrixz"Object3D.hierarchical_model_matrixs ; $ $+,#/  ! !."7"D E E E+2N#/y""4(@(@#A#ABBBr/c<|j|jddddfS|jg}|}|j-||jj|j}|j-tjtt|ddddfSrq) r(rIr9rHr<rrrnr)rLnormal_matricesrs r.hierarchical_normal_matrixz#Object3D.hierarchical_normal_matrixs ; %bqb"1"f- --.#/  " ">#8#E F F F+2N#/y""4(A(A#B#BCCBQBFKKr/c>g|_g|_d|_d|_dS)NF)time_based_updatersnon_time_updaters has_updatersupdating_suspendedrts r.rKzObject3D.init_updaterss(?A @B!"'r/rdtc||jr|jr|S|jD]}||||jD] }|||SrX)rrrr)rLrtime_based_updaternon_time_based_updaters r.updatezObject3D.updatesq  D$; K"&": ) )   tR ( ( ( (&*&< ) ) " " "4 ( ( ( ( r/list[MeshTimeBasedUpdater]c|jSrX)rrts r.get_time_based_updatersz Object3D.get_time_based_updaterss ''r/c2t|jdkSr])lenrrts r.has_time_based_updaterzObject3D.has_time_based_updaters4+,,q00r/list[MeshUpdater]c |j|jzSrX)rrrts r. get_updaterszObject3D.get_updaterss'$*@@@r/Nupdate_functionrindex int | None call_updatercdtj|jvr|||n|||||r||S)Nr)inspect signature parameters_add_time_based_updater_add_non_time_updaterrefresh_has_updater_statusr)rLrrrs r. add_updaterzObject3D.add_updatersx 7$_55@ @ @  ( (% @ @ @ @  & & > > > '')))   KKMMM r/rcx||j|dS|j||dSrX)rr9insertrLrrs r.rz Object3D._add_time_based_updatersE =  $ + +O < < < < <  $ + +E? C C C C Cr/rcx||j|dS|j||dSrX)rr9rrs r.rzObject3D._add_non_time_updatersE =  " ) )/ : : : : :  " ) )% A A A A Ar/c||jvr#|j|||jv#||jvr#|j|||jv#||SrX)rrgrr)rLrs r.remove_updaterzObject3D.remove_updaters!999  $ + +O < < <!999!777  " ) )/ : : :!777 ''))) r/cJg|_g|_||SrX)rrrrts r.clear_updaterszObject3D.clear_updaterss)#% !# ''))) r/meshc||D]}|||SrX)rrr)rLrupdaters r.match_updaterszObject3D.match_updaterssK ((** & &G   W % % % % r/cd|_|SNT)rrts r.suspend_updatingzObject3D.suspend_updating s"& r/cDd|_|r|d|S)NFr)r)rr)rLrs r.resume_updatingzObject3D.resume_updatings+"'   KK1K    r/cZt|dk|_|Sr])rrrrts r.rz#Object3D.refresh_has_updater_statuss( 1 1 3 344q8 r/)rDr ) rNr rOr rPrQrRr rrS)rr )rDr rrS)rDr rerirrS)rr)rvrrr )rry)rr)rRr rrS)rrrrS)r)rrQrr )rr)rri)rrr)rrrrrrirr rX)rrrrrrS)rrrrrrS)rrrr )rr )rr rr )T)rrirr )__name__ __module__ __qualname__rMrVr[rYrJrgrurxr~rrrrrKrrrrrrrrrrrrrrlr/r.r r Us<<<<  """"     IM       ++++((((((((     C C C C L L L L(((( ((((1111AAAA !! "JNDDDDDMQBBBBB   r/cReZdZddddddejfdfd ZddZddZddZxZ S)rNTshader Shader | Noner2npt.NDArray | Nonegeometry Mesh | Nonematerialindicesuse_depth_testri primitiveintc4t||||_||_||_n3|"| ||_|j|_|j|_nt d||_||_d|_| dS)NzIMesh requires either attributes and a Shader or a Geometry and a MaterialF) superrMrr2rrfrr skip_renderrK) rLrr2rrrrr __class__s r.rMz Mesh.__init__s   *"8"(DK(DO"DLL  !h&:"DK&1DO#+DLL -"!& r/rcJt|j|j|j|jnd|j|j}|j|_|j|_|j |_ |S)N)r2rrrr) rr2rYrrrrrrHrIrZs r.r[zMesh.single_copy9s++--;+/<+CDL%%'''.n     + -2244!/4466 r/rendererr rSc|jdtj|j|jd|jj|jd|jjdS)Nu_model_matrix u_view_matrixu_projection_matrix)r set_uniformrmatrix_to_shader_inputrHcameraformatted_view_matrixprojection_matrix)rLrs r. set_uniformszMesh.set_uniformsGs~    )$*; < <    1VWWW  ! O -     r/c|jrdS|jr*|jjt jn)|jjt jg}|jjj D]4\}}t|t j r| |5t|j|}|jj|}|jd}nP|jd}|r |jj|}nd}|jjj|jj|g|jjRd|i}||j||||dSdS)Ni4 index_buffer)rrrcontextenablemoderngl DEPTH_TESTdisableshader_program_membersitemsr{ Attributer9rBr2buffertobytesrastypesimple_vertex_arrayr6r8renderrrelease) rLshader_attribute_names member_namememberfiltered_shader_attributesvertex_buffer_objectindex_buffer_objectvert_index_datavertex_array_objects r.rz Mesh.renderRs    F   = K  & &x': ; ; ; ; K  ' '(; < < <,.#';#=#F#L#L#N#N ; ; K&("455 ;&--k:::%6 O3& & " ${299 & . . 0 0   < "&  "l11$77??AAO +&*k&9&@&@&Q&Q##&*#Edk1E K &  ( - 3   -    ""4>222$$&&&##%%%  *  ' ' ) ) ) ) ) + *r/)rrr2rrrrrrrrrirr)rr)rr rrSr) rrrr TRIANGLESrMr[rr __classcell__rs@r.rrs!%)- $"&&*#!+:         &*&*&*&*&*&*&*&*r/rc$eZdZ dddZddZdS)rNrmoderngl.Contextname str | Noner*dict[str, Any] | NonecJ||_||_||_|jtvr8t|jj|jkrt|j|_n|j|jdi|j|_n{|jei}dddd}t|jz }|D]&}||z }t|} | |||j <'|jdi||_ntd|j$|jtvr|jt|j<dSdSdS)N vertex_shaderfragment_shadergeometry_shader)vertfraggeom.Must either pass shader name or shader source.rl) rrr*rctxrprogram SHADER_FOLDERiterdirr'stemrf) rLrrr* source_dictsource_dict_key shader_folder shader_fileshader_file_path shader_sources r.rMzShader.__init__|s\    I- - -$TY/3t|CC"6ty"AD   [ $"1'/"@"@DK"@"@D   Y "K'))O *DI5M,4466 T T #0;#> 9:J K K FS O,<,ABCC"1'/"@"@K"@"@D  LMM M 9 TY6J%J%J.2.A  + + + ! %J%Jr/rvaluer rrSc~tjt5||j|<ddddS#1swxYwYdSrX) contextlibsuppressKeyErrorr)rLrrs r.rzShader.set_uniforms   * * . .(-D  % . . . . . . . . . . . . . . . . . .s 266NN)rrrrr*r)rrrr rrS)rrrrMrrlr/r.rr{sO (, 'B'B'B'B'BR......r/rc2eZdZ d d fd Zd fd ZxZS)rNrrfragment_shader_sourcerfragment_shader_namec ||td|t|dz }t|}n(|&tj|}t |d|d}tjddtj dfg }tj td  td  d d gtd  td d d gtd td d d gtd  td  d d gtd td  d d gtd td d d gg|d<| dtj| dtjt!||dS)Nrz.fraga #version 330 in vec4 in_vert; uniform mat4 u_model_view_matrix; uniform mat4 u_projection_matrix; void main() {{ vec4 camera_space_vertex = u_model_view_matrix * in_vert; vec4 clip_space_vertex = u_projection_matrix * camera_space_vertex; gl_Position = clip_space_vertex; }} )rr)r*in_vert)rFr5frame_x_radiusframe_y_radiusrrUu_model_view_matrixr)rfr r'textwrapdedentlstriprr<r=float32arrayrrr view_matrixorthographic_projection_matrixrrM)rLrrrrrr2rs r.rMzFullScreenQuad.__init__s " ).B.JLMM M  +,2F/M/M/MM %>?O%P%P " " # /%-_5K5R5R5T5T%U%U "  "$:     "XaBJ(E'FGGG ")**V4D-E,Eq!L)**F3C,DaK()62B+CQJ)**V4D-E,Eq!L()F3C,D+DaK()62B+CQJ   ! ! 9 0&2D2F2FGGG !  1 3 3    ,,,,,r/rrScHtdSrX)rr)rLrs r.rzFullScreenQuad.renders r/r)rrrrrrr)rrrrMrrrs@r.rrsi.2+/ 1-1-1-1-1-1-1-fr/r)rrrr)r0r1r2r3rr1)0 __future__rrrr#r"collections.abcrrrpathlibrtypingrr r r rnumpyr< numpy.typingnptmanim.renderer.opengl_rendererr rQr__annotations__rr manim.typingrrrutilsr__file__r(r rr__all__r'rBr rrrrlr/r.r8s""""""" 8888888888666666666666L======&. E/BD/H&IIIII)1:,2D)EEEEE47KKKKKKK********X% 1 466666)+++++   &6CCCCCCCCL]*]*]*]*]*8]*]*]*@,.,.,.,.,.,.,.,.^55555T55555r/