trace-graph: Add find_record to plots for popup windows

The popup needs a record to use to determin if a task should be
added to the filter or not. Add the method find_record to the
plot callback structure.

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

4 files changed, 66 insertions, 6 deletions

diff --git a/trace-graph.c b/trace-graph.c
index 026d940..67cae96 100644
--- a/trace-graph.c
+++ b/trace-graph.c
@@ -456,13 +456,14 @@ do_pop_up(GtkWidget *widget, GdkEventButton *event, gpointer data)
         static GtkWidget *menu_filter_hide_task;
         static GtkWidget *menu_filter_clear_tasks;
         struct record *record = NULL;
+        struct graph_plot *plot;
         const char *comm;
         guint64 time;
         gchar *text;
         gint pid;
         gint len;
         gint x, y;
-        gint cpu;
+        gint i;
 
         x = event->x;
         y = event->y;
@@ -523,10 +524,11 @@ do_pop_up(GtkWidget *widget, GdkEventButton *event, gpointer data)
 
         time =  convert_x_to_time(ginfo, x);
 
-        for (cpu = 0; cpu < ginfo->cpus; cpu++) {
-                if (y >= (PLOT_TOP(cpu) - PLOT_GIVE) &&
-                    y <= (PLOT_BOTTOM(cpu) + PLOT_GIVE)) {
-                        record = find_record_on_cpu(ginfo, cpu, time);
+        for (i = 0; i < ginfo->plots; i++) {
+                if (y >= (PLOT_TOP(i) - PLOT_GIVE) &&
+                    y <= (PLOT_BOTTOM(i) + PLOT_GIVE)) {
+                        plot = ginfo->plot_array[i];
+                        record = trace_graph_plot_find_record(ginfo, plot, time);
                         break;
                 }
         }
diff --git a/trace-graph.h b/trace-graph.h
index 343a769..7427785 100644
--- a/trace-graph.h
+++ b/trace-graph.h
@@ -46,6 +46,9 @@ struct graph_plot;
  *   a event, the plot may ask to display that event.
  *   The time will be given to find the event, the time may be before
  *   the given event.
+ *
+ * find_record:
+ *   return a tracecmd record for a given time.
  */
 struct plot_callbacks {
         int (*match_time)(struct graph_info *, struct graph_plot *,
@@ -62,6 +65,8 @@ struct plot_callbacks {
         void (*end)(struct graph_info *, struct graph_plot *);
         int (*display_last_event)(struct graph_info *ginfo, struct graph_plot *plot,
                                   struct trace_seq *s, unsigned long long time);
+        struct record *(*find_record)(struct graph_info *, struct graph_plot *,
+                                      unsigned long long time);
 };
 
 struct graph_plot {
@@ -239,6 +244,11 @@ int trace_graph_plot_event(struct graph_info *ginfo,
 void trace_graph_plot_end(struct graph_info *ginfo,
                           struct graph_plot *plot);
 
+struct record *
+trace_graph_plot_find_record(struct graph_info *ginfo,
+                             struct graph_plot *plot,
+                             unsigned long long time);
+
 /* cpu plot */
 void graph_plot_init_cpus(struct graph_info *ginfo, int cpus);
 
diff --git a/trace-plot-cpu.c b/trace-plot-cpu.c
index b1adb91..5d37950 100644
--- a/trace-plot-cpu.c
+++ b/trace-plot-cpu.c
@@ -275,12 +275,49 @@ static int cpu_plot_event(struct graph_info *ginfo,
         return ret;
 }
 
+static struct record *
+find_record_on_cpu(struct graph_info *ginfo, gint cpu, guint64 time)
+{
+        struct record *record = NULL;
+        guint64 offset = 0;
+
+        tracecmd_set_cpu_to_timestamp(ginfo->handle, cpu, time);
+        do {
+                if (record) {
+                        offset = record->offset;
+                        free_record(record);
+                }
+                record = tracecmd_read_data(ginfo->handle, cpu);
+        } while (record && record->ts <= (time - 1 / ginfo->resolution));
+
+        if (record) {
+                if (record->ts > (time + 1 / ginfo->resolution) && offset) {
+                        free_record(record);
+                        record = tracecmd_read_at(ginfo->handle, offset, NULL);
+                }
+        }
+
+        return record;
+}
+
+static struct record *
+cpu_plot_find_record(struct graph_info *ginfo, struct graph_plot *plot,
+                     unsigned long long time)
+{
+        struct cpu_plot_info *cpu_info = plot->private;
+        int cpu;
+
+        cpu = cpu_info->cpu;
+
+        return find_record_on_cpu(ginfo, cpu, time);
+}
 
 static const struct plot_callbacks cpu_plot_cb = {
         .match_time             = cpu_plot_match_time,
         .plot_event             = cpu_plot_event,
         .start                  = cpu_plot_start,
-        .display_last_event     = cpu_plot_display_last_event
+        .display_last_event     = cpu_plot_display_last_event,
+        .find_record            = cpu_plot_find_record,
 };
 
 void graph_plot_init_cpus(struct graph_info *ginfo, int cpus)
diff --git a/trace-plot.c b/trace-plot.c
index 7a3e5dd..0cc8919 100644
--- a/trace-plot.c
+++ b/trace-plot.c
@@ -121,3 +121,14 @@ int trace_graph_plot_display_last_event(struct graph_info *ginfo,
 
         return plot->cb->display_last_event(ginfo, plot, s, time);
 }
+
+struct record *
+trace_graph_plot_find_record(struct graph_info *ginfo,
+                             struct graph_plot *plot,
+                             unsigned long long time)
+{
+        if (!plot->cb->find_record)
+                return 0;
+
+        return plot->cb->find_record(ginfo, plot, time);
+}