X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=include%2Fbabeltrace%2Fgraph%2Fgraph-internal.h;h=cd1b557d946a9287a705abf333a5fe454397f2c1;hb=312c056ae3d374b253fa0cfe5ed576c0b0e5e569;hp=85c196be6e2bf4793d09da989f7b9b96b80b6307;hpb=f345f8bb507db96ad6f068f404603f6390fa34ce;p=babeltrace.git

diff --git a/include/babeltrace/graph/graph-internal.h b/include/babeltrace/graph/graph-internal.h
index 85c196be..cd1b557d 100644
--- a/include/babeltrace/graph/graph-internal.h
+++ b/include/babeltrace/graph/graph-internal.h
@@ -1,5 +1,5 @@
-#ifndef BABELTRACE_COMPONENT_COMPONENT_GRAPH_INTERNAL_H
-#define BABELTRACE_COMPONENT_COMPONENT_GRAPH_INTERNAL_H
+#ifndef BABELTRACE_GRAPH_GRAPH_INTERNAL_H
+#define BABELTRACE_GRAPH_GRAPH_INTERNAL_H
 
 /*
  * BabelTrace - Component Graph Internal
@@ -28,8 +28,11 @@
  */
 
 #include <babeltrace/graph/graph.h>
+#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>
 
 struct bt_component;
@@ -55,6 +58,22 @@ struct bt_graph {
 	/* Queue of pointers (weak references) to sink bt_components. */
 	GQueue *sinks_to_consume;
 
+	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;
 		GArray *port_removed;
@@ -63,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);
 
@@ -81,4 +114,80 @@ void bt_graph_notify_ports_disconnected(struct bt_graph *graph,
 		struct bt_port *upstream_port,
 		struct bt_port *downstream_port);
 
-#endif /* BABELTRACE_COMPONENT_COMPONENT_GRAPH_INTERNAL_H */
+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)
+{
+	switch (status) {
+	case BT_GRAPH_STATUS_CANCELED:
+		return "BT_GRAPH_STATUS_CANCELED";
+	case BT_GRAPH_STATUS_AGAIN:
+		return "BT_GRAPH_STATUS_AGAIN";
+	case BT_GRAPH_STATUS_END:
+		return "BT_GRAPH_STATUS_END";
+	case BT_GRAPH_STATUS_OK:
+		return "BT_GRAPH_STATUS_OK";
+	case BT_GRAPH_STATUS_INVALID:
+		return "BT_GRAPH_STATUS_INVALID";
+	case BT_GRAPH_STATUS_NO_SINK:
+		return "BT_GRAPH_STATUS_NO_SINK";
+	case BT_GRAPH_STATUS_ERROR:
+		return "BT_GRAPH_STATUS_ERROR";
+	case BT_GRAPH_STATUS_COMPONENT_REFUSES_PORT_CONNECTION:
+		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)";
+	}
+}
+
+static inline
+enum bt_graph_status bt_graph_status_from_component_status(
+		enum bt_component_status comp_status)
+{
+	switch (comp_status) {
+	case BT_COMPONENT_STATUS_OK:
+		return BT_GRAPH_STATUS_OK;
+	case BT_COMPONENT_STATUS_END:
+		return BT_GRAPH_STATUS_END;
+	case BT_COMPONENT_STATUS_AGAIN:
+		return BT_GRAPH_STATUS_AGAIN;
+	case BT_COMPONENT_STATUS_REFUSE_PORT_CONNECTION:
+		return BT_GRAPH_STATUS_COMPONENT_REFUSES_PORT_CONNECTION;
+	case BT_COMPONENT_STATUS_ERROR:
+		return BT_GRAPH_STATUS_ERROR;
+	case BT_COMPONENT_STATUS_UNSUPPORTED:
+		return BT_GRAPH_STATUS_ERROR;
+	case BT_COMPONENT_STATUS_INVALID:
+		return BT_GRAPH_STATUS_INVALID;
+	case BT_COMPONENT_STATUS_NOMEM:
+		return BT_GRAPH_STATUS_NOMEM;
+	case BT_COMPONENT_STATUS_NOT_FOUND:
+		return BT_GRAPH_STATUS_ERROR;
+	default:
+#ifdef BT_LOGF
+		BT_LOGF("Unknown component status: status=%d", comp_status);
+#endif
+		abort();
+	}
+}
+
+#endif /* BABELTRACE_GRAPH_GRAPH_INTERNAL_H */