#----------------------------------------------------------------------- #Copyright 2013 Centrum Wiskunde & Informatica, Amsterdam # #Author: Daniel M. Pelt #Contact: D.M.Pelt@cwi.nl #Website: http://dmpelt.github.io/pyastratoolbox/ # # #This file is part of the Python interface to the #All Scale Tomographic Reconstruction Antwerp Toolbox ("ASTRA Toolbox"). # #The Python interface to the ASTRA Toolbox is free software: you can redistribute it and/or modify #it under the terms of the GNU General Public License as published by #the Free Software Foundation, either version 3 of the License, or #(at your option) any later version. # #The Python interface to the ASTRA Toolbox is distributed in the hope that it will be useful, #but WITHOUT ANY WARRANTY; without even the implied warranty of #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #GNU General Public License for more details. # #You should have received a copy of the GNU General Public License #along with the Python interface to the ASTRA Toolbox. If not, see . # #----------------------------------------------------------------------- from . import plugin_c as p from . import log class base(object): def astra_init(self, cfg): try: try: req = self.required_options except AttributeError: log.warn("Plugin '" + self.__class__.__name__ + "' does not specify required options") req = {} try: opt = self.optional_options except AttributeError: log.warn("Plugin '" + self.__class__.__name__ + "' does not specify optional options") opt = {} try: optDict = cfg['options'] except KeyError: optDict = {} cfgKeys = set(optDict.keys()) reqKeys = set(req) optKeys = set(opt) if not reqKeys.issubset(cfgKeys): for key in reqKeys.difference(cfgKeys): log.error("Required option '" + key + "' for plugin '" + self.__class__.__name__ + "' not specified") raise ValueError("Missing required options") if not cfgKeys.issubset(reqKeys | optKeys): log.warn(self.__class__.__name__ + ": unused configuration option: " + str(list(cfgKeys.difference(reqKeys | optKeys)))) self.initialize(cfg) except Exception as e: log.error(str(e)) raise def register(name, className): """Register plugin with ASTRA. :param name: Plugin name to register :type name: :class:`str` :param className: Class name or class object to register :type className: :class:`str` or :class:`class` """ p.register(name,className) def get_registered(): """Get dictionary of registered plugins. :returns: :class:`dict` -- Registered plugins. """ return p.get_registered() def get_help(name): """Get help for registered plugin. :param name: Plugin name to get help for :type name: :class:`str` :returns: :class:`str` -- Help string (docstring). """ return p.get_help(name)