File: //home/arjun/projects/env/lib/python3.10/site-packages/PIL/__pycache__/ImageFile.cpython-310.pyc
o
we�[ � @ s� d dl Z d dlZd dlZd dlZddlmZ ddlmZ dZdZ dZ
dd d
d d d
�Z dd� Z dd� Z
G dd� dej�ZG dd� de�ZG dd� d�Zd&dd�Zd'dd�Zd d � ZG d d � d �ZG d d!� d!�ZG d"d#� d#e�ZG d$d%� d%e�ZdS )(� N� )�Image)�is_pathi i Fzimage buffer overrun errorzdecoding errorz
unknown errorzbad configurationzout of memory error)������������i����c C sL zt j�| �}W n ty t�| �}Y nw |s d| � �}|d7 }t|��)Nzdecoder error z when reading image file)r �core�getcodecstatus�AttributeError�ERRORS�get�OSError)�error�msg� r �F/home/arjun/projects/env/lib/python3.10/site-packages/PIL/ImageFile.py�
raise_oserror@ s �
r c C s | d S )N� r )�tr r r � _tilesortK s r c s^ e Zd ZdZd� fdd� Zdd� Z� fdd�Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z � Z S )� ImageFilez*Base class for image file format handlers.Nc s� t � �� d| _d | _d | _ d| _d| _t| _t |�r)t
|d�| _ || _ d| _
n || _ || _ d | _
z8z| �� W n tttttjfyP } zt|�|�d }~ww | jrb| jd dksb| jd dkrhd}t|��W d S tyz | j
ry| j �� � w )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$ �vr �� __class__r r r X sJ
�
��" �
�zImageFile.__init__c C s, | j r| j S | jd urtj�| j�� �S d S �N)r �formatr �MIMEr
�upper�r1 r r r �get_format_mimetype� s
�z ImageFile.get_format_mimetypec s g | _ t� �|� d S r5 )r r � __setstate__)r1 �stater3 r r r; � s zImageFile.__setstate__c C s | j r| j�� d| _dS )zCheck file integrityN)r% r# r0 r9 r r r �verify� s
zImageFile.verifyc C s� | j du r d}t|��tj�| �}| j s|S d| _| jo"t| j �dk}|o*ttd� }d}z| j }d}W n
t
yA | j j }Y nw z| j
}d}W n
t
yV | j j}Y nw |r�| j d \}}} }
|dkr�t|
�dkr�|
d | jkr�|
d tjv r�zOddl} t| j��} | j| �� d| jd �| _W d � n1 s�w Y | | jd |
d | j�� kr�t�tj�| j| j|| |
�| _d}| jr�d| j_W n t
ttfy� d| _Y nw | � � d
}
| j�s�| j j t d � z| j }W n t
y� d }Y nw d
d� t �!| j dd� �D �| _ | j D ]�\}}} }
|| � t�"| j||
| j#�}zw|�$| j|� |j%�r;|�&| j � |�'d �d }
nW|} z|| j(�}W n t)t*j+f�ye } zt,�rZW Y d}~n8d}t|�|�d}~ww |�syt,�rmn%dt|�� d�}t|��|| }|�'|�\}}
|dk �r�n||d� }�q>W |�-� �q
|�-� w g | _ || _.| �/� | j0�r�| j1�r�| j �2� d| _ | j�s�t,�s�|
dk �r�t3|
� tj�| �S )z"Load image data based on tile listNzcannot load this imager �pypy_version_infor F�raw� )�accessr ��key� c S s g | ]
\}}t |�d �qS )r )�list)�.0�_�tilesr r r �
<listcomp>� s
��z"ImageFile.load.<locals>.<listcomp>c S s | d | d | d fS )Nr r r@ r )r r r r �<lambda>� s z ImageFile.load.<locals>.<lambda>Tzimage file is truncatedzimage file is truncated (z bytes not processed))4r r r �load�mapr$ �len�hasattr�sys� load_readr r# �read� load_seek�seekr- � _MAPMODES�mmapr" �fileno� ACCESS_READr. r �
map_buffer�im�palette�dirty� ImportError� load_prepare�sortr � tile_prefix� itertools�groupby� _getdecoderr �setimage�pulls_fd�setfd�decoder! r' r+ r �LOAD_TRUNCATED_IMAGES�cleanupr �load_endr% �!_close_exclusive_fp_after_loadingr0 r )r1 r �pixel�use_mmapr rQ rS � decoder_name�extents�offset�argsrU r# �err_code�prefix�decoder�b�s�e�nr r r rK � s�
� � � ��
�
�
�� �
����
��
zImageFile.loadc C sT | j r| j j| jks| j j| jkrtj�| j| j�| _ | jdkr(tj�| � d S d S )N�P)rY r- r. r r �newrK r9 r r r r] $ s
"
�zImageFile.load_preparec C s d S r5 r r9 r r r ri , s zImageFile.load_endc C sF || j k st| d�r| jd u s || j| j kr d}t|��| �� |kS )N� _n_framesz attempt to seek outside sequence)r rN rz �n_framesr* �tell)r1 �framer r r r � _seek_check8 s
�
zImageFile._seek_check)NN)
�__name__�
__module__� __qualname__�__doc__r r: r; r= rK r] ri r~ �
__classcell__r r r3 r r U s 0
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 � d}t |��)Nz+StubImageFile subclass must implement _open��NotImplementedError�r1 r r r r r& P s zStubImageFile._openc C sT | � � }|d u rd| j� d�}t|��|�| �}|d us J �|j| _|j| _|�� S )Nz cannot find loader for this z file)�_loadr6 r rK r4 �__dict__)r1 �loaderr �imager r r rK T s
zStubImageFile.loadc C r� )z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadr� r� r r r r� ` s zStubImageFile._loadN)r r� r� r� r& rK r� r r r r r� H s
r� 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 du s J d��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)�datar9 r r r �resets s z Parser.resetc C s� | j rdS | jdu r|| _n| j| | _| jrh| jdkr=tt| j�| j�}| j|d� | _| j| | _| jdks;| js=dS | j�| j�\}}|dk r^d| _d| _ |dk r\d| _t|� ndS | j|d� | _dS | jrmdS z t �
| j��
}t � |�}W d � n1 s�w Y W n
t
y� Y dS w t|d�p�t|d�}|s�t|j�dkr�d| _n:|�� |jd \}}} }
g |_t �|j||
|j�| _| j�|j|� | | _| jt| j�kr�| j| jd� | _d| _|| _dS )z�
(Consumer) Feed data to the parser.
:param data: A string buffer.
:exception OSError: If the parser failed to parse the image file.
Nr r rR rP )�finishedr� rs ro �minrM rf r� r �io�BytesIOr r" r rN r r] rb r- r rc rY ) r1 r� �skiprw rv r# rY �flag�d�o�ar r r �feed{ sX
�� �
z Parser.feedc C s | S r5 r r9 r r r � __enter__� s zParser.__enter__c G s | � � d S r5 )r0 �r1 rp r r r �__exit__� s zParser.__exit__c C s� | j r| �d� d | _| _ | jsd}t|��| js d}t|��| jrOt�| j�� }z
t� |�| _W | j�
� n| j�
� w W d � | jS 1 sJw Y | jS )a
(Consumer) Close the stream.
:returns: An image object.
:exception OSError: If the parser failed to parse the image file either
because it cannot be identified or cannot be
decoded.
rD Nzimage was incompletezcannot parse this image) rs r� r� r� r r� r� r� r r" rK )r1 r r# r r r r0 � s$
��z Parser.close)r r� r� r� � incrementalr� r� rs ro r� r� r� r� r� r0 r r r r r� f s M r� c
C s� | � � t| d�s d| _|jtd� tt|| jd d �}z|�� }|� � t
| ||||� W n t t j
fyM } zt
| |||d|� W Y d}~nd}~ww t|d�rY|� � dS 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 rB r � N�flush)rK rN r� r^ r �maxr r. rV r� � _encode_tiler r� �UnsupportedOperation)rY r# r �bufsize�fh�excr r r �_save� s
��
�r� c C s� |D ]g\}}}} |dkr|� |� t�| j|| | j�}
zI|
�| j|� |
jr2|
�|� |
� � d } n |rJ |
�
|�dd � \} } |� | � | rHnq5n|
� ||�} | dk r_d| � d�}
t
|
�|�W |
�� q|
�� w d S )Nr r Tzencoder error z when writing image file)rS r � _getencoderr- r� rc rY � pushes_fdre �encode_to_pyfd�encode�write�encode_to_filer rh )rY r# r r� r� r� rv rt r� r� �encoder�errcoder� r r r r r� s0
�
��r� c C s� |dkrdS |t kr | �|�}t|�|k rd}t|��|S g }|}|dkr?| �t|t ��}|s0n|�|� |t|�8 }|dks%tdd� |D ��|k rPd}t|��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 cannot be completed
r rD zTruncated File Readc s s � | ]}t |�V qd S r5 )rM )rF r� r r r � <genexpr>E s � z _safe_read.<locals>.<genexpr>)� SAFEBLOCKrQ rM r r� �append�sum�join)r# r. r� r �remaining_size�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�yoffr9 r r r r L s
zPyCodecState.__init__c C s | j | j| j | j | j| j fS r5 )r� r� r� r� r9 r r r rn R s zPyCodecState.extentsN)r r� r� r rn r r r r r� K s r� c @ s6 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
d �Zd S )
�PyCodecc G s( d | _ t� | _d | _|| _| �|� d S r5 )rY r� r< �fdr- �init)r1 r- rp r r r r W s
zPyCodec.__init__c C �
|| _ dS )z�
Override to perform codec specific initialization
:param args: Array of args items from the tile entry
:returns: None
N)rp r� r r r r� ^ �
z PyCodec.initc C s dS )zT
Override to perform codec specific cleanup
:returns: None
Nr r9 r r r rh g s zPyCodec.cleanupc C r� )z�
Called from ImageFile to set the Python file-like object
:param fd: A Python file-like object
:returns: None
N)r� )r1 r� r r r re o r� z
PyCodec.setfdNc C s� || _ |r |\}}}}nd\}}}}|dkr%|dkr%| j j\| j_| j_n|| j_|| j_|| | j_|| | j_| jjdksE| jjdkrKd}t|��| jj| jj | j jd ksg| jj| jj | j jd krmd}t|��dS )z�
Called from ImageFile to set the core output image for the codec
: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)rY r. r< r� r� r� r� �
ValueError)r1 rY rn �x0�y0�x1�y1r r r r rc x s$ �zPyCodec.setimager5 )r r� r� r r� rh re rc r r r r r� V s r� c @ s2 e Zd ZdZdZedd� �Zdd� Zd
dd �ZdS ) � PyDecoderz�
Python implementation of a format decoder. Override this class and
add the decoding logic in the :meth:`decode` method.
See :ref:`Writing Your Own File Codec in Python<file-codecs-py>`
Fc C � | j S r5 )� _pulls_fdr9 r r r rd � � zPyDecoder.pulls_fdc C � t � �)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 -1 for the bytes consumed.
Err codes are from :data:`.ImageFile.ERRORS`.
r� )r1 �bufferr r r rf � � zPyDecoder.decodeNc C sl |s| j }t�| j d|�}|�| j| j�� � |�|�}|d dkr(d}t|��|d dkr4d}t|��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
r? r znot enough image datar zcannot decode image dataN) r- r rb rc rY r< rn rf r� )r1 r� �rawmoder� ru r r r r �
set_as_raw� s
�zPyDecoder.set_as_rawr5 ) r r� r� r� r� �propertyrd rf r� r r r r r� � s
r� c @ s8 e Zd ZdZdZedd� �Zdd� Zdd� Zd d
� Z d S ) � PyEncoderz�
Python implementation of a format encoder. Override this class and
add the decoding logic in the :meth:`encode` method.
See :ref:`Writing Your Own File Codec in Python<file-codecs-py>`
Fc C r� r5 )�
_pushes_fdr9 r r r r� � r� zPyEncoder.pushes_fdc C r� )a
Override to perform the encoding process.
:param bufsize: Buffer size.
:returns: A tuple of ``(bytes encoded, errcode, bytes)``.
If finished with encoding return 1 for the error code.
Err codes are from :data:`.ImageFile.ERRORS`.
r� )r1 r� r r r r� � r� zPyEncoder.encodec C s2 | j sdS | �d�\}}}|r| j�|� ||fS )z�
If ``pushes_fd`` is ``True``, then this method will be used,
and ``encode()`` will only be called once.
:returns: A tuple of ``(bytes consumed, errcode)``.
Err codes are from :data:`.ImageFile.ERRORS`.
)r r r )r� r� r� r� )r1 �bytes_consumedr� r� r r r r� � s zPyEncoder.encode_to_pyfdc C sB d}|dkr | � |�\}}}|dkr|�||d� � |dks|S )z�
:param fh: File handle.
:param bufsize: Buffer size.
:returns: If finished successfully, return 0.
Otherwise, return an error code. Err codes are from
:data:`.ImageFile.ERRORS`.
r N)r� r� )r1 r� r� r� �status�bufr r r r� � s �zPyEncoder.encode_to_fileN)
r r� r� r� r� r� r� r� r� r� r r r r r� � s
r� )r r5 )r� r` r+ rO � r �_utilr r r� rg r r r r r� r� r� r� r� r� r� r� r� r r r r �<module> s> �
t
# H1