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

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

diff --git a/include/babeltrace/graph/notification-event.h b/include/babeltrace/graph/notification-event.h
index af2666b8ebd0d83dbeb4328a3479cf5ea50dea00..1df54eb2a6a098bee40b265cf0b94a06b745f524 100644 (file)
--- a/include/babeltrace/graph/notification-event.h
+++ b/include/babeltrace/graph/notification-event.h
@@ -27,6 +27,9 @@
  * SOFTWARE.
  */
 
+/* For bt_get() */
+#include <babeltrace/ref.h>
+
 #ifdef __cplusplus
 extern "C" {
 #endif
@@ -35,7 +38,6 @@ struct bt_notification;
 struct bt_event;
 struct bt_clock_class_priority_map;
 
-/***BT_NOTIFICATION_TYPE_EVENT ***/
 /**
  * Create an event notification.
  *
@@ -48,6 +50,9 @@ extern struct bt_notification *bt_notification_event_create(
                struct bt_event *event,
                struct bt_clock_class_priority_map *clock_class_priority_map);
 
+extern struct bt_event *bt_notification_event_borrow_event(
+               struct bt_notification *notification);
+
 /**
  * Get an event notification's event.
  *
@@ -56,13 +61,26 @@ extern struct bt_notification *bt_notification_event_create(
  *
  * @see #bt_event
  */
-extern struct bt_event *bt_notification_event_get_event(
-               struct bt_notification *notification);
+static inline
+struct bt_event *bt_notification_event_get_event(
+               struct bt_notification *notification)
+{
+       return bt_get(bt_notification_event_borrow_event(notification));
+}
 
 extern struct bt_clock_class_priority_map *
-bt_notification_event_get_clock_class_priority_map(
+bt_notification_event_borrow_clock_class_priority_map(
                struct bt_notification *notification);
 
+static inline
+struct bt_clock_class_priority_map *
+bt_notification_event_get_clock_class_priority_map(
+               struct bt_notification *notification)
+{
+       return bt_get(bt_notification_event_borrow_clock_class_priority_map(
+               notification));
+}
+
 #ifdef __cplusplus
 }
 #endif