X-Git-Url: http://git.tremily.us/?p=igor.git;a=blobdiff_plain;f=bin%2Figorbinarywave.py;fp=bin%2Figorbinarywave.py;h=70c1bc3132411b4e826586f52d1c842544d35c44;hp=6215628175bdde07d805217fba0ba1387e4f8b0e;hb=b290d098fee408c7811a1afe507740d70f22f6c4;hpb=e0a5ed57f0569a41ee72ed81c1242703cc67706e

diff --git a/bin/igorbinarywave.py b/bin/igorbinarywave.py
index 6215628..70c1bc3 100755
--- a/bin/igorbinarywave.py
+++ b/bin/igorbinarywave.py
@@ -27,13 +27,16 @@ from igor.binarywave import load
 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()