o Ԧ/fe_@s<ddlmZddlZddlZddlZddlZddlmZmZm Z ddl m Z ddl m Z ddlmZdZd Zd Z d d d dddZ ddZddZddZGddde ZGddde j ZGdddeZGdddZd1d2d!d"Zd3d4d%d&Zd'd(ZGd)d*d*ZGd+d,d,ZGd-d.d.eZ Gd/d0d0eZ!dS)5) annotationsN)IOAny NamedTuple)Image) deprecate)is_pathiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)icCshztj|}Wntyt|}Ynw|s$|rdndd|}|d|r*dndd7}t|S)Nencoderdecoderz error z when ZwritingZreadingz image file)rcoreZgetcodecstatusAttributeErrorERRORSgetOSError)errorrmsgr-lib/python3.10/site-packages/PIL/ImageFile.py _get_oserrorCs rcCstddddt|dd)N raise_oserror zIt is only useful for translating error codes returned by a codec's decode() method, which ImageFile already does automatically.)actionFr)rr)rrrrrNs  rcCs|dS)Nr)trrr _tilesortXsr c@s.eZdZUded<ded<ded<ded<d S) _TilestrZ codec_nameztuple[int, int, int, int]extentsintoffsetztuple[Any, ...] | str | NoneargsN)__name__ __module__ __qualname____annotations__rrrrr!]s  r!cs^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 decoderconfigMAXBLOCKdecodermaxblockr openfpfilename _exclusive_fp_open IndexError TypeErrorKeyErrorEOFErrorstructr SyntaxErrormodesize BaseExceptionclose)selfr7r8vr __class__rrr.lsJ    "   zImageFile.__init__cCs,|jr|jS|jdurtj|jSdSN)r0formatrZMIMErupperrErrrget_format_mimetypes  zImageFile.get_format_mimetypecsg|_t|dSrI)r1r- __setstate__)rEstaterGrrrNszImageFile.__setstate__cCs|jr|jd|_dS)zCheck file integrityN)r9r7rDrLrrrverifys  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\}}} } t| trl| ddf} |dkrt| dkr| d|jkr| dtjvrzSddl} t|j} | j| d| jd |_Wdn1swY| |jd| d|jkrd }t|tj|j|j|| | |_d}|jrd|j_Wnt ttfyd|_Ynw|d } |js|jjt d z|j!}Wn t y d }Ynwddt"#|jddD|_|jD]\}}} } || t$|j|| |j%}zw|&|j||j'rJ|(|j |)d d} nW|} z||j*}Wn t+t,j-fyt}zt.riWYd}~n8d}t||d}~ww|st.r|n%dt|d}t|||}|)|\}} |dkrn||d}qMW|/q|/wg|_||_0|1|j2r|j3r|j 4d|_ |jst.s| dkrt5| ddtj|S)z"Load image data based on tile listNzcannot load this imagerZpypy_version_inforFraw)accesszbuffer is not large enoughr keycSsg|] \}}t|dqS)r )list).0_Ztilesrrr s z"ImageFile.load..cSs|d|d|dfS)NrrrRr)r1rrrsz ImageFile.load..Tzimage file is truncatedzimage file is truncated (z bytes not processed)r)6r1rrloadmapr8lenhasattrsys load_readrr7read load_seekseek isinstancer"rAZ _MAPMODESmmapr6filenoZ ACCESS_READrBrZ map_bufferimZpaletteZdirty ImportError load_preparesortr Z tile_prefix itertoolsgroupby _getdecoderr3setimagepulls_fdsetfddecoder5r;r?rLOAD_TRUNCATED_IMAGEScleanupr2load_endr9Z!_close_exclusive_fp_after_loadingrDr)rErZpixelZuse_mmapr2rbrdZ decoder_namer#r%r&rfr7Zerr_codeprefixrbsenrrrr\s                       zImageFile.loadcCsT|jr|jj|jks|jj|jkrtj|j|j|_|jdkr(tj|dSdS)NP)rhrArBrrnewr\rLrrrrj:s " zImageFile.load_preparecCsdSrIrrLrrrruBszImageFile.load_endcCsF||jkst|dr|jdus||j|jkrd}t|||kS)N _n_framesz attempt to seek outside sequence)r/r_r}Zn_framesr>tell)rEframerrrr _seek_checkNs   zImageFile._seek_check)NN) r'r(r)__doc__r.rMrNrPr\rjrur __classcell__rrrGrr+is0    r+c@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 _openNotImplementedErrorrErrrrr:fszStubImageFile._opencCsT|}|durd|jd}t|||}|dusJ|j|_|j|_|S)Nzcannot find loader for this z file)_loadrJrr\rH__dict__)rEloaderrimagerrrr\js  zStubImageFile.loadcCr)z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadrrrrrrvszStubImageFile._loadN)r'r(r)rr:r\rrrrrr^s  rc@sZeZdZUdZdZdZded<dZdZdZ dZ ddZ dd Z d d Z d d ZddZdS)Parserzj Incremental image parser. This class implements the standard feed/close consumer interface. NzImage.Image | NonerrcCs|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)datarLrrrresetsz Parser.resetc Cs|jrdS|jdur||_n|j||_|jri|jdkr=tt|j|j}|j|d|_|j||_|jdks;|js=dS|j|j\}}|dkr_d|_d|_|dkr]d|_t|dddS|j|d|_dS|jrndSzt |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. NrrFrrcra)finishedrrr%minr^rrrrioBytesIOrr6rr_r1rjrnrAr3rorh) rErskiprzryr7rhflagdoarrrfeedsX         z Parser.feedcCs|SrIrrLrrr __enter__szParser.__enter__cGs |dSrI)rDrEr&rrr__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. rVNzimage was incompletezcannot parse this image) rrrrrrrrrr6r\)rErr7rrrrDs$   z Parser.close)r'r(r)rZ incrementalrr*rrr%rrrrrrDrrrrr|s  L rreturnNonec 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 encoderconfigrrTrNflush)r\r_rrkr maxr4rBrgr _encode_tilerrUnsupportedOperation)rhr7r1bufsizefhexcrrr_saves     rr1 list[_Tile]c Cs|D]c\}}}} |dkr||t|j|| |j} zE| |j|| jr2| || d} n|rJ | |dd\} } | | | rHnq5n| ||} | dkr[t | dd|W| q| wdS)NrrTr)rdrZ _getencoderrArrorh pushes_fdrqencode_to_pyfdencodewriteencode_to_filerrt) rhr7r1rrrZ encoder_namer#r%r&rerrcoderrrrr!s.    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 rrVzTruncated File Readcss|]}t|VqdSrI)r^)rXrrrr Ysz_safe_read..) SAFEBLOCKrbr^rrappendsumjoin)r7rBrrZremaining_sizeblockrrr _safe_read<s*      rc@seZdZddZddZdS) PyCodecStatecCsd|_d|_d|_d|_dS)Nr)xsizeysizexoffyoffrLrrrr.`s zPyCodecState.__init__cCs |j|j|j|j|j|jfSrI)rrrrrLrrrr#fs zPyCodecState.extentsN)r'r(r)r.r#rrrrr_s rc@s@eZdZUded<ddZddZddZd d Zdd d Zd S)PyCodeczIO[bytes] | NonefdcGs(d|_t|_d|_||_||dSrI)rhrrOrrAinit)rErAr&rrrr.ms zPyCodec.__init__cC ||_dS)z Override to perform codec specific initialization :param args: Array of args items from the tile entry :returns: None N)r&rrrrrt z PyCodec.initcCsdS)zT Override to perform codec specific cleanup :returns: None NrrLrrrrt}szPyCodec.cleanupcCr)z Called from ImageFile to set the Python file-like object :param fd: A Python file-like object :returns: None N)r)rErrrrrqrz 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)rhrBrOrrrr ValueError)rErhr#Zx0Zy0Zx1Zy1rrrrros$    zPyCodec.setimagerI) r'r(r)r*r.rrtrqrorrrrrjs   rc@s4eZdZdZdZeddZddZddd d 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|jSrI) _pulls_fdrLrrrrpzPyDecoder.pulls_fdcC d}t|)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`. zunavailable in base decoderr)rEbufferrrrrrr zPyDecoder.decodeNrbytesrrcCsl|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 rQrznot enough image datarzcannot decode image dataN) rArrnrorhrOr#rrr)rErZrawmoderrxrrrr set_as_raws    zPyDecoder.set_as_rawrI)rrrr) r'r(r)rrpropertyrprrrrrrrrs  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` FcCrrI) _pushes_fdrLrrrrrzPyEncoder.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`. zunavailable in base encoderr)rErrrrrrrzPyEncoder.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`. )rr r)rrrr)rEZbytes_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)rErrrstatusbufrrrrs zPyEncoder.encode_to_fileN) r'r(r)rrrrrrrrrrrrs  r)r)rrrI)r1r)"Z __future__rrrlr?r`typingrrrrZ _deprecaterZ_utilr r4rrsrrrr r!r+rrrrrrrrrrrrrsH        v # J2