B 6d[@sddlZddlZddlZddlZddlmZddlmZdZdZ dZ dd d d d d Z ddZ ddZ GdddejZGdddeZGdddZd&ddZd'ddZddZGdddZGd d!d!ZGd"d#d#eZGd$d%d%eZdS)(N)Image)is_pathiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)iicCsRytj|}Wntk r.t|}YnX|s>d|}|d7}t|dS)Nzdecoder error z when reading image file)rcoreZgetcodecstatusAttributeErrorERRORSgetOSError)errormsgr1/tmp/pip-unpacked-wheel-lxgm_btu/PIL/ImageFile.py raise_oserror@s rcCs|dS)Nr)trrr _tilesortKsrcs^eZdZdZdfdd ZddZfddZd d Zd d Zd dZ ddZ ddZ Z S) ImageFilez*Base class for image file format handlers.Nc std|_d|_d|_d|_d|_t|_t |rPt |d|_ ||_ d|_ n||_ ||_ d|_ yxy |Wn8tttttjfk r}zt||Wdd}~XYnX|jr|jddks|jddkrd}t|Wn(tk r|j r|j YnXdS)NrrrrbTznot identified by this driver)super__init__ _min_framecustom_mimetypetilereadonly decoderconfigMAXBLOCKdecodermaxblockropenfpfilename _exclusive_fp_open IndexError TypeErrorKeyErrorEOFErrorstructr SyntaxErrormodesize BaseExceptionclose)selfr!r"vr) __class__rrrXs>   "  zImageFile.__init__cCs,|jr |jS|jdk r(tj|jSdS)N)rformatrZMIMEr upper)r/rrrget_format_mimetypes zImageFile.get_format_mimetypecsg|_t|dS)N)rr __setstate__)r/state)r1rrr5szImageFile.__setstate__cCs|jr|jd|_dS)zCheck file integrityN)r#r!r.)r/rrrverifys zImageFile.verifyc Cs|jdkrd}t|tj|}|js,|Sd|_|joDt|jdk}|oTttd }d}y|j }d}Wnt k r|j j }YnXy|j }d}Wnt k r|j j}YnX|r|jd\}}} } |dkrt| dkr| d|jkr| dtjkryddl} t|j} | j| d| jd |_WdQRX| |jd| d|jkr`ttj|j|j|| | |_d}|jrd|j_Wn"t ttfk rd|_YnX|d } |jsJ|jjtd y |j}Wnt k rd }YnXd dt !|jddD|_x,|jD] \}}} } || t"|j|| |j#}z|$|j||j%r|&|j |'d d} n|}xy||j(}WnBt)t*j+fk r}zt,rPnd}t||Wdd}~XYnX|st,rPndt|d}t|||}|'|\}} |dkr$P||d}qWWd|-Xq$Wg|_||_.|/|j0rx|j1rx|j 2d|_ |jst,s| dkrt3| tj|S)z"Load image data based on tile listNzcannot load this imagerpypy_version_inforFraw)accessr)keycSsg|]\}}t|dqS)r)list).0_Ztilesrrr sz"ImageFile.load..cSs|d|d|dfS)Nrrr:r)rrrrr=z ImageFile.load..zimage file is truncatedzimage file is truncated (z bytes not processed))4rr rloadmapr"lenhasattrsys load_readr r!read load_seekseekr+Z _MAPMODESmmapr filenoZ ACCESS_READr,rZ map_bufferimZpaletteZdirty ImportError load_preparesortrZ tile_prefix itertoolsgroupby _getdecoderrsetimagepulls_fdsetfddecoderr%r)r LOAD_TRUNCATED_IMAGEScleanuprload_endr#Z!_close_exclusive_fp_after_loadingr.r)r/rZpixelZuse_mmaprrIrKZ decoder_nameextentsoffsetargsrLr!Zerr_codeprefixdecoderbsenrrrrCs    ""       zImageFile.loadcCsP|jr"|jj|jks"|jj|jkr6tj|j|j|_|jdkrLtj|dS)NP)rNr+r,rrnewrC)r/rrrrP$s" zImageFile.load_preparecCsdS)Nr)r/rrrr[,szImageFile.load_endcCsF||jks.t|dr|jdks:||j|jkr:d}t|||kS)N _n_framesz attempt to seek outside sequence)rrFrgZn_framesr(tell)r/framerrrr _seek_check8s  zImageFile._seek_check)NN) __name__ __module__ __qualname____doc__rr4r5r7rCrPr[rj __classcell__rr)r1rrUs0    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. cCsd}t|dS)Nz+StubImageFile subclass must implement _open)NotImplementedError)r/rrrrr$PszStubImageFile._opencCsT|}|dkr&d|jd}t|||}|dk s|rx8| |dd\} } | | | rlPqlWn | ||} | dkrd| d} t | |Wd| XqWdS)Nrrzencoder error z when writing image file)rKrZ _getencoderr+rrUrN pushes_fdrWencode_to_pyfdencodewriteencode_to_filer rZ)rNr!rrrrrcrarrencodererrcoderxrrrrr s(     rcCs|dkr dS|tkr:||}t||kr6d}t||Sg}|}x8|dkrz|t|t}|sbP|||t|8}qDWtdd|D|krd}t|d|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 rr=zTruncated File Readcss|]}t|VqdS)N)rE)r?rrrr Esz_safe_read..) SAFEBLOCKrIrEr r{appendsumjoin)r!r,rxrZremaining_sizeblockrrr _safe_read(s(     rc@seZdZddZddZdS) PyCodecStatecCsd|_d|_d|_d|_dS)Nr)xsizeysizexoffyoff)r/rrrrLszPyCodecState.__init__cCs |j|j|j|j|j|jfS)N)rrrr)r/rrrr\RszPyCodecState.extentsN)rkrlrmrr\rrrrrKsrc@s6eZdZddZddZddZddZd d d Zd S) PyCodeccGs(d|_t|_d|_||_||dS)N)rNrr6fdr+init)r/r+r^rrrrWs zPyCodec.__init__cCs ||_dS)z Override to perform codec specific initialization :param args: Array of args items from the tile entry :returns: None N)r^)r/r^rrrr^sz PyCodec.initcCsdS)zT Override to perform codec specific cleanup :returns: None Nr)r/rrrrZgszPyCodec.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/rrrrrWosz PyCodec.setfdNcCs||_|r|\}}}}n d\}}}}|dkrJ|dkrJ|jj\|j_|j_n(||j_||j_|||j_|||j_|jjdks|jjdkrd}t||jj|jj|jjdks|jj|jj|jjdkrd}t|dS)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)rNr,r6rrrr ValueError)r/rNr\Zx0Zy0x1y1rrrrrUxs"    zPyCodec.setimage)N)rkrlrmrrrZrWrUrrrrrVs   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|jS)N) _pulls_fd)r/rrrrVszPyDecoder.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`. N)rq)r/bufferrrrrXs zPyDecoder.decodeNcCsl|s |j}t|jd|}||j|j||}|ddkrPd}t||ddkrhd}t|dS)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 r9rznot enough image datarzcannot decode image dataN) r+rrTrUrNr6r\rXr)r/rxrawmoderrbrrrr set_as_raws    zPyDecoder.set_as_raw)N) rkrlrmrnrpropertyrVrXrrrrrrs   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|jS)N) _pushes_fd)r/rrrrszPyEncoder.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`. N)rq)r/rrrrrs 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`. )rir)rrrr)r/Zbytes_consumedrrxrrrrs  zPyEncoder.encode_to_pyfdcCs@d}x6|dkr:||\}}}|dkr|||dqW|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)r/rrrstatusbufrrrrs  zPyEncoder.encode_to_fileN) rkrlrmrnrrrrrrrrrrrs   r)r)N)r|rRr)rGrZ_utilrrrrYr rrrrprwrrrrrrrrrrrs6    t  # H1