U Ev fe_@s@ddlmZddlZddlZddlZddlZddlmZmZm Z ddl m Z ddl m Z ddlmZdZd Zd Zd d d dddZddZddZddZGddde ZGddde j ZGdddeZGdddZd1dd d!d"Zd2d#d$d%d&Zd'd(ZGd)d*d*ZGd+d,d,ZGd-d.d.eZ Gd/d0d0eZ!dS)3) annotationsN)IOAny NamedTuple)Image) deprecate)is_pathiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)icCsjztj|}Wntk r.t|}YnX|sJ|rsz"ImageFile.load..cSs|d|d|dfS)NrrrUrr3rrrrYz ImageFile.load..zimage file is truncatedzimage file is truncated (z bytes not processed)r)6r3rrloadmapr:lenhasattrsys load_readrr9read load_seekseek isinstancer$rCZ _MAPMODESmmapr8filenoZ ACCESS_READrDrZ map_bufferimZpaletteZdirty ImportError load_preparesortr"Z tile_prefix itertoolsgroupby _getdecoderr5cleanupsetimagepulls_fdsetfddecoder7r=rArLOAD_TRUNCATED_IMAGESr4load_endr;Z!_close_exclusive_fp_after_loadingrFr)rGrZpixelZuse_mmapr4rfrhZ decoder_namer%r'r(rjr9Zerr_codeprefixrbsenrrrr`s        ""         zImageFile.loadcCsP|jr"|jj|jks"|jj|jkr6tj|j|j|_|jdkrLtj|dS)NP)rlrCrDrrnewr`rNrrrrn:s" zImageFile.load_preparecCsdSrKrrNrrrryBszImageFile.load_endcCsF||jks.t|dr|jdks:||j|jkr:d}t|||kS)N _n_framesz attempt to seek outside sequence)r1rcrZn_framesr@tell)rGframerrrr _seek_checkNs zImageFile._seek_check)NN) r)r*r+__doc__r0rOrPrRr`rnryr __classcell__rrrIrr-is0    r-c@s(eZdZdZddZddZddZdS) 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 _openNotImplementedErrorrGrrrrr<fszStubImageFile._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 rrYzTruncated File Readcss|]}t|VqdSrK)rb)r[rrrr Ysz_safe_read..) SAFEBLOCKrfrbrrappendsumjoin)r9rDrrZremaining_sizeblockrrr _safe_read<s(    rc@seZdZddZddZdS) PyCodecStatecCsd|_d|_d|_d|_dS)Nr)xsizeysizexoffyoffrNrrrr0`szPyCodecState.__init__cCs |j|j|j|j|j|jfSrK)rrrrrNrrrr%fszPyCodecState.extentsN)r)r*r+r0r%rrrrr_src@s@eZdZUded<ddZddZddZd d Zdd d Zd S)PyCodeczIO[bytes] | NonefdcGs(d|_t|_d|_||_||dSrK)rlrrQrrCinit)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(rrrrrtsz PyCodec.initcCsdS)zT Override to perform codec specific cleanup :returns: None NrrNrrrrs}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)rGrrrrrvsz PyCodec.setfdNcCs||_|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)rlrDrQrrrr ValueError)rGrlr%Zx0Zy0x1y1rrrrrts&    zPyCodec.setimage)N) r)r*r+r,r0rrsrvrtrrrrrjs   rc@s:eZdZdZdZeddZddZd 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_fdrNrrrruszPyDecoder.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)rGbufferrrrrrws zPyDecoder.decodeNbytesr)rrcCsl|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 rTrznot enough image datarzcannot decode image dataN) rCrrrrtrlrQr%rwr)rGrrawmoderr|rrrr set_as_raws    zPyDecoder.set_as_raw)N) r)r*r+rrpropertyrurwrrrrrrs   rc@s8eZdZdZdZeddZddZddZd 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_fdrNrrrrszPyEncoder.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)rGrrrrrrs 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)rrrr)rGZbytes_consumedrrrrrrs  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)rGrrrstatusbufrrrrs zPyEncoder.encode_to_fileN) r)r*r+rrrrrrrrrrrrs  r)r)N)" __future__rrrprArdtypingrrrrZ _deprecaterZ_utilr r6rrxrrrr"r#r-rrrrrrrrrrrrrsB       v # J2