1 files changed, 30 insertions, 15 deletions

diff --git a/plot_rtas12.py b/plot_rtas12.py
index f4deef9..5cbcabd 100755
--- a/plot_rtas12.py
+++ b/plot_rtas12.py
@@ -2,6 +2,7 @@
 
 import sys
 import time
+import optparse
 
 from os.path import basename
 from os import listdir
@@ -27,6 +28,11 @@ COLS = {'plugin':       0,
         'iqr_min':      12}
 
 
+def parse_args():
+    p = optparse.OptionParser(description='make graphs')
+    p.add_option('--paper', dest='paper', action='store_true', default=False)
+    return p.parse_args()
+
 def usage(msg):
     print >>sys.stderr, msg
     sys.exit(1)
@@ -56,11 +62,9 @@ def get_overhead_title(ov):
           'LVLA-RELEASE': '(A)'}
     return OV[ov]
 
-def set_plot_opts(p):
+def set_plot_opts(opts, p):
     p.rounded_caps = True
     p.font = 'Helvetica'
-    p.font_size = '5pt'
-    p.size = ('8.5cm', '5.25cm')
     p.default_style = 'linespoints lw 2.5'
     p.default_style += ' smooth bezier'
     p.key = 'off'
@@ -69,6 +73,16 @@ def set_plot_opts(p):
     p.xrange = (18, 122)
     p.yrange = (0, '')
 
+    if opts.paper:
+        ext = 'pdf'
+        p.size = ('8.5cm', '5.25cm')
+        p.font_size = '5pt'
+    else:
+        ext = 'png'
+        p.size = (1024, 768)
+    p.output = '{0}.{1}'.format(p.output, ext)
+    p.format = ext
+
     for i, c in enumerate(p.curves):
         c.style = "linespoints ls %d" % (i + 1)
 
@@ -141,10 +155,9 @@ def include_level_a_releases(data_dir, o_type, scheduler, ycol):
     return (f.name, f)
 
 
-def plot_release(data_dir, ycol, title, fname):
+def plot_release(opts, data_dir, ycol, title, fname):
     p = Plot()
     p.output = '{0}/{1}'.format(data_dir, fname)
-    p.format = 'pdf'
     refs = [] # need to save reference to file handle so it is not deleted
 
     for o_type in ['RELEASE', 'LVLA-RELEASE']:
@@ -162,7 +175,7 @@ def plot_release(data_dir, ycol, title, fname):
                 ycol=ycol, title=ti)]
     p.xlabel = 'number of tasks'
     p.ylabel = 'overhead (microseconds)'
-    set_plot_opts(p)
+    set_plot_opts(opts, p)
     if gnuplot_col('avg') == ycol:
         # make this graph's y-scale match the scale on the average-case
         # scheduling overhead graph because they are next to each other
@@ -171,10 +184,9 @@ def plot_release(data_dir, ycol, title, fname):
         p.key = 'top left'
     p.gnuplot_exec()
 
-def plot_sched(data_dir, ycol, title, fname):
+def plot_sched(opts, data_dir, ycol, title, fname):
     p = Plot()
     p.output = '{0}/{1}'.format(data_dir, fname)
-    p.format = 'pdf'
 
     for o_type in ['SCHED', 'LVLA-SCHED']:
         for sched in SCHEDULERS:
@@ -184,19 +196,22 @@ def plot_sched(data_dir, ycol, title, fname):
                 ycol=ycol, title=ti)]
     p.xlabel = 'number of tasks'
     p.ylabel = 'overhead (microseconds)'
-    set_plot_opts(p)
+    set_plot_opts(opts, p)
     p.gnuplot_exec()
 
 
 def main():
-    if len(sys.argv) < 2:
+
+    opts, extra = parse_args()
+
+    if len(extra) < 1:
         usage('missing args')
 
-    data_dir = sys.argv[1]
-    plot_sched(data_dir, gnuplot_col('max'), 'worst-case scheduling overhead', 'overhead=SCHED_type=MAX.pdf')
-    plot_release(data_dir, gnuplot_col('max'), 'worst-case release overhead', 'overhead=RELEASE_type=MAX.pdf')
-    plot_sched(data_dir, gnuplot_col('avg'), 'average-case scheduling overhead', 'overhead=SCHED_type=AVG.pdf')
-    plot_release(data_dir, gnuplot_col('avg'), 'average-case release overhead', 'overhead=RELEASE_type=AVG.pdf')
+    data_dir = extra[0]
+    plot_sched(opts, data_dir, gnuplot_col('max'), 'worst-case scheduling overhead', 'overhead=SCHED_type=MAX')
+    plot_release(opts, data_dir, gnuplot_col('max'), 'worst-case release overhead', 'overhead=RELEASE_type=MAX')
+    plot_sched(opts, data_dir, gnuplot_col('avg'), 'average-case scheduling overhead', 'overhead=SCHED_type=AVG')
+    plot_release(opts, data_dir, gnuplot_col('avg'), 'average-case release overhead', 'overhead=RELEASE_type=AVG')
     
 if __name__ == '__main__':
     main()