o
     Pj&                     @  s  d dl mZ d dlmZ ddlmZmZ z
ddlmZ dZW n e	y)   dZY nw dZ
e
r6d dlmZmZ 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reeje eeje eejd! eejd" d#S d#S )'    )annotations)BytesIO   )Image	ImageFile)_webpTF)IOAny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   is_riff_file_formatis_webp_fileis_valid_vp8_mode r   H/home/thesage/.local/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 )*WebPImageFileWEBPz
WebP imager   r   Nonec                 C  s   | j d usJ | j  }t|sd}t|t|| _| j \| _| j	d< }| _
| _|d? d@ |d? d@ |d@ |d? d@ f| j	d< | j
dk| _| jd	krPd
n| j| _dddd D ]\}}| j| }rm|| j	|< q\| jdd d S )Nznot a WEBP fileloopr      r      
backgroundr   RGBXr
   ICCPEXIFzXMP )icc_profileexifxmpF)reset)fpreadr   SyntaxErrorr   WebPAnimDecoder_decoderget_info_sizeinfon_framesrawmodeis_animated_modeitems	get_chunk_reset)selfsmsgbgcolorkey
chunk_namevaluer   r   r   _open-   s2   






zWebPImageFile._opendict[int, Any] | Nonec                 C  s   d| j vrd S |   S )Nr'   )r1   getexif_get_merged_dictr9   r   r   r   _getexifQ   s   
zWebPImageFile._getexifframeintc                 C  s   |  |sd S || _d S N)_seek_check_WebPImageFile__logical_framer9   rF   r   r   r   seekV   s   

zWebPImageFile.seekTr)   boolc                 C  s$   |r| j   d| _d| _d| _d S )Nr   )r.   r)   _WebPImageFile__physical_frame_WebPImageFile__loaded_WebPImageFile__timestamp)r9   r)   r   r   r   r8   ]   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.   get_nextrO   r8   rL   EOFErrorrQ   )r9   retr;   data	timestampdurationr   r   r   	_get_nextd   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 rH   )rO   r8   rY   rK   r   r   r   _seeky   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 )NrW   rX   raw)r   r   r   )rP   rJ   rZ   rY   r1   r*   _exclusive_fpcloser   r   _Tilesizer3   tilesuperload)r9   rV   	__class__r   r   rc      s   


zWebPImageFile.loadposc                 C  s   d S rH   r   )r9   rf   r   r   r   	load_seek   s   zWebPImageFile.load_seekc                 C  s   | j S rH   )rJ   rD   r   r   r   tell   s   zWebPImageFile.tell)r   r   )r   rA   )rF   rG   r   r   )T)r)   rM   r   r   )r   rR   )r   r[   )rf   rG   r   r   )r   rG   )__name__
__module____qualname__formatformat_descriptionrP   rJ   r@   rE   rL   r8   rY   rZ   rc   rg   rh   __classcell__r   r   rd   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
   )modeconverthas_transparency_data)ro   r   r   r   _convert_frame   s   
rt   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   r2   r   )r   r   r   r   r"      r    rX   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_saver1   
isinstancerG   
getpaletter   ExiftobytestuplelenallOSErrorr   WebPAnimEncoderr`   rh   rangerL   rt   addgetimroundassemblewrite)'ro   r*   rv   r   rx   totalimsr"   palettergbrX   r   rz   r{   r|   r}   verboser~   r   r   r   r&   r'   r(   r;   bg_rbg_gbg_bbg_aenc	frame_idxrW   cur_idxnfridxrF   rV   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 )Nr~   Fr   r   r   r   r&   r   r'       s   Exif     r(   r   r   exactr   r   r   )r   r   r   r   r   r   r   rt   r   
WebPEncoder   floatr   r   )ro   r*   rv   r~   r   r   r&   r'   r(   r   r   rV   r;   r   r   r   r     s8   
r   z.webpz
image/webpN)r   r   r   r   )ro   rp   r   rp   )ro   rp   r*   ru   rv   rw   r   r   )
__future__r   ior   r   r   r   r   r   ImportErrorTYPE_CHECKINGtypingr   r	   r   r   r   rt   r   r   register_openrl   register_saveregister_save_allregister_extensionregister_mimer   r   r   r   <module>   s8    

q

|"