lib: Reset libbabeltrace2 to SONANE 0

[babeltrace.git] / include / babeltrace / graph / component-source-const.h

diff --git a/include/babeltrace/graph/component-source-const.h b/include/babeltrace/graph/component-source-const.h
index daaaa4ef78e5368c79b05471df67eecbac994225..8604359abe048efb72e46f95f6b218a021a6fbe4 100644 (file)
--- a/include/babeltrace/graph/component-source-const.h
+++ b/include/babeltrace/graph/component-source-const.h
@@ -26,31 +26,56 @@
 
 #include <stdint.h>
 
+/*
+ * For bt_component, bt_component_filter, bt_port_output,
+ * __BT_UPCAST_CONST, bt_component_class_source
+ */
+#include <babeltrace/types.h>
+
 #ifdef __cplusplus
 extern "C" {
 #endif
 
-struct bt_component;
-struct bt_component_source;
-struct bt_port_output;
-
 static inline
-const struct bt_component *bt_component_source_as_component_const(
-               const struct bt_component_source *component)
+const bt_component *bt_component_source_as_component_const(
+               const bt_component_source *component)
 {
-       return (void *) component;
+       return __BT_UPCAST_CONST(bt_component, component);
 }
 
+extern const bt_component_class_source *
+bt_component_source_borrow_class_const(
+               const bt_component_source *component);
+
 extern uint64_t bt_component_source_get_output_port_count(
-               const struct bt_component_source *component);
+               const bt_component_source *component);
 
-extern const struct bt_port_output *
+extern const bt_port_output *
 bt_component_source_borrow_output_port_by_name_const(
-               const struct bt_component_source *component, const char *name);
+               const bt_component_source *component, const char *name);
 
-extern const struct bt_port_output *
+extern const bt_port_output *
 bt_component_source_borrow_output_port_by_index_const(
-               const struct bt_component_source *component, uint64_t index);
+               const bt_component_source *component, uint64_t index);
+
+extern void bt_component_source_get_ref(
+               const bt_component_source *component_source);
+
+extern void bt_component_source_put_ref(
+               const bt_component_source *component_source);
+
+#define BT_COMPONENT_SOURCE_PUT_REF_AND_RESET(_var)            \
+       do {                                                    \
+               bt_component_source_put_ref(_var);              \
+               (_var) = NULL;                                  \
+       } while (0)
+
+#define BT_COMPONENT_SOURCE_MOVE_REF(_var_dst, _var_src)       \
+       do {                                                    \
+               bt_component_source_put_ref(_var_dst);          \
+               (_var_dst) = (_var_src);                        \
+               (_var_src) = NULL;                              \
+       } while (0)
 
 #ifdef __cplusplus
 }