2 files changed, 305 insertions, 0 deletions

diff --git a/compact_pm_ovd.py b/compact_pm_ovd.py
new file mode 100755
index 0000000..5024d61
--- /dev/null
+++ b/compact_pm_ovd.py
@@ -0,0 +1,111 @@
+#!/usr/bin/env python
+
+from os.path import splitext, basename, dirname
+from optparse import OptionParser
+
+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 build_name(dirname, conf, wss_list):
+    if dirname == '':
+        dirname = '.'
+    outname = dirname + '/'
+    if 'hard' in conf:
+        outname += 'hard'
+    else:
+        outname += 'soft'
+    outname += '_pm_plugin=' + conf['plugin'] + '_dist=' + conf['dist']
+    # atm we use only uniform for PM (we also use only light, but...)
+    if 'light' in conf:
+        outname += '_light'
+    elif 'medium' in conf:
+        outname += '_medium'
+    elif 'heavy' in conf:
+        outname += '_heavy'
+
+    outname += '_wss='
+    for wss in wss_list:
+        if wss != wss_list[len(wss_list) - 1]:
+            outname += str(wss) + ','
+        else:
+            outname += str(wss)
+    outname += '_ovd=' + conf['ovd'] + '.csv'
+    return outname
+
+
+class WSSCompactor():
+    def __init__(self):
+        self.ovhead_table = {}
+        self.wss_list = []
+
+    def add_wss(self, wss):
+        self.wss_list.append(int(wss))
+
+    def read_csv(self, datafile):
+        csvf = open(datafile, 'r')
+        # csvf format is: tss, max, avg
+        for line in csvf:
+            tmp = line.split(', ')
+            if tmp[0] in self.ovhead_table.keys():
+                self.ovhead_table[tmp[0]] += [tmp[1], tmp[2].rstrip('\n')]
+            else:
+                self.ovhead_table[tmp[0]] = [tmp[1], tmp[2].rstrip('\n')]
+        csvf.close()
+
+    def write_csv(self, dirname, conf):
+        self.wss_list.sort()
+        outname = build_name(dirname, conf, self.wss_list)
+        csvf = open(outname, 'w')
+        csvf.write('# preemption and migration overhead for ' + \
+                conf['plugin'] + '\n')
+
+        wstr = '# tssize, '
+        for wss in self.wss_list:
+            wstr += 'wss=' + str(wss) + ':(max, avg), '
+        csvf.write(wstr[0:-2] + '\n')
+
+        tkeys = self.ovhead_table.keys()
+        for i in range(0,len(tkeys)):
+            tkeys[i] = int(tkeys[i])
+        tkeys.sort()
+
+        for key in tkeys:
+            wstr = str(key) + ', '
+            for value in self.ovhead_table[str(key)]:
+                wstr += value + ', '
+            wstr = wstr[0:-2]
+            csvf.write(wstr + '\n')
+
+        csvf.close()
+
+def main(args):
+    wsscmpctr = WSSCompactor()
+    for datafile in args:
+        bname = basename(datafile)
+        fname, ext = splitext(bname)
+        if ext != '.csv':
+            self.err("Warning: '%s' doesn't look like a CSV file."
+                     % bname)
+        conf = decode(fname)
+        wsscmpctr.add_wss(conf['wss'])
+        wsscmpctr.read_csv(datafile)
+
+    # assume all files in same directory and same kind of overheads
+    wsscmpctr.write_csv(dirname(datafile), conf)
+
+if __name__ == "__main__":
+    usage = "Usage: %prog list-of-file"
+    parser = OptionParser(usage=usage)
+    (options, args) = parser.parse_args()
+    if len(args) < 1:
+        parser.error("Argument missing")
+        sys.exit(-1)
+    main(args)
diff --git a/plot_pm.py b/plot_pm.py
new file mode 100755
index 0000000..7388dce
--- /dev/null
+++ b/plot_pm.py
@@ -0,0 +1,194 @@
+#!/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(None, '--shared-l3', action='store_true', dest='sharedL3'),
+    o('-f', '--format', action='store', dest='format', type='choice',
+      choices=FORMATS, help='output format'),
+    o(None, '--paper', action='store_true', dest='paper'),
+    o(None, '--wide', action='store_true', dest='wide'),
+    ]
+
+defaults = {
+    'sharedL3' : False,
+    'format' : 'show',
+    'xrange' : (49.5, 250.5),
+    'yrange' : (0, 500.05),
+    'xticks' : (0, 10),
+    'yticks' : (0, 50),
+    'title'  : None,
+    'xlabel' : 'number of tasks',
+    'ylabel' : 'overhead (us)',
+    'paper'  : False,
+    'split'  : False,
+    'wide'   : 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, want_period=False, sharedL3=False):
+    ovd_type = 'unknowkn overhead'
+    if 'ovd' in conf:
+        if conf['ovd'] == 'preemption':
+            ovd_type = 'Preemption ovhead'
+        elif conf['ovd'] == 'onchip':
+            if sharedL3:
+                ovd_type = 'Shared L3 migration ovhead'
+            else:
+                ovd_type = 'Shared L2 migration ovhead'
+        elif conf['ovd'] == 'l2cache':
+            ovd_type = 'Shared L2 migration ovhead'
+        elif conf['ovd'] == 'offchip':
+            ovd_type = 'Off Chip migration ovhead'
+
+    dist = 'unknown distribution'
+    if 'dist' in conf:
+        if conf['dist'] == 'uni':
+            dist = 'utilization uniformly '
+            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 = 'util. bimodially '
+            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)'
+    if want_period:
+        if '33' in conf:
+            dist += '; period uniformly in [3, 33]'
+        elif '250' in conf:
+            dist += '; period uniformly in [50, 250]'
+        else:
+            dist += '; period uniformly in [10, 100]'
+
+    return ovd_type + '; ' + dist
+
+class PmPlotter(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 +
+                (' [avg]' if 'soft' in conf else ' [max]'),
+                xrange=self.options.xrange,