k jJdZddlZddlmZGddeZGddZGd d ZGd d ZdS) aU Simple row based texture atlas created for fast runtime allocation. * This atlas is partly based on the texture atlas in the Arcade project * The allocator is based on Pyglet's row allocator https://github.com/pyglet/pyglet/blob/master/pyglet/image/atlas.py https://github.com/pythonarcade/arcade/blob/development/arcade/texture_atlas.py N) BaseImageceZdZdS)AllocatorExceptionN)__name__ __module__ __qualname__]/home/agentuser/manim-venv/lib/python3.11/site-packages/moderngl_window/atlas/simple_atlas.pyrrsDr rcVeZdZdZdZdededdfdZded edeeeffd Zd d Z dS) _Rowz% A row in the texture atlas. )xyy2 max_heightrrreturnNc>d|_||_||_||_dSNr)rrrr)selfrrs r __init__z _Row.__init__s"$r widthheightc|dks|dkr#td||||jkrtd|j|j}}|xj|z c_t |j|z|j|_||fS)z/Add a region to the row and return the positionrzCannot allocate size: [{}, {}]z#Cannot allocate past the max height)rformatrrrmaxr)rrrrrs r addz_Row.add!s A::1$%E%L%LUTZ%[%[\\ \ DO # #$%JKK Kvtv1 %dfvotw//!t r c.|j|jz |_dS)z Compacts the row to the smallest height. Should only be done once when the row is filled before adding a new row. N)rrrrs r compactz _Row.compact.s 'DF*r rN) rrr __doc__ __slots__intrtuplerr r r r rrs/I#34  c eCHo    ++++++r rcFeZdZdZdedefdZdededeeeffdZdS) AllocatorzRow based allocatorrrcX||_||_td|jg|_dSr)rrrrowsrrrs r rzAllocator.__init__9s,  !T[))* r rc|jD]8}|j|jz |kr#|j|kr|||cS9|j|kry|j|jz |krf|t|j|j|jz }|j ||||Std |||)z Allocate a region. Returns: tuple[int, int]: The x,y location Raises: AllocatorException: if no more space z$No more space in {} for box [{}, {}]) r)rrrrrrr rappendrr)rrrrownew_rows r alloczAllocator.alloc@s9 . .CzCE!U**s~/G/Gwwuf----- :  4;#76#A#A KKMMM364;#788G I  W % % %;;uf-- - !G!N!NtUZ\b!c!cdddr N)rrr r"r$rr%r/r r r r'r'6sv+c+3++++e3eec3heeeeeer r'c \eZdZdZ ddejdededed ed ef d Ze d ejfd Z e d ej fdZ e d efdZ e d efdZe d eeeffdZe d eeeffdZded dfdZded dfdZdeded dfdZddZdS) TextureAtlasz A simple texture atlas using a row based allocation. There are more efficient ways to pack textures, but this is normally sufficient for dynamic atlases were textures are added on the fly runtime. rTctxrr componentsborder auto_resizecT||_||_||_||_||_||_|jjd|_|j|j |j|_ |j |j g|_ t|||_dS)NGL_MAX_VIEWPORT_DIMS)r4)color_attachments)_ctx_width_height _components_border _auto_resizeinfo _max_sizetexturesize_texture framebuffer_fbor' _allocator)rr3rrr4r5r6s r rzTextureAtlas.__init__ds   % '+/).9O*P ))$)@P)QQ I))T]O)LL #E622r rc|jS)z)The moderngl contex this atlas belongs to)r:rs r r3zTextureAtlas.ctxs yr c|jS)z,The moderngl texture with the atlas contents)rDrs r textruezTextureAtlas.textrues }r c|jS)z!int: Width of the atlas in pixels)r;rs r rzTextureAtlas.widths {r c|jS)z"int: Height of the atlas in pixels)r<rs r rzTextureAtlas.heights |r c|j|jfS)z5tuple[int, int]: The size of he atlas (width, height))r;r<rs r rCzTextureAtlas.sizes{DL((r c|jS)zP tuple[int,int]: The maximum size of the atlas in pixels (x, y) )rArs r max_sizezTextureAtlas.max_sizes ~r imageNcdSNr rrPs r rzTextureAtlas.add r cdSrRr rSs r removezTextureAtlas.removerTr cdSrRr r*s r resizezTextureAtlas.resizerTr cdSrRr rs r rebuildzTextureAtlas.rebuildrTr )r2rTr!)rrr r"modernglContextr$boolrpropertyr3TexturerJrrr%rCrOrrrVrXrZr r r r1r1\s 33  33 3  3  333336X%X)XsXX)eCHo)))X)%S/X   t     I $     C            r r1) r"r[baser Exceptionrrr'r1r r r rbs        ++++++++D#e#e#e#e#e#e#e#eLM M M M M M M M M M r