o
�J�h� � @ s� d dl mZ d dlZddlmZmZ ddlmZ dd
d �Z G d d
� d
ej�Z
G dd� dej �Z e�
e
je
e � e�de � e�e
jd� dS )� )� annotationsN� )�Image� ImageFile)�i32be�prefix�bytes�return�boolc C s | d d� dkS )N� s qoif� )r r r �PC:\pinokio\api\whisper-webui.git\app\env\lib\site-packages\PIL\QoiImagePlugin.py�_accept s r c @ s e Zd ZdZdZddd�ZdS ) � QoiImageFileZQOIzQuite OK Imager �Nonec C s� t | j�d��sd}t|��t| j�d��t| j�d��f| _| j�d�d }|dkr-dnd| _| j�dtj � t
� dd | j | j� � d �g| _
d S )
Nr znot a QOI filer r � �RGB�RGBA�qoi)r r )r �fp�read� SyntaxError�i32�_size�_mode�seek�os�SEEK_CURr �_Tile�tell�tile)�self�msg�channelsr r r
�_open s "$zQoiImageFile._openN)r r )�__name__�
__module__� __qualname__�format�format_descriptionr$ r r r r
r s r c @ s> e Zd ZU dZdZded< i Zded< dd d �Zddd�ZdS )�
QoiDecoderTNzbytes | bytearray | None�_previous_pixelz dict[int, bytes | bytearray]�_previously_seen_pixels�value�bytes | bytearrayr r c C sD || _ |\}}}}|d |d |d |d d }|| j|< d S )Nr � � � �@ )r+ r, )r! r- �r�g�b�a�
hash_valuer r r
�_add_to_previous_pixels+ s $z"QoiDecoder._add_to_previous_pixels�buffer�$bytes | Image.SupportsArrayInterface�tuple[int, int]c
sN �j d usJ �i �_��td�� t� }t��j�}�jj�jj | }t
|�|k �r �j � d�d }|dkrJ�j rJt�j � d���j dd � }n�|dkrU�j � d�}n�|d? }|dkrk|d @ }�j�
|td
��}n�|dkr��j r�t�j d |d @ d? d d
�j d |d@ d ? d d
�j d |d@ d d
�j d f�}nc|d kr�j r�j � d�d } |d @ d � | d@ d? d }
| d@ d } tt� �fdd�t|
d| f�D ���}|�j dd � 7 }n#|dk�r�j �r|d @ d } �j }|dkr�|d d� }||| 7 }q$��|� |dk�r|d d� }||7 }t
|�|k s+��|� dS )N)r r r �� r r �� r r< r � �? )r r r r �0 � � � � �� � � c 3 s* � | ]\}}�j | � | d V qdS )rB N)r+ )�.0�i�diff�Z
diff_greenr! r r
� <genexpr>[ s
� �
�z$QoiDecoder.decode.<locals>.<genexpr>)�����r )�fdr, r8 � bytearrayr � getmodebands�mode�state�xsize�ysize�lenr r+ �get�tuple� enumerate�
set_as_raw)
r! r9 �data�bands� dest_length�byter- �opZop_indexZ second_byteZdiff_redZ diff_blueZ
run_lengthr rK r
�decode2 sn �����
��
�
2zQoiDecoder.decode)r- r. r r )r9 r: r r; ) r% r& r' � _pulls_fdr+ �__annotations__r, r8 r_ r r r r
r* & s
r* r z.qoi)r r r r
)�
__future__r r � r r �_binaryr r r r � PyDecoderr* �
register_openr( �register_decoder�register_extensionr r r r
�<module> s
K