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()