#!/usr/bin/env python import defapp from os.path import splitext, basename from optparse import make_option as o from tempfile import NamedTemporaryFile as Tmp from gnuplot import gnuplot, FORMATS options = [ o('-f', '--format', action='store', dest='format', type='choice', choices=FORMATS, help='output format'), o(None, '--paper', action='store_true', dest='paper'), ] defaults = { 'format' : 'show', 'xrange' : (0.5, 32.5), 'yrange' : (-0.05, 1.05), 'xticks' : (0, 2), 'yticks' : (0, 0.1), 'title' : None, 'xlabel' : 'task set utilization cap (prior to inflation)', 'ylabel' : 'ratio of schedulable task sets', 'paper' : False, } def decode(name): params = {} parts = name.split('_') for p in parts: kv = p.split('=') k = kv[0] v = kv[1] if len(kv) > 1 else None params[k] = v return params def get_data_tmpfile(datafile): """Removes all comments form datafile, stores result in a temp file. The temp file is returned.""" count = 0 f = open(datafile, 'r') d = Tmp() for line in f: if len(line) > 1 and line[0] != '#': d.write(line) count += 1 f.close() d.flush() if count > 0: return d else: del d # removes temp file return None def scenario_heading(conf): dist = 'unknown distribution' if 'dist' in conf: if conf['dist'] == 'uni': dist = 'uniformly distributed ' if 'light' in conf: dist = dist + 'in [0.001, 0.1]' elif 'medium' in conf: dist = dist + 'in [0.1, 0.4]' elif 'heavy' in conf: dist = dist + 'in [0.5, 0.9]' elif conf['dist'] == 'bimo': dist = 'bimodially distributed ' if 'light' in conf: dist = dist + 'in [0.001, 0.5] (8/9) and [0.5, 0.9] (1/9)' elif 'medium' in conf: dist = dist + 'in [0.001, 0.5] (6/9) and [0.5, 0.9] (3/9)' elif 'heavy' in conf: dist = dist + 'in [0.001, 0.5] (4/9) and [0.5, 0.9] (5/9)' return dist class SchedPlotter(defapp.App): def __init__(self): defapp.App.__init__(self, options, defaults, no_std_opts=True) def plot(self, graphs, title, name, conf, **xtra): gnuplot(graphs, title=title, xlabel=self.options.xlabel, ylabel=self.options.ylabel + (' [soft]' if 'soft' in conf else ' [hard]'), xrange=self.options.xrange, yrange=self.options.yrange, xticks=self.options.xticks, yticks=self.options.yticks, format=self.options.format, fname=name, **xtra) def plot_paper(self, graphs, title, name, conf, **xtra): tops = 'color solid font "Helvetica,12" linewidth 1.0 rounded size 16cm,8.5cm' gnuplot(graphs, title=title, xlabel=self.options.xlabel, ylabel=self.options.ylabel + (' [soft]' if 'soft' in conf else ' [hard]'), xrange=self.options.xrange, yrange=self.options.yrange, xticks=self.options.xticks, yticks=self.options.yticks, format=self.options.format, fname=name, key='off', style='lines lw 7', term_opts=tops) def plot_spec(self, tmpfile, name, conf): title = 'G-EDF overhead speculation: ' + scenario_heading(conf) graphs = [ (tmpfile, 1, 2, 'G-EDF (100% release)'), (tmpfile, 1, 3, 'G-EDF ( 75% release)'), (tmpfile, 1, 4, 'G-EDF ( 50% release)'), (tmpfile, 1, 5, 'G-EDF ( 25% release)'), (tmpfile, 1, 6, 'G-EDF ( 0% release)'), (tmpfile, 1, 7, 'G-EDF (no overheads)'), ] if self.options.paper and self.options.format == 'pdf': self.plot_paper(graphs, title, name, conf) else: self.plot(graphs, title, name, conf) def plot_spec2(self, tmpfile, name, conf): title = 'service processor speculation: ' + scenario_heading(conf) graphs = [ (tmpfile, 1, 2, 'G-EDF ( 25% release, 100% rest)'), (tmpfile, 1, 3, 'G-EDF (100% release, 0% rest)'), (tmpfile, 1, 4, 'G-EDF (service cpu, 100% preempt)'), (tmpfile, 1, 5, 'G-EDF (service cpu, 150% preempt)'), ] self.plot(graphs, title, name, conf) def plot_spec3(self, tmpfile, name, conf): title = 'linear overhead charge speculation: ' + scenario_heading(conf) graphs = [ (tmpfile, 1, 2, 'G-EDF (100% release)'), (tmpfile, 1, 3, 'G-EDF ( 25% release)'), (tmpfile, 1, 4, 'G-EDF ( Lin release)'), ] self.plot(graphs, title, name, conf) def plot_spec4(self, tmpfile, name, conf): title = 'linear overhead charge speculation: ' + scenario_heading(conf) graphs = [ (tmpfile, 1, 2, 'G-EDF (100% release)'), (tmpfile, 1, 3, 'G-EDF-S (100% release, all cpus)'), (tmpfile, 1, 4, 'G-EDF-S (100% release, one cpu)'), (tmpfile, 1, 5, 'G-EDF (25% release)'), (tmpfile, 1, 6, 'G-EDF-S (25% release, all cpus)'), (tmpfile, 1, 7, 'G-EDF-S (25% release, one cpu)'), ] self.plot(graphs, title, name, conf) def plot_irq(self, tmpfile, name, conf): graphs = [ (tmpfile, 1, 2, 'quantum-centric (100% overhead)'), (tmpfile, 1, 3, 'task-centric (100% overhead)'), (tmpfile, 1, 4, 'processor-centric (100% overhead)'), (tmpfile, 1, 5, 'quantum-centric (20% overhead)'), (tmpfile, 1, 6, 'task-centric (20% overhead)'), (tmpfile, 1, 7, 'processor-centric (20% overhead)'), ] if self.options.paper and self.options.format == 'pdf': title = scenario_heading(conf) self.plot_paper(graphs, title, name, conf) else: title = 'Interrupt accounting under G-EDF: utilization ' + scenario_heading(conf) self.plot(graphs, title, name, conf) def plot_file(self, datafile): bname = basename(datafile) name, ext = splitext(bname) if ext != '.csv': self.err("Warning: '%s' doesn't look like a CSV file." % bname) conf = decode(name) tmpfile = get_data_tmpfile(datafile) if tmpfile: if 'spec' in conf: self.plot_spec(tmpfile.name, name, conf) elif 'spec2' in conf: self.plot_spec2(tmpfile.name, name, conf) elif 'spec3' in conf: self.plot_spec3(tmpfile.name, name, conf) elif 'spec4' in conf: self.plot_spec4(tmpfile.name, name, conf) elif 'irq' in conf: self.plot_irq(tmpfile.name, name, conf) else: self.err("Skipped '%s'; unkown experiment type." % bname) del tmpfile # removes temp file else: self.err("Skipped '%s'; it dosn't appear to contain data." % bname) def default(self, _): for datafile in self.args: self.plot_file(datafile) if __name__ == "__main__": SchedPlotter().launch()