o
    'j6h                     @  sR  d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dlm	Z	 d dl
mZmZmZmZmZ ddlmZmZmZmZmZmZmZ ddlmZ dd	lmZ dd
lmZ ddlmZ dZ e rjddlm!Z! ddl"m#Z# G dd deZ$e$j%Z&dcddZ'G dd dejZ(ddddZ)ddddZ*ee+e,e-e. ejf Z/ded$d%Z0dfd)d*Z1dgd.d/Z2G d0d1 d1eZ3dhd2d3Z4did6d7Z5	djdkd9d:Z6dld<d=Z7dmdAdBZ8didCdDZ9dZ:dndFdGZ;dodIdJZ<dpdKdLZ=dqdMdNZ>drdQdRZ?dsdTdUZ@dtdWdXZA	dudvd[d\ZB	]dwdxd_d`ZCeDe(jEe(e' eFe(jEe6 eGe(jEe5 eHe(jEda eIe(jEdb dS )y    )annotationsN)IntEnum)cached_property)IOAnyLiteral
NamedTupleUnion   )Image
ImageChops	ImageFile	ImageMathImageOpsImagePaletteImageSequence)i16le)o8)o16le)DeferredErrorF)_imaging)Bufferc                   @  s   e Zd ZdZdZdZdZdS )LoadingStrategyz.. versionadded:: 9.1.0r   r
      N)__name__
__module____qualname____doc__RGB_AFTER_FIRST RGB_AFTER_DIFFERENT_PALETTE_ONLY
RGB_ALWAYS r!   r!   a/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/PIL/GifImagePlugin.pyr   8   s
    r   prefixbytesreturnboolc                 C  s
   |  dS )N)s   GIF87as   GIF89a)
startswith)r#   r!   r!   r"   _acceptG   s   
r(   c                      s   e Zd ZdZdZdZdZd#ddZd$ddZd%ddZ	e
d&ddZed'ddZd(ddZd)d*ddZd% fddZd%dd Zd&d!d"Z  ZS )+GifImageFileGIFzCompuserve GIFFNr%   bytes | Nonec                 C  s,   | j d}|r|d r| j |d S d S )Nr
   r   )fpread)selfsr!   r!   r"   dataW   s   zGifImageFile.datapr$   r&   c                 C  s\   t dt|dD ]#}|d ||   kr'||d    kr'||d  ks+ dS   dS qdS )Nr      r
   r   TF)rangelen)r.   r1   ir!   r!   r"   _is_palette_needed]   s   4zGifImageFile._is_palette_neededNonec                 C  s   | j d}t|sd}t||d d | jd< t|dt|df| _|d }|d@ d }|d	@ rR|d
 | jd< | j d|> }| |rRt	d|}| | _
| _| j | _| j  | _d | _| d d S )N   znot a GIF file   version   
      r
         
backgroundr2   RGBr   )r,   r-   r(   SyntaxErrorinfoi16_sizer6   r   rawglobal_palettepalette_fptell_GifImageFile__rewind	_n_frames_seek)r.   r/   msgflagsbitsr1   r!   r!   r"   _openc   s$   
zGifImageFile._openintc                 C  s\   | j d u r+|  }z	 | |  d d q ty%   |  d | _ Y nw | | | j S )NTr
   FrL   rJ   rM   EOFErrorseek)r.   currentr!   r!   r"   n_frames}   s   

zGifImageFile.n_framesc                 C  s^   | j d ur
| j dkS |  }|rdS z
| dd d}W n ty'   d}Y nw | | |S )Nr
   TFrS   )r.   rV   is_animatedr!   r!   r"   rX      s   


zGifImageFile.is_animatedframec                 C  s   |  |sd S || jk rd | _| d | j}t| jd |d D ]!}z| | W q! tyB } z| | d}t||d }~ww d S )Nr   r
   no more images in GIF file)_seek_check_GifImageFile__frame_imrM   r3   rT   rU   )r.   rY   
last_frameferN   r!   r!   r"   rU      s    




