lib: remove unused _NO_SINK graph status

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

diff --git a/include/babeltrace/graph/component-class-source-const.h b/include/babeltrace/graph/component-class-source-const.h
index aaf6d65860fd28cbe8eef6416954ea0dd46eca6e..f3c87ff6a60235c1d266b6583fcdc9a6e2bdb065 100644 (file)
--- a/include/babeltrace/graph/component-class-source-const.h
+++ b/include/babeltrace/graph/component-class-source-const.h
@@ -24,26 +24,29 @@
  * SOFTWARE.
  */
 
+/*
+ * For bt_component_class, bt_component_class_source,
+ * __BT_UPCAST_CONST
+ */
+#include <babeltrace/types.h>
+
 #ifdef __cplusplus
 extern "C" {
 #endif
 
-struct bt_component_class;
-struct bt_component_class_source;
-
 static inline
-const struct bt_component_class *
+const bt_component_class *
 bt_component_class_source_as_component_class_const(
-               const struct bt_component_class_source *comp_cls_source)
+               const bt_component_class_source *comp_cls_source)
 {
-       return (const void *) comp_cls_source;
+       return __BT_UPCAST_CONST(bt_component_class, comp_cls_source);
 }
 
 extern void bt_component_class_source_get_ref(
-               const struct bt_component_class_source *component_class_source);
+               const bt_component_class_source *component_class_source);
 
 extern void bt_component_class_source_put_ref(
-               const struct bt_component_class_source *component_class_source);
+               const bt_component_class_source *component_class_source);
 
 #define BT_COMPONENT_CLASS_SOURCE_PUT_REF_AND_RESET(_var)      \
        do {                                                    \