trace-graph: Show running sched out latencies

Add a empty red box when scheduled out, and make wakeup latencies
green.

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

3 files changed, 31 insertions, 6 deletions

diff --git a/trace-graph.c b/trace-graph.c
index cdf84e8..7f982bd 100644
--- a/trace-graph.c
+++ b/trace-graph.c
@@ -893,6 +893,7 @@ int trace_graph_check_sched_switch(struct graph_info *ginfo,
                         return 0;
 
                 ginfo->event_sched_switch_id = event->id;
+                ginfo->event_prev_state = pevent_find_field(event, "prev_state");
                 ginfo->event_pid_field = pevent_find_field(event, "next_pid");
                 ginfo->event_comm_field = pevent_find_field(event, "next_comm");
 
diff --git a/trace-graph.h b/trace-graph.h
index cb8286c..c501353 100644
--- a/trace-graph.h
+++ b/trace-graph.h
@@ -206,6 +206,7 @@ struct graph_info {
         gint                    event_sched_switch_id;
         gint                    event_wakeup_id;
         gint                    event_wakeup_new_id;
+        struct format_field     *event_prev_state;
         struct format_field     *event_pid_field;
         struct format_field     *event_comm_field;
         struct format_field     *ftrace_pid_field;
diff --git a/trace-plot-task.c b/trace-plot-task.c
index 2aa03d4..0c9f8e0 100644
--- a/trace-plot-task.c
+++ b/trace-plot-task.c
@@ -24,6 +24,7 @@
 #include "trace-filter.h"
 
 #define RED 0xff
+#define GREEN (0xff<<16)
 
 struct task_plot_info {
         int                     pid;
@@ -31,6 +32,7 @@ struct task_plot_info {
         unsigned long long      last_time;
         unsigned long long      wake_time;
         unsigned long long      display_wake_time;
+        int                     wake_color;
         int                     last_cpu;
 };
 
@@ -55,6 +57,19 @@ static int hash_cpu(int cpu)
         return trace_hash(cpu + 124);
 }
 
+static gboolean is_running(struct graph_info *ginfo, struct record *record)
+{
+        unsigned long long val;
+        int id;
+
+        id = pevent_data_type(ginfo->pevent, record);
+        if (id != ginfo->event_sched_switch_id)
+                return FALSE;
+
+        pevent_read_number_field(ginfo->event_prev_state, record->data, &val);
+        return val ? FALSE : TRUE;
+}
+
 static gboolean record_matches_pid(struct graph_info *ginfo,
                                    struct record *record, int match_pid,
                                    int *pid, int *sched_pid,
@@ -357,6 +372,7 @@ static int task_plot_event(struct graph_info *ginfo,
                         }
 
                         task_info->wake_time = record->ts;
+                        task_info->wake_color = GREEN;
                         task_info->display_wake_time = record->ts;
 
                         return 1;
@@ -384,8 +400,10 @@ static int task_plot_event(struct graph_info *ginfo,
                                         info->bfill = FALSE;
                                         info->bstart = task_info->wake_time;
                                         info->bend = record->ts;
-                                        info->bcolor = RED;
-                                }
+                                        info->bcolor = task_info->wake_color;
+                                } else
+                                        task_info->wake_time = 0;
+
                         } else if (!info->box) {
                                 /* just got scheduled out */
                                 info->box = TRUE;
@@ -393,10 +411,15 @@ static int task_plot_event(struct graph_info *ginfo,
                                 info->bstart = task_info->last_time;
                                 info->bend = record->ts;
                                 task_info->last_cpu = -1;
-                        }
-                }
-
-                task_info->wake_time = 0;
+                                if (is_running(ginfo, record)) {
+                                        task_info->wake_time = record->ts;
+                                        task_info->wake_color = RED;
+                                } else
+                                        task_info->wake_time = 0;
+                        } else
+                                task_info->wake_time = 0;
+                } else
+                        task_info->wake_time = 0;
 
                 return 1;
         }