From f48482529e34cc5d9f7d11681febb57905251b47 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Tue, 20 Apr 2021 09:10:23 +0200
Subject: add dot export of energy models

---
 bin/analyze-archive.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'bin')

diff --git a/bin/analyze-archive.py b/bin/analyze-archive.py
index 6f5fc37..8b58478 100755
--- a/bin/analyze-archive.py
+++ b/bin/analyze-archive.py
@@ -614,6 +614,12 @@ if __name__ == "__main__":
         type=str,
         help="Load DFA hardware model from JSON or YAML FILE",
     )
+    parser.add_argument(
+        "--export-dot",
+        metavar="FILE",
+        type=str,
+        help="Export PTA representation suitable for Graphviz dot to FILE",
+    )
     parser.add_argument(
         "--export-energymodel",
         metavar="FILE",
@@ -1222,4 +1228,13 @@ if __name__ == "__main__":
         with open(args.export_energymodel, "w") as f:
             json.dump(json_model, f, indent=2, sort_keys=True, cls=NpEncoder)
 
+    if args.export_dot:
+        if not pta:
+            print(
+                "Note: v0 measurements do not embed the PTA used for benchmark generation. Estimating PTA from recorded observations."
+            )
+        json_model = model.to_json()
+        with open(args.export_dot, "w") as f:
+            f.write(model.to_dot())
+
     sys.exit(0)
-- 
cgit v1.2.3