Replace assert() -> BT_ASSERT() and some preconditions with BT_ASSERT_PRE()

[babeltrace.git] / include / babeltrace / graph / graph-internal.h

diff --git a/include/babeltrace/graph/graph-internal.h b/include/babeltrace/graph/graph-internal.h
index 1e8b1b79ab3407671b39848e4b1d4baa89f119be..c8be19aace1d6d91876a6641c7fd4fb66432e6b7 100644 (file)
--- a/include/babeltrace/graph/graph-internal.h
+++ b/include/babeltrace/graph/graph-internal.h
@@ -31,6 +31,7 @@
 #include <babeltrace/graph/component-status.h>
 #include <babeltrace/babeltrace-internal.h>
 #include <babeltrace/object-internal.h>
+#include <babeltrace/assert-internal.h>
 #include <stdlib.h>
 #include <glib.h>
 
@@ -59,6 +60,19 @@ struct bt_graph {
 
        bt_bool canceled;
        bt_bool in_remove_listener;
+       bt_bool has_sink;
+
+       /*
+        * If this is BT_FALSE, then the public API's consuming
+        * functions (bt_graph_consume() and bt_graph_run()) return
+        * BT_GRAPH_STATUS_CANNOT_CONSUME. The internal "no check"
+        * functions always work.
+        *
+        * In bt_output_port_notification_iterator_create(), on success,
+        * this flag is cleared so that the iterator remains the only
+        * consumer for the graph's lifetime.
+        */
+       bt_bool can_consume;
 
        struct {
                GArray *port_added;
@@ -68,6 +82,20 @@ struct bt_graph {
        } listeners;
 };
 
+static inline
+void bt_graph_set_can_consume(struct bt_graph *graph, bt_bool can_consume)
+{
+       BT_ASSERT(graph);
+       graph->can_consume = can_consume;
+}
+
+BT_HIDDEN
+enum bt_graph_status bt_graph_consume_no_check(struct bt_graph *graph);
+
+BT_HIDDEN
+enum bt_graph_status bt_graph_consume_sink_no_check(struct bt_graph *graph,
+               struct bt_component *sink);
+
 BT_HIDDEN
 void bt_graph_notify_port_added(struct bt_graph *graph, struct bt_port *port);
 
@@ -90,6 +118,18 @@ BT_HIDDEN
 void bt_graph_remove_connection(struct bt_graph *graph,
                struct bt_connection *connection);
 
+/*
+ * This only works with a component which is not connected at this
+ * point.
+ *
+ * Also the reference count of `component` should be 0 when you call
+ * this function, which means only `graph` owns the component, so it
+ * is safe to destroy.
+ */
+BT_HIDDEN
+int bt_graph_remove_unconnected_component(struct bt_graph *graph,
+               struct bt_component *component);
+
 static inline
 const char *bt_graph_status_string(enum bt_graph_status status)
 {
@@ -112,6 +152,8 @@ const char *bt_graph_status_string(enum bt_graph_status status)
                return "BT_GRAPH_STATUS_COMPONENT_REFUSES_PORT_CONNECTION";
        case BT_GRAPH_STATUS_NOMEM:
                return "BT_GRAPH_STATUS_NOMEM";
+       case BT_GRAPH_STATUS_CANNOT_CONSUME:
+               return "BT_GRAPH_STATUS_CANNOT_CONSUME";
        default:
                return "(unknown)";
        }