U 0[c- @sddlZddlZddlZddlZddlmZmZmZmZe dZ e rRddlm Z dZ dZ ddZd d Zd d Zd dZddZGdddZGdddejZddZddZeejeeeejdeejeeejdedkreejdkrede e!ejddpZ"ee"Z#e#j$d D]Z%e%e#_%e#&d!e%q:e!ejdZ'e'&d"W5QRXej(d#kre)d"W5QRXdS)$N)Image ImageFilePngImagePluginfeaturesZjpg_2000)Jpeg2KImagePluginsicnscCstd|tS)Nz>4sI)structZunpackread HEADERSIZE)fobjr 2lib/python3.8/site-packages/PIL/IcnsImagePlugin.py nextheader#srcCsD|\}}|||d}|dkr,tdt||d|df|S)Nsz'Unknown signature, expecting 0x00000000)seekr SyntaxErrorread_32)r start_lengthsizestartlengthsigr r r read_32t's   rc Csl|\}}|||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| dkr6td t| d 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. rRGBrawN}zError reading channel [z left]L) rr r frombuffernewrangeappendrreprjoinimZputband)r rrrr pixel_sizesizesqZindatar(Zband_ixdataZ bytesleftZbyte blocksizeibandr r r r1s>          rc Csf|d}|||d|d|d|df}|d|d}td|||dddd}d|iS)Nrrrr rA)rrr"r )r rrrr)r*r.r r r read_mk[s   r0c Cs|\}}|||d}|dddkrT||t|}t|jd|iS|dddks||dddks||dkrtstd ||||}t |}t |}t|j|j dkr|d}d|iStd dS) N rsPNG  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 ValueErrorioBytesIOrZJpeg2KImageFilemodeZconvert) r rrrrrr(Z jp2kstreamfr r r read_png_or_jpeg2000es6            r9c@seZdZdefgdefgdefgdefgdefgdefdefdefgd efgd efgd efd efgd efdefdefgdefgdefdefdefgd ZddZddZ ddZ ddZ d ddZ dS)!IcnsFileic10ic09ic14ic08ic13ic07sit32st8mksicp6ic12sih32sh8mksicp5sil32sl8mkic11sicp4sis32ss8mk) )rCr)rCrCr)rDr)rDrDr)rrr)rrr)@rEr) rFr)0rGr)rFrFr)rHr)rHrHrcCsi|_}||_t|\}}t|s,tdt}||krt|\}}|dkrTtd|t7}|t8}||f||<||tj||7}q0dS)z@ fobj is a file-like object as an icns resource znot an icns filerzinvalid block headerN) dctr r_acceptrr rr5SEEK_CUR)selfr rIrZfilesizer-r,r r r __init__s    zIcnsFile.__init__cCsBg}|jD].\}}|D] \}}||jkr||qqq|SN)SIZESitemsrIr%)rLsizesrZfmtsZfmtreaderr r r itersizess   zIcnsFile.itersizescCs|}|stdt|S)NzNo 32bit icon resources found)rSrmax)rLrQr r r bestsizeszIcnsFile.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)rOrIgetupdater )rLrrIcoderRZdescr r r dataforsizes  zIcnsFile.dataforsizeNcCs|dkr|}t|dkr.|d|ddf}||}|dd}|rL|S|d}z||dWntk rYnX|S)Nrrrr2rr/)rUlenrYrVcopyZputalphaKeyError)rLrZchannelsr(r r r getimages   zIcnsFile.getimage)N) __name__ __module__ __qualname__r9rr0rrOrMrSrUrYr]r r r r r:s6  r:c@sBeZdZdZdZdZddZeddZej ddZd 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 resourcecCs\t|j|_d|_|j|jd<|j|_|jd|jd|jd|jdf|_dS)Nr2rQrrr) r:fpicnsr7rSinforU best_sizerrLr r r _opens  zIcnsImageFile._opencCs|jSrN)_sizerfr r r rszIcnsImageFile.sizecCs|}||jdkr0t|dkr0|d|ddf}||jdkrt|dkr|ddkrdd|jdD}||kr|jd||}||jdkrtd||_dS) NrQrrrrcSs,g|]$}|d|d|d|dfqS)rrrr ).0rr r r sz&IcnsImageFile.size..z2This is not one of the allowed sizes of this image)rdrZindexr4rh)rLvalueZ info_sizeZ simple_sizesr r r rs"   cCst|jdkr@|j|_|jd|jd|jd|jdf|_tj|}|jdk rh|jj|jkrh|S||j|j}|}|j|_|j |_ |j|_|S)Nrrrr) rZrrerloadr(Z load_preparercr]r7)rLZpxr(r r r rms zIcnsImageFile.loadN) r^r_r`__doc__formatZformat_descriptionrgpropertyrsetterrmr r r r ras    rac Cst|dr|ddddddddd}d d |jd gD}i}t|D]B}||krd||n |||f}t}| |d | ||<qPg} | D]*\} }||} | | t t| | d q|tt } | t dt| 7} | tdd| D7} |td| |d|tdt t| t | D]*} || d|td| dqB| D]8} || d|td| d|| dqrt|dr|dS)za Saves the image as a series of PNG files, that are then combined into a .icns file. flushrrDrCirFrE)r@r>r<r;rBrAr?r=cSsi|] }|j|qSr )width)rir(r r r Esz_save..Z append_imagesZpng)typerstreamrcss|]}|dVqdS)rNr )rientryr r r ]sz_save..z>isTOC rurrvN)hasattrrrZ encoderinforVsetvaluesZresizer5r6savegetvaluerPr%r rZwriteMAGICsumrZpack)r(rbfilenamerQZprovided_imagesZ size_streamsrZimageZtempentriesrurvZ file_lengthrwr r r _save3sV        rcCs|ddtkS)Nr)r)prefixr r r rJqsrJz.icnsz image/icns__main__rz)Syntax: python3 IcnsImagePlugin.py [file]rrbrQzout-%s-%s-%s.pngzout.pngZwindows)*r5osrsysZPILrrrrZ check_codecr3rrr rrrr0r9r:rarrJZ register_openroZregister_extensionZ register_saveZ register_mimer^rZargvprintexitopenrbZimfrdrr|r(platformZ startfiler r r r sF   *  eI>