a
    Ag                     @  s  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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 erddlmZ ddl m!Z! G dd deZ"e"j#Z$dddddZ%G dd dejZ&ddddZ'dddddZ(ee)e*e+e, ejf Z-dddddd d!Z.dd"dd#d$d%d&Z/ddd'd(d)d*Z0G d+d, d,eZ1dd"ddd$d-d.Z2dd"d/d#d0d1d2Z3d`dd"d/dd#d4d5d6Z4dd7dd8d9Z5d"dd:d7d#d;d<d=Z6dd"d/d#d0d>d?Z7d3Z8ddd@dAdBdCZ9dd7dDdEdFZ:dddDdGdHZ;ddddIdJZ<ddKd7dLdMdNZ=dddOdAdPdQZ>d"dd:dd#dRdSdTZ?dadddUdVddWdXZ@dbdd:dZdOd[d\d]ZAeBe&jCe&e% eDe&jCe4 eEe&jCe3 eFe&jCd^ eGe&jCd_ dS )c    )annotationsN)IntEnum)cached_property)IOTYPE_CHECKINGAnyLiteral
NamedTupleUnion   )Image
ImageChops	ImageFile	ImageMathImageOpsImagePaletteImageSequence)i16le)o8)o16le)_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!   Q/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/PIL/GifImagePlugin.pyr   6   s   r   bytesbool)prefixreturnc                 C  s   | d d dv S )N   )s   GIF87as   GIF89ar!   )r%   r!   r!   r"   _acceptE   s    r(   c                      s   e Zd ZdZdZdZdZddddZd	d
dddZddddZ	e
ddddZed
dddZdddddZd#dd
ddddZdd fddZdddd Zddd!d"Z  ZS )$GifImageFileZGIFzCompuserve GIFFNzbytes | None)r&   c                 C  s,   | j d}|r(|d r(| j |d S d S )Nr   r   )fpread)selfsr!   r!   r"   dataU   s    zGifImageFile.datar#   r$   )pr&   c                 C  sX   t dt|dD ]B}|d ||   krJ||d    krJ||d  ksn  dS qdS )Nr      r   r   TF)rangelen)r,   r/   ir!   r!   r"   _is_palette_needed[   s    8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	@ r|d
 | jd< | j d|> }| |rt	d|}| | _
| _| j | _| j  | _d | _| d d S )N   znot a GIF filer'   version   
      r         
backgroundr0   RGBr   )r*   r+   r(   SyntaxErrorinfoi16_sizer4   r   rawglobal_palettepalette_fptell_GifImageFile__rewind	_n_frames_seek)r,   r-   msgflagsbitsr/   r!   r!   r"   _opena   s$    
zGifImageFile._openintc                 C  s^   | j d u rX|  }z| |  d d qW n  tyL   |  d | _ Y n0 | | | j S )Nr   FrI   rG   rJ   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yN   d}Y n0 | | |S )Nr   TFrP   )r,   rS   is_animatedr!   r!   r"   rU      s    



zGifImageFile.is_animated)framer&   c                 C  s   |  |sd S || jk r(d | _| d | j}t| jd |d D ]P}z| | W qB ty } z$| | d}t||W Y d }~qBd }~0 0 qBd S )Nr   r   no more images in GIF file)Z_seek_check_GifImageFile__frame_imrJ   r1   rQ   rR   )r,   rV   Z
last_frameferK   r!   r!   r"   rR      s    



zGifImageFile.seekT)rV   update_imager&   c                   s0  |dkrBd _ d  _d _ j j d _d jv rT jd= n jrT|rT 	  | jd krtd| }t
| j _ j r j j    rqd _  jd}|r|dkrd}t|d }i }d }d }d }	|s jd}|r|dkrqn|dkrj jd}  }
|d d	kr||
d ur||
d }|d@ rL|
d
 }t|
dd |d< d|@ }|d? }|rX| _n|d dkrd}|
r||
7 }  }
qd|v r|d  d| 7  < n||d< d }qn|d dkrX|dkrX|
d urX|
 j f|d< |
