lib: add "borrow" functions where "get" functions exist

[babeltrace.git] / include / babeltrace / graph / notification-packet.h

diff --git a/include/babeltrace/graph/notification-packet.h b/include/babeltrace/graph/notification-packet.h
index 3a98091018c32c7f66c79443a355ec120b270959..6b237c3b2e93fa1eef2083022212652198e6d334 100644 (file)
--- a/include/babeltrace/graph/notification-packet.h
+++ b/include/babeltrace/graph/notification-packet.h
@@ -27,27 +27,42 @@
  * SOFTWARE.
  */
 
+/* For bt_get() */
+#include <babeltrace/ref.h>
+
 #ifdef __cplusplus
 extern "C" {
 #endif
 
 struct bt_notification;
-struct bt_ctf_packet;
+struct bt_packet;
 
 extern struct bt_notification *bt_notification_packet_begin_create(
-               struct bt_ctf_packet *packet);
+               struct bt_packet *packet);
 
 extern struct bt_notification *bt_notification_packet_end_create(
-               struct bt_ctf_packet *packet);
+               struct bt_packet *packet);
 
-/*** BT_NOTIFICATION_TYPE_PACKET_BEGIN ***/
-extern struct bt_ctf_packet *bt_notification_packet_begin_get_packet(
+extern struct bt_packet *bt_notification_packet_begin_borrow_packet(
                struct bt_notification *notification);
 
-/*** BT_NOTIFICATION_TYPE_PACKET_END ***/
-extern struct bt_ctf_packet *bt_notification_packet_end_get_packet(
+static inline
+struct bt_packet *bt_notification_packet_begin_get_packet(
+               struct bt_notification *notification)
+{
+       return bt_get(bt_notification_packet_begin_borrow_packet(notification));
+}
+
+extern struct bt_packet *bt_notification_packet_end_borrow_packet(
                struct bt_notification *notification);
 
+static inline
+struct bt_packet *bt_notification_packet_end_get_packet(
+               struct bt_notification *notification)
+{
+       return bt_get(bt_notification_packet_end_borrow_packet(notification));
+}
+
 #ifdef __cplusplus
 }
 #endif