e_lddlmZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl mZdZd Zd Z d d d dddZ dZdZdZGddeZGdde jZGddeZGddZd'dZd(d)dZdZGdd ZGd!d"ZGd#d$eZGd%d&eZ y)*) annotationsN)Any NamedTuple)Image) deprecate)is_pathiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)ic tjj|}|s |rdndd|}|d|rdnddz }t |S#t$rtj |}YEwxYw)Nencoderdecoderz error z when writingreadingz image file)rcoregetcodecstatusAttributeErrorERRORSgetOSError)errorrmsgs //usr/lib/python3/dist-packages/PIL/ImageFile.py _get_oserrorrCsv jj''. %95WUGDVIi8 DDC 3<  jj sAA'&A'c8tdddt|d)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)rs rrrNs%  G ue ,,c |dS)N)ts r _tilesortr'Xs Q4Kr"c6eZdZUded<ded<ded<ded<y ) _Tilestr encoder_nameztuple[int, int, int, int]extentsintoffsetztuple[Any, ...] | str | NoneargsN)__name__ __module__ __qualname____annotations__r%r"rr)r)]s && K &&r"r)cReZdZdZd fd ZdZfdZdZdZdZ dZ d Z xZ S) ImageFilez*Base class for image file format handlers.c~t|d|_d|_d|_ d|_d|_t|_t|r t|d|_ ||_ d|_ n||_ ||_ d|_ |j|j,r$|j.ddks|j.ddkr d}t+|y#tt t"t$t&j(f$r}t+||d}~wwxYw#t0$r(|jr|jj3wxYw)Nrrr%rbTznot identified by this driver)super__init__ _min_framecustom_mimetypetilereadonly decoderconfigMAXBLOCKdecodermaxblockr openfpfilename _exclusive_fp_open IndexError TypeErrorKeyErrorEOFErrorstructr SyntaxErrormodesize BaseExceptionclose)selfrBrCvr __class__s rr9zImageFile.__init__ls' # 7 ' 2;2tnDGDM!%D DG$DM!%D   , 99 !  1TYYq\Q5F5!#&&6G  ,"!n!+ , !!    s*C=D (D7 DDD 1D<c|jr |jS|j7tjj |jj SyN)r;formatrMIMErupperrPs rget_format_mimetypezImageFile.get_format_mimetypesF   '' ' ;; "::>>$++"3"3"56 6 #r"c2g|_t| |yrT)r<r8 __setstate__)rPstaterRs rr[zImageFile.__setstate__s  U#r"c^|jr|jjd|_y)zCheck file integrityN)rDrBrOrXs rverifyzImageFile.verifys!    GGMMOr"cD |j d}t|tjj|}|js|Sd|_|j xrt |jdk(}|xrttd }d} |j}d} |j}d}|rU|jd\}}} } t| t r| ddf} |dk(r&t | dk\r| d|j"k(r| dtj$vr ddl} t)|j 5} | j'| j+d| j, |_ddd| |j.d| dzz|jj/kDr d }t|tj0j3|j|j.|| | |_d}|j6rd|j6_|j=d } |js|jj?t@  |jB}tEjF|jdDcgc]\}}tI|dc}}|_|jD]\}}} } || tjJ|j"|| |jL} |jO|j4||jPr0|jS|j|jUd d} n^|} ||jV}|s!t^rn?dt |d}t|||z}|jU|\}} |dkrn||d}[|jag|_||_1|je|jfr&|jhr|jjkd|_ |jst^s| dkr tm| dtjj|S#t$r|jj}YwxYw#t$r|jj}YwxYw#1swY/xYw#ttt:f$r d|_YwxYw#t$rd }Y{wxYwcc}}w#tXtZj\f$r}t^rYd}~ed}t||d}~wwxYw#|jawxYw)z"Load image data based on tile listNzcannot load this imagerpypy_version_inforFraw)accesszbuffer is not large enoughr keyr"c|d|d|dfS)Nrrrbr%)r<s rz ImageFile.load..sT!Wd1gtAw,Gr"r zimage file is truncatedzimage file is truncated (z bytes not processed)r!)7r<rrloadmaprClenhasattrsys load_readrrBread load_seekseek isinstancer*rL _MAPMODESmmaprAfileno ACCESS_READrMr map_bufferimpalettedirty ImportError load_preparesortr' tile_prefix itertoolsgroupbylist _getdecoderr>setimagepulls_fdsetfddecoder@rFrJrLOAD_TRUNCATED_IMAGEScleanupr=load_endrD!_close_exclusive_fp_after_loadingrOr)rPrpixeluse_mmapr=rnrp decoder_namer,r.r/rsrBerr_codeprefix_tilesrbsens rrhzImageFile.loads 99 *C#,    &yyL==8S^q%8EGC1D$E E >>DH >>DH 26))A, /L'64$$a|%INGtyy(Gu.$dmm,V#'99RYY[!DDTDT9#UV ! tAw 66H:%cl*#jj33$))\64DG !H||-. * xx IINNyN ) ))!* 1 1IIG!AuU BDI 8>##6q#9"">$()=)=$>$%#8$)),,/F83H)J%(+2#,$6 !AA*1..*;KAx 1u % !!"A3#8OO%Q( &T       $"H"H GGMMOxx 5(Q,x7 7{{%%e 77<#8$)*CC*1#,A$= >2OO%s P?Q%R,2R B#R R:1S AT 8S AT ?Q"!Q"%RR RRR76R7: S S T +T1T 7TT  T  Tcx|jrF|jj|jk7s#|jj|jk7r9tjj |j|j|_|jdk(r tjj |yy)NP)rwrLrMrrnewrhrXs rr{zImageFile.load_prepare:spww$'',,$))3tww||tyy7PjjnnTYY :DG 99  KK  T " r"cyrTr%rXs rrzImageFile.load_endBs r"c||jks4t|dr |j)||j|jzk\r d}t ||j |k7S)N _n_framesz attempt to seek outside sequence)r:rkrn_framesrItell)rPframers r _seek_checkzImageFile._seek_checkNsZ DOO #T;/DNN4JT]]T__<<5C3- yy{e##r")NN) r0r1r2__doc__r9rYr[r^rhr{rr __classcell__)rRs@rr5r5is14.`7 $I&V#  $r"r5c"eZdZdZdZdZdZy) 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. cd}t|)Nz+StubImageFile subclass must implement _openNotImplementedErrorrPrs rrEzStubImageFile._openfs;!#&&r"c|j}|d|jd}t||j|}|J|j|_|j |_|jS)Nzcannot find loader for this z file)_loadrUrrhrR__dict__)rPloaderrimages rrhzStubImageFile.loadjsj >0 UCC#,  D!    zz|r"cd}t|)z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadrrs rrzStubImageFile._loadvs;!#&&r"N)r0r1r2rrErhrr%r"rrr^s' 'r"rcFeZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZy) Parserzj Incremental image parser. This class implements the standard feed/close consumer interface. Nrc*|jJdy)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)datarXs rresetz Parser.resets yy 8"88 r"cT|jry|j||_n|j|z|_|jr|jdkDrmt t |j|j}|j|d|_|j|z |_|jdkDs |jsy|jj |j\}}|dkr(d|_d|_|dkrd|_t|dy|j|d|_y|jry tj|j5}tj|}dddtdxs t|d}|st |jdk7rd|_||_y|j|jd\}}} } g|_tj |j"|| |j$|_|jj'|j(|| |_|jt |jkr%|j|jd|_d|_||_y#1swY%xYw#t*$rYywxYw)z (Consumer) Feed data to the parser. :param data: A string buffer. :exception OSError: If the parser failed to parse the image file. NrrFr!rorm)finishedrrr.minrjrrrioBytesIOrrArkr<r{rrLr>rrwr) rPrskiprrrBrwflagdoas rfeedz Parser.feeds( ==  99 DI D(DI <<{{Q3tyy>4;;7 IIde, "kkD0 ;;?$))<<&&tyy1DAq1u  ! q5!%DJ&q%88 !" DI ZZ  ZZ *(bBB( r;/K72{3K3rww<1,"&DK  OO%!#JAq!Q BG#(#4#4RWWaBDTDT#UDLLL))"%%3#$DK{{c$))n4$(IIdkkm$< &'  /((  s*6JJ+JJJ J'&J'c|SrTr%rXs r __enter__zParser.__enter__s r"c$|jyrT)rOrPr/s r__exit__zParser.__exit__s  r"c,|jr8|jddx|_|_|js d}t ||j s d}t ||jrjt j|j5} tj||_|j j ddd|j S|j S#|j jwxYw#1swY|j SxYw)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) rrrrrrrrrrArh)rPrrBs rrOz Parser.closes << IIcN'+ +DI ==,cl"zz+C#,  99DII& &"&!&BDJJJOO%  & zztzzJJOO%  & zzs$ C? C %C? C<<C??D)r0r1r2r incrementalrrrr.rrrrrrOr%r"rrr|sB K E DG FH9J Xr"rc |jt|dsd|_|jtt t ||jddz} |j}|jt|||||t|dr|jyy#ttjf$r}t||||d|Yd}~Ld}~wwxYw)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 encoderconfigr%rdrNflush)rhrkrr|r'maxr?rMrtr _encode_tilerrUnsupportedOperation)rwrBr<bufsizefhexcs r_saversGGI 2 'II)I (GRWWQZ!^4G7 YY[  RT7B/r7   B33 47RT7D#667s/B))CCCc0|D]\}}}} |dkDr|j|tj|j|| |j} | j |j || jr%| j|| jd} nA|r- | j|dd\} } |j| | rn,| j||} | dkrt| d| | jy#| jwxYw)NrrTr!)rpr _getencoderrLrrrw pushes_fdrencode_to_pyfdencodewriteencode_to_filerr) rwrBr<rrrr+r,r.r/rerrcoders rrr!s /3+ gvt A: GGFO##BGG\4AQAQR    RUUG ,   b!!00215(/w(?(C "! &44RAG{"7D9sB OO /. OO s B!DDc|dkry|tkr.|j|}t||kr d}t||Sg}|}|dkDrG|jt |t}|sn%|j ||t|z}|dkDrGt d|D|kr d}t|dj|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 Readc32K|]}t|ywrT)rj).0rs r z_safe_read..Ys a3q6 s) SAFEBLOCKrnrjrrappendsumjoin)rBrMrrremaining_sizeblocks r _safe_readr<s qy ywwt} t9t 'C#,  DN 1 NI67  E#e*$ 1   4  4'#cl 88D>r"ceZdZdZdZy) PyCodecStatec<d|_d|_d|_d|_y)Nr)xsizeysizexoffyoffrXs rr9zPyCodecState.__init__`s    r"c|j|j|j|jz|j|jzfSrT)rrrrrXs rr,zPyCodecState.extentsfs3yy$))TYY%;TYY=SSSr"N)r0r1r2r9r,r%r"rrr_s  Tr"rc,eZdZdZdZdZdZddZy)PyCodeccnd|_t|_d|_||_|j |yrT)rwrr\fdrLinit)rPrLr/s rr9zPyCodec.__init__ks+!^   $r"c||_y)z Override to perform codec specific initialization :param args: Array of args items from the tile entry :returns: None N)r/rs rrz PyCodec.initrs  r"cy)zT Override to perform codec specific cleanup :returns: None Nr%rXs rrzPyCodec.cleanup{s r"c||_y)z Called from ImageFile to set the Python file-like object :param fd: A Python file-like object :returns: None N)r)rPrs rrz PyCodec.setfds r"Nc||_|r|\}}}}nd\}}}}|dk(r=|dk(r8|jj\|j_|j_nJ||j_||j_||z |j_||z |j_|jjdks|jjdkr d}t||jj|jj z|jjdkDsG|jj|jj z|jjdkDr d}t|y)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)rwrMr\rrrr ValueError)rPrwr,x0y0x1y1rs rrzPyCodec.setimages, & RR+ RR 7rQw15 .DJJ djj. DJJO DJJO!BwDJJ !BwDJJ  ::  q DJJ$4$4$9+CS/ ! JJ  tzz .a @zz$**//1DGGLLOC4CS/ !Dr"rT)r0r1r2r9rrrrr%r"rrrjs #"r"rc2eZdZdZdZedZdZddZy) 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` Fc|jSrT) _pulls_fdrXs rrzPyDecoder.pulls_fds ~~r"cd}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)rPbufferrs rrzPyDecoder.decode,!#&&r"Nc@|s |j}tj|jd|}|j|j|j j |j|}|ddk\r d}t||ddk7r d}t|y)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 rarznot enough image datarzcannot decode image dataN) rLrrrrwr\r,rr)rPrrawmoderrrs r set_as_rawzPyDecoder.set_as_rawsiiG   dii 8 477DJJ..01 HHTN Q419)CS/ ! Q419,CS/ ! r"rT) r0r1r2rrpropertyrrrr%r"rrrs*I  '"r"rc6eZdZdZdZedZdZdZdZ y) 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` Fc|jSrT) _pushes_fdrXs rrzPyEncoder.pushes_fds r"cd}t|)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)rPrrs rrzPyEncoder.encoderr"c|jsy|jd\}}}|r|jj|||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)rPbytes_consumedrrs rrzPyEncoder.encode_to_pyfds@~~(, A%  GGMM$ w&&r"c|d}|dk(r4|j|\}}}|dkDr|j||d|dk(r4|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)rPrrrstatusbufs rrzPyEncoder.encode_to_file sNl#';;w#7 FGSzVW&lr"N) r0r1r2rrrrrrrr%r"rrrs/J  ' 'r"r)rrT)r<z list[_Tile])! __future__rrr~rJrltypingrrr _deprecater_utilr r?rrrrrr'r)r5rrrrrrrrrr%r"rrs:#  "!   N %  - 'J'r$ r$j'I'