o
    pc                  
   @   s  d dl Z d dlm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ziedkrQd dlmZmZ d dlmZmZmZ nPed
krhd dlmZmZ d dlmZmZmZ n9edkrd dlmZmZ d dlmZmZmZ eddd nedkrd dlmZmZ d dlmZmZmZ eddd W n eefy   Y q5w dZ ndZdZd+ddZdd Zdd  Zd!d" Z d#d$ Z!erG d%d& d&eZ"d'd( Z#d)d* Z$dS ),    N)BytesIO   )Image)	deprecate)is_path6ZPyQt6Zside6ZPySide65ZPyQt5Zside2ZPySide2c                 C   s   | d t jv S )Nr   )sysmodules)
qt_version r   +lib/python3.10/site-packages/PIL/ImageQt.py<lambda>"   s    r   T)keyreverse)QBuffer	QIODevice)QImageQPixmapqRgbazSupport for PyQt5
   zPyQt6 or PySide6zSupport for PySide2F   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   rgb>   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   ZpngZppmr   )r   r   r   ZOpenModeFlagAttributeErrorZOpenModeopenZ	ReadWriteZhasAlphaChannelZsaver   writedatacloseseekr   )imbufferZqt_openmoder   r   r   r   
fromqimageE   s"   



r%   c                 C      t | S N)r%   r#   r   r   r   fromqpixmapa      r)   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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   
align8to32o   s   

r=   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=nq| j	d	krq|j}g }|  }td
t|dD ]}|t|||d    q`nJ| j	dkr| d} | dd}|j}n6| j	dkr| dd}|j}n'| j	dkrt |dr| dd } |j}n|r|   dt| j	 }t|| j}	|pt|  |	d
 | j	}
|r|   |
|	||dS )NFtoUtf8zutf-8Tr   r-   r.      r/   r      ZRGBZRGBArawZBGRAr0   Format_Grayscale16c                 S   s   | d S )Nr?   r   )r<   r   r   r   r      s    z"_toqclass_helper.<locals>.<lambda>zunsupported image mode )r    sizeformat
colortable)hasattrstrr>   r   r   r   r   r   ZFormatr;   ZFormat_MonoZFormat_Indexed8r5   r7   r   Z
getpaletter6   ZconverttobytesZFormat_RGB32ZFormat_ARGB32ZpointrB   r!   repr
ValueErrorrC   r=   )r#   r    rE   Zexclusive_fpZ	qt_formatrD   r<   ZpalettemsgrC   Z__datar   r   r   _toqclass_helper   sV   







rL   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    rC   r   r   rD   rE   N)rL   Z_ImageQt__datasuper__init__ZsetColorTable)selfr#   Zim_data	__class__r   r   rO      s   


zImageQt.__init__)__name__
__module____qualname__rO   __classcell__r   r   rQ   r   rM      s    rM   c                 C   r&   r'   )rM   r(   r   r   r   toqimage   r*   rW   c                 C   s   t | }t|S r'   )rW   r   Z	fromImage)r#   Zqimager   r   r   	toqpixmap   s   
rX   )r   )%r	   ior    r   Z
_deprecater   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=   rL   rM   rW   rX   r   r   r   r   <module>   sX   
5