File: //usr/lib/python3/dist-packages/PIL/__pycache__/IcoImagePlugin.cpython-38.pyc
U
tCf�' � @ s� d dl Z d dlZd dlmZ d dlmZmZ ddlmZm Z m
Z
m Z ddl m
Z
mZmZ dZdd � Zd
d � ZG d d
� d
�ZG dd� de
j
�Ze �ejee� e �eje� e �ejd� e �ejd� dS )� N)�BytesIO)�ceil�log� )�BmpImagePlugin�Image� ImageFile�PngImagePlugin)�i8�i16le�i32les c
s� |� t� | j�ddddddddg�}| j\�� t� �fd d
�|�}t|�}|� t�d t |��� |�
� t |�d }|D �]}|\�� |� t�d
�dk r��nd�� |� t�d
� dk r�� nd�� |� d� |� d� |� d� |� t�d d�� t � }| � � }|j
|tjd d� |�|d� |�d� |�� }t |�} |� t�d| �� |� t�d|�� |�
� }
|�|� |� |� || }|�|
� qzd S )N�sizes)� r )� r )� r )�0 r )�@ r )� r )� r c s8 | d �ks0| d � ks0| d dks0| d dkr4dS dS )Nr r r FT� ��x��height�widthr �4/usr/lib/python3/dist-packages/PIL/IcoImagePlugin.py�<lambda>/ s
�
�
�
�z_save.<locals>.<lambda>z<Hr �Br r � s r )Z reducing_gapZpngz<I)�write�_MAGICZ encoderinfo�get�size�filter�list�structZpack�len�tellr �copyZ thumbnailr ZLANCZOSZsave�seek�read) �im�fp�filenamer
�offsetr" Zimage_ioZtmpZ image_bytesZ bytes_lenZcurrentr r r �_save' sF
�
�
r/ c C s | d d� t kS )N� )r )�prefixr r r �_acceptR s r2 c @ s8 e Zd Zdd� Zdd� Zd
dd�Zddd �Zd
d � Zd S )�IcoFilec C sj |� d�}t|�std��|| _g | _t|dd� �| _t| j�D ]�}|� d�}t|d �t|d �t|d �t|d �t|dd� �t|dd� �t |d
d� �t |d d� �d �}d
D ]}|| s�d||< q�|d p�|d dkr�t
t |d d��p�d|d< |d |d f|d< |d |d |d< | j� |� qBt
| jdd� d�| _t
| jdd� d�| _| j�� dS )zL
Parse image from file-like object containing ico file data
� znot an ICO filer0 Nr r r � � � � )r r �nb_colorZreservedZplanes�bppr" r. �r r r r: r9 � color_depthr r �dim�squarec S s | d S )Nr< r r r r r r � � z"IcoFile.__init__.<locals>.<lambda>)�keyc S s | d S )Nr> r r r r r r � r? )r* r2 � SyntaxError�buf�entry�i16Znb_items�ranger
�i32r r �append�sorted�reverse)�selfrB �s�iZ icon_header�jr r r �__init__W s@
�
�� zIcoFile.__init__c C s dd� | j D �S )zJ
Get a list of all available icon sizes and color depths.
c S s h | ]}|d |d f�qS r; r )�.0�hr r r � <setcomp>� s z IcoFile.sizes.<locals>.<setcomp>)rC �rJ r r r r
� s z
IcoFile.sizesFc C s@ t | j�D ]0\}}||d kr
|dks2||d kr
| S q
dS )Nr= Fr< r )� enumeraterC )rJ r" r: rL rP r r r �
getentryindex� s
zIcoFile.getentryindexc C s | � | �||��S )z,
Get an image from the icon
)�framerT )rJ r" r: r r r �getimage� s zIcoFile.getimagec
C s | j | }| j�|d � | j�d�}| j�|d � |dd� tjkrdt�| j�}t�|j � �n�t
� | j�}t�|j � |j d t |j d d �f|_
|jd \}}}}|d|j ||f|jd< |d } d}
t
j�� D ] \} } | | d kr�| }
q�q�d|
k�rN| j�|� | j�|j d |j d d �d
dd � }
t�d |j |
d d
�}n�|j d }|d dk�r||d|j d d 7 }|t |j d |j d |
d � }t ||j d d �}| j�|� | j�|�}t�d|j |d dt |d �df�}|�d�}|�|� |S )z-
Get an image from frame idx
r. r7 Nr r r5 )r r r r0 r6 �L�raw)rW r ���g @�1z1;IrY ZRGBA)rC rB r) r* r r Z PngImageFiler Z_decompression_bomb_checkr" r Z DibImageFile�int�_sizeZtileZBIT2MODE�itemsZ
frombufferZconvertZputalpha)rJ �idx�header�datar+ �d�e�o�a�moder: �k�vZ alpha_bytes�mask�wZand_mask_offsetZ total_bytesZ mask_datar r r rU � sZ
*�
$ �
z
IcoFile.frameN)F)F)�__name__�
__module__� __qualname__rN r
rT rV rU r r r r r3 V s
:
r3 c @ sJ e Zd ZdZdZdZdd� Zedd� �Zej dd� �Zd d
� Z
d d � Z d
S )� IcoImageFilea�
PIL read-only image support for Microsoft Windows .ico files.
By default the largest resolution image in the file will be loaded. This
can be changed by altering the 'size' attribute before calling 'load'.
The info dictionary has a key 'sizes' that is a list of the sizes available
in the icon file.
Handles classic, XP and Vista icon formats.
When saving, PNG compression is used. Support for this was only added in
Windows Vista.
This plugin is a refactored version of Win32IconImagePlugin by Bryan Davis
<casadebender@gmail.com>.
https://code.google.com/archive/p/casadebender/wikis/Win32IconImagePlugin.wiki
ZICOz Windows Iconc C s: t | j�| _| j�� | jd< | jjd d | _| �� d S )Nr
r r= )r3 r, �icor
�inforC r" �loadrR r r r �_open s zIcoImageFile._openc C s | j S �N)r\ rR r r r r" s zIcoImageFile.sizec C s || j d krtd��|| _d S )Nr
z2This is not one of the allowed sizes of this image)ro �
ValueErrorr\ )rJ �valuer r r r" s c C s� | j r| j j| jkrd S | j�| j�}|�� |j | _ |j| _|j| jkr�t�d� | j�| j�}t | j
d �}|j||< t |�| j
d< |j| _d S )Nz Image was not the expected sizer
) r+ r" rn rV rp re �warnings�warnrT r$ ro �set)rJ r+ �indexr
r r r rp $ s
zIcoImageFile.loadc C s d S rr r rR r r r � load_seek7 s zIcoImageFile.load_seekN) rj rk rl �__doc__�formatZformat_descriptionrq �propertyr" �setterrp ry r r r r rm � s
rm z.icoz image/x-icon) r% ru �ior Zmathr r � r r r r Z_binaryr
r rD r rF r r/ r2 r3 rm Z
register_openr{ Z
register_saveZregister_extensionZ
register_mimer r r r �<module> s + (D