lib: graph: add "self" and some "private" APIs

[babeltrace.git] / lib / graph / notification / notification.c

diff --git a/lib/graph/notification/notification.c b/lib/graph/notification/notification.c
index 57da6c333c3ae95c80d24babd2eec2bcff210ccf..2b8c12d77ffcbacd2a503af35dbbc654885ab5ab 100644 (file)
--- a/lib/graph/notification/notification.c
+++ b/lib/graph/notification/notification.c
@@ -1,6 +1,4 @@
 /*
- * Babeltrace Plug-in Notification
- *
  * Copyright 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
  *
  * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com>
@@ -27,14 +25,16 @@
 #define BT_LOG_TAG "NOTIF"
 #include <babeltrace/lib-logging-internal.h>
 
+#include <babeltrace/graph/private-notification.h>
 #include <babeltrace/graph/notification-internal.h>
+#include <babeltrace/graph/graph-internal.h>
 #include <babeltrace/assert-internal.h>
 #include <babeltrace/assert-pre-internal.h>
 
 BT_ASSERT_PRE_FUNC
 static inline void _init_seq_num(struct bt_notification *notification)
 {
-       notification->seq_num = -1ULL;
+       notification->seq_num = UINT64_C(-1);
 }
 
 #ifdef BT_DEV_MODE
@@ -46,12 +46,18 @@ static inline void _init_seq_num(struct bt_notification *notification)
 BT_HIDDEN
 void bt_notification_init(struct bt_notification *notification,
                enum bt_notification_type type,
-               bt_object_release_func release)
+               bt_object_release_func release,
+               struct bt_graph *graph)
 {
        BT_ASSERT(type >= 0 && type < BT_NOTIFICATION_TYPE_NR);
        notification->type = type;
        init_seq_num(notification);
-       bt_object_init(&notification->base, release);
+       bt_object_init_shared(&notification->base, release);
+       notification->graph = graph;
+
+       if (graph) {
+               bt_graph_add_notification(graph, notification);
+       }
 }
 
 enum bt_notification_type bt_notification_get_type(
@@ -60,3 +66,10 @@ enum bt_notification_type bt_notification_get_type(
        BT_ASSERT_PRE_NON_NULL(notification, "Notification");
        return notification->type;
 }
+
+BT_HIDDEN
+void bt_notification_unlink_graph(struct bt_notification *notif)
+{
+       BT_ASSERT(notif);
+       notif->graph = NULL;
+}