U
|
T ±d)0 ã
|
@ s d dl Z d dlZd dlZd dlZd dlmZmZmZmZ e d¡Z
|
e
|
rRd dlm Z dZ dZ dd Zd d
|
Zd d Zd d Zdd ZG dd dZG dd dejZdd Zdd Ze ejee¡ e ejd¡ e eje¡ e ejd¡ e dkre ej d k re d e ¡ e!ej d d pZ"ee"Z#e#j$d D ]Z%e%e#_%e# &d!e% ¡ q:e !ej d ¡Z'e' &d"¡ W 5 Q R X ej(d#kre )d"¡ W 5 Q R X dS )$é N)ÚImageÚ ImageFileÚPngImagePluginÚfeaturesZjpg_2000)ÚJpeg2KImagePlugins icnsé c C s t d| t¡¡S )Nz>4sI)ÚstructÚunpackÚreadÚ
|
HEADERSIZE)Úfobj© r úJd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\PIL/IcnsImagePlugin.pyÚ
|
nextheader# s r c C sH |\}}| |¡ | d¡}|dkr0d}t|t| |d |d f|S )Né s z'Unknown signature, expecting 0x00000000)Úseekr
|
Ú SyntaxErrorÚread_32)r Ú start_lengthÚsizeÚstartÚlengthÚsigÚmsgr r r Úread_32t' s
|
|
r c
|
C sp |\}}| |¡ |d |d |d |d f}|d |d }||d krp| |¡}t d||dddd¡}nøt d|d¡}tdD ]à} g }
|
|} | dkr| d¡} | s®q| d } | d@ rê| d } | d¡} t| D ]}|
|
| ¡ qØn| d } |
|
| | ¡¡ | | 8 } | dkrqq| dkr:d
|
t| d }t|t d |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.
|
r é é é ÚRGBÚrawNé é} zError reading channel [z left]ÚLó ) r r
|
r Ú
|
frombufferÚnewÚrangeÚappendÚreprr ÚjoinÚimZputband)r r r r r Ú
|
pixel_sizeÚsizesqZindatar* Zband_ixÚdataZ bytesleftÚbyteÚ blocksizeÚir Úbandr r r r 2 s@
|
|
|
|
|
|
r c C sf |d }| |¡ |d |d |d |d f}|d |d }t d|| |¡dddd¡}d|iS )Nr r r r" r ÚA)r r r$ r
|
)r r r r r+ r, r1 r r r Úread_mk] s
|
r3 c
|
C sì |\}}| |¡ | d¡}|d d
dkrT| |¡ t | ¡}t |j¡ d|iS |d d
dks||d d
dks||dkrÜtsd }t|| |¡ | |¡}t
|
|¡} t | ¡}t |j¡ |j dkrÔ| d¡}d|iS d
|
}t|d S ) Né r s PNG
|
|
ÚRGBAr s ÿOÿQs
|
|
s jP
|
|
zQUnsupported icon subimage format (rebuild PIL with JPEG 2000 support to fix this)z Unsupported icon subimage format)r r
|
r Z PngImageFiler Z_decompression_bomb_checkr Ú enable_jpeg2kÚ
|
ValueErrorÚioÚBytesIOr ZJpeg2KImageFileÚmodeÚconvert)
|
r r r r r r r* r Z
|
jp2kstreamÚfr r r Úread_png_or_jpeg2000g s8
|
|
|
|
ÿþýÿ
|
|
|
|
|
|
r= c @ sÆ e Zd Zdefgdefgdefgdefgdefgdefdefdefgd efgd
|
efgd efd efgd efdefdefgdefgdefdefdefgd Zdd Zdd Z dd Z
|
dd Z d d d Z d S )!ÚIcnsFileó ic10ó ic09ó ic14ó ic08ó ic13ó ic07s it32s t8mks icp6ó ic12s ih32s h8mks icp5s il32s l8mkó ic11s icp4s is32s s8mk) )é rG r )rG rG r )é rH r )rH rH r )r r r )r r r )é@ rI r )é rJ r )é0 rK r )rJ rJ r )é rL r )rL rL r c C s i | _ }|| _t|\}}t|s0d}t|t}||k rt|\}}|dkr\d}t||t7 }|t8 }||f||< | |tj¡ ||7 }q4dS )z@
|
fobj is a file-like object as an icns resource
|
znot an icns filer zinvalid block headerN) Údctr r Ú_acceptr r r r8 ÚSEEK_CUR)Úselfr rM r Úfilesizer r0 r/ r r r Ú__init__¥ s"
|
zIcnsFile.__init__c C sB g }| j ¡ D ].\}}|D ] \}}|| jkr| |¡ qqq|S ©N)ÚSIZESÚitemsrM r' )rP Úsizesr ZfmtsÚfmtÚreaderr r r Ú itersizes¼ s
|
|
zIcnsFile.itersizesc C s | ¡ }|sd}t|t|S )Nz No 32bit icon resources found)rY r Úmax)rP rV r r r r ÚbestsizeÅ s
|
zIcnsFile.bestsizec C sD i }| 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)rT rM ÚgetÚupdater )rP r rM ÚcoderX Údescr r r Ú dataforsizeÌ s zIcnsFile.dataforsizeNc C s |d kr| ¡ }t|dkr.|d |d df}| |¡}| dd ¡}|rL|S | d¡ ¡ }z| |d ¡ W n tk
|
r Y nX |S )Nr r r r5 r r2 )r[ Úlenr` r\ ÚcopyZputalphaÚKeyError)rP r Zchannelsr* r r r ÚgetimageÙ s
|
zIcnsFile.getimage)N) Ú__name__Ú
|
__module__Ú __qualname__r= r r3 r rT rR rY r[ r` rd r r r r r> s6 ýýýì r> c @ sB e Zd ZdZdZdZdd Zedd Zej dd 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 resourcec C s\ t | j| _d| _| j ¡ | jd< | j ¡ | _| jd | jd | jd | jd f| _d S )Nr5 rV r r r ) r> ÚfpÚicnsr: rY Úinfor[ Ú best_sizer ©rP r r r Ú_openþ s þzIcnsImageFile._openc C s | j S rS )Ú_sizerm r r r r s zIcnsImageFile.sizec C sª |}|| j d kr0t|dkr0|d |d df}|| j d krt|dkr|d dkrdd | j d D }||kr| j d | |¡ }|| j d kr d}t||| _d S ) NrV r r r r c S s, g | ]$}|d |d |d |d fqS )r r r r )Ú.0r r r r Ú
|
<listcomp> s z&IcnsImageFile.size.<locals>.<listcomp>z2This is not one of the allowed sizes of this image)rk ra Úindexr7 ro )rP ÚvalueZ info_sizeZ simple_sizesr r r r r s$ ÿ
|
þ
|
ýÿc C s¢ t | jdkr@| j| _| jd | jd | jd | jd f| _tj | ¡}| jd k rh| jj| jkrh|S | ¡ | j | j¡}| ¡ }|j| _|j | _ |j| _|S )Nr r r r )
|
ra r rl r Úloadr* Z load_preparerj rd r: )rP Zpxr* r r r rt s þ zIcnsImageFile.loadN) re rf rg Ú__doc__ÚformatÚformat_descriptionrn Úpropertyr Úsetterrt r r r r rh ð s
|
|
|
|
rh c C sÄ t |dr| ¡ ddddddddd}d d
|
| j d g ¡D }i }t| ¡ D ]B}||krd|| n | ||f¡}t ¡ }| |d ¡ |
|
¡ ||< qPg } | ¡ D ]*\}
|
}|| } | |
|
t t| | d ¡ q | t¡ t } | t dt| 7 } | tdd | D 7 } | t d| ¡¡ | d¡ | t dt t| t ¡¡ | D ]*} | | d ¡ | t d| d ¡¡ qB| D ]8} | | d ¡ | t d| d ¡¡ | | d ¡ qrt |drÀ| ¡ dS )za
|
Saves the image as a series of PNG files,
|
that are then combined into a .icns file.
|
Úflushr rH rG i rJ rI )rD rB r@ r? rF rE rC rA c S s i | ] }|j |qS r )Úwidth)rp r* r r r Ú
|
<dictcomp>L s z_save.<locals>.<dictcomp>Z append_imagesZpng)Útyper Ústreamr c s s | ]}|d V qdS )r Nr )rp Úentryr r r Ú <genexpr>d s z_save.<locals>.<genexpr>z>is TOC r} r r~ N)Úhasattrrz Z encoderinfor\ ÚsetÚvaluesÚresizer8 r9 ÚsaveÚgetvaluerU r' r ra ÚwriteÚMAGICÚsumr Úpack)r* ri ÚfilenamerV Zprovided_imagesZ size_streamsr ÚimageÚtempÚentriesr} r~ Z file_lengthr r r r Ú_save: sV
|
ø
|
ÿ
|
ý ÿ
|
|
r c C s | d d
t kS )Nr )r )Úprefixr r r rN x s rN z.icnsz
|
image/icnsÚ__main__r z)Syntax: python3 IcnsImagePlugin.py [file]r ÚrbrV zout-%s-%s-%s.pngzout.pngZwindows)*r8 Úosr ÚsysZPILr r r r Z check_codecr6 r r r r r r r3 r= r> rh r rN Z register_openrv Zregister_extensionZ register_saveZ register_mimere ra ÚargvÚprintÚexitÚopenri Zimfrk r r
r* ÚplatformÚ startfiler r r r Ú<module> sF
|
+
|
"gJ>
|
|
