#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>
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
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
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)
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;
}
}
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);