[chattyparser] add tracking of threads

1 files changed, 17 insertions, 2 deletions

diff --git a/chattyparser.py b/chattyparser.py
index ff71c69..6ea7c1f 100755
--- a/chattyparser.py
+++ b/chattyparser.py
@@ -142,6 +142,7 @@ def record_allocation(trace, context):
        trace - Trace object ro record
        context - dict holding all data structures used for parsing
            allocations - dict of life allocations mapping their pointer to their size
+           threads - set of all used tid's
            hists - dict mapping allocation sizes to their occurrence
            total_size - list of total requested memory till last recorded function call
            cache_lines - dict of cache lines mapped to the owning tids
@@ -152,6 +153,7 @@ def record_allocation(trace, context):
     allocations = context.setdefault("allocations", [])
 
     # optional
+    threads = context.get("threads", None)
     hists = context.get("hists", None)
     total_size = context.get("total_size", None)
     cache_lines = context.get("cache_lines", None)
@@ -166,6 +168,9 @@ def record_allocation(trace, context):
     if trace.func == Function.uninitialized:
         return "WARNING: empty entry\n"
 
+    if threads is not None:
+        threads.add(trace.tid)
+
     # (potential) free of a pointer
     if trace.func in (Function.free, Function.realloc):
         if trace.func == Function.realloc:
@@ -231,6 +236,7 @@ def record_allocation(trace, context):
 
 def parse(path="chattymalloc.txt",
           hists=True,
+          threads=True,
           track_total=True,
           track_calls=True,
           cache_lines=False,
@@ -247,6 +253,9 @@ def parse(path="chattymalloc.txt",
     # Dictionary to track all live allocations
     context["allocations"] = {}
 
+    if threads:
+        context["threads"] = set()
+
     if track_calls:
         # function call histogram
         context["calls"] = {f: 0 for f in Function}
@@ -325,7 +334,11 @@ def plot(path):
             else:
                 total_hist[size] = total_hist.get(amount, 0) + 1
 
-    plot_ascii_summary(f"{path}.hist", hists, total_hist, result["calls"])
+    plot_ascii_summary(f"{path}.hist",
+                       hists,
+                       total_hist,
+                       result["calls"],
+                       threads=len(result["threads"]))
 
     if PLOT_PROFILE:
         top5 = [
@@ -368,10 +381,12 @@ def plot_profile(trace_path, plot_path, sizes):
     plt.clf()
 
 
-def plot_ascii_summary(path, hists, total_hist, calls):
+def plot_ascii_summary(path, hists, total_hist, calls, threads=None):
     """Create an ascii summary of the trace"""
 
     with open(path, "w") as hist_file:
+        if threads:
+            print(f"Number of threads: {threads}\n", file=hist_file)
         print("Total function calls:", sum(calls.values()), file=hist_file)
         for func, func_calls in calls.items():
             print(func.name, func_calls, file=hist_file)