Add -p/--plot option so scripts will plot waves.

[igor.git] / bin / igorbinarywave.py

diff --git a/bin/igorbinarywave.py b/bin/igorbinarywave.py
index 6215628175bdde07d805217fba0ba1387e4f8b0e..70c1bc3132411b4e826586f52d1c842544d35c44 100755 (executable)
--- a/bin/igorbinarywave.py
+++ b/bin/igorbinarywave.py
@@ -27,13 +27,16 @@ from igor.binarywave import load
 from igor.script import Script
 
 
 from igor.script import Script
 
 

-def run(args):
-    wave = load(args.infile)
-    numpy.savetxt(args.outfile, wave['wave']['wData'], fmt='%g', delimiter='\t')
-    if args.verbose > 0:
-        wave['wave'].pop('wData')
-        pprint.pprint(wave)
-
-s = Script(description=__doc__)
-s._run = run
+class WaveScript (Script):
+    def _run(self, args):
+        wave = load(args.infile)
+        numpy.savetxt(
+            args.outfile, wave['wave']['wData'], fmt='%g', delimiter='\t')
+        self.plot_wave(args, wave)
+        if args.verbose > 0:
+            wave['wave'].pop('wData')
+            pprint.pprint(wave)
+
+
+s = WaveScript(description=__doc__)

 s.run()
 s.run()