o
    Z#ûa*  ã                
   @   sš  d dl Z d dlmZ ddlmZ ddlmZ ddgdd	gd
dgddggZejdd„ dd� eD ]p\Z	Z
z]e
dkrKd dlmZmZ d dlmZmZmZ nDe
d	krbd dlmZmZ d dlmZmZmZ n-e
dkryd dlmZmZ d dlmZmZmZ ne
dkr�d dlmZmZ d dlmZmZmZ W n eefy›   Y q/w dZ ndZdZ	d&dd„Zdd„ Zdd„ Zdd„ Zdd„ ZerÃG d d!„ d!eƒZ d"d#„ Z!d$d%„ Z"dS )'é    N)ÚBytesIOé   )ÚImage)ÚisPathÚ6ÚPyQt6Úside6ÚPySide6Ú5ÚPyQt5Úside2ÚPySide2c                 C   s   | d t jv S )Nr   )ÚsysÚmodules)Ú
qt_version© r   ú-/usr/lib/python3/dist-packages/PIL/ImageQt.pyÚ<lambda>!   s    r   T)ÚkeyÚreverse)ÚQBufferÚ	QIODevice)ÚQImageÚQPixmapÚqRgbaFéÿ   c                 C   s   t | |||ƒd@ S )zA(Internal) Turns an RGB color into a Qt compatible color integer.l   ÿÿ )r   )ÚrÚgÚbÚar   r   r   Úrgb9   s   r    c                 C   s”   t ƒ }tdkrztj}W n ty   tj}Y nw t}| |j¡ |  ¡ r,|  	|d¡ n|  	|d¡ t
ƒ }| | ¡ ¡ | ¡  | d¡ t |¡S )z1
    :param im: QImage or PIL ImageQt object
    r   ÚpngÚppmr   )r   r   r   ÚOpenModeFlagÚAttributeErrorÚOpenModeÚopenÚ	ReadWriteÚhasAlphaChannelÚsaver   ÚwriteÚdataÚcloseÚseekr   )ÚimÚbufferÚqt_openmoder   r   r   r   Ú
fromqimage@   s"   

ÿ

r1   c                 C   ó   t | ƒS ©N)r1   ©r.   r   r   r   Úfromqpixmap\   ó   r5   c                 C   s”   dddddœ| }|| }t |dƒ\}}||rdnd }| d }|s%| S g }	tt| ƒ| ƒD ]}
|	 | |
| |
d | … d|  ¡ q/d |	¡S )	zE
    converts each scanline of data from 8 bit to 32 bit aligned
    r   é   é   )Ú1ÚLÚPúI;16r   é   ó    ó    )ÚdivmodÚrangeÚlenÚappendÚjoin)ÚbytesÚwidthÚmodeÚbits_per_pixelÚbits_per_lineÚfull_bytes_per_lineÚremaining_bits_per_lineÚbytes_per_lineÚextra_paddingÚnew_dataÚir   r   r   Ú
align8to32j   s   
ÿÿ
rP   c           
      C   s¬  d }d }d}t | dƒrt|  ¡ dƒ} t| ƒrt | ¡} d}tdkr$tjnt}| j	dkr/|j
}nŠ| j	dkrJ|j}g }tdƒD ]}| t|||ƒ¡ q=no| j	d	krq|j}g }|  ¡ }td
t|ƒdƒD ]}| t|||d … Ž ¡ q`nH| j	dkr…|  d¡} |  dd¡}|j}n4| j	dkr”|  dd¡}|j}n%| j	dkr©t |dƒr©|  dd„ ¡} |j}n|r¯|  ¡  tdt| j	ƒ› �ƒ‚| j}|pÈt|  ¡ |d
 | j	ƒ}	|rÏ|  ¡  |	|||dœS )NFÚtoUtf8zutf-8Tr   r9   r:   é   r;   r   é   ÚRGBÚRGBAÚrawÚBGRAr<   ÚFormat_Grayscale16c                 S   s   | d S )NrR   r   )rO   r   r   r   r   «   s    z"_toqclass_helper.<locals>.<lambda>zunsupported image mode )r+   ÚsizeÚformatÚ
colortable)ÚhasattrÚstrrQ   r   r   r&   r   r   ÚFormatrG   ÚFormat_MonoÚFormat_Indexed8rA   rC   r    Ú
getpaletterB   ÚconvertÚtobytesÚFormat_RGB32ÚFormat_ARGB32ÚpointrX   r,   Ú
ValueErrorÚreprrY   rP   )
r.   r+   r[   Úexclusive_fpÚ	qt_formatrZ   rO   ÚpaletterY   Ú__datar   r   r   Ú_toqclass_helper†   sT   



ÿ
ÿ


rm   c                       s   e Zd Z‡ fdd„Z‡  ZS )ÚImageQtc                    sX   t |ƒ}|d | _tƒ  | j|d d |d d |d ¡ |d r*|  |d ¡ dS dS )zò
            An PIL image wrapper for Qt.  This is a subclass of PyQt's QImage
            class.

            :param im: A PIL Image object, or a file name (given either as
                Python string or a PyQt string object).
            r+   rY   r   r   rZ   r[   N)rm   Ú_ImageQt__dataÚsuperÚ__init__ÚsetColorTable)Úselfr.   Úim_data©Ú	__class__r   r   rq   ½   s   


üÿzImageQt.__init__)Ú__name__Ú
__module__Ú__qualname__rq   Ú__classcell__r   r   ru   r   rn   ¼   s    rn   c                 C   r2   r3   )rn   r4   r   r   r   ÚtoqimageÕ   r6   r{   c                 C   s   t | ƒ}t |¡S r3   )r{   r   Ú	fromImage)r.   Úqimager   r   r   Ú	toqpixmapÙ   s   
r~   )r   )#r   Úior   Ú r   Ú_utilr   Úqt_versionsÚsortr   Ú	qt_moduleÚPyQt6.QtCorer   r   ÚPyQt6.QtGuir   r   r   ÚPySide6.QtCoreÚPySide6.QtGuiÚPyQt5.QtCoreÚPyQt5.QtGuiÚPySide2.QtCoreÚPySide2.QtGuiÚImportErrorÚRuntimeErrorÚqt_is_installedr    r1   r5   rP   rm   rn   r{   r~   r   r   r   r   Ú<module>   sR   ü€ÿ
4