U ?hx@sdZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z ddl mZmZmZmZdd lmZdd lmZe Zd d ZGd ddeZGdddeZdS)a .. note:: imageio is under construction, some details with regard to the Reader and Writer classes may change. These are the main classes of imageio. They expose an interface for advanced users and plugin developers. A brief overview: * imageio.FormatManager - for keeping track of registered formats. * imageio.Format - representation of a file format reader/writer * imageio.Format.Reader - object used during the reading of a file. * imageio.Format.Writer - object used during saving a file. * imageio.Request - used to store the filename and other info. Plugins need to implement a Format class and register a format object using ``imageio.formats.add_format()``. N)Path)Arrayasarray) ImageMode) known_pluginsknown_extensions PluginConfig FileExtension)_original_order)imopencCsd}t|jtkr&t|j}nN|tkr0nD|tkrF|}n.d|tkrdd|}ntd|d|dk rddt|DD]}t|jr|}qqt|S)zTOld Plugin resolution logic. Remove once we remove the old format manager. N.zNo format known by name `z`.cSsg|]}|jD]}|qqS)priority).0Zfile_extensionxrrE/var/www/html/venv/lib/python3.8/site-packages/imageio/core/format.py Hsz_get_config..)rsuffixlowerr r IndexError is_legacy)pluginZextension_nameZ plugin_namerrr _get_config4s$    rc@seZdZdZd%ddZddZddZed d Zed d Z ed dZ eddZ eddZ ddZ ddZddZddZddZddZGdd d eZGd!d"d"eZGd#d$d$eZdS)&Formatu+Represents an implementation to read/write a particular file format A format instance is responsible for 1) providing information about a format; 2) determining whether a certain file can be read/written with this format; 3) providing a reader/writer class. Generally, imageio will select the right format and use that to read/write an image. A format can also be explicitly chosen in all read/write functions. Use ``print(format)``, or ``help(format_name)`` to see its documentation. To implement a specific format, one should create a subclass of Format and the Format.Reader and Format.Writer classes. See :class:`imageio.plugins` for details. Parameters ---------- name : str A short name of this format. Users can select a format using its name. description : str A one-line description of the format. extensions : str | list | None List of filename extensions that this format supports. If a string is passed it should be space or comma separated. The extensions are used in the documentation and to allow users to select a format by file extension. It is not used to determine what format to use for reading/saving a file. modes : str A string containing the modes that this format can handle ('iIvV'), “i” for an image, “I” for multiple images, “v” for a volume, “V” for multiple volumes. This attribute is used in the documentation and to select the formats when reading/saving a file. NcCs||_||_|dkrg}nt|tr:|ddd}t|ttfr^tdd|D|_ nt d|pld|_ t|j tst d|j D]}|d krt d qdS) uInitialize the Plugin. Parameters ---------- name : str A short name of this format. Users can select a format using its name. description : str A one-line description of the format. extensions : str | list | None List of filename extensions that this format supports. If a string is passed it should be space or comma separated. The extensions are used in the documentation and to allow users to select a format by file extension. It is not used to determine what format to use for reading/saving a file. modes : str A string containing the modes that this format can handle ('iIvV'), “i” for an image, “I” for multiple images, “v” for a volume, “V” for multiple volumes. This attribute is used in the documentation and to select the formats when reading/saving a file. N, cSs"g|]}|rd|dqS)r)stripr)rerrrrsz#Format.__init__..z#Invalid value for extensions given.zInvalid value for modes given.ziIvV?zInvalid value for mode given.) upper_name _description isinstancestrreplacesplittuplelist _extensions ValueError_modes)selfname description extensionsmodesmrrr__init__xs"      zFormat.__init__cCsd|j|jfS)Nz)r.r/r-rrr__repr__szFormat.__repr__cCs|jSN)docr4rrr__str__szFormat.__str__cCsd|j|j|jfS)zCThe documentation for this format (name + description + docstring).z%s - %s %s )r.r/__doc__rr4rrrr7s z Format.doccCs|jS)zThe name of this format.)r"r4rrrr.sz Format.namecCs|jS)z#A short description of this format.)r#r4rrrr/szFormat.descriptioncCs|jS)zpA list of file extensions supported by this plugin. These are all lowercase with a leading dot. )r*r4rrrr0szFormat.extensionscCs|jS)z:A string specifying the modes that this format can handle.)r,r4rrrr1sz Format.modescCsN|jddkr|jdnd}||jkrBtd|jd|jjd|||S)zget_reader(request) Return a reader object that can be used to read data and info from the given file. Users are encouraged to use imageio.get_reader() instead. riIvVr Format z cannot read in  mode)moder1 RuntimeErrorr. image_modeReaderr-requestZ select_moderrr get_readers  zFormat.get_readercCsN|jddkr|jdnd}||jkrBtd|jd|jjd|||S)zget_writer(request) Return a writer object that can be used to write data and info to the given file. Users are encouraged to use imageio.get_writer() instead. rr:r r;z cannot write in r<)r=r1r>r.r?WriterrArrr get_writers  zFormat.get_writercCs ||S)zacan_read(request) Get whether this format can read data from the specified uri. ) _can_readr-rBrrrcan_readszFormat.can_readcCs ||S)zacan_write(request) Get whether this format can write data to the speciefed uri. ) _can_writerGrrr can_writeszFormat.can_writecCsdS)aNCheck if Plugin can read from ImageResource. This method is called when the format manager is searching for a format to read a certain image. Return True if this format can do it. The format manager is aware of the extensions and the modes that each format can handle. It will first ask all formats that *seem* to be able to read it whether they can. If none can, it will ask the remaining formats if they can: the extension might be missing, and this allows formats to provide functionality for certain extensions, while giving preference to other plugins. If a format says it can, it should live up to it. The format would ideally check the request.firstbytes and look for a header of some kind. Parameters ---------- request : Request A request that can be used to access the ImageResource and obtain metadata about it. Returns ------- can_read : bool True if the plugin can read from the ImageResource, False otherwise. NrrGrrrrFszFormat._can_readcCsdS)alCheck if Plugin can write to ImageResource. Parameters ---------- request : Request A request that can be used to access the ImageResource and obtain metadata about it. Returns ------- can_read : bool True if the plugin can write to the ImageResource, False otherwise. NrrGrrrrIszFormat._can_writec@sveZdZdZddZeddZeddZdd Zd d Z d d Z ddZ eddZ dddZ ddZddZdS)zFormat._BaseReaderWriterzBase class for the Reader and Writer class to implement common functionality. It implements a similar approach for opening/closing and context management as Python's file objects. cCs0d|_d|_||_||_|jf|jjdS)NF)_BaseReaderWriter__closed_BaseReaderWriter_last_index_format_request_openrBkwargscopy)r-formatrBrrrr32s z!Format._BaseReaderWriter.__init__cCs|jS)zgThe :class:`.Format` object corresponding to the current read/write operation. )rNr4rrrrS:szFormat._BaseReaderWriter.formatcCs|jS)zhThe :class:`.Request` object corresponding to the current read/write operation. )rOr4rrrrBAsz Format._BaseReaderWriter.requestcCs ||Sr6) _checkClosedr4rrr __enter__Hsz"Format._BaseReaderWriter.__enter__cCs|dkr|dSr6)close)r-typevalue tracebackrrr__exit__Lsz!Format._BaseReaderWriter.__exit__cCs&z |Wntk r YnXdSr6)rV Exceptionr4rrr__del__Qs z Format._BaseReaderWriter.__del__cCs&|jr dSd|_||jdS)znFlush and close the reader/writer. This method has no effect if it is already closed. NT)rL_closerBfinishr4rrrrVWs zFormat._BaseReaderWriter.closecCs|jS)z$Whether the reader/writer is closed.)rLr4rrrclosedbszFormat._BaseReaderWriter.closedNcCs&|jr"|jj}|pd|}t|dS)z8Internal: raise an ValueError if reader/writer is closedzI/O operation on closed %s.N)r_ __class____name__r>)r-msgwhatrrrrTgs z%Format._BaseReaderWriter._checkClosedcKs tdS)aA_open(**kwargs) Plugins should probably implement this. It is called when reader/writer is created. Here the plugin can do its initialization. The given keyword arguments are those that were given by the user at imageio.read() or imageio.write(). NNotImplementedErrorr-rQrrrrPps zFormat._BaseReaderWriter._opencCs tdS)z_close() Plugins should probably implement this. It is called when the reader/writer is closed. Here the plugin can do a cleanup, flush, etc. Nrdr4rrrr]|s zFormat._BaseReaderWriter._close)N)ra __module__ __qualname__r9r3propertyrSrBrUrZr\rVr_rTrPr]rrrr_BaseReaderWriter,s      rjc@sjeZdZdZddZddZddZdd Zdd d Zd dZ ddZ ddZ ddZ ddZ ddZd S)z Format.Readerah The purpose of a reader object is to read data from an image resource, and should be obtained by calling :func:`.get_reader`. A reader can be used as an iterator to read multiple images, and (if the format permits) only reads data from the file when new data is requested (i.e. streaming). A reader can also be used as a context manager so that it is automatically closed. Plugins implement Reader's for different formats. Though rare, plugins may provide additional functionality (beyond what is provided by the base reader class). cCs|S)aget_length() Get the number of images in the file. (Note: you can also use ``len(reader_object)``.) The result can be: * 0 for files that only have meta data * 1 for singleton images (e.g. in PNG, JPEG, etc.) * N for image series * inf for streams (series of unknown length) ) _get_lengthr4rrr get_lengths zFormat.Reader.get_lengthcKsL|||_z|j|f|\}}Wntk r@t|YnXt||S)awget_data(index, **kwargs) Read image data from the file, using the image index. The returned image has a 'meta' attribute with the meta data. Raises IndexError if the index is out of range. Some formats may support additional keyword arguments. These are listed in the documentation of those formats. )rTrM _get_data StopIterationrr)r-indexrQimmetarrrget_datas zFormat.Reader.get_datacKs|j|jdf|S)zget_next_data(**kwargs) Read the next image from the series. Some formats may support additional keyword arguments. These are listed in the documentation of those formats. r)rrrMrfrrr get_next_dataszFormat.Reader.get_next_datacKs*||}tt|dd||_dS)zset_image_index(index) Set the internal pointer such that the next call to get_next_data() returns the image specified by the index rrKN)rTrlminmaxrM)r-rorQnrrrset_image_indexszFormat.Reader.set_image_indexNcCs0|||}t|ts,td|jj|S)aVget_meta_data(index=None) Read meta data from the file. using the image index. If the index is omitted or None, return the file's (global) meta data. Note that ``get_data`` also provides the meta data for the returned image as an attribute of that image. The meta data is a dict, which shape depends on the format. E.g. for JPEG, the dict maps group names to subdicts and each group is a dict with name-value pairs. The groups represent the different metadata formats (EXIF, XMP, etc.). z Meta data must be a dict, not %r)rT_get_meta_datar$dictr+r`ra)r-rorqrrr get_meta_datas   zFormat.Reader.get_meta_dataccs||}d}||krz||\}}Wn<tk rDYdStk rj|tdkrdYdSYnXt||V|d7}qdS)ziter_data() Iterate over all images in the series. (Note: you can also iterate over the reader object.) rNinfr)rTrlrmrnrfloatr)r-rvirprqrrr iter_datas  zFormat.Reader.iter_datacCs|Sr6)r~r4rrr__iter__szFormat.Reader.__iter__cCs|}|tdkrtj}|S)Nr{)rlr|sysmaxsize)r-rvrrr__len__s zFormat.Reader.__len__cCs tdS)z_get_length() Plugins must implement this. The returned scalar specifies the number of images in the series. See Reader.get_length for more information. Nrdr4rrrrkszFormat.Reader._get_lengthcCs tdS)z_get_data() Plugins must implement this, but may raise an IndexError in case the plugin does not support random access. It should return the image and meta data: (ndarray, dict). Nrdr-rorrrrmszFormat.Reader._get_datacCs tdS)z_get_meta_data(index) Plugins must implement this. It should return the meta data as a dict, corresponding to the given index, or to the file's (global) meta data if index is None. Nrdrrrrrxs zFormat.Reader._get_meta_data)N)rargrhr9rlrrrsrwrzr~rrrkrmrxrrrrr@s    r@c@s2eZdZdZd ddZddZddZd d ZdS) z Format.Writera2 The purpose of a writer object is to write data to an image resource, and should be obtained by calling :func:`.get_writer`. A writer will (if the format permits) write data to the file as soon as new data is provided (i.e. streaming). A writer can also be used as a context manager so that it is automatically closed. Plugins implement Writer's for different formats. Though rare, plugins may provide additional functionality (beyond what is provided by the base writer class). NcCs~|t|tjstdi}t|drBt|jtrB||j|dkrLnt|ts`tdn ||t |}| ||S)zappend_data(im, meta={}) Append an image (and meta data) to the file. The final meta data that is used consists of the meta data on the given image (if applicable), updated with the given meta data. z)append_data requires ndarray as first argrqNMeta must be a dict.) rTr$npZndarrayr+hasattrrqryupdater _append_data)r-rprqZ total_metarrr append_data5s     zFormat.Writer.append_datacCs*|t|tstdn ||SdS)aset_meta_data(meta) Sets the file's (global) meta data. The meta data is a dict which shape depends on the format. E.g. for JPEG the dict maps group names to subdicts, and each group is a dict with name-value pairs. The groups represents the different metadata formats (EXIF, XMP, etc.). Note that some meta formats may not be supported for writing, and individual fields may be ignored without warning if they are invalid. rN)rTr$ryr+_set_meta_datar-rqrrr set_meta_dataPs   zFormat.Writer.set_meta_datacCs tdSr6rd)r-rprqrrrreszFormat.Writer._append_datacCs tdSr6rdrrrrriszFormat.Writer._set_meta_data)N)rargrhr9rrrrrrrrrD&s  rD)NN)rargrhr9r3r5r8rir7r.r/r0r1rCrErHrJrFrIobjectrjr@rDrrrrrTs0# 1     ]rc@sveZdZdZeddZddZddZdd Zd d Z d d Z ddZ dddZ ddZ ddZddZddZdS) FormatManagera The FormatManager is a singleton plugin factory. The format manager supports getting a format object using indexing (by format name or extension). When used as an iterator, this object yields all registered format objects. See also :func:`.help`. c CsHt}tD]4}tt |jr8|jdk r8||W5QRXq|Sr6) r)rvalues contextlibsuppress ImportErrorrrSappend)r-Zavailable_formatsconfigrrr_formatsys   zFormatManager._formatscCsdt|jdS)Nzlenrr4rrrr5szFormatManager.__repr__cCstdd|jDS)Ncss|] }|jVqdSr6)rSrrrrr sz)FormatManager.__iter__..)iterrr4rrrrszFormatManager.__iter__cCs t|jSr6rr4rrrrszFormatManager.__len__cCsPg}|jD]:}|jd}|jd}|jd|d|d}||q d|S)Nr0r/z - z [] )r legacy_argsr.rjoin)r-ssrextdescsrrrr8s    zFormatManager.__str__c Cstjdtddt|ts"td|dkr2tdt|rfzt|ddd j WStk rdYnXt | }z|j WSt k rt d |jd |jd YnXdS) NzjThe usage of `FormatManager` is deprecated and it will be removed in Imageio v3. Use `iio.imopen` instead.r stacklevelz;Looking up a format should be done by name or by extension.r z#No format matches the empty string.rTZ legacy_modezThe `z4` format is not installed. Use `pip install imageio[z]` to install it.)warningswarnDeprecationWarningr$r%r+ris_filer rNrr!rSrr. install_name)r-r.rrrr __getitem__s.   zFormatManager.__getitem__cs"tjdtdd|D]4tts*tdtfdddDrtdqt|d k}|r^t }d d |D}d d t D}|D]Z}|r| qt |D]>d d |jDD](}|r|j||jd |qqqt}t|D] |d}|dk r|t<qt|dS)aksort(name1, name2, name3, ...) Sort the formats based on zero or more given names; a format with a name that matches one of the given names will take precedence over other formats. A match means an equal name, or ending with that name (though the former counts higher). Case insensitive. Format preference will match the order of the given names: using ``sort('TIFF', '-FI', '-PIL')`` would prefer the FreeImage formats over the Pillow formats, but prefer TIFF even more. Each time this is called, the starting point is the default format order, and calling ``sort()`` with no arguments will reset the order. Be aware that using the function can affect the behavior of other code that makes use of imageio. Also see the ``IMAGEIO_FORMAT_ORDER`` environment variable. a6`FormatManager` is deprecated and it will be removed in ImageIO v3. Migrating `FormatManager.sort` depends on your use-case: - modify `iio.config.known_plugins` to specify the search order for unrecognized formats. - modify `iio.config.known_extensions[].priority` to control a specific extension.rrz)formats.sort() accepts only string names.c3s|]}|kVqdSr6r)rcr.rrrsz%FormatManager.sort..z.,zHNames given to formats.sort() should not contain dots `.` or commas `,`.rcSs g|]}|dkr|qS)r )rr!)rr.rrrrsz&FormatManager.sort..cSsg|]}|D]}|q qSrr)rZext_listrrrrrscSsg|]}|qSrrrrrrrsN)rrrr$r% TypeErroranyr+rr r rresetreversedZdefault_priorityendswithrremoveinsertrrRclearpopr)r-namesZ should_resetZ sane_namesZflat_extensions extensionrZ old_orderrrrsortsF        zFormatManager.sortFc Cstjdtddt|ts$tdn$|sH|j|krHtd|jdt|j |j j |j j dd|j|j d |j|jd d }|t|j<|jD]6}t||jgd d |jdd}t|t|qdS)zadd_format(format, overwrite=False) Register a format, so that imageio can use it. If a format with the same name already exists, an error is raised, unless overwrite is True, in which case the current format is replaced. z`FormatManager` is deprecated and it will be removed in ImageIO v3.To migrate `FormatManager.add_format` add the plugin directly to `iio.config.known_plugins`.rrz/add_format needs argument to be a Format objectzA Format named z8 is already registered, use `overwrite=True` to replace.Tunknownr)r.r/r0r1)r. class_name module_namerrrz Unique Formatz7A format inserted at runtime. It is being read by the `z ` plugin.)rrr.r/N)rrrr$rr+r.get_format_namesr r!r`rargr/rr0r1rr r setdefaultr)r)r-Z iio_format overwriterrrrrr add_formats@       zFormatManager.add_formatcCsVzt||jjddjWStk r<tjdtddYdStk rPYdSXdS)zsearch_read_format(request) Search a format that can read a file according to the given request. Returns None if no appropriate format was found. (used internally) TrzrImageIO now uses a v3 plugin when reading this format. Please migrate to the v3 API (preferred) or use imageio.v2.rrN r r=Zio_moderNAttributeErrorrrrr+rGrrrsearch_read_format5sz FormatManager.search_read_formatcCsVzt||jjddjWStk r<tjdtddYdStk rPYdSXdS)zsearch_write_format(request) Search a format that can write a file according to the given request. Returns None if no appropriate format was found. (used internally) TrzrImageIO now uses a v3 plugin when writing this format. Please migrate to the v3 API (preferred) or use imageio.v2.rrNrrGrrrsearch_write_formatLsz!FormatManager.search_write_formatcCs tjdtdddd|jDS)z(Get the names of all registered formats.z`FormatManager` is deprecated and it will be removed in ImageIO v3.To migrate `FormatManager.get_format_names` use `iio.config.known_plugins.keys()` instead.rrcSsg|] }|jqSrr)rfrrrrmsz2FormatManager.get_format_names..)rrrrr4rrrrcs zFormatManager.get_format_namescCs t|dS)z1Show a nicely formatted list of available formatsN)printr4rrrshowoszFormatManager.showN)F)rargrhr9rirr5rrr8rrrrrrrrrrrrns   #H 3 r)r9rrrnumpyrpathlibrr rrrBrrrr r r Zconfig.pluginsr r Z MODENAMESrrrrrrrrs$