zGifImageFile.seekTupdate_imagec                   s2  t  jtr
 jj|dkr+d _d  _d _ j j d _	d j
v r* j
d= n	 jr4|r4   | jd krDd| }t| j _ jr^ j j   r[	   sVd _ jd}|rj|dkrpd}t|d }i }d }d }d }		 |s jd}|r|dkrnB|d	kr: jd}  }
|d d
kr|
d ur|
d }|d@ r|
d }t|
dd |d< d|@ }|d? }|r| _	ng|d dkrd}|
r||
7 }  }
|
sd|v r|d  d| 7  < n||d< d }qz|d dkr.|dkr.|
d ur.|
 j f|d< |
dr.  }
|
r.t|
dkr.|
d dkr.t|
d j
d<   r9	   s3n|dkrʈ jd}t|dt|d}}|t|d |t|d }}| jd kso| jd kr|rt| jd t| jd f _t j ||||f}	|d }|d@ dk}|d@ r|d@ d } jd|> } |rtd|}nd } jdd } j  _nd }q{|d u rd!}t|| _|sd S g  _ jr j j j  |d ur|n j! _"| _#|dkr; j"rt$t%j&kr|d urd"nd _'nd# _'nd$ _'|r&| _(nS j!r7dd%l)m)} | j! _(nBd  _(n> j*d#kryt$t%j+ksJ|ryd& j
v rl j, j
d& d  j-d"tj.j/ _d" _' j
d&= nd _' j-dtj.j/ _d2 fd+d,}d  _|	 _  j r5 j	dkr5z j	dkr j \}}}}|| || f}t| d#} j
0d&|}|d urȈ j*d-v rd"}||d. }n j
0d/d} j*d-v rd}||}tj12||| _nD j3d ur 4 j j  _n4|d ur) j \}}}}|| || f}t| d#}|} j*d-v r d"}||d. }tj12||| _W n
 t5y4   Y nw |d urld}|d urZ|dkrRt$t%j&krQ| j
d&< n j*d-vrZ|}t67d0||||f j|||fg _|0dry|d  j
d< d1D ]}||v r||  j
|< q{| j
v r j
|= q{d S )3Nr   commentr
   zcannot seek to frame    ;rZ   T   !   r2   r<   duration   r             
   	extension   NETSCAPE2.0loop   ,	      r9   r;   @   r>   r=   rA   Fzimage not found in GIF frameRGBAPL)copytransparencycolorrR   r%   tuple[int, int, int]c                   sL    j r!| d d t j jkrd} t j j| d | d d  S | | | fS )Nr2   r   )_frame_paletter4   rH   tuple)ry   r.   r!   r"   _rgb]  s
    
z GifImageFile._seek.<locals>._rgbrA   rt   )r   r@   gif)rg   rm   )ry   rR   r%   rz   )8
isinstancerI   r   ex_GifImageFile__offsetdisposer\   rU   rK   disposal_methodrC   tileload
ValueErrorr,   r0   r-   rT   rD   rJ   r'   r4   sizemaxrE   r   _decompression_bomb_checkr6   r   rF   impastedispose_extentrG   r{   _frame_transparencyLOADING_STRATEGYr   r    _moderH   rw   moder   putpalettealphaconvertDitherFLOYDSTEINBERGgetcorefillr]   _cropAttributeErrorr   _Tile)r.   rY   ra   rN   r/   rH   rC   frame_transparency	interlaceframe_dispose_extentblockrO   dispose_bitsrc   x0y0x1y1rP   r1   rw   r~   dispose_sizedispose_modery   rx   kr!   r}   r"   rM      s|  



""


&"

[










	
zGifImageFile._seekc                   s  | j rdnd}d | _| jdkr | jd urtj|| j| j| _n*| j	dv rJ| j| _| j rGtjd| j| jp6d| _| jj
dg| j  R   nd | _| js| jd ur| j| jjkrtj| jj	| j}| j rr|j
dg| j  R   || jd| jj  || _|| _d | _ t   d S )Nru   rv   r   r   rA   r   r   )r{   _prev_imr\   r   r   r   r   r   r   r   
putpalettegetdatar]   r   r   superload_prepare)r.   	temp_modeexpanded_im	__class__r!   r"   r     s.   



