From 7014fba3239b270673e80791d62bff71b2cce9d9 Mon Sep 17 00:00:00 2001
From: Austin Bingham <austin.bingham@gmail.com>
Date: Tue, 5 Aug 2014 19:46:15 +0200
Subject: [PATCH] Added command-line arguments to test.

---
 testsegy.py | 43 +++++++++++++++++++++++++++++++------------
 1 file changed, 31 insertions(+), 12 deletions(-)

diff --git a/testsegy.py b/testsegy.py
index 47b6c35..1b8fca9 100644
--- a/testsegy.py
+++ b/testsegy.py
@@ -46,16 +46,35 @@ def test_render(d_ieee, d_ibm, data):
     pylab.title('TEST')
     pylab.show()
 
-if __name__ == '__main__':
-    filename = 'ld0057_file_00095.sgy'
-    # filename = 'mini.sgy'
-    # filename = 'data_IEEE.segy'
-    # filename = 'data_IBM_REV1.segy'
-    # filename = 'data_IBM_REV0.segy'
-    # filename = 'data_1byteINT.segy'
-    # filename = 'data_2byteINT.segy'
-    # filename = 'data_4byteINT.segy'
 
-    read_results = test_read(filename)
-    # write_results = test_write('testout.segy', *read_results)
-    # test_render(*write_results)
+def parse_args():
+    import argparse
+
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('filename',
+                        help='The SEGY input file')
+
+    parser.add_argument('--output', '-o',
+                        dest='outfile',
+                        default='',
+                        help='The output file (optional).')
+
+    parser.add_argument('--render_test', '-r',
+                        dest='render_test',
+                        action='store_true',
+                        help='Whether to test rendering (only applies if '
+                             'output is produced.)')
+
+    return parser.parse_args()
+
+if __name__ == '__main__':
+    args = parse_args()
+
+    read_results = test_read(args.filename)
+
+    if args.outfile:
+        write_results = test_write(args.outfile, *read_results)
+
+        if args.render_test:
+            test_render(*write_results)