Support standard timestamp formats for begin/end

[babeltrace.git] / lib / plugin-system / component.c

diff --git a/lib/plugin-system/component.c b/lib/plugin-system/component.c
index 643e17ad21b3ad909862f556d35aae55e00fbff8..ec9390793b3407a1bc818905c7bda7f71dce5436 100644 (file)
--- a/lib/plugin-system/component.c
+++ b/lib/plugin-system/component.c
@@ -29,6 +29,8 @@
 #include <babeltrace/plugin/component.h>
 #include <babeltrace/plugin/component-internal.h>
 #include <babeltrace/plugin/source-internal.h>
+#include <babeltrace/plugin/filter-internal.h>
+#include <babeltrace/plugin/notification/iterator-internal.h>
 #include <babeltrace/plugin/sink-internal.h>
 #include <babeltrace/babeltrace-internal.h>
 #include <babeltrace/compiler.h>
@@ -39,6 +41,7 @@ struct bt_component * (* const component_create_funcs[])(
                struct bt_component_class *, struct bt_value *) = {
        [BT_COMPONENT_TYPE_SOURCE] = bt_component_source_create,
        [BT_COMPONENT_TYPE_SINK] = bt_component_sink_create,
+       [BT_COMPONENT_TYPE_FILTER] = bt_component_filter_create,
 };
 
 static
@@ -46,6 +49,7 @@ enum bt_component_status (* const component_validation_funcs[])(
                struct bt_component *) = {
        [BT_COMPONENT_TYPE_SOURCE] = bt_component_source_validate,
        [BT_COMPONENT_TYPE_SINK] = bt_component_sink_validate,
+       [BT_COMPONENT_TYPE_FILTER] = bt_component_filter_validate,
 };
 
 static
@@ -100,6 +104,75 @@ enum bt_component_type bt_component_get_type(struct bt_component *component)
        return component ? component->class->type : BT_COMPONENT_TYPE_UNKNOWN;
 }
 
+BT_HIDDEN
+struct bt_notification_iterator *bt_component_create_iterator(
+               struct bt_component *component)
+{
+       enum bt_notification_iterator_status ret_iterator;
+       enum bt_component_type component_type;
+       struct bt_notification_iterator *iterator = NULL;
+
+       if (!component) {
+               goto error;
+       }
+
+       component_type = bt_component_get_type(component);
+       if (component_type != BT_COMPONENT_TYPE_SOURCE &&
+                       component_type != BT_COMPONENT_TYPE_FILTER) {
+               /* Unsupported operation. */
+               goto error;
+       }
+
+       iterator = bt_notification_iterator_create(component);
+       if (!iterator) {
+               goto error;
+       }
+
+       switch (component_type) {
+       case BT_COMPONENT_TYPE_SOURCE:
+       {
+               struct bt_component_source *source;
+               enum bt_component_status ret_component;
+
+               source = container_of(component, struct bt_component_source, parent);
+               assert(source->init_iterator);
+               ret_component = source->init_iterator(component, iterator);
+               if (ret_component != BT_COMPONENT_STATUS_OK) {
+                       goto error;
+               }
+               break;
+
+               break;
+       }
+       case BT_COMPONENT_TYPE_FILTER:
+       {
+               struct bt_component_filter *filter;
+               enum bt_component_status ret_component;
+
+               filter = container_of(component, struct bt_component_filter, parent);
+               assert(filter->init_iterator);
+               ret_component = filter->init_iterator(component, iterator);
+               if (ret_component != BT_COMPONENT_STATUS_OK) {
+                       goto error;
+               }
+               break;
+       }
+       default:
+               /* Unreachable. */
+               assert(0);
+       }
+
+       ret_iterator = bt_notification_iterator_validate(iterator);
+       if (ret_iterator != BT_NOTIFICATION_ITERATOR_STATUS_OK) {
+               goto error;
+       }
+
+       return iterator;
+error:
+       BT_PUT(iterator);
+       return iterator;
+}
+
 struct bt_component *bt_component_create(
                struct bt_component_class *component_class, const char *name,
                struct bt_value *params)
@@ -114,8 +187,7 @@ struct bt_component *bt_component_create(
 
        type = bt_component_class_get_type(component_class);
        if (type <= BT_COMPONENT_TYPE_UNKNOWN ||
-                       type >= BT_COMPONENT_TYPE_FILTER) {
-               /* Filter components are not supported yet. */
+                       type > BT_COMPONENT_TYPE_FILTER) {
                goto end;
        }
 
@@ -132,8 +204,12 @@ struct bt_component *bt_component_create(
        }
 
        component->initializing = true;
-       component_class->init(component, params);
+       ret = component_class->init(component, params);
        component->initializing = false;
+       if (ret != BT_COMPONENT_STATUS_OK) {
+               BT_PUT(component);
+               goto end;
+       }
        ret = component_validation_funcs[type](component);
        if (ret != BT_COMPONENT_STATUS_OK) {
                BT_PUT(component);