B 0\WQ@sddlmZddlmZddlZddlZddlZdZdZdZ dd d d d d Z ddZ ddZ GdddejZ Gddde ZGdddeZd ddZddZGdddeZGdddeZdS)!)Image)isPathNiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)iicCsLytj|}Wntk r.t|}YnX|sstdoutr~rEr rqUnsupportedOperationrZ _getencoderr+rBrKrCZ pushes_fdrMZencode_to_pyfdencodewriter rPZencode_to_file) rCr rbufsizeZfhrtrWrwrxlrXrvrrr_savesT                 rcCsf|dkr dS|tkr||Sg}x8|dkrZ|t|t}|sBP|||t|8}q$Wd|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. rr:) SAFEBLOCKr@rpappendr<join)r r,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-rrrrR.szPyCodecState.extentsN)r`rarbrrRrrrrr'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|_||_||dS)N)rCrstatefdr+init)r-r+rTrrrr=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)rT)r-rTrrrrDszPyDecoder.initcCs|jS)N) _pulls_fd)r-rrrrLMszPyDecoder.pulls_fdcCs tdS)a 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)re)r-bufferrrrrNQs zPyDecoder.decodecCsdS)zV Override to perform decoder specific cleanup :returns: None Nr)r-rrrrP^szPyDecoder.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-rrrrrMfszPyDecoder.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)rCr,rrrrr ValueError)r-rCrRZx0Zy0Zx1Zy1rrrrKos    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 r4rznot enough image datarzcannot decode image dataN) r+rrJrKrCrrRrNr)r-rmZrawmodervrXrrr set_as_raws    zPyDecoder.set_as_raw)N)N)r`rarbrcrrrpropertyrLrNrPrMrKrrrrrr3s   !r)r)rZ_utilrrqr>r(rrrOr rrrrdobjectrlrrrrrrrrs.   _ >