zGifImageFile.load_preparec                 C  s  | j dkr0| jdkr.ttjkr.| jd ur | j| jd d| _nd| _| j	| jt
jj| _d S | js5d S | j| jjkrv| jd urJt
jd| j}nt
jd| j}|dd| j  |	d}|| jd| jj  || _| jd usvJ | jd ur| j| jd | j	d}n| j	d}| jd usJ | || j}| j| _| jj| _|jdkr| j|| j| d S | j|| j d S )Nr   ru   rt   rA   r   )r\   r   r   r   r    r   r   r   r   r   r   r   r   r   r   r   r   r   
getpaletter   r   r   )r.   r   frame_imr!   r!   r"   load_end  s<   






zGifImageFile.load_endc                 C  s   | j S N)r\   r}   r!   r!   r"   rJ     s   zGifImageFile.tell)r%   r+   )r1   r$   r%   r&   )r%   r7   )r%   rR   )r%   r&   )rY   rR   r%   r7   )T)rY   rR   ra   r&   r%   r7   )r   r   r   formatformat_description!_close_exclusive_fp_after_loadingrG   r0   r6   rQ   propertyrW   r   rX   rU   rM   r   r   rJ   __classcell__r!   r!   r   r"   r)   P   s&    



  
'r)   rv   ru   )1rv   ru   r   Image.Imagec                 C  s   | j tv r|   | S t| j dkrD| jdtjjd} | jdus#J | jj dkrB| jj	D ]}|d dkrA| jj	| | j
d<  | S q-| S | d	S )
a  
    Takes an image (or frame), returns an image in a mode that is appropriate
    for saving in a Gif.

    It may return the original image, or it may return an image converted to
    palette or 'L' mode.

    :param im: Image object
    :returns: Image object
    rA   ru   rH   Nrt   r2   r   rx   rv   )r   RAWMODEr   r   getmodebaser   PaletteADAPTIVErH   colorsrC   )r   rgbar!   r!   r"   _normalize_mode  s   

r   rH   _Palette | NonerC   dict[str, Any]c                 C  s  d}|rt |tttfrt|dd }t |tjrt|j}| jdkr6|s5| d}|dus1J t|}n|sCtdd tdD }tjd|d| _|dusQJ |rg }| jdus\J tdt	|d	D ]}t
|||d	  }| jj|}||v r}d}|| qdt|D ]\}}|du rtt	|D ]}	|	|vr|	||<  nqqg }
|D ]}|dusJ |
| q| |
} n/t| |}|dur| ||} d
|v rz||d
 |d
< W | S  ty   |d
= Y | S w | S | jdusJ || j_| S )at  
    Normalizes the palette for image.
      - Sets the palette to the incoming palette, if provided.
      - Ensures that there's a palette for L mode images
      - Optimizes the palette if necessary/desired.

    :param im: Image object
    :param palette: bytes object containing the source palette, or ....
    :param info: encoderinfo
    :returns: Image object
    Ni   ru   c                 s  s    | ]}|d  V  qdS )r2   Nr!   .0r5   r!   r!   r"   	<genexpr>4  s    z%_normalize_palette.<locals>.<genexpr>rA   r   r   r2   rx   )r   r$   	bytearraylistr   rH   r   r   r3   r4   r|   r   r   append	enumerateremap_palette_get_optimizeindexr   )r   rH   rC   source_palette
im_paletteused_palette_colorsr5   source_colorr   jdest_mapoptimized_palette_colorsr!   r!   r"   _normalize_palette  sn   




