1 """The driver module provides :class:`Driver`\s for identifying and
4 This allows Hooke to be data-file agnostic. Drivers for various
5 commercial force spectroscopy microscopes are provided, and it's easy
6 to write your own to handle your lab's specific format.
9 from ..config import Setting
10 from ..util.graph import Node, Graph
14 # ('csvdriver', True),
16 # ('hemingclamp', True),
19 # ('mfp1dexport', True),
21 # ('picoforce', True),
22 # ('picoforcealt', True),
25 """List of driver modules and whether they should be included by
26 default. TODO: autodiscovery
31 """Base class for file format drivers.
33 :attr:`name` identifies your driver, and should match the module
36 def __init__(self, name):
38 self.setting_section = '%s driver' % self.name
40 def dependencies(self):
41 """Return a list of :class:`Driver`\s we require."""
44 def default_settings(self):
45 """Return a list of :class:`hooke.config.Setting`\s for any
46 configurable driver settings.
48 The suggested section setting is::
50 Setting(section=self.setting_section, help=self.__doc__)
54 def is_me(self, path):
55 """Read the file and return True if the filetype can be
56 managed by the driver. Otherwise return False.
61 """Read data from `path` and return a
62 (:class:`hooke.curve.Data`, `info`) tuple.
64 The `info` :class:`dict` must contain values for the keys:
65 'filetype' and 'experiment'. See :class:`hooke.curve.Curve`
68 raise NotImplementedError
70 # Construct driver dependency graph and load default drivers.
73 """(name, instance) :class:`dict` of all possible :class:`Driver`\s.
76 for driver_modname,default_include in DRIVER_MODULES:
77 assert len([mod_name for mod_name,di in DRIVER_MODULES]) == 1, \
78 'Multiple %s entries in DRIVER_MODULES' % mod_name
79 this_mod = __import__(__name__, fromlist=[driver_modname])
80 driver_mod = getattr(this_mod, driver_modname)
81 for objname in dir(driver_mod):
82 obj = getattr(driver_mod, objname)
84 subclass = issubclass(obj, Driver)
87 if subclass == True and obj != Driver:
89 if d.name != driver_modname:
90 raise Exception('Driver name %s does not match module name %s'
91 % (d.name, driver_modname))
94 DRIVER_GRAPH = Graph([Node([DRIVERS[name] for name in d.dependencies()],
96 for d in DRIVERS.values()])
97 DRIVER_GRAPH.topological_sort()
100 def default_settings():
102 'drivers', help='Enable/disable default drivers.')]
103 for dnode in DRIVER_GRAPH:
105 default_include = [di for mod_name,di in DRIVER_MODULES
106 if mod_name == driver.name][0]
107 help = driver.__doc__.split('\n', 1)[0]
108 settings.append(Setting(
111 value=str(default_include),
114 for dnode in DRIVER_GRAPH:
116 settings.extend(driver.default_settings())