Move trace collection pointer to bt_trace_descriptor

[babeltrace.git] / lib / trace-collection.c

diff --git a/lib/trace-collection.c b/lib/trace-collection.c
index fc06dcd961c36aa3af3d723dae8fd85d20dfbfda..6743ce6ba7467ef8c21daeea02d81b8073e9b632 100644 (file)
--- a/lib/trace-collection.c
+++ b/lib/trace-collection.c
@@ -156,7 +156,7 @@ static void clock_add(gpointer key, gpointer value, gpointer user_data)
  * convert the index from cycles to real time.
  */
 int bt_trace_collection_add(struct trace_collection *tc,
-                               struct trace_descriptor *td)
+                               struct bt_trace_descriptor *td)
 {
        struct ctf_trace *trace;
 
@@ -164,8 +164,6 @@ int bt_trace_collection_add(struct trace_collection *tc,
                return -EINVAL;
 
        trace = container_of(td, struct ctf_trace, parent);
-       g_ptr_array_add(tc->array, td);
-       trace->collection = tc;
 
        if (tc->array->len > 1) {
                struct clock_match clock_match = {
@@ -187,6 +185,9 @@ int bt_trace_collection_add(struct trace_collection *tc,
                }
        }
 
+       g_ptr_array_add(tc->array, td);
+       trace->parent.collection = tc;
+
        {
                struct clock_match clock_match = {
                        .clocks = tc->clocks,
@@ -209,7 +210,7 @@ error:
 }
 
 int bt_trace_collection_remove(struct trace_collection *tc,
-                           struct trace_descriptor *td)
+                           struct bt_trace_descriptor *td)
 {
        if (!tc || !td)
                return -EINVAL;