List detected component classes

[babeltrace.git] / plugins / component.c

diff --git a/plugins/component.c b/plugins/component.c
index c931ddf62053d42c400e5966d459aca109211795..5cfbbf0e3403e9d2d7f87b4f562864a04e168c66 100644 (file)
--- a/plugins/component.c
+++ b/plugins/component.c
@@ -28,10 +28,26 @@
 
 #include <babeltrace/plugin/component.h>
 #include <babeltrace/plugin/component-internal.h>
+#include <babeltrace/plugin/source-internal.h>
+#include <babeltrace/plugin/sink-internal.h>
 #include <babeltrace/babeltrace-internal.h>
 #include <babeltrace/compiler.h>
 #include <babeltrace/ref.h>
 
+static
+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,
+};
+
+static
+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,
+};
+
 static
 void bt_component_destroy(struct bt_object *obj)
 {
@@ -44,48 +60,85 @@ void bt_component_destroy(struct bt_object *obj)
 
        component = container_of(obj, struct bt_component, base);
 
-       /**
+       assert(component->destroy);
+       component_class = component->class;
+
+       /*
         * User data is destroyed first, followed by the concrete component
         * instance.
         */
-       assert(!component->user_data || component->user_destroy);
-       component->user_destroy(component->user_data);
-
-       g_string_free(component->name, TRUE);
-
-       assert(component->destroy);
-       component_class = component->class;
+       if (component->user_destroy) {
+               component->user_destroy(component->user_data);
+       }
 
-       /* Frees the component, which becomes invalid */
        component->destroy(component);
-       component = NULL;
-
+       g_string_free(component->name, TRUE);
        bt_put(component_class);
+       g_free(component);
 }
 
 BT_HIDDEN
 enum bt_component_status bt_component_init(struct bt_component *component,
-               struct bt_component_class *class, const char *name,
                bt_component_destroy_cb destroy)
 {
        enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
 
-       if (!component || !class || !name || name[0] == '\0' || !destroy) {
+       if (!component || !destroy) {
                ret = BT_COMPONENT_STATUS_INVAL;
                goto end;
        }
 
+       component->destroy = destroy;
+end:
+       return ret;
+}
+
+BT_HIDDEN
+enum bt_component_type bt_component_get_type(struct bt_component *component)
+{
+       return component ? component->class->type : BT_COMPONENT_TYPE_UNKNOWN;
+}
+
+struct bt_component *bt_component_create(
+               struct bt_component_class *component_class, const char *name,
+               struct bt_value *params)
+{
+       int ret;
+       struct bt_component *component = NULL;
+       enum bt_component_type type;
+
+       if (!component_class) {
+               goto end;
+       }
+
+       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. */
+               goto end;
+       }
+
+       component = component_create_funcs[type](component_class, params);
+       if (!component) {
+               goto end;
+       }
+
        bt_object_init(component, bt_component_destroy);
-       bt_get(class);
-       component->class = class;
+       component->class = bt_get(component_class);
        component->name = g_string_new(name);
-       if (!component->name) {
-               ret = BT_COMPONENT_STATUS_NOMEM;
+       if (component->name) {
+               BT_PUT(component);
+               goto end;
+       }
+
+       component_class->init(component, params);
+       ret = component_validation_funcs[type](component);
+       if (ret) {
+               BT_PUT(component);
                goto end;
        }
-       component->destroy = destroy;
 end:
-       return ret;
+       return component;
 }
 
 const char *bt_component_get_name(struct bt_component *component)
@@ -116,17 +169,10 @@ end:
        return ret;
 }
 
-enum bt_component_type bt_component_get_type(struct bt_component *component)
+struct bt_component_class *bt_component_get_class(
+               struct bt_component *component)
 {
-       enum bt_component_type type = BT_COMPONENT_TYPE_UNKNOWN;
-
-       if (!component) {
-               goto end;
-       }
-
-       type = component->class->type;
-end:
-       return type;
+       return component ? bt_get(component->class) : NULL;
 }
 
 enum bt_component_status bt_component_set_error_stream(
@@ -146,15 +192,7 @@ end:
 
 void *bt_component_get_private_data(struct bt_component *component)
 {
-       void *ret = NULL;
-
-       if (!component) {
-               goto end;
-       }
-
-       ret = component->user_data;
-end:
-       return ret;
+       return component ? component->user_data : NULL;
 }
 
 enum bt_component_status