a xdR@sddlZddlZddlZddlZddlmZddlmZdZdZ dZ dd d d d d Z ddZ ddZ ddZGdddejZGdddeZGdddZd"ddZddZGdddZGd d!d!ZdS)#N)Image)isPathiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)iicCsLztj|}Wnty,t|}Yn0|s        zImageFile.__init__cCs,|jr |jS|jdur(tj|jSdSN)rformatrZMIMEr upperr3rrrget_format_mimetypes zImageFile.get_format_mimetypecCs|jr|jd|_dS)zCheck file integrityN)r'r%r2r:rrrverifys zImageFile.verifyc Csz|jdurtdtj|}|js(|Sd|_|jo@t|jdk}|oPttd }d}z|j }d}Wnt y~|j j }Yn0z|j }d}Wnt y|j j}Yn0|r|jd\}}}} |dkrt| dkr| d|jkr| dtjvrz~ddl} t|j(} | j| d| jd |_Wdn1s>0Ytj|j|j||| |_d}|jrvd|j_Wn t ttfyd|_Yn0|d } |js|jjtd z |j} Wnt yd } Yn0|jD]0\}}}} t |j|| |j!}z|||"|j||j#rF|$|j |%d \}} n| }z||j&}WnLt't(j)fy}z,t*rWYd}~qn td |WYd}~n d}~00|st*rqntdt|d||}|%|\}} |dkrq||d}qJW|+n |+0qg|_||_,|-|j.rH|j/rH|j 0d|_ |jsnt*sn| dkrnt1| tj|S)z"Load image data based on tile listNzcannot load this imagerpypy_version_inforFraw)accessrkeyzimage file is truncatedzimage file is truncated (z bytes not processed))2rr rloadmapr&lenhasattrsys load_readr r%read load_seekseekr/Z _MAPMODESmmapr$filenoZ ACCESS_READrZ map_bufferr0imZpaletteZdirty ImportError load_preparesortrZ tile_prefix _getdecoderr!setimagepulls_fdsetfddecoder#r)r-r LOAD_TRUNCATED_IMAGEScleanupr load_endr'Z!_close_exclusive_fp_after_loadingr2r)r3ZpixelZuse_mmapr rJrLZ decoder_nameextentsoffsetargsrMr%Zerr_codeprefixdecoderstatusbsenrrrrDs        8         zImageFile.loadcCsP|jr"|jj|jks"|jj|jkr6tj|j|j|_|jdkrLtj|dS)NP)rOr/r0rrnewrDr:rrrrQs" zImageFile.load_preparecCsdSr7rr:rrrrZszImageFile.load_endcCsB||jks.t|dr|jdus6||j|jkr6td||kS)N _n_framesz attempt to seek outside sequence)rrGrgZn_framesr,tell)r3framerrr _seek_check*s zImageFile._seek_check)NN) __name__ __module__ __qualname____doc__rr;r<rDrQrZrj __classcell__rrr5rrYs/ | rc@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. cCs tddS)Nz+StubImageFile subclass must implement _openNotImplementedErrorr:rrrr(AszStubImageFile._opencCsL|}|dur"td|jd||}|dus8J|j|_|j|_dS)Nzcannot find loader for this z file)_loadr r8rDr6__dict__)r3loaderimagerrrrDDs  zStubImageFile.loadcCs tddS)z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadNrqr:rrrrsNszStubImageFile._loadN)rkrlrmrnr(rDrsrrrrrp9s rpc@sPeZdZdZdZdZdZdZdZdZ ddZ ddZ dd Z d d Z d d ZdS)Parserzj Incremental image parser. This class implements the standard feed/close consumer interface. NrcCs|jdusJddS)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)datar:rrrreset`sz Parser.resetc Cs|jr dS|jdur||_n |j||_|jr|jdkrztt|j|j}|j|d|_|j||_|jdksv|jszdS|j|j\}}|dkrd|_d|_|dkrd|_t|ndS|j|d|_n |jrڐnzread 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 rrCzTruncated File Readcss|]}t|VqdSr7)rF).0rrrr ;rCz_safe_read..) SAFEBLOCKrJrFr r{appendsumjoin)r%r0rxblockrrr _safe_read s"    rc@seZdZddZddZdS) PyCodecStatecCsd|_d|_d|_d|_dS)Nr)xsizeysizexoffyoffr:rrrrAszPyCodecState.__init__cCs |j|j|j|j|j|jfSr7)rrrrr:rrrr[GszPyCodecState.extentsN)rkrlrmrr[rrrrr@src@s\eZdZdZdZddZddZeddZd d Z d d Z d dZ dddZ dddZ dS) 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 Decoder in Python` FcGs(d|_t|_d|_||_||dSr7)rOrstatefdr/init)r3r/r]rrrrUs zPyDecoder.__init__cCs ||_dS)z Override to perform decoder specific initialization :param args: Array of args items from the tile entry :returns: None N)r]rrrrr\szPyDecoder.initcCs|jSr7) _pulls_fdr:rrrrUeszPyDecoder.pulls_fdcCs tdS)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 <0 for the bytes consumed. Err codes are from :data:`.ImageFile.ERRORS`. Nrq)r3rrrrrWis zPyDecoder.decodecCsdS)zV Override to perform decoder specific cleanup :returns: None Nrr:rrrrYtszPyDecoder.cleanupcCs ||_dS)z Called from ImageFile to set the python file-like object :param fd: A python file-like object :returns: None N)r)r3rrrrrV|szPyDecoder.setfdNcCs||_|r|\}}}}n d\}}}}|dkrJ|dkrJ|jj\|j_|j_n(||j_||j_|||j_|||j_|jjdks|jjdkrtd|jj|jj|jjdks|jj|jj|jjdkrtddS)z Called from ImageFile to set the core output image for the decoder :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)rOr0rrrrr ValueError)r3rOr[Zx0Zy0x1y1rrrrTs"    zPyDecoder.setimagecCsd|s |j}t|jd|}||j|j||}|ddkrLtd|ddkr`tddS)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>rznot enough image datarzcannot decode image dataN) r/rrSrTrOrr[rWr)r3rxrawmoderrbrrr set_as_raws    zPyDecoder.set_as_raw)N)N)rkrlrmrnrrrpropertyrUrWrYrVrTrrrrrrKs    #r)r)r|r-rHrrZ_utilrr"rrXr rrrrrprwrrrrrrrrs4     a B