3 jQcQ @sddlmZddlmZddlZddlZddlZddlZdZd#Z dZ d$dd%d d&d d'dd(diZ ddZ ddZ GdddejZGdddeZGdddeZd)ddZddZGdd d eZGd!d"d"eZdS)*)Image)isPathNiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configuration zout of memory errorc CsLytjj|}Wntk r.tj|}YnX|sZ decoder_nameextentsoffsetargsr@rr+Zerr_codeprefixdecoderZstatusbsnrrrr7s  "            zImageFile.loadcCsR|j s$|jj|jks$|jj|jkr8tjj|j|j|_|jdkrNtjj|dS)NP)r?r*r+rr newr7)r,rrrrGs  zImageFile.load_preparecCsdS)Nr)r,rrrrPszImageFile.load_endcCsD||jks0t|do|jdk r8||j|jkr8td|j|kS)N _n_framesz attempt to seek outside sequence)rr9r[Zn_framesr&tell)r,framerrr _seek_checks  zImageFile._seek_check)NN) __name__ __module__ __qualname____doc__rr.r/r7rGrPr^rrrrrKs & |  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 _open)NotImplementedError)r,rrrr"&szStubImageFile._opencCsH|j}|dkrtd|j|j|}|dk s4t|j|_|j|_dS)Nz#cannot find loader for this %s file)_loadr formatr7AssertionError __class____dict__)r,loaderimagerrrr7+s  zStubImageFile.loadcCs tddS)z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadN)rd)r,rrrre5szStubImageFile._loadN)r_r`rarbr"r7rerrrrrcs rcc@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)datarg)r,rrrresetHsz Parser.resetc Cs|jr dS|jdkr||_n |j||_|jr|jdkr|tt|j|j}|j|d|_|j||_|jdksx|j r|dS|jj|j\}}|dkrd|_d|_|dkrd|_t|ndS|j|d|_n|jrny&t j |j}t j |}WdQRXWnt k rYnXt|dp*t|d}|sBt|jdkrJd|_nv|j|jd\}}} } g|_t j|j|| |j|_|jj|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. Nrrr=r;)finishedrmrUrRminr8rMrkrioBytesIOrrr r9rrGrIr*rrJr?) r,rmskiprXerr?flagdoarrrfeedPsT      z Parser.feedcCs|S)Nr)r,rrr __enter__szParser.__enter__cGs |jdS)N)r()r,rSrrr__exit__szParser.__exit__cCsz|jr*|jdd|_|_|js*td|js8td|jrttj|j$}ztj ||_Wd|jj XWdQRX|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. r5Nzimage was incompletezcannot parse this image) rUryrmror rkrqrrrrr7)r,rrrrr(s   z Parser.close)r_r`rarbZ incrementalrkrmrUrRrornryrzr{r(rrrrrl<sRrlc Cs|jt|dsf|_|jtdtt||jdd}|tj krP|j dSy|j }|j Wnt t jfk r.x|D]\}}}}tj|j|||j}|dkr|j|d|j|j||jr|j||j\} } n&x$|j|\} } } |j| | rPqW| dkrtd| |jqWYnXx|D]\}}}}tj|j|||j}|dkrl|j|d|j|j||jr|j||j\} } n |j||} | dkrtd| |jq6Wt|dr|j 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 encoderconfig)r4rNz(encoder error %d when writing image fileflush)r7r9r|rHrmaxrr+r:stdoutr~rDr rqUnsupportedOperationrZ _getencoderr*r>rJr?Z pushes_fdrLZencode_to_pyfdencodewriter rOZencode_to_file) r?rrbufsizeZfhrtrVrwrxlrWrvrrr_savesT                 rcCs|dkr dS|tkr6|j|}t||kr2td|Sg}x8|dkrr|jt|t}|sZP|j||t|8}qread 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 can not be completed rr5zTruncated File Readcss|]}t|VqdS)N)r8).0rvrrr sz_safe_read..) SAFEBLOCKr<r8OSErrorrpappendsumjoin)rr+rmblockrrr _safe_reads"     rc@seZdZddZddZdS) PyCodecStatecCsd|_d|_d|_d|_dS)Nr)xsizeysizexoffyoff)r,rrrr&szPyCodecState.__init__cCs |j|j|j|j|j|jfS)N)rrrr)r,rrrrQ,szPyCodecState.extentsN)r_r`rarrQrrrrr%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 `decode` method. See :ref:`Writing Your Own File Decoder in Python` FcGs(d|_t|_d|_||_|j|dS)N)r?rstatefdr*init)r,r*rSrrrr;s zPyDecoder.__init__cCs ||_dS)z Override to perform decoder specific initialization :param args: Array of args items from the tile entry :returns: None N)rS)r,rSrrrrBszPyDecoder.initcCs|jS)N) _pulls_fd)r,rrrrKKszPyDecoder.pulls_fdcCs tdS)az Override to perform the decoding process. :param buffer: A bytes object with the data to be decoded. If `handles_eof` is set, then `buffer` will be empty and `self.fd` will be set. :returns: A tuple of (bytes consumed, errcode). If finished with decoding return <0 for the bytes consumed. Err codes are from `ERRORS` N)rd)r,bufferrrrrMOs zPyDecoder.decodecCsdS)zV Override to perform decoder specific cleanup :returns: None Nr)r,rrrrOZszPyDecoder.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,rrrrrLbszPyDecoder.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 rzSize cannot be negativerz Tile cannot extend outside imageN)rrrr)r?r+rrrrr ValueError)r,r?rQZx0Zy0Zx1Zy1rrrrJks    zPyDecoder.setimagecCsd|s |j}tj|jd|}|j|j|jj|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 r0rznot enough image datarzcannot decode image dataN) r*rrIrJr?rrQrMr)r,rmZrawmodervrWrrr set_as_raws    zPyDecoder.set_as_raw)N)N)r_r`rarbrrrpropertyrKrMrOrLrJrrrrrr1s   !rir6ii)r)rZ_utilrrqrAr:r'rrrNr rrrrcobjectrlrrrrrrrrs0   T >