o
    f                  
   @  s  U d dl mZ 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ed	< d
dgddggZejdd dd eD ]X\ZZzCded< ded< ded< ded< ded< edkrjd dlmZmZ d dlmZmZmZ nedkrd 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e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*d+Z#d,d- Z$dS )0    )annotationsN)BytesIO)Callable   )Image)is_pathz
str | None
qt_version6PyQt6side6PySide6c                 C  s   | d t jv S )Nr   )sysmodules)version r   D/home/ubuntu/webapp/venv/lib/python3.10/site-packages/PIL/ImageQt.py<lambda>"   s    r   T)keyreversetypeQBuffer	QIODeviceQImageQPixmapz#Callable[[int, int, int, int], int]qRgba)r   r   )r   r   r   F   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	   pngppmr   )r   r   r   OpenModeFlagAttributeErrorOpenModeopen	ReadWritehasAlphaChannelsaver   writedatacloseseekr   )imbufferqt_openmoder   r   r   r   
fromqimageA   s"   



r1   c                 C     t | S N)r1   r.   r   r   r   fromqpixmap]      r5   c                   sv   ddddd| }|| }t |d\}}||rdnd  d s% S  fddtt  D }d	|S )
zE
    converts each scanline of data from 8 bit to 32 bit aligned
    r         )1LPI;16r      c                   s,   g | ]} | |d    d  qS )r       r   .0ibytesbytes_per_lineextra_paddingr   r   
<listcomp>s   s    zalign8to32.<locals>.<listcomp>    )divmodrangelenjoin)rC   widthmodebits_per_pixelbits_per_linefull_bytes_per_lineremaining_bits_per_linenew_datar   rB   r   
align8to32a   s   

rS   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
}ns| j	dkrA|j}dd	 td
D }na| j	dkr]|j}|    fdd	tdt dD }nE| j	dkrq| d} | dd}|j}n1| j	dkr| dd}|j}n"| j	dkr| dd } |j}n|r|   dt| j	 }t|| j}|pt|  |d | j	}|r|   ||||dS )NFtoUtf8zutf-8Tr	   r9   r:   c                 S  s   g | ]}t |||qS r   r    r?   r   r   r   rF      s    z$_toqclass_helper.<locals>.<listcomp>   r;   c                   s    g | ]}t  ||d    qS )   rU   r?   paletter   r   rF      s     r   rW   RGBRGBArawBGRAr<   c                 S  s   | d S )NrV   r   )rA   r   r   r   r      s    z"_toqclass_helper.<locals>.<lambda>zunsupported image mode )r+   sizeformat
colortable)hasattrstrrT   r   r   r&   r   r   FormatrM   Format_MonoFormat_Indexed8rI   
getpaletterJ   converttobytesFormat_RGB32Format_ARGB32pointFormat_Grayscale16r,   repr
ValueErrorr^   rS   )	r.   r+   r`   exclusive_fp	qt_formatr_   msgr^   __datar   rX   r   _toqclass_helper{   sJ   




 



rs   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+   r^   r   r   r_   r`   N)rs   _ImageQt__datasuper__init__setColorTable)selfr.   im_data	__class__r   r   rw      s   


zImageQt.__init__)__name__
__module____qualname__rw   __classcell__r   r   r{   r   rt      s    rt   returnc                 C  r2   r3   )rt   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   rt   )%
__future__r   r   ior   typingr    r   _utilr   __annotations__qt_versionssortr   	qt_modulePyQt6.QtCorer   r   PyQt6.QtGuir   r   r   PySide6.QtCorePySide6.QtGuiImportErrorRuntimeErrorqt_is_installedr   r    r1   r5   rS   rs   rt   r   r   r   r   r   r   <module>   sT   
1
