"c[HddlZddlZddlZddlZddlmZddlmZdZdZ dZ dd d d d d Z dZ dZ GddejZGddeZGddZd!dZd"dZdZGddZGddZGddeZGdd eZdS)#N)Image)is_pathiiFzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)ic tj|}n*#t$rt|}YnwxYw|sd|}|dz }t |)Nzdecoder error z when reading image file)rcoregetcodecstatusAttributeErrorERRORSgetOSError)errormsgs //usr/lib/python3/dist-packages/PIL/ImageFile.py raise_oserrorr@sz j''..    jj '&u&& %%C #,,s"$A A c|dS)N)ts r _tilesortrKs Q4KcReZdZdZd fd ZdZfdZdZdZdZ d Z d Z xZ S) ImageFilez*Base class for image file format handlers.Nctd|_d|_d|_ d|_d|_t|_t|r$t|d|_ ||_ d|_ n||_ ||_ d|_ |n@#tt t"t$t&jf$r}t+||d}~wwxYw|jr"|jddks|jddkrd}t+|dS#t0$r"|j r|j wxYw)NrrrrbTznot identified by this driver)super__init__ _min_framecustom_mimetypetilereadonly decoderconfigMAXBLOCKdecodermaxblockropenfpfilename _exclusive_fp_open IndexError TypeErrorKeyErrorEOFErrorstructr SyntaxErrormodesize BaseExceptionclose)selfr)r*vr __class__s rr zImageFile.__init__Xsm # 7 ' 2;; &2tnnDGDM!%D  DG$DM!%D   ,   , , ,"!nn!+ ,9 ' !  1 1TYq\Q5F5F5!#&&&6G5F   !     s*B10D-1(C.C))C..=D--,Ec|jr|jS|j6tj|jSdSN)r"formatrMIMErupperr7s rget_format_mimetypezImageFile.get_format_mimetypesH   (' ' ; ":>>$+"3"3"5"566 6 # "rcXg|_t|dSr;)r#r __setstate__)r7stater9s rrBzImageFile.__setstate__s(  U#####rcT|jr|jd|_dS)zCheck file integrityN)r+r)r6r?s rverifyzImageFile.verifys)    GMMOOOrcl |jd}t|tj|}|js|Sd|_|jot |jdk}|ottd }d} |j }d}n#t$r|j j }YnwxYw |j }d}n#t$r|j j}YnwxYw|r`|jd\}}} } |dkrGt | dkr3| d|jkr!| dtjvr ddl} t%|j5} | | d| j |_dddn #1swxYwY| |jd| dzz|jkrttj|j|j|| | |_d}|jr d|j_n$#ttt6f$r d|_YnwxYw|d } |js|jt<  |j}n#t$rd }YnwxYwd tAj!|jdD|_|jD]f\}}} } || tj"|j|| |j#} |$|j||j%r6|&|j |'d d} n|} ||j(}n<#tRtTj+f$r#}tXrYd}~nud}t||d}~wwxYw|s*tXrnQdt |d}t|||z}|'|\}} |dkrn ||d}|-O#|-wxYwg|_||_.|/|j0r |j1r|j 2d|_ |jstXs| dkrtg| tj|S)z"Load image data based on tile listNzcannot load this imagerpypy_version_inforFraw)accessrkeyrc>g|]\}}t|dS)r)list).0_tiless r z"ImageFile.load..s6AuU Brc0|d|d|dfS)NrrrIr)r#s rz ImageFile.load..sT!Wd1gtAw,GrTzimage file is truncatedzimage file is truncated (z bytes not processed))4r#rrloadmapr*lenhasattrsys load_readr r)read load_seekseekr3 _MAPMODESmmapr(fileno ACCESS_READr4r map_bufferimpalettedirty ImportError load_preparesortr tile_prefix itertoolsgroupby _getdecoderr%setimagepulls_fdsetfddecoder'r-r1rLOAD_TRUNCATED_IMAGEScleanupr$load_endr+!_close_exclusive_fp_after_loadingr6r)r7rpixeluse_mmapr$r[r] decoder_nameextentsoffsetargsr_r)err_codeprefixdecoderbsens rrUzImageFile.loadsw 9 *C#,,    &&y L=8S^^q%8EGC1D$E$E E >DHH   7DHH   7>##6#6q#9"&>$(D)=$>$>$. #=>>>#8!>$)EEEE*CC*1#,,A$= >$%7#8!7$))J,/FF)J)J)J%(+2#,,$6 !AA*1..*;*;KAx 1uu % !!""A3&8OO%%%%GOO%%%%       $"H  GMMOOOx $ 5 $(Q,, ( # # #{%%%s B B%$B%) B33C  C 'H?5F4 HFHFBHH/.H/1I99 JJ1AP  MP N4N;P NNAP  P!c|jr*|jj|jks|jj|jkr/tj|j|j|_|jdkr!tj|dSdS)NP)rcr3r4rr newrUr?s rrgzImageFile.load_prepare$sxw ;$',$)33tw|ty7P7PjnnTY ::DG 9   K  T " " " " "  rcdSr;rr?s rrszImageFile.load_end,s rc||jks*t|dr|j$||j|jzkrd}t |||kS)N _n_framesz attempt to seek outside sequence)r!rXrn_framesr0tell)r7framers r _seek_checkzImageFile._seek_check8sg DO # #T;// $59N4JT]T_<<<5C3-- yy{{e##r)NN) __name__ __module__ __qualname____doc__r r@rBrErUrgrsr __classcell__)r9s@rrrUs44......`777 $$$$$G&G&G&R###    $ $ $ $ $ $ $rrc$eZdZdZdZdZdZdS) 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. c$d}t|)Nz+StubImageFile subclass must implement _openNotImplementedErrorr7rs rr,zStubImageFile._openPs;!#&&&rc|}|d|jd}t|||}|J|j|_|j|_|S)Nzcannot find loader for this z file)_loadr<rrUr9__dict__)r7loaderrimages rrUzStubImageFile.loadTso >CCCCC#,,  D!!    zz||rc$d}t|)z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadrrs rrzStubImageFile._load`s;!#&&&rN)rrrrr,rUrrrrrrHsK'''   '''''rrcHeZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZdS) Parserzj Incremental image parser. This class implements the standard feed/close consumer interface. Nrc(|j 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)datar?s rresetz Parser.resetss! y  "8     rc|jrdS|j||_n|j|z|_|jr|jdkr^t t |j|j}|j|d|_|j|z |_|jdks|jsdS|j|j\}}|dkr-d|_d|_|dkrd|_t|ndS|j|d|_dS|jrdS tj |j5}tj |}dddn #1swxYwYt|dpt|d}|st |jdkrd|_n||jd\}}} } g|_tj|j|| |j|_|j|j|| |_|jt |jkr |j|jd|_d|_||_dS#t*$rYdSwxYw)z (Consumer) Feed data to the parser. :param data: A string buffer. :exception OSError: If the parser failed to parse the image file. Nrrr\rZ)finishedrr}ryminrWrprrioBytesIOrr(rXr#rgrlr3r%rmrcr) r7rskiprrr)rcflagdoas rfeedz Parser.feed{s =  F 9 DII D(DI << {Q3ty>>4;77 Idee, "kD0 ;??$)?F<&&ty11DAq1uu  ! q55!%DJ!!$$$$F !"" DIII Z# D  Z **(bBB((((((((((((((( r;//K72{3K3K(3rw<<1,,"&DKKOO%%%!#JAq!Q BG#(#4RWaBDT#U#UDLL))"%333#$DK{c$)nn44$(Idkmm$< &'  -    s6 I&E; IE  IE I I"!I"c|Sr;rr?s r __enter__zParser.__enter__s rc.|dSr;)r6r7rzs r__exit__zParser.__exit__s rc|jr;|ddx|_|_|jsd}t ||jsd}t ||jrt j|j5} tj ||_|j n#|j wxYw dddn #1swxYwY|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. rNzimage was incompletezcannot parse this image) r}rrrrrrrrr(rU)r7rr)s rr6z Parser.closes < # IIcNNN'+ +DI = #,cll"z +C#,,  9 &DI&& &"&!&BDJJOO%%%%DJOO%%%%%  & & & & & & & & & & & & & & & zs*;C=B0C0C  CC"C)rrrr incrementalrrr}ryrrrrrr6rrrrrfs K E DG FH999N N N `rrc |t|dsd|_|tt t ||jddz} |}| t|||||n7#ttj f$r}t||||d|Yd}~nd}~wwxYwt|dr| 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 encoderconfigrrKrNflush)rUrXrrhrmaxr&r4r`r _encode_tiler rUnsupportedOperation)rcr)r#bufsizefhexcs r_saversGGIII 2 ' 'II)I (GRWQZ!^44G7 YY[[  RT7B//// B3 4777RT7D#666666667r7  s,;B((C>CCcb|D]*\}}}} |dkr||tj|j|| |j} | |j|| jr-| || \} } nL|r4 | |\} } } | | | rn2n| ||} | dkrd| d}t|| | #| wxYwdS)NrTzencoder error z when writing image file)r]r _getencoderr3rrmrc pushes_fdroencode_to_pyfdencodewriteencode_to_filerrr)rcr)r#rrrrr~rrencoderlrrrs rrrs] 1a q55 GGAJJJ#BGQ23CDD    RUA & & & < b!!!--//11 <"").."9"91a "! "" ..r7;;A1uuBqBBBcll+ OO    GOO    1s B7DD,c|dkrdS|tkr;||}t||krd}t||Sg}|}|dkrX|t |t}|sn-|||t|z}|dkXt d|D|krd}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 rrzTruncated File Readc34K|]}t|VdSr;)rW)rOrs r z_safe_read..Hs( a3q66 r) SAFEBLOCKr[rWrrappendsumjoin)r)r4rrremaining_sizeblocks r _safe_readr+s qyys ywwt}} t99t  'C#,,  DN 1  NI6677   E#e**$ 1    4   4''#cll 88D>>rceZdZdZdZdS) PyCodecStatec>d|_d|_d|_d|_dS)Nr)xsizeysizexoffyoffr?s rr zPyCodecState.__init__Os"    rcV|j|j|j|jz|j|jzfSr;)rrrrr?s rrxzPyCodecState.extentsUs'y$)TY%;TY=SSSrN)rrrr rxrrrrrNs7 TTTTTrrc.eZdZdZdZdZdZddZdS)PyCodeccd|_t|_d|_||_||dSr;)rcrrCfdr3init)r7r3rzs rr zPyCodec.__init__Zs5!^^   $rc||_dS)z Override to perform codec specific initialization :param args: Array of args items from the tile entry :returns: None N)rzrs rrz PyCodec.initas rcdS)zT Override to perform codec specific cleanup :returns: None Nrr?s rrrzPyCodec.cleanupjs rc||_dS)z Called from ImageFile to set the Python file-like object :param fd: A Python file-like object :returns: None N)r)r7rs rroz PyCodec.setfdrsrNc:||_|r|\}}}}nd\}}}}|dkr*|dkr$|jj\|j_|j_n6||j_||j_||z |j_||z |j_|jjdks|jjdkrd}t||jj|jjz|jjdks-|jj|jjz|jjdkrd}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)rcr4rCrrrr ValueError)r7rcrxx0y0x1y1rs rrmzPyCodec.setimage{s  ,& RR+ RR 77rQww15 .DJ dj.. DJO DJO!BwDJ !BwDJ  : q DJ$4$9$9+CS// ! J tz .a @ @z$*/1DGLOCC4CS// !DCrr;)rrrr rrrrormrrrrrYsd   #"#"#"#"#"#"rrc:eZdZdZdZedZdZddZdS) 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|jSr;) _pulls_fdr?s rrnzPyDecoder.pulls_fds ~rct)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)r7buffers rrpzPyDecoder.decode"###rNcP|s|j}tj|jd|}||j|j||}|ddkrd}t||ddkrd}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 rHrznot enough image datarzcannot decode image dataN) r3rrlrmrcrCrxrpr)r7rrawmoderrrs r set_as_rawzPyDecoder.set_as_raws iG  di 8 8 47DJ..00111 HHTNN Q4199)CS// ! Q4199,CS// ! 9rr;) rrrrrpropertyrnrprrrrrrsaI X $ $ $""""""rrc>eZdZdZdZedZdZdZdZ dS) 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|jSr;) _pushes_fdr?s rrzPyEncoder.pushes_fds rct)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)r7rs rrzPyEncoder.encoderrc|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)r7bytes_consumederrcoders rrzPyEncoder.encode_to_pyfdsP~ 5(, A%  GMM$   w&&rcd}|dkrB||\}}}|dkr|||d|dkB|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)r7rrrstatusbufs rrzPyEncoder.encode_to_files_ll#';;w#7#7 FGSzzVWW&&&llrN) rrrrrrrrrrrrrrrskJ X $ $ $ ' ' 'rr)rr;)rrjr1rYr_utilrr&rrqrrrrrrrrrrrrrrrrrs-<     N %   p$p$p$p$p$ p$p$p$f'''''I'''