k jzpdZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z Gd d Z dS) z1 Wrapper for a loaded mesh / vao with properties ) annotations)OptionalN)VAO)Camera)MeshceZdZdZ d7d8d Zed9d Zejd:dZed;dZejddZ ed?dZ e jd@dZ ed?dZ e jd@dZ edAdZ dBd!Z dCdDd'ZdEd,ZdFd-ZdGd3ZdHd5ZdId6ZdS)JNodezA generic scene node containing a mesh or camera and/or a container for other nodes. Nodes and their children represents the scene tree. Nname Optional[str]cameraOptional[Camera]meshOptional[Mesh]matrixOptional[glm.mat4]c~||_||_||_||_t jd|_g|_dS)zCreate a node. Keyword Args: name: Name of the node camera: Camera to store in the node mesh: Mesh to store in the node matrix: The node's matrix g?N)_name_camera_mesh_matrixglmmat4_matrix_global _children)selfr r rrs U/home/agentuser/manim-venv/lib/python3.11/site-packages/moderngl_window/scene/node.py__init__z Node.__init__s;    !hsmm')returnc|jS)zstr: Get or set the node namerrs rr z Node.name0 zrvaluestrNonec||_dSNr"rr%s rr z Node.name5  rc|jS)z<:py:class:`~moderngl_window.scene.Mesh`: The mesh if presentrr#s rrz Node.mesh9r$rrc||_dSr)r-r*s rrz Node.mesh>r+rc|jS)z@:py:class:`~moderngl_window.scene.Camera`: The camera if presentrr#s rr z Node.cameraB |rrc||_dSr)r0r*s rr z Node.cameraG  rc|jS)zglm.mat4x4: Note matrix (local)rr#s rrz Node.matrixKr1rglm.mat4c||_dSr)r5r*s rrz Node.matrixPr3rc|jS)zNglm.matx4: The global node matrix containing transformations from parent nodesrr#s r matrix_globalzNode.matrix_globalTs ""rc||_dSr)r9r*s rr:zNode.matrix_globalYs#r list['Node']c|jS)zlist: List of children)rr#s rchildrenz Node.children]s ~rnode'Node'c:|j|dS)z`Add a child to this node Args: node (Node): Node to add as a child N)rappend)rr?s r add_childzNode.add_childbs d#####rprojection_matrix camera_matrixtimefloatc|jr#|j||j|||jD]}||||dS)zDraw node and children. Keyword Args: projection_matrix: projection matrix camera_matrix: camera_matrix time: The current time )rE model_matrixrFrG)rErFrGN)rdrawrr)rrErFrGchilds rrKz Node.drawjs :  JOO"3!0+    ^  E JJ"3+       rprogrammoderngl.Programvaorc|jrL| Jd|j Jd| Jd|j||j||||jD]}|||||dS)a0Draw bounding box around the node and children. Keyword Args: projection_matrix: projection matrix camera_matrix: camera_matrix program (moderngl.Program): The program to render the bbox vao: The vertex array representing the bounding box N1Can not draw bbox, the projection matrix is empty-Can not draw bbox, the global matrix is emptyz-Can not draw bbox, the camera matrix is empty)rr draw_bboxr>)rrErFrMrOrLs rrSzNode.draw_bboxs : !--B.--&224c222 ,,.],,, J !4#6 wPS   ] L LE OO-}gs K K K K L Lrc|jr>| Jd|j Jd|j||j||jD]}|||j|dS)zRender the node as wireframe. Keyword Args: projection_matrix (bytes): projection matrix camera_matrix (bytes): camera_matrix program (moderngl.Program): The program to render wireframe NrQrR)rrdraw_wireframer>)rrErFrMrLs rrUzNode.draw_wireframes : W!--B.--&224c222 J % %&79Lg V V V] R RE  !2D4G Q Q Q Q R Rr view_matrixbbox_minglm.vec3 | Nonebbox_maxtuple[glm.vec3, glm.vec3]c|j |j|z}|jr|j|||\}}|jD]}||||\}}||fS)zRecursive calculation of scene bbox. Keyword Args: view_matrix (numpy.ndarray): view matrix bbox_min: min bbox values bbox_max: max bbox values )rrcalc_global_bboxr)rrVrWrYrLs rr\zNode.calc_global_bboxs~ < #,4K : ^!%!D  : :$$T%89999 : :#0D  4 4$$]3333 4 4rc6d|jS)Nz)formatr r#s r__repr__z Node.__repr__s&&ty111r)NNNN)r r r rrrrr)r r )r%r&r r')r r)r%rr r')r r)r%rr r')r r)r%r6r r')r r<)r?r@r r')rD)rEr6rFr6rGrHr r') rErrFrrMrNrOrr r')rErrFrrMrNr r')rVr6rWrXrYrXr rZ)r]r6r r')r r&)__name__ __module__ __qualname____doc__rpropertyr setterrr rr:r>rCrKrSrUr\r_rbrrr r sO##'#%) *****2X [[X [[X ]]X ]]###X#$$$$X$$$$ 8LLLL8RRRR.""""*4444"222222rr )rf __future__rtypingrrmodernglmoderngl_window.opengl.vaorr rrrr rirrrns#""""" ******O2O2O2O2O2O2O2O2O2O2r