trace-graph: Add trace_graph_plot_insert() and remove()

Add a way to insert a plot in between other plots.
Also add the removal of a plot.

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

2 files changed, 79 insertions, 3 deletions

diff --git a/trace-graph.h b/trace-graph.h
index 22ae29b..3344f87 100644
--- a/trace-graph.h
+++ b/trace-graph.h
@@ -224,6 +224,11 @@ void trace_graph_plot_init(struct graph_info *ginfo);
 void trace_graph_plot_append(struct graph_info *ginfo,
                              const char *label, const struct plot_callbacks *cb,
                              void *data);
+void trace_graph_plot_insert(struct graph_info *ginfo,
+                             int pos,
+                             const char *label, const struct plot_callbacks *cb,
+                             void *data);
+void trace_graph_plot_remove(struct graph_info *ginfo, int pos);
 
 int trace_graph_plot_match_time(struct graph_info *ginfo,
                                 struct graph_plot *plot,
diff --git a/trace-plot.c b/trace-plot.c
index a158745..e684621 100644
--- a/trace-plot.c
+++ b/trace-plot.c
@@ -26,9 +26,10 @@ void trace_graph_plot_init(struct graph_info *ginfo)
         ginfo->plot_list = NULL;
 }
 
-void trace_graph_plot_append(struct graph_info *ginfo,
-                             const char *label, const struct plot_callbacks *cb,
-                             void *data)
+static struct graph_plot *
+allocate_plot(struct graph_info *ginfo,
+              const char *label, const struct plot_callbacks *cb,
+              void *data)
 {
         struct graph_plot *plot;
         char *name;
@@ -47,6 +48,17 @@ void trace_graph_plot_append(struct graph_info *ginfo,
         plot->next = ginfo->plot_list;
         ginfo->plot_list = plot;
 
+        return plot;
+}
+
+void trace_graph_plot_append(struct graph_info *ginfo,
+                             const char *label, const struct plot_callbacks *cb,
+                             void *data)
+{
+        struct graph_plot *plot;
+
+        plot = allocate_plot(ginfo, label, cb, data);
+
         if (!ginfo->plots) {
                 ginfo->plot_array = malloc_or_die(sizeof(ginfo->plot_array[0]));
                 ginfo->plot_array[0] = plot;
@@ -64,6 +76,65 @@ void trace_graph_plot_append(struct graph_info *ginfo,
         ginfo->plots++;
 }
 
+void trace_graph_plot_insert(struct graph_info *ginfo,
+                             int pos,
+                             const char *label, const struct plot_callbacks *cb,
+                             void *data)
+{
+        struct graph_plot *plot;
+
+        if (pos >= ginfo->plots)
+                return trace_graph_plot_append(ginfo, label, cb, data);
+
+        if (pos < 0)
+                pos = 0;
+
+        plot = allocate_plot(ginfo, label, cb, data);
+
+        ginfo->plot_array = realloc(ginfo->plot_array,
+                                    sizeof(ginfo->plot_array[0]) *
+                                    (ginfo->plots + 1));
+
+        if (!ginfo->plot_array)
+                die("unable to resize plot array");
+
+        memmove(&ginfo->plot_array[pos+1], &ginfo->plot_array[pos],
+                sizeof(ginfo->plot_array[0]) * (ginfo->plots - pos));
+
+        ginfo->plot_array[pos] = plot;
+
+        ginfo->plots++;
+}
+
+void trace_graph_plot_remove(struct graph_info *ginfo, int pos)
+{
+        struct graph_plot **pplot;
+
+        if (pos < 0 || pos >= ginfo->plots || !ginfo->plots)
+                return;
+
+        for (pplot = &ginfo->plot_list; *pplot; pplot = &((*pplot)->next)) {
+
+                if (*pplot != ginfo->plot_array[pos])
+                        continue;
+
+                *pplot = (*pplot)->next;
+                break;
+        }
+
+        free(ginfo->plot_array[pos]);
+
+        ginfo->plots--;
+
+        if (ginfo->plots) {
+                memmove(&ginfo->plot_array[pos], &ginfo->plot_array[pos+1],
+                        sizeof(ginfo->plot_array[0]) * (ginfo->plots - pos));
+        } else {
+                free(ginfo->plot_array);
+                ginfo->plot_array = NULL;
+        }
+}
+
 int trace_graph_plot_match_time(struct graph_info *ginfo,
                                 struct graph_plot *plot,
                                 unsigned long long time)