U ?ia@shddlmZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z ddl mZddlmZdZd Zd Zd d d dddZddddddZdddddZddZGddde ZGddde j ZGd d!d!ZGd"d#d#eZGd$d%d%Zd8d&d'd(d)Zd9d*d+d,d-Zd.d/ZGd0d1d1Z Gd2d3d3Z!Gd4d5d5e!Z"Gd6d7d7e!Z#dS):) annotationsN)IOAny NamedTuple)Image) deprecate)is_pathiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)iintboolOSError)errorencoderreturncCsjztj|}Wntk r.t|}YnX|sJ|rvr __class__rrr4ms@   "  zImageFile.__init__z str | NonercCs,|jr |jS|jdk r(tj|jSdSN)r6formatrMIMErupperrKrrrget_format_mimetypes  zImageFile.get_format_mimetypecsg|_t|dSrP)r7r3 __setstate__)rKstaterMrrrVszImageFile.__setstate__NonecCs|jr|jd|_dS)zCheck file integrityN)r?r=rJrTrrrverifys zImageFile.verifyc Cs|jdkrd}t|tj|}|js,|Sd|_|joDt|jdk}|oTttd }d}z|j }d}Wnt k r|j j }YnXz|j }d}Wnt k r|j j}YnX|r|jd\}}} } t| tr| ddf} |dkrt| dkr| d|jkr| dtjkrzddl} t|j} | j| d| jd |_W5QRX| |jd| d|jkr|d }t|tj|j|j|| | |_d}|jrd|j_Wn"t ttfk rd|_YnX|d } |jsh|jjt d z |j!}Wnt k rd }YnXddt"#|jddD|_|jD](\}}} } || t$|j|| |j%}z|'|j||j(r|)|j |*d d} n|}z||j+}WnJt,t-j.fk r}z$t/rWYqVnd}t||W5d}~XYnX|s"t/r qVndt|d}t|||}|*|\}} |dkrFqV||d}qW5|&Xqsz"ImageFile.load..cSs|d|d|dfS)Nrrr[rr7rrrr_z ImageFile.load..zimage file is truncatedzimage file is truncated (z bytes not processed)r#)6r7rrloadmapr>lenhasattrsys load_readrr=read load_seekseek isinstancer(rG _MAPMODESmmapr<fileno ACCESS_READrHr map_bufferimpalettedirty ImportError load_preparesortr&Z tile_prefix itertoolsgroupby _getdecoderr9cleanupsetimagepulls_fdsetfddecoder;rArErLOAD_TRUNCATED_IMAGESr8load_endr?!_close_exclusive_fp_after_loadingrJr)rKrZpixelZuse_mmapr8rlrn decoder_namer*r+r,rqr=Zerr_codeprefixrbsenrrrrfs        ""         zImageFile.loadcCsP|jr"|jj|jks"|jj|jkr6tj|j|j|_|jdkrLtj|dS)NP)rurGrHrrnewrfrTrrrry<s" zImageFile.load_preparecCsdSrPrrTrrrrDszImageFile.load_endcCsF||jks.t|dr|jdks:||j|jkr:d}t|||kS)N _n_framesz attempt to seek outside sequence)r5rirn_framesrDtell)rKframerrrr _seek_checkPs zImageFile._seek_check)NN) r-r.r/__doc__r4rUrVrYrfryrr __classcell__rrrMrr1js0    r1c@s2eZdZdddddZejdddddZd S) StubHandler StubImageFilerX)rurcCsdSrPrrKrurrrr<aszStubHandler.openz Image.ImagecCsdSrPrrrrrrfdszStubHandler.loadN)r-r.r/r<abcabstractmethodrfrrrrr`src@s4eZdZdZddddZddZddd d Zd S) rz 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. rXrOcCsd}t|dS)Nz+StubImageFile subclass must implement _openNotImplementedErrorrKrrrrr@qszStubImageFile._opencCsT|}|dkr&d|jd}t|||}|dk sread method. :param size: Number of bytes to read. :returns: A string containing size bytes of data. Raises an OSError if the file is truncated and the read cannot be completed rr_zTruncated File Readcss|]}t|VqdSrP)rh)rarrrr dsz_safe_read..) SAFEBLOCKrlrhrrappendsumjoin)r=rHrrZremaining_sizeblockrrr _safe_readGs(    rc@s(eZdZddddZddddZdS) PyCodecStaterXrOcCsd|_d|_d|_d|_dS)Nr)xsizeysizexoffyoffrTrrrr4kszPyCodecState.__init__r)cCs |j|j|j|j|j|jfSrP)rrrrrTrrrr*qszPyCodecState.extentsN)r-r.r/r4r*rrrrrjsrc@sNeZdZUded<ddZddZddd d Zd d ZddddddZd S)PyCodeczIO[bytes] | NonefdcGs(d|_t|_d|_||_||dSrP)rurrWrrGinit)rKrGr,rrrr4xs zPyCodec.__init__cCs ||_dS)z Override to perform codec specific initialization :param args: Array of args items from the tile entry :returns: None N)r,rrrrrsz PyCodec.initrXrOcCsdS)zT Override to perform codec specific cleanup :returns: None NrrTrrrr~szPyCodec.cleanupcCs ||_dS)z Called from ImageFile to set the Python file-like object :param fd: A Python file-like object :returns: None N)r)rKrrrrrsz PyCodec.setfdNz tuple[int, int, int, int] | None)r*rcCs||_|r|\}}}}n d\}}}}|dkrJ|dkrJ|jj\|j_|j_n(||j_||j_|||j_|||j_|jjdks|jjdkrd}t||jj|jj|jjdks|jj|jj|jjdkrd}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 )rrrrrzSize cannot be negativerz Tile cannot extend outside imageN)rurHrWrrrr ValueError)rKrur*x0y0x1y1rrrrrs&    zPyCodec.setimage)N) r-r.r/r0r4rr~rrrrrrrus   rc@sHeZdZdZdZeddddZddd d d Zddd dddZd S) 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` FrrOcCs|jSrP) _pulls_fdrTrrrrszPyDecoder.pulls_fdbytestuple[int, int])bufferrcCsd}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`. zunavailable in base decoderNr)rKrrrrrrs zPyDecoder.decodeNrX)rrcCsz|s |j}t|jd|}|jdk s(t||j|j||}|ddkr^d}t ||ddkrvd}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 rZNrznot enough image datarzcannot decode image data) rGrr}rurrrWr*rr)rKrrawmoderrrrrr set_as_raws    zPyDecoder.set_as_raw)N) r-r.r/rrpropertyrrrrrrrrs  rc@sLeZdZdZdZeddddZddd d d Zd dd dZddZ dS) 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` FrrOcCs|jSrP) _pushes_fdrTrrrrszPyEncoder.pushes_fdrztuple[int, int, bytes])rrcCsd}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`. zunavailable in base encoderNr)rKrrrrrrs zPyEncoder.encodercCs@|js dS|d\}}}|r8|jdk s,t|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`. )rr rN)rrrrr)rKbytes_consumedrrrrrr s zPyEncoder.encode_to_pyfdcCs<d}|dkr8||\}}}|dkr|||dq|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`. rN)rr)rKrrrstatusbufrrrrs zPyEncoder.encode_to_fileN) r-r.r/rrrrrrrrrrrrs r)r)N)$ __future__rrrr{rErjtypingrrrr _deprecater_utilr r:rrrrr r&r'r1rrrrrrrrrrrrrrsF       w  # J3