#include <babeltrace/compiler-internal.h>
#include <babeltrace/values.h>
+#include <babeltrace/graph/private-component.h>
#include <babeltrace/graph/component-sink-internal.h>
#include <babeltrace/graph/component-internal.h>
#include <babeltrace/graph/notification.h>
+#include <babeltrace/graph/graph.h>
+#include <babeltrace/assert-internal.h>
+#include <babeltrace/assert-internal.h>
BT_HIDDEN
void bt_component_sink_destroy(struct bt_component *component)
BT_HIDDEN
struct bt_component *bt_component_sink_create(
- struct bt_component_class *class, struct bt_value *params)
+ struct bt_component_class *class)
{
struct bt_component_sink *sink = NULL;
return sink ? &sink->parent : NULL;
}
-BT_HIDDEN
-enum bt_component_status bt_component_sink_consume(
- struct bt_component *component)
-{
- enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
- struct bt_component_class_sink *sink_class = NULL;
-
- if (!component) {
- BT_LOGW_STR("Invalid parameter: component is NULL.");
- ret = BT_COMPONENT_STATUS_INVALID;
- goto end;
- }
-
- if (bt_component_get_class_type(component) != BT_COMPONENT_CLASS_TYPE_SINK) {
- BT_LOGW("Invalid parameter: component's class is not a sink component class: "
- "comp-addr=%p, comp-name=\"%s\", comp-class-type=%s",
- component, bt_component_get_name(component),
- bt_component_class_type_string(component->class->type));
- ret = BT_COMPONENT_STATUS_UNSUPPORTED;
- goto end;
- }
-
- sink_class = container_of(component->class, struct bt_component_class_sink, parent);
- assert(sink_class->methods.consume);
- BT_LOGD("Calling user's consume method: "
- "comp-addr=%p, comp-name=\"%s\"",
- component, bt_component_get_name(component));
- ret = sink_class->methods.consume(bt_private_component_from_component(component));
- BT_LOGD("User method returned: status=%s",
- bt_component_status_string(ret));
- if (ret < 0) {
- BT_LOGW_STR("Consume method failed.");
- }
-
-end:
- return ret;
-}
-
int64_t bt_component_sink_get_input_port_count(struct bt_component *component)
{
int64_t ret;
/* bt_component_sink_get_input_port_by_index() logs details/errors */
return bt_private_port_from_port(
bt_component_sink_get_input_port_by_index(
- bt_component_from_private(private_component), index));
+ bt_component_borrow_from_private(private_component), index));
}
struct bt_private_port *
/* bt_component_sink_get_input_port_by_name() logs details/errors */
return bt_private_port_from_port(
bt_component_sink_get_input_port_by_name(
- bt_component_from_private(private_component), name));
+ bt_component_borrow_from_private(private_component), name));
}
-struct bt_private_port *bt_private_component_sink_add_input_private_port(
+enum bt_component_status bt_private_component_sink_add_input_private_port(
struct bt_private_component *private_component,
- const char *name, void *user_data)
+ const char *name, void *user_data,
+ struct bt_private_port **user_priv_port)
{
+ enum bt_component_status status = BT_COMPONENT_STATUS_OK;
struct bt_port *port = NULL;
struct bt_component *component =
- bt_component_from_private(private_component);
+ bt_component_borrow_from_private(private_component);
+ struct bt_graph *graph;
if (!component) {
BT_LOGW_STR("Invalid parameter: component is NULL.");
+ status = BT_COMPONENT_STATUS_INVALID;
goto end;
}
"comp-addr=%p, comp-name=\"%s\", comp-class-type=%s",
component, bt_component_get_name(component),
bt_component_class_type_string(component->class->type));
+ status = BT_COMPONENT_STATUS_INVALID;
+ goto end;
+ }
+
+ graph = bt_component_borrow_graph(component);
+
+ if (graph && bt_graph_is_canceled(graph)) {
+ BT_LOGW("Cannot add input port to sink component: graph is canceled: "
+ "comp-addr=%p, comp-name=\"%s\", graph-addr=%p",
+ component, bt_component_get_name(component),
+ bt_component_borrow_graph(component));
+ status = BT_COMPONENT_STATUS_GRAPH_IS_CANCELED;
goto end;
}
/* bt_component_add_input_port() logs details/errors */
port = bt_component_add_input_port(component, name, user_data);
+ if (!port) {
+ status = BT_COMPONENT_STATUS_NOMEM;
+ goto end;
+ }
+
+ if (user_priv_port) {
+ /* Move reference to user */
+ *user_priv_port = bt_private_port_from_port(port);
+ port = NULL;
+ }
end:
- return bt_private_port_from_port(port);
+ bt_object_put_ref(port);
+ return status;
}