bt2: by-pass distutils changed file check

[babeltrace.git] / include / babeltrace / graph / notification-stream-internal.h

diff --git a/include/babeltrace/graph/notification-stream-internal.h b/include/babeltrace/graph/notification-stream-internal.h
index 839c220ccaa445773150d33b20d61697bb53e7c6..5c0f72260c3c50e1ce2d8bfd07c2ec38ad8e3eab 100644 (file)
--- a/include/babeltrace/graph/notification-stream-internal.h
+++ b/include/babeltrace/graph/notification-stream-internal.h
@@ -27,12 +27,43 @@
  * SOFTWARE.
  */
 
+#include <babeltrace/compiler-internal.h>
 #include <babeltrace/ctf-ir/packet.h>
 #include <babeltrace/graph/notification-internal.h>
+#include <babeltrace/assert-internal.h>
+
+struct bt_notification_stream_begin {
+       struct bt_notification parent;
+       struct bt_stream *stream;
+};
 
 struct bt_notification_stream_end {
        struct bt_notification parent;
-       struct bt_ctf_stream *stream;
+       struct bt_stream *stream;
 };
 
+static inline
+struct bt_stream *bt_notification_stream_begin_borrow_stream(
+               struct bt_notification *notif)
+{
+       struct bt_notification_stream_begin *notif_stream_begin =
+               container_of(notif,
+                       struct bt_notification_stream_begin, parent);
+
+       BT_ASSERT(notif_stream_begin);
+       return notif_stream_begin->stream;
+}
+
+static inline
+struct bt_stream *bt_notification_stream_end_borrow_stream(
+               struct bt_notification *notif)
+{
+       struct bt_notification_stream_end *notif_stream_end =
+               container_of(notif,
+                       struct bt_notification_stream_end, parent);
+
+       BT_ASSERT(notif_stream_end);
+       return notif_stream_end->stream;
+}
+
 #endif /* BABELTRACE_COMPONENT_NOTIFICATION_STREAM_INTERNAL_H */