o pc. @sddlZddlZddlZddlZddlmZmZmZmZe dZ e r)ddlm Z dZ dZ ddZd d Zd d Zd dZddZGdddZGdddejZddZddZeejeeeejdeejeeejdedkreejdkrede e!ejddNZ"ee"Z#e#j$d D] Z%e%e#_%e#&d!e%qe!ejd Z'e'&d"Wdn1swYej(d#kre)d"WddSWddS1swYdSdS)$N)Image ImageFilePngImagePluginfeaturesZjpg_2000)Jpeg2KImagePluginsicnscCstd|tS)Nz>4sI)structunpackread HEADERSIZE)fobjr 3lib/python3.10/site-packages/PIL/IcnsImagePlugin.py nextheader#srcCsH|\}}|||d}|dkrd}t|t||d|df|S)Nsz'Unknown signature, expecting 0x00000000)seekr SyntaxErrorread_32)r start_lengthsizestartlengthsigmsgr r rread_32t's  rc Cst|\}}|||d|d|d|df}|d|d}||dkr;||}td||dddd}d|iStd|d}tdD]o} g} |} | dkr|d} | sXn7| d} | d@rv| d } |d} t| D]}| | qmn | d} | || | | 8} | dkrn| dksP| dkrd t| d }t|td |d | dd dd}|j |j | qFd|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 frombuffernewrangeappendreprrjoinimZputband)r rrrr pixel_sizesizesqZindatar*Zband_ixdataZ bytesleftZbyte blocksizeirbandr r rr2sF         rc Csf|d}|||d|d|d|df}|d|d}td|||dddd}d|iS)Nrrrr"rA)rrr$r )r rrrr+r,r0r r rread_mk]s   r2c Cs|\}}|||d}|dddkr*||t|}t|jd|iS|dddks>|dddks>|dkrntsFd }t|||||}t |} t | }t|j|j dkrj|d}d|iSd }t|) 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*rZ jp2kstreamfr r rread_png_or_jpeg2000gs2            r;c@seZdZdefgdefgdefgdefgdefgdefdefdefgd efgd efgd efd efgd efdefdefgdefgdefdefdefgd ZddZddZ ddZ ddZ d ddZ dS)!IcnsFileic10ic09ic14ic08ic13ic07sit32st8mksicp6ic12sih32sh8mksicp5sil32sl8mkic11sicp4sis32ss8mk) )rEr)rErEr)rFr)rFrFr)r r r)r r r)@rGr) rHr)0rIr)rHrHr)rJr)rJrJrcCsi|_}||_t|\}}t|sd}t|t}||krMt|\}}|dkr.d}t||t7}|t8}||f||<||tj||7}||ksdSdS)z@ fobj is a file-like object as an icns resource znot an icns filerzinvalid block headerN) dctr r_acceptrr rr7SEEK_CUR)selfr rKrZfilesizerr/r.r r r__init__s$    zIcnsFile.__init__cCsBg}|jD]\}}|D]\}}||jvr||nq q|SN)SIZESitemsrKr')rNsizesrZfmtsZfmtreaderr r r itersizess   zIcnsFile.itersizescCs |}|s d}t|t|S)NzNo 32bit icon resources found)rUrmax)rNrSrr r rbestsizes zIcnsFile.bestsizecCsDi}|j|D]\}}|j|}|dur|||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)rQrKgetupdater )rNrrKcoderTZdescr r r dataforsizes zIcnsFile.dataforsizeNcCs|dur|}t|dkr|d|ddf}||}|dd}|r&|S|d}z ||dW|StyAY|Sw)Nrrrr4rr1)rWlenr[rXcopyZputalphaKeyError)rNrZchannelsr*r r rgetimages     zIcnsFile.getimagerP) __name__ __module__ __qualname__r;rr2rrQrOrUrWr[r_r r r rr<s8  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)Nr4rSrrr) r<fpicnsr9rUinforW best_sizerrNr r r_opens   zIcnsImageFile._opencCs|jSrP)_sizerhr r rr szIcnsImageFile.sizecCs|}||jdvrt|dkr|d|ddf}||jdvrCt|dkrC|ddkrCdd|jdD}||vrC|jd||}||jdvrPd}t|||_dS) NrSrrrrcSs,g|]}|d|d|d|dfqS)rrrr ).0rr r r s z&IcnsImageFile.size..z2This is not one of the allowed sizes of this image)rfr\indexr6rj)rNvalueZ info_sizeZ simple_sizesrr r rr s   cCst|jdkr |j|_|jd|jd|jd|jdf|_tj|}|jdur4|jj|jkr4|S||j|j}|}|j|_|j |_ |j|_|S)Nrrrr) r\rrgrloadr*Z load_preparerer_r9)rNZpxr*r r rro!s zIcnsImageFile.loadN) r`rarb__doc__formatZformat_descriptionripropertyrsetterror r r rrcs    rcc Cst|dr |ddddddddd}d d |jd gD}i}t|D]!}||vr2||n|||f}t}| |d | ||<q(g} | D]\} }||} | | t t| | d qP|tt } | t dt| 7} | tdd| D7} |td| |d|tdt t| t | D]} || d|td| dq| D]} || d|td| d|| dqt|dr|dSdS)za Saves the image as a series of PNG files, that are then combined into a .icns file. flushr rFrEirHrG)rBr@r>r=rDrCrAr?cSsi|]}|j|qSr )width)rkr*r r r Msz_save..Z append_imagesZpng)typerstreamrcss|]}|dVqdS)rNr )rkentryr r r esz_save..z>isTOC rwrrxN)hasattrrtZ encoderinforXsetvaluesZresizer7r8savegetvaluerRr'r r\writeMAGICsumrZpack)r*rdfilenamerSZprovided_imagesZ size_streamsrZimageZtempentriesrwrxZ file_lengthryr r r_save;sX        rcCs|ddtkS)Nr)r)prefixr r rrLysrLz.icnsz image/icns__main__rz)Syntax: python3 IcnsImagePlugin.py [file]rrbrSzout-%s-%s-%s.pngzout.pngZwindows)*r7osrsysZPILrrrrZ check_codecr5rrr rrrr2r;r<rcrrLZ register_openrqZregister_extensionZ register_saveZ register_mimer`r\argvprintexitopenrdZimfrfrr~r*platformZ startfiler r r rsR   + "hJ>   "