U &^W @s<ddlZddlmZddlmZddlmZddgdd ggZejd d d d eD]\Z Z z^e dkrddl m Z m Z mZddlmZmZn,e d krddlm Z m Z mZddlmZmZWneefk rYqRYnXd ZqqRdZdZ d"ddZddZddZddZddZer(Gddde ZddZd d!ZdS)#N)BytesIO)Image)isPath5ZPyQt5Zside2ZPySide2cCs|dtjkS)Nr)sysmodules) qt_versionr //tmp/pip-install-a1j0c_p1/Pillow/PIL/ImageQt.pyr T)keyreverse)QImageqRgbaQPixmap)QBuffer QIODeviceFcCst||||d@S)zA(Internal) Turns an RGB color into a Qt compatible color integer.l)r)rgbar r r rgb.srcCsdt}|tj|r(||dn ||dt}||| | dt |S)zs :param im: A PIL Image object, or a file name (given either as Python string or a PyQt string object) Zpngppmr) ropenrZ ReadWriteZhasAlphaChannelsaverwritedatacloseseekr)imbufferrr r r fromqimage5s   r$cCst|SN)r$r"r r r fromqpixmapKsr'c Csdddd|}||}t|d\}}||r0dnd}| d}|sH|Sg} tt||D]*} | || || d|d|q\d| S)zE converts each scanline of data from 8 bit to 32 bit aligned r)1LPrr )divmodrangelenappendjoin) byteswidthmodeZbits_per_pixelZ bits_per_lineZfull_bytes_per_lineZremaining_bits_per_lineZbytes_per_lineZ extra_paddingZnew_datair r r align8to32Ys r7c Csd}d}t|dr t|d}t|r2t|}|jdkrFtj}n|jdkr|tj }g}t dD]}| t |||qbn|jdkrtj }g}| }t dt|dD]}| t |||dqn|jd kr|d d }tj}nn|jd krHz|d d }Wn:tk r>|\}}}} td |||| f}YnXtj}ntd|j|prt||jd|j} | |||dS)NtoUtf8zutf-8r)r*r+rRGBrawZBGRXRGBAZBGRAzunsupported image mode %r)rr"format colortable)hasattrstrr8rrrr5rZ Format_MonoZFormat_Indexed8r/r1rZ getpaletter0tobytesZ Format_RGB32 SystemErrorsplitmergeZ Format_ARGB32 ValueErrorr7size) r"rr?r>r6ZpaletterrrrZ__datar r r _toqclass_helperus@          rHcseZdZfddZZS)ImageQtcsXt|}|d|_t|j|djd|djd|d|drT||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). rr"rrr>r?N)rHZ_ImageQt__datasuper__init__rGZ setColorTable)selfr"Zim_data __class__r r rKs   zImageQt.__init__)__name__ __module__ __qualname__rK __classcell__r r rMr rIsrIcCst|Sr%)rIr&r r r toqimagesrScCs&|jdkr|d}t|}t|S)Nr;r=)r5convertrSrZ fromImage)r"Zqimager r r toqpixmaps  rU)r)riorrZ_utilrZ qt_versionssortr Z qt_moduleZ PyQt5.QtGuirrrZ PyQt5.QtCorerrZ PySide2.QtGuiZPySide2.QtCore ImportError RuntimeErrorZqt_is_installedrr$r'r7rHrIrSrUr r r r s8      *