B
    àz÷bë-  ã            
   @   s¤  d dl Z d dlZd dlZd dlZd dlmZmZmZmZ e 	d¡Z
e
rRd dlmZ dZdZdd„ Zd	d
„ Zdd„ Zdd„ Zdd„ ZG dd„ dƒZG dd„ dejƒZdd„ Zdd„ Ze ejee¡ e ejd¡ e eje¡ e ejd¡ edkr eejƒdk redƒ e  ¡  e!ejd dƒtZ"ee"ƒZ#x(e#j$d  D ]Z%e%e#_%e# &d!e% ¡ q<W e !ejd ¡Z'e' &d"¡ W dQ R X ej(d#kr–e )d"¡ W dQ R X dS )$é    N)ÚImageÚ	ImageFileÚPngImagePluginÚfeaturesZjpg_2000)ÚJpeg2KImagePlugins   icnsé   c             C   s   t  d|  t¡¡S )Nz>4sI)ÚstructZunpackÚreadÚ
HEADERSIZE)Úfobj© r   ú2lib/python3.7/site-packages/PIL/IcnsImagePlugin.pyÚ
nextheader#   s    r   c             C   sD   |\}}|   |¡ |  d¡}|dkr,tdƒ‚t| |d |d f|ƒS )Né   s       z'Unknown signature, expecting 0x00000000)Úseekr	   ÚSyntaxErrorÚread_32)r   Ústart_lengthÚsizeÚstartÚlengthÚsigr   r   r   Úread_32t'   s    

r   c          
   C   st  |\}}|   |¡ |d |d  |d |d  f}|d |d  }||d krp|  |¡}t d||dddd¡}nüt d|d¡}xìtdƒD ]à}	g }
|}xˆ|dkr|  d¡}|s°P |d }|d@ rð|d	 }|  d¡}x4t|ƒD ]}|
 |¡ qÜW n|d }|
 |  |¡¡ ||8 }|dkr–P q–W |dkr<td
t|ƒ› dƒ‚t d|d 	|
¡dddd¡}|j
 |j
|	¡ qˆW d|iS )zn
    Read a 32bit RGB icon resource.  Seems to be either uncompressed or
    an RLE packbits-like scheme.
    r   é   é   é   ÚRGBÚrawNé€   é}   zError reading channel [z left]ÚLó    )r   r	   r   Ú
frombufferÚnewÚrangeÚappendr   ÚreprÚjoinÚimZputband)r   r   r   r   r   Ú
pixel_sizeÚsizesqZindatar(   Zband_ixÚdataZ	bytesleftZbyteÚ	blocksizeÚiÚbandr   r   r   r   1   s>    
 



r   c          	   C   sf   |d }|   |¡ |d |d  |d |d  f}|d |d  }t d||  |¡dddd¡}d|iS )Nr   r   r   r    r   ÚA)r   r   r"   r	   )r   r   r   r   r)   r*   r.   r   r   r   Úread_mk[   s    
 r0   c       	      C   sä   |\}}|   |¡ |  d¡}|d d… dkrT|   |¡ t | ¡}t |j¡ d|iS |d d… dks||d d… dks||dkrØtsˆtd	ƒ‚|   |¡ |  |¡}t	 
|¡}t |¡}t |j¡ |jdkrÐ| d¡}d|iS td
ƒ‚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   ZJpeg2KImageFileÚmodeZconvert)	r   r   r   r   r   r   r(   Z
jp2kstreamÚfr   r   r   Úread_png_or_jpeg2000e   s.    









r9   c               @   sÆ   e Zd Zdefgdefgdefgdefgdefgdefdefdefgd	efgd
efgdefdefgdefdefdefgdefgdefdefdefgdœZdd„ Zdd„ Z	dd„ Z
dd„ Zd dd„ZdS )!ÚIcnsFiles   ic10s   ic09s   ic14s   ic08s   ic13s   ic07s   it32s   t8mks   icp6s   ic12s   ih32s   h8mks   icp5s   il32s   l8mks   ic11s   icp4s   is32s   s8mk))i   i   r   )i   i   r   )é   r;   r   )r;   r;   r   )r   r   r   )r   r   r   )é@   r<   r   )é    r=   r   )é0   r>   r   )r=   r=   r   )é   r?   r   )r?   r?   r   c             C   s   i  | _ }|| _t|ƒ\}}t|ƒs,tdƒ‚t}xZ||k rŠt|ƒ\}}|dkrVtdƒ‚|t7 }|t8 }||f||< | |tj¡ ||7 }q2W dS )z@
        fobj is a file-like object as an icns resource
        znot an icns filer   zinvalid block headerN)	Údctr   r   Ú_acceptr   r
   r   r5   ÚSEEK_CUR)Úselfr   r@   r   Zfilesizer-   r,   r   r   r   Ú__init__¢   s    

zIcnsFile.__init__c             C   sH   g }x>| j  ¡ D ]0\}}x&|D ]\}}|| jkr| |¡ P qW qW |S )N)ÚSIZESÚitemsr@   r%   )rC   Úsizesr   ZfmtsZfmtÚreaderr   r   r   Ú	itersizes·   s    


