a d(@sdZddlZddlZddlmZddlmZddlmZgdZ da iZ iZ iZ gdd gd gd Zd d Zed dZddZddZddZddZed&ddZeZddZd'ddZddZd d!Zd"d#Zd$d%ZdS)(aAHandle image reading, writing and plotting plugins. To improve performance, plugins are only loaded as needed. As a result, there can be multiple states for a given plugin: available: Defined in an *ini file located in `skimage.io._plugins`. See also `skimage.io.available_plugins`. partial definition: Specified in an *ini file, but not defined in the corresponding plugin module. This will raise an error when loaded. available but not on this system: Defined in `skimage.io._plugins`, but a dependent library (e.g. Qt, PIL) is not available on your system. This will raise an error when loaded. loaded: The real availability is determined when it's explicitly loaded, either because it's one of the default plugins, or because it's loaded explicitly by the user. N) ConfigParser)glob)imread_collection_wrapper) use_plugin call_plugin plugin_info plugin_order reset_pluginsfind_available_pluginsavailable_plugins)ZimageioZpil matplotlibr )allimshowimshow_collectioncCsggggggdadS)zMClear the plugin state to the default, i.e., where no plugins are loaded )imreadimsaverimread_collectionr _app_showN plugin_storerrb/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/skimage/io/manage_plugins.py_clear_plugins/srcCsNgd}|D]}t|tdq ddtD}|D]}t|t|q6dS)N)rrrrrrcss|]}|dkr|VqdS)rNr.0prrr Ez*_load_preferred_plugins..) _set_pluginpreferred_pluginskeys)Zio_typesZp_typeZ plugin_typesrrr_load_preferred_plugins>s r"c CsH|D]>}|tvrqzt||dWqDWqtttfy@Yq0qdS)N)kind)r r ImportError RuntimeErrorOSError)Z plugin_typeZ plugin_listpluginrrrrJs  rcCsttdS)N)rr"rrrrr Usr cCsHt}|||d}i}||D]}|||||<q*||fS)z>Return plugin name and meta-data dict from plugin config file.r)rreadsectionsoptionsget)filenameparsername meta_dataoptrrr_parse_config_fileZs  r1c Cstjt}ttj|dd}|D]}t|\}}d|vrPtd|dq$|t |<dd|d dD}d d|D}|D]"}|t vrt d |d |d qd |vod|v}|r| d |t|<tj|ddt|<q$dS)z]Scan the plugins directory for .ini files and parse them to gather plugin meta-data. Z_pluginsz*.iniprovideszfile z6 not recognized as a scikit-image io plugin, skipping.cSsg|] }|qSr)strip)rsrrr trz!_scan_plugins..,cSsg|]}|tvr|qSrrrrrrr5urzPlugin `z!` wants to provide non-existent `z `. Ignoring.rrN)ospathdirname__file__rjoinr1warningswarnplugin_meta_datasplitrprintappendplugin_providesbasenameplugin_module_name) pd config_filesr,r.r/r2Zvalid_providesrZneed_to_add_collectionrrr _scan_pluginsgs(   rHFcCs`t}tD]}|D]\}}||qqi}tD]&}|rD||vr4ddt|D||<q4|S)a1List available plugins. Parameters ---------- loaded : bool If True, show only those plugins currently loaded. By default, all plugins are shown. Returns ------- p : dict Dictionary with plugin names as keys and exposed functions as values. cSsg|]}|ds|qS)_) startswith)rfrrrr5s z*find_available_plugins..)setrvaluesaddrC)loadedZactive_pluginsZ plugin_funcr'funcdrrrr s   r c s|tvrtd|dt|}t|dkr@d|d}t||dddurb|d\}}nLtzfdd |Dd}Wn(tytd d |d Yn0||i|S) aFind the appropriate plugin of 'kind' and execute it. Parameters ---------- kind : {'imshow', 'imsave', 'imread', 'imread_collection'} Function to look up. plugin : str, optional Plugin to load. Defaults to None, in which case the first matching plugin is used. *args, **kwargs : arguments and keyword arguments Passed to the plugin function. zInvalid function (z ) requested.rz"No suitable plugin registered for z. You may load I/O plugins with the `skimage.io.use_plugin` command. A list of all available plugins are shown in the `skimage.io` docstring.r'Ncsg|]\}}|kr|qSrr)rrrKr'rrr5rzcall_plugin..zCould not find the plugin "z" for .)r ValueErrorlenr%pop_load IndexError)r#argskwargsZ plugin_funcsmsgrIrPrrRrrs    rcs|durt}n:|tvr4tdd|d|dkrF|dg}n|g}t|D]P}|tvrttd|dt|}fd d |Dfd d |D}|t|<qXdS) aSSet the default plugin for a specified operation. The plugin will be loaded if it hasn't been already. Parameters ---------- name : str Name of plugin. kind : {'imsave', 'imread', 'imshow', 'imread_collection', 'imshow_collection'}, optional Set the plugin for this function. By default, the plugin is set for all functions. See Also -------- available_plugins : List of available plugins Examples -------- To use Matplotlib as the default image reader, you would write: >>> from skimage import io >>> io.use_plugin('matplotlib', 'imread') To see a list of available plugins run ``io.available_plugins``. Note that this lists plugins that are defined, but the full list may not be usable if your system does not have the required libraries installed. NPlugin z does not support `z`.rr'z!' is not a known plugin function.cs g|]\}}|kr||fqSrrrnrKr.rrr5rzuse_plugin..cs g|]\}}|kr||fqSrrr^r`rrr5r)rr!rCr%rW)r.r#kfuncsrr`rrs    rcCs6t|ds2t|dr2t|d}t|}t|d|dS)z9Add `imread_collection` to module if not already present.rrN)hasattrgetattrrsetattr)modulerrPrrr#_inject_imread_collection_if_neededs rgcCs|tddvrdS|tvr,td|dnt|}td||gd}t|}|D]d}|dkrht|n"t||std|d |d qRt|}t ||}||f|vrR| ||fqRdS) zLoad the given plugin. Parameters ---------- plugin : str Name of plugin to load. See Also -------- plugins : List of available plugins T)rONr\z not found.zskimage.io._plugins.)fromlistrz does not provide z as advertised. Ignoring.) r rErT __import__rCrgrcrArrdrB)r'modnameZ plugin_moduler2rstorerPrrrrWs&     rWcCs2z t|WSty,td|dYn0dS)zReturn plugin meta-data. Parameters ---------- plugin : str Name of plugin. Returns ------- m : dict Meta data as specified in plugin ``.ini``. zNo information on plugin ""N)r?KeyErrorrTrRrrrr3s  rcCs(i}tD]}ddt|D||<q|S)zReturn the currently preferred plugin order. Returns ------- p : dict Dictionary of preferred plugin order, with function name as key and plugins (in order of preference) as value. cSsg|] \}}|qSrr)rZ plugin_namerKrrrr5Srz plugin_order..r)rrPrrrr Gs r )F)N)__doc__os.pathr8r= configparserrrZ collectionr__all__rrCrEr?r rr"rr r1rHr r rrrgrWrr rrrrs<        & 7$