add script to print all facts about a result directory

1 files changed, 70 insertions, 0 deletions

diff --git a/scripts/print_facts.py b/scripts/print_facts.py
new file mode 100755
index 0000000..831cc78
--- /dev/null
+++ b/scripts/print_facts.py
@@ -0,0 +1,70 @@
+#!/usr/bin/env python3
+
+# Copyright 2018-2020 Florian Fischer <florian.fl.fischer@fau.de>
+#
+# This file is part of allocbench.
+#
+# allocbench is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# allocbench is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with allocbench.  If not, see <http://www.gnu.org/licenses/>.
+
+"""Print facts about an allocbench result directory"""
+
+import argparse
+import importlib
+import inspect
+import os
+import sys
+
+CURRENTDIR = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
+PARENTDIR = os.path.dirname(CURRENTDIR)
+sys.path.insert(0, PARENTDIR)
+
+import src.facter
+from src.plots import print_facts, print_common_facts
+from src.util import print_error
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Print facts about an allocbench result directory")
+    parser.add_argument("results", help="path to allocbench results", type=str)
+    args = parser.parse_args()
+
+    # Load common facts
+    src.facter.load_facts(args.results)
+
+    print_common_facts()
+
+    cwd = os.getcwd()
+    os.chdir(args.results)
+
+    for benchmark in src.globalvars.benchmarks:
+        bench_module = importlib.import_module(f"src.benchmarks.{benchmark}")
+
+        if not hasattr(bench_module, benchmark):
+            print_error(f"{benchmark} has no member {benchmark}")
+            print_error(f"Skipping {benchmark}.")
+
+        bench = getattr(bench_module, benchmark)
+
+        try:
+            bench.load()
+        except FileNotFoundError:
+            continue
+
+        print_facts(bench, print_allocators=True, print_common=False)
+
+    os.chdir(cwd)
+
+
+if __name__ == "__main__":
+    main()