U
    ­A›df,  ã                   @   sì   d dl mZ ddlmZmZ zddlmZ dZW n ek
rH   dZY nX ddddœZdddœZ	d	d
d
dœZ
dd„ ZG dd„ dejƒZdd„ Zdd„ Ze ejee¡ erèe eje¡ ejrÌe eje¡ e ejd¡ e ejd¡ dS )é    )ÚBytesIOé   )ÚImageÚ	ImageFile)Ú_webpTF)ÚRGBXÚRGBAÚRGB)r	   r   r	   r   )s   VP8 s   VP8Xs   VP8Lc                 C   sL   | d d… dk}| dd… dk}| dd… t k}|rH|rH|rHtsDdS dS d S )	Né   s   RIFFé   é   s   WEBPé   zEimage file could not be identified because WEBP support not installedT)Ú_VP8_MODES_BY_IDENTIFIERÚ	SUPPORTED)ÚprefixZis_riff_file_formatZis_webp_fileZis_valid_vp8_mode© r   úL/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/PIL/WebPImagePlugin.pyÚ_accept   s    ÿr   c                       sr   e Zd ZdZdZdZdZdd„ Zdd„ Zdd	„ Z	d
d„ Z
ddd„Zdd„ Zdd„ Z‡ fdd„Z‡ fdd„Z‡  ZS )ÚWebPImageFileZWEBPz
WebP imager   c                 C   s�  t js|t  | j ¡ ¡\}}}| _}}|r2|| jd< |r@|| jd< ||f| _t|ƒ| _dd| j	 d| jfg| _
d| _d| _d S t  | j ¡ ¡| _| j ¡ \}}}}}}	||f| _|| jd< |d	? d
@ |d? d
@ |d? d
@ |d
@ f\}
}}}||||
f| jd< || _| jdk| _|	dk�rdn|	| _|	| _g | _
| j d¡}| j d¡}| j d¡}|�r`|| jd< |�rp|| jd< |�r€|| jd< | jdd� d S )NÚicc_profileÚexifÚraw©r   r   r   r   FÚloopé   éÿ   r   r   Ú
backgroundr   r	   ZICCPZEXIFzXMP Úxmp)Úreset)r   ÚHAVE_WEBPANIMZ
WebPDecodeÚfpÚreadÚmodeÚinfoÚ_sizer   ÚsizeÚtileÚn_framesZis_animatedZWebPAnimDecoderÚ_decoderÚget_infoÚrawmodeZ	get_chunkÚ_reset)ÚselfÚdataÚwidthÚheightr   r   Z
loop_countÚbgcolorZframe_countr"   Úbg_aÚbg_rÚbg_gÚbg_br   r   r   r   Ú_open+   sN    ÿ








ü


zWebPImageFile._openc                 C   s   d| j krd S |  ¡  ¡ S )Nr   )r#   ZgetexifZ_get_merged_dict©r,   r   r   r   Ú_getexif_   s    
zWebPImageFile._getexifc                 C   s   d| j kr|  | j d ¡S i S )z™
        Returns a dictionary containing the XMP tags.
        Requires defusedxml to be installed.

        :returns: XMP tags in a dictionary.
        r   )r#   Z_getxmpr6   r   r   r   Úgetxmpd   s    zWebPImageFile.getxmpc                 C   s   |   |¡sd S || _d S ©N)Z_seek_checkÚ_WebPImageFile__logical_frame©r,   Úframer   r   r   Úseekm   s    
zWebPImageFile.seekTc                 C   s$   |r| j  ¡  d| _d| _d| _d S )Nr   éÿÿÿÿ)r(   r   Ú_WebPImageFile__physical_frameÚ_WebPImageFile__loadedÚ_WebPImageFile__timestamp)r,   r   r   r   r   r+   t   s
    
zWebPImageFile._resetc                 C   sh   | j  ¡ }|  jd7  _|d kr>|  ¡  |  d¡ d}t|ƒ‚|\}}|| j }|| _||8 }|||fS )Nr   r   z(failed to decode next frame in WebP file)r(   Zget_nextr?   r+   r=   ÚEOFErrorrA   )r,   ÚretÚmsgr-   Ú	timestampÚdurationr   r   r   Ú	_get_next{   s    


zWebPImageFile._get_nextc                 C   s8   | j |krd S || j k r |  ¡  | j |k r4|  ¡  q d S r9   )r?   r+   rG   r;   r   r   r   Ú_seek�   s    


zWebPImageFile._seekc                    sŠ   t jr€| j| jkr€|  | j¡ |  ¡ \}}}|| jd< || jd< | j| _| jr^| jr^| j 	¡  t
|ƒ| _dd| j d| jfg| _tƒ  ¡ S )NrE   rF   r   r   r   )r   r   r@   r:   rH   rG   r#   r    Z_exclusive_fpÚcloser   r%   r*   r&   ÚsuperÚload)r,   r-   rE   rF   ©Ú	__class__r   r   rK   ˜   s    



zWebPImageFile.loadc                    s   t jstƒ  ¡ S | jS r9   )r   r   rJ   Útellr:   r6   rL   r   r   rN   «   s    
zWebPImageFile.tell)T)Ú__name__Ú
__module__Ú__qualname__ÚformatÚformat_descriptionr@   r:   r5   r7   r8   r=   r+   rG   rH   rK   rN   Ú__classcell__r   r   rL   r   r   %   s   4	
r   c           (      C   sî  | j  ¡ }t| dg ¡ƒ}d}| g| D ]}|t|ddƒ7 }q(|dkrVt| ||ƒ d S d}d|krl|d }nbd| jkrÎ| jd }t|tƒrÎ|  	¡ }|rÂ||d |d d … \}	}
}|	|
|df}n|||df}| j  d	| j d	d¡¡}| j  d
d¡}| j  dd¡}| j  dd ¡}| j  dd ¡}| j  dd¡}d}| j  dd¡}| j  dd¡}| j  dd¡}| j  d¡�pjd}| j  dd¡}t|t
jƒ�r�| ¡ }| j  dd¡}|�r¨d}|d k�rÀ|�r¼dnd}|d k�rØ|�rÔdnd}t|ttfƒ�r
t|ƒdk�s
tdd„ |D ƒƒ�sd|› �}t|ƒ‚|\}}}}|d> |d > B |d!> B |d> B }t | jd | jd |||||||¡	}d}d} |  ¡ }!�z| g| D �] }t|ddƒ}"t|"ƒD ]ä}#| |#¡ | ¡  |}$|j}%|jtk�rd"|jk�pd#|jk�p|jd$k�od"|j ¡ k}&|&�rd%nd&}%| |%¡}$|%d&k�r*d'}%| |$ d(|%¡t| ƒ|$jd |$jd |%|||¡ t|ttfƒ�rx| || 7 } n| |7 } |d7 }�q¦�qŒW 5 |  |!¡ X | d t| ƒddd||d¡ | |||¡}'|'d k�ràd)}t|ƒ‚|  |'¡ d S )*NÚappend_imagesr   r'   r   )r   r   r   r   r   é   r   rF   r   Úminimize_sizeFÚkminÚkmaxÚallow_mixedÚlosslessÚqualityéP   Úmethodr   Ú r   r   é	   é   é   r
   c                 s   s&   | ]}d |  kodk n  V  qdS )r   é   Nr   )Ú.0Úvr   r   r   Ú	<genexpr>ê   s     z_save_all.<locals>.<genexpr>z:Background color is not an RGBA tuple clamped to (0-255): r   r   r   ÚAÚaÚPr   r	   r   r   ú1cannot write file as WebP (encoder returned None))!ÚencoderinfoÚcopyÚlistÚgetÚgetattrÚ_saver#   Ú
isinstanceÚintZ
getpaletter   ÚExifÚtobytesÚtupleÚlenÚallÚOSErrorr   ZWebPAnimEncoderr%   rN   r=   ÚrangerK   r"   Ú_VALID_WEBP_MODESÚimZgetpalettemodeÚconvertÚaddÚroundÚassembleÚwrite)(r{   r    Úfilenamerk   rU   ÚtotalZimsr   ÚpaletteÚrÚgÚbrF   r   rW   rX   rY   rZ   Úverboser[   r\   r^   r   r   r   rD   r2   r3   r4   r1   ÚencZ	frame_idxrE   Zcur_idxZnfrÚidxr<   r*   Úalphar-   r   r   r   Ú	_save_all²   sÎ    






ÿ
þý
 ÷
ÿý


ø
r‹   c                 C   s0  | j  dd¡}| j  dd¡}| j  d¡p*d}| j  dd¡}t|tjƒrN| ¡ }| d	¡rd|d
d … }| j  dd¡}| j  dd¡}| j  d¡r�dnd}	| jtkrØd| jkpÄd| jkpÄ| jdkoÄd| j	k}
|  
|
rÒdnd¡} t |  ¡ | jd | jd |t|ƒ| j|||	||¡}|d k�r"d}t|ƒ‚| |¡ d S )Nr[   Fr\   r]   r   r_   r   ó    s   Exif  é   r   r^   r
   Úexactr   r   rg   rh   ri   Ztransparencyr   r	   rj   )rk   rn   rq   r   rs   rt   Ú
startswithr"   Ú_VALID_WEBP_LEGACY_MODESr#   r|   r   Z
WebPEncoder%   Úfloatrx   r€   )r{   r    r�   r[   r\   r   r   r   r^   rŽ   rŠ   r-   rD   r   r   r   rp   ?  sF    


ÿýõ
rp   z.webpz
image/webpN)Úior   r_   r   r   r   r   ÚImportErrorrz   r�   r   r   r   r‹   rp   Zregister_openrR   Zregister_saver   Zregister_save_allZregister_extensionZregister_mimer   r   r   r   Ú<module>   s2   

ý  )