Compute abs path to overheads directory

1 files changed, 14 insertions, 4 deletions

diff --git a/ecrts14/ecrts14.py b/ecrts14/ecrts14.py
index 68200a9..b0a381a 100755
--- a/ecrts14/ecrts14.py
+++ b/ecrts14/ecrts14.py
@@ -8,6 +8,7 @@ import sys
 import os
 import math
 import time
+import inspect
 
 import sqlite3 as lite
 import json
@@ -233,18 +234,26 @@ def complete(results, n):
 def update_mean(old_mean, n, new_sample):
     return (old_mean*n + new_sample)/(n+1)
 
+def get_ovh_dir():
+    parent = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
+    ovh_dir = os.path.join(parent, 'overheads')
+    return ovh_dir
+
 def get_consumer_overheads(dp, _system):
-    co_file = 'overheads/%s/consumer/dco_host=%s_lvl=mem_polluters=%s_walk=%s_hpages=%s_upages=%s_type=%s.csv' % (dp.host, dp.host, str(dp.polluters), dp.walk, str(dp.huge_pages), str(dp.uncached), dp.ovh_type)
+    co_file = '%s/consumer/dco_host=%s_lvl=mem_polluters=%s_walk=%s_hpages=%s_upages=%s_type=%s.csv' % (dp.host, dp.host, str(dp.polluters), dp.walk, str(dp.huge_pages), str(dp.uncached), dp.ovh_type)
+    co_file = os.path.join(get_ovh_dir(), co_file)
     co = ConsumerOverheads.from_file(co_file, non_decreasing=False, system=_system)
     return co
 
 def get_producer_overheads(dp):
-    po_file = 'overheads/%s/producer/dpo_host=%s_type=%s.csv' % (dp.host, dp.host, dp.ovh_type)
+    po_file = '%s/producer/dpo_host=%s_type=%s.csv' % (dp.host, dp.host, dp.ovh_type)
+    po_file = os.path.join(get_ovh_dir(), po_file)
     po = ProducerOverheads.from_file(po_file, non_decreasing=False)
     return po
 
 def get_cpmds(dp):
-    cpmd_file = 'overheads/%s/cpmd/dpmo_host=%s_lvl=mem_wcycle=%s_polluters=%s_walk=%s_hpages=%s_upages=%s_type=%s.csv' % (dp.host, dp.host, str(dp.wcycle), str(dp.polluters), dp.walk, str(dp.huge_pages), str(dp.uncached), dp.ovh_type)
+    cpmd_file = '%s/cpmd/dpmo_host=%s_lvl=mem_wcycle=%s_polluters=%s_walk=%s_hpages=%s_upages=%s_type=%s.csv' % (dp.host, dp.host, str(dp.wcycle), str(dp.polluters), dp.walk, str(dp.huge_pages), str(dp.uncached), dp.ovh_type)
+    cpmd_file = os.path.join(get_ovh_dir(), cpmd_file)
     cpmds = CacheDelay.from_file(cpmd_file, non_decreasing=False)
     return cpmds
 
@@ -253,7 +262,8 @@ def get_overheads(dp, system = None):
     max_dist = dp.system.distance(0, cluster_size-1)
     lvl = dp.system.levels[max_dist]
     max_wss = dp.system.max_wss()
-    ovh_file = 'overheads/ovh_host=%s_sched=%s_lvl=%s_type=%s.csv' % (dp.host, dp.sched, lvl, dp.ovh_type)
+    ovh_file = 'ovh_host=%s_sched=%s_lvl=%s_type=%s.csv' % (dp.host, dp.sched, lvl, dp.ovh_type)
+    ovh_file = os.path.join(get_ovh_dir(), ovh_file)
     ovh = Overheads.from_file(ovh_file)
     ovh.shared_cache = dp.system.schedcat_distance(0, max_dist)
     ovh.cache_affinity_loss = get_cpmds(dp)