lib: make packets and packet messages optional, disabled by default

[babeltrace.git] / src / lib / trace-ir / event.h

diff --git a/src/lib/trace-ir/event.h b/src/lib/trace-ir/event.h
index 7b5c2544dc57e17b7a3fa44cbdb9b03f107863a6..71318f95b626b3b8e11b8db83f3e603d2f00d9f9 100644 (file)
--- a/src/lib/trace-ir/event.h
+++ b/src/lib/trace-ir/event.h
@@ -25,8 +25,8 @@
  */
 
 /* Protection: this file uses BT_LIB_LOG*() macros directly */
-#ifndef BABELTRACE_LIB_LOGGING_INTERNAL_H
-# error Please include <lib/lib-logging.h> before including this file.
+#ifndef BT_LIB_LOG_SUPPORTED
+# error Please include "lib/logging.h" before including this file.
 #endif
 
 #include "lib/assert-pre.h"
@@ -56,9 +56,12 @@ struct bt_event {
        /* Owned by this */
        struct bt_event_class *class;
 
-       /* Owned by this */
+       /* Owned by this (can be `NULL`) */
        struct bt_packet *packet;
 
+       /* Owned by this */
+       struct bt_stream *stream;
+
        struct bt_field *common_context_field;
        struct bt_field *specific_context_field;
        struct bt_field *payload_field;
@@ -118,8 +121,13 @@ void bt_event_reset(struct bt_event *event)
        BT_ASSERT(event);
        BT_LIB_LOGD("Resetting event: %!+e", event);
        bt_event_set_is_frozen(event, false);
-       bt_object_put_no_null_check(&event->packet->base);
-       event->packet = NULL;
+       bt_object_put_no_null_check(&event->stream->base);
+       event->stream = NULL;
+
+       if (event->packet) {
+               bt_object_put_no_null_check(&event->packet->base);
+               event->packet = NULL;
+       }
 }
 
 static inline
@@ -170,7 +178,7 @@ void bt_event_set_packet(struct bt_event *event, struct bt_packet *packet)
                event->class) == packet->stream->class,
                "Packet's stream class and event's stream class differ: "
                "%![event-]+e, %![packet-]+a", event, packet);
-
+       BT_ASSERT(event->stream->class->supports_packets);
        BT_ASSERT(!event->packet);
        event->packet = packet;
        bt_object_get_no_null_check_no_parent_check(&event->packet->base);
@@ -178,16 +186,35 @@ void bt_event_set_packet(struct bt_event *event, struct bt_packet *packet)
                event, packet);
 }
 
+static inline
+void bt_event_set_stream(struct bt_event *event, struct bt_stream *stream)
+{
+       BT_ASSERT_PRE_NON_NULL(event, "Event");
+       BT_ASSERT_PRE_NON_NULL(stream, "Stream");
+       BT_ASSERT_PRE_EVENT_HOT(event);
+       BT_ASSERT_PRE(bt_event_class_borrow_stream_class(
+               event->class) == stream->class,
+               "Stream's class and event's stream class differ: "
+               "%![event-]+e, %![stream-]+s", event, stream);
+       BT_ASSERT(!event->stream);
+       event->stream = stream;
+       bt_object_get_no_null_check_no_parent_check(&event->stream->base);
+       BT_LIB_LOGD("Set event's stream: %![event-]+e, %![stream-]+s",
+               event, stream);
+}
+
 static inline
 struct bt_event *bt_event_create(struct bt_event_class *event_class,
-               struct bt_packet *packet)
+               struct bt_packet *packet, struct bt_stream *stream)
 {
        struct bt_event *event = NULL;
 
        BT_ASSERT(event_class);
+       BT_ASSERT(stream);
        event = bt_object_pool_create_object(&event_class->event_pool);
        if (G_UNLIKELY(!event)) {
-               BT_LIB_LOGE("Cannot allocate one event from event class's event pool: "
+               BT_LIB_LOGE_APPEND_CAUSE(
+                       "Cannot allocate one event from event class's event pool: "
                        "%![ec-]+E", event_class);
                goto end;
        }
@@ -197,8 +224,13 @@ struct bt_event *bt_event_create(struct bt_event_class *event_class,
                bt_object_get_no_null_check(&event_class->base);
        }
 
-       BT_ASSERT(packet);
-       bt_event_set_packet(event, packet);
+       bt_event_set_stream(event, stream);
+
+       if (packet) {
+               BT_ASSERT(packet);
+               bt_event_set_packet(event, packet);
+       }
+
        goto end;
 
 end: