2?jFn dZddlZddlZddlmZddlZddlZddlZddlZddl Z ddl m Z m Z ddl mZddlmZddlmZd ZdZd Zd Zd Zd ZGddeejZGddeejZejGddeejZdZdZ iddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@Z!GdAdBe"Z#dCZ$GdDdEZ%GdFdGe&Z'dS)Hzx Definition of the Request object, which acts as a kind of bridge between what the user wants and what the plugins can. N)BytesIO)urlopenget_remote_file)Path)urlparse)OptionalceZdZdZdZdZdS)IOModeakAvailable Image modes This is a helper enum for ``Request.Mode`` which is a composite of a ``Request.ImageMode`` and ``Request.IOMode``. The IOMode that tells the plugin if the resource should be read from or written to. Available values are - read ("r"): Read from the specified resource - write ("w"): Write to the specified resource rwN)__name__ __module__ __qualname____doc__readwriteEC:\Projects\Historical\venv\Lib\site-packages\imageio/core/request.pyrr!s$   D EEErrc&eZdZdZdZdZdZdZdZdS) ImageModea2Available Image modes This is a helper enum for ``Request.Mode`` which is a composite of a ``Request.ImageMode`` and ``Request.IOMode``. The image mode that tells the plugin the desired (and expected) image shape. Available values are - single_image ("i"): Return a single image extending in two spatial dimensions - multi_image ("I"): Return a list of images extending in two spatial dimensions - single_volume ("v"): Return an image extending into multiple dimensions. E.g. three spatial dimensions for image stacks, or two spatial and one time dimension for videos - multi_volume ("V"): Return a list of images extending into multiple dimensions. - any_mode ("?"): Return an image in any format (the plugin decides the appropriate action). iIvV?N) rrrr single_image multi_image single_volume multi_volumeany_moderrrrr1s3(LKMLHHHrrceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z ed Zed efdZed efdZdZdS)ModeaThe mode to use when interacting with the resource ``Request.Mode`` is a composite of ``Request.ImageMode`` and ``Request.IOMode``. The image mode that tells the plugin the desired (and expected) image shape and the ``Request.IOMode`` tells the plugin the way the resource should be interacted with. For a detailed description of the available modes, see the documentation for ``Request.ImageMode`` and ``Request.IOMode`` respectively. Available modes are all combinations of ``Request.IOMode`` and ``Request.ImageMode``: - read_single_image ("ri") - read_multi_image ("rI") - read_single_volume ("rv") - read_multi_volume ("rV") - read_any ("r?") - write_single_image ("wi") - write_multi_image ("wI") - write_single_volume ("wv") - write_multi_volume ("wV") - write_any ("w?") Examples -------- >>> Request.Mode("rI") # a list of simple images should be read from the resource >>> Request.Mode("wv") # a single volume should be written to the resource rirIrvrVr?wiwIwvwVw?cj|dkr |dS|dkr |dSt|d)aEnable Mode("r") and Mode("w") The sunder method ``_missing_`` is called whenever the constructor fails to directly look up the corresponding enum value from the given input. In our case, we use it to convert the modes "r" and "w" (from the v3 API) into their legacy versions "r?" and "w?". More info on _missing_: https://docs.python.org/3/library/enum.html#supported-sunder-names rr.rr3z is no valid Mode.) ValueError)clsvalues r _missing_zMode._missing_wsI C<<3t99  c\\3t99 999:: :rreturnc6t|jdS)Nr)rr7selfs rio_modez Mode.io_modesdjm$$$rc6t|jdS)Nr )rr7r;s r image_modezMode.image_modesA'''rcZ|dkr|jS|dkr|jStd|)z7For backwards compatibility with the old non-enum modesrr zMode has no item )r=r? IndexError)r<keys r __getitem__zMode.__getitem__s; !88<  AXX? "66677 7rN)rrrrread_single_imageread_multi_imageread_single_volumeread_multi_volumeread_anywrite_single_imagewrite_multi_imagewrite_single_volumewrite_multi_volume write_any classmethodr8propertyrr=rr?rCrrrr)r)Ms:HI;;[;&%%%%X%(I(((X(88888rr))zz z astronaut.pngz%Image of the astronaut Eileen Collinsz camera.pngz#A grayscale image of a photographerzcheckerboard.pngz'Black and white image of a checkerboardzwood.jpgz'A (repeatable) texture of wooden planksz bricks.jpgz&A (repeatable) texture of stone bricksz clock.pngz7Photo of a clock with motion blur (Stefan van der Walt)z coffee.pngz*Image of a cup of coffee (Rachel Michetti)z chelsea.pngzImage of Stefan's catz wikkie.pngzImage of Almar's catz coins.pngz&Image showing greek coins from Pompeiiz horse.pngz8Image showing the silhouette of a horse (Andreas Preuss)zhubble_deep_field.pngz+Photograph taken by Hubble telescope (NASA)zimmunohistochemistry.pngz"Immunohistochemical (IHC) stainingzmoon.pngz2Image showing a portion of the surface of the moonzpage.pngzA scanned page of textztext.pngzA photograph of handdrawn textzbacterial_colony.tifz+Multi-page TIFF image of a bacterial colonyzNeuronal calcium imaging videoz*The chelsea.png in a zipfile (for testing)z+The chelsea.png in a BSDF file(for testing)z!Animated GIF of a newton's cradlezVideo file of a cockatooz1Video file of a cockatoo with yuv420 pixel formatz2Volumetric image showing a stented abdominal aortaz5A cubemap image of a meadow, e.g. to render a skybox.)zcalcium_imaging.tifz chelsea.zipz chelsea.bsdfznewtonscradle.gifz cockatoo.mp4zcockatoo_yuv420.mp4z stent.npzzmeadow_cube.jpgceZdZdZddddefdZdZedZedefd Z ede efd Z e j d eddfd Z ed Z edZdZdZddZdZedZddZdS)RequestaUImageResource handling utility. Represents a request for reading or saving an image resource. This object wraps information to that request and acts as an interface for the plugins to several resources; it allows the user to read from filenames, files, http, zipfiles, raw bytes, etc., but offer a simple interface to the plugins via ``get_file()`` and ``get_local_filename()``. For each read/write operation a single Request instance is used and passed to the can_read/can_write method of a format, and subsequently to the Reader/Writer class. This allows rudimentary passing of information between different formats and between a format and associated reader/writer. Parameters ---------- uri : {str, bytes, file} The resource to load the image from. mode : str The first character is "r" or "w", indicating a read or write request. The second character is used to indicate the kind of data: "i" for an image, "I" for multiple images, "v" for a volume, "V" for multiple volumes, "?" for don't care. N) extension format_hintrTc ||_d|_d|_d|_d|_||_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ t||_n #t $rt!d|wxYw|||'|ddkrt!d|d||_nr|jk|jt$t&fvr|j}nt)|jj}t-|j}|dkr|nd|_|t3jdt6||ddkrt!d |d||_dS) NFzInvalid Request.Mode: r.zD`extension` should be a file extension starting with a `.`, but is `z`.zdThe usage of `format_hint` is deprecated and will be removed in ImageIO v3. Use `extension` instead.zF`format_hint` should be a file extension starting with a `.`, but is `)raw_uri _uri_type _filename _extension _format_hint_kwargs_result _filename_zip_bytes_zipfile_file_file_is_local_filename_local _firstbytesr)_moder5 _parse_uri URI_FILENAME URI_ZIPPEDrpathrsuffixlowerwarningswarnDeprecationWarningrT)r<urimoderSrTkwargsrjexts r__init__zRequest.__init__s    "   ##  >dDJJ > > >   |s"" . )...(DOO ^ '~, !;;;~//4t**#))++C%(BYYccDDO  " M:"     "{1~'<'<,',,,  's $A99Bcb |jjtju}|jjtju}t |t r|dr|rtd| ddd }| d\}}}|r|dz }|tvrtd|zt|_t!d |zd |_|r|xjd |zz c_nL|d s|drt$|_||_n |ds|drt&|_||_n|drt|_|dd|_n|t(r|rt*|_||_nk|t,r|rt*|_||_n:t|_||_n%t |t.r/|r-t*|_d|_||_nt |t4r|rt*|_d|_||_nt |t6r!t|_t ||_ny|r;t9|dr*t9|drt:|_d|_||_n<|r:t9|dr*t9|drt:|_d|_||_|jtkrC|jdr)t>j !|j|_|jtkrdD]}|j "|}|dkr|dz }|jd|} t>j #| rh|st>j $| r7tJ|_| |j|d&df|_'n|jAtQ|} tS| dkr | ddd z} tUd!| zt$t&g} |r|j| vrtUd"|rn|jttJfvrY|j}|j'r |j'd}t>j +|s|tvrtUd#|d$|d%|jttJfvrm|j'r=t>j ,|j'd|j'df|_'n)t>j ,|j|_|jttJfvr|j}|j'r |j'd}|rj +|st[t\j/d&|dSt>j 0|} t>j +| st[t\j/d'| dSdS)(z$Try to figure our what we were givenzimageio:z$Cannot write to the standard images.:r .zip/z.zipzUnknown standard image %r.zimages/T)auto/zhttp://zhttps://zftp://zftps://zfile://NrPrclosezr~)rxz.zip\rr z/\<9z...z Cannot understand given URI: %s.z-imageio does not support writing to http/ftp.zNo such file: z}. This file looks like one of the standard images, but from imageio 2.1, standard images have to be specified using "imageio:z".zNo such file or directoryzThe directory does not exist)1rqr=rrr isinstancestr startswith RuntimeErrorsplitrl partitionEXAMPLE_IMAGESr5rhrYrrZURI_HTTPURI_FTPSPECIAL_READ_URIS URI_BYTES RETURN_BYTES memoryviewtobytesr`bytesrhasattrURI_FILErbosrj expanduserfindisdirisfilerilstripr_reprlenIOErrorexistsabspathFileNotFoundErrorerrnoENOENTdirname) r<rpis_read_requestis_write_requestfn_zip_partneedlezip_izip_pathuri_r noWritingdns rrgzRequest._parse_uris_)+v{:9, < c3  7 !~~j)) %#O&'MNNNYYsA&&r*0022"$,,w"7"7Ax!&LB^++$%AB%FGGG!-!0Rd!K!K!K5NNcHn4NN ** %cnnZ.H.H %!)!$)) %S^^I-F-F %!(!$ ** %!-!$QRR 122 % %!*!$ -- %2B %!*!$!-!$ Z ( ( !_ !&DN&DN++--DKK U # # ! !&DN&DNDKK T " " !)DN XXDNN  !sF## !W(=(= !!)!)   !sG$$ !g)>)> !!)!)   >\ ) )dn.G.G.L.L )W//??DN >\ ) )-  ,,..33F;;199QJE#~fuf5Hw}}X..)RW^^H-E-E)3$ N566299%@@.* > !IIE5zzBcrc U*-A-Ag(*rr222/ >lJ7 7 7! AGOOD$6q$9::&q)&"" "$!@!@ >lJ7 7 7B! +'* w~~b))+ &A2 W__R((w~~b))+ &Db 8 7rc|jS)a(Name of the ImageResource. The uri for which reading/saving was requested. This can be a filename, an http address, or other resource identifier. Do not rely on the filename to obtain the data, but use ``get_file()`` or ``get_local_filename()`` instead. )rZr;s rfilenamezRequest.filenames ~rr9c|jS)zThe (lowercase) extension of the requested filename. Suffixes in url's are stripped. Can be None if the request is not based on a filename. )r[r;s rrSzRequest.extensions rc|jSN)r\r;s rrTzRequest.format_hints   rformatc4||_|j ||_dSdSr)r\r[)r<rs rrTzRequest.format_hints%" ? "$DOOO # "rc|jS)a.The mode of the request. The first character is "r" or "w", indicating a read or write request. The second character is used to indicate the kind of data: "i" for an image, "I" for multiple images, "v" for a volume, "V" for multiple volumes, "?" for don't care. )rfr;s rrqz Request.modes zrc|jS)z3The dict of keyword arguments supplied by the user.)r]r;s rrrzRequest.kwargss |rc|jjtju}|j|jS|jt kr9|rt|_d|_nwt|j |_n\|jtkr:|rt|j d|_n.t|j d|_n|jtkr{|j\}}|rt|_d|_ntj|d|_|j |d|_t%|j|_n|jt&pt(fvrq|rJt+jd}||sd}t1|j t3||_t%|j|_|jS) aget_file() Get a file object for the resource associated with this request. If this is a reading request, the file is in read mode, otherwise in write mode. This method is not thread safe. Plugins should not close the file when done. This is the preferred way to read/write the data. But if a format cannot handle file-like objects, they should use ``get_local_filename()``. NTwbrbrIMAGEIO_REQUEST_TIMEOUTr )timeout)rqr=rrrbrYrrrcr`rhopenrrir_zipfileZipFileraSeekableFileObjectrrrgetenvisdigitrfloat)r< want_to_writernamers rget_filezRequest.get_files )V\9 : !:  >Y & & 2$YY &*##$T[11 ^| + + 7!$-66 !$-66 ^z ) )!/NHd <$YY &*##!(# > > !]//c:: / ;; ^ 3G4 4 4$ $ $$i 9::Ggoo&7&7 gGGGDJ+DJ77DJzrc|jtkr|jS|j|j}n*tj|jd}tj|d\}|_ t j ||j j tjkrTt|j d5}t!j||dddn #1swxYwY|j S)zget_local_filename() If the filename is an existing file on this filesystem, return that. Otherwise a temporary file is created on the local file system which can be used by the format to read from or write to. Nr imageio_r)rYrhrZrSrrjsplitexttempfilemkstemprdr|rqr=rrrshutil copyfileobjr)r<rsfdfiles rget_local_filenamezRequest.get_local_filename s  >\ ) )> !~)ng&&t~66q9'/'7Z'H'H $B$ HRLLLy FK//$.55>&t}}===>>>>>>>>>>>>>>>' 's1(C%%C),C)c|jjtjkrd}|jr't |j}n'|jr d|_|j }||j tkr||_ n|j tkr|j|ne|j tkrUtj|jdd}||jd|||jr0|j tkr |jd|_|jr |jd|_|jrO t+j|jn-#t.$r t1jd|jdYnwxYwd|_d|_dS)zWrap up this request. Finishes any pending reads or writes, closes any open files and frees any resources allocated by this request. NFrar z(Failed to delete the temporary file at `z`. Please report this issue.)rqr=rrrdr read_bytesrcrbgetvaluerYrr^rrirrr_writestrr|rarremove Exceptionrmrnr`)r<rzfs rfinishzRequest.finish$s 9  , ,E# .T122==??$ .&+# ++-- >Y..#(DLL^x//J$$U++++^z11 );A)>DDBKK 21 5u===HHJJJ : $.H44 J     DJ = ! M   ! ! ! DM   (  $.////    K,KKK  $(D  sF'GGc$d|jc|_}|S)zFor internal use. In some situations a write action can have a result (bytes data). That is obtained with this function. N)r^)r<ress r get_resultzRequest.get_resultWs !$, c rcF|j||jS)zuThe first 256 bytes of the file. These can be used to parse the header to determine the file-format. )re_read_first_bytesr;s r firstbyteszRequest.firstbytes_s(   #  " " $ $ $rc4|j|jd||_dS |}nI#t$r<tj|jrt|_YdSwxYw | }n#t$rd}YnwxYwt|||_ |td| |dS#t$r*d|_ |jtkrtdYdSwxYw)Nzcannot seek with Nonez*Cannot seek back after getting firstbytes!)r`rerrrrjrrrtellr read_n_bytesseekrbrYr)r<Nfrs rrzRequest._read_first_byteshsE ; "#{2A2D    MMOO   7==//',wwD$FF    FFHH     ,Aq11D  P9#$;<<<q  P P P! >X--!"NOOO.--  Ps44AA:8A:>B B"!B";&C##0DD)r9N)r)rrrrrrtrgrOrrSr rTsetterrqrrrrrrrrrrrrRrRs604>'>'>'#>'>'>'>'@HHHT  X 3X!Xc]!!!X!%#%$%%%% XX 666p(((.1111f  X PPPPPPrrRct}t||kr@||t|z }|sn||z }t||k@|S)zread_n_bytes(file, n) Read n bytes from the given file, or less if the file has less bytes. Returns zero bytes if the file is closed. )rrr)rrbb extra_bytess rrrsd B b''A++ffQR[))    k b''A++ IrcHeZdZdZdZd dZddZdZdd Zd Z d Z d Z dS)ra)A readonly wrapper file object that add support for seeking, even if the wrapped file object does not. The allows us to stream from http and still use Pillow. readline matches BytesIO: it returns bytes up to the next newline, EOF, or size, using the same buffer as read/seek. cL||_d|_d|_d|_d|_dS)NrrF)r_i_buffer _have_allclosed)r<rs rrtzSeekableFileObject.__init__s)  rNc|nt|}|dkrd}|jsd}|!|j}d|_n[|j|z}|t |jz }|dkr4|j|}t ||krd|_|xj|z c_||j|jd}n|j|j|j|z}|xjt |z c_|S)NrrT)intrrrrrr)r<nmorewant_i want_morers rrzSeekableFileObject.reads 9 AA1uu~ !Dyv{{}}!%1"S%6%66 q==6;;y11D4yy9,,)- LLD LL 9,twyy)CC,tw145C 3s88 rrwc|d}nt|}|dkrdS|js|jd|j}t |j|jz }|dkr}|dks||krq|dkr!|j}d|_n:|j||z }t |||z krd|_|xj|z c_|jd|j}|dkrt |j|jz }n |dz|jz }|dkrt||}||S)Nrwrr Tr ) rrrrrrrrmin)r<sizenlunreadrrs rreadlinezSeekableFileObject.readlinesM <DDt99Dqyys~ %""5$'22B&&0FRxxTAXX$!886;;==D%)DNN6;;tf}55D4yy4&=00)- $ \  udg . . 88DL!!DG+AAQ A 199At Ayy||rc|jSr)rr;s rrzSeekableFileObject.tells wrrct|}|dkr(|dkrtdt|z|}nx|dkrtd|j|z}nY|dkrA|js|tdt|j|z}ntd|z|t|jkrn1|js*||jksJ|||jz ||_|jS)Nrznegative seek value r rz(invalid whence (%s, should be 0, 1 or 2)) rr5rmaxrrrrr)r<rrqreal_is rrzSeekableFileObject.seeks FF 1991uu !7#a&&!@AAAFF QYYDGaK((FF QYY>  C --122FFG!KLL L S&& & &  (DG#### IIftw& ' ' 'wrcFd|_|jdSNT)rrr|r;s rr|zSeekableFileObject.closes   rcdS)NFrr;s risattyzSeekableFileObject.isattysurcdSrrr;s rseekablezSeekableFileObject.seekable strr)rw)r) rrrrrtrrrrr|rrrrrrrsB<8rrceZdZdZdS)InitializationErroraThe plugin could not initialize from the given request. This is a _internal_ error that is raised by plugins that fail to handle a given request. We use this to differentiate incompatibility between a plugin and a request from an actual error/bug inside a plugin. N)rrrrrrrrrs Drr)(rrriorrrrenumrmcorerrpathlibr urllib.parsertypingr rrrhrirrrEnumrruniquer)rrrobjectrRrrrrrrrrs8   ++++++++!!!!!!          S$)    TY8L8L8L8L8L83 L8L8 L8^8 <7A9  :  J >*(9KJ DD( 0!"I#$<?A<.NEN3:DPDPDPDPDPfDPDPDPN   vvvvvvvvr      )      r