File: //lib/python3/dist-packages/PIL/__pycache__/WebPImagePlugin.cpython-38.pyc
U
g�
^+ � @ s� d dl mZ ddlmZmZ zddlmZ dZW n ek
rH dZY nX dddd�Zddd�Z d d
d
d �Z
d d
� Z G dd� dej�Z dd� Z
dd� Ze�e je e � er�e�e je� ejr�e�e je
� e�e jd� e�e jd� dS )� )�BytesIO� )�Image� ImageFile)�_webpTF)�RGBX�RGBA�RGB)r r r r )s VP8 s VP8Xs VP8Lc C sL | d d� dk}| dd� dk}| dd� t k}|rH|rH|rHtsDdS dS d S ) N� s RIFF� � s WEBP� zEimage file could not be identified because WEBP support not installedT)�_VP8_MODES_BY_IDENTIFIER� SUPPORTED)�prefixZis_riff_file_formatZ is_webp_fileZis_valid_vp8_mode� r �5/usr/lib/python3/dist-packages/PIL/WebPImagePlugin.py�_accept s �r c s~ e Zd ZdZdZdd� Zdd� Zedd� �Zed d
� �Z � fd d �Z
ddd�Z dd� Z dd� Z
� fdd�Z� fdd�Z� ZS )�
WebPImageFileZWEBPz
WebP imagec C s� t jsvt �| j�� �\}}}| _}}|r2|| jd< |r@|| jd< ||f| _t|�| _dd| j d| jfg| _
d| _ d S t � | j�� �| _
| j
�� \}}}}}} ||f| _|| jd< |d? d @ |d
? d @ |d ? d @ |d @ f\}
} } }
| | |
|
f| jd < || _ | d
k�rdn| | _| | _g | _
| j
�d�}| j
�d�}| j
�d�}|�rN|| jd< |�r^|| jd< |�rn|| jd< | jdd� | �d� d S )N� icc_profile�exif�raw�r r r r �loop� � r
r �
backgroundr r ZICCPZEXIFzXMP �xmpF)�reset)r �
HAVE_WEBPANIMZ
WebPDecode�fp�read�mode�infoZ_sizer �size�tile� _n_framesZWebPAnimDecoder�_decoderZget_info�rawmodeZ get_chunk�_reset�seek)�self�data�widthZheightr r Z
loop_countZbgcolorZ frame_countr" �bg_a�bg_r�bg_g�bg_br r r r �_open* sL �
�
zWebPImageFile._openc C s d| j krd S t| �� �S )Nr )r# �dictZgetexif�r+ r r r �_getexif] s
zWebPImageFile._getexifc C s | j S �N�r& r4 r r r �n_framesb s zWebPImageFile.n_framesc C s
| j dkS )Nr r7 r4 r r r � is_animatedf s zWebPImageFile.is_animatedc s> t jst� �|�S || jkr$td��|dk r4td��|| _d S )Nz(attempted to seek beyond end of sequencer z!negative frame index is not valid)r r �superr* r& �EOFError� _WebPImageFile__logical_frame�r+ �frame�� __class__r r r* j s
zWebPImageFile.seekTc C s$ |r| j �� d| _d| _d| _d S )Nr ���)r' r � _WebPImageFile__physical_frame�_WebPImageFile__loaded�_WebPImageFile__timestamp)r+ r r r r r) w s
zWebPImageFile._resetc C sd | j �� }| jd7 _|d kr:| �� | �d� td��|\}}|| j }|| _||8 }|||fS )Nr r z(failed to decode next frame in WebP file)r' Zget_nextrB r) r* r; rD )r+ Zretr, � timestamp�durationr r r � _get_next~ s
zWebPImageFile._get_nextc C s8 | j |krd S || j k r | �� | j |k r4| �� q d S r6 )rB r) rG r= r r r �_seek� s
zWebPImageFile._seekc s� t jr�| j| jkr�| �| j� | �� \}}}|| jd< || jd< | j| _| jr^| jr^| j� � t
|�| _dd| j d| j fg| _
t� �� S )NrE rF r r r )r r rC r< rH rG r# r Z
_exclusive_fp�closer r$ r( r% r: �load)r+ r, rE rF r? r r rJ � s
zWebPImageFile.loadc s t jst� �� S | jS r6 )r r r: �tellr< r4 r? r r rK � s
zWebPImageFile.tell)T)�__name__�
__module__� __qualname__�formatZformat_descriptionr2 r5 �propertyr8 r9 r* r) rG rH rJ rK �
__classcell__r r r? r r % s 3
r c ' C s� | j �� }t|�dg ��}d}| g| D ]}|t|dd�7 }q(|dkrVt| ||� d S d}d|krl|d }nTd| jkr�| jd }t|t�r�| � � }|r�||d |d d � \} }
} | |
| df}| j �dd�} | j �d d�}
| j �d
d �}| j �d d �}| j �d
d �}| j �dd �}d }| j �dd �}| j �dd�}| j �dd�}| j �dd�}| j �dd�}t|t
j ��rt|� � }| j �dd�}|�r�d }|d k�r�|�r�dnd}|d k�r�|�r�dnd}t|tt
f��r�t|�dk�s�tdd � |D ���s�td t|� ��|\}}}} | d > |d > B |d > B |d> B }t�| jd | jd ||
|||||� } d} d} | �� } �z | g| D ]�}t|dd�}!t|!�D ]�}"|�|"� |�� |}#|j}$|jtk�r�d!|jk�p�d"|jk�p�|jd#k�o�d!|j� � k}%|%�r�d$nd%}$|� |$�}#|$d%k�r
d&}$| � |#� d'|$�| |#jd |#jd |$|||� t| tt
f��rT| | | 7 } n| | 7 } | d7 } �q��qnW 5 | �| � X | � d | ddd||d� | � |||�}&|&d k�r�td(��|� |&� d S ))N�
append_imagesr r8 r )r r r r r � rF r �
minimize_sizeF�kmin�kmax� allow_mixed�lossless�quality�P �methodr � r r � � � r
c s s | ]}|d ko|dk V qdS )r � Nr )�.0�vr r r � <genexpr>� s z _save_all.<locals>.<genexpr>z<Background color is not an RGBA tuple clamped to (0-255): %sr r
r �A�a�Pr r r r �1cannot write file as WebP (encoder returned None))!� encoderinfo�copy�list�get�getattr�_saver# �
isinstance�intZ
getpaletter �Exif�tobytes�tuple�len�all�OSError�strr ZWebPAnimEncoderr$ rK r* �rangerJ r" �_VALID_WEBP_MODES�im�getpalettemode�convert�addZassemble�write)'ry r �filenamerh rR ZtotalZimsr Zpalette�r�g�brF r rT rU rV rW �verboserX rY r[ r r r r/ r0 r1 r. �encZ frame_idxrE Zcur_idxZnfr�idxr> r( �alphar, r r r � _save_all� s�
�
���� �
��
�
r� c
C s� | j �dd�}| j �dd�}| j �dd�}| j �dd�}t|tj�rL|�� }| j �dd�}| jtkr�d | jkp�d
| jkp�| jd ko�d | j� � k}| �
|r�d nd
�} t � | �� | j
d | j
d |t|�| j|||� } | d kr�td��|�| � d S )NrX FrY rZ r r\ r r rd re rf r r r r rg )rh rk rn r rp rq r" �_VALID_WEBP_LEGACY_MODESry rz r{ r Z
WebPEncoder$ �floatru r} )
ry r r~ rX rY r r r r� r, r r r rm @ s8
��� rm z.webpz
image/webpN)�ior r\ r r r r � ImportErrorrx r� r r r r� rm Z
register_openrO Z
register_saver Zregister_save_allZregister_extensionZ
register_mimer r r r �<module> s2
�
"