o _#aSR@sddlZddlZddlZddlZddlmZddlmZdZdZ dZ dd d d d d Z ddZ ddZ GdddejZGdddeZGdddZd ddZddZGdddZGdddZdS)!N)Image)isPathiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)iicCsHztj|}Wntyt|}Ynw|sd|}t|d)Nzdecoder error z when reading image file)rcoregetcodecstatusAttributeErrorERRORSgetOSError)errormessager//usr/lib/python3/dist-packages/PIL/ImageFile.py raise_oserror<s   rcCs|dS)Nr)trrr _tilesortFrcsReZdZdZdfdd 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 |r)t |d|_ ||_ d|_ n ||_ ||_ d|_ z/z|WntttttjfyP}zt||d}~ww|jr[|jddkr_tdWdStyq|j rp|j w)NrrrrbTznot identified by this driver)super__init__ _min_framecustom_mimetypetilereadonly decoderconfigMAXBLOCKdecodermaxblockropenfpfilename _exclusive_fp_open IndexError TypeErrorKeyErrorEOFErrorstructr SyntaxErrormodesize BaseExceptionclose)selfr#r$v __class__rrrSsH       zImageFile.__init__cCs,|jr|jS|jdurtj|jSdSN)rformatrMIMEr upperr1rrrget_format_mimetypes  zImageFile.get_format_mimetypecCs|jr|jd|_dS)zCheck file integrityN)r%r#r0r9rrrverifys  zImageFile.verifyc Csj|jdur tdtj|}|js|Sd|_|jo t|jdk}|o(ttd }d}z|j }d}Wn t y?|j j }Ynwz|j }d}Wn t yT|j j}Ynw|r|jd\}}}} |dkrt| dkr| d|jkr| dtjvrz=ddl} t|j} | j| d| jd |_Wdn1swYtj|j|j||| |_d}|jrd|j_Wnt ttfyd|_Ynw|d } |js|jjtd z|j} Wn t yd } Ynwd dt !|jddD|_|jD]\}}}} t"|j|| |j#}zv|||$|j||j%r'|&|j |'d \}} nS| } z||j(}Wnt)t*j+fyO}zt,rFWYd}~n4td|d}~ww|sat,rWn#tdt|d||}|'|\}} |dkrrn||d}q*W|-q|-wg|_||_.|/|j0r|j1r|j 2d|_ |jst,s| dkrt3| tj|S)z"Load image data based on tile listNzcannot load this imagerpypy_version_inforFraw)accessrkeycSsg|] \}}t|dqS)r)list).0_tilesrrr s z"ImageFile.load..cSs|d|d|dfS)Nrrr>r)rrrrsz ImageFile.load..Tzimage file is truncatedzimage file is truncated (z bytes not processed))4rr rloadmapr$lenhasattrsys load_readr r#read load_seekseekr- _MAPMODESmmapr"fileno ACCESS_READr map_bufferr.impalettedirty ImportError load_preparesortr tile_prefix itertoolsgroupby _getdecoderrsetimagepulls_fdsetfddecoder!r'r+rLOAD_TRUNCATED_IMAGEScleanuprload_endr%!_close_exclusive_fp_after_loadingr0r)r1pixeluse_mmaprrOrQ decoder_nameextentsoffsetargsrSr#err_codeprefixdecoderstatusbsenrrrrIs                    zImageFile.loadcCsT|jr|jj|jks|jj|jkrtj|j|j|_|jdkr(tj|dSdS)NP)rWr-r.rrnewrIr9rrrr[s " zImageFile.load_preparecCsdSr5rr9rrrrgszImageFile.load_endcCsB||jkst|dr|jdus||j|jkrtd||kS)N _n_framesz attempt to seek outside sequence)rrLryn_framesr*tell)r1framerrr _seek_check(s   zImageFile._seek_check)NN) __name__ __module__ __qualname____doc__rr:r;rIr[rgr} __classcell__rrr3rrPs/  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. cCtd)Nz+StubImageFile subclass must implement _openNotImplementedErrorr9rrrr&?szStubImageFile._opencCsL|}|durtd|jd||}|dusJ|j|_|j|_dS)Nzcannot find loader for this z file)_loadr r6rIr4__dict__)r1loaderimagerrrrIBs   zStubImageFile.loadcCr)z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadrr9rrrrLrzStubImageFile._loadN)r~rrrr&rIrrrrrr7s  rc@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|jdus JddS)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)datar9rrrreset^sz Parser.resetc Cs|jrdS|jdur||_n|j||_|jrh|jdkr=tt|j|j}|j|d|_|j||_|jdks;|js=dS|j|j\}}|dkr^d|_d|_|dkr\d|_t|ndS|j|d|_dS|jrmdSzt |j }t |}Wdn1swYWn t yYdSwt|dpt|d}|st|jdkrd|_n:||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. NrrrPrN)finishedrrqrmminrKrdrrioBytesIOrr"r rLrr[r`r-rrarW) r1rskiprvrur#rWflagdoarrrfeedfsX         z Parser.feedcCs|Sr5rr9rrr __enter__szParser.__enter__cGs |dSr5)r0r1rnrrr__exit__s zParser.__exit__c Cs|jr|dd|_|_|jstd|jstd|jrKt|j}z t ||_W|j n|j wWd|jS1sFwY|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. rBNzimage was incompletezcannot parse this image) rqrrrr rrrrr"rI)r1r#rrrr0s    z Parser.close)r~rrr incrementalrrrqrmrrrrrr0rrrrrQsP rc Cs|t|ds d|_|jtdtt||jdd}z |}| Wnjt t j fy}z[|D]R\}}}} t |j|| |j}|dkrM||||j||jrc|||\} } n ||\} } } || | rtnqd| dkrtd| d||q4WYd }~nMd }~ww|D]E\}}}} t |j|| |j}|dkr||||j||jr|||\} } n|||} | dkrtd| d|qt|d r| d Sd S) 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@rTzencoder error z when writing image fileNflush)rIrLrr\rmaxr r.rTrr rUnsupportedOperationr _getencoderr-rQrarW pushes_fdrcencode_to_pyfdencodewriter rfencode_to_file) rWr#rbufsizefhexcrursrrlrtrrrr_savesX              rcCs|dkrdS|tkr||}t||krtd|Sg}|}|dkr=|t|t}|s.n|||t|8}|dks#tdd|D|krLtdd|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 rrBzTruncated File Readcss|]}t|VqdSr5)rK)rDrrrr 3sz_safe_read..) SAFEBLOCKrOrKr rappendsumjoin)r#r.rremaining_sizeblockrrr _safe_reads&      rc@seZdZddZddZdS) PyCodecStatecCsd|_d|_d|_d|_dS)Nr)xsizeysizexoffyoffr9rrrr9s zPyCodecState.__init__cCs |j|j|j|j|j|jfSr5)rrrrr9rrrrl?s zPyCodecState.extentsN)r~rrrrlrrrrr8s rc@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 :meth:`decode` method. See :ref:`Writing Your Own File Decoder in Python` FcGs(d|_t|_d|_||_||dSr5)rWrstatefdr-init)r1r-rnrrrrMs zPyDecoder.__init__cC ||_dS)z Override to perform decoder specific initialization :param args: Array of args items from the tile entry :returns: None N)rnrrrrrT zPyDecoder.initcCs|jSr5) _pulls_fdr9rrrrb]szPyDecoder.pulls_fdcCst)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 :data:`.ImageFile.ERRORS`. r)r1bufferrrrrdas zPyDecoder.decodecCsdS)zV Override to perform decoder specific cleanup :returns: None Nrr9rrrrflszPyDecoder.cleanupcCr)z Called from ImageFile to set the python file-like object :param fd: A python file-like object :returns: None N)r)r1rrrrrctrzPyDecoder.setfdNcCs||_|r |\}}}}nd\}}}}|dkr%|dkr%|jj\|j_|j_n||j_||j_|||j_|||j_|jjdksE|jjdkrItd|jj|jj|jjdkse|jj|jj|jjdkritddS)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)rWr.rrrrr ValueError)r1rWrlx0y0x1y1rrrra}s     zPyDecoder.setimagecCsd|s|j}t|jd|}||j|j||}|ddkr&td|ddkr0tddS)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-rr`rarWrrlrdr)r1rrawmoderrtrrr set_as_raws    zPyDecoder.set_as_rawr5)r~rrrrrrpropertyrbrdrfrcrarrrrrrCs   #r)r)rr^r+rMr_utilrr rrer rrrrrrrrrrrrrs8    h ;!