X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=include%2Fbabeltrace%2Fgraph%2Fcomponent-source-const.h;h=8604359abe048efb72e46f95f6b218a021a6fbe4;hb=7e69c04f557e46fb50e5a888a3bc3451afbd3b8e;hp=daaaa4ef78e5368c79b05471df67eecbac994225;hpb=e2f7325d1e58710ee928373592adcee466f93d06;p=babeltrace.git

diff --git a/include/babeltrace/graph/component-source-const.h b/include/babeltrace/graph/component-source-const.h
index daaaa4ef..8604359a 100644
--- 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
 }