r   r,   	IO[bytes]r7   c                 C  s   t | }|j D ]\}}t|tr| j|| q	t||| j}t|| jD ]}|	| q'd}t
| r9|dB }t|| d| dt
| f|_t||tdd| j dt|j g |	d d S )Nr   rs   r   r;   r       )r   rC   itemsr   strencoderinfo
setdefaultr   _get_global_headerwriteget_interlace_write_local_headerencoderconfigr   _saver   r   r   r   )r   r,   rH   im_outr   vr/   rO   r!   r!   r"   _write_single_frame^  s"   
 r   base_imim_frame4tuple[Image.Image, tuple[int, int, int, int] | None]c                 C  sR   dd | |fD }|d |d kr| d}|  d} t|| }||jddfS )Nc                 S  s"   g | ]}|j rt|j j nd qS )rj   )rH   r$   )r   r   r!   r!   r"   
<listcomp>}  s    z_getbbox.<locals>.<listcomp>r   r
   rt   F)
alpha_only)r   r   subtract_modulogetbbox)r   r   palette_bytesdeltar!   r!   r"   _getbboxz  s   

r   c                   @  s&   e Zd ZU ded< ded< ded< dS )_Framer   r   z tuple[int, int, int, int] | Nonebboxr   r   N)r   r   r   __annotations__r!   r!   r!   r"   r     s   
 r   c              
   C  s  | j d}| j d| jd}g }d }d}d }t| g| j dg D ]}	t|	D ]w}
t|
 }
|dkrU|
j	 D ]\}}|dkrHq?t
|trT| j || q?| j  }d|
jv rh|d|
jd  t|
||}
t
|ttfr||| |d< n|d u rd|
jv r|
jd |d< t
|ttfr|| |d< |d7 }d }|r|rt||
\}}|s|dr|d j d  |d 7  < q-|d j ddkr| j d| jd}|d ur	|d u rt|
|}td	|
j|}|d jj}|d usJ |||j t||
d }nd
|
j }n|dr|
jdkrd|vr=|
jd us(J z
|
j|
|d< W n
 ty<   Y nw d|v r|
 }td	|j|d }|jdkrk| \}}}}tjdd ||||d}n|jd	krtd|j}||  |}tjdd |d}|j |t!"|d nd }|
}|#t$|p|
|| q-q%t%|dkrd| j v r|d j d | j d< dS |D ]B}|j}
|j&st'|
|j D ]}|(| qd
}n|sd|j d< |j&d
|
j kr|
)|j&}
|j&d d }t*||
||j  qdS )Nrg   disposalr   append_imagesrx   r
   rb   r   ru   r   optimizer   rt   c                 S  s@   | d | d | d | d | d | d | d | d d dS )	Nr   r   rgbarl   r   r!   argsr!   r!   r"   <lambda>  s    z(_write_multiple_frames.<locals>.<lambda>)r   r   r   r   rv   c                 S  s   | d | d d dS )Nr   r   rl   r   r!   r   r!   r!   r"   r     s    r   )maskFTinclude_color_table)+r   r   rC   	itertoolschainr   Iteratorr   rw   r   r   r   r   r   r   r|   r   _get_backgroundr   newr   r   rH   r   r   _new_color_indexr   splitr   lambda_evalputdatar   r   r   invertr   r   r4   r   r   r   crop_write_frame_data)r   r,   rH   rg   r   	im_framesprevious_imframe_countbackground_im
imSequencer   r   r   r   
diff_framer   r   ry   r@   first_paletter   r   r   r   r   r  delta_l
frame_datar/   offsetr!   r!   r"   _write_multiple_frames  s   











^
r  filenamestr | bytesc                 C  s   t | ||dd d S )NT)save_all)r   )r   r,   r  r!   r!   r"   	_save_all  s   r  r  c                 C  s~   d| j v s
d| jv r| j d| jd}n	d }| j dd |r't| ||s-t| || |d t|dr=|  d S d S )NrH   r   Trd   flush)	r   rC   r   r   r  r   r   hasattrr  )r   r,   r  r  rH   r!   r!   r"   r     s   

r   rR   c                 C  s$   | j dd}t| jdk rd}|S )Nr   r
      r   )r   r   minr   )r   r   r!   r!   r"   r   &  s   r   r  tuple[int, int]rO   c                 C  sj  z|j d }W n ty   d }Y nw d|j v r"t|j d d }nd}t|j dd}|d us7|dks7|rd|d ur=dnd}||d> O }| dtd	 td
 t| t| t|p\d td  |j d}|r~t|}	t|	}
