U ڲg @sUddlmZddlZddlmZddlmZddlmZddl m Z de d <d d gd d ggZ e j dddde D]\ZZzde d<de d<de d<de d<de d<ed krddlmZmZddlmZmZmZn.ed krddlmZmZddlmZmZmZWneefk r"YqtYnXdZeZq"rT)keyreversetypeQBuffer QIODeviceQImageQPixmapz#Callable[[int, int, int, int], int]qRgba)rr)rrrFcCst||||d@S)zA(Internal) Turns an RGB color into a Qt compatible color integer.l)r)rgbar r rrgb:srcCst}tdkr6z tj}Wq:tk r2tj}Yq:Xnt}||j|r\| |dn | |dt }| | | |dt|S)z1 :param im: QImage or PIL ImageQt object r Zpngppmr)rrrZ OpenModeFlagAttributeErrorZOpenModeopenZ ReadWriteZhasAlphaChannelsaverwritedatacloseseekr)imbufferZ qt_openmoderr r r fromqimageAs     r)cCst|SN)r)r'r r r fromqpixmap]sr,csvddddd|}||}t|d\}}||r2dnd dsJSfddttD}d |S) zE converts each scanline of data from 8 bit to 32 bit aligned r)1LPI;16rcs,g|]$}||ddqS)rr .0ibytesZbytes_per_lineZ extra_paddingr r sszalign8to32..r)divmodrangelenjoin)r9widthmodeZbits_per_pixelZ bits_per_lineZfull_bytes_per_lineZremaining_bits_per_lineZnew_datar r8r align8to32as rAc sd}d}d}t|dr$t|d}t|r:t|}d}tdkrHtjnt}|j dkr^|j }n|j dkr|j }dd t d D}n|j d kr|j }| fd d t d tdD}n|j dkr|d}|dd}|j}nh|j dkr|dd}|j}nH|j dkr$|dd}|j}n&|r2|dt|j }t||j}|pjt||d |j }|rz|||||dS)NFtoUtf8zutf-8Tr r/r0cSsg|]}t|||qSr rr5r r rr:sz$_toqclass_helper..r1cs g|]}t||dqS)rCr5Zpaletter rr:srrEZRGBZRGBArawZBGRAr2cSs|dS)NrDr )r7r r rrrz"_toqclass_helper..zunsupported image mode )r$sizeformat colortable)hasattrstrrBrrr!rrZFormatr@Z Format_MonoZFormat_Indexed8r<Z getpaletter=converttobytesZ Format_RGB32Z Format_ARGB32ZpointZFormat_Grayscale16r%repr ValueErrorrHrA) r'r$rJZ exclusive_fpZ qt_formatrImsgrHZ__datar rFr_toqclass_helper{sJ            rRcseZdZfddZZS)ImageQtcsTt|}|d|_t|j|dd|dd|d|drP||ddS)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$rHrrrIrJN)rRZ_ImageQt__datasuper__init__Z setColorTable)selfr'Zim_data __class__r rrUs   zImageQt.__init__)__name__ __module__ __qualname__rU __classcell__r r rWrrSsrS)returncCst|Sr*)rSr+r r rtoqimagesr^cCst|}t|Sr*)r^rZ fromImage)r'Zqimager r r toqpixmapsr_)r)% __future__rr iortypingrrZ_utilr__annotations__Z qt_versionssortr Z qt_moduleZ PyQt6.QtCorerrZ PyQt6.QtGuirrrZPySide6.QtCoreZ PySide6.QtGui ImportError RuntimeErrorZqt_is_installedrrr)r,rArRrSr^r_r r r rsP         1