k jxddlmZddlmZmZmZddlZddlZddlm Z ddl m Z erddl m Z Gdd ZdS) ) annotations) TYPE_CHECKINGAnyOptionalN)VAO)Material) MeshProgramceZdZdZdddejejfd-dZ d.d/dZd0dZd1dZ d2d#Z d3d'Z d4d)Z d5d6d,Z dS)7MeshzMesh info and geometryNnamestrvao Optional[VAO]materialOptional[Material] attributesOptional[dict[str, Any]]bbox_minglm.vec3bbox_maxreturnNonecl||_||_||_|pi|_||_||_d|_dS)a;Initialize mesh. Args: name (str): name of the mesh Keyword Args: vao (VAO): geometry material (Material): material for the mesh attributes (dict): Details info about each mesh attribute (dict) bbox_min: xyz min values bbox_max: xyz max values Attributes example:: { "NORMAL": {"name": "in_normal", "components": 3, "type": GL_FLOAT}, "POSITION": {"name": "in_position", "components": 3, "type": GL_FLOAT} } N)r rrrrr mesh_program)selfr rrrrrs U/home/agentuser/manim-venv/lib/python3.11/site-packages/moderngl_window/scene/mesh.py__init__z Mesh.__init__sA6   $*    59projection_matrixglm.mat4 model_matrix camera_matrixtimefloatcV|j!|j|||||dSdS)zDraw the mesh using the assigned mesh program Keyword Args: projection_matrix (bytes): projection_matrix view_matrix (bytes): view_matrix camera_matrix (bytes): camera_matrix N)r!r#r$r%)rdraw)rr!r#r$r%s rr(z Mesh.draw5sN   (   " ""3)+ #      ) (r proj_matrix cam_matrixprogrammoderngl.Programrc|d||d||d||d|j|d|j||dS)a4Renders the bounding box for this mesh. Args: proj_matrix: Projection matrix model_matrix: View/model matrix cam_matrix: Camera matrix program: The moderngl.Program rendering the bounding box vao: The vao mesh for the bounding box m_projm_modelm_cambb_minbb_maxN)writeto_bytesrrrender)rr)r#r*r+rs r draw_bboxzMesh.draw_bboxLs"  4 4 6 6777   !6!6!8!8999z2244555 6 6 8 8999 6 6 8 8999 7rc|j Jd|d||d||j|dS)zRender the mesh as wireframe. proj_matrix: Projection matrix model_matrix: View/model matrix program: The moderngl.Program rendering the wireframe Nz(Can not draw the wireframe, vao is emptyr.r/)rr3r4r5)rr)r#r+s rdraw_wireframezMesh.draw_wireframedsx##%O### 4 4 6 6777   !6!6!8!8999      r attr_type componentsintc ||d|j|<dS)z Add metadata about the mesh :param attr_type: POSITION, NORMAL etc :param name: The attribute name used in the program :param components: Number of floats )r r:Nr)rr9r r:s r add_attributezMesh.add_attributers/3*%M%M """r view_matrixglm.vec3 | Nonetuple[glm.vec3, glm.vec3]cbtj|jd}tj|jd}||z}||z}t dD].}||||z dkr||||c||<||</||Jtj|j|j|jtj|j|j|jfSt dD]!}t||||||<"t dD]!}t||||||<"||fS)zCalculates the global bounding. Args: view_matrix: View matrix bbox_min: xyz min bbox_max: xyz max Returns: bbox_min, bbox_max: Combined bbox g?r) glmvec4rrrangevec3xyzminmax) rr?rrbb1bb2bminbmaxis rcalc_global_bboxzMesh.calc_global_bbox{s9ht}c**ht}c**S S q 4 4AAwa 1$$#'7DG Qa  x/HTVTVTV44chtvtvtv6V6VW Wq 4 4Ahqk4733HQKKq 4 4Ahqk4733HQKK!!rboolcd|jvS)zJ Returns: bool: Does the mesh have a normals? NORMALr=)rs r has_normalszMesh.has_normalss 4?**rrlayerc:d||jvS)zT Returns: bool: Does the mesh have texture coordinates? z TEXCOORD_{})formatr)rrWs rhas_uvsz Mesh.has_uvss ##E**do==r)r rrrrrrrrrrrrr)r ) r!r"r#r"r$r"r%r&rr) r)r"r#r"r*r"r+r,rrrr)r)r"r#r"r+r,rr)r9rr rr:r;rr)r?r"rr@rr@rrA)rrS)r)rWr;rrS)__name__ __module__ __qualname____doc__rDrGrr(r6r8r>rRrVrZrrr r s  "'+/3%SXZZ%SXZZ!:!:!:!:!:P .0 ! ! ! !NNNN""""""""H++++>>>>>>>rr ) __future__rtypingrrrrDmodernglmoderngl_window.opengl.vaorrr programsr r r_rrres""""""////////// ******&%%%%%%\>\>\>\>\>\>\>\>\>\>r