include/babeltrace/graph: fix some include guards

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

diff --git a/include/babeltrace/graph/notification-event-internal.h b/include/babeltrace/graph/notification-event-internal.h
index 939b77111f610641080704546f2f67a21079fe72..62b8a3ab727ffd5c92c3a9f2597b4463886d2a04 100644 (file)
--- a/include/babeltrace/graph/notification-event-internal.h
+++ b/include/babeltrace/graph/notification-event-internal.h
@@ -1,5 +1,5 @@
-#ifndef BABELTRACE_COMPONENT_NOTIFICATION_EVENT_INTERNAL_H
-#define BABELTRACE_COMPONENT_NOTIFICATION_EVENT_INTERNAL_H
+#ifndef BABELTRACE_GRAPH_NOTIFICATION_EVENT_INTERNAL_H
+#define BABELTRACE_GRAPH_NOTIFICATION_EVENT_INTERNAL_H
 
 /*
  * BabelTrace - Plug-in Event Notification internal
@@ -27,8 +27,11 @@
  * SOFTWARE.
  */
 
+#include <babeltrace/compiler-internal.h>
 #include <babeltrace/ctf-ir/event.h>
 #include <babeltrace/graph/notification-internal.h>
+#include <babeltrace/graph/clock-class-priority-map.h>
+#include <babeltrace/assert-internal.h>
 
 #ifdef __cplusplus
 extern "C" {
@@ -36,11 +39,36 @@ extern "C" {
 
 struct bt_notification_event {
        struct bt_notification parent;
-       struct bt_ctf_event *event;
+       struct bt_event *event;
+       struct bt_clock_class_priority_map *cc_prio_map;
 };
 
+static inline
+struct bt_event *bt_notification_event_borrow_event(
+               struct bt_notification *notif)
+{
+       struct bt_notification_event *notif_event = container_of(notif,
+                       struct bt_notification_event, parent);
+
+       BT_ASSERT(notif_event);
+       return notif_event->event;
+}
+
+static inline
+struct bt_clock_class_priority_map *
+bt_notification_event_borrow_clock_class_priority_map(
+               struct bt_notification *notif)
+{
+       struct bt_notification_event *notif_event = container_of(notif,
+                       struct bt_notification_event, parent);
+
+       BT_ASSERT(notif_event);
+       return notif_event->cc_prio_map;
+}
+
+
 #ifdef __cplusplus
 }
 #endif
 
-#endif /* BABELTRACECOMPONENTPLUGIN_NOTIFICATION_EVENT_INTERNAL_H */
+#endif /* BABELTRACE_GRAPH_NOTIFICATION_EVENT_INTERNAL_H */