3 jQcE+@sddlmZmZmZddlmZddlZddlZddlZddl Z ddl Z ddl Z e ej dZerlddlmZdZddZd d Zd d Zd dZddZGdddeZGdddejZddZejejeddejejde jdkr ejejeejejde dkre!e j"dkr6e#de j$ee%e j"dd Z&x6e&j'd!D](Z(e(e&_(e&j)e&j*Z*e*j+d"e(qVWej%e j"dZ*e*j+d#e jd$krej,d#dS)%)Image ImageFilePngImagePlugin)i8NZjp2klib_version)Jpeg2KImagePlugincCstjd|jtS)Nz>4sI)structZunpackread HEADERSIZE)fobjr 5/usr/lib/python3/dist-packages/PIL/IcnsImagePlugin.py nextheader"srcCsD|\}}|j||jd}|dkr,tdt||d|df|S)Nsz'Unknown signature, expecting 0x00000000)seekr SyntaxErrorread_32)r start_lengthsizestartlengthsigr r r read_32t&s   rc Csl|\}}|j||d|d|d|df}|d|d}||dkrp|j|}tjd||dddd}ntjd|d}xtdD]} g} |} x| dkr|jd} | sPt| } | d@r| d } |jd} x4t| D]}| j| qWn| d} | j|j| | | 8} | dkrPqW| dkr4td | tjd |d j | dd dd}|j j |j | qWd|iS) zn Read a 32bit RGB icon resource. Seems to be either uncompressed or an RLE packbits-like scheme. rRGBrawN}zError reading channel [%r left]L) rr r frombuffernewrangerappendrjoinimZputband)r rrrr pixel_sizesizesqZindatar'Zband_ixdataZ bytesleftZbyte blocksizeibandr r r r0sB         rcCsf|d}|j||d|d|d|df}|d|d}tjd||j|dddd}d|iS)Nrrrr rA)rrr"r )r rrrr(r)r-r r r read_mk^s  r/c Cs|\}}|j||jd}|dddkrT|j|tj|}tj|jd|iS|dddks||dddks||dkrtstd |j||j|}t j |}t j |}tj|j|j dkr|jd}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_jpeg2000js,            r8c@seZdZd(defgd)defgd*defgd+defgd,d efgd-d efd efd efgd.defgd/defgd0defdefgd1defdefdefgd2defgd3defdefdefgi ZddZdd Z d!d"Z d#d$Z d4d&d'Z d%S)5IcnsFilersic10rsic09sic14sic08rsic13sic07sit32st8mk@sicp6 sic120sih32sh8mksicp5sil32sl8mksic11sicp4sis32ss8mkcCsi|_}||_t|\}}|dkr,tdt}xX||krt|\}}|dkrVtd|t7}|t8}||f||<|j|d||7}q2WdS)z@ fobj is a file-like object as an icns resource sicnsznot an icns filerzinvalid block headerrN)dctr rrr r)selfr r@rZfilesizer,r+r r r __init__s      zIcnsFile.__init__cCsHg}x>|jjD]0\}}x&|D]\}}||jkr|j|PqWqW|S)N)SIZESitemsr@r%)rAsizesrZfmtsZfmtreaderr r r itersizess   zIcnsFile.itersizescCs|j}|stdt|S)NzNo 32bit icon resources found)rGrmax)rArEr r r bestsizeszIcnsFile.bestsizecCsHi}x>|j|D]0\}}|jj|}|dk r|j||j||qW|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)rCr@getupdater )rArr@coderFZdescr r r dataforsizes  zIcnsFile.dataforsizeNc Cs|dkr|j}t|dkr.|d|ddf}|j|}|jdd}|rL|S|jdj}y|j|dWntk rYnX|S)Nrrrr1rr.)rIlenrMrJcopyZputalphaKeyError)rArZchannelsr'r r r getimages   zIcnsFile.getimage)r:r:r)r:r:r)r;r;r)r;r;r)rrr)rrr)r<r<r)r=r=r)r>r>r)r=r=r)r?r?r)r?r?r)N) __name__ __module__ __qualname__r8rr/rrCrBrGrIrMrQr r r r r9sH   r9c@s(eZdZdZdZdZddZddZdS) 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 resourcecCsbt|j|_d|_|jj|_|jd|jd|jd|jdf|_|jj|jd<d|_ dS)Nr1rrrrE)rV) r9fpicnsr6rI best_sizerrGinfotile)rAr r r _open s  zIcnsImageFile._opencCst|jdkr@|j|_|jd|jd|jd|jdf|_tjj||jsVdS|j|jj|j}|j|j |_ |j |_ |j|_d|_ d|_f|_|j dS)Nrrrr) rNrrYrloadr[Z load_preparerXrQr'r6rWZload_end)rAr'r r r r]s" zIcnsImageFile.loadN)rRrSrT__doc__formatZformat_descriptionr\r]r r r r rUs   rUc Csbt|dr|jtjd}dd|jjdgD}d}xdD]}d j||}tjj ||d }||krvt j | |n$|j||j ||ft j} | j|tjj ||d} |j|d|j |d|dft j} | j| |d}q>Wddlm} m} m}dddd||g}ttjd}| || |d}WdQRX|jj|j}t j||r^|||dS)z Saves the image as a series of PNG files, that are then converted to a .icns file using the macOS command line utility 'iconutil'. macOS only. flushz.iconsetcSsi|] }||jqSr )width).0r'r r r ;sz_save..Z append_imagesNr?r=rr;z icon_{}x{}z.pngz@2x.pngrr)PopenPIPECalledProcessErrorZiconutilz-crXz-owb)stdoutstderr)r?r=rr;rd)hasattrr`tempfileZmkdtempZ encoderinforJr_ospathr&shutilZcopyfileZresizerZLANCZOSsave subprocessrerfrgopendevnullriclosewaitZrmtree)r'rWfilenameZiconsetZprovided_imagesZlast_wwprefixZ first_pathZ second_pathZim_wZim_w2rerfrgZ convert_cmdrsZ convert_procZretcoder r r _save.s4     &    rycCs|dddkS)Nrsicnsr )xr r r `sr{z.icnsdarwinz image/icns__main__rz(Syntax: python IcnsImagePlugin.py [file]rrbrEzout-%s-%s-%s.pngzout.pngZwindows)-ZPILrrrZ PIL._binaryrr4rmrorsysrlrkZcorer2rr rrrr/r8objectr9rUryZ register_openr_Zregister_extensionplatformZ register_saveZ register_mimerRrNargvprintexitrrZimfrZrr]r'rpZ startfiler r r r sN    . w11