From bb0b4049f7864c75494532500c32a6a41e052f98 Mon Sep 17 00:00:00 2001
From: Christophe Bedard <fixed-term.christophe.bourquebedard@de.bosch.com>
Date: Fri, 2 Aug 2019 10:36:49 +0200
Subject: [PATCH] Include parent function name in profiling data

---
 tracetools_analysis/tracetools_analysis/data_model/profile.py | 3 +++
 tracetools_analysis/tracetools_analysis/processor/profile.py  | 2 ++
 2 files changed, 5 insertions(+)

diff --git a/tracetools_analysis/tracetools_analysis/data_model/profile.py b/tracetools_analysis/tracetools_analysis/data_model/profile.py
index 035c6e0..1037edf 100644
--- a/tracetools_analysis/tracetools_analysis/data_model/profile.py
+++ b/tracetools_analysis/tracetools_analysis/data_model/profile.py
@@ -29,6 +29,7 @@ class ProfileDataModel(DataModel):
             'tid',
             'depth',
             'function_name',
+            'parent_name',
             'start_timestamp',
             'duration',
         ])
@@ -38,6 +39,7 @@ class ProfileDataModel(DataModel):
         tid: int,
         depth: int,
         function_name: str,
+        parent_name: str,
         start_timestamp: int,
         duration: int,
     ) -> None:
@@ -45,6 +47,7 @@ class ProfileDataModel(DataModel):
             'tid': tid,
             'depth': depth,
             'function_name': function_name,
+            'parent_name': parent_name,
             'start_timestamp': start_timestamp,
             'duration': duration,
         }
diff --git a/tracetools_analysis/tracetools_analysis/processor/profile.py b/tracetools_analysis/tracetools_analysis/processor/profile.py
index eee4db5..e9ce6d4 100644
--- a/tracetools_analysis/tracetools_analysis/processor/profile.py
+++ b/tracetools_analysis/tracetools_analysis/processor/profile.py
@@ -99,11 +99,13 @@ class ProfileHandler(EventHandler):
         function_depth = len(tid_functions) - 1
         (start_timestamp, start_function_name) = tid_functions.pop()
         # Add to data model
+        parent_name = tid_functions[-1][1] if function_depth > 0 else None
         duration = metadata.timestamp - start_timestamp
         self._data.add_duration(
             tid,
             function_depth,
             start_function_name,
+            parent_name,
             start_timestamp,
             duration
         )