File: //lib/python3/dist-packages/PIL/__pycache__/QoiImagePlugin.cpython-312.pyc
�
��eV � �6 � d dl mZ d dlZddlmZmZ ddlmZ ddlm Z d� Z
G d� d ej
� Z G d
� d ej � Z
ej e j e e
� ej d e
� ej" e j d
� y)� )� annotationsN� )�Image� ImageFile)�i32be��o8c � � | d d dk( S )N� s qoif� )�prefixs �4/usr/lib/python3/dist-packages/PIL/QoiImagePlugin.py�_acceptr s � � �"�1�:�� � � c � � e Zd ZdZdZd� Zy)� QoiImageFile�QOIzQuite OK Imagec � � � t � j j d� � s
d}t |� �t � fd�t d� D � � � _ � j j d� d }|dk( rdnd � _ � j j dt j � d
d � j z � j j � d fg� _ y ) Nr znot a QOI filec 3 �f �K � | ]( }t �j j d � � �� �* y�w)r N)�i32�fp�read)�.0�i�selfs �r � <genexpr>z%QoiImageFile._open.<locals>.<genexpr> s" �� �� �B�A�3�t�w�w�|�|�A��/�B�s �.1� r r � �RGB�RGBA�qoi)r r )
r r r � SyntaxError�tuple�range�_size�_mode�seek�os�SEEK_CUR�tell�tile)r �msg�channelss ` r �_openzQoiImageFile._open s� �� ��t�w�w�|�|�A��'�"�C� �c�"� "��B��q��B�B��
��7�7�<�<��?�1�%�� &�!�m�U���
� ��� � �Q�� � �$��V�d�j�j� 0�$�'�'�,�,�.�$�G�H�� r N)�__name__�
__module__� __qualname__�format�format_descriptionr. r r r r r s � �
�F�)�� Ir r c � � e Zd ZdZd� Zd� Zy)�
QoiDecoderTc �r � || _ |\ }}}}|dz |dz z |dz z |dz z dz }|| j |< y )Nr � � � �@ )�_previous_pixel�_previously_seen_pixels)r �value�r�g�b�a�
hash_values r �_add_to_previous_pixelsz"QoiDecoder._add_to_previous_pixels* sO � � $�� ��
��1�a���!�e�a�!�e�m�a�!�e�+�a�"�f�4��:�
�38��$�$�Z�0r c � � � � i � _ d � _ � j dj d� dD � � � t � }t j � j � }t |� � j j � j j z |z k �r$� j j d� d }|dk( r-� j j d� � j dd z }�nv|dk( r � j j d � }�nT|d
z }|dk( r#|d z }� j j |d � }�n|dk( rj� j d |d
z d z z dz
dz � j d |dz dz z dz
dz � j d |dz z dz
dz f}|� j d fz
}n�|dk( rn� j j d� d }|d z dz
� |dz d z dz
} |dz dz
}
t � � fd�t! | d|
f� D � � }|� j d fz
}n-|dk( r(|d z dz } � j }|dk( r|d d }||| z z
}���dj d� D � � }� j |� |dk( r|d d }||z
}t |� � j j � j j z |z k r��$� j# t% |� � y)Nr c 3 �2 K � | ] }t |� �� � y �w�Nr �r r s r r z$QoiDecoder.decode.<locals>.<genexpr>4 s � �� �-L��b��e�-L�� �)r r r � r r � r rI r � �? )r r r r �0 r � � � �� � � c 3 �T �K � | ] \ }}�j | �z |z d z �� �! y�w)rN N)r; )r r �diff�
diff_greenr s ��r r z$QoiDecoder.decode.<locals>.<genexpr>R s7 �� �� � "� #�A�t� �-�-�a�0�:�=��D��K�"�s �%(c 3 �2 K � | ] }t |� �� � y �wrF r rG s r r z$QoiDecoder.decode.<locals>.<genexpr>^ s � �� � 6�1��A�� 6�rH )���r )r< r; rC �join� bytearrayr � getmodebands�mode�len�state�xsize�ysize�fdr �getr# � enumerate�
set_as_raw�bytes)
r �buffer�data�bands�byter= �op�op_index� second_byte�diff_red� diff_blue�
run_lengthrV s
` @r �decodezQoiDecoder.decode1 s� �� �')��$� #�� � �$�$�S�X�X�-L�^�-L�%L�M��{���"�"�4�9�9�-���$�i�$�*�*�*�*�T�Z�Z�-=�-=�=��E�E��7�7�<�<��?�1�%�D��z�!� ��� � �Q��$�*>�*>�q�r�*B�B����#� ��� � �Q����Q�Y����7� #�j� 0�H� � 8� 8� <� <�X�|� T�E��1�W� �-�-�a�0�T�J�5F�1�4L�M�PQ�Q� � � �-�-�a�0�T�J�5F�1�4L�M�PQ�Q� � � �-�-�a�0�D�:�4E�F��J�c�Q� �E� �d� 2� 2�1� 5� 7�7�E��1�W�"&�'�'�,�,�q�/�!�"4�K�"&��"3�r�!9�J�!,�z�!9�a� ?�1� D�H�!,�z�!9�Q� >�I� !� "�'0�(�A�y�1I�'J�"� �E� �d� 2� 2�1� 5� 7�7�E��1�W�"&��"3�q�!8�J� � 0� 0�E� ��z� %�b�q� ���E�J� .�.�D� ���� 6�� 6�6�� � (� (�� /���z� �b�q� �� �E�M�D�W �$�i�$�*�*�*�*�T�Z�Z�-=�-=�=��E�E�X
����d� �$�r N)r/ r0 r1 � _pulls_fdrC rp r r r r5 r5 ' s � ��I�9�4r r5 r! z.qoi)�
__future__r r( � r r �_binaryr r r r r � PyDecoderr5 �
register_openr2 �register_decoder�register_extensionr r r �<module>ry s� �� #� � � !� �!�I�9�&�&� I�$>��$�$� >�B �� � �L�'�'� �w� ?� �� � �u�j� )� �� � � �,�,�f� 5r