Add ports to the source, filter and sink component interfaces

[babeltrace.git] / lib / component / source.c

diff --git a/lib/component/source.c b/lib/component/source.c
index 393b12f8b322907fcbbc6ef8239f762a7d57bf51..0baa3f9b7fdcc4ae695a5a2283a080adab66f637 100644 (file)
--- a/lib/component/source.c
+++ b/lib/component/source.c
@@ -30,6 +30,7 @@
 #include <babeltrace/compiler.h>
 #include <babeltrace/component/component-source-internal.h>
 #include <babeltrace/component/component-internal.h>
+#include <babeltrace/component/component-port-internal.h>
 #include <babeltrace/component/notification/iterator.h>
 #include <babeltrace/component/notification/iterator-internal.h>
 
@@ -58,12 +59,25 @@ end:
        return ret;
 }
 
+static
+void bt_component_source_destroy(struct bt_component *component)
+{
+       struct bt_component_source *source = container_of(component,
+                       struct bt_component_source, parent);
+
+       if (source->output_ports) {
+               g_ptr_array_free(source->output_ports, TRUE);
+       }
+}
+
+
 BT_HIDDEN
 struct bt_component *bt_component_source_create(
                struct bt_component_class *class, struct bt_value *params)
 {
+       int ret;
        struct bt_component_source *source = NULL;
-       enum bt_component_status ret;
+       enum bt_component_status status;
 
        source = g_new0(struct bt_component_source, 1);
        if (!source) {
@@ -71,13 +85,21 @@ struct bt_component *bt_component_source_create(
        }
 
        source->parent.class = bt_get(class);
-       ret = bt_component_init(&source->parent, NULL);
-       if (ret != BT_COMPONENT_STATUS_OK) {
-               BT_PUT(source);
-               goto end;
+       status = bt_component_init(&source->parent, bt_component_source_destroy);
+       if (status != BT_COMPONENT_STATUS_OK) {
+               goto error;
        }
+
+       ret = bt_component_init_output_ports(&source->parent, &source->output_ports);
+       if (ret) {
+               goto error;
+       }
+
 end:
        return source ? &source->parent : NULL;
+error:
+       BT_PUT(source);
+       goto end;
 }
 
 struct bt_notification_iterator *bt_component_source_create_notification_iterator(
@@ -91,3 +113,103 @@ struct bt_notification_iterator *bt_component_source_create_notification_iterato
 {
        return bt_component_create_iterator(component, init_method_data);
 }
+
+int bt_component_source_get_output_port_count(struct bt_component *component)
+{
+       int ret;
+       struct bt_component_source *source;
+
+       if (!component) {
+               ret = -1;
+               goto end;
+       }
+
+       if (component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
+               ret = -1;
+               goto end;
+       }
+
+       source = container_of(component, struct bt_component_source, parent);
+       ret = source->output_ports->len;
+end:
+       return ret;
+}
+
+struct bt_port *bt_component_source_get_output_port(
+               struct bt_component *component, const char *name)
+{
+       struct bt_component_source *source;
+       struct bt_port *ret_port = NULL;
+
+       if (!component || !name ||
+                       component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
+               goto end;
+       }
+
+       source = container_of(component, struct bt_component_source, parent);
+       ret_port = bt_component_get_port(source->output_ports, name);
+end:
+       return ret_port;
+}
+
+struct bt_port *bt_component_source_get_output_port_at_index(
+               struct bt_component *component, int index)
+{
+       struct bt_port *port = NULL;
+       struct bt_component_source *source;
+
+       if (!component ||
+                       component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
+               goto end;
+       }
+
+       source = container_of(component, struct bt_component_source, parent);
+       port = bt_component_get_port_at_index(source->output_ports, 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_port *bt_component_source_add_output_port(
+               struct bt_component *component, const char *name)
+{
+       struct bt_port *port;
+       struct bt_component_source *source;
+
+       if (!component ||
+                       component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
+               port = NULL;
+               goto end;
+       }
+
+       source = container_of(component, struct bt_component_source, parent);
+       port = bt_component_add_port(component, source->output_ports,
+                       BT_PORT_TYPE_OUTPUT, name);
+end:
+       return port;
+}
+
+enum bt_component_status bt_component_source_remove_output_port(
+               struct bt_component *component, const char *name)
+{
+       enum bt_component_status status;
+       struct bt_component_source *source;
+
+       if (!component ||
+                       component->class->type != BT_COMPONENT_CLASS_TYPE_SOURCE) {
+               status = BT_COMPONENT_STATUS_INVALID;
+               goto end;
+       }
+
+       source = container_of(component, struct bt_component_source, parent);
+       status = bt_component_remove_port(component, source->output_ports,
+                       name);
+end:
+       return status;
+}