Added an [-x FILE] option to igorpackedexperiment.py.

bin/igorpackedexperiment.py

@@ -42,11 +42,12 @@ def _run(self, args):
                 f.close()
         if args.verbose > 0:
             pprint.pprint(filesystem)
-        walk(filesystem, self._plot_wave_callback)
+        walk(filesystem, self._plot_wave_and_execute_callback)
 
-    def _plot_wave_callback(self, dirpath, key, value):
+    def _plot_wave_and_execute_callback(self, dirpath, key, value):
         if isinstance(value, WaveRecord):
             self.plot_wave(self.args, value.wave, title=dirpath + [key])
+            self.execute(self.args, value.wave, dirpath + [key])
 
 
 s = PackedScript(

igor/script.py

@@ -52,6 +52,13 @@ def __init__(self, description=None, filetype='IGOR Binary Wave (.ibw) file'):
         self.parser.add_argument(
             '-V', '--verbose', action='count', default=0,
             help='increment verbosity')
+        self.parser.add_argument(
+            '-x', '--execute', metavar='FILE', default='-',
+            help=('Path to a custom Python script, which will be executed on '
+                  'every wave file. The script can access the wave and the '
+                  'corresponding path through the global variables wave and '
+                  'dirpath.')
+        )
         self._num_plots = 0
 
     def run(self, *args, **kwargs):
@@ -86,6 +93,15 @@ def plot_wave(self, args, wave, title=None):
             pass
         self._num_plots += 1
 
+    def execute(self, args, wave, dirpath):
+        if args.execute == '-':
+            return  # no-op
+        try:
+            exec(open(args.execute).read(), {"wave": wave, "dirpath": dirpath})
+        except Exception as e:
+            _LOG.error('error executing on wave {} {}'.format(dirpath, e))
+            raise
+
     def display_plots(self):
         if self._num_plots:
             _matplotlib_pyplot.show()