^ jQjdZddlmZddlmZddlZddlmZmZm Z m Z m Z m Z m Z mZddlmZmZmZmZmZerddlmZmZmZddlmZmZmZGd d ejjZejjd ejj d iZ!ejjd ejj diZ"ddZ#Gddej$j%Z&dS)zdAllows for a somewhat generic multitextured sprite. This sprite behaves just like regular sprites. ) annotations) TYPE_CHECKINGN)glActiveTexture glBindTextureglEnableGL_BLEND glBlendFunc glDisable glGetIntegervGLint) GL_SRC_ALPHA GL_TEXTURE0GL_ONE_MINUS_SRC_ALPHA GL_TRIANGLESGL_MAX_TEXTURE_IMAGE_UNITS)Texture AbstractImage Animation)BatchGroup ShaderProgramcPeZdZdZ ddfd ZddZddZddZddZddZ xZ S)MultiTextureSpriteGroupaShared Multi-texture Sprite rendering Group. The Group defines custom ``__eq__`` and ``__hash__`` methods, and so will be automatically coalesced with other MultiTexture Sprite Groups sharing the same parent Group, Textures, and blend parameters. Ntexturesdict[str, Texture] blend_srcint blend_destprogramShaderProgram | Noneparent Group | NonereturnNonect|||_||_||_||_dS)aCreate a sprite group for multiple textures and samplers. All textures must share the same target type. The group is created internally when a :py:class:`~pyglet.multitexture_sprite.MultiTextureSprite` is created; applications usually do not need to explicitly create it. Args: textures: A dictionary of textures, with the keys being the GLSL sampler name. blend_src: OpenGL blend source mode; for example, ``GL_SRC_ALPHA``. blend_dest: OpenGL blend destination mode; for example, ``GL_ONE_MINUS_SRC_ALPHA``. program: A custom ShaderProgram. parent: Optional parent group. )r!N)super__init__ _texturesrrr)selfrrrrr! __class__s b/home/agentuser/manim-venv/lib/python3.11/site-packages/pyglet/experimental/multitexture_sprite.pyr'z MultiTextureSpriteGroup.__init__s@( '''!"$ c|jt|jD]\}}||j|<t|jD]6\}}t t |zt|j|j 7ttt|j |jdS)zCalled before this group is drawn to setup the shader state. The shader program is activated than then all textures are bound and the blend mode is setup. N)ruse enumerater(valuesrrrtargetidrrr rr)r)idxnameitextures r+ set_statez!MultiTextureSpriteGroup.set_state4s "4>22 % %IC!$DL  #DN$9$9$;$;<< 6 6JAw K!O , , , '.'* 5 5 5 5DNDO44444r,ctt|jt t dS)zCalled after all draw calls for the group have been made. When done the blend mode is disabled, the shader program is deactivated, and all textures are deactivated. N)r rrstoprrr)s r+ unset_statez#MultiTextureSpriteGroup.unset_stateFs: (  $$$$$r,strch|jjdd|jDdS)N(cg|] \}}||f Sr@).0r4r6s r+ z4MultiTextureSpriteGroup.__repr__..Qs ,g,g,g Wd7^,g,g,gr,))r*__name__r(itemsr:s r+__repr__z MultiTextureSpriteGroup.__repr__Ps=.)jj,g,gPTP^PdPdPfPf,g,g,gjjjjr,other(object | Group | MultiTextureSpriteGroupboolc|j|jurdSt|jt|jkrdS|jD]J\}}||jvrdS|j|j|jks|j|j|jkrdSK|j|juo|j|jko|j|jkS)zDetermines if this group is the same as the other group. The method is broken out like this to make it easier to follow and read. F) r*lenr(rEr2r1rrr)r)rGr4r6s r+__eq__zMultiTextureSpriteGroup.__eq__Ss ?$. 0 05 t~  #eo"6"6 6 65!^1133  MD'5?**uuzU_T25555?[_K`Kg9g9guu:h  -4%/145#33 5r,ct|j|j|j|jft d|jDzt d|jDzS)Ncg|] }|j Sr@)r2rAr6s r+rBz4MultiTextureSpriteGroup.__hash__..ns S S S S S Sr,cg|] }|j Sr@)r1rOs r+rBz4MultiTextureSpriteGroup.__hash__..os W W WG W W Wr,)hashr!rrrtupler(r0r:s r+__hash__z MultiTextureSpriteGroup.__hash__jsT[$,>?, S S4>;P;P;R;R S S STTU W Wt~?T?T?V?V W W WXX YZZ Zr,)NN) rrrrrrrr r!r"r#r$r#r$)r#r<)rGrHr#rI)r#r) rD __module__ __qualname____doc__r'r7r;rFrLrS __classcell__r*s@r+rrsOS45555$%%%%kkkk5555.ZZZZZZZZr,r sampler2Dsampler2DArrayz.xyimagesrr#rclt}tt|t||jksJd|jddd|D}dd|D}dd|D}d|d|d |d }dd |D}dd |D}dd |D}d|d|d|d}tj j |df|dfS)aCreates the default multi-texture shader based on the dict of textures passed in. The default shader program will 'overlay' each texture layer on top of each other taking into account of the alpha channel of each texture. Textures can be either normal 2D textures or 2D texture arrays. The maximum number of textures you can layer is determined by the maximum number of samplers you can have in a fragment shader. zOnly z Texture Units are available. cg|]}d|d S)in vec3 _coords;r@rAr4s r+rBz*_get_default_mt_shader..s$SSST8$888SSSr,cg|]}d|d S)z out vec3 _coords_frag;r@rcs r+rBz*_get_default_mt_shader..s$ZZZD ?D ? ? ?ZZZr,cg|] }|d|d S)z_coords_frag = rbr@rcs r+rBz*_get_default_mt_shader..s*'i'i'iSW4(N(N(N(N(N'i'i'ir,z #version 150 core in vec3 translate; in vec4 colors; in vec2 scale; in vec3 position; in float rotation; z# out vec4 vertex_colors; a uniform WindowBlock { mat4 projection; mat4 view; } window; mat4 m_scale = mat4(1.0); mat4 m_rotation = mat4(1.0); mat4 m_translate = mat4(1.0); void main() { m_scale[0][0] = scale.x; m_scale[1][1] = scale.y; m_translate[3][0] = translate.x; m_translate[3][1] = translate.y; m_translate[3][2] = translate.z; m_rotation[0][0] = cos(-radians(rotation)); m_rotation[0][1] = sin(-radians(rotation)); m_rotation[1][0] = -sin(-radians(rotation)); m_rotation[1][1] = cos(-radians(rotation)); gl_Position = window.projection * window.view * m_translate * m_rotation * m_scale * vec4(position, 1.0); vertex_colors = colors; z } cg|]}d|d S)rarer@rcs r+rBz*_get_default_mt_shader..s$XXX$=$===XXXr,cFg|]\}}dt|jd|dS)zuniform  ;)_SAMPLER_TYPESr1rAr4texs r+rBz*_get_default_mt_shader..s:!q!q!qV^VZ[^"Q^CJ-G"Q"Q$"Q"Q"Q!q!q!qr,c Lg|]!\}}d|d|dt|jd"S)z color = layer(texture(z, _coords_fragz ), color);)_SAMPLER_COORDSr1rls r+rBz*_get_default_mt_shader..sc ^ ^ ^BKBFHK |4 | |4 | |UdeheoUp | | | ^ ^ ^r,z7 #version 150 core in vec4 vertex_colors; z" out vec4 final_colors; z vec4 layer(vec4 foreground, vec4 background) { return foreground * foreground.a + background * (1.0 - foreground.a); } void main() { vec4 color = vec4(0.0, 0.0, 0.0, 1.0); z9 final_colors = color * vertex_colors; } vertexfragment) r r rrKvaluejoinkeysrEpygletglcurrent_contextcreate_program) r]max_tex in_tex_coordsout_tex_coordstex_coords_assignments vertex_sourceuniform_samplerstex_operationsfragment_sources r+_get_default_mt_shaderrsggG,g666 v;;'- ' ' ')])])])] ' ' 'IISSV[[]]SSSTTMYYZZFKKMMZZZ[[N!YY'i'i[a[f[f[h[h'i'i'ijj'''''J K'''MRIIXX&++--XXXYYMyy!q!qbhbnbnbpbp!q!q!qrrYY ^ ^OUO[O[O]O] ^ ^ ^__N  !O* 9 $ 3 3]H4MP_akOl m mmr,c 2eZdZdZeZdddeeddddf d3dZd4fd Z d5dZ d4dZ d6dZ d4d Z d7d$Zd8d%Zd9d'Zd:d*Zed4d+Zejd;d-Zed4d.Zejdd2ZxZS)?MultiTextureSpriteaCreates a multi-textured sprite. Multi-textured sprites behave just like regular sprites except they can contain multiple texture layers. The default behavior is to overlay each layer on top of each other. Each texture layer can be either a static image or an animation. If the default behavior is not desired then a custom shader program can be supplied overriding the default program. The following complete example loads a 2 layer sprite and draws it to the screen:: import pyglet batch = pyglet.graphics.Batch() logo_image = pyglet.image.load('logo.png') kitten_image = pyglet.image.load('kitten.png') sprite = pyglet.experimental.MultiTextureSprite({'kitten': kitten_image, 'logo': logo_image}, x=50, y=50, batch=batch) window = pyglet.window.Window() @window.event def on_draw(): batch.draw() pyglet.app.run() If a custom program is provided then several assumptions are made by the MultiTextureSprite class. * The vertex shader is expected to have each texture coordinates passed in with variables named ``_coords`` as vec3 types where ```` is the name of the layer given to the constructor. * The fragment shader is expected to have samplers of the appropriate type with variables named ```` where ```` is the name of the layer given to the constructor. rNFr]$dict[str, AbstractImage | Animation]xfloatyzrrrbatch Batch | Nonegroupr"subpixelrIrr r#r$c 6i|_d|_|D]\} } t| tjjr-| jdj|j| <n| |j| <|jr@|j| j |j| j z|jj |jj zkr|j| |_| st|j|_ n| |_ ||_ ||_||_i|_|D]\} } t| tjjrg| d| jdjd|j| <| jdjr7tj|j|j| d| ||_||_||_||_||_| |_|dS)aCreate a Sprite instance. If the texture layers are different sizes then the largest texture by area is picked for size of the sprite and then the other layers are scaled to that size. Args: images: A dict object with the key being the name of the texture and the value is either an Animation or AbstractImage. Currently, each item must be of the same size. x: X coordinate of the sprite. y: Y coordinate of the sprite. z: Z coordinate of the sprite. blend_src: OpenGL blend source mode. The default is suitable for compositing sprites drawn from back-to-front. blend_dest: OpenGL blend destination mode. The default is suitable for compositing sprites drawn from back-to-front. batch: Optional batch to add the sprite to. group: Optional parent group of the sprite. subpixel: Allow floating-point coordinates for the sprite. By default, coordinates are restricted to integer values. program: A specific shader program to initialize the sprite with. The default multi-texture overlay shader will be used if one is not provided. Nr animation frame_idxnext_dtr)r(_texturerE isinstancervimagerframes get_textureheightwidthr_program_x_y_z _animationsdurationclock schedule_once_animate_batch _blend_src _blend_dest _user_groupget_sprite_group_group _subpixel_create_vertex_list) r)r]rrrrrrrrrr4imgs r+r'zMultiTextureSprite.__init__sX  5 5ID##v|566 9'*z!}':'F'F'H'Ht$$'*'8'8t$= 5T^D%9%@4>RVCWC]%]bfbobvy}zGzMcM%N%N $t 4  $24>BBDMM#DM g gID##v|566 g9Nt>TU^>_aefff #% ++-- !   """""r,c|jr$tj|jt dS)zForce immediate removal of the sprite from video memory. This method removes any scheduled animation calls and then calls pyglet.sprite.Sprite.delete method. N)rrvr unschedulerr&delete)r)r*s r+rzMultiTextureSprite.deleteTsB   3 L # #DM 2 2 2 r,rcf||j|j|j|j|jS)zCreates and returns a group to be used to render the sprite. This is used internally to create a consolidated group for rendering. ) group_classr(rrrrr:s r+rz#MultiTextureSprite.get_sprite_group_s/ AQSWS`bfbrsssr,cR||jvr|j|}|dxxdz cc<|dt|djkr#d|d<|d|jdS|dj|d}|j|j|d|z z }t td||j}||d<|||j tj |j|d|dSdSdS)Nrrron_animation_endr)rrKrdispatch_event _vertex_listrminmax_set_multi_texturerrrvrrr)r)dtkeyrframers r+rzMultiTextureSprite._animatefs< $" " "(-I k " " "a ' " " "%Y{-C-J)K)KKK)* +&##$6777$,Fk*1)K2HIE~) >Yy-AB-FGs1h//@@'/ )$''U[-D-D-F-FGGG **4=)I:NPSTTTTT! # "*)r,new_texrc|j|j|jurp|j|_||j|<|j||_|dS||j|<|j|jt|j|ddd<dS)N_coords) r2r(copyrrrrr tex_coordsgetattr)r)rrs r+rz%MultiTextureSprite._set_multi_textureys :T^C03 3 3 "^0022DN")DN3    $ $ & & &//11DK  $ $ & & & & &")DN3 <@N3$' 'tr,rAbstractImage | Animationc||jvr>tj|j|j|d}t |tjjr|d|j dj d|j|<|j dj r>|j s7tj |j|j|d||j dj }n| }|r]||||j|jz|jj|jjzkr||_|dSdSdS)aSets the layer to the new image or animation. This method has no effect if name is not a valid layer. Note: if you want to swap out a layer which is an animation then this will cause all other animated layers for this sprite to pause until the swap is done. Args: name: The dict key given for the texture layer in the constructor img: The Image or Animation to set Nrrr)rrvrrrpoprrrrr_pausedrrrrrr_update_position)r)r4rrms r+ set_layerzMultiTextureSprite.set_layersh 4# # # L # #DM 2 2 2    & & & c6<1 2 2 $36QSVS]^_S`Si%j%jD T "z!}% cdl c **4=$:J4:PQZ:[]abbb*Q-%1133CC//##C  (  # #D# . . . CJ&4=+>AU+UVV # %%'''''  ( (WVr,c td)Nz[MultiTextureSprite does not support the frame_index property. Use get_frame_index instead.NotImplementedErrorr:s r+ frame_indexzMultiTextureSprite.frame_index!"AA Ar,indexc td)Nz[MultiTextureSprite does not support the frame_index property. Use set_frame_index instead.r)r)rs r+rzMultiTextureSprite.frame_indexrr,c td)NzOMultiTextureSprite does not support the image property. Use get_layer instead.rr:s r+rzMultiTextureSprite.image!"stttr,c td)NzOMultiTextureSprite does not support the image property. Use set_layer instead.r)r)rs r+rzMultiTextureSprite.imagerr,c|jS)N)rr:s r+pausedzMultiTextureSprite.pauseds |r,pausect|dr ||jkrdS||_|r&tj|jdS|jD]f\}}|dj|d}|j rA|j |d<tj |j|j|d|gdS)Nrrrr) hasattrrrvrrrrrErrr)r)rr4rrs r+rzMultiTextureSprite.pausedst]++ u /D/D F  g L # #DM 2 2 2 2 2$(#3#9#9#;#; g gi!+.5i 6LM>g+0>Ii(L..t}d>Nt>TU^>_aefff  g gr,)r]rrrrrrrrrrrrrrr"rrIrr r#r$rT)r#r)rrr#r$)r4r<rrr#r$)r4r<r#r)r4r<r#r)r4r<rrr#r$)rrr#r$)rrr#r$)r#rI)rrIr#r$)rDrUrVrWrrr rr'rrrrrrrrrpropertyrsetterrrrXrYs@r+rrs&&N*K!".#9'+'+"'15T#T#T#T#T#l      ttttUUUU& [ [ [ ["IIII$        $($($($(LAAAXAAAAAuuuXu \uuu\uX ]ggg]gggggr,r)r]rr#r)'rW __future__rtypingrrv pyglet.glrrrrr r r r r rrrr pyglet.imagerrrpyglet.graphicsrrrgraphicsrrw GL_TEXTURE_2DGL_TEXTURE_2D_ARRAYrkrprspriteSpriterr@r,r+rs#""""" vvvvvvvvvvvvvvvvvvvvqqqqqqqqqqqqqq<>>>>>>>>>>;;;;;;;;;;]Z]Z]Z]Z]Zfo3]Z]Z]ZD I[ I!#3 IU I!2 RnRnRnRnj}g}g}g}g}g-}g}g}g}g}gr,