measurestack: Add a `dbg_dumpgraph` toggle

1 files changed, 70 insertions, 0 deletions

diff --git a/build-aux/measurestack/analyze.py b/build-aux/measurestack/analyze.py
index 3996b3b..86c51f2 100644
--- a/build-aux/measurestack/analyze.py
+++ b/build-aux/measurestack/analyze.py
@@ -9,6 +9,8 @@ import typing
 
 from . import vcg
 
+# Whether to print the graph in a /* comment */ before processing it
+dbg_dumpgraph = False
 # Whether to print "//dbg-nstatic:" lines that trace nstatic() execution
 dbg_nstatic = False
 
@@ -24,6 +26,72 @@ __all__ = [
     "analyze",
 ]
 
+
+def dumps(x: typing.Any, depth: int = 0, compact: bool = False) -> str:
+    match x:
+        case int() | str() | None:
+            return repr(x)
+        case dict():
+            if len(x) == 0:
+                return "{}"
+            ret = "{"
+            if not compact:
+                ret += "\n"
+            for k, v in x.items():
+                if not compact:
+                    ret += "\t" * (depth + 1)
+                ret += dumps(k, depth + 1, True)
+                ret += ":"
+                if not compact:
+                    ret += " "
+                ret += dumps(v, depth + 1, compact)
+                ret += ","
+                if not compact:
+                    ret += "\n"
+            if not compact:
+                ret += "\t" * depth
+            ret += "}"
+            return ret
+        case list():
+            if len(x) == 0:
+                return "[]"
+            ret = "["
+            if not compact:
+                ret += "\n"
+            for v in x:
+                if not compact:
+                    ret += "\t" * (depth + 1)
+                ret += dumps(v, depth + 1, compact)
+                ret += ","
+                if not compact:
+                    ret += "\n"
+            if not compact:
+                ret += "\t" * depth
+            ret += "]"
+            return ret
+        case set():
+            if len(x) == 0:
+                return "set()"
+            ret = "{"
+            if not compact:
+                ret += "\n"
+            for v in x:
+                if not compact:
+                    ret += "\t" * (depth + 1)
+                ret += dumps(v, depth + 1, compact)
+                ret += ","
+                if not compact:
+                    ret += "\n"
+            if not compact:
+                ret += "\t" * depth
+            ret += "}"
+            return ret
+        case _:
+            if hasattr(x, "__dict__"):
+                return f"{x.__class__.__name__}(*{dumps(x.__dict__, depth, compact)})"
+            return f"TODO({x.__class__.__name__})"
+
+
 # types ########################################################################
 
 
@@ -335,6 +403,8 @@ def analyze(
     cfg_max_call_depth: int,
 ) -> AnalyzeResult:
     graphdata = _make_graph(ci_fnames, app)
+    if dbg_dumpgraph:
+        print(f"/* {dumps(graphdata)} */")
 
     missing: set[QName] = set()
     dynamic: set[QName] = set()