|
r~|dB }||
B }| dt|d  t|d  t|j	d  t|j	d  t|  |r|
r| t
|	 | td d S )Nrx   rg   r<   r   r   r
   r   re   rf   rr   r  r>   rp   r;   )r   KeyErrorrR   r   r   r   o16_get_palette_bytes_get_color_table_sizer   _get_header_palette)r,   r   r  rO   rx   rg   r   packed_flagr  r   color_table_sizer!   r!   r"   r   0  sd   




r   c           
      C  s"  |   }z{t|dZ}| jdkrtjd|g|tjd nAdd|g}dg}tj|tjtjd}tj||j|tjd}|jd us@J |j	  |
 }	|	rQt|	||
 }	|	r]t|	|W d    n1 sgw   Y  W zt| W d S  ty   Y d S w zt| W w  ty   Y w w )NwbrA   ppmtogif)stdoutstderrppmquant256)stdinr,  r-  )_dumpopenr   
subprocess
check_callDEVNULLPopenPIPEr,  closewaitCalledProcessErrorosunlinkOSError)
r   r,   r  tempfiler_   	quant_cmd	togif_cmd
quant_proc
togif_procretcoder!   r!   r"   _save_netpbmb  sN   



 rD  list[int] | Nonec                 C  s   | j dv rg|rg|drgtp| j dk}|s| j| j dk rgg }t|  D ]\}}|r0|| q%|s;t|t	|kr=|S | j
dusDJ t	| j
j
t| j
j  }d|d  > }t	||d krg|dkrg|S dS )aL  
    Palette optimization is a potentially expensive operation.

    This function determines if the palette should be optimized using
    some heuristics, then returns the list of palette entries in use.

    :param im: Image object
    :param info: encoderinfo
    :returns: list of indexes of palette entries in use, or None
    )ru   rv   r   rv   i   Nr
   r   )r   r   _FORCE_OPTIMIZEwidthheightr   	histogramr   r   r4   rH   r   getmodebands
bit_length)r   rC   optimiser   r5   countnum_palette_colorscurrent_palette_sizer!   r!   r"   r     s&   
r   r   c                 C  s6   | sdS t | dk rdS ttt | d dd S )Nr   rq   r
   r2   r   )r4   mathceillog)r   r!   r!   r"   r&    s
   r&  c                 C  s<   t | }d|> t| d  }|dkr| tdd | 7 } | S )z
    Returns the palette, null padded to the next power of 2 (*3) bytes
    suitable for direct inclusion in the GIF header

    :param palette_bytes: Unpadded palette bytes, in RGBRGB form
    :returns: Null padded palette
    r   r2   r   )r&  r4   r   )r   r)  actual_target_size_diffr!   r!   r"   r'    s
   r'  c                   sJ   | j sdS t| j j  | j jdkr#d fddtt d D   S )z
    Gets the palette for inclusion in the gif header

    :param im: Image object
    :returns: Bytes, len<=768 suitable for inclusion in gif header
    rj   rt   c                 3  s(    | ]} |d  |d  d  V  qdS )rr   r2   Nr!   r   r   r!   r"   r     s   & z%_get_palette_bytes.<locals>.<genexpr>r2   )rH   r$   r   joinr3   r4   r   r!   r   r"   r%    s   $r%  info_background=int | tuple[int, int, int] | tuple[int, int, int, int] | Nonec              
   C  sr   d}|r7t |tr5| jd usJ z
