B zbX@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)iicCsNytj|}Wntk r.t|}YnX|s>d|}t|ddS)Nzdecoder error z when reading image file)rcoreZgetcodecstatusAttributeErrorERRORSgetOSError)errormessager,lib/python3.7/site-packages/PIL/ImageFile.py raise_oserror@s rcCs|dS)Nr)trrr _tilesortJsrcsReZdZdZdfdd ZddZddZd d Zd d Zd dZ ddZ Z S) ImageFilez*Base class for image file format handlers.Nc std|_d|_d|_d|_d|_t|_t |rPt |d|_ ||_ d|_ n||_ ||_ d|_ yty |Wn8tttttjfk r}zt||Wdd}~XYnX|jr|jddks|jddkrtdWn&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"v) __class__rrrWs<   "  zImageFile.__init__cCs,|jr |jS|jdk r(tj|jSdS)N)rformatrZMIMEr upper)r/rrrget_format_mimetypes zImageFile.get_format_mimetypecCs|jr|jd|_dS)zCheck file integrityN)r#r!r.)r/rrrverifys zImageFile.verifyc Csx|jdkrtdtj|}|js(|Sd|_|jo@t|jdk}|oPttd }d}y|j }d}Wnt k r|j j }YnXy|j }d}Wnt k r|j j}YnX|r|jd\}}}} |dkrt| dkr| d|jkr| dtjkryhddl} t|j} | j| d| jd |_WdQRXtj|j|j||| |_d}|jrdd|j_Wn"t ttfk rd|_YnX|d } |js|jjtd y |j} Wnt k rd } YnXd dt !|jddD|_x$|jD]\}}}} ||t"|j|| |j#}z|$|j||j%rX|&|j |'d d} n| }xy||j(}Wn>t)t*j+fk r}zt,rPn td|Wdd}~XYnX|st,rPntdt|d||}|'|\}} |dkrP||d}q^WWd|-XqWg|_||_.|/|j0rF|j1rF|j 2d|_ |jslt,sl| dkrlt3| tj|S)z"Load image data based on tile listNzcannot load this imagerZpypy_version_inforFraw)accessr)keycSsg|]\}}t|dqS)r)list).0_Ztilesrrr sz"ImageFile.load..cSs|d|d|dfS)Nrrr7r)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_READrZ map_bufferr,imZpaletteZdirty ImportError load_preparesortrZ tile_prefix itertoolsgroupby _getdecoderrsetimagepulls_fdsetfddecoderr%r)r LOAD_TRUNCATED_IMAGEScleanuprload_endr#Z!_close_exclusive_fp_after_loadingr.r)r/ZpixelZuse_mmaprrFrHZ decoder_nameextentsoffsetargsrIr!Zerr_codeprefixdecoderbsenrrrr@s    "       zImageFile.loadcCsP|jr"|jj|jks"|jj|jkr6tj|j|j|_|jdkrLtj|dS)NP)rKr+r,rrnewr@)r/rrrrMs" zImageFile.load_preparecCsdS)Nr)r/rrrrX szImageFile.load_endcCsB||jks.t|dr|jdks6||j|jkr6td||kS)N _n_framesz attempt to seek outside sequence)rrCrdZn_framesr(tell)r/framerrr _seek_check,s  zImageFile._seek_check)NN) __name__ __module__ __qualname____doc__rr4r5r@rMrXrg __classcell__rr)r1rrTs/  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$CszStubImageFile._opencCsP|}|dkr"td|jd||}|dk s8t|j|_|j|_|S)Nzcannot find loader for this z file)_loadr r2r@AssertionErrorr1__dict__)r/loaderimagerrrr@Fs  zStubImageFile.loadcCs tddS)z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadN)rn)r/rrrroQszStubImageFile._loadN)rhrirjrkr$r@rorrrrrm;s rmc@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)datarp)r/rrrresetcsz 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|jrny&t |j}t |}WdQRXWnt 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 OSError: If the parser failed to parse the image file. NrrrGrE)finishedrur]rZminrBrUrsrioBytesIOrr r rCrrMrQr+rrRrK) r/ruskiprar`r!rKflagdoarrrfeedksR      z Parser.feedcCs|S)Nr)r/rrr __enter__szParser.__enter__cGs |dS)N)r.)r/r[rrr__exit__szParser.__exit__c Csz|jr*|dd|_|_|js*td|js8td|jrtt|j$}zt ||_Wd|j XWdQRX|jS)a (Consumer) Close the stream. :returns: An image object. :exception OSError: If the parser failed to parse the image file either because it cannot be identified or cannot be decoded. r:Nzimage was incompletezcannot parse this image) r]rrurwr rsryrzrr r@)r/r!rrrr.s   z Parser.close)rhrirjrkZ incrementalrsrur]rZrwrvrrrr.rrrrrtVsPrtc Csf|t|dsd|_|jtdtt||jdd}y|}| d}Wn,t t j fk r~}z|}Wdd}~XYnXx|D]\}}}} |dkr| |t|j|| |j} z| |j|| jr| || \} } n:|rx2| |\} } } || | rPqWn | ||} | dkrread 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)rB)r<r}rrr 1sz_safe_read..) SAFEBLOCKrFrBr rxappendsumjoin)r!r,ruZremaining_sizeblockrrr _safe_reads$     rc@seZdZddZddZdS) PyCodecStatecCsd|_d|_d|_d|_dS)Nr)xsizeysizexoffyoff)r/rrrr7szPyCodecState.__init__cCs |j|j|j|j|j|jfS)N)rrrr)r/rrrrY=szPyCodecState.extentsN)rhrirjrrYrrrrr6src@s6eZdZddZddZddZddZd d d Zd S) PyCodeccGs(d|_t|_d|_||_||dS)N)rKrstatefdr+init)r/r+r[rrrrBs 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[rrrrIsz PyCodec.initcCsdS)zT Override to perform codec specific cleanup :returns: None Nr)r/rrrrWRszPyCodec.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/rrrrrTZsz 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)rKr,rrrrr ValueError)r/rKrYZx0Zy0Zx1Zy1rrrrRcs    zPyCodec.setimage)N)rhrirjrrrWrTrRrrrrrAs   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/rrrrSszPyDecoder.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)rn)r/bufferrrrrUs 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 r6rznot enough image datarzcannot decode image dataN) r+rrQrRrKrrYrUr)r/ruZrawmoder}r_rrr set_as_raws    zPyDecoder.set_as_raw)N) rhrirjrkrpropertyrSrUrrrrrrs   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)rn)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_consumederrcoderurrrrs  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/rrrZstatusZbufrrrrs  zPyEncoder.encode_to_fileN) rhrirjrkrrrrrrrrrrrs   r)r)ryrOr)rDrZ_utilrrrrVr rrrrmrtrrrrrrrrrrs4    h 4! F/