X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=include%2Fbabeltrace%2Fgraph%2Fnotification-event.h;h=d6614fb6ad5dec10b0dc196c77e311281094974c;hp=f386abb2845c3c13933408453d29df7043037da5;hb=312c056ae3d374b253fa0cfe5ed576c0b0e5e569;hpb=599faa1c75b3c429745e0d4a983670c3dd205092

diff --git a/include/babeltrace/graph/notification-event.h b/include/babeltrace/graph/notification-event.h
index f386abb2..d6614fb6 100644
--- a/include/babeltrace/graph/notification-event.h
+++ b/include/babeltrace/graph/notification-event.h
@@ -1,5 +1,5 @@
-#ifndef BABELTRACE_COMPONENT_NOTIFICATION_EVENT_H
-#define BABELTRACE_COMPONENT_NOTIFICATION_EVENT_H
+#ifndef BABELTRACE_GRAPH_NOTIFICATION_EVENT_H
+#define BABELTRACE_GRAPH_NOTIFICATION_EVENT_H
 
 /*
  * BabelTrace - Plug-in Event Notification
@@ -27,44 +27,41 @@
  * SOFTWARE.
  */
 
+/* For bt_get() */
+#include <babeltrace/ref.h>
+
 #ifdef __cplusplus
 extern "C" {
 #endif
 
 struct bt_notification;
-struct bt_ctf_event;
+struct bt_event;
+struct bt_event_class;
 struct bt_clock_class_priority_map;
 
-/***BT_NOTIFICATION_TYPE_EVENT ***/
-/**
- * Create an event notification.
- *
- * @param event			The event
- * @returns			An event notification instance
- *
- * @see #bt_notification_type
- */
 extern struct bt_notification *bt_notification_event_create(
-		struct bt_ctf_event *event,
+		struct bt_event_class *event_class,
+		struct bt_packet *packet,
 		struct bt_clock_class_priority_map *clock_class_priority_map);
 
-/**
- * Get an event notification's event.
- *
- * @param notification	Event notification instance
- * @returns		An event instance
- *
- * @see #bt_ctf_event
- */
-extern struct bt_ctf_event *bt_notification_event_get_event(
+extern struct bt_event *bt_notification_event_borrow_event(
 		struct bt_notification *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
 
-#endif /* BABELTRACE_COMPONENT_NOTIFICATION_EVENT_H */
+#endif /* BABELTRACE_GRAPH_NOTIFICATION_EVENT_H */