Allow to save figure directly to disk

2 files changed, 20 insertions, 4 deletions

diff --git a/scripts/parse-depmod b/scripts/parse-depmod
index 4753969..e5df12d 100755
--- a/scripts/parse-depmod
+++ b/scripts/parse-depmod
@@ -2,6 +2,8 @@
 
 import sys
 import os.path
+import argparse
+
 
 def parse_table(f):
     curr = 0
@@ -37,8 +39,14 @@ def add_at(ax, t, loc=2):
     return _at
 
 
+parser = argparse.ArgumentParser()
+parser.add_argument('--save', action='store_true')
+parser.add_argument('files', nargs='*')
+args = parser.parse_args()
+
 figs = []
-for fn in sys.argv[1:]:
+
+for fn in args.files:
     fig = plt.figure()
     figs += [fig]
 
@@ -62,5 +70,8 @@ for fn in sys.argv[1:]:
     fig.tight_layout()
     fig.subplots_adjust(top=0.9)
 
-
-plt.show()
+if args.save:
+    for fig, fn in zip(figs, args.files):
+        fig.savefig(os.path.splitext(fn)[0] + '.png')
+else:
+    plt.show()
diff --git a/scripts/plot-timing b/scripts/plot-timing
index 297cf6e..972df18 100755
--- a/scripts/plot-timing
+++ b/scripts/plot-timing
@@ -23,6 +23,7 @@ def parse_table(f):
 
 parser = argparse.ArgumentParser()
 parser.add_argument('--name', type=str, default='')
+parser.add_argument('--save', action='store_true')
 parser.add_argument('file', nargs='*')
 args = parser.parse_args()
 
@@ -44,4 +45,8 @@ fig.suptitle('Hash function timings: %s' % args.name, weight='bold',
              size='large')
 fig.tight_layout()
 plt.subplots_adjust(top=0.9)
-plt.show()
+
+if args.save:
+    fig.savefig(os.path.dirname(args.file[0]) + '/plot.png')
+else:
+    plt.show()