B \(@s dZddlZejdr&ddlmZn ddlmZddlZddlmZddl m Z dd d d d d dgZ da iZ iZiZdddgdgdgdZddZeddZddZdd ZddZddZed(dd ZeZd d Zd)d!dZd"d#Zd$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. N3) ConfigParser)glob)imread_collection_wrapper use_plugin call_plugin plugin_info plugin_order reset_pluginsfind_available_pluginsavailable_pluginsZpilZ matplotlibZqt)allimshowimshow_collectioncCsggggggdadS)zNClear the plugin state to the default, i.e., where no plugins are loaded )imreadimsaverimread_collectionr _app_showN) plugin_storerr8lib/python3.7/site-packages/skimage/io/manage_plugins.py_clear_plugins6s rcCs\dddddg}x|D]}t|tdqWddtD}x|D]}t|t|qBWdS) Nrrrrrrcss|]}|dkr|VqdS)rNr).0prrr Msz*_load_preferred_plugins..) _set_pluginpreferred_pluginskeys)Zio_typesZp_typeZ plugin_typesrrr_load_preferred_pluginsFs  rc CsJxD|D]<}|tkrqyt||dPWqtttfk r@YqXqWdS)N)kind)r r ImportError RuntimeErrorOSError)Z plugin_typeZ plugin_listpluginrrrrRs  rcCsttdS)N)rrrrrrr ]scCsLt}|||d}i}x"||D]}|||||<q,W||fS)z>Return plugin name and meta-data dict from plugin config file.r)rreadZsectionsZoptionsget)filenameparsername meta_dataZoptrrr_parse_config_filebs  r+c Cstjt}ttj|dd}x|D]}t|\}}|t|<dd|ddD}dd|D}x$|D]}|t krjt d||fqjWd |kod |k}|r| d |t |<tj |d d t|<q&Wd S) z^Scan the plugins directory for .ini files and parse them to gather plugin meta-data. Z_pluginsz*.inicSsg|] }|qSr)strip)rsrrr {sz!_scan_plugins..provides,cSsg|]}|tkr|qSr)r)rrrrrr.|sz9Plugin `%s` wants to provide non-existent `%s`. Ignoring.rrN)ospathdirname__file__rjoinr+plugin_meta_datasplitrprintappendplugin_providesbasenameplugin_module_name) ZpdZ config_filesr'r)r*r/Zvalid_providesrZneed_to_add_collectionrrr _scan_pluginsos"     r>FcCslt}x,tD] }x|D]\}}||qWqWi}x.tD]&}|rN||kr>ddt|D||<q>W|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.sz*find_available_plugins..)setrvaluesaddr;)loadedZactive_pluginsZ plugin_funcr$funcdrrrr s  cs|tkrtd|t|}t|dkr8d}t|||dddkrZ|d\}}nHtyfdd|Dd}Wn$tk rtd|fYnX|||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)rrrA)r$rrr.szcall_plugin..z&Could not find the plugin "%s" for %s.)r ValueErrorlenr"pop_load IndexError)r argskwargsZ plugin_funcsmsgr?rFr)r$rrs     cs|dkrt}n4|tkr.td|f|dkr@|dg}n|g}txT|D]L}|tkrltd|t|}fdd|Dfdd|D}|t|<qTWdS) aTSet 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||fqSrr)rnrA)r)rrr. szuse_plugin..cs g|]\}}|kr||fqSrr)rrPrA)r)rrr. s)rrr;r"rK)r)r kZfuncsr)r)rrs       cCs6t|ds2t|dr2t|d}t|}t|d|dS)z9Add `imread_collection` to module if not already present.rrN)hasattrgetattrrsetattr)modulerrFrrr#_inject_imread_collection_if_neededs rVcCs|tddkrdS|tkr(td|nt|}td||gd}t|}xf|D]^}|dkrft|nt||std||fqPt|}t ||}||f|krP| ||fqPWdS) zLoad the given plugin. Parameters ---------- plugin : str Name of plugin to load. See Also -------- plugins : List of available plugins T)rENzPlugin %s not found.zskimage.io._plugins.)fromlistrz7Plugin %s does not provide %s as advertised. Ignoring.) r r=rH __import__r;rVrRr9rrSr:)r$modnameZ plugin_moduler/rZstorerFrrrrKs&        rKcCs.yt|Stk r(td|YnXdS)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"N)r7KeyErrorrH)r$rrrr ?scCs,i}x"tD]}ddt|D||<q W|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_namerArrrr._sz plugin_order..)r)rrFrrrr Ss  )F)N)__doc__sysversionr@Z configparserros.pathr2rZ collectionr__all__rr;r=r7rrrrr r+r>r r rrrVrKr r rrrrs@           ' 9%