X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=include%2Fbabeltrace%2Fgraph%2Fself-component-class-source.h;h=f78b4f78701369ebe178b36c52a80f3e47a2c037;hb=005f1204d7620018bb2c1406b6194940f0ee119f;hp=07bb19121251f026942106da54799acf64cac639;hpb=e2f7325d1e58710ee928373592adcee466f93d06;p=babeltrace.git

diff --git a/include/babeltrace/graph/self-component-class-source.h b/include/babeltrace/graph/self-component-class-source.h
index 07bb1912..f78b4f78 100644
--- a/include/babeltrace/graph/self-component-class-source.h
+++ b/include/babeltrace/graph/self-component-class-source.h
@@ -24,19 +24,23 @@
  * SOFTWARE.
  */
 
+/*
+ * For bt_component_class_source, bt_self_component_class_source,
+ * __BT_UPCAST_CONST
+ */
+#include <babeltrace/types.h>
+
 #ifdef __cplusplus
 extern "C" {
 #endif
 
-struct bt_component_class_source;
-struct bt_self_component_class_source;
-
 static inline
-const struct bt_component_class_source *
+const bt_component_class_source *
 bt_self_component_class_source_as_component_class_source(
-		struct bt_self_component_class_source *self_comp_cls_source)
+		bt_self_component_class_source *self_comp_cls_source)
 {
-	return (const void *) self_comp_cls_source;
+	return __BT_UPCAST_CONST(bt_component_class_source,
+		self_comp_cls_source);
 }
 
 #ifdef __cplusplus