File: //lib/python3/dist-packages/PIL/__pycache__/ImageFile.cpython-38.pyc
U
tCf
R � @ s� d dl Z d dlZd dlZddlmZ ddlmZ dZdZdZ dd d
d d d
�Z
dd� Z dd� Z G dd� dej�Z
G dd� de
�ZG dd� d�Zd dd�Zdd� ZG d d � d �ZG d d � d �ZdS )!� N� )�Image)�isPathi i Fzimage buffer overrun errorzdecoding errorz
unknown errorzbad configurationzout of memory error)���������i����i����c C sL zt j�| �}W n tk
r. t�| �}Y nX |s<d| }t|d ��d S )Nzdecoder error %dz when reading image file)r �coreZgetcodecstatus�AttributeError�ERRORS�get�OSError)�error�message� r �//usr/lib/python3/dist-packages/PIL/ImageFile.py�
raise_ioerror4 s r c C s | d S )N� r )�tr r r � _tilesortC s r c sR e Zd ZdZd� fdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
� Z S )� ImageFilez*Base class for image file format handlers.Nc s� t � �� d| _d | _d | _d| _d| _t| _t |�rPt
|d�| _ || _ d| _
n|| _ || _ d | _
zdz | �� W n6 tttttjfk
r� } z t|��W 5 d }~X Y nX | jr�| jd dkr�td��W n& tk
r� | j
r�| j �� � Y nX d S )Nr r r �rbTz not identified by this driver)�super�__init__�
_min_frame�custom_mimetype�tile�readonly�
decoderconfig�MAXBLOCK�decodermaxblockr �open�fp�filename�
_exclusive_fp�_open�
IndexError� TypeError�KeyError�EOFError�structr
� SyntaxError�mode�size�
BaseException�close)�selfr! r" �v�� __class__r r r P s>
�
zImageFile.__init__c C s, | j r | j S | jd k r(tj�| j�� �S d S �N)r �formatr ZMIMEr �upper�r/ r r r �get_format_mimetype} s
z ImageFile.get_format_mimetypec C s | j r| j�� d| _dS )zCheck file integrityN)r# r! r. r6 r r r �verify� s
zImageFile.verifyc
C s� t j �| �}| jdkr td��| js(|S d| _| jo@t| j�dk}|oPttd� }d}z| j }d}W n t
k
r� | j j }Y nX z| j
}d}W n t
k
r� | j j}Y nX |�r�| jd \}}}} |dk�r�t| �dk�r�| d | jk�r�| d t jk�r�z�tt jd ��rLt j�| j�| _| j�|� | j�| j| j| d | d
�| _nTddl}
t| jd �� } |
j| �� d|
jd �| _W 5 Q R X t j�| j| j|||| �| _d}| j�r�d| j_W n" t
tt fk
�r� d| _Y nX | � � d
} | j�s4| jj t d� z
| j }
W n t
k
�r d}
Y nX | jD �]\}}}} t �!| j|| | j"�}z�||� |�$| j|� |j%�r�|�&| j � |�'d�\}} n�|
}z|| j(�}W n2 t)t*j+fk
�r� t,�r�Y �q"ntd��Y nX |�s�t,�rސq"ntdt|� ��|| }|�'|�\}} |dk �r�q"||d� }�q�W 5 |�#� X �q(g | _|| _-| �.� | j/�rb| j0�rb| j �1� d| _ | j�s�t,�s�| dk �r�t2| � t j �| �S )z"Load image data based on tile listNzcannot load this imager Zpypy_version_infor F�raw� �mapr �r)�accessr ��key� zimage file is truncatedz0image file is truncated (%d bytes not processed))3r �loadr r r; r" �len�hasattr�sys� load_readr r! �read� load_seek�seekr+ Z _MAPMODESr Z readimager, �im�mmapr �filenoZ ACCESS_READZ
map_bufferZpaletteZdirty� ImportError� load_prepare�sortr Z tile_prefix� _getdecoderr �cleanup�setimage�pulls_fd�setfd�decoder r% r) r
�LOAD_TRUNCATED_IMAGESr �load_endr# Z!_close_exclusive_fp_after_loadingr. r )r/ ZpixelZuse_mmapr rF rH Z decoder_name�extents�offset�argsrJ r! Zerr_code�prefix�decoderZstatus�b�s�nr r r rA � s�
�
� � � � � �
� ��
zImageFile.loadc C sP | j r"| j j| jks"| j j| jkr6tj�| j| j�| _ | jdkrLtj�| � d S )N�P)rI r+ r, r r �newrA r6 r r r rM s "
zImageFile.load_preparec C s d S r3 r r6 r r r rV s zImageFile.load_endc C sB || j k s.t| d�r | jd ks6|| j| j kr6td��| �� |kS )N� _n_framesz attempt to seek outside sequence)r rC ra Zn_framesr( �tell)r/ �framer r r � _seek_check& s ���� zImageFile._seek_check)NN) �__name__�
__module__� __qualname__�__doc__r r7 r8 rA rM rV rd �
__classcell__r r r1 r r M s - r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �
StubImageFilez�
Base class for stub image loaders.
A stub loader is an image loader that can identify files of a
certain format, but relies on external code to load the file.
c C s t d��d S )Nz+StubImageFile subclass must implement _open��NotImplementedErrorr6 r r r r$ = s zStubImageFile._openc C sH | � � }|d kr td| j ��|�| �}|d k s4t�|j| _|j| _d S )Nz#cannot find loader for this %s file)�_loadr r4 rA �AssertionErrorr2 �__dict__)r/ �loader�imager r r rA @ s
zStubImageFile.loadc C s t d��dS )z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadNrk r6 r r r rm J s zStubImageFile._loadN)re rf rg rh r$ rA rm r r r r rj 5 s
rj c @ sP e Zd ZdZdZdZdZdZdZdZ dd� Z
dd� Z dd � Z d
d � Z
d d
� ZdS )�Parserzj
Incremental image parser. This class implements the standard
feed/close consumer interface.
Nr c C s | j dkstd��dS )z�
(Consumer) Reset the parser. Note that you can only call this
method immediately after you've created a parser; parser
instances cannot be reused.
Nzcannot reuse parsers)�datarn r6 r r r �reset\ s z Parser.resetc C s� | j r
dS | jdkr || _n | j| | _| jr�| jdkrztt| j�| j�}| j|d� | _| j| | _| jdksv| jszdS | j�| j�\}}|dk r�d| _d| _ |dk r�d| _t|� ndS | j|d� | _n�| jr�n�z&t �
| j��}t � |�}W 5 Q R X W n t
k
�r Y n�X t|d��p(t|d�}|�s@t|j�dk�rHd| _nv|�� |jd \}}} }
g |_t �|j||
|j�| _| j�|j|� | | _| jt| j�k�r�| j| jd� | _d| _|| _dS )z�
(Consumer) Feed data to the parser.
:param data: A string buffer.
:exception IOError: If the parser failed to parse the image file.
Nr r rG rE )�finishedrs r[ rX �minrB rT rq r �io�BytesIOr r r rC r rM rO r+ r rQ rI ) r/ rs �skipr^ �er! rI �flag�d�o�ar r r �feedd sR
z Parser.feedc C s | S r3 r r6 r r r � __enter__� s zParser.__enter__c G s | � � d S r3 )r. �r/ rY r r r �__exit__� s zParser.__exit__c C sz | j r*| �d� d | _| _ | js*td��| js8td��| jrtt�| j��$}zt �
|�| _W 5 | j�� X W 5 Q R X | jS )a
(Consumer) Close the stream.
:returns: An image object.
:exception IOError: If the parser failed to parse the image file either
because it cannot be identified or cannot be
decoded.
r@ Nzimage was incompletezcannot parse this image) r[ r rs ru r rq rw rx rA r r )r/ r! r r r r. � s
z Parser.close)re rf rg rh Z incrementalrq rs r[ rX ru rt r r� r� r. r r r r rr O s Prr c
C s� | � � t| d�sd| _|jtd� tt|| jd d �}|tj krP|�
� dS z|� � }|�
� W n� t t
jfk
�r& |D ]�\}}}}t�| j||| j�}|dkr�|�|� |�| j|� |jr�|�|� |�� \} }
n$|�|�\} }
} |�| � |
rސqq�|
dk �rtd|
��|�� q�Y n�X |D ]�\}}}}t�| j||| j�}|dk�r`|�|� |�| j|� |j�r�|�|� |�� \} }
n |� ||�}
|
dk �r�td|
��|�� �q,t|d��r�|�
� dS ) z�Helper to save image based on tile list
:param im: Image object.
:param fp: File object.
:param tile: Tile list.
:param bufsize: Optional buffer size
�
encoderconfigr r> r � Nz(encoder error %d when writing image file�flush) rA rC r� rN r �maxr r, rD �stdoutr� rK r rw �UnsupportedOperationr Z _getencoderr+ rH rQ rI Z pushes_fdrS Zencode_to_pyfd�encode�writer rP Zencode_to_file) rI r! r �bufsizeZfhrz r\ r} r~ �lr] r| r r r �_save� sR
r� c C s� |dkr dS |t kr6| �|�}t|�|k r2td��|S g }|dkrp| �t|t ��}|sXqp|�|� |t|�8 }q:tdd� |D ��|k r�td��d�|�S )a�
Reads large blocks in a safe way. Unlike fp.read(n), this function
doesn't trust the user. If the requested size is larger than
SAFEBLOCK, the file is read block by block.
:param fp: File handle. Must implement a <b>read</b> method.
:param size: Number of bytes to read.
:returns: A string containing <i>size</i> bytes of data.
Raises an OSError if the file is truncated and the read can not be completed
r r@ zTruncated File Readc s s | ]}t |�V qd S r3 )rB )�.0r| r r r � <genexpr>3 s z _safe_read.<locals>.<genexpr>)� SAFEBLOCKrF rB r rv �append�sum�join)r! r, rs �blockr r r �
_safe_read s"
r� c @ s e Zd Zdd� Zdd� ZdS )� PyCodecStatec C s d| _ d| _d| _d| _d S )Nr )�xsize�ysize�xoff�yoffr6 r r r r ; s zPyCodecState.__init__c C s | j | j| j | j | j| j fS r3 )r� r� r� r� r6 r r r rW A s zPyCodecState.extentsN)re rf rg r rW r r r r r� : s r� c @ s\ e Zd ZdZdZdd� Zdd� Zedd� �Zd d
� Z d d � Z
d
d� Z ddd�Z ddd�Z
dS )� PyDecoderz�
Python implementation of a format decoder. Override this class and
add the decoding logic in the `decode` method.
See :ref:`Writing Your Own File Decoder in Python<file-decoders-py>`
Fc G s( d | _ t� | _d | _|| _| �|� d S r3 )rI r� �state�fdr+ �init)r/ r+ rY r r r r O s
zPyDecoder.__init__c C s
|| _ dS )z�
Override to perform decoder specific initialization
:param args: Array of args items from the tile entry
:returns: None
N)rY r� r r r r� V s zPyDecoder.initc C s | j S r3 )� _pulls_fdr6 r r r rR _ s zPyDecoder.pulls_fdc C s
t � �dS )a'
Override to perform the decoding process.
:param buffer: A bytes object with the data to be decoded.
:returns: A tuple of (bytes consumed, errcode).
If finished with decoding return <0 for the bytes consumed.
Err codes are from `ERRORS`
Nrk )r/ �bufferr r r rT c s zPyDecoder.decodec C s dS )zV
Override to perform decoder specific cleanup
:returns: None
Nr r6 r r r rP n s zPyDecoder.cleanupc C s
|| _ dS )z�
Called from ImageFile to set the python file-like object
:param fd: A python file-like object
:returns: None
N)r� )r/ r� r r r rS v s zPyDecoder.setfdNc C s� || _ |r|\}}}}n d\}}}}|dkrJ|dkrJ| j j\| j_| j_n(|| j_|| j_|| | j_|| | j_| jjdks�| jjdkr�td��| jj| jj | j jd ks�| jj| jj | j jd kr�td��dS )z�
Called from ImageFile to set the core output image for the decoder
:param im: A core image object
:param extents: a 4 tuple of (x0, y0, x1, y1) defining the rectangle
for this tile
:returns: None
)r r r r r zSize cannot be negativer z Tile cannot extend outside imageN)rI r, r� r� r� r� r� �
ValueError)r/ rI rW Zx0Zy0Zx1Zy1r r r rQ s" ��zPyDecoder.setimagec C sd |s
| j }t�| j d|�}|�| j| j�� � |�|�}|d dkrLtd��|d dkr`td��dS )a
Convenience method to set the internal image from a stream of raw data
:param data: Bytes to be set
:param rawmode: The rawmode to be used for the decoder.
If not specified, it will default to the mode of the image
:returns: None
r9 r znot enough image datar zcannot decode image dataN) r+ r rO rQ rI r� rW rT r� )r/ rs Zrawmoder| r] r r r �
set_as_raw� s
zPyDecoder.set_as_raw)N)N)re rf rg rh r� r r� �propertyrR rT rP rS rQ r� r r r r r� E s
#r� )r )rw r) rD � r Z_utilr r r� rU r
r r r rj rr r� r� r� r� r r r r �<module> s0 �
i
>"