U
    Å½„b*  ã                
   @   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 ]î\Z	Z
z¼e
dkr–d 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 n\e
dkròd 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k
�r@   Y q^Y nX dZ �qVq^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Ú6ZPyQt6Zside6ZPySide6Ú5ZPyQt5Zside2ZPySide2c                 C   s   | d t jkS )Nr   )ÚsysÚmodules)Ú
qt_version© r   ú//tmp/pip-unpacked-wheel-_wxctax1/PIL/ImageQt.pyÚ<lambda>!   ó    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r6z
tj}W q: tk
r2   tj}Y q:X nt}| |j¡ |  ¡ r\|  	|d¡ n|  	|d¡ t
ƒ }| | ¡ ¡ | ¡  | d¡ t |¡S )z1
    :param im: QImage or PIL ImageQt object
    r   ZpngÚppmr   )r   r
   r   ZOpenModeFlagÚAttributeErrorZOpenModeÚopenZ	ReadWriteZhasAlphaChannelÚsaver   ÚwriteÚdataÚcloseÚseekr   )ÚimÚbufferZqt_openmoder   r   r   r   Ú
fromqimage@   s     

r&   c                 C   s   t | ƒS ©N)r&   ©r$   r   r   r   Úfromqpixmap\   s    r)   c                 C   s”   dddddœ| }|| }t |dƒ\}}||r2dnd }| d }|sJ| 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   é   ó    r   )ÚdivmodÚrangeÚlenÚappendÚjoin)ÚbytesÚwidthÚmodeZbits_per_pixelZbits_per_lineZfull_bytes_per_lineZremaining_bits_per_lineZbytes_per_lineZextra_paddingZnew_dataÚir   r   r   Ú
align8to32j   s    
ÿÿr;   c           
      C   s¼  d }d }d}t | dƒr$t|  ¡ dƒ} t| ƒr:t | ¡} d}tdkrHtjnt}| j	dkr`|j
}�n| j	dkr–|j}g }tdƒD ]}| t|||ƒ¡ q|nè| j	d	krä|j}g }|  ¡ }td
t|ƒdƒD ]}| t|||d … Ž ¡ qÂnš| j	dk�r|  d¡} |  dd¡}|j}np| j	dk�r.|  dd¡}|j}nP| j	dk�r\t |dƒ�r\|  dd„ ¡} |j}n"|�rj|  ¡  tdt| j	ƒ› �ƒ‚| j}|�pžt|  ¡ |d
 | j	ƒ}	|�r®|  ¡  |	|||dœS )NFÚtoUtf8zutf-8Tr   r,   r-   é   r.   r   é   ZRGBZRGBAÚrawZBGRAr/   ÚFormat_Grayscale16c                 S   s   | d S )Nr=   r   )r:   r   r   r   r   «   r   z"_toqclass_helper.<locals>.<lambda>zunsupported image mode )r!   ÚsizeÚformatÚ
colortable)ÚhasattrÚstrr<   r   r   r   r
   r   ZFormatr9   ZFormat_MonoZFormat_Indexed8r3   r5   r   Z
getpaletter4   ÚconvertÚtobytesZFormat_RGB32ZFormat_ARGB32Zpointr@   r"   Ú
ValueErrorÚreprrA   r;   )
r$   r!   rC   Zexclusive_fpZ	qt_formatrB   r:   ZpaletterA   Z__datar   r   r   Ú_toqclass_helper†   sP    






rJ   c                       s   e Zd Z‡ fdd„Z‡  ZS )ÚImageQtc                    sT   t |ƒ}|d | _tƒ  | j|d d |d d |d ¡ |d rP|  |d ¡ 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!   rA   r   r   rB   rC   N)rJ   Z_ImageQt__dataÚsuperÚ__init__ZsetColorTable)Úselfr$   Zim_data©Ú	__class__r   r   rM   ½   s    


üzImageQt.__init__)Ú__name__Ú
__module__Ú__qualname__rM   Ú__classcell__r   r   rO   r   rK   ¼   s   rK   c                 C   s   t | ƒS r'   )rK   r(   r   r   r   ÚtoqimageÕ   s    rU   c                 C   s   t | ƒ}t |¡S r'   )rU   r   Z	fromImage)r$   Zqimager   r   r   Ú	toqpixmapÙ   s    rV   )r   )#r   Úior   Ú r   Z_utilr   Zqt_versionsÚsortr
   Z	qt_moduleZPyQt6.QtCorer   r   ZPyQt6.QtGuir   r   r   ZPySide6.QtCoreZPySide6.QtGuiZPyQt5.QtCoreZPyQt5.QtGuiZPySide2.QtCoreZPySide2.QtGuiÚImportErrorÚRuntimeErrorZqt_is_installedr   r&   r)   r;   rJ   rK   rU   rV   r   r   r   r   Ú<module>   sL   ü


4