trace-graph: Show actual task color at start of graph

If the first event in the graph is a sched_switch, then the color
for the block was showing the new task. Not the previous task.

Signed-off-by: Steven Rostedt <rostedt@goodmis.org>

1 files changed, 6 insertions, 2 deletions

diff --git a/trace-graph.c b/trace-graph.c
index e5b22a8..458f5bf 100644
--- a/trace-graph.c
+++ b/trace-graph.c
@@ -1454,8 +1454,12 @@ static void draw_cpu(struct graph_info *ginfo, gint cpu,
                 if (last_pid != pid) {
 
                         if (last_pid < 0) {
-                                last_pid = pid;
-                                set_color_by_pid(ginfo->draw, gc, pid);
+                                /* if we hit a sched switch, use the original pid */
+                                if (is_sched_switch)
+                                        last_pid = pevent_data_pid(ginfo->pevent, record);
+                                else
+                                        last_pid = pid;
+                                set_color_by_pid(ginfo->draw, gc, last_pid);
                         }
                                 
                         filter = graph_filter_on_task(ginfo, last_pid);