lib: use object pool for event and packet notifications

[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 7fdca46ae8cdfb105ad7b30cd3188c95f9c36045..9e6012ec641e18225640869a8429db2205ec15fd 100644 (file)
--- 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
@@ -29,8 +29,11 @@
 
 #include <babeltrace/graph/graph.h>
 #include <babeltrace/graph/component-status.h>
+#include <babeltrace/graph/notification.h>
 #include <babeltrace/babeltrace-internal.h>
 #include <babeltrace/object-internal.h>
+#include <babeltrace/object-pool-internal.h>
+#include <babeltrace/assert-internal.h>
 #include <stdlib.h>
 #include <glib.h>
 
@@ -60,6 +63,17 @@ 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 {
@@ -68,8 +82,47 @@ struct bt_graph {
                GArray *ports_connected;
                GArray *ports_disconnected;
        } listeners;
+
+       /* Pool of `struct bt_notification_event *` */
+       struct bt_object_pool event_notif_pool;
+
+       /* Pool of `struct bt_notification_packet_begin *` */
+       struct bt_object_pool packet_begin_notif_pool;
+
+       /* Pool of `struct bt_notification_packet_end *` */
+       struct bt_object_pool packet_end_notif_pool;
+
+       /*
+        * Array of `struct bt_notification *` (weak).
+        *
+        * This is an array of all the notifications ever created from
+        * this graph. Some of them can be in one of the pools above,
+        * some of them can be at large. Because each notification has a
+        * weak pointer to the graph containing its pool, we need to
+        * notify each notification that the graph is gone on graph
+        * destruction.
+        *
+        * TODO: When we support a maximum size for object pools,
+        * add a way for a notification to remove itself from this
+        * array (on destruction).
+        */
+       GPtrArray *notifications;
 };
 
+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);
 
@@ -92,6 +145,22 @@ 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);
+
+BT_HIDDEN
+void bt_graph_add_notification(struct bt_graph *graph,
+               struct bt_notification *notif);
+
 static inline
 const char *bt_graph_status_string(enum bt_graph_status status)
 {
@@ -152,4 +221,4 @@ enum bt_graph_status bt_graph_status_from_component_status(
        }
 }
 
-#endif /* BABELTRACE_COMPONENT_COMPONENT_GRAPH_INTERNAL_H */
+#endif /* BABELTRACE_GRAPH_GRAPH_INTERNAL_H */