X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=lib%2Fgraph%2Fnotification%2Fstream.c;h=f82761952171bd40f968662fc5fc138a4d113ae7;hb=e22b45d0f7d3ce1311bf96a930bc42326f555202;hp=a1ce7a5aa995f1ebf98ead1048b0efe2e4b97c18;hpb=bf55043c2e742cafb86d3a3404d0d35c4cf294a3;p=babeltrace.git

diff --git a/lib/graph/notification/stream.c b/lib/graph/notification/stream.c
index a1ce7a5a..f8276195 100644
--- a/lib/graph/notification/stream.c
+++ b/lib/graph/notification/stream.c
@@ -27,12 +27,12 @@
 #define BT_LOG_TAG "NOTIF-STREAM"
 #include <babeltrace/lib-logging-internal.h>
 
+#include <babeltrace/assert-pre-internal.h>
 #include <babeltrace/compiler-internal.h>
 #include <babeltrace/ctf-ir/stream-internal.h>
 #include <babeltrace/ctf-ir/stream-class.h>
 #include <babeltrace/graph/notification-stream-internal.h>
 #include <babeltrace/assert-internal.h>
-#include <babeltrace/assert-pre-internal.h>
 #include <inttypes.h>
 
 static
@@ -45,6 +45,7 @@ void bt_notification_stream_end_destroy(struct bt_object *obj)
 		notification);
 	BT_LOGD_STR("Putting stream.");
 	BT_PUT(notification->stream);
+	bt_clock_value_set_finalize(&notification->cv_set);
 	g_free(notification);
 }
 
@@ -53,6 +54,7 @@ struct bt_notification *bt_notification_stream_end_create(
 {
 	struct bt_notification_stream_end *notification;
 	struct bt_stream_class *stream_class;
+	int ret;
 
 	BT_ASSERT_PRE_NON_NULL(stream, "Stream");
 	stream_class = bt_stream_borrow_class(stream);
@@ -75,6 +77,11 @@ struct bt_notification *bt_notification_stream_end_create(
 			BT_NOTIFICATION_TYPE_STREAM_END,
 			bt_notification_stream_end_destroy, NULL);
 	notification->stream = bt_get(stream);
+	ret = bt_clock_value_set_initialize(&notification->cv_set);
+	if (ret) {
+		goto error;
+	}
+
 	BT_LOGD("Created stream end notification object: "
 		"stream-addr=%p, stream-name=\"%s\", "
 		"stream-class-addr=%p, stream-class-name=\"%s\", "
@@ -101,6 +108,38 @@ struct bt_stream *bt_notification_stream_end_borrow_stream(
 	return stream_end->stream;
 }
 
+int bt_notification_stream_end_set_clock_value(struct bt_notification *notif,
+		struct bt_clock_class *clock_class, uint64_t raw_value,
+		bt_bool is_default)
+{
+	struct bt_notification_stream_end *stream_end = (void *) notif;
+
+	BT_ASSERT_PRE_NON_NULL(notif, "Notification");
+	BT_ASSERT_PRE_NON_NULL(clock_class, "Clock class");
+	BT_ASSERT_PRE_HOT(notif, "Notification", ": %!+n", notif);
+	BT_ASSERT_PRE_NOTIF_IS_TYPE(notif, BT_NOTIFICATION_TYPE_STREAM_END);
+	BT_ASSERT_PRE(is_default,
+		"You can only set a default clock value as of this version.");
+	return bt_clock_value_set_set_clock_value(&stream_end->cv_set,
+		clock_class, raw_value, is_default);
+}
+
+struct bt_clock_value *bt_notification_stream_end_borrow_default_clock_value(
+		struct bt_notification *notif)
+{
+	struct bt_notification_stream_end *stream_end = (void *) notif;
+	struct bt_clock_value *clock_value = NULL;
+
+	BT_ASSERT_PRE_NON_NULL(notif, "Notification");
+	BT_ASSERT_PRE_NOTIF_IS_TYPE(notif, BT_NOTIFICATION_TYPE_STREAM_END);
+	clock_value = stream_end->cv_set.default_cv;
+	if (!clock_value) {
+		BT_LIB_LOGV("No default clock value: %![notif-]+n", notif);
+	}
+
+	return clock_value;
+}
+
 static
 void bt_notification_stream_begin_destroy(struct bt_object *obj)
 {
@@ -111,12 +150,14 @@ void bt_notification_stream_begin_destroy(struct bt_object *obj)
 		notification);
 	BT_LOGD_STR("Putting stream.");
 	BT_PUT(notification->stream);
+	bt_clock_value_set_finalize(&notification->cv_set);
 	g_free(notification);
 }
 
 struct bt_notification *bt_notification_stream_begin_create(
 		struct bt_graph *graph, struct bt_stream *stream)
 {
+	int ret;
 	struct bt_notification_stream_begin *notification;
 	struct bt_stream_class *stream_class;
 
@@ -141,6 +182,11 @@ struct bt_notification *bt_notification_stream_begin_create(
 			BT_NOTIFICATION_TYPE_STREAM_BEGIN,
 			bt_notification_stream_begin_destroy, NULL);
 	notification->stream = bt_get(stream);
+	ret = bt_clock_value_set_initialize(&notification->cv_set);
+	if (ret) {
+		goto error;
+	}
+
 	BT_LOGD("Created stream beginning notification object: "
 		"stream-addr=%p, stream-name=\"%s\", "
 		"stream-class-addr=%p, stream-class-name=\"%s\", "
@@ -166,3 +212,35 @@ struct bt_stream *bt_notification_stream_begin_borrow_stream(
 			struct bt_notification_stream_begin, parent);
 	return stream_begin->stream;
 }
+
+int bt_notification_stream_begin_set_clock_value(struct bt_notification *notif,
+		struct bt_clock_class *clock_class, uint64_t raw_value,
+		bt_bool is_default)
+{
+	struct bt_notification_stream_begin *stream_begin = (void *) notif;
+
+	BT_ASSERT_PRE_NON_NULL(notif, "Notification");
+	BT_ASSERT_PRE_NON_NULL(clock_class, "Clock class");
+	BT_ASSERT_PRE_HOT(notif, "Notification", ": %!+n", notif);
+	BT_ASSERT_PRE_NOTIF_IS_TYPE(notif, BT_NOTIFICATION_TYPE_STREAM_BEGIN);
+	BT_ASSERT_PRE(is_default,
+		"You can only set a default clock value as of this version.");
+	return bt_clock_value_set_set_clock_value(&stream_begin->cv_set,
+		clock_class, raw_value, is_default);
+}
+
+struct bt_clock_value *bt_notification_stream_begin_borrow_default_clock_value(
+		struct bt_notification *notif)
+{
+	struct bt_notification_stream_begin *stream_begin = (void *) notif;
+	struct bt_clock_value *clock_value = NULL;
+
+	BT_ASSERT_PRE_NON_NULL(notif, "Notification");
+	BT_ASSERT_PRE_NOTIF_IS_TYPE(notif, BT_NOTIFICATION_TYPE_STREAM_BEGIN);
+	clock_value = stream_begin->cv_set.default_cv;
+	if (!clock_value) {
+		BT_LIB_LOGV("No default clock value: %![notif-]+n", notif);
+	}
+
+	return clock_value;
+}