U ûØfe_ã@s@ddlmZddlZddlZddlZddlZddlmZmZm Z ddl m Z ddl m Z ddlmZdZd Zd Zd d d dddœZdd„Zdd„Zdd„ZGdd„de ƒZGdd„de j ƒZGdd„deƒZGdd„dƒZd1dd œd!d"„Zd2d#d$œd%d&„Zd'd(„ZGd)d*„d*ƒZGd+d,„d,ƒZGd-d.„d.eƒZ Gd/d0„d0eƒZ!dS)3é)Ú annotationsN)ÚIOÚAnyÚ NamedTupleé)ÚImage)Ú deprecate)Úis_pathiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)éÿÿÿÿéþÿÿÿéýÿÿÿéøÿÿÿi÷ÿÿÿcCsjztj |¡}Wntk r.t |¡}YnX|sJ|rûsÿz"ImageFile.load..cSs|d|d|dfS)NrrrVr©r3rrrÚþrZz ImageFile.load..zimage file is truncatedzimage file is truncated (z bytes not processed)r)6r3rrÚloadÚmapr:ÚlenÚhasattrÚsysÚ load_readrr9ÚreadÚ load_seekÚseekÚ isinstancer$rCZ _MAPMODESÚmmapr8ÚfilenoÚ ACCESS_READrDrZ map_bufferÚimÚpaletteZdirtyÚ ImportErrorÚ load_prepareÚsortr"Z tile_prefixÚ itertoolsÚgroupbyÚ _getdecoderr5ÚcleanupÚsetimageÚpulls_fdÚsetfdÚdecoder7r=rArÚLOAD_TRUNCATED_IMAGESr4Úload_endr;Z!_close_exclusive_fp_after_loadingrFr)rGrZpixelZuse_mmapr4rgriZ decoder_namer%r'r(rkr9Zerr_codeÚprefixrÚbÚsÚeÚnrrrra¯sÖ    ÿ þ ý ü ""ÿ    ÿþÿ  ÿ   zImageFile.loadcCsP|jr"|jj|jks"|jj|jkr6tj |j|j¡|_|jdkrLtj |¡dS)NÚP)rnrCrDrrÚnewrarOrrrrq:s" zImageFile.load_preparecCsdSrKrrOrrrr|BszImageFile.load_endcCsF||jks.t|dƒr|jdks:||j|jkr:d}t|ƒ‚| ¡|kS)NÚ _n_framesz attempt to seek outside sequence)r1rdr„Zn_framesr@Útell)rGÚframerrrrÚ _seek_checkNsÿûûú zImageFile._seek_check)NN) r)r*r+Ú__doc__r0rPrQrSrarqr|r‡Ú __classcell__rrrIrr-is0    r-c@s(eZdZdZdd„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. cCsd}t|ƒ‚dS)Nz+StubImageFile subclass must implement _open©ÚNotImplementedError©rGrrrrr<fszStubImageFile._opencCsT| ¡}|dkr&d|j›d}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 rrZzTruncated File Readcss|]}t|ƒVqdSrK)rc)r\rœrrrÚ Ysz_safe_read..)Ú SAFEBLOCKrgrcrr—ÚappendÚsumÚjoin)r9rDr”rZremaining_sizeÚblockrrrÚ _safe_read<s(    r¼c@seZdZdd„Zdd„ZdS)Ú PyCodecStatecCsd|_d|_d|_d|_dS)Nr)ÚxsizeÚysizeÚxoffÚyoffrOrrrr0`szPyCodecState.__init__cCs |j|j|j|j|j|jfSrK)rÀrÁr¾r¿rOrrrr%fszPyCodecState.extentsN)r)r*r+r0r%rrrrr½_sr½c@s@eZdZUded<dd„Zdd„Zdd„Zd d „Zdd d „Zd S)ÚPyCodeczIO[bytes] | NoneÚfdcGs(d|_tƒ|_d|_||_| |¡dSrK)rnr½rRrÃrCÚinit)rGrCr(rrrr0ms 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(r¡rrrrÄtsz PyCodec.initcCsdS)zT Override to perform codec specific cleanup :returns: None NrrOrrrrv}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Ã)rGrÃrrrry…sz PyCodec.setfdNcCsÞ||_|r|\}}}}n d\}}}}|dkrJ|dkrJ|jj\|j_|j_n(||j_||j_|||j_|||j_|jjdksŠ|jjdkr–d}t|ƒ‚|jj|jj|jjdksÎ|jj|jj|jjdkrÚ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 )rrrrrzSize cannot be negativerz Tile cannot extend outside imageN)rnrDrRr¾r¿rÀrÁÚ ValueError)rGrnr%Zx0Zy0Úx1Úy1rrrrrwŽs&    ÿþzPyCodec.setimage)N) r)r*r+r,r0rÄrvryrwrrrrrÂjs   rÂc@s:eZdZdZdZedd„ƒZdd„Zd dd d œd d „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` FcCs|jSrK)Ú _pulls_fdrOrrrrx¾szPyDecoder.pulls_fdcCsd}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‹)rGÚbufferrrrrrzÂs zPyDecoder.decodeNÚbytesr¤)r”r¥cCsl|s |j}t |jd|¡}| |j|j ¡¡| |¡}|ddkrPd}t|ƒ‚|ddkrhd}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 rUrznot enough image datarzcannot decode image dataN) rCrrurwrnrRr%rzrÅ)rGr”ÚrawmoderœrrrrrÚ set_as_rawÎs    zPyDecoder.set_as_raw)N) r)r*r+rˆrÉÚpropertyrxrzrÍrrrrrÈ´s   rÈc@s8eZdZdZdZedd„ƒ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` FcCs|jSrK)Ú _pushes_fdrOrrrr°ðszPyEncoder.pushes_fdcCsd}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‹)rGr¬rrrrr²ôs zPyEncoder.encodecCs2|js 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`. )rr r)r°r²rÃr³)rGZbytes_consumedrµr”rrrr±s  zPyEncoder.encode_to_pyfdcCs<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`. rN)r²r³)rGr­r¬rµÚstatusÚbufrrrr´s zPyEncoder.encode_to_fileN) r)r*r+rˆrÐrÎr°r²r±r´rrrrrÏæs  rÏ)r)N)"Ú __future__rr˜rsrAreÚtypingrrrÚrZ _deprecaterZ_utilr r6r·r{rrrr"r#r-rŠr“r¯rªr¼r½rÂrÈrÏrrrrÚsB    û   v # J2