o
�J�h�( � @ s� d dl mZ d dlmZ d dlmZmZ ddlmZm Z z
ddlm
Z
dZ W n e y1 dZ Y nw d d
d
d �Z
d$dd�ZG dd� de j �Zd%dd�Zd&d d �Zd&d d �Ze�ejee� e r}e�eje� e�eje� e�ejd!� e�ejd"� d#S d#S )'� )� annotations)�BytesIO)�IO�Any� )�Image� ImageFile)�_webpTF�RGB�RGBA)s VP8 s VP8Xs VP8L�prefix�bytes�return�
bool | strc C sN | d d� dk}| dd� dk}| dd� t v }|r%|r%|r%ts# dS dS d S )
N� s RIFF� � s WEBP� zEimage file could not be identified because WEBP support not installedTF)�_VP8_MODES_BY_IDENTIFIER� SUPPORTED)r Zis_riff_file_formatZ is_webp_fileZis_valid_vp8_mode� r �QC:\pinokio\api\whisper-webui.git\app\env\lib\site-packages\PIL\WebPImagePlugin.py�_accept s �r c s� e Zd ZdZdZdZdZd!dd�Zd"d d
�Zd#d
d�Z d$d%dd�Z
d&dd�Z d#dd�Z d'� fdd� Z
d(d d �Zd)d d �Z� ZS )*�
WebPImageFile�WEBPz
WebP imager r �Nonec C s t �| j�� �| _| j�� \}}}}}}||f| _|| jd< |d? d@ |d? d@ |d? d@ |d@ f\}}} }
|| |
|f| jd< || _| jdk| _ |dkrNd n|| _
|| _ g | _ | j�
d
�} | j�
d �} | j�
d �}
| rp| | jd
< | rw| | jd< |
r~|
| jd< | jdd� d S )N�loop� �� r r �
backgroundr �RGBXr
ZICCPZEXIFzXMP � icc_profile�exif�xmpF)�reset)r ZWebPAnimDecoder�fp�read�_decoder�get_info�_size�info�n_frames� is_animated�_mode�rawmode�tileZ get_chunk�_reset)�self�width�heightZ
loop_count�bgcolor� frame_count�mode�bg_a�bg_r�bg_g�bg_br! r" r# r r r �_open+ s2
�
zWebPImageFile._open�dict[int, Any] | Nonec C s d| j vrd S | �� �� S )Nr" )r* �getexif�_get_merged_dict�r1 r r r �_getexifO s
zWebPImageFile._getexif�frame�intc C s | � |�sd S || _d S �N)� _seek_check� _WebPImageFile__logical_frame�r1 rA r r r �seekT s
zWebPImageFile.seekTr$ �boolc C s$ |r| j �� d| _d| _d| _d S )Nr �����)r' r$ � _WebPImageFile__physical_frame�_WebPImageFile__loaded�_WebPImageFile__timestamp)r1 r$ r r r r0 [ s
zWebPImageFile._reset�tuple[bytes, int, int]c C sh | j �� }| jd7 _|d u r | �� | �d� d}t|��|\}}|| j }|| _||8 }|||fS )Nr r z(failed to decode next frame in WebP file)r' Zget_nextrJ r0 rG �EOFErrorrL )r1 �ret�msg�data� timestamp�durationr r r � _get_nextb s
zWebPImageFile._get_nextc C sD | j |krd S || j k r| �� | j |k r | �� | j |k sd S d S rC )rJ r0 rT rF r r r �_seekw s
�zWebPImageFile._seek� Image.core.PixelAccess | Nonec s� | j | jkr?| �| j� | �� \}}}|| jd< || jd< | j| _ | jr,| jr,| j�� t|�| _t �
dd| j d| j �g| _
t� �� S )NrR rS �raw)r r r )rK rE rU rT r* r% �
_exclusive_fp�closer r �_Tile�sizer. r/ �super�load)r1 rQ rR rS �� __class__r r r] s
zWebPImageFile.load�posc C s d S rC r )r1 r` r r r � load_seek� s zWebPImageFile.load_seekc C s | j S rC )rE r? r r r �tell� s zWebPImageFile.tell)r r )r r< )rA rB r r )T)r$ rH r r )r rM )r rV )r` rB r r )r rB )�__name__�
__module__� __qualname__�format�format_descriptionrK rE r; r@ rG r0 rT rU r] ra rb �
__classcell__r r r^ r r % s
$
r �im� Image.Imagec C s" | j dvr| �| jr dnd�} | S )N)r r r
r r
)r6 �convert�has_transparency_data)ri r r r �_convert_frame� s
rm r% � IO[bytes]�filename� str | bytesr c ' C sn | j �� }t|�dg ��}d}| g| D ]
}|t|dd�7 }q|dkr+t| ||� d S d}d|v r6|d }n1d| jv rg| jd }t|t�rg| � � }|ra||d |d d � \} }
} | |
| df}n|||df}| j �d | j�d d��} | j �d
d�}
| j �d d �}| j �d
d �}| j �dd �}| j �dd �}d }| j �dd �}| j �dd�}| j �dd�}| j �dd�}| j �d�p�d}| j �dd�}t|t
j �r�|� � }| j �dd�}|r�d }|d u r�|r�dnd}|d u r�|r�dnd }t|tt
f��rt|�d k�std d � |D ���sd |� �}t|��|\}} } } | d!> |d"> B | d#> B | d> B }t�| jd | jd ||
|||||� } d} d}!| �� }"zL| g| D ]?}t|dd�}#t|#�D ]1}$|�|$� t|�}%| �|%�� t|!�||||� t| tt
f��ry|!| | 7 }!n|!| 7 }!| d7 } �qQ�qEW | �|"� n| �|"� w | �d t|!�|||d� | �|||�}&|&d u �r�d$}t|��|� |&� d S )%N�
append_imagesr r+ r )r r r r r � r rS r �
minimize_sizeF�kmin�kmax� allow_mixed�lossless�quality�P �
alpha_quality�d �methodr! � r" r# � � � r c s s( � | ]}d | ko
dk n V qdS )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* �
isinstancerB �
getpaletter �Exif�tobytes�tuple�len�all�OSErrorr ZWebPAnimEncoderr[ rb �rangerG rm �add�getim�round�assemble�write)'ri r% ro r� rq �total�imsr �palette�r�g�brS r rs rt ru rv �verboserw rx rz r| r! r" r# rP r8 r9 r: r7 �encZ frame_idxrR �cur_idx�nfr�idxrA rQ r r r � _save_all� s�
��
�
�
��
r� c
C s� | j �dd�}| j �dd�}| j �dd�}| j �d�p d}| j �d d
�}t|tj�r.|�� }|�d �r9|d d � }| j �d
d�}| j �dd�} | j �d�rOdnd}
t| �} t� | �
� |t |�t |�|| |
||� } | d u rrd} t | ��|�
| � d S )Nrw Frx ry rz r{ r! r} r" � s Exif � r# r| r �exactr r r� )r� r� r� r r� r� �
startswithrm r Z
WebPEncoder� �floatr� r� )
ri r% ro rw rx rz r! r" r# r| r� rQ rP r r r r� s8
� r� z.webpz
image/webpN)r r
r r )ri rj r rj )ri rj r% rn ro rp r r )�
__future__r �ior �typingr r r} r r r r � ImportErrorr r r rm r� r� �
register_openrf �
register_save�register_save_all�register_extension�
register_mimer r r r �<module> s4 ��
s
}"