d d dkrX  }
|
rXt|
d
krX|
d dkrXt|
d jd<   rqXn"|dkr jd}t|dt|d }}|t|d |t|d  }}| jd ks| jd kr|rt| jd t| jd f _t j ||||f}	|d }|d@ dk}|d@ rl|d@ d } jd
|> } |rhtd|}nd } jdd } j  _ qd }q|d u rd!}t|| _|sd S g  _ jr؈ j j j |d ur|n j _| _|dkrn jr0t t!j"kr(|d ur d"nd _#nd# _#nd$ _#|rD| _$n( jrfdd%l%m%} | j _$nd  _$n| j&d#krt t!j'ks|rd& jv rЈ j( jd& d  j)d"tj*j+ _d" _# jd&= nd _# j)dtj*j+ _d'd(d) fd*d+}d  _|	 _ jrj jdkrjz4 jdkrʈ j\}}}}|| || f}t| d#} j,d&|}|d ur j&d,v rd"}||d- }n& j,d.d} j&d,v rd}||}tj-.||| _n j/d ur 0 j j _nh|d urR j\}}}}|| || f}t| d#}|} j&d,v r@d"}||d- }tj-.||| _W n t1yh   Y n0 |d urd}|d ur|dkrt t!j"kr| jd&< n j&d,vr|}t23d/||||f j |||fg _|,dr|d  jd< d0D ]4}||v r||  j|< n| jv r j|= qd S )1Nr   commentr   zcannot seek to frame    ;rW      !   r0   r9   duration   r             
   	extensionr<      NETSCAPE2.0loop   ,	      r'   r8   @   r;   r:   r>   Fzimage not found in GIF frameRGBAPL)copytransparencyrO   ztuple[int, int, int])colorr&   c                   sP    j rB| d d t j jkr"d} t j j| d | d d  S | | | fS d S )Nr0   r   )_frame_paletter2   rE   tuple)rt   r,   r!   r"   _rgbY  s
     z GifImageFile._seek.<locals>._rgbr>   ro   )r   r=   gif)rb   rh   )4Z_GifImageFile__offsetZdisposerX   rF   rR   rH   Zdisposal_methodr@   Ztileload
ValueErrorr*   r.   r+   rQ   rA   rG   r2   sizemaxrB   r   Z_decompression_bomb_checkr4   r   rC   impastedispose_extentrD   ru   _frame_transparencyLOADING_STRATEGYr   r    _moderE   rr   moder   putpalettealphaconvertDitherFLOYDSTEINBERGgetcorefillrY   _cropAttributeErrorr   _Tile)r,   rV   r\   rK   r-   rE   r@   Zframe_transparency	interlaceZframe_dispose_extentblockrL   Zdispose_bitsr^   Zx0Zy0x1y1rM   r/   rr   rx   Zdispose_sizeZdispose_modert   rs   kr!   rw   r"   rJ      s^   






""

&"












	
zGifImageFile._seekc                   s   | j r
dnd}d | _| jdkr@| jd urtj|| j| j| _nT| j	dv r| j| _| j rtjd| j| jpld| _| jj
dg| j  R   nd | _| js| jd ur| j| jjkrtj| jj	| j}| j r|j
dg| j  R   || jd| jj  || _|| _d | _ t   d S )Nrp   rq   r   ry   r>   r   r   )ru   _prev_imrX   r   r   r   r   r}   r   r   
putpalettegetdatarY   r   r   superload_prepare)r,   Z	temp_modeexpanded_im	__class__r!   r"   r     s,    



