add batch task runner

bin/benchmark.py

@@ -0,0 +1,65 @@
+#!/usr/bin/env python
+
+import subprocess, json, os
+
+POLICIES = ["LRU", "Random", "LIP", "BIP", "DIP"]
+
+RESULT_DIR = "result"
+
+LOCAL_CFG_FILE = "./cfg.json"
+LOCAL_TRACE_CFG_FILE = "./traces.json"
+
+TEMPLATE_CFG_FILE = "../cfg/cfg.json"
+TEMPLATE_TRACE_CFG_FILE = "../cfg/traces.json"
+
+def set_policy_lambda(policy):
+    def f(node):
+        if node['name'].find('L1') != -1:
+            node['policy'] = policy
+    return f
+
+def create_cfg_file(filename, policy):
+    with open(TEMPLATE_CFG_FILE, "r") as f:
+        buf = f.read()
+        c = json.loads(buf)
+        map(set_policy_lambda(policy), c["nodes"])
+        with open(filename, "w") as wf:
+            wf.write(json.dumps(c))
+
+def create_trace_cfg_file(filename, traces):
+    with open(TEMPLATE_TRACE_CFG_FILE, "r") as f:
+        buf = f.read()
+        c = json.loads(buf)
+        c["traces"] = traces
+        with open(filename, "w") as wf:
+            wf.write(json.dumps(c))
+
+def run(policy, traces):
+    create_cfg_file(LOCAL_CFG_FILE, policy)
+    create_trace_cfg_file(LOCAL_TRACE_CFG_FILE, traces)
+    batcmd="./lightsim -c %s -t %s -p %d" % \
+            (LOCAL_CFG_FILE, LOCAL_TRACE_CFG_FILE, len(traces))
+    result = subprocess.check_output(batcmd, shell=True)
+    os.remove(LOCAL_CFG_FILE)
+    os.remove(LOCAL_TRACE_CFG_FILE)
+    return result
+
+def run_stand_alone_trace(trace):
+    trace_name = trace.split("/")[-1].split(".")[0]
+
+    if not os.path.exists(RESULT_DIR):
+        os.makedirs(RESULT_DIR)
+
+    sub_dir = '/'.join([RESULT_DIR, trace_name])
+    if os.path.exists(sub_dir):
+        raise Exception("%s exists, delete the dir first"%sub_dir)
+    os.makedirs(sub_dir)
+
+    for policy in POLICIES:
+        result_file = '/'.join([sub_dir, policy])
+        with open(result_file, "w") as f:
+            ret = run(policy, [trace])
+            f.write(ret)
+
+if __name__ == "__main__":
+    run_stand_alone_trace("../traces/gzip.trace.gz")