X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;ds=sidebyside;f=include%2Fbabeltrace%2Fctf-ir%2Fstream-class-internal.h;h=b8366aaabd64a757d75384ccb533265b51a53996;hb=dc3fffef7b84cc4af1a7c99828fd57a106cd2257;hp=453b08515a8be31819a1233f78f6888696211d9a;hpb=09840de5d022697671dafe2bed427fa5c09b40b9;p=babeltrace.git

diff --git a/include/babeltrace/ctf-ir/stream-class-internal.h b/include/babeltrace/ctf-ir/stream-class-internal.h
index 453b0851..b8366aaa 100644
--- a/include/babeltrace/ctf-ir/stream-class-internal.h
+++ b/include/babeltrace/ctf-ir/stream-class-internal.h
@@ -33,7 +33,8 @@
 #include <babeltrace/ctf-ir/trace.h>
 #include <babeltrace/object-internal.h>
 #include <babeltrace/babeltrace-internal.h>
-#include <babeltrace/ctf/types.h>
+#include <babeltrace/ctf-ir/trace-internal.h>
+#include <assert.h>
 #include <glib.h>
 
 struct bt_ctf_stream_class {
@@ -41,6 +42,8 @@ struct bt_ctf_stream_class {
 	GString *name;
 	struct bt_ctf_clock *clock;
 	GPtrArray *event_classes; /* Array of pointers to bt_ctf_event_class */
+	/* event class id (int64_t) to event class */
+	GHashTable *event_classes_ht;
 	int id_set;
 	uint32_t id;
 	uint32_t next_event_id;
@@ -78,4 +81,12 @@ BT_HIDDEN
 int bt_ctf_stream_class_set_id_no_check(
 		struct bt_ctf_stream_class *stream_class, uint32_t id);
 
+static inline
+struct bt_ctf_trace *bt_ctf_stream_class_borrow_trace(
+		struct bt_ctf_stream_class *stream_class)
+{
+	assert(stream_class);
+	return (void *) bt_object_borrow_parent(stream_class);
+}
+
 #endif /* BABELTRACE_CTF_IR_STREAM_CLASS_INTERNAL_H */