X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=src%2Flib%2Fgraph%2Fmessage%2Fstream.c;h=a110338aa1b5b42bfe1d0bf1d1816b70d6efa29b;hb=98b15851a941e7342b8bb19e265cdc3a40fabfb8;hp=ff7d1c83730a40c2a78d20837ab80f2f06524fbd;hpb=870631a2db01676b476dbee615aade0a22926bcd;p=babeltrace.git

diff --git a/src/lib/graph/message/stream.c b/src/lib/graph/message/stream.c
index ff7d1c83..a110338a 100644
--- a/src/lib/graph/message/stream.c
+++ b/src/lib/graph/message/stream.c
@@ -45,6 +45,14 @@ void destroy_stream_message(struct bt_object *obj)
 	struct bt_message_stream *message = (void *) obj;
 
 	BT_LIB_LOGD("Destroying stream message: %!+n", message);
+
+	if (message->default_cs) {
+		BT_LIB_LOGD("Putting default clock snapshot: %!+k",
+			message->default_cs);
+		bt_clock_snapshot_destroy(message->default_cs);
+		message->default_cs = NULL;
+	}
+
 	BT_LIB_LOGD("Putting stream: %!+s", message->stream);
 	BT_OBJECT_PUT_REF_AND_RESET(message->stream);
 	g_free(message);
