ir: add bt_ctf_event_get_clock_value()

[babeltrace.git] / formats / ctf / ir / event.c

diff --git a/formats/ctf/ir/event.c b/formats/ctf/ir/event.c
index d18b1d7a0582750b4c386febca5ed9ea90e6dd68..63769c935b4895279383a5ba236a95beef4c6620 100644 (file)
--- a/formats/ctf/ir/event.c
+++ b/formats/ctf/ir/event.c
@@ -33,6 +33,9 @@
 #include <babeltrace/ctf-ir/event-class-internal.h>
 #include <babeltrace/ctf-ir/stream-class.h>
 #include <babeltrace/ctf-ir/stream-class-internal.h>
+#include <babeltrace/ctf-ir/stream-internal.h>
+#include <babeltrace/ctf-ir/packet.h>
+#include <babeltrace/ctf-ir/packet-internal.h>
 #include <babeltrace/ctf-ir/trace-internal.h>
 #include <babeltrace/ctf-ir/validation-internal.h>
 #include <babeltrace/ctf-ir/packet-internal.h>
@@ -152,6 +155,8 @@ struct bt_ctf_event *bt_ctf_event_create(struct bt_ctf_event_class *event_class)
         * lifetime.
         */
        event->event_class = bt_get(event_class);
+       event->clock_values = g_hash_table_new_full(g_direct_hash,
+               g_direct_equal, NULL, g_free);
        event_header =
                bt_ctf_field_create(validation_output.event_header_type);
        if (!event_header) {
@@ -296,7 +301,7 @@ int bt_ctf_event_set_payload(struct bt_ctf_event *event,
 {
        int ret = 0;
 
-       if (!event || !payload) {
+       if (!event || !payload || event->frozen) {
                ret = -1;
                goto end;
        }
@@ -344,7 +349,7 @@ int bt_ctf_event_set_payload_field(struct bt_ctf_event *event,
        int ret = 0;
        struct bt_ctf_field_type *payload_type = NULL;
 
-       if (!event || !payload) {
+       if (!event || !payload || event->frozen) {
                ret = -1;
                goto end;
        }
@@ -427,7 +432,7 @@ int bt_ctf_event_set_header(struct bt_ctf_event *event,
        struct bt_ctf_field_type *field_type = NULL;
        struct bt_ctf_stream_class *stream_class = NULL;
 
-       if (!event || !header) {
+       if (!event || !header || event->frozen) {
                ret = -1;
                goto end;
        }
@@ -475,7 +480,7 @@ int bt_ctf_event_set_event_context(struct bt_ctf_event *event,
        int ret = 0;
        struct bt_ctf_field_type *field_type = NULL;
 
-       if (!event || !context) {
+       if (!event || !context || event->frozen) {
                ret = -1;
                goto end;
        }
@@ -516,7 +521,7 @@ int bt_ctf_event_set_stream_event_context(struct bt_ctf_event *event,
        struct bt_ctf_field_type *field_type = NULL;
        struct bt_ctf_stream_class *stream_class = NULL;
 
-       if (!event || !stream_event_context) {
+       if (!event || !stream_event_context || event->frozen) {
                ret = -1;
                goto end;
        }
@@ -566,6 +571,7 @@ void bt_ctf_event_destroy(struct bt_object *obj)
                 */
                bt_put(event->event_class);
        }
+       g_hash_table_destroy(event->clock_values);
        bt_put(event->event_header);
        bt_put(event->stream_event_context);
        bt_put(event->context_payload);
@@ -688,7 +694,7 @@ int bt_ctf_event_populate_event_header(struct bt_ctf_event *event)
        int ret = 0;
        struct bt_ctf_field *id_field = NULL, *timestamp_field = NULL;
 
-       if (!event) {
+       if (!event || event->frozen) {
                ret = -1;
                goto end;
        }
@@ -736,66 +742,13 @@ end:
        return ret;
 }
 
-struct bt_ctf_event *bt_ctf_event_copy(struct bt_ctf_event *event)
-{
-       struct bt_ctf_event *copy = NULL;
-
-       if (!event) {
-               goto error;
-       }
-
-       copy = g_new0(struct bt_ctf_event, 1);
-       if (!copy) {
-               goto error;
-       }
-
-       bt_object_init(copy, bt_ctf_event_destroy);
-       copy->event_class = bt_get(event->event_class);
-
-       if (event->event_header) {
-               copy->event_header = bt_ctf_field_copy(event->event_header);
-               if (!copy->event_header) {
-                       goto error;
-               }
-       }
-
-       if (event->stream_event_context) {
-               copy->stream_event_context =
-                       bt_ctf_field_copy(event->stream_event_context);
-               if (!copy->stream_event_context) {
-                       goto error;
-               }
-       }
-
-       if (event->context_payload) {
-               copy->context_payload = bt_ctf_field_copy(
-                       event->context_payload);
-               if (!copy->context_payload) {
-                       goto error;
-               }
-       }
-
-       if (event->fields_payload) {
-               copy->fields_payload = bt_ctf_field_copy(event->fields_payload);
-               if (!copy->fields_payload) {
-                       goto error;
-               }
-       }
-
-       return copy;
-
-error:
-       BT_PUT(copy);
-       return copy;
-}
-
 int bt_ctf_event_set_packet(struct bt_ctf_event *event,
                struct bt_ctf_packet *packet)
 {
        struct bt_ctf_stream *stream = NULL;
        int ret = 0;
 
-       if (!event || !packet) {
+       if (!event || !packet || event->frozen) {
                ret = -1;
                goto end;
        }
@@ -823,3 +776,70 @@ end:
 
        return ret;
 }
+
+BT_HIDDEN
+void bt_ctf_event_freeze(struct bt_ctf_event *event)
+{
+       assert(event);
+       bt_ctf_packet_freeze(event->packet);
+       bt_ctf_field_freeze(event->event_header);
+       bt_ctf_field_freeze(event->stream_event_context);
+       bt_ctf_field_freeze(event->context_payload);
+       bt_ctf_field_freeze(event->fields_payload);
+       event->frozen = 1;
+}
+
+static
+void insert_stream_clock_value_into_event_clock_values(gpointer key,
+               gpointer value,
+               gpointer data)
+{
+       struct bt_ctf_event *event = data;
+       uint64_t *clock_value;
+
+       assert(event);
+
+       /* Copy clock value because it belongs to the hash table */
+       clock_value = g_new0(uint64_t, 1);
+       *clock_value = *((uint64_t *) value);
+
+       /* Insert copy into event clock values */
+       g_hash_table_insert(event->clock_values, key, clock_value);
+}
+
+BT_HIDDEN
+int bt_ctf_event_register_stream_clock_values(struct bt_ctf_event *event)
+{
+       int ret = 0;
+       struct bt_ctf_stream *stream;
+
+       stream = bt_ctf_event_get_stream(event);
+       assert(stream);
+       g_hash_table_remove_all(event->clock_values);
+       g_hash_table_foreach(stream->clock_values,
+               insert_stream_clock_value_into_event_clock_values, event);
+       BT_PUT(stream);
+
+       return ret;
+}
+
+uint64_t bt_ctf_event_get_clock_value(struct bt_ctf_event *event,
+               struct bt_ctf_clock *clock)
+{
+       uint64_t ret = -1ULL;
+       uint64_t *clock_value;
+
+       if (!event || !clock) {
+               goto end;
+       }
+
+       clock_value = g_hash_table_lookup(event->clock_values, clock);
+       if (!clock_value) {
+               goto end;
+       }
+
+       ret = *clock_value;
+
+end:
+       return ret;
+}