$zGifImageFile.load_preparec                 C  s  | j dkr`| 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sjd S | j| jjkr| jd urt
jd| j}n.t
jd| j}|dd| j  |	d}|| jd| jj  || _| jd usJ | jd ur| j| jd | j	d}n| j	d}| jd us2J | || j}| j| _| jj| _|jdkrr| j|| j| n| j|| j d S )Nr   rp   ro   r>   r   )rX   r   r   r   r    r   r   r   r   r   r   r   r   r   r}   r   r   r   
getpaletter   r   r   )r,   r   Zframe_imr!   r!   r"   load_end  s<    




zGifImageFile.load_endc                 C  s   | j S N)rX   rw   r!   r!   r"   rG     s    zGifImageFile.tell)T)r   r   r   formatformat_descriptionZ!_close_exclusive_fp_after_loadingrD   r.   r4   rN   propertyrT   r   rU   rR   rJ   r   r   rG   __classcell__r!   r!   r   r"   r)   N   s$     'r)   rq   rp   )1rq   rp   Image.Image)r   r&   c                 C  s   | j tv r|   | S t| j dkr| jdtjjd} | jdusFJ | jj dkr| jj	D ]&}|d dkrZ| jj	| | j
d<  qqZ| 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
    r>   rp   rE   Nro   r0   r   rs   rq   )r   RAWMODEr{   r   Zgetmodebaser   ZPaletteZADAPTIVErE   colorsr@   )r   Zrgbar!   r!   r"   _normalize_mode  s    
r   z_Palette | Nonedict[str, Any])r   rE   r@   r&   c                 C  s  d}|r>t |tttfr(t|dd }t |tjr>t|j}| jdkrl|s| d}|dusbJ t|}n*|stdd tdD }tjd|d| _|dusJ |rg }| jdusJ tdt	|d	D ]<}t
|||d	  }| jj|}||v rd}|| qt|D ]@\}}|du rtt	|D ]}	|	|vr.|	||<  qq.qg }
|D ]}|duslJ |
| qZ| |
} n`t| |}|dur| ||} d
|v rz||d
 |d
< W n ty   |d
= Y n0 | 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   rp   c                 s  s   | ]}|d  V  qdS )r0   Nr!   .0r3   r!   r!   r"   	<genexpr>0  re   z%_normalize_palette.<locals>.<genexpr>r>   r   r   r0   rs   )
isinstancer#   	bytearraylistr   rE   r   r   r1   r2   rv   r   r   append	enumerateZremap_palette_get_optimizeindexr|   )r   rE   r@   Zsource_paletteZ
im_paletteused_palette_colorsr3   Zsource_colorr   jZdest_mapZoptimized_palette_colorsr!   r!   r"   _normalize_palette  sb    








r   z	IO[bytes]r5   )r   r*   rE   r&   c                 C  s   t | }|j D ] \}}t|tr| j|| qt||| j}t|| jD ]}|	| qNd}t
| rr|dB }t|| d| dt
| f|_t||tdd| j dt|j g |	d d S )Nr   rn   r   r8   rz       )r   r@   itemsr   strencoderinfo
setdefaultr   _get_global_headerwriteget_interlace_write_local_headerZencoderconfigr   _saver   r}   r   r   )r   r*   rE   Zim_outr   vr-   rL   r!   r!   r"   _write_single_frameZ  s     
 r   z4tuple[Image.Image, tuple[int, int, int, int] | None])base_imim_framer&   c                 C  sR   dd | |fD }|d |d kr6| d}|  d} t|| }||jddfS )Nc                 S  s"   g | ]}|j rt|j j nd qS )re   )rE   r#   )r   r   r!   r!   r"   
<listcomp>y  s   z_getbbox.<locals>.<listcomp>r   r   ro   F)Z
alpha_only)r   r   Zsubtract_moduloZgetbbox)r   r   palette_bytesdeltar!   r!   r"   _getbboxv  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 ]}
t|
 }
|dkr|
j	 D ]*\}}|dkrq~t
|tr~| j || q~| j  }d|
jv r|d|
jd  t|
||}
t
|ttfr|| |d< n$|d u rd|
jv r|
jd |d< t
|ttfr8|| |d< |d7 }d }|r"|r"t||
\}}|s|drZ|d j d  |d 7  < qZ|d j ddkr|d u r| j d| jdd	}t|
|}td
|
j|}|d jj}|d usJ |||j t||
d }q&|dr&|
jdkr&d|vrr|
jd usHJ z|
j|
|d< W n typ   Y n0 d|v r&|
 }td
