o :ήc2)@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)imageiopil matplotlibqtr)allimshowimshow_collectioncCsggggggdadS)zMClear the plugin state to the default, i.e., where no plugins are loaded )imreadimsaverimread_collectionr _app_showN plugin_storerr@/tmp/pip-target-vg8gfxp4/lib/python/skimage/io/manage_plugins.py_clear_plugins/s rcCsNgd}|D] }t|tdqddtD}|D] }t|t|qdS)N)rrrrrrcss|] }|dkr|VqdS)rNr.0prrr Esz*_load_preferred_plugins..) _set_pluginpreferred_pluginskeys)io_typesp_type plugin_typesrrr_load_preferred_plugins>sr'c CsD|D]}|tvr qz t||dWdStttfyYqwdS)N)kind)r r ImportError RuntimeErrorOSError) plugin_type plugin_listpluginrrrr!Js r!cCsttdSN)rr'rrrrr Us r 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  r9c Cstjt}ttj|dd}|D]]}t|\}}d|vr(td|dq|t |<dd|d dD}d d|D}|D]}|t vrOt d ||fqAd |voWd |v}|r_| d |t|<tj|d dt|<qd S)z]Scan the plugins directory for .ini files and parse them to gather plugin meta-data. _pluginsz*.iniprovideszfile z6 not recognized as a scikit-image io plugin, skipping.cSsg|]}|qSr)strip)rsrrr tz!_scan_plugins..,cSsg|]}|tvr|qSrrrrrrr>usz9Plugin `%s` wants to provide non-existent `%s`. Ignoring.rrN)ospathdirname__file__rjoinr9warningswarnplugin_meta_datasplitrprintappendplugin_providesbasenameplugin_module_name) pd config_filesr4r6r7r;valid_providesrneed_to_add_collectionrrr _scan_pluginsgs0   rTFcCs`t}tD]}|D] \}}||q qi}tD]}|r"||vr-ddt|D||<q|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)rfrrrr>s z*find_available_plugins..)setrvaluesaddrM)loadedactive_plugins plugin_funcr.funcdrrrr s    r cs|tvr td|t|}t|dkrd}t|||dddur-|d\}}n!tz fdd|Dd}WntyMtd|fw||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. z Invalid function (%s) requested.rzNo suitable plugin registered for %s. 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)rrrWr.rrr>szcall_plugin..z&Could not find the plugin "%s" for %s.)r ValueErrorlenr*pop_load IndexError)r(argskwargs plugin_funcsmsgrUr^rr`rrs$     rcs|dur t}n|tvrtd|f|dkr |dg}n|g}t|D]&}|tvr5td|t|}fdd|Dfdd|D}|t|<q)dS) 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. Nz Plugin %s does not support `%s`.rrz$'%s' is not a known plugin function.cs g|] \}}|kr||fqSrrrnrWr6rrr> zuse_plugin..cs g|] \}}|kr||fqSrrrjrlrrr>rm)rr#rMr*rd)r6r(kfuncsrrlrrs&     rcCs>t|dst|drt|d}t|}t|d|dSdSdS)z9Add `imread_collection` to module if not already present.rrN)hasattrgetattrrsetattr)modulerr^rrr#_inject_imread_collection_if_needed s  rtcCs|tddvr dS|tvrtd|t|}td||gd}t|}|D]/}|dkr1t|nt||s?td||fq&t|}t ||}||f|vrU| ||fq&dS) zLoad the given plugin. Parameters ---------- plugin : str Name of plugin to load. See Also -------- plugins : List of available plugins T)r[NzPlugin %s not found.zskimage.io._plugins.)fromlistrz7Plugin %s does not provide %s as advertised. Ignoring.) r rOra __import__rMrtrprKrrqrL)r.modname plugin_moduler;rstorer^rrrrds.      rdcCs&zt|WStytd|w)zReturn plugin meta-data. Parameters ---------- plugin : str Name of plugin. Returns ------- m : dict Meta data as specified in plugin ``.ini``. zNo information on plugin "%s")rIKeyErrorrar`rrrr7s    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)r plugin_namerWrrrr>Wr?z plugin_order..r)rr^rrrr Ks r )Fr/)__doc__os.pathrBrG configparserrr collectionr__all__rrMrOrIr"rr'r!r r9rTr r rrrtrdrr rrrrs>         '8%