X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=formats%2Fctf%2Fir%2Fstream-class.c;h=07f15651361280752a38998033b55ace8e5d03c7;hb=9f476966aa40bd0de2cd0654623ea03f8a3254eb;hp=058f4b29ecd1765f6a7d0521d58c02eb559a73c5;hpb=36336d93377e7e3c9a5077b32f63bcef9a08e81c;p=babeltrace.git

diff --git a/formats/ctf/ir/stream-class.c b/formats/ctf/ir/stream-class.c
index 058f4b29..07f15651 100644
--- a/formats/ctf/ir/stream-class.c
+++ b/formats/ctf/ir/stream-class.c
@@ -306,6 +306,47 @@ end:
 	return ret;
 }
 
+struct bt_ctf_field_type *bt_ctf_stream_class_get_event_header_type(
+		struct bt_ctf_stream_class *stream_class)
+{
+	struct bt_ctf_field_type *ret = NULL;
+
+	if (!stream_class || !stream_class->event_header_type) {
+		goto end;
+	}
+
+	assert(stream_class->event_header_type);
+	bt_ctf_field_type_get(stream_class->event_header_type);
+	ret = stream_class->event_header_type;
+end:
+	return ret;
+}
+
+int bt_ctf_stream_class_set_event_header_type(
+		struct bt_ctf_stream_class *stream_class,
+		struct bt_ctf_field_type *event_header_type)
+{
+	int ret = 0;
+
+	if (!stream_class || !event_header_type || stream_class->frozen) {
+		ret = -1;
+		goto end;
+	}
+
+	if (bt_ctf_field_type_get_type_id(event_header_type) !=
+		CTF_TYPE_STRUCT) {
+		/* An event header must be a structure */
+		ret = -1;
+		goto end;
+	}
+
+	bt_ctf_field_type_put(stream_class->event_header_type);
+	bt_ctf_field_type_get(event_header_type);
+	stream_class->event_header_type = event_header_type;
+end:
+	return ret;
+}
+
 struct bt_ctf_field_type *bt_ctf_stream_class_get_event_context_type(
 		struct bt_ctf_stream_class *stream_class)
 {
@@ -480,7 +521,6 @@ void bt_ctf_stream_class_destroy(struct bt_ctf_ref *ref)
 	}
 
 	bt_ctf_field_type_put(stream_class->event_header_type);
-	bt_ctf_field_put(stream_class->event_header);
 	bt_ctf_field_type_put(stream_class->packet_context_type);
 	if (stream_class->event_context_type) {
 		bt_ctf_field_type_put(stream_class->event_context_type);
@@ -528,11 +568,6 @@ int init_event_header(struct bt_ctf_stream_class *stream_class,
 	}
 
 	stream_class->event_header_type = event_header_type;
-	stream_class->event_header = bt_ctf_field_create(
-		stream_class->event_header_type);
-	if (!stream_class->event_header) {
-		ret = -1;
-	}
 end:
 	if (ret) {
 		bt_ctf_field_type_put(event_header_type);