#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
}
component = container_of(obj, struct bt_component, base);
-
- assert(component->destroy);
component_class = component->class;
/*
* instance.
*/
if (component->user_destroy) {
- component->user_destroy(component->user_data);
+ component->user_destroy(component);
+ }
+
+ if (component->destroy) {
+ component->destroy(component);
}
- component->destroy(component);
g_string_free(component->name, TRUE);
bt_put(component_class);
g_free(component);
{
enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
- if (!component || !destroy) {
- ret = BT_COMPONENT_STATUS_INVAL;
+ if (!component) {
+ ret = BT_COMPONENT_STATUS_INVALID;
goto end;
}
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;
}
}
bt_object_init(component, bt_component_destroy);
- component->class = bt_get(component_class);
component->name = g_string_new(name);
- if (component->name) {
+ if (!component->name) {
BT_PUT(component);
goto end;
}
- component_class->init(component, params);
+ component->initializing = true;
+ 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) {
+ if (ret != BT_COMPONENT_STATUS_OK) {
BT_PUT(component);
goto end;
}
enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
if (!component || !name || name[0] == '\0') {
- ret = BT_COMPONENT_STATUS_INVAL;
+ ret = BT_COMPONENT_STATUS_INVALID;
goto end;
}
return component ? bt_get(component->class) : NULL;
}
-enum bt_component_status bt_component_set_error_stream(
- struct bt_component *component, FILE *stream)
+void *bt_component_get_private_data(struct bt_component *component)
+{
+ return component ? component->user_data : NULL;
+}
+
+enum bt_component_status
+bt_component_set_private_data(struct bt_component *component,
+ void *data)
{
enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
- if (!component) {
- ret = BT_COMPONENT_STATUS_INVAL;
+ if (!component || !component->initializing) {
+ ret = BT_COMPONENT_STATUS_INVALID;
goto end;
}
- component->error_stream = stream;
+ component->user_data = data;
end:
return ret;
}
-void *bt_component_get_private_data(struct bt_component *component)
-{
- return component ? component->user_data : NULL;
-}
-
-enum bt_component_status
-bt_component_set_private_data(struct bt_component *component,
- void *data)
+enum bt_component_status bt_component_set_destroy_cb(
+ struct bt_component *component, bt_component_destroy_cb destroy)
{
enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
- if (!component) {
- ret = BT_COMPONENT_STATUS_INVAL;
+ if (!component || !component->initializing) {
+ ret = BT_COMPONENT_STATUS_INVALID;
goto end;
}
- component->user_data = data;
+ component->user_destroy = destroy;
end:
return ret;
}