k j7*ddlmZmZmZddlZddlZddlmZddlZ ddl m Z ej dej dejdejdejdejd ejd ejd ejd ejd i ZGddZGddZGddeZdS))AnyOptionalUnionN)types TRIANGLES TRIANGLE_FANTRIANGLE_STRIPTRIANGLES_ADJACENCYTRIANGLE_STRIP_ADJACENCYPOINTSLINES LINE_STRIP LINE_LOOPLINES_ADJACENCYc eZdZdZgdfdejdedeedefdZ e de fd Z deede eed ffd Zd edefd ZdS) BufferInfoz/Container for a vbo with additional informationFbuffer buffer_format attributes per_instancec:||_tj||_||_||_|jj|jzdkr;td ||j|jj|jz|jj|jz|_ dS)z_ :param buffer: The vbo object :param format: The format of the buffer rz@Buffer with type {} has size not aligning with {}. Remainder: {}N) rrparse_attribute_formatsattrib_formatsrrsize vertex_sizeVAOErrorformatvertices)selfrrrrs U/home/agentuser/manim-venv/lib/python3.11/site-packages/moderngl_window/opengl/vao.py__init__zBufferInfo.__init__s #;MJJ$( ; d. .! 3 3RYY!4#3T[5EHX5X   (D,<< returnc>td|jDS)Nc3$K|] }|jV dSN) bytes_total).0fs r z)BufferInfo.vertex_size..8s$>>Q1=>>>>>>r")sumrrs r rzBufferInfo.vertex_size6s">>$*=>>>>>>r".cg}g}t|j|jD]u\}}||vr(||1||j||||vt|dkrdS|jdd ||j rdndg|RS)zBuild content tuple for the buffer Returns The first value is the moderngl buffer From the third to the end, it is the attributes of the classrNz{}{} z/i) ziprrappendpad_strrremovelenrjoinr)rrformatsattrs attrib_formatattribs r contentzBufferInfo.content:s %()r"r rrs99 !#" ====I =  ====4?S???X? $s) vs{9K0L    6'#'$''''''r"rc>eZdZdZdejfdedefdZe dej fdZ d d ej de ed edededdf dZ d!d dd ej dejde edededdf dZ d d ej dejde ed edededdfdZdeejejeefdedeeeefdejfdZ d"deejejeefdeddfdZd ej dejfdZd#deddfdZdede efdZdS)$VAOa Represents a vertex array object. This is a wrapper class over ``moderngl.VertexArray`` to make interactions with programs/shaders simpler. Named buffers are added corresponding with attribute names in a vertex shader. When rendering the VAO an internal ``moderngl.VertextArray`` is created automatically mapping the named buffers compatible with the supplied program. This program is cached internally. The shader program doesn't need to use all the buffers registered in this wrapper. When a subset is used only the used buffers are mapped and the appropriate padding is calculated when interleaved data is used. You are not required to use this class, but most methods in the system creating vertexbuffers will return this type. You can obtain a single ``moderngl.VertexBuffer`` instance by calling :py:meth:`VAO.instance` method if you prefer to work directly on moderngl instances. Example:: # Separate buffers vao = VAO(name="test", mode=moderngl.POINTS) vao.buffer(positions, '3f', ['in_position']) vao.buffer(velocities, '3f', ['in_velocities']) # Interleaved vao = VAO(name="test", mode=moderngl.POINTS) vao.buffer(interleaved_data, '3f 3f', ['in_position', 'in_velocities']) .. code:: glsl # GLSL vertex shader in attributes in vec3 in_position; in vec3 in_velocities; r/r;modec||_||_ t|jnG#t$r:t dtwxYwg|_d|_d|_ d|_ i|_ dS)aCreate and empty VAO with a name and default render mode. Example:: VAO(name="cube", mode=moderngl.TRIANGLES) Keyword Args: name (str): Optional name for debug purposes mode (int): Default draw mode z!Invalid draw mode. Options are {}Nr) r;rO DRAW_MODESKeyErrorrrvalues_buffers _index_buffer_index_element_size vertex_countvaos)rr;rOs r r!z VAO.__init__s   \ ty ! ! ! \ \ \>EEjFWFWFYFYZZ[[ [ \+- 8<26 57 s #AA'r#c(tjS)z-moderngl.Context: The active moderngl context)mglwctxr,s r r[zVAO.ctxsxzzr"Nrprogramrfirst instancesct||}||j}|||||dS)aRender the VAO. An internal ``moderngl.VertexBuffer`` with compatible buffer bindings is automatically created on the fly and cached internally. Args: program: The ``moderngl.Program`` Keyword Args: mode: Override the draw mode (``TRIANGLES`` etc) vertices (int): The number of vertices to transform first (int): The index of the first vertex to start with instances (int): The number of instances N)rr_r`)instancerOrender)rr^rOrr_r`vaos r rcz VAO.rendersA*mmG$$ <9D 4(%9 MMMMMr")r_rcountct||}||j}|||||dS)aA The render primitive (mode) must be the same as the input primitive of the GeometryShader. The draw commands are 5 integers: (count, instanceCount, firstIndex, baseVertex, baseInstance). Args: program: The ``moderngl.Program`` buffer: The ``moderngl.Buffer`` containing indirect draw commands Keyword Args: mode (int): By default :py:data:`TRIANGLES` will be used. count (int): The number of draws. first (int): The index of the first indirect draw command. N)rOrer_)rbrOrender_indirect)rr^rrOrer_rds r rgzVAO.render_indirectsD0mmG$$ <9D FU%HHHHHr"cv||}||j}||||||dS)aTransform vertices. Stores the output in a single buffer. Args: program: The ``moderngl.Program`` buffer: The ``moderngl.buffer`` to store the output Keyword Args: mode: Draw mode (for example ``moderngl.POINTS``) vertices (int): The number of vertices to transform first (int): The index of the first vertex to start with instances (int): The number of instances N)rOrr_r`)rbrO transform)rr^rrOrr_r`rds r riz VAO.transformsD(mmG$$ <9D f4(%S\ ]]]]]r"rattribute_namesct|ts|g}t|tjt jtfvr/td t|t|t jr,|j | }t|tr|j |}| }t|t|kr#td |||jt#||||jdj|_|S)atRegister a buffer/vbo for the VAO. This can be called multiple times. adding multiple buffers (interleaved or not). Args: buffer: The buffer data. Can be ``numpy.array``, ``moderngl.Buffer`` or ``bytes``. buffer_format (str): The format of the buffer. (eg. ``3f 3f`` for interleaved positions and normals). attribute_names: A list of attribute names this buffer should map to. Returns: The ``moderngl.Buffer`` instance object. This is handy when providing ``bytes`` and ``numpy.array``. zSbuffer parameter must be a moderngl.Buffer, numpy.ndarray or bytes instance(not {})dataz+Format '{}' does not describe attributes {}r\) isinstancerFtyperCrDnumpyndarraybytesrrr[rtobytessplitr4rTr1rrrW)rrrrjr6s r rz VAO.buffersM(/400 O <<F F F%vd6ll33  fem , , 7X__V^^%5%566F fe $ $ 2X__&_11F%%'' w<<3// / /=DD]Tcdd  Z OOPPP M"-6 r"index_element_sizect|tjtjt fvrt dt|tjr,|j | }t|t r|j |}||_ ||_ dS)zSet the index buffer for this VAO. Args: buffer: ``moderngl.Buffer``, ``numpy.array`` or ``bytes`` Keyword Args: index_element_size (int): Byte size of each element. 1, 2 or 4 zKbuffer parameter must be a moderngl.Buffer, numpy.ndarray or bytes instancerlN) rorCrDrprqrrrrnr[rrsrUrV)rrrvs r index_bufferzVAO.index_buffer&s <<F F F]  fem , , 7X__V^^%5%566F fe $ $ 2X__&_11F##5   r"c |j|j}|t|tjr|Sd|jD}|D]\tfd|j Ds:td |j ||d|j D]g}|j D].}| |}|r||/|r,td d|D|jr(|j|||j|j}n|j||}||j|j<|S)a Obtain the ``moderngl.VertexArray`` instance for the program. The instance is only created once and cached internally. Args: program (moderngl.Program): The program Returns: ``moderngl.VertexArray``: instance Ncg|];\}}t|tj|jd9|z VAO.instance..Ms[   d$ 233 =AI)r(r attrib_names r r*zVAO.instance..Vs1UUVv++K88UUUUUUr"z[VAO {} doesn't have attribute {} for program {}. Program attributes: {}. VAO attributes: {}c&g|]}|jD]}|SrLr=)r(buffr~s r rz VAO.instance..as'TTT$DOTTDTTTTr"z$Did not find a buffer mapping for {}cg|]}|SrLrL)r(ns r rz VAO.instance..ps>]>]>]Qq>]>]>]r")rXgetglornrC VertexArray_membersitemsr+rTrrr;r:r1rUr[ vertex_arrayrV)rr^rdprogram_attributes vao_contentrr:rs @r rbz VAO.instance>simmGK(( ?z#x/CDD?J  %.4466   .  KUUUUt}UUUUU -f #*TT$-TTT      m , ,Fnn%788G ,""7+++  6==>]>]J\>]>]>]^^    >(''"( CC(''==C!$ '+ r"Tc|jD]\}}|i|_|rC|jD]}|j|jr|jg|_dS)zDestroy all internally cached vaos and release all buffers. Keyword Args: buffers (bool): also release buffers N)rXrreleaserTrrU)rr_rdrs r rz VAO.releases ioo''  FAs KKMMMM  -  & & ##%%%%! -"**,,, r"c4|jD]}||jvr|cSdS)aGet the BufferInfo associated with a specific attribute name If no buffer is associated with the name `None` will be returned. Args: name (str): Name of the mapped attribute Returns: BufferInfo: BufferInfo instance N)rTr)rr;rs r get_buffer_by_namezVAO.get_buffer_by_names8m  Fv((( )tr")Nr\rr])Nr\)ru)T) r?r@rArBrCrrErIr!rHContextr[ProgramrrcrDrgrirnptNDArrayrrrrFrrxrrbrGrrrrLr"r rNrNYs##J$&83E88S8S88884X%X# NN!NsmN N  N  N NNNN@# IIII!IIsm I  II IIIIF#^^!^^sm ^  ^  ^^ ^^^^60hos{3'7>?00tCy#~. 0  0000fbc66HOS[-=uDE6[^6 66660A 0AX5IAAAAFdd(sx /Cr"rNceZdZdS)rN)r?r@rArLr"r rrsDr"r)typingrrrrCrp numpy.typingrmoderngl_windowrZmoderngl_window.openglrrrr r r r r rrrrQrrN ExceptionrrLr"r rsN'''''''''' ((((((   > -  "7 %'A OX NG    /  ='='='='='='='='@JJJJJJJJZ      y     r"