o g[@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)icCsLztj|}Wntyt|}Ynw|sd|}|d7}t|)Nzdecoder error z when reading image file)rcoregetcodecstatusAttributeErrorERRORSgetOSError)errormsgrK/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/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 |r)t |d|_ ||_ d|_ n ||_ ||_ d|_ z8z|WntttttjfyP}zt||d}~ww|jrb|jddksb|jddkrhd}t|WdStyz|j ry|j w)NrrrrbTznot identified by this driver)super__init__ _min_framecustom_mimetypetilereadonly decoderconfigMAXBLOCKdecodermaxblockropenfpfilename _exclusive_fp_open IndexError TypeErrorKeyErrorEOFErrorstructr SyntaxErrormodesize BaseExceptionclose)selfr#r$vr __class__rrrXsJ    "   zImageFile.__init__cCs,|jr|jS|jdurtj|jSdSN)rformatrMIMEr upperr1rrrget_format_mimetypes  zImageFile.get_format_mimetypecsg|_t|dSr5)rr __setstate__)r1stater3rrr;szImageFile.__setstate__cCs|jr|jd|_dS)zCheck file integrityN)r%r#r0r9rrrverifys  zImageFile.verifyc Cs|jdur d}t|tj|}|js|Sd|_|jo"t|jdk}|o*ttd }d}z|j }d}Wn t yA|j j }Ynwz|j }d}Wn t yV|j j}Ynw|r|jd\}}} } |dkrt| dkr| d|jkr| dtjvrzOddl} t|j} | j| d| jd |_Wdn1swY| |jd| d|jkrttj|j|j|| | |_d}|jrd|j_Wnt ttfyd|_Ynw|d } |js|jjtd z|j}Wn t yd }Ynwd dt !|jddD|_|jD]\}}} } || t"|j|| |j#}zw|$|j||j%r;|&|j |'d d} nW|} z||j(}Wn t)t*j+fye}zt,rZWYd}~n8d}t||d}~ww|syt,rmn%dt|d}t|||}|'|\}} |dkrn||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))4rrrloadmapr$lenhasattrsys load_readr r#read load_seekseekr- _MAPMODESmmapr"fileno ACCESS_READr.r map_bufferimpalettedirty ImportError load_preparesortr tile_prefix itertoolsgroupby _getdecoderrsetimagepulls_fdsetfddecoder!r'r+rLOAD_TRUNCATED_IMAGEScleanuprload_endr%!_close_exclusive_fp_after_loadingr0r)r1rpixeluse_mmaprrQrS decoder_nameextentsoffsetargsrUr#err_codeprefixdecoderbsenrrrrKs                     zImageFile.loadcCsT|jr|jj|jks|jj|jkrtj|j|j|_|jdkr(tj|dSdS)NP)rYr-r.rr newrKr9rrrr]$s " zImageFile.load_preparecCsdSr5rr9rrrri,szImageFile.load_endcCsF||jkst|dr|jdus||j|jkrd}t|||kS)N _n_framesz attempt to seek outside sequence)rrNrzn_framesr*tell)r1framerrrr _seek_check8s   zImageFile._seek_check)NN) __name__ __module__ __qualname____doc__rr:r;r=rKr]rir~ __classcell__rrr3rrUs0    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. cC d}t|)Nz+StubImageFile subclass must implement _openNotImplementedErrorr1rrrrr&PszStubImageFile._opencCsT|}|durd|jd}t|||}|dusJ|j|_|j|_|S)Nzcannot find loader for this z file)_loadr6rrKr4__dict__)r1loaderrimagerrrrKTs  zStubImageFile.loadcCr)z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadrrrrrr`szStubImageFile._loadN)rrrrr&rKrrrrrrHs  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)datar9rrrresetssz 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. NrrrRrP)finishedrrsrominrMrfrrioBytesIOrr"rrNrr]rbr-rrcrY) r1rskiprwrvr#rYflagdoarrrfeed{sX         z Parser.feedcCs|Sr5rr9rrr __enter__szParser.__enter__cGs |dSr5)r0r1rprrr__exit__s zParser.__exit__c Cs|jr|dd|_|_|jsd}t||js d}t||jrOt|j}z t ||_W|j n|j wWd|jS1sJwY|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. rDNzimage was incompletezcannot parse this image) rsrrrrrrrrr"rK)r1rr#rrrr0s$   z Parser.close)rrrr incrementalrrrsrorrrrrr0rrrrrfsM rc Cs|t|ds d|_|jtdtt||jdd}z|}| t |||||Wnt t j fyM}zt ||||d|WYd}~nd}~wwt|drY| dSdS)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 encoderconfigrrBrNflush)rKrNrr^rmaxr r.rVr _encode_tiler rUnsupportedOperation)rYr#rbufsizefhexcrrr_saves     rc Cs|D]g\}}}} |dkr||t|j|| |j} zI| |j|| jr2| || d} n|rJ | |dd\} } | | | rHnq5n| ||} | dkr_d| d} t | |W| q| wdS)NrrTzencoder error z when writing image file)rSr _getencoderr-rrcrY pushes_fdreencode_to_pyfdencodewriteencode_to_filerrh)rYr#rrrrrvrtrrencodererrcoderrrrrr s0      rcCs|dkrdS|tkr||}t||krd}t||Sg}|}|dkr?|t|t}|s0n|||t|8}|dks%tdd|D|krPd}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 rrDzTruncated File Readcss|]}t|VqdSr5)rM)rFrrrr Esz_safe_read..) SAFEBLOCKrQrMrrappendsumjoin)r#r.rrremaining_sizeblockrrr _safe_read(s*      rc@seZdZddZddZdS) PyCodecStatecCsd|_d|_d|_d|_dS)Nr)xsizeysizexoffyoffr9rrrrLs zPyCodecState.__init__cCs |j|j|j|j|j|jfSr5)rrrrr9rrrrnRs zPyCodecState.extentsN)rrrrrnrrrrrKs rc@s6eZdZddZddZddZddZd d d Zd S) PyCodeccGs(d|_t|_d|_||_||dSr5)rYrr<fdr-init)r1r-rprrrrWs zPyCodec.__init__cC ||_dS)z Override to perform codec specific initialization :param args: Array of args items from the tile entry :returns: None N)rprrrrr^ z PyCodec.initcCsdS)zT Override to perform codec specific cleanup :returns: None Nrr9rrrrhgszPyCodec.cleanupcCr)z Called from ImageFile to set the Python file-like object :param fd: A Python file-like object :returns: None N)r)r1rrrrreorz PyCodec.setfdNcCs||_|r |\}}}}nd\}}}}|dkr%|dkr%|jj\|j_|j_n||j_||j_|||j_|||j_|jjdksE|jjdkrKd}t||jj|jj|jjdksg|jj|jj|jjdkrmd}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)rYr.r<rrrr ValueError)r1rYrnx0y0x1y1rrrrrcxs$    zPyCodec.setimager5)rrrrrrhrercrrrrrVs   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` FcC|jSr5) _pulls_fdr9rrrrdzPyDecoder.pulls_fdcCt)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`. r)r1bufferrrrrf zPyDecoder.decodeNcCsl|s|j}t|jd|}||j|j||}|ddkr(d}t||ddkr4d}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 r?rznot enough image datarzcannot decode image dataN) r-rrbrcrYr<rnrfr)r1rrawmoderrurrrr set_as_raws    zPyDecoder.set_as_rawr5) rrrrrpropertyrdrfrrrrrrs  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` FcCrr5) _pushes_fdr9rrrrrzPyEncoder.pushes_fdcCr)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`. r)r1rrrrrrzPyEncoder.encodecCs2|jsdS|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`. )rrr)rrrr)r1bytes_consumedrrrrrrs  zPyEncoder.encode_to_pyfdcCsBd}|dkr||\}}}|dkr|||d|dks|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)r1rrrstatusbufrrrrs zPyEncoder.encode_to_fileN) rrrrrrrrrrrrrrrs  r)rr5)rr`r+rOr_utilrr rrgr rrrrrrrrrrrrrrrrs>     t # H1