From: Philippe Proulx <eeppeliteloop@gmail.com>
Date: Thu, 6 Apr 2017 07:31:55 +0000 (-0400)
Subject: Add bt_component_class_is_source/filter/sink()
X-Git-Url: http://drtracing.org/?a=commitdiff_plain;h=2154eb7df06eb48c6db3a1447e657feccf47fa86;p=deliverable%2Fbabeltrace.git

Add bt_component_class_is_source/filter/sink()

Signed-off-by: Philippe Proulx <eeppeliteloop@gmail.com>
Signed-off-by: Jérémie Galarneau <jeremie.galarneau@efficios.com>
---

diff --git a/include/babeltrace/graph/component-class.h b/include/babeltrace/graph/component-class.h
index fcb171fc4..a3ca7ccc6 100644
--- a/include/babeltrace/graph/component-class.h
+++ b/include/babeltrace/graph/component-class.h
@@ -26,6 +26,7 @@
  */
 
 #include <stdint.h>
+#include <stdbool.h>
 #include <babeltrace/graph/component-status.h>
 #include <babeltrace/graph/notification-iterator.h>
 
@@ -170,6 +171,27 @@ extern struct bt_value *bt_component_class_query(
 extern enum bt_component_class_type bt_component_class_get_type(
 		struct bt_component_class *component_class);
 
+static inline
+bool bt_component_class_is_source(struct bt_component_class *component_class)
+{
+	return bt_component_class_get_type(component_class) ==
+		BT_COMPONENT_CLASS_TYPE_SOURCE;
+}
+
+static inline
+bool bt_component_class_is_filter(struct bt_component_class *component_class)
+{
+	return bt_component_class_get_type(component_class) ==
+		BT_COMPONENT_CLASS_TYPE_FILTER;
+}
+
+static inline
+bool bt_component_class_is_sink(struct bt_component_class *component_class)
+{
+	return bt_component_class_get_type(component_class) ==
+		BT_COMPONENT_CLASS_TYPE_SINK;
+}
+
 #ifdef __cplusplus
 }
 #endif