U &^Q@sddlZddlZddlZddlmZddlmZdZdZdZ dd d d d d Z ddZ ddZ GdddejZ Gddde ZGdddZd ddZddZGdddZGdddZdS)!N)Image)isPathiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)iicCsLztj|}Wntk r.t|}YnX|s     zImageFile.__init__cCs,|jr |jS|jdk r(tj|jSdSN)rformatrZMIMEr upperr/rrrget_format_mimetype}s zImageFile.get_format_mimetypecCs|jr|jd|_dS)zCheck file integrityN)r#r!r.r6rrrverifys zImageFile.verifyc Cstj|}|jdkrtd|js(|Sd|_|jo@t|jdk}|oPttd }d}z|j }d}Wnt k r|j j }YnXz|j }d}Wnt k r|j j}YnX|r|jd\}}}} |dkrt| dkr| d|jkr| dtjkrzttjd rLtj|j|_|j||j|j|j| d| d |_nTddl} t|jd } | j| d| jd |_W5QRXtj|j|j|||| |_d}|jrd|j_Wn"t ttfk rd|_YnX|d } |js4|jjtdz |j } Wnt k r d} YnX|jD]\}}}} t!|j|| |j"}z|||$|j||j%r|&|j |'d\}} n| }z||j(}Wn2t)t*j+fk rt,rYq"ntdYnX|st,rސq"ntdt|||}|'|\}} |dkrq"||d}qW5|#Xq(g|_||_-|.|j/rb|j0rb|j 1d|_ |jst,s| dkrt2| tj|S)z"Load image data based on tile listNzcannot load this imagerpypy_version_inforFrawmaprr)accessrkeyzimage file is truncatedz0image file is truncated (%d bytes not processed))3rloadrr r<r"lenhasattrsys load_readr r!read load_seekseekr+Z _MAPMODESrZ readimager,immmapr filenoZ ACCESS_READZ map_bufferZpaletteZdirty ImportError load_preparesortrZ tile_prefix _getdecoderrcleanupsetimagepulls_fdsetfddecoderr%r)r LOAD_TRUNCATED_IMAGESrload_endr#Z!_close_exclusive_fp_after_loadingr.r)r/ZpixelZuse_mmaprrGrIZ decoder_nameextentsoffsetargsrKr!Zerr_codeprefixdecoderstatusbsnrrrrBs             zImageFile.loadcCsP|jr"|jj|jks"|jj|jkr6tj|j|j|_|jdkrLtj|dS)NP)rJr+r,rrnewrBr6rrrrNs" zImageFile.load_preparecCsdSr3rr6rrrrWszImageFile.load_endcCsB||jks.t|dr|jdks6||j|jkr6td||kS)N _n_framesz attempt to seek outside sequence)rrDrcZn_framesr(tell)r/framerrr _seek_check&s zImageFile._seek_check)NN) __name__ __module__ __qualname____doc__rr7r8rBrNrWrf __classcell__rrr1rrMs-  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 _openNotImplementedErrorr6rrrr$=szStubImageFile._opencCsH|}|dkrtd|j||}|dk s4t|j|_|j|_dS)Nz#cannot find loader for this %s file)_loadr r4rBAssertionErrorr2__dict__)r/loaderimagerrrrB@s  zStubImageFile.loadcCs tddS)z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadNrmr6rrrroJszStubImageFile._loadN)rgrhrirjr$rBrorrrrrl5s rlc@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|jdkstddS)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)datarpr6rrrreset\sz Parser.resetc Cs|jr dS|jdkr||_n |j||_|jr|jdkrztt|j|j}|j|d|_|j||_|jdksv|jszdS|j|j\}}|dkrd|_d|_|dkrd|_t|ndS|j|d|_n|jrnz&t |j}t |}W5QRXWnt k rYnXt|dp(t|d}|s@t|jdkrHd|_nv||jd\}}} } g|_t |j|| |j|_|j|j|| |_|jt|jkr|j|jd|_d|_||_dS)z (Consumer) Feed data to the parser. :param data: A string buffer. :exception IOError: If the parser failed to parse the image file. NrrrHrF)finishedrur\rYminrCrUrsrioBytesIOrr r rDrrNrPr+rrRrJ) r/ruskipr`er!rJflagdoarrrfeeddsR      z Parser.feedcCs|Sr3rr6rrr __enter__szParser.__enter__cGs |dSr3)r.r/rZrrr__exit__szParser.__exit__c Csz|jr*|dd|_|_|js*td|js8td|jrtt|j$}zt ||_W5|jXW5QRX|jS)a (Consumer) Close the stream. :returns: An image object. :exception IOError: If the parser failed to parse the image file either because it cannot be identified or cannot be decoded. rANzimage was incompletezcannot parse this image) r\rrurwr rsryrzrBrr )r/r!rrrr.s   z Parser.close)rgrhrirj incrementalrsrur\rYrwrvrrrr.rrrrrtOsPrtc Cs|t|dsd|_|jtdtt||jdd}|tj krP| dSz| }| Wnt t jfk r&|D]\}}}}t|j|||j}|dkr||||j||jr|||\} } n$||\} } } || | rސqq| dkrtd| |qYnX|D]\}}}}t|j|||j}|dkr`||||j||jr|||\} } n |||} | dkrtd| |q,t|dr| dS) zHelper to save image based on tile list :param im: Image object. :param fp: File object. :param tile: Tile list. :param bufsize: Optional buffer size encoderconfigrr?rNz(encoder error %d when writing image fileflush)rBrDrrOrmaxrr,rEstdoutrrLr ryUnsupportedOperationrZ _getencoderr+rIrRrJZ pushes_fdrTZencode_to_pyfdencodewriter rQZencode_to_file) rJr!rbufsizefhr|r^rrlr_r~rrr_savesR                  rcCsb|dkr dS|tkr||Sg}|dkrX|t|t}|s@qX|||t|8}q"d|S)ao 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 up to size bytes of data. rrA) SAFEBLOCKrGrxappendrCjoin)r!r,rublockrrr _safe_reads   rc@seZdZddZddZdS) PyCodecStatecCsd|_d|_d|_d|_dS)Nr)xsizeysizexoffyoffr6rrrr1szPyCodecState.__init__cCs |j|j|j|j|j|jfSr3)rrrrr6rrrrX7szPyCodecState.extentsN)rgrhrirrXrrrrr0src@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 `decode` method. See :ref:`Writing Your Own File Decoder in Python` FcGs(d|_t|_d|_||_||dSr3)rJrstatefdr+init)r/r+rZrrrrEs zPyDecoder.__init__cCs ||_dS)z Override to perform decoder specific initialization :param args: Array of args items from the tile entry :returns: None N)rZrrrrrLszPyDecoder.initcCs|jSr3) _pulls_fdr6rrrrSUszPyDecoder.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 `ERRORS` Nrm)r/bufferrrrrUYs zPyDecoder.decodecCsdS)zV Override to perform decoder specific cleanup :returns: None Nrr6rrrrQdszPyDecoder.cleanupcCs ||_dS)z Called from ImageFile to set the python file-like object :param fd: A python file-like object :returns: None N)r)r/rrrrrTlszPyDecoder.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)rJr,rrrrr ValueError)r/rJrXZx0Zy0x1y1rrrrRus"    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+rrPrRrJrrXrUr)r/rurawmoder~r_rrr set_as_raws    zPyDecoder.set_as_raw)N)N)rgrhrirjrrrpropertyrSrUrQrTrRrrrrrr;s    #r)r)ryr)rErZ_utilrrrrVr rrrrlrtrrrrrrrrs0    i >