zIcnsFile.itersizesc             C   s   |   ¡ }|stdƒ‚t|ƒS )NzNo 32bit icon resources found)rI   r   Úmax)rC   rG   r   r   r   ÚbestsizeÀ   s    zIcnsFile.bestsizec             C   sH   i }x>| j | D ]0\}}| j |¡}|dk	r| || j||ƒ¡ qW |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)rE   r@   ÚgetÚupdater   )rC   r   r@   ÚcoderH   Zdescr   r   r   ÚdataforsizeÆ   s    zIcnsFile.dataforsizeNc             C   s†   |d kr|   ¡ }t|ƒdkr.|d |d df}|  |¡}| dd ¡}|rL|S | d¡ ¡ }y| |d ¡ W n tk
r€   Y nX |S )Nr   r   r   r2   r   r/   )rK   ÚlenrO   rL   ÚcopyZputalphaÚKeyError)rC   r   Zchannelsr(   r   r   r   ÚgetimageÓ   s    
zIcnsFile.getimage)N)Ú__name__Ú
__module__Ú__qualname__r9   r   r0   r   rE   rD   rI   rK   rO   rS   r   r   r   r   r:   …   s.   	r:   c               @   sB   e Zd ZdZdZdZdd„ Zedd„ ƒZej	dd„ ƒZd	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.
    ZICNSzMac OS icns resourcec             C   s\   t | jƒ| _d| _| j ¡ | jd< | j ¡ | _| jd | jd  | jd | jd  f| _d S )Nr2   rG   r   r   r   )	r:   ÚfpÚicnsr7   rI   ÚinforK   Ú	best_sizer   )rC   r   r   r   Ú_openø   s    zIcnsImageFile._openc             C   s   | j S )N)Ú_size)rC   r   r   r   r     s    zIcnsImageFile.sizec             C   s¦   |}|| j d kr0t|ƒdkr0|d |d df}|| j d kr†t|ƒdkr†|d dkr†dd„ | j d D ƒ}||kr†| j d | |¡ }|| j d krœtdƒ‚|| _d S )	NrG   r   r   r   r   c             S   s,   g | ]$}|d  |d  |d |d  f‘qS )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)rZ   rP   Úindexr4   r]   )rC   ÚvalueZ	info_sizeZsimple_sizesr   r   r   r     s    c             C   s¢   t | jƒdkr@| j| _| jd | jd  | jd | jd  f| _tj | ¡}| jd k	rh| jj| jkrh|S |  ¡  | j | j¡}| ¡ }|j| _|j	| _	|j| _|S )Nr   r   r   r   )
rP   r   r[   r   Úloadr(   Zload_preparerY   rS   r7   )rC   Zpxr(   r   r   r   rb     s    zIcnsImageFile.loadN)rT   rU   rV   Ú__doc__ÚformatZformat_descriptionr\   Úpropertyr   Úsetterrb   r   r   r   r   rW   ê   s   	
rW   c          	   C   sÔ  t |dƒr| ¡  dddddddddœ}d	d
„ | j dg ¡D ƒ}i }xRt| ¡ ƒD ]B}||krf|| n|  ||f¡}t ¡ }| 	|d¡ | 
¡ ||< qRW g }	x6| ¡ D ]*\}
}|| }|	 |
tt|ƒ |dœ¡ q¦W | t¡ t}|tdt|	ƒ  7 }|tdd„ |	D ƒƒ7 }| t d|¡¡ | d¡ | t dtt|	ƒt  ¡¡ x2|	D ]*}| |d ¡ | t d|d ¡¡ qLW x@|	D ]8}| |d ¡ | t d|d ¡¡ | |d ¡ q€W t |dƒrÐ| ¡  dS )za
    Saves the image as a series of PNG files,
    that are then combined into a .icns file.
    Úflushr   r;   i   i   r=   r<   )s   ic07s   ic08s   ic09s   ic10s   ic11s   ic12s   ic13s   ic14c             S   s   i | ]}||j “qS r   )Úwidth)r^   r(   r   r   r   ú
<dictcomp>E  s    z_save.<locals>.<dictcomp>Zappend_imagesZpng)Útyper   Ústreamr   c             s   s   | ]}|d  V  qdS )r   Nr   )r^   Úentryr   r   r   ú	<genexpr>]  s    z_save.<locals>.<genexpr>z>is   TOC rj   r   rk   N)Úhasattrrg   ZencoderinforL   ÚsetÚvaluesZresizer5   r6   ÚsaveÚgetvaluerF   r%   r
   rP   ÚwriteÚMAGICÚsumr   Zpack)r(   rX   ÚfilenamerG   Zprovided_imagesZsize_streamsr   ZimageZtempÚentriesrj   rk   Zfile_lengthrl   r   r   r   Ú_save3  sN    




rx   c             C   s   | d d… t kS )Nr   )rt   )Úprefixr   r   r   rA   q  s    rA   z.icnsz
image/icnsÚ__main__r   z)Syntax: python3 IcnsImagePlugin.py [file]r   ÚrbrG   zout-%s-%s-%s.pngzout.pngZwindows)*r5   Úosr   ÚsysZPILr   r   r   r   Zcheck_codecr3   r   rt   r
   r   r   r   r0   r9   r:   rW   rx   rA   Zregister_openrd   Zregister_extensionZregister_saveZregister_mimerT   rP   ÚargvÚprintÚexitÚopenrX   ZimfrZ   r   rq   r(   ÚplatformZ	startfiler   r   r   r   Ú<module>   sF   

*
 eI>
