B
    -¦aL-  ã            
   @   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)ÚstructÚunpackÚreadÚ
HEADERSIZE)Úfobj© r   ú7/tmp/pip-unpacked-wheel-rysjrues/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	bytesleftÚ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+   r0   r   r   r   Úread_mk[   s    
 r2   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ÚmodeÚconvert)	r   r   r   r   r   r   r)   Z
jp2kstreamÚfr   r   r   Úread_png_or_jpeg2000e   s.    









r<   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   rA   r   )r@   r@   r   )é   rB   r   )rB   rB   r   c             C   s   i  | _ }|| _t|ƒ\}}|tkr,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   ÚMAGICr   r   r   r7   ÚSEEK_CUR)Úselfr   rC   r   Ú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ÚitemsrC   r&   )rF   Úsizesr   ZfmtsÚfmtÚreaderr   r   r   Ú	itersizes·   s    


zIcnsFile.itersizesc             C   s   |   ¡ }|stdƒ‚t|ƒS )NzNo 32bit icon resources found)rN   r   Úmax)rF   rK   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)rI   rC   ÚgetÚupdater   )rF   r   rC   ÚcoderM   Ú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   r4   r   r1   )rP   ÚlenrU   rQ   ÚcopyZputalphaÚKeyError)rF   r   Zchannelsr)   r   r   r   ÚgetimageÓ   s    
zIcnsFile.getimage)N)Ú__name__Ú
__module__Ú__qualname__r<   r   r2   r   rI   rH   rN   rP   rU   rY   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 )Nr4   rK   r   r   r   )	r=   ÚfpÚicnsr9   rN   ÚinforP   Ú	best_sizer   )rF   r   r   r   Ú_openø   s    zIcnsImageFile._openc             C   s   | j S )N)Ú_size)rF   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 )	NrK   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)r`   rV   Úindexr6   rc   )rF   Ú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rd| jj| jkrdd S |  ¡  | j | j¡}| ¡  |j| _|j	| _	|j| _|  
¡  d S )Nr   r   r   r   )rV   r   ra   r   Úloadr)   Zload_preparer_   rY   r9   Zload_end)rF   r)   r   r   r   rh     s    zIcnsImageFile.loadN)rZ   r[   r\   Ú__doc__ÚformatÚformat_descriptionrb   Úpropertyr   Úsetterrh   r   r   r   r   r]   ê   s   	
r]   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 }	x2| ¡ D ]&\}
}|| }|	 |
t|ƒ|dœ¡ q¦W | t¡ | t dtdd„ |	D ƒƒ¡¡ | d¡ | t dtt|	ƒt  ¡¡ x6|	D ].}| |d ¡ | t dt|d  ¡¡ q(W xD|	D ]<}| |d ¡ | t dt|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)rd   r)   r   r   r   ú
<dictcomp>D  s    z_save.<locals>.<dictcomp>Zappend_imagesZpng)Útyper   Ústreamz>ic             s   s   | ]}|d  V  qdS )r   Nr   )rd   Úentryr   r   r   ú	<genexpr>X  s    z_save.<locals>.<genexpr>s   TOC rq   r   rr   N)Úhasattrrn   ZencoderinforQ   ÚsetÚvaluesÚresizer7   r8   ÚsaveÚgetvaluerJ   r&   rV   ÚwriterD   r   ÚpackÚsumr   )r)   r^   ÚfilenamerK   Zprovided_imagesZsize_streamsr   ÚimageÚtempÚentriesrq   rr   rs   r   r   r   Ú_save2  sF    

 

 
r‚   c             C   s   | d d… t kS )Nr   )rD   )Úprefixr   r   r   Ú_acceptk  s    r„   z.icnsz
image/icnsÚ__main__r   z)Syntax: python3 IcnsImagePlugin.py [file]r   ÚrbrK   zout-%s-%s-%s.pngzout.pngZwindows)*r7   Úosr   ÚsysZPILr   r   r   r   Zcheck_codecr5   r   rD   r   r   r   r   r2   r<   r=   r]   r‚   r„   Zregister_openrj   Zregister_extensionZregister_saveZregister_mimerZ   rV   ÚargvÚprintÚexitÚopenr^   Zimfr`   r   ry   r)   ÚplatformZ	startfiler   r   r   r   Ú<module>   sF   

*
 eH9
