U ڲg. @sddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z e dZerjddlmZdZd Zd d Zd d ZddZddZddZGdddZGddde j ZdddddddZdd d!d"d#Ze ejeee ejd$e ejee ejd%e d&kre!ej"d'krBe#d(e$e%ej"dd)Z&ee&Z'e'j(d*D]4Z)e)\Z*Z+Z,e'_)e'-d+e*d,e+d,e,d-qfe %ej"dZ.e.-d.W5QRXej/d/kre0d.W5QRXdS)0) annotationsN)IO)Image ImageFilePngImagePluginfeaturesZjpg_2000)Jpeg2KImagePluginsicnscCstd|tS)Nz>4sI)structunpackread HEADERSIZE)fobjr7/tmp/pip-unpacked-wheel-xela2va7/PIL/IcnsImagePlugin.py nextheader%srcCsH|\}}|||d}|dkr0d}t|t||d|df|S)Nsz'Unknown signature, expecting 0x00000000)seekr SyntaxErrorread_32)r start_lengthsizestartlengthsigmsgrrrread_32t)s  rc Csp|\}}|||d|d|d|df}|d|d}||dkrp||}td||dddd}ntd|d}tdD]} g} |} | dkr|d} | sq| d} | d@r| d } |d} t| D]}| | qn| d} | || | | 8} | dkrqq| dkr:d t| d }t|td |d | dd dd}|j |j | qd|iS)zn Read a 32bit RGB icon resource. Seems to be either uncompressed or an RLE packbits-like scheme. rrRGBrawN}zError reading channel [z left]L) rr r frombuffernewrangeappendreprrjoinimZputband)rrrrr pixel_sizesizesqZindatar,Zband_ixdataZ bytesleftbyte blocksizeirbandrrrr4s@          rc Csf|d}|||d|d|d|df}|d|d}td|||dddd}d|iS)Nrrrr$r!A)rrr&r )rrrrr-r.r3rrrread_mk_s   r5c Cs|\}}|||d}|dddkrT||t|}t|jd|iS|dddks||dddks||dkrtsd }t|||||}t |} t | }t|j|j dkr|d}d|iSd }t|dS) N r sPNG  RGBArsOQs s jP zQUnsupported icon subimage format (rebuild PIL with JPEG 2000 support to fix this)z Unsupported icon subimage format)rr rZ PngImageFilerZ_decompression_bomb_checkr enable_jpeg2k ValueErrorioBytesIOr ZJpeg2KImageFilemodeconvert) rrrrrrr,rZ jp2kstreamfrrrread_png_or_jpeg2000is8            r?c@seZdZdefgdefgdefgdefgdefgdefdefdefgd efgd efgd efd efgd efdefdefgdefgdefdefdefgd ZddZddZ ddZ ddZ d ddZ dS)!IcnsFileic10ic09ic14ic08ic13ic07sit32st8mksicp6ic12sih32sh8mksicp5sil32sl8mkic11sicp4sis32ss8mk) )rIr)rIrIr)rJr)rJrJr)r"r"r)r"r"r)@rKr) rLr)0rMr)rLrLr)rNr)rNrNrcCsi|_}||_t|\}}t|s0d}t|t}||krt|\}}|dkr\d}t||t7}|t8}||f||<||tj||7}q4dS)z@ fobj is a file-like object as an icns resource znot an icns filerzinvalid block headerN) dctrr_acceptrrrr:SEEK_CUR)selfrrOrfilesizerr2r1rrr__init__s"    zIcnsFile.__init__cCsBg}|jD].\}}|D] \}}||jkr||qqq|SN)SIZESitemsrOr))rRsizesrZfmtsfmtreaderrrr itersizess   zIcnsFile.itersizescCs |}|sd}t|t|S)NzNo 32bit icon resources found)r[rmax)rRrXrrrrbestsizes zIcnsFile.bestsizecCsDi}|j|D]0\}}|j|}|dk r|||j||q|S)z Get an icon resource as {channel: array}. Note that the arrays are bottom-up like windows bitmaps and will likely need to be flipped or transposed in some way. N)rVrOgetupdater)rRrrOcoderZdescrrr dataforsizes  zIcnsFile.dataforsizeNcCs|dkr|}t|dkr.|d|ddf}||}|dd}|rL|S|d}z||dWntk rYnX|S)Nrrrr7r r4)r]lenrbr^copyZputalphaKeyError)rRrZchannelsr,rrrgetimages   zIcnsFile.getimage)N) __name__ __module__ __qualname__r?rr5rrVrTr[r]rbrfrrrrr@s6  r@c@sHeZdZdZdZdZddddZedd Zej d d Zd d Z d S) IcnsImageFilea! PIL image support for Mac OS .icns files. Chooses the best resolution, but will possibly load a different size image if you mutate the size attribute before calling 'load'. The info dictionary has a key 'sizes' that is a list of sizes that the icns file has. ZICNSzMac OS icns resourceNone)returncCs\t|j|_d|_|j|jd<|j|_|jd|jd|jd|jdf|_dS)Nr7rXrrr) r@fpicns_moder[infor] best_sizerrRrrr_opens  zIcnsImageFile._opencCs|jSrU)_sizerrrrrr szIcnsImageFile.sizecCs|}||jdkr0t|dkr0|d|ddf}||jdkrt|dkr|ddkrdd|jdD}||kr|jd||}||jdkrd}t|||_dS) NrXrrrrcSs,g|]$}|d|d|d|dfqS)rrrr).0rrrr sz&IcnsImageFile.size..z2This is not one of the allowed sizes of this image)rprcindexr9rt)rRvalueZ info_sizeZ simple_sizesrrrrrs$   cCst|jdkr@|j|_|jd|jd|jd|jdf|_tj|}|jdk rh|jj|jkrh|S||j|j}|}|j|_|j |_ |j|_|S)Nrrrr) rcrrqrloadr,Z load_preparernrfr<ro)rRZpxr,rrrry"s zIcnsImageFile.loadN) rgrhri__doc__formatformat_descriptionrspropertyrsetterryrrrrrjs    rjz Image.Imagez IO[bytes]z str | bytesrk)r,rmfilenamerlc Cst|dr|ddddddddd}d d |jd gD}i}t|D]B}||krd||n |||f}t}| |d | ||<qPg} | D](\} }||} | | t t| | fq|tt } | t d t| 7} | tdd| D7} |td| |d|tdt t| t | D]*} || d|td| dq@| D]8} || d|td| d|| dqpt|dr|dS)za Saves the image as a series of PNG files, that are then combined into a .icns file. flushr"rJrIirLrK)rFrDrBrArHrGrErCcSsi|] }|j|qSr)width)rur,rrr Nsz_save..Z append_imagesZpngr css|]}|dVqdS)rNr)ruentryrrr dsz_save..z>isTOC rrrN)hasattrrZ encoderinfor^setvaluesresizer:r;savegetvaluerWr)rrcwriteMAGICsumr pack)r,rmrrXZprovided_imagesZ size_streamsrimagetempentriestypestreamZ file_lengthrrrr_save<sR        rbytesbool)prefixrlcCs|ddtkS)Nr)r)rrrrrPxsrPz.icnsz image/icns__main__rz)Syntax: python3 IcnsImagePlugin.py [file]rbrXzout--z.pngzout.pngZwindows)1 __future__rr:osr systypingrrrrrZ check_codecr8r rrrrrr5r?r@rjrrPZ register_openr{Zregister_extensionZ register_saveZ register_mimergrcargvprintexitopenrmZimfrprrheightZscalerr,platformZ startfilerrrrsJ     + "gJ< "