lib: use object pool for event and packet notifications
[babeltrace.git] / lib / graph / source.c
index 0c4f1817c91e7584efda10238a43d6c47c11dad6..52ac8c81da502dd2725f365ed0e350d623bb0073 100644 (file)
  * SOFTWARE.
  */
 
+#define BT_LOG_TAG "COMP-SOURCE"
+#include <babeltrace/lib-logging-internal.h>
+
 #include <babeltrace/ref.h>
 #include <babeltrace/compiler-internal.h>
+#include <babeltrace/graph/private-component.h>
 #include <babeltrace/graph/component-source-internal.h>
 #include <babeltrace/graph/component-internal.h>
 #include <babeltrace/graph/port-internal.h>
 #include <babeltrace/graph/notification-iterator.h>
 #include <babeltrace/graph/notification-iterator-internal.h>
-
-BT_HIDDEN
-enum bt_component_status bt_component_source_validate(
-               struct bt_component *component)
-{
-       enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
-
-       if (!component) {
-               ret = BT_COMPONENT_STATUS_INVALID;
-               goto end;
-       }
-
-       if (!component->class) {
-               ret = BT_COMPONENT_STATUS_INVALID;
-               goto end;
-       }
-
-       if (component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
-               ret = BT_COMPONENT_STATUS_INVALID;
-               goto end;
-       }
-
-end:
-       return ret;
-}
+#include <babeltrace/graph/graph.h>
 
 BT_HIDDEN
 void bt_component_source_destroy(struct bt_component *component)
@@ -66,12 +46,13 @@ void bt_component_source_destroy(struct bt_component *component)
 
 BT_HIDDEN
 struct bt_component *bt_component_source_create(
-               struct bt_component_class *class, struct bt_value *params)
+               struct bt_component_class *class)
 {
        struct bt_component_source *source = NULL;
 
        source = g_new0(struct bt_component_source, 1);
        if (!source) {
+               BT_LOGE_STR("Failed to allocate one source component.");
                goto end;
        }
 
@@ -79,98 +60,160 @@ end:
        return source ? &source->parent : NULL;
 }
 
-enum bt_component_status bt_component_source_get_output_port_count(
-               struct bt_component *component, uint64_t *count)
+int64_t bt_component_source_get_output_port_count(
+               struct bt_component *component)
 {
-       enum bt_component_status status = BT_COMPONENT_STATUS_OK;
+       int64_t ret;
 
-       if (!component || !count ||
-                       component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
-               status = BT_COMPONENT_STATUS_INVALID;
+       if (!component) {
+               BT_LOGW_STR("Invalid parameter: component is NULL.");
+               ret = (int64_t) -1;
+               goto end;
+       }
+
+       if (component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
+               BT_LOGW("Invalid parameter: component's class is not a source 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 = (int64_t) -1;
                goto end;
        }
 
-       *count = bt_component_get_output_port_count(component);
+       /* bt_component_get_output_port_count() logs details/errors */
+       ret = bt_component_get_output_port_count(component);
+
 end:
-       return status;
+       return ret;
 }
 
-struct bt_port *bt_component_source_get_output_port(
+struct bt_port *bt_component_source_get_output_port_by_name(
                struct bt_component *component, const char *name)
 {
        struct bt_port *port = NULL;
 
-       if (!component || !name ||
-                       component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
+       if (!component) {
+               BT_LOGW_STR("Invalid parameter: component is NULL.");
+               goto end;
+       }
+
+       if (!name) {
+               BT_LOGW_STR("Invalid parameter: name is NULL.");
+               goto end;
+       }
+
+       if (component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
+               BT_LOGW("Invalid parameter: component's class is not a source 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));
                goto end;
        }
 
-       port = bt_component_get_output_port(component, name);
+       /* bt_component_get_output_port_by_name() logs details/errors */
+       port = bt_component_get_output_port_by_name(component, name);
+
 end:
        return port;
 }
 
-struct bt_port *bt_component_source_get_output_port_at_index(
-               struct bt_component *component, int index)
+struct bt_port *bt_component_source_get_output_port_by_index(
+               struct bt_component *component, uint64_t index)
 {
        struct bt_port *port = NULL;
 
-       if (!component ||
-                       component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
+       if (!component) {
+               BT_LOGW_STR("Invalid parameter: component is NULL.");
                goto end;
        }
 
-       port = bt_component_get_output_port_at_index(component, index);
+       if (component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
+               BT_LOGW("Invalid parameter: component's class is not a source 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));
+               goto end;
+       }
+
+       /* bt_component_get_output_port_by_index() logs details/errors */
+       port = bt_component_get_output_port_by_index(component, index);
+
 end:
        return port;
 }
 
-struct bt_port *bt_component_source_get_default_output_port(
-               struct bt_component *component)
-{
-       return bt_component_source_get_output_port(component,
-                       DEFAULT_OUTPUT_PORT_NAME);
-}
-
-struct bt_private_port *bt_private_component_source_get_output_private_port(
+struct bt_private_port *
+bt_private_component_source_get_output_private_port_by_name(
                struct bt_private_component *private_component,
                const char *name)
 {
-       return bt_private_port_from_port(bt_component_source_get_output_port(
-               bt_component_from_private(private_component), name));
-}
-
-struct bt_private_port *
-bt_private_component_source_get_output_private_port_at_index(
-               struct bt_private_component *private_component, int index)
-{
+       /* bt_component_source_get_output_port_by_name() logs details/errors */
        return bt_private_port_from_port(
-               bt_component_source_get_output_port_at_index(
-                       bt_component_from_private(private_component), index));
+               bt_component_source_get_output_port_by_name(
+                       bt_component_borrow_from_private(private_component), name));
 }
 
-struct bt_private_port *bt_private_component_source_get_default_output_private_port(
-               struct bt_private_component *private_component)
+struct bt_private_port *
+bt_private_component_source_get_output_private_port_by_index(
+               struct bt_private_component *private_component, uint64_t index)
 {
+       /* bt_component_source_get_output_port_by_index() logs details/errors */
        return bt_private_port_from_port(
-               bt_component_source_get_default_output_port(
-                       bt_component_from_private(private_component)));
+               bt_component_source_get_output_port_by_index(
+                       bt_component_borrow_from_private(private_component), index));
 }
 
-struct bt_private_port *bt_private_component_source_add_output_private_port(
+enum bt_component_status bt_private_component_source_add_output_private_port(
                struct bt_private_component *private_component,
-               const char *name)
+               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;
+       }
+
+       if (component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
+               BT_LOGW("Invalid parameter: component's class is not a source 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));
+               status = BT_COMPONENT_STATUS_INVALID;
+               goto end;
+       }
+
+       graph = bt_component_borrow_graph(component);
 
-       if (!component ||
-                       component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
+       if (graph && bt_graph_is_canceled(graph)) {
+               BT_LOGW("Cannot add output port to source 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;
        }
 
-       port = bt_component_add_output_port(component, name);
+       /* bt_component_add_output_port() logs details and errors */
+       port = bt_component_add_output_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_put(port);
+       return status;
 }
This page took 0.028808 seconds and 4 git commands to generate.