trace-graph: Use ref_count instead of comparisons

In find_previous_record in the trace-plot, instead of checking if
the last_record is equal to the start_record, simply up the ref count.
Then the free will not free the start_record.

This makes the code a bit cleaner.

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

1 files changed, 5 insertions, 5 deletions

diff --git a/trace-plot-task.c b/trace-plot-task.c
index 0c9f8e0..c8d0905 100644
--- a/trace-plot-task.c
+++ b/trace-plot-task.c
@@ -464,7 +464,9 @@ find_previous_record(struct graph_info *ginfo, struct record *start_record,
         gint sched_pid;
         int count = 0;
 
-        if (!last_record)
+        if (last_record)
+                last_record->ref_count++;
+        else
                 last_record = tracecmd_read_cpu_last(ginfo->handle, cpu);
 
         while ((record = tracecmd_read_prev(ginfo->handle, last_record))) {
@@ -475,8 +477,7 @@ find_previous_record(struct graph_info *ginfo, struct record *start_record,
                 if (match)
                         break;
 
-                if (last_record != start_record)
-                        free_record(last_record);
+                free_record(last_record);
 
                 if (count > MAX_SEARCH) {
                         free_record(record);
@@ -485,8 +486,7 @@ find_previous_record(struct graph_info *ginfo, struct record *start_record,
                 last_record = record;
         }
 
-        if (last_record != start_record)
-                free_record(last_record);
+        free_record(last_record);
 
         return record;
 }