lib: make public reference count functions have strict types

[babeltrace.git] / include / babeltrace / trace-ir / stream-class-const.h

diff --git a/include/babeltrace/trace-ir/stream-class-const.h b/include/babeltrace/trace-ir/stream-class-const.h
index 8452c56c85c49db75b6f6cbaa882d22a182ce906..b740b936002e291ffb3a197cdb6c043a44805fbb 100644 (file)
--- a/include/babeltrace/trace-ir/stream-class-const.h
+++ b/include/babeltrace/trace-ir/stream-class-const.h
@@ -39,7 +39,7 @@ extern "C" {
 struct bt_trace_class;
 struct bt_stream_class;
 struct bt_event_class;
-struct bt_clock_class;
+struct bt_stream_class;
 struct bt_event_header_field;
 struct bt_packet_context_field;
 
@@ -100,6 +100,23 @@ extern bt_bool bt_stream_class_packets_have_default_beginning_clock_value(
 extern bt_bool bt_stream_class_packets_have_default_end_clock_value(
                const struct bt_stream_class *stream_class);
 
+extern void bt_stream_class_get_ref(const struct bt_stream_class *stream_class);
+
+extern void bt_stream_class_put_ref(const struct bt_stream_class *stream_class);
+
+#define BT_STREAM_CLASS_PUT_REF_AND_RESET(_var)                \
+       do {                                            \
+               bt_stream_class_put_ref(_var);          \
+               (_var) = NULL;                          \
+       } while (0)
+
+#define BT_STREAM_CLASS_MOVE_REF(_var_dst, _var_src)   \
+       do {                                            \
+               bt_stream_class_put_ref(_var_dst);      \
+               (_var_dst) = (_var_src);                \
+               (_var_src) = NULL;                      \
+       } while (0)
+
 #ifdef __cplusplus
 }
 #endif