a SicX@sddlZddlZddlZddlZddlmZddlmZdZdZ dZ dd d d d d Z ddZ ddZ GdddejZGdddeZGdddZd$ddZddZGdddZGdddZGd d!d!eZGd"d#d#eZdS)%N)Image)is_pathiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)icCsLztj|}Wnty,t|}Yn0|s     "   zImageFile.__init__cCs,|jr |jS|jdur(tj|jSdSN)rformatrMIMEr upperr1rrrget_format_mimetypes zImageFile.get_format_mimetypecCs|jr|jd|_dS)zCheck file integrityN)r%r#r0r9rrrverifys zImageFile.verifyc Cs|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 } |js6|jjtd z |j} Wnt yd } Yn0d dt !|jddD|_|jD].\}}}} ||t"|j|| |j#}z|$|j||j%rb|&|j |'d d} n| }z||j(}WnLt)t*j+fy}z,t,rWYd}~qn td|WYd}~n d}~00|st,rԐqntdt|d||}|'|\}} |dkr q||d}qfW|-n |-0qg|_||_.|/|j0rd|j1rd|j 2d|_ |jst,s| dkrt3| tj|S)z"Load image data based on tile listNzcannot load this imagerZpypy_version_inforFraw)accessrkeycSsg|]\}}t|dqS)r)list).0_tilesrrr sz"ImageFile.load..cSs|d|d|dfS)Nrrr=r)rrrrrAz ImageFile.load..zimage file is truncatedzimage file is truncated (z bytes not processed))4rrrloadmapr$lenhasattrsys load_readr r#read load_seekseekr- _MAPMODESmmapr"fileno ACCESS_READr map_bufferr.impalettedirty ImportError load_preparesortrZ tile_prefix itertoolsgroupby _getdecoderrsetimagepulls_fdsetfddecoder!r'r+rLOAD_TRUNCATED_IMAGEScleanuprload_endr%!_close_exclusive_fp_after_loadingr0r)r1pixelZuse_mmaprrNrP decoder_nameextentsoffsetargsrRr#Zerr_codeprefixdecoderbsenrrrrHs        8          zImageFile.loadcCsP|jr"|jj|jks"|jj|jkr6tj|j|j|_|jdkrLtj|dS)NP)rVr-r.rr newrHr9rrrrZs" zImageFile.load_preparecCsdSr5rr9rrrre szImageFile.load_endcCsB||jks.t|dr|jdus6||j|jkr6td||kS)N _n_framesz attempt to seek outside sequence)rrKrtn_framesr*tell)r1framerrr _seek_check,s zImageFile._seek_check)NN) __name__ __module__ __qualname____doc__rr:r;rHrZrerx __classcell__rrr3rrTs/  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 _openNotImplementedErrorr9rrrr&CszStubImageFile._opencCsP|}|dur"td|jd||}|dus8J|j|_|j|_|S)Nzcannot find loader for this z file)_loadrr6rHr4__dict__)r1loaderimagerrrrHFs  zStubImageFile.loadcCs tddS)z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadNrr9rrrrQszStubImageFile._loadN)ryrzr{r|r&rHrrrrrr~;s r~c@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)datar9rrrresetcsz 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ڐnz         rcCs|dkr dS|tkr6||}t||kr2td|Sg}|}|dkrt|t|t}|s\qt|||t|8}q>tdd|D|krtdd|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 read 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 rrAzTruncated File Readcss|]}t|VqdSr5)rJ)rCrrrr 1rAz_safe_read..) SAFEBLOCKrNrJrrappendsumjoin)r#r.rZremaining_sizeblockrrr _safe_reads$    rc@seZdZddZddZdS) PyCodecStatecCsd|_d|_d|_d|_dS)Nr)xsizeysizexoffyoffr9rrrr7szPyCodecState.__init__cCs |j|j|j|j|j|jfSr5)rrrrr9rrrri=szPyCodecState.extentsN)ryrzr{rrirrrrr6src@s6eZdZddZddZddZddZd d d Zd S) PyCodeccGs(d|_t|_d|_||_||dSr5)rVrstatefdr-init)r1r-rkrrrrBs zPyCodec.__init__cCs ||_dS)z Override to perform codec specific initialization :param args: Array of args items from the tile entry :returns: None N)rkrrrrrIsz PyCodec.initcCsdS)zT Override to perform codec specific cleanup :returns: None Nrr9rrrrdRszPyCodec.cleanupcCs ||_dS)z Called from ImageFile to set the Python file-like object :param fd: A Python file-like object :returns: None N)r)r1rrrrraZsz PyCodec.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 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)rVr.rrrrr ValueError)r1rVrix0y0x1y1rrrr_cs"    zPyCodec.setimage)N)ryrzr{rrrdrar_rrrrrAs   rc@s2eZdZdZdZeddZddZd 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` FcCs|jSr5) _pulls_fdr9rrrr`szPyDecoder.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 -1 for the bytes consumed. Err codes are from :data:`.ImageFile.ERRORS`. Nr)r1bufferrrrrbs zPyDecoder.decodeNcCsd|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-rr^r_rVrrirbr)r1rrawmoderrorrr set_as_raws    zPyDecoder.set_as_raw)N) ryrzr{r|rpropertyr`rbrrrrrrs   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|jSr5) _pushes_fdr9rrrrszPyEncoder.pushes_fdcCs tdS)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`. Nr)r1rrrrrs 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`. )rrr)rrrr)r1bytes_consumederrcoderrrrrs  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)r1rrrstatusbufrrrrs zPyEncoder.encode_to_fileN) ryrzr{r|rrrrrrrrrrrs  r)r)rr\r+rLr_utilrr rrcr rrrr~rrrrrrrrrrrs6    h 4! F/