^ j0LdZddlmZGddeZGddZdS)aJMemory allocation algorithm for vertex arrays and buffers. The region allocator is used to allocate vertex indices within a vertex domain's multiple buffers. ("Buffer" refers to any abstract buffer presented by :py:mod:`pyglet.graphics.vertexbuffer`. The allocator will at times request more space from the buffers. The current policy is to double the buffer size when there is not enough room to fulfil an allocation. The buffer is never resized smaller. The allocator maintains references to free space only; it is the caller's responsibility to maintain the allocated regions. ) annotationsceZdZdZddZdS) AllocatorMemoryExceptiona"The buffer is not large enough to fulfil an allocation. Raised by `Allocator` methods when the operation failed due to lack of buffer space. The buffer should be increased to at least requested_capacity and then the operation retried (guaranteed to pass second time). requested_capacityintreturnNonec||_dS)z&Requested capacity failed to allocate.N)r)selfrs U/home/agentuser/manim-venv/lib/python3.11/site-packages/pyglet/graphics/allocation.py__init__z!AllocatorMemoryException.__init__1s"4N)rrrr )__name__ __module__ __qualname____doc__r rr rr(s2555555rrceZdZUdZded<ded<dZdd Zdd Zdd Zd dZ d!dZ d"dZ d#dZ d#dZ d$dZd$dZd%dZd%dZdS)& Allocatorz'Buffer space allocation implementation.z list[int]sizesstartscapacityrrrrrr c0||_g|_g|_dS)zHCreate an allocator for a buffer of the specified maximum capacity size.Nr)r rs r r zAllocator.__init__=s  $  rsizec.||jksJ||_dS)zQResize the maximum buffer size. The capacity cannot be reduced. N)r)r rs r set_capacityzAllocator.set_capacityTs" dm#### rc\|dksJ|dkrdS|jsP||jkr6|jd|j|dSt ||jd|kr8|jdd|jd|dS|jd|jdz}t t|jdd|jddD]i\}\}}||z }||kr2|j|xx||zz cc<|j|dz=|j|dz=|cS||kr|j|xx|z cc<|cS||z}j|j|z }||kr|jdxx|z cc<|St |j|z|z )aAllocate memory in the buffer. Raises `AllocatorMemoryException` if the allocation cannot be fulfilled. Args: size: Size of region to allocate. Returns: Starting index of the allocated region. rN)rrappendrrinsert enumeratezip)r r free_starti alloc_start alloc_size free_sizes r alloczAllocator.alloc\sqyyyy 1991{ 1t}$$ ""1%%% !!$'''q*400 0 ;q>D K  q! $ $ $ J  a & & &1[^djm3 ,5c$+abb/4:VWVXVX>6Z6Z,[,[ 2 2 (A( Z $j0ID   1 Z!77 KA&Jq1u%!!!!T!! 1 % !!!!$z1JJMJ.    JrNNNd "NNN &t}t';i'GHHHrstartnew_sizec(|dkr|dksJ|dkr|dkr|||dS|dkr||S||kr|||z||z |Stt|j|jfD]\}\}}||z }|dkr |||z krn|dkr |||z ksVt tt|j|jt |||t ||||dkr |||z ks Jd|||z kr|t|jdz k}|s|j|dz||zz } n |j ||zz } | ||z kr@|s>|j|xx| |j|dzzz cc<|j|dz=|j|dz=|S| ||z kr|j|xx||z z cc<|S||} |||| S)a&Reallocate a region of the buffer. This is more efficient than separate `dealloc` and `alloc` calls, as the region can often be resized in-place. Raises `AllocatorMemoryException` if the allocation cannot be fulfilled. Args: start: Current starting index of the region. size: Current size of the region. new_size: int New size of the region. Returns: Starting index of the re-allocated region. rRegion not allocatedr) deallocr*r#r$rrprintlistlenr) r r+rr,r&r'r(pis_final_blockr)results r realloczAllocator.reallocs(qyyX]]]* q==qyy UD)))1 199::h'' ' d?? LL)4(? ; ; ;L-6cDK;T6U,V,V   (A( Z #AAvv$*q.00Q4:>11 $s4; 3344 5 5 5 %x ( ( ( ![* - - -Avv$*q.0002H000 :> ! !#dk"2"2Q"66N! ; KA.%$,?  MUT\: HtO++N+ 1 TZA->!>> KA&Jq1u% 8d?** 1 D0  H%% UD!!! rcj|dksJ|dkrdS|jsJtt|j|jfD]\}\}}||z }|dkr |||z krn|dkr |||z ks Jd|dkr||kr|j|=|j|=dS|dkr,|j|xx|z cc<|j|xx|zcc<dS|||z kr|j|xx|zcc<dS||j|<|j|dz||z|j|dz|||zz dS)zFree a region of the buffer. Args: start: Starting index of the region. size: Size of the region. rNr.r)rr#r$rr")r r+rr&r'r(r3s r r/zAllocator.deallocsqyyyy 199 F{{-6cDK;T6U,V,V   (A( Z #AAvv$*q.00Avv$*q.0002H000 66dj(( A 1 !VV KNNNd "NNN JqMMMT !MMMMM Z!^ # # JqMMMT !MMMMM&DJqM K  q1uedl 3 3 3 J  a!eZ1t8%< = = = = =rtuple[list, list]c|j|jfS)zGet a list of (aggregate) allocated regions. The result of this method is ``(starts, sizes)``, where ``starts`` is a list of starting indices of the regions and ``sizes`` their corresponding lengths. )rrr s r get_allocated_regionszAllocator.get_allocated_regions+s{DJ&&rc|jsdSd}|jd|jdz}tt|jdd|jddD]\}\}}|||z z }||z}|S)zGReturns the amount of space unused, not including the final free block.rrN)rrr#r$)r total_freer%r&r'r(s r get_fragmented_free_sizez"Allocator.get_fragmented_free_size4s{ 1 [^djm3 ,5c$+abb/4:VWVXVX>6Z6Z,[,[ 2 2 (A( Z + 2 2J$z1JJrc|js|jS|j|jd|jdzz }||zS)z"Return the amount of space unused.r )rrrr>)r free_ends r get_free_sizezAllocator.get_free_sizeBsH{ != =DKOdjn$DE,,..99rfloatcZd|t|jz z S)z0Return fraction of capacity currently allocated.g?)rArBrr:s r get_usagezAllocator.get_usageJs(D&&((5+?+????rc|}|dkrdS|t|z S)z5Return fraction of free space that is not expandable.rg)rAr>rB)r r)s r get_fragmentationzAllocator.get_fragmentationNsL&&(( >>2,,..t7I7I7K7K1L1LLLrstrc pdttt|j|jzS)Nzallocs=)reprr1r$rrr:s r __str__zAllocator.__str__Us+4Sdj%A%A B BCCCCrc(d|jjd|dS)N< >) __class__rr:s r __repr__zAllocator.__repr__Xs 64>*66T6666rN)rrrr )rrrr )rrrr)r+rrrr,rrr)r+rrrrr )rr8)rr)rrB)rrG)rrrr__annotations__ __slots__r rr*r6r/r;r>rArDrFrJrPrrr rr6s@11-I.<I<I<I<I|TTTTl9>9>9>9>v''''    ::::@@@@MMMMDDDD777777rrN)r __future__r Exceptionrrrrr rUs  H#""""" 5 5 5 5 5y 5 5 5c7c7c7c7c7c7c7c7c7c7r