lib: Add functions to borrow specialized component classes from specialized components

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

diff --git a/include/babeltrace/graph/component-sink-const.h b/include/babeltrace/graph/component-sink-const.h
index bda32846b2e440b4af9b50af603fd833bc833a41..5ce6798284d8e742ea1a2205b3fced8890245d4a 100644 (file)
--- a/include/babeltrace/graph/component-sink-const.h
+++ b/include/babeltrace/graph/component-sink-const.h
@@ -26,37 +26,43 @@
 
 #include <stdint.h>
 
+/*
+ * For bt_component, bt_component_filter, bt_port_input,
+ * __BT_UPCAST_CONST, bt_component_class_sink
+ */
+#include <babeltrace/types.h>
+
 #ifdef __cplusplus
 extern "C" {
 #endif
 
-struct bt_component;
-struct bt_component_sink;
-struct bt_port_input;
-
 static inline
-const struct bt_component *bt_component_sink_as_component_const(
-               const struct bt_component_sink *component)
+const bt_component *bt_component_sink_as_component_const(
+               const bt_component_sink *component)
 {
-       return (const void *) component;
+       return __BT_UPCAST_CONST(bt_component, component);
 }
 
+extern const bt_component_class_sink *
+bt_component_sink_borrow_class_const(
+               const bt_component_sink *component);
+
 extern uint64_t bt_component_sink_get_input_port_count(
-               const struct bt_component_sink *component);
+               const bt_component_sink *component);
 
-extern const struct bt_port_input *
+extern const bt_port_input *
 bt_component_sink_borrow_input_port_by_name_const(
-               const struct bt_component_sink *component, const char *name);
+               const bt_component_sink *component, const char *name);
 
-extern const struct bt_port_input *
+extern const bt_port_input *
 bt_component_sink_borrow_input_port_by_index_const(
-               const struct bt_component_sink *component, uint64_t index);
+               const bt_component_sink *component, uint64_t index);
 
 extern void bt_component_sink_get_ref(
-               const struct bt_component_sink *component_sink);
+               const bt_component_sink *component_sink);
 
 extern void bt_component_sink_put_ref(
-               const struct bt_component_sink *component_sink);
+               const bt_component_sink *component_sink);
 
 #define BT_COMPONENT_SINK_PUT_REF_AND_RESET(_var)              \
        do {                                                    \