Change behaviour of stream-intersection with multiple traces

[babeltrace.git] / formats / ctf / iterator.c

diff --git a/formats/ctf/iterator.c b/formats/ctf/iterator.c
index e5d32a890b503220a9ecefde2c31ae8464b12030..7ef3e0b1537479941193a26708ef42f605e03563 100644 (file)
--- a/formats/ctf/iterator.c
+++ b/formats/ctf/iterator.c
@@ -63,6 +63,53 @@ struct bt_ctf_iter *bt_ctf_iter_create(struct bt_context *ctx,
        return iter;
 }
 
+struct bt_ctf_iter *bt_ctf_iter_create_intersect(struct bt_context *ctx,
+               struct bt_iter_pos **inter_begin_pos,
+               struct bt_iter_pos **inter_end_pos)
+{
+       int ret;
+       uint64_t begin, end;
+
+       /*
+        * The iterator's range is the union of each trace's intersection of
+        * streams. This means that we determine the "active" region of each
+        * trace (that is the region where all of its streams are active), and
+        * use the TraceCollection to merge all of these active regions.
+        *
+        * This results in a union of the traces' active regions.
+        */
+       ret = ctf_find_tc_stream_packet_intersection_union(ctx, &begin, &end);
+       if (ret == 1) {
+               fprintf(stderr, "[error] No intersection found between trace files.\n");
+               goto error;
+       } else if (ret != 0) {
+               goto error;
+       }
+       *inter_begin_pos = bt_iter_create_time_pos(NULL, begin);
+       if (!(*inter_begin_pos)) {
+               goto error;
+       }
+       *inter_end_pos = bt_iter_create_time_pos(NULL, end);
+       if (!(*inter_end_pos)) {
+               goto error;
+       }
+
+       ret = ctf_tc_set_stream_intersection_mode(ctx);
+       if (ret) {
+               goto error;
+       }
+
+       /*
+        * bt_ctf_iter does not take ownership of begin and end positions,
+        * so we return them to the caller who must still assume their ownership
+        * until the iterator is destroyed.
+        */
+       return bt_ctf_iter_create(ctx, *inter_begin_pos,
+                       *inter_end_pos);
+error:
+       return NULL;
+}
+
 void bt_ctf_iter_destroy(struct bt_ctf_iter *iter)
 {
        struct bt_stream_callbacks *bt_stream_cb;