^ jL`ddlmZddlZddlZddlmZmZmZmZddlTe edoej Z dZ dZ dZd Zd Zd ZGd d ejZGddejZeddS)) annotationsN)clockeventgraphicsimage)*is_pyglet_doc_runa#version 150 in vec3 position; in vec4 size; in vec2 scale; in vec4 color; in vec4 texture_uv; in float rotation; out vec4 geo_size; out vec2 geo_scale; out vec4 geo_color; out vec4 geo_tex_coords; out float geo_rotation; void main() { gl_Position = vec4(position, 1); geo_size = size; geo_scale = scale; geo_color = color; geo_tex_coords = texture_uv; geo_rotation = rotation; } a #version 150 // We are taking single points from the vertex shader // and emitting 4 new vertices to create a quad/sprite. layout (points) in; layout (triangle_strip, max_vertices = 4) out; uniform WindowBlock { mat4 projection; mat4 view; } window; // Since geometry shader can take multiple values from a vertex // shader we need to define the inputs from it as arrays. // For our purposes, we just take single values (points). in vec4 geo_size[]; in vec2 geo_scale[]; in vec4 geo_color[]; in vec4 geo_tex_coords[]; in float geo_rotation[]; out vec2 uv; out vec4 frag_color; void main() { // Unpack the image size and anchor vec2 size = geo_size[0].xy; vec2 anchor = geo_size[0].zw; // Grab the position value from the vertex shader vec3 center = gl_in[0].gl_Position.xyz; // This matrix controls the actual position of the sprite mat4 m_translate = mat4(1.0); m_translate[3][0] = center.x; m_translate[3][1] = center.y; m_translate[3][2] = center.z; mat4 m_rotation = mat4(1.0); m_rotation[0][0] = cos(radians(-geo_rotation[0])); m_rotation[0][1] = sin(radians(-geo_rotation[0])); m_rotation[1][0] = -sin(radians(-geo_rotation[0])); m_rotation[1][1] = cos(radians(-geo_rotation[0])); mat4 m_scale = mat4(1.0); m_scale[0][0] = geo_scale[0].x; m_scale[1][1] = geo_scale[0].y; // Final UV coords (left, bottom, right, top): float uv_l = geo_tex_coords[0].s; float uv_b = geo_tex_coords[0].t; float uv_r = geo_tex_coords[0].p; float uv_t = geo_tex_coords[0].q; // Emit a triangle strip to create a quad (4 vertices). // Prepare and reuse the transformation matrix and fragment color: mat4 m_pv = window.projection * window.view * m_translate * m_rotation * m_scale; frag_color = geo_color[0]; // Upper left gl_Position = m_pv * vec4(vec2(0.0, size.y) - anchor, 0.0, 1.0); uv = vec2(uv_l, uv_t); EmitVertex(); // lower left gl_Position = m_pv * vec4(vec2(0.0, 0.0) - anchor, 0.0, 1.0); uv = vec2(uv_l, uv_b); EmitVertex(); // upper right gl_Position = m_pv * vec4(vec2(size.x, size.y) - anchor, 0.0, 1.0); uv = vec2(uv_r, uv_t); EmitVertex(); // lower right gl_Position = m_pv * vec4(vec2(size.x, 0.0) - anchor, 0.0, 1.0); uv = vec2(uv_r, uv_b); EmitVertex(); // We are done with this triangle strip now EndPrimitive(); } z#version 150 in vec2 uv; in vec4 frag_color; out vec4 final_color; uniform sampler2D sprite_texture; void main() { final_color = texture(sprite_texture, uv) * frag_color; } z#version 150 core in vec2 uv; in vec4 frag_color; out vec4 final_colors; uniform sampler2DArray sprite_texture; void main() { final_colors = texture(sprite_texture, uv) * frag_color; } cxtjjtdft dft dfSNvertexgeometryfragment)pygletglcurrent_contextcreate_program vertex_sourcegeometry_sourcefragment_source_/home/agentuser/manim-venv/lib/python3.11/site-packages/pyglet/experimental/geoshader_sprite.pyget_default_shaderrs9 9 $ 3 3]H4M5Dj4Q5Dj4Q S SSrcxtjjtdft dft dfSr )rrrrrrfragment_array_sourcerrrget_default_array_shaderrs9 9 $ 3 3]H4M5Dj4Q5JJ4W Y YYrcBeZdZdZd fd ZdZdZdZdZdZ xZ S) SpriteGroupzShared sprite rendering group. The group is automatically coalesced with other sprite groups sharing the same parent group, texture and blend parameters. Nct|||_||_||_||_dS)aiCreate a sprite group. The group is created internally when a :py:class:`~pyglet.sprite.Sprite` is created; applications usually do not need to explicitly create it. :Parameters: `texture` : `~pyglet.image.Texture` The (top-level) texture containing the sprite image. `blend_src` : int OpenGL blend source mode; for example, ``GL_SRC_ALPHA``. `blend_dest` : int OpenGL blend destination mode; for example, ``GL_ONE_MINUS_SRC_ALPHA``. `program` : `~pyglet.graphics.shader.ShaderProgram` A custom ShaderProgram. `order` : int Change the order to render above or below other Groups. `parent` : `~pyglet.graphics.Group` Optional parent group. )parentN)super__init__texture blend_src blend_destprogram)selfr#r$r%r&r __class__s rr"zSpriteGroup.__init__s@, ''' "$ rc|jttt |jj|jjttt|j |j dSN) r&useglActiveTexture GL_TEXTURE0 glBindTexturer#targetidglEnableGL_BLEND glBlendFuncr$r%r's r set_statezSpriteGroup.set_statese  $$$dl)414  -4 u|+4 #u}';;4 5=#33 4 %/1 4 5#33  5rc|t|j|j|jj|jj|j|jfSr*)hashr&r r#r0r/r$r%r4s r__hash__zSpriteGroup.__hash__s8T\4;\_dl&9^T_677 7rr*) r= __module__ __qualname____doc__r"r5r9r>rArD __classcell__)r(s@rrrs 8555<<<5557777777rrc eZdZdZdZdZdZdZdZdZ dZ dZ dZ e Zdddeeddddf dZdZedZejd Zd Zd Zed Zejd ZedZejdZedZejdZdZedZejdZedZejdZedZejdZedZejdZedZ e jdZ edZ!e!jdZ!edZ"e"jd Z"ed!Z#e#jd"Z#d5d#Z$ed$Z%e%jd%Z%ed&Z&e&jd'Z&ed(Z'e'jd)Z'ed*Z(e(jd+Z(ed,Z)e)jd-Z)ed.Z*e*jd/Z*ed0Z+e+jd1Z+d2Z,d3Z-e.rd4Z/dSdS)6SpriteNrFg?Tc ||_||_||_||_gd|_t |t jrn||_|j dj |_ |j dj |_ |j rtj|j|j n| |_ | sBt |t jrt%|_nt)|_n| |_|pt+j|_||_||j |||j||_| |_|dS)atCreate a sprite. :Parameters: `img` : `~pyglet.image.AbstractImage` or `~pyglet.image.Animation` Image or animation to display. `x` : int X coordinate of the sprite. `y` : int Y coordinate of the sprite. `z` : int Z coordinate of the sprite. `blend_src` : int OpenGL blend source mode. The default is suitable for compositing sprites drawn from back-to-front. `blend_dest` : int OpenGL blend destination mode. The default is suitable for compositing sprites drawn from back-to-front. `batch` : `~pyglet.graphics.Batch` Optional batch to add the sprite to. `group` : `~pyglet.graphics.Group` Optional parent group of the sprite. `subpixel` : bool Allow floating-point coordinates for the sprite. By default, coordinates are restricted to integer values. `program` : `~pyglet.graphics.shader.ShaderProgram` A custom shader to use. This shader program must contain the exact same attribute names and types as the default shader. The class methods and properties depend on this, and will crash otherwise. )rLrLrLrN)_x_y_z_img_rgba isinstancer Animation _animationframes get_texture_textureduration_next_dtr schedule_once_animateTextureArrayRegionr_programrrget_default_batch_batch _user_group group_classr&_group _subpixel_create_vertex_list) r'imgxyzr$r%batchgroupsubpixelr&s rr"zSprite.__init__sID ))) c5? + + .!DOJqM/;;==DMJqM2DM} B#DM4=AAAOO--DM $#u788 5 8 : : 2 4 4 #DM;x9;;  &&t}iT\[`aa !   """""rc (|j}|jdt|j|jd|j|j|jffd|j |j |j |j ffd|j |jffd|jfd|jfd|jff |_dS)NfBn)positionsizescalecolor texture_uvrotation)rWr& vertex_list GL_POINTSr_rbrMrNrOwidthheightanchor_xanchor_y_scale_x_scale_yrQuv _rotation _vertex_listr'r#s rrdzSprite._create_vertex_list1s- L44 y$+t{DGTWdg67 w~w7GIYZ[ 67$WZ(DN,-5//rc|jSr*)r]r4s rr&zSprite.program<s }rc\|j|krdS||j|jj|jj||j|_|jr.|j|j t|j|rdS|j | dSr*) r]rarWrbr$r%r`r_ update_shaderrrwdeleterd)r'r&s rr&zSprite.program@s =G # # F&&t}'+{'<'+{'='.'+'7 99 K  ))$*;Y U\]]  F   """   """""rc|jrtj|j|jd|_d|_d|_dS)zForce immediate removal of the sprite from video memory. This is often necessary when using batches, as the Python garbage collector will not necessarily call the finalizer as soon as the sprite is garbage. N)rTr unscheduler[rrrWrbr4s rrz Sprite.deleteRsP ? ,  T] + + +   """   rc"|xjdz c_|jt|jjkr%d|_|d|jdS|jj|j}||j|j X|j |j |z z }ttd||j }tj|j|||_ dS|ddS)Nrmron_animation_end) _frame_indexlenrTrUdispatch_eventr _set_texturerrVrXrYminmaxrrZr[)r'dtframerXs rr[zSprite._animate`s Q  DO$: ; ; ; ; !D     2 3 3 3 (&t'89 %+1133444 > %~);$>%DN,4D  &qqq )  DK"O  #DM"O  #DM"O  b)-t})DdkTXTaFa)aD  #AAA & & & b brc|jjt|jzt|jz}|jr|nt |S)z[Scaled width of the sprite. Invariant under rotation. :type: int )rWrxabsr|rrcint)r'ws rrxz Sprite.widthsC M #dm"4"4 4s4;7G7G GN.qqA.rcX||jjt|jzz |_dSr*)rWrxrrr)r'rxs rrxz Sprite.widths& 3c$+6F6F FG rc|jjt|jzt|jz}|jr|nt |S)z\Scaled height of the sprite. Invariant under rotation. :type: int )rWryrr}rrcr)r'hs rryz Sprite.heightsC M 3t}#5#5 5DK8H8H HN.qqA.rcX||jjt|jzz |_dSr*)rWryrrr)r'rys rryz Sprite.heights&!5DK8H8H!HI rc|jdS)aBlend opacity. This property sets the alpha component of the colour of the sprite's vertices. With the default blend mode (see the constructor), this allows the sprite to be drawn with fractional opacity, blending with the background. An opacity of 255 (the default) has no effect. An opacity of 128 will make the sprite appear translucent. :type: int rQr4s ropacityzSprite.opacitysz!}rcF||jd<|j|jjdd<dSNr)rQrrs)r'rs rrzSprite.opacitys) 1 %)Z"""rc |jddS)aTBlend color. This property sets the color of the sprite's vertices. This allows the sprite to be drawn with a color tint. The color is specified as an RGB tuple of integers '(red, green, blue)'. Each color component must be in the range 0 (dark) to 255 (saturated). :type: (int, int, int) Nrrr4s rrsz Sprite.colorsz"1"~rcttt||jdd<|j|jjdd<dSr)listmaprrQrrs)r'rgbs rrsz Sprite.colors=c#smm,, 2A2%)Z"""rc|jS)z?True if the sprite will be drawn. :type: bool )_visibler4s rvisiblezSprite.visibles }rcR||_|sdn |jj|jjdd<dS)N)rrrr)rrWr~rrt)r'rs rrzSprite.visibles1 >E*[,,4=K[$QQQ'''rc|jS)zPause/resume the Sprite's Animation If `Sprite.image` is an Animation, you can pause or resume the animation by setting this property to True or False. If not an Animation, this has no effect. :type: bool )_pausedr4s rpausedz Sprite.pauseds |rct|dr ||jkrdS|durtj|jnI|jj|j}|j|_ |j rtj |j|j ||_dS)NrTT) hasattrrrrr[rTrUrrXrYrZ)r'pausers rrz Sprite.pausedst\** et|.C.C F D==  T] + + + +O*4+<=E!NDM} B#DM4=AAA rc|jS)zThe current Animation frame. If the `Sprite.image` is an `Animation`, you can query or set the current frame. If not an Animation, this will always be 0. :type: int )rr4s r frame_indexzSprite.frame_indexs   rc |jdStdt|t|jjdz |_dS)Nrrm)rTrrrrUr)r'indexs rrzSprite.frame_indexsE ? " F3uc$/2H.I.IA.M#N#NOOrc|j|jt|jdS)zDraw the sprite at its current position. See the module documentation for hints on drawing multiple sprites efficiently. N)rbset_state_recursiverdrawrwunset_state_recursiver4s rrz Sprite.drawsI ''))) y))) ))+++++rc\ |j|jdSdS#YdSxYwr*)rrr4s r__del__zSprite.__del__ sC  ,!((*****-,  DDs &+cdS)a The sprite animation reached the final frame. The event is triggered only if the sprite has an animation, not an image. For looping animations, the event is triggered each time the animation loops. :event: Nrr4s rrzSprite.on_animation_endsr)NNNNNNN)0r=rErFr_rTrrrrr|r}rrrra GL_SRC_ALPHAGL_ONE_MINUS_SRC_ALPHAr"rdpropertyr&setterrr[rirjrrrprfrgrhrurrrrrrxryrrsrrrrr_is_pyglet_doc_runrrrrrJrJsP FJLGI FHHHLKQ!'4J4%?#?#?#?#B / / /X ^##^#"   444&  X  \ ' '\ '""X" \TT\TX \ 1 1\ 1     ) )X )_11_1XX<<X<XX<<X<XX<<X<X_77_7X \RR\RX ^XX^XX ^XX^X8b8b8b8bt//X/ \HH\H//X/ ]JJ]J  X  ^00^0  X  \00\0X ^\\^\  X  ]  ]  ! !X !PPP ,,,        rrJr) __future__rsysrrrrr pyglet.glrr rrrrrrrGrouprEventDispatcherrJregister_event_typerrrrs7"""""" 000000000000WS"566P3;P 0Tl  SSS YYY ?7?7?7?7?7(.?7?7?7DvvvvvU "vvvr-.....r