o
    n¨ iF'  ã                   @  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   Zis_riff_file_formatZis_webp_fileZis_valid_vp8_mode© r   úN/home/app/PyTorch/.pytorch/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   Ú_getexifO   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   ÚseekT   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-   [   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_nextb   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   Ú_seekw   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¡p»d}| j  dd¡}t|t
jƒrÍ| ¡ }| j  dd¡}|rØd}|d u râ|ràdnd}|d u 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} 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 rªd$}t|ƒ‚| |&¡ d S )%NÚappend_imagesr   r*   r   )r   r   r   r   r   é   r   rJ   r   Úminimize_sizeFÚkminÚkmaxÚallow_mixedÚlosslessÚqualityéP   Úalpha_qualityéd   Ú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   rq   r   r   r   r   ÚImportErrorÚTYPE_CHECKINGÚtypingr   r	   r   r   r   ra   r–   r€   Zregister_openr[   Zregister_saveZregister_save_allZregister_extensionZregister_mimer   r   r   r   Ú<module>   s8    ÿý

q

|"ü