o 6Wjm0 @sddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z e dZer5ddlmZdZd Zd:ddZd;ddZd;ddZd;ddZd;ddZGdddZGdd d e j Zd) annotationsN)IO)Image ImageFilePngImagePluginfeaturesjpg_2000)Jpeg2KImagePluginsicnsfobj IO[bytes]returntuple[bytes, int]cCstd|tS)Nz>4sI)structunpackread HEADERSIZE)r rb/home/esfera/Documents/content_generation/venv/lib/python3.10/site-packages/PIL/IcnsImagePlugin.py nextheader%sr start_lengthtuple[int, int]sizetuple[int, int, int]dict[str, Image.Image]cCsH|\}}|||d}|dkrd}t|t||d|df|S)Nsz'Unknown signature, expecting 0x00000000)seekr SyntaxErrorread_32)r rrstartlengthsigmsgrrrread_32t)s  r$c Csl|\}}|||d|d|d|df}|d|d}||dkr;||}td||dddd}d|iStd|d}tdD]k} g} |} | dkr|d} | sXn3| d} | d@rr| d }|d} | | g|n | 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. rrRGBrawN}zError reading channel [z left]L) rrr frombuffernewrangeextendappendreprrjoinimputband)r rrr r! pixel_sizesizesqindatar4band_ixdata bytesleftbytebyte_int blocksizer#bandrrrr6sB       rc Csf|d}|||d|d|d|df}|d|d}td|||dddd}d|iS)Nrr%rr+r(A)rrr-r)r rrr r6r7r?rrrread_mkbs   rAc Cs|\}}|||d}|dr'||t|}t|jd|iS|ds0|dkr`ts8d}t |||||}t |} t | }t|j|jdkr\|d}d|iSd}t |)N sPNG  RGBA)sOQs s jP zQUnsupported icon subimage format (rebuild PIL with JPEG 2000 support to fix this)z Unsupported icon subimage format)rr startswithr PngImageFiler_decompression_bomb_checkr enable_jpeg2k ValueErrorioBytesIOr Jpeg2KImageFilemodeconvert) r rrr r!r"r4r# jp2kstreamfrrrread_png_or_jpeg2000ns2             rPc@seZdZdefgdefgdefgdefgdefgdefdefdefgd efgd efgd efd efgd efdefdefgdefgdefdefdefgd Zd*ddZd+ddZ d,dd Z d-d#d$Z %d.d/d(d)Z d%S)0IcnsFileic10ic09ic14ic08ic13ic07sit32st8mksicp6ic12sih32sh8mksicp5sil32sl8mkic11sicp4sis32ss8mk) )rZr%)rZrZr)r[r%)r[r[r)r)r)r%)r)r)r)@r\r) r]r%)0r^r)r]r]r)r_r%)r_r_rr r rNonecCsi|_||_t|\}}t|sd}t|t}||krLt|\}}|dkr,d}t||t7}|t8}||f|j|<||tj||7}||ksdSdS)z@ fobj is a file-like object as an icns resource znot an icns filerzinvalid block headerN) dctr r_acceptrrrrISEEK_CUR)selfr r"filesizer#ir>rrr__init__s$  zIcnsFile.__init__list[tuple[int, int, int]]cCsBg}|jD]\}}|D]\}}||jvr||nq q|SN)SIZESitemsrar1)rdsizesrfmtsfmtreaderrrr itersizess   zIcnsFile.itersizesrcCs |}|s d}t|t|S)NzNo 32bit icon resources found)rprmax)rdrlr#rrrbestsizes zIcnsFile.bestsizerrcCsDi}|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)rjragetupdater )rdrracoderodescrrr dataforsizes zIcnsFile.dataforsizeN-tuple[int, int] | tuple[int, int, int] | None Image.ImagecCs|dur |}nt|dkr|d|ddf}||}|d}|r&|S|d}z ||dW|Sty@Y|Sw)Nr%rrrCr'r@)rrlenrwrscopyputalphaKeyError)rdrchannelsr4rrrgetimages       zIcnsFile.getimage)r r rr`)rrh)rr)rrrrri)rrxrry) __name__ __module__ __qualname__rPr$rArrjrgrprrrwrrrrrrQs:   rQc@sLeZdZdZdZdZdddZedd d Zej dd d ZddddZ 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. ICNSzMac OS icns resourcerr`cCsj|jdusJt|j|_d|_|j|jd<|j|_|jd|jd|jd|jdf|_dS)NrCrlrr%r) fprQicns_moderpinforr best_sizerrdrrr_open s   zIcnsImageFile._openrcCs|jSri)_sizerrrrrszIcnsImageFile.sizevaluecCsl|jdD]*}|d|d|d|df}|d|d}|d|d|kr/||_dSqd}t|)Nrlrr%rz2This is not one of the allowed sizes of this image)rrrH)rdrr simple_sizescaler#rrrrs Nr int | NoneImage.core.PixelAccess | NonecCs|dur|jdd\}}||||f|_|||f|_tj|}|jdur0|jj|jkr0|S||j|j}|}|j|_|j |_ |j|_|S)Nr%) rrrload_imr4 load_preparerrrLr)rdrwidthheightpxr4rrrr&s  zIcnsImageFile.load)rr`)rr)rrrr`ri)rrrr) rrr__doc__formatformat_descriptionrpropertyrsetterrrrrrrs     rr4ryrfilename str | bytesr`c Cst|dr |ddddddddd}d d |jd gD}i}t|D]!}||vr2||n|||f}t}| |d | ||<q(g} | D]\} }||} | | t t| | fqP|tt } | t d t| 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)r[rZir]r\)rWrUrSrRrYrXrVrTcSsi|]}|j|qSr)r).0r4rrr Psz_save.. append_imagespngr css|]}|dVqdS)rNr)rentryrrr fsz_save..z>isTOC rrr%N)hasattrr encoderinforssetvaluesresizerIrJsavegetvaluerkr1rrzwriteMAGICsumrpack)r4rrrlprovided_images size_streamsrimagetempentriestypestream file_lengthrrrr_save>sT        rprefixbytesboolcCs |tSri)rDr)rrrrrbzs rbz.icnsz image/icns__main__r%z)Syntax: python3 IcnsImagePlugin.py [file]rbrlzout--z.pngzout.pngwindows)r r rr)r r rrrrrr)r4ryrr rrrr`)rrrr)1 __future__rrIosrsystypingrrrrr check_codecrGr rrrr$rrArPrQrrrb register_openrregister_extension register_save register_mimerrzargvprintexitopenrimfrrrrrrr4platform startfilerrrrsV       , %i B <     $