@@ -75,15 +83,30 @@ struct bt_message *create_stream_message(
 	bt_message_init(&message->parent, type,
 		destroy_stream_message, NULL);
 	message->stream = stream;
-	bt_object_get_no_null_check(message->stream);
+	bt_object_get_ref_no_null_check(message->stream);
+
+	if (stream_class->default_clock_class) {
+		message->default_cs = bt_clock_snapshot_create(
+			stream_class->default_clock_class);
+		if (!message->default_cs) {
+			goto error;
+		}
+	}
+
 	BT_LIB_LOGD("Created stream message object: "
 		"%![msg-]+n, %![stream-]+s, %![sc-]+S", message,
 		stream, stream_class);
 
-	return (void *) &message->parent;
+	goto end;
 
 error:
-	return NULL;
+	if (message) {
+		g_free(message);
+		message = NULL;
+	}
+
+end:
+	return &message->parent;
 }
 
 struct bt_message *bt_message_stream_beginning_create(
@@ -107,7 +130,7 @@ struct bt_stream *borrow_stream_message_stream(struct bt_message *message)
 {
 	struct bt_message_stream *stream_msg;
 
-	BT_ASSERT(message);
+	BT_ASSERT_DBG(message);
 	stream_msg = (void *) message;
 	return stream_msg->stream;
 }
@@ -115,16 +138,18 @@ struct bt_stream *borrow_stream_message_stream(struct bt_message *message)
 struct bt_stream *bt_message_stream_beginning_borrow_stream(
 		struct bt_message *message)
 {
-	BT_ASSERT_PRE_NON_NULL(message, "Message");
-	BT_ASSERT_PRE_MSG_IS_TYPE(message, BT_MESSAGE_TYPE_STREAM_BEGINNING);
+	BT_ASSERT_PRE_DEV_NON_NULL(message, "Message");
+	BT_ASSERT_PRE_DEV_MSG_IS_TYPE(message,
+		BT_MESSAGE_TYPE_STREAM_BEGINNING);
 	return borrow_stream_message_stream(message);
 }
 
 struct bt_stream *bt_message_stream_end_borrow_stream(
 		struct bt_message *message)
 {
-	BT_ASSERT_PRE_NON_NULL(message, "Message");
-	BT_ASSERT_PRE_MSG_IS_TYPE(message, BT_MESSAGE_TYPE_STREAM_END);
+	BT_ASSERT_PRE_DEV_NON_NULL(message, "Message");
+	BT_ASSERT_PRE_DEV_MSG_IS_TYPE(message,
+		BT_MESSAGE_TYPE_STREAM_END);
 	return borrow_stream_message_stream(message);
 }
 
@@ -141,3 +166,113 @@ const struct bt_stream *bt_message_stream_end_borrow_stream_const(
 	return bt_message_stream_end_borrow_stream(
 		(void *) message);
 }
+
+static
+void bt_message_stream_set_default_clock_snapshot(
+		struct bt_message *msg, uint64_t raw_value)
+{
+	struct bt_message_stream *stream_msg = (void *) msg;
+	struct bt_stream_class *sc;
+
+	BT_ASSERT(msg);
+	BT_ASSERT_PRE_DEV_HOT(msg, "Message", ": %!+n", msg);
+	sc = stream_msg->stream->class;
+	BT_ASSERT(sc);
+	BT_ASSERT_PRE(sc->default_clock_class,
+		"Message's stream's class has no default clock class: "
+		"%![msg-]+n, %![sc-]+S", msg, sc);
+	BT_ASSERT(stream_msg->default_cs);
+	bt_clock_snapshot_set_raw_value(stream_msg->default_cs, raw_value);
+	stream_msg->default_cs_state = BT_MESSAGE_STREAM_CLOCK_SNAPSHOT_STATE_KNOWN;
+	BT_LIB_LOGD("Set stream message's default clock snapshot: "
+		"%![msg-]+n, value=%" PRIu64, msg, raw_value);
+}
+
+void bt_message_stream_beginning_set_default_clock_snapshot(
+		struct bt_message *message, uint64_t raw_value)
+{
+	BT_ASSERT_PRE_NON_NULL(message, "Message");
+	BT_ASSERT_PRE_MSG_IS_TYPE(message, BT_MESSAGE_TYPE_STREAM_BEGINNING);
+
+	bt_message_stream_set_default_clock_snapshot(message, raw_value);
+}
+
+void bt_message_stream_end_set_default_clock_snapshot(
+		struct bt_message *message, uint64_t raw_value)
+{
+	BT_ASSERT_PRE_NON_NULL(message, "Message");
+	BT_ASSERT_PRE_MSG_IS_TYPE(message, BT_MESSAGE_TYPE_STREAM_END);
+
+	return bt_message_stream_set_default_clock_snapshot(message, raw_value);
+}
+
+static enum bt_message_stream_clock_snapshot_state
+bt_message_stream_borrow_default_clock_snapshot_const(
+		const bt_message *msg, const bt_clock_snapshot **snapshot)
+{
+	struct bt_message_stream *stream_msg = (void *) msg;
+	struct bt_stream_class *sc;
+
+	BT_ASSERT_DBG(msg);
+	sc = stream_msg->stream->class;
+	BT_ASSERT_DBG(sc);
+	BT_ASSERT_PRE_DEV(sc->default_clock_class,
+		"Message's stream's class has no default clock class: "
+		"%![msg-]+n, %![sc-]+S", msg, sc);
+	BT_ASSERT_DBG(stream_msg->default_cs);
+
+	*snapshot = stream_msg->default_cs;
+
+	return stream_msg->default_cs_state;
+}
+
+enum bt_message_stream_clock_snapshot_state
+bt_message_stream_beginning_borrow_default_clock_snapshot_const(
+		const bt_message *message, const bt_clock_snapshot **snapshot)
+{
+	BT_ASSERT_PRE_DEV_NON_NULL(message, "Message");
+	BT_ASSERT_PRE_DEV_MSG_IS_TYPE(message,
+		BT_MESSAGE_TYPE_STREAM_BEGINNING);
+	return bt_message_stream_borrow_default_clock_snapshot_const(
+		message, snapshot);
+}
+
+enum bt_message_stream_clock_snapshot_state
+bt_message_stream_end_borrow_default_clock_snapshot_const(
+		const bt_message *message, const bt_clock_snapshot **snapshot)
+{
+	BT_ASSERT_PRE_DEV_NON_NULL(message, "Message");
+	BT_ASSERT_PRE_DEV_MSG_IS_TYPE(message, BT_MESSAGE_TYPE_STREAM_END);
+	return bt_message_stream_borrow_default_clock_snapshot_const(
+		message, snapshot);
+}
+
+static inline
+const struct bt_clock_class *
+borrow_stream_message_stream_class_default_clock_class(
+		const struct bt_message *msg)
+{
+	struct bt_message_stream *stream_msg = (void *) msg;
+
+	BT_ASSERT_DBG(msg);
+	return stream_msg->stream->class->default_clock_class;
+}
+
+const struct bt_clock_class *
+bt_message_stream_beginning_borrow_stream_class_default_clock_class_const(
+		const struct bt_message *msg)
+{
+	BT_ASSERT_PRE_DEV_NON_NULL(msg, "Message");
+	BT_ASSERT_PRE_DEV_MSG_IS_TYPE(msg,
+		BT_MESSAGE_TYPE_STREAM_BEGINNING);
+	return borrow_stream_message_stream_class_default_clock_class(msg);
+}
+
+const struct bt_clock_class *
+bt_message_stream_end_borrow_stream_class_default_clock_class_const(
+		const struct bt_message *msg)
+{
+	BT_ASSERT_PRE_DEV_NON_NULL(msg, "Message");
+	BT_ASSERT_PRE_DEV_MSG_IS_TYPE(msg, BT_MESSAGE_TYPE_STREAM_END);
+	return borrow_stream_message_stream_class_default_clock_class(msg);
+}