| j|| }W |S  ty4 } zt|dvr) W Y d }~|S d }~ww |}|S )Nr   )z$cannot allocate more than 256 colorsz/cannot add non-opaque RGBA color to RGB palette)r   r|   rH   getcolorr   r   )r   rU  r@   r`   r!   r!   r"   r    s    



r  list[bytes]c                 C  s~  d}| j ddks!|r#d|v s!|ddus!|ds!|dr#d}t| |d	}t| }t|}d
| t| jd  t| jd  t|d t|td t|g}|ddur}|	dtd td d td td t|d  td  |drdtd }|d }t
|tr| }tdt|dD ]}	||	|	d  }
|tt|
|
 7 }q|td7 }|	| |S )z2Return a list of strings representing a GIF headers   87ar:   s   89arx   ro   Nrg   rc   r@   s   GIFr   r
   r>   re   rl   r?   rn   r2   ri   )rC   r   r  r%  r&  r$  r   r   r'  r   r   r   encoder3   r4   )r   rC   r:   r@   r   r)  headercomment_blockrc   r5   subblockr!   r!   r"   r     sl   





r   paramsc                 C  sX   z(||_ t| ||d t|| tdd|j dt|j g | d W |` d S |` w )Nr   r   r   r   )	r   r   r   r   r   r   r   r   r   )r,   r   r  r]  r!   r!   r"   r  L  s   r  dict[str, Any] | None$tuple[list[bytes], list[int] | None]c                 C  sd   |du ri }t | |}d|vrd| jv r| jd |d< t| ||}|j| _|j| _t| |}||fS )a  
    Legacy Method to get Gif data from image.

    Warning:: May modify image data.

    :param im: Image object
    :param palette: bytes object containing the source palette, or ....
    :param info: encoderinfo
    :returns: tuple of(list of header items, optimized palette)

    Nr@   )r   rC   r   rH   r   r   )r   rH   rC   r   im_modrZ  r!   r!   r"   	getheaderg  s   

ra  r   r   c                 K  s>   ddl m} G dd d|}|   | }t|| || |jS )a  
    Legacy Method

    Return a list of strings representing this image.
    The first string is a local image header, the rest contains
    encoded image data.

    To specify duration, add the time in milliseconds,
    e.g. ``getdata(im_frame, duration=1000)``

    :param im: Image object
    :param offset: Tuple of (x, y) pixels. Defaults to (0, 0)
    :param \**params: e.g. duration or other encoder info parameters
    :returns: List of bytes containing GIF encoded frame data

    r   )BytesIOc                   @  s   e Zd Zg ZdddZdS )	zgetdata.<locals>.Collectorr0   r   r%   rR   c                 S  s   | j | t|S r   )r0   r   r4   )r.   r0   r!   r!   r"   r     s   z getdata.<locals>.Collector.writeN)r0   r   r%   rR   )r   r   r   r0   r   r!   r!   r!   r"   	Collector  s    rc  )iorb  r   r  r0   )r   r  r]  rb  rc  r,   r!   r!   r"   r     s   r   z.gifz	image/gif)r#   r$   r%   r&   )r   r   r%   r   )r   r   rH   r   rC   r   r%   r   )r   r   r,   r   rH   r   r%   r7   )r   r   r   r   r%   r   )r   r   r,   r   rH   r   r%   r&   )r   r   r,   r   r  r  r%   r7   )F)
r   r   r,   r   r  r  r  r&   r%   r7   )r   r   r%   rR   )
r,   r   r   r   r  r"  rO   rR   r%   r7   )r   r   rC   r   r%   rE  )r   r$   r%   rR   )r   r$   r%   r$   )r   r   r%   r$   )r   r   rU  rV  r%   rR   )r   r   rC   r   r%   rX  )
r,   r   r   r   r  r"  r]  r   r%   r7   )NN)r   r   rH   r   rC   r^  r%   r_  )r   )r   r   r  r"  r]  r   r%   rX  )J
__future__r   r  rP  r;  r3  enumr   	functoolsr   typingr   r   r   r   r	    r   r   r   r   r   r   r   _binaryr   rD   r   r   r$  _utilr   TYPE_CHECKINGr   _typingr   r   r   r   r(   r)   r   r   r$   r   r   rR   _Paletter   r   r   r   r  r  r   r   r   rD  rF  r   r&  r'  r%  r  r   r  ra  r   register_openr   register_saveregister_save_allregister_extensionregister_mimer!   r!   r!   r"   <module>   sp   $		
	   *


G

 




22

0





>(