lib: add "borrow" functions where "get" functions exist

[babeltrace.git] / include / babeltrace / ctf-ir / stream.h

diff --git a/include/babeltrace/ctf-ir/stream.h b/include/babeltrace/ctf-ir/stream.h
index 4b4ac3cfe42d9859d283aa612764c2f25ae16d31..5f0fd26bfda28c1522f946ce7cec5136032c2fab 100644 (file)
--- a/include/babeltrace/ctf-ir/stream.h
+++ b/include/babeltrace/ctf-ir/stream.h
@@ -30,6 +30,9 @@
  * http://www.efficios.com/ctf
  */
 
+/* For bt_get() */
+#include <babeltrace/ref.h>
+
 #include <stdint.h>
 
 #ifdef __cplusplus
@@ -153,6 +156,9 @@ extern const char *bt_stream_get_name(struct bt_stream *stream);
 */
 extern int64_t bt_stream_get_id(struct bt_stream *stream);
 
+extern struct bt_stream_class *bt_stream_borrow_class(
+               struct bt_stream *stream);
+
 /**
 @brief Returns the parent CTF IR stream class of the CTF IR
        stream \p stream.
@@ -169,8 +175,12 @@ bt_stream_create().
 @postrefcountsame{stream}
 @postsuccessrefcountretinc
 */
-extern struct bt_stream_class *bt_stream_get_class(
-               struct bt_stream *stream);
+static inline
+struct bt_stream_class *bt_stream_get_class(
+               struct bt_stream *stream)
+{
+       return bt_get(bt_stream_borrow_class(stream));
+}
 
 /** @} */