* SOFTWARE.
*/
-#include <babeltrace/graph/notification.h>
-#include <babeltrace/ctf-ir/stream.h>
+/* For bt_get() */
+#include <babeltrace/ref.h>
#ifdef __cplusplus
extern "C" {
#endif
-extern struct bt_notification *bt_notification_stream_begin_create(
- struct bt_ctf_stream *stream);
+struct bt_notification;
+struct bt_stream;
-extern struct bt_notification *bt_notification_stream_end_create(
- struct bt_ctf_stream *stream);
+extern
+struct bt_notification *bt_notification_stream_begin_create(
+ struct bt_graph *graph,
+ struct bt_stream *stream);
-extern struct bt_ctf_stream *bt_notification_stream_begin_get_stream(
+extern
+struct bt_notification *bt_notification_stream_end_create(
+ struct bt_graph *graph,
+ struct bt_stream *stream);
+
+extern struct bt_stream *bt_notification_stream_begin_borrow_stream(
struct bt_notification *notification);
-extern struct bt_ctf_stream *bt_notification_stream_end_get_stream(
+static inline
+struct bt_stream *bt_notification_stream_begin_get_stream(
+ struct bt_notification *notification)
+{
+ return bt_get(bt_notification_stream_begin_borrow_stream(notification));
+}
+
+extern struct bt_stream *bt_notification_stream_end_borrow_stream(
struct bt_notification *notification);
+static inline
+struct bt_stream *bt_notification_stream_end_get_stream(
+ struct bt_notification *notification)
+{
+ return bt_get(bt_notification_stream_end_borrow_stream(notification));
+}
+
#ifdef __cplusplus
}
#endif