a
    Qh.                  	   @   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reejdk red e   e!ejd dZ"ee"Z#e#j$d  D ]Z%e%e#_%e#&d!e%  q:e!ejd Z'e'&d" W d   n1 s0    Y  ej(d#kre)d" W d   n1 s0    Y  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   L/var/www/sistema_ama/venv/lib/python3.9/site-packages/PIL/IcnsImagePlugin.py
nextheader#   s    r   c                 C   sH   |\}}|  | | d}|dkr0d}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 krp| |}td||dddd}ntd|d}tdD ]}	g }
|}|dkr| d}|sq|d }|d@ r|d	 }| d}t|D ]}|
| qn|d }|
| | ||8 }|dkrqq|dkr:d
t| d}t|td|d	|
dddd}|j
|j
|	 qd|iS )zn
    Read a 32bit RGB icon resource.  Seems to be either uncompressed or
    an RLE packbits-like scheme.
    r      r      RGBrawN   }   zError reading channel [z left]L    )r   r   r   
frombuffernewrangeappendreprr   joinimputband)r   r   r   r   r   
pixel_sizesizesqZindatar*   Zband_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|| |dddd}d|iS )Nr   r   r   r"   r   A)r   r   r$   r   )r   r   r   r   r,   r-   r2   r   r   r   read_mk]   s    
 r4   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rtsd	}t||  | | |}t	
|}	t|	}t|j |jdkr|d}d|iS d
}t|d S )N   r   s   PNG

RGBAr   s   OQs   

s      jP  

zQUnsupported icon subimage format (rebuild PIL with JPEG 2000 support to fix this)z Unsupported icon subimage format)r   r   r   PngImageFiler   _decompression_bomb_checkr   enable_jpeg2k
ValueErrorioBytesIOr   ZJpeg2KImageFilemodeconvert)
r   r   r   r   r   r   r*   r   Z
jp2kstreamfr   r   r   read_png_or_jpeg2000g   s8    









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 )!IcnsFile   ic10   ic09   ic14   ic08   ic13   ic07s   it32s   t8mks   icp6   ic12s   ih32s   h8mks   icp5s   il32s   l8mk   ic11s   icp4s   is32s   s8mk))   rJ   r   )rJ   rJ   r   )   rK   r   )rK   rK   r   )r    r    r   )r    r    r   )@   rL   r   )    rM   r   )0   rN   r   )rM   rM   r   )   rO   r   )rO   rO   r   c                 C   s   i  | _ }|| _t|\}}t|s0d}t|t}||k rt|\}}|dkr\d}t||t7 }|t8 }||f||< ||tj ||7 }q4d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   r;   SEEK_CUR)selfr   rP   r   filesizer   r1   r0   r   r   r   __init__   s"    
zIcnsFile.__init__c                 C   sB   g }| j  D ].\}}|D ] \}}|| jv r||  qqq|S N)SIZESitemsrP   r'   )rS   sizesr   Zfmtsfmtreaderr   r   r   	itersizes   s    

zIcnsFile.itersizesc                 C   s    |   }|sd}t|t|S )NzNo 32bit icon resources found)r\   r   max)rS   rY   r   r   r   r   bestsize   s
    zIcnsFile.bestsizec                 C   sD   i }| j | D ]0\}}| j|}|du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)rW   rP   getupdater   )rS   r   rP   coder[   descr   r   r   dataforsize   s    zIcnsFile.dataforsizeNc                 C   s   |d u r|   }t|dkr.|d |d df}| |}|dd }|rL|S |d }z||d  W n ty~   Y n0 |S )Nr   r   r   r6   r   r3   )r^   lenrc   r_   copyputalphaKeyError)rS   r   channelsr*   r   r   r   getimage   s    
zIcnsFile.getimage)N)__name__
__module____qualname__r@   r   r4   r   rW   rU   r\   r^   rc   ri   r   r   r   r   rA      s6   	rA   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 )Nr6   rY   r   r   r   )	rA   fpicns_moder\   infor^   	best_sizer   rS   r   r   r   _open   s    zIcnsImageFile._openc                 C   s   | j S rV   )_sizers   r   r   r   r     s    zIcnsImageFile.sizec                 C   s   |}|| j d vr0t|dkr0|d |d df}|| j d vrt|dkr|d dkrdd | j d D }||v r| j d || }|| j d vrd}t||| _d S )	NrY   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)rq   rd   indexr:   ru   )rS   valueZ	info_sizeZsimple_sizesr   r   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 urh| jj| jkrh|S |   | j| j}| }|j| _|j	| _
|j| _|S )Nr   r   r   r   )rd   r   rr   r   loadr*   load_preparero   ri   r=   rp   )rS   pxr*   r   r   r   rz      s    zIcnsImageFile.loadN)rj   rk   rl   __doc__formatformat_descriptionrt   propertyr   setterrz   r   r   r   r   rm      s   



rm   c              	   C   s  t |dr|  ddddddddd}d	d
 | jdg D }i }t| D ]B}||v rd|| n| ||f}t }|	|d |
 ||< qPg }	| D ]*\}
}|| }|	|
tt| |d q|t t}|tdt|	  7 }|tdd |	D 7 }|td| |d |tdtt|	t   |	D ]*}||d  |td|d  qB|	D ]8}||d  |td|d  ||d  qrt |dr|  dS )za
    Saves the image as a series of PNG files,
    that are then combined into a .icns file.
    flushr    rK   rJ   i   rM   rL   )rG   rE   rC   rB   rI   rH   rF   rD   c                 S   s   i | ]}|j |qS r   )width)rv   r*   r   r   r   
<dictcomp>L  r#   z_save.<locals>.<dictcomp>append_imagespng)typer   streamr   c                 s   s   | ]}|d  V  qdS )r   Nr   )rv   entryr   r   r   	<genexpr>d  r#   z_save.<locals>.<genexpr>z>is   TOC r   r   r   N)hasattrr   encoderinfor_   setvaluesresizer;   r<   savegetvaluerX   r'   r   rd   writeMAGICsumr	   pack)r*   rn   filenamerY   Zprovided_imagesZsize_streamsr   imagetempentriesr   r   Zfile_lengthr   r   r   r   _save:  sV    




r   c                 C   s   | d d t kS )Nr   )r   )prefixr   r   r   rQ   x  s    rQ   z.icnsz
image/icns__main__r   z)Syntax: python3 IcnsImagePlugin.py [file]rbrY   zout-%s-%s-%s.pngzout.pngwindows)*r;   osr	   sys r   r   r   r   Zcheck_codecr9   r   r   r   r   r   r   r4   r@   rA   rm   r   rQ   register_openr~   register_extensionregister_saveregister_mimerj   rd   argvprintexitopenrn   Zimfrq   r   r   r*   platformZ	startfiler   r   r   r   <module>   sF   
+
"gJ>
*