|j|d }|jdkr| \}}}}tjdd ||||d}n>|jd
krtd|j}||  |}tjdd |d}|j |t!"|d nd }|
}|#t$|p8|
|| qZqJt%|dkrxd| j v rt|d j d | j d< dS |D ]r}|j}
|j&st'|
|j D ]}|(| qd}n*|sd|j d< |
)|j&}
|j&d d }t*||
||j  q|dS )Nrb   disposalr   Zappend_imagesrs   r   r]   r   )r   r   r   rp   optimizer   ro   c                 S  s@   | d | d | d | d | d | d | d | d d dS )	Nr   r~   rgbarg   r   r!   argsr!   r!   r"   <lambda>  s   z(_write_multiple_frames.<locals>.<lambda>)r   r   r   r   rq   c                 S  s   | d | d d dS )Nr   r   rg   r   r!   r   r!   r!   r"   r     re   r   )maskFr   Tinclude_color_table)+r   r   r@   	itertoolschainr   Iteratorr   rr   r   r   r   r   r   r   rv   r   _get_backgroundr   newr}   r   rE   r   r   Z_new_color_indexr|   splitr   Zlambda_evalZputdatar   r   r   invertr   r   r2   r   r   r   Zcrop_write_frame_data)r   r*   rE   rb   r   Z	im_framesZprevious_imZframe_countZbackground_imZ
imSequencer   r   r   r   Z
diff_framer   r   rt   r=   Zfirst_paletter   r   r   r   r   r   Zdelta_lZ
frame_datar-   offsetr!   r!   r"   _write_multiple_frames  s    











r   zstr | bytes)r   r*   filenamer&   c                 C  s   t | ||dd d S )NT)save_all)r   )r   r*   r   r!   r!   r"   	_save_all  s    r   F)r   r*   r   r   r&   c                 C  sz   d| j v sd| jv r,| j d| jd}nd }| j dd |rNt| ||sZt| || |d t|drv|  d S )NrE   r   Tr_   flush)	r   r@   r   r   r   r   r   hasattrr   )r   r*   r   r   rE   r!   r!   r"   r   	  s    

r   rO   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   ztuple[int, int])r*   r   r   rL   r&   c                 C  sn  z|j d }W n ty$   d }Y n0 d|j v rDt|j d d }nd}t|j dd}|d usn|dksn|r|d urzdnd}||d> O }| dtd	 td
 t| t| t|pd td  |j d}|rt|}	t|	}
