U
|
T ±dø^ ã @ sð d dl Z d dlZd dlZd dlZddlmZ ddlmZ dZdZ dZ
|
dd d
|
d d d Z dd Z dd Z G dd dejZG dd deZG dd dZd&ddZd'ddZd d ZG d d d ZG d d! d!ZG d"d# d#eZG d$d% d%eZdS )(é Né )ÚImage)Úis_pathi i Fzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)éÿÿÿÿéþÿÿÿéýÿÿÿéøÿÿÿi÷ÿÿÿc C sR zt j | ¡}W n tk
|
r. t | ¡}Y nX |s>d| }|d7 }t|d S )Nzdecoder error z when reading image file)r ÚcoreZgetcodecstatusÚAttributeErrorÚERRORSÚgetÚOSError)ÚerrorÚmsg© r úDd:\z\workplace\vscode\pyvenv\venv\Lib\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 fddZd 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 |rPt
|
|d| _ || _ d| _ n|| _ || _ d | _ zxz | ¡ W n8 tttttjfk
|
r¨ } zt||W 5 d }~X Y nX | jrÌ| jd dksÌ| jd dkrØd}t|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# Úvr ©Ú __class__r r r X s@
|
û "
|
zImageFile.__init__c C s, | j r | j S | jd k r(tj | j ¡ ¡S d S ©N)r Úformatr ZMIMEr Úupper©r0 r r r Úget_format_mimetype s
|
z ImageFile.get_format_mimetypec s g | _ t |¡ d S r4 )r r Ú __setstate__)r0 Ústater2 r r r9 s zImageFile.__setstate__c C s | j r| j ¡ d| _dS )zCheck file integrityN)r$ r" r/ r7 r r r Úverify s
|
zImageFile.verifyc C s¬ | j dkrd}t|tj | ¡}| j s,|S d| _| joDt| j dk}|oTttd }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´| j d \}}} }
|
|dkr´t|
|
dkr´|
|
d | jkr´|
|
d tjkr´zddl} t| j } | j| ¡ d| jd | _W 5 Q R X | | jd |
|
d | j ¡ kr`ttj | j| j|| |
|
¡| _d}| jrd| j_W n" t
|
ttfk
|
r² d| _Y nX | ¡ d
|
} | jsL| j j t d z
|
| j }W n t
|
k
|
rú d }Y nX d d t !| j dd ¡D | _ | j D ](\}}} }
|
|| t "| j||
|
| j#¡}zò| %| j|¡ |j&r~| '| j ¡ | (d ¡d } n¼|}z|| j)}W nJ t*t+j,fk
|
rÚ } z$t-r¼W Y ¢q:nd}t||W 5 d}~X Y nX |st-rîq:ndt| d}t||| }| (|¡\}} |dk r*q:||d
}qW 5 | $¡ X q g | _ || _.| /¡ | j0rz| j1rz| j 2¡ d| _ | js 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Ú_Ztilesr 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>è rB z ImageFile.load.<locals>.<lambda>zimage 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, Z _MAPMODESÚmmapr! ÚfilenoZ ACCESS_READr- r Z
|
map_bufferÚimZpaletteZdirtyÚ ImportErrorÚ load_prepareÚsortr Z tile_prefixÚ itertoolsÚgroupbyÚ _getdecoderr ÚcleanupÚsetimageÚpulls_fdÚsetfdÚdecoder r& r* r ÚLOAD_TRUNCATED_IMAGESr Úload_endr$ Z!_close_exclusive_fp_after_loadingr/ r )r0 r ZpixelZuse_mmapr rN rP Z decoder_nameÚextentsÚoffsetÚargsrQ r" Zerr_codeÚprefixÚdecoderÚbÚsÚeÚnr r r rH sÐ
|
ÿ
|
þ ý ü "" ÿ
|
|
ÿþ ÿ ÿ
|
|
zImageFile.loadc C sP | j r"| j j| jks"| j j| jkr6tj | j| j¡| _ | jdkrLtj | ¡ d S )NÚP)rS r, r- r r ÚnewrH r7 r r r rU $ s "
|
zImageFile.load_preparec C s d S r4 r r7 r r r r` , s zImageFile.load_endc C sF || j k s.t| dr | jd ks:|| j| j kr:d}t|| ¡ |kS )NÚ _n_framesz attempt to seek outside sequence)r rK rl Zn_framesr) Útell)r0 Úframer r r r Ú _seek_check8 s ÿûûú zImageFile._seek_check)NN) Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r r8 r9 r; rH rU r` ro Ú __classcell__r r r2 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 s d}t |d S )Nz+StubImageFile subclass must implement _open©ÚNotImplementedError©r0 r r r r r% P s zStubImageFile._openc C sT | ¡ }|d kr&d| j d}t|| | ¡}|d k s<t|j| _|j| _| ¡ S )Nz cannot find loader for this z file)Ú_loadr5 r rH ÚAssertionErrorr3 Ú__dict__)r0 Úloaderr Úimager r r rH T s
|
zStubImageFile.loadc C s d}t |dS )z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadNrv rx r r r ry ` s zStubImageFile._loadN)rp rq rr rs r% rH ry r r r r ru H s ru 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 dkstddS )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)Údatarz r7 r r r Úresets 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|dp(t|d}|s@t|jdkrHd| _nv| ¡ |jd \}}} }
|
g |_t |j||
|
|j¡| _| j |j|¡ | | _| jt| jkr¾| 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 rO rM )Úfinishedr re rb ÚminrJ r^ r} r ÚioÚBytesIOr r! r rK r rU rY r, r r[ rS ) r0 r Úskipri rh r" rS ÚflagÚdÚoÚar r r Úfeed{ sR
|
|
|
z Parser.feedc C s | S r4 r r7 r r r Ú __enter__È s zParser.__enter__c G s | ¡ d S r4 )r/ ©r0 rc r r r Ú__exit__Ë s zParser.__exit__c C s | j r.| d¡ d | _| _ | js.d}t|| js@d}t|| jr|t | j¡$}zt
|
|¡| _W 5 | j ¡ X W 5 Q R X | 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.
|
rB Nzimage was incompletezcannot parse this image) re r r r r r} r r rH r r! )r0 r r" r r r r/ Î s
|
|
z Parser.close)rp rq rr rs Ú incrementalr} r re rb r r r r r r/ r r r r r~ f s Mr~ c
|
C s° | ¡ t| dsd| _|jtd tt|| jd d }z$| ¡ }| ¡ t
|
| |||| W n: t t j fk
|
r } zt
|
| |||d| W 5 d}~X Y nX t|dr¬| ¡ 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 é NÚflush)rH rK r rV r Úmaxr r- rR r Ú _encode_tiler
|
r ÚUnsupportedOperation)rS r" r ÚbufsizeÚfhÚexcr r r Ú_saveð s
|
$
|
r c C sÐ |D ]Æ\}}}} |dkr"| |¡ t | j|| | j¡}
|
z|
|
| j|¡ |
|
jrd|
|
|¡ |
|
|
¡ d } n:|r|
|
|¡dd
\} } | | ¡ | rhqqhn |
|
||¡} | dk r¼d| d} t| |W 5 |
|
¡ X qd S )Nr r zencoder error z when writing image file)rP r Z _getencoderr, r rZ r[ rS Ú pushes_fdr] Úencode_to_pyfdÚencodeÚwriteÚencode_to_filer )rS r" r r r r rh rf r r ÚencoderÚerrcoder r r r r r s&
|
|
|
r c C s¤ |dkr dS |t kr:| |¡}t||k r6d}t||S g }|}|dkrx| t|t ¡}|s`qx| |¡ |t|8 }qBtdd |D |k rd}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 rB zTruncated File Readc s s | ]}t |V qd S r4 )rJ )rD r r r r Ú <genexpr>E s z _safe_read.<locals>.<genexpr>)Ú SAFEBLOCKrN rJ r r ÚappendÚsumÚjoin)r" r- r r Zremaining_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Úyoffr7 r r r r L s zPyCodecState.__init__c C s | j | j| j | j | j| j fS r4 )rª r« r¨ r© r7 r r r ra R s zPyCodecState.extentsN)rp rq rr r ra 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 r4 )rS r§ r: Úfdr, Úinit)r0 r, rc r r r r W s
|
zPyCodec.__init__c C s
|
|| _ dS )z
|
Override to perform codec specific initialization
|
|
:param args: Array of args items from the tile entry
|
:returns: None
|
N)rc r r r r r® ^ s z PyCodec.initc C s dS )zT
|
Override to perform codec specific cleanup
|
|
:returns: None
|
Nr r7 r r r rZ g s zPyCodec.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 )r0 r r r r r] o s z PyCodec.setfdNc C sÞ || _ |r|\}}}}n d\}}}}|dkrJ|dkrJ| j j\| j_| j_n(|| j_|| j_|| | j_|| | j_| jjdks| jjdkrd}t|| jj| jj | j jd ksÎ| jj| jj | j jd krÚd}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)rS r- r: r¨ r© rª r« Ú
|
ValueError)r0 rS ra Zx0Zy0Úx1Úy1r r r r r[ x s& ÿþzPyCodec.setimage)N)rp rq rr r r® rZ r] r[ 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 s | j S r4 )Ú _pulls_fdr7 r r r r\ ¨ 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 -1 for the bytes consumed.
|
Err codes are from :data:`.ImageFile.ERRORS`.
|
Nrv )r0 Úbufferr r r r^ ¬ s zPyDecoder.decodeNc C sl |s
|
| j }t | j d|¡}| | j| j ¡ ¡ | |¡}|d dkrPd}t||d dkrhd}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 rY r[ rS r: ra r^ r¯ )r0 r Úrawmoder rg r r r r Ú
|
set_as_raw· s
|
|
zPyDecoder.set_as_raw)N) rp rq rr rs r³ Úpropertyr\ r^ 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 s | j S r4 )Ú
|
_pushes_fdr7 r r r r Ù s zPyEncoder.pushes_fdc C s
|
t dS )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`.
|
Nrv )r0 r r r r r Ý s zPyEncoder.encodec C s2 | j s
|
dS | 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 )r0 Zbytes_consumedr r r r r r è s zPyEncoder.encode_to_pyfdc C s< d}|dkr8| |¡\}}}|dkr| ||d
¡ q|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 )r0 r r r ÚstatusÚbufr r r r ÷ s zPyEncoder.encode_to_fileN)
|
rp rq rr rs r¹ r· r r r r r r r r r¸ Ï s
|
r¸ )r )N)r rW r* rL Ú r Z_utilr r r¡ r_ r r r r ru r~ r r r¦ r§ r¬ r² r¸ r r r r Ú<module> s8 û
|
t
|
|
# H1
|
