o
     h'                     @  s   d dl mZ d dlmZ d dlmZmZ ddlmZm	Z	 z
ddlm
Z
 dZW n ey1   dZY nw d	d
d
dZd$ddZG dd de	j	Zd%ddZd&ddZd&dd Zeejee er}eeje eeje eejd! eejd" d#S d#S )'    )annotations)BytesIO)IOAny   )Image	ImageFile)_webpTFRGBRGBA)s   VP8 s   VP8Xs   VP8Lprefixbytesreturn
bool | strc                 C  sH   |  d}| dd dk}| dd tv }|r"|r"|r"ts 	 dS dS dS )	Ns   RIFF      s   WEBP   zEimage file could not be identified because WEBP support not installedTF)
startswith_VP8_MODES_BY_IDENTIFIER	SUPPORTED)r   Zis_riff_file_formatZis_webp_fileZis_valid_vp8_mode r   Y/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/PIL/WebPImagePlugin.py_accept   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	d$d%ddZ
d&ddZd#ddZd' fddZd(ddZd)dd Z  ZS )*WebPImageFileZWEBPz
WebP imager   r   Nonec                 C  s   t | j | _| j \| _}}}}|| jd< |d? d@ |d? d@ |d? d@ |d@ f\}}}}||||f| jd< || _| jdk| _	|dkrId	n|| _
|| _| jd
}	| jd}
| jd}|	rh|	| jd< |
ro|
| jd< |rv|| jd< | jdd d S )Nloop      r   r   
backgroundr   RGBXr
   ZICCPZEXIFzXMP icc_profileexifxmpF)reset)r	   ZWebPAnimDecoderfpread_decoderget_info_sizeinfon_framesZis_animated_moderawmodeZ	get_chunk_reset)selfZ
loop_countbgcolorZframe_countmodebg_abg_rbg_gbg_br    r!   r"   r   r   r   _open+   s.   






zWebPImageFile._opendict[int, Any] | Nonec                 C  s   d| j vrd S |   S )Nr!   )r)   ZgetexifZ_get_merged_dictr.   r   r   r   _getexifM   s   
zWebPImageFile._getexifframeintc                 C  s   |  |sd S || _d S N)Z_seek_check_WebPImageFile__logical_framer.   r9   r   r   r   seekR   s   

zWebPImageFile.seekTr#   boolc                 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-   Y   s
   

zWebPImageFile._resettuple[bytes, int, int]c                 C  sh   | j  }|  jd7  _|d u r|   | d d}t||\}}|| j }|| _||8 }|||fS )Nr   r   z(failed to decode next frame in WebP file)r&   Zget_nextrA   r-   r>   EOFErrorrC   )r.   retmsgdata	timestampdurationr   r   r   	_get_next`   s   



zWebPImageFile._get_nextc                 C  sD   | j |krd S || j k r|   | j |k r |   | j |k sd S d S r;   )rA   r-   rK   r=   r   r   r   _seeku   s   


zWebPImageFile._seekImage.core.PixelAccess | Nonec                   s   | j | jkr?| | j |  \}}}|| jd< || jd< | j| _ | jr,| jr,| j  t|| _t	
dd| j d| jg| _t  S )NrI   rJ   raw)r   r   r   )rB   r<   rL   rK   r)   r$   Z_exclusive_fpcloser   r   Z_Tilesizer,   Ztilesuperload)r.   rH   rI   rJ   	__class__r   r   rR   }   s   




zWebPImageFile.loadposc                 C  s   d S r;   r   )r.   rU   r   r   r   	load_seek   s   zWebPImageFile.load_seekc                 C  s   | j S r;   )r<   r7   r   r   r   tell   s   zWebPImageFile.tell)r   r   )r   r6   )r9   r:   r   r   )T)r#   r?   r   r   )r   rD   )r   rM   )rU   r:   r   r   )r   r:   )__name__
__module____qualname__formatformat_descriptionrB   r<   r5   r8   r>   r-   rK   rL   rR   rV   rW   __classcell__r   r   rS   r   r   %   s    

"



r   imImage.Imagec                 C  s"   | j dvr| | jrdnd} | S )N)r   r   r
   r   r
   )r0   convertZhas_transparency_data)r^   r   r   r   _convert_frame   s   
ra   r$   	IO[bytes]filenamestr | bytesr   c           '   
   C  sb  | j  }t|dg }d}| g| D ]
}|t|dd7 }q|dkr+t| || d S d}d|v r6|d }n1d| jv rg| jd }t|trg| 	 }|ra||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d}| j dpd}| j dd}t|t
jr| }| j dd}|rd}|d u r|rdnd}|d u r|rd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} d}!|  }"zL| g| D ]?}t|dd}#t|#D ]1}$||$ t|}%||% t|!|||| t|ttfrs|!||  7 }!n|!|7 }!| d7 } qKq?W | |" n| |" w |d t|!|||d ||||}&|&d u rd$}t|||& d S )%Nappend_imagesr   r*   r   )r   r   r   r   r      r   rJ   r   minimize_sizeFkminkmaxallow_mixedlosslessqualityP   alpha_qualityd   methodr     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   1cannot write file as WebP (encoder returned None))encoderinfocopylistgetgetattr_saver)   
isinstancer:   Z
getpaletter   ExiftobytestuplelenallOSErrorr	   ZWebPAnimEncoderrP   rW   ranger>   ra   addgetimroundassemblewrite)'r^   r$   rc   r{   re   totalZimsr   palettergbrJ   r   rg   rh   ri   rj   verboserk   rl   rn   rp   r    r!   r"   rG   r2   r3   r4   r1   encZ	frame_idxrI   Zcur_idxZnfridxr9   rH   r   r   r   	_save_all   s   






 


r   c                 C  s   | j dd}| j dd}| j dd}| j dpd}| j d	d
}t|tjr.| }|dr9|dd  }| j dd}| j dd}	| j drOdnd}
t| } t	| 
 |t|t|||	|
||	}|d u rrd}t||| d S )Nrk   Frl   rm   rn   ro   r    rq   r!       s   Exif     r"   rp   ru   exactr   r   rz   )r{   r~   r   r   r   r   r   ra   r	   Z
WebPEncoder   floatr   r   )r^   r$   rc   rk   rl   rn   r    r!   r"   rp   r   rH   rG   r   r   r   r     s8   
r   z.webpz
image/webpN)r   r   r   r   )r^   r_   r   r_   )r^   r_   r$   rb   rc   rd   r   r   )
__future__r   ior   typingr   r   rq   r   r   r	   r   ImportErrorr   r   r   ra   r   r   Zregister_openr[   Zregister_saveZregister_save_allZregister_extensionZregister_mimer   r   r   r   <module>   s4    

q

|"