|
r|dB }||
B }| dt|d  t|d  t|j	d  t|j	d  t|  |r\|
r\| t
|	 | td d S )Nrs   rb   r9   r   r   r   r   r`   ra   rm   r   r;   rk   r8   )r   KeyErrorrO   r   r   r   o16_get_palette_bytes_get_color_table_sizer}   _get_header_palette)r*   r   r   rL   rs   rb   r   Zpacked_flagr   r   color_table_sizer!   r!   r"   r   &  sb    





r   c           
      C  s(  |   }zt|d}| jdkr8tjd|g|tjd ndd|g}dg}tj|tjtjd}tj||j|tjd}|jd usJ |j	  |
 }	|	rt|	||
 }	|	rt|	|W d    n1 s0    Y  W zt| W n ty   Y n0 n&zt| W n ty    Y n0 0 d S )Nwbr>   Zppmtogif)stdoutstderrZppmquantZ256)stdinr   r   )_dumpopenr   
subprocess
check_callDEVNULLPopenPIPEr   closewaitCalledProcessErrorosunlinkOSError)
r   r*   r   tempfilerZ   Z	quant_cmdZ	togif_cmdZ
quant_procZ
togif_procretcoder!   r!   r"   _save_netpbmX  sD    



,r  zlist[int] | None)r   r@   r&   c                 C  s   | j dv r|r|drtp$| j dk}|s:| j| j dk rg }t|  D ]\}}|rJ|| qJ|svt|t	|krz|S | j
dusJ t	| j
j
t| j
j  }d|d  > }t	||d kr|dkr|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
    )rp   rq   r   rq   i   Nr   r   )r   r   _FORCE_OPTIMIZEwidthheightr   Z	histogramr   r~   r2   rE   r   Zgetmodebands
bit_length)r   r@   Zoptimiser   r3   countZnum_palette_colorsZcurrent_palette_sizer!   r!   r"   r     s(    r   )r   r&   c                 C  s:   | sdS t | dk rdS ttt | d dd S d S )Nr   rl   r   r0   r   )r2   mathceillog)r   r!   r!   r"   r     s
    r   c                 C  s<   t | }d|> t| d  }|dkr8| 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   r0   r   )r   r2   r   )r   r   Zactual_target_size_diffr!   r!   r"   r     s
    r   c                   sJ   | j s
dS t| j j  | j jdkrFd 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
    re   ro   c                 3  s&   | ]} |d  |d  d  V  qdS )rm   r0   Nr!   r   r   r!   r"   r     re   z%_get_palette_bytes.<locals>.<genexpr>r0   )rE   r#   r   joinr1   r2   r   r!   r   r"   r     s    $r   z=int | tuple[int, int, int] | tuple[int, int, int, int] | None)r   info_backgroundr&   c              
   C  sp   d}|rlt |trh| jd us J z| j|| }W ql tyd } zt|dvrP W Y d }~qld }~0 0 n|}|S )Nr   )z$cannot allocate more than 256 colorsz/cannot add non-opaque RGBA color to RGB palette)r   rv   rE   Zgetcolorr|   r   )r   r  r=   r[   r!   r!   r"   r     s    
r   zlist[bytes]c                 C  s  d}| j ddksB|rFd|v sB|ddusB|dsB|drFd}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   87ar7   s   89ars   rj   Nrb   r^   r=   s   GIFr   r   r;   r`   rg   r<   ri   r0   rd   )r@   r   r   r   r   r   r}   r   r   r   r   r   encoder1   r2   )r   r@   r7   r=   r   r   headerZcomment_blockr^   r3   Zsubblockr!   r!   r"   r     sp    	



r   )r*   r   r   paramsr&   c                 C  sZ   zN||_ t| ||d t|| tdd|j dt|j g | d W |` n|` 0 d S )Nr   rz   r   r   )	r   r   r   r   r   r}   r   r   r   )r*   r   r   r  r!   r!   r"   r   B  s    r   zdict[str, Any] | Nonez$tuple[list[bytes], list[int] | None]c                 C  sd   |du ri }t | |}d|vr6d| jv r6| 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   r@   r   rE   r   r   )r   rE   r@   r   Zim_modr  r!   r!   r"   	getheader]  s    

r  r   r   )r   r   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ddZdS )zgetdata.<locals>.Collectorr   rO   )r.   r&   c                 S  s   | j | t|S r   )r.   r   r2   )r,   r.   r!   r!   r"   r     s    z getdata.<locals>.Collector.writeN)r   r   r   r.   r   r!   r!   r!   r"   	Collector  s   r  )ior  r{   r   r.   )r   r   r  r  r  r*   r!   r!   r"   r   {  s    r   z.gifz	image/gif)F)NN)r   )H
__future__r   r   r
  r   r   enumr   	functoolsr   typingr   r   r   r   r	   r
    r   r   r   r   r   r   r   _binaryr   rA   r   r   r   r   _typingr   r   r   r   r(   r)   r   r   r#   r   r   rO   Z_Paletter   r   r   r   r   r   r   r   r   r  r  r   r   r   r   r   r   r   r  r   Zregister_openr   Zregister_saveZregister_save_allZregister_extensionZregister_mimer!   r!   r!   r"   <module>   sh    $			   (G| 
220
>  (