k j'dZddlmZddlZddlmZmZmZddlZddl Z ddl Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZmZmZmZmZmZmZejeZer dd lm Z mZm!Z!mZGd dZ"dS)z- Wrapper for a loaded scene with properties. ) annotationsN) TYPE_CHECKINGAnyOptional)geometry)ProgramDescription)programs)Material)Node)ColorLightProgramFallbackProgram MeshProgramTextureLightProgramTextureProgramTextureVertexColorProgramVertexColorProgram)Camerar Meshr ceZdZdZd2dZed3d Zed4d Zejd5dZ d6d7dZ d8d9dZ d:d;dZ dd%Z d?d'Zd>d(Zd@dAd*Zd@dBd,Zd>d-Zd>d.ZdCd0ZdCd1ZdS)DScenez Generic scenename Optional[str]kwargsrc ||_g|_g|_g|_g|_g|_t j|_t j|_ d|_ tj |_ |jj i|j_|jjd|_|js;t%jt)d|_|j|jjd<|jjd|_|js;t%jt)d|_|j|jjd<t j|_dS)zmCreate a scene with a name. Args: name (str): Unique name or path for the scene ?NDEFAULT_BBOX_PROGRAMzscene_default/bbox.glsl)pathDEFAULT_WIREFRAME_PROGRAMzscene_default/wireframe.glsl)r root_nodesnodes materialsmeshescamerasglmvec3bbox_minbbox_max diagonal_sizerbboxbbox_vaoctxextraget bbox_programr loadrwireframe_programmat4_matrix)selfrrs V/home/agentuser/manim-venv/lib/python3.11/site-packages/moderngl_window/scene/scene.py__init__zScene.__init__&sC  &("$ )+"$ %' "%(** "%(**    8> !DHN!HN../EFF  G ( "(ABBB!!D 6:5FDHN1 2"&!3!34O!P!P% Q%-]"(FGGG&&D ";?:PDHN6 7xzz returnmoderngl.Contextc(tjS)z%moderngl.Context: The current context)mglwr,r4s r5r,z Scene.ctxPsxzzr7glm.mat4c|jS)zQglm.mat4x4: The current model matrix This property is settable. )r3r<s r5matrixz Scene.matrixUs |r7r?Nonec\||_|jD]}||jdSN)r3r calc_model_mat)r4r?nodes r5r?z Scene.matrix]s= O . .D    - - - - . .r7projection_matrixOptional[glm.mat4] camera_matrixtimefloatc|jD]}|||||jdddS)zDraw all the nodes in the scene. Args: projection_matrix (ndarray): projection matrix (bytes) camera_matrix (ndarray): camera_matrix (bytes) time (float): The current time )rFrHrIrN)r drawr,clear_samplers)r4rFrHrIrDs r5rMz Scene.drawcs^O  D II"3+      1%%%%%r7NT?rPrPchildrencolortuple[float, float, float]c.|}|}|jd||jd|j|jd||jd|j|jd|j||jd_|j|j|sdS|jD]$}| |||j|j%dS)a.Draw scene and mesh bounding boxes. Args: projection_matrix (glm.mat4): mat4 projection camera_matrix (glm.mat4): mat4 camera matrix children (bool): Will draw bounding boxes for meshes as well color (tuple): Color of the bounding boxes m_projm_modelm_cambb_minbb_maxrRN) r/writer3r'r(valuer+renderr draw_bbox)r4rFrHrQrRrDs r5r]zScene.draw_bboxys.%  (#))*;<<< )$**4<888 '"((777 (#))$-888 (#))$-888+0'"( T.///  FO _ _D NN,mT=NPTP] ^ ^ ^ ^ _ _r7rPrPrPr!tuple[float, float, float, float]ch|}|}|jd||jd|j|jd|||jd<d|j_|jD]}||||jd|j_dS)a(Render the scene in wireframe mode. Args: projection_matrix (ndarray): mat4 projection camera_matrix (ndarray): mat4 camera matrix children (bool): Will draw bounding boxes for meshes as well color (tuple): Color of the wireframes rUrVrWrRTFN)r1rZr3r, wireframer draw_wireframe)r4rFrHrRrDs r5rbzScene.draw_wireframes.%  x(../@AAA y)// === w'--m<<<*/w'"O Z ZD    1=$BX Y Y Y Y"r7 mesh_programsOptional[list[MeshProgram]]clearboolc d|r|jD] }d|_ |s|jjd}|s_t t ttttg}||jjd<|jD]}|D]f}| |}|Mt|tr ||_n0tdt!|g|js t"d|jdS)zApplies mesh programs to meshes. If not mesh programs are passed in we assign default ones. Args: mesh_programs (list): List of mesh programs to assign clear (bool): Clear all assigned mesh programs NDEFAULT_PROGRAMSz2apply() must return a MeshProgram instance, not {}z'WARING: No mesh program applied to '%s')r# mesh_programr,r-r.rrrrr rapply isinstancer ValueErrorformattypeloggerwarningr)r4rcremesh mesh_proginstances r5apply_mesh_programszScene.apply_mesh_programss^  )  ) )$(!! C HN../ABBM C'))"$$&((-//%''#%% ! 6C12K U UD*   $??400'!(K88,5)(PWW $X ($ UH$)TTT U Ur7cd}d}|jD]-}|tj||\}}.||_||_tj|j|jz |_dS)zCalculate scene bboxN)r calc_global_bboxr%r2r'r(lengthr))r4r'r(rDs r5calc_scene_bboxzScene.calc_scene_bboxsr$($(O W WD!%!6!6sxzz8X!V!V Hhh     Z  (EFFr7glm.vec3c6|j|j|jz dz zS)z6Calculate the center of the scene using bounding boxesg@)r'r(r<s r5 get_centerzScene.get_centers}  =DDDr7c^|tj|_dS)zprepare the scene for rendering. Calls ``apply_mesh_programs()`` assigning default meshprograms if needed and sets the model matrix. N)rtr%r2r?r<s r5preparez Scene.prepares'   """hjj r7Optional[Node]c8|jD]}|j|kr|cSdS)zFinds a :py:class:`~moderngl_window.scene.Node` Keyword Args: name (str): Case sensitive name Returns: A :py:class:`~moderngl_window.scene.Node` or ``None`` if not found. N)r!r)r4rrDs r5 find_nodezScene.find_nodes7J  DyD   !tr7Optional[Material]c8|jD]}|j|kr|cSdS)zFinds a :py:class:`~moderngl_window.scene.Material` Keyword Args: name (str): Case sensitive material name Returns: A :py:class:`~moderngl_window.scene.Material` or ``None`` N)r"r)r4rmats r5 find_materialzScene.find_materials7>  Cx4  tr7c.|dSz*Destroys the scene data and vertex buffersN)destroyr<s r5releasez Scene.releases r7c|jD]"}|j|j#|jD]}|g|_g|_dSr)r#vaorr"r )r4rqrs r5rz Scene.destroy"sdK # #Dx#  """>  C KKMMMM r7strc6d|jS)Nz )rmrr<s r5__str__z Scene.__str__0s##DI...r7c t|SrB)rr<s r5__repr__zScene.__repr__3s4yyr7)rrrr)r8r9)r8r=)r?r=r8r@)rE)rFrGrHrGrIrJr8r@)NNTrO) rFrGrHrGrQrJrRrSr8r@)NNr^)rFrGrHrGrRr_r8r@)NT)rcrdrerfr8r@)r8r@)r8ryrB)rrr8r~)rrr8r)r8r)__name__ __module__ __qualname____doc__r6propertyr,r?setterrMr]rbrtrxr{r}rrrrrrr7r5rr#s("("("("TXX ]...]. &&&&&015,0,> _ _ _ _ _H15,03J #####>PT,U,U,U,U,U\ G G G GEEEE!!!!              ////r7r)#r __future__rloggingtypingrrrr%modernglmoderngl_windowr;rmoderngl_window.metar"moderngl_window.resources.programsr materialr rDr r rrrrrr getLoggerrromoderngl_window.scenerrrrr7r5rs#"""""////////// $$$$$$333333777777  8 $ $CBBBBBBBBBBBBQQQQQQQQQQr7