Rename correlator to muxer

[babeltrace.git] / plugins / ctf / fs / fs.c

diff --git a/plugins/ctf/fs/fs.c b/plugins/ctf/fs/fs.c
index 6f0a3c0a77271d994a97a0a1fea28b0575958859..b80e4e152095c4e3ed19a0abf80b72dc2fe3974b 100644 (file)
--- a/plugins/ctf/fs/fs.c
+++ b/plugins/ctf/fs/fs.c
@@ -27,32 +27,188 @@
  */
 
 #include <babeltrace/plugin/plugin-system.h>
+#include <babeltrace/plugin/notification/iterator.h>
 #include <glib.h>
 #include <assert.h>
-#include "fs-internal.h"
+#include <unistd.h>
+#include "fs.h"
+#include "metadata.h"
+#include "data-stream.h"
 
 static bool ctf_fs_debug;
 
 static
-struct ctf_fs_component *ctf_fs_create(struct bt_value *params)
+struct bt_notification *ctf_fs_iterator_get(
+               struct bt_notification_iterator *iterator)
 {
-       return g_new0(struct ctf_fs_component, 1);
+       struct bt_notification *notification = NULL;
+       struct ctf_fs_component *ctf_fs;
+       struct bt_component *component = bt_notification_iterator_get_component(
+                       iterator);
+
+       if (!component) {
+               goto end;
+       }
+
+       ctf_fs = bt_component_get_private_data(component);
+       if (!ctf_fs) {
+               goto end;
+       }
+
+       notification = bt_get(ctf_fs->current_notification);
+end:
+       BT_PUT(component);
+       return notification;
 }
 
 static
-void ctf_fs_destroy(void *data)
+enum bt_notification_iterator_status ctf_fs_iterator_next(
+               struct bt_notification_iterator *iterator)
 {
-       g_free(data);
+       enum bt_notification_iterator_status ret;
+       struct bt_notification *notification = NULL;
+       struct ctf_fs_component *ctf_fs;
+       struct bt_component *component = bt_notification_iterator_get_component(
+                       iterator);
+
+       if (!component) {
+               ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
+               goto end;
+       }
+
+       ctf_fs = bt_component_get_private_data(component);
+       assert(ctf_fs);
+
+       ret = ctf_fs_data_stream_get_next_notification(ctf_fs, &notification);
+       if (ret || !notification) {
+               goto end;
+       }
+
+       bt_put(ctf_fs->current_notification);
+       ctf_fs->current_notification = notification;
+end:
+       BT_PUT(component);
+       return ret;
 }
 
-BT_HIDDEN
+static
+void ctf_fs_iterator_destroy_data(struct ctf_fs_iterator *ctf_it)
+{
+       g_free(ctf_it);
+}
+
+static
+void ctf_fs_iterator_destroy(struct bt_notification_iterator *it)
+{
+       void *data = bt_notification_iterator_get_private_data(it);
+
+       ctf_fs_iterator_destroy_data(data);
+}
+
+static
 enum bt_component_status ctf_fs_iterator_init(struct bt_component *source,
                struct bt_notification_iterator *it)
 {
+       struct ctf_fs_iterator *ctf_it;
        enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
 
        assert(source && it);
+       ctf_it = g_new0(struct ctf_fs_iterator, 1);
+       if (!ctf_it) {
+               ret = BT_COMPONENT_STATUS_NOMEM;
+               goto end;
+       }
+
+       ret = bt_notification_iterator_set_get_cb(it, ctf_fs_iterator_get);
+       if (ret) {
+               goto error;
+       }
+
+       ret = bt_notification_iterator_set_next_cb(it, ctf_fs_iterator_next);
+       if (ret) {
+               goto error;
+       }
+
+       ret = bt_notification_iterator_set_destroy_cb(it,
+                       ctf_fs_iterator_destroy);
+       if (ret) {
+               goto error;
+       }
+
+       ret = bt_notification_iterator_set_private_data(it, ctf_it);
+       if (ret) {
+               goto error;
+       }
+end:
        return ret;
+error:
+       (void) bt_notification_iterator_set_private_data(it, NULL);
+       ctf_fs_iterator_destroy_data(ctf_it);
+       return ret;
+}
+
+static
+void ctf_fs_destroy_data(struct ctf_fs_component *component)
+{
+       if (component->trace_path) {
+               g_string_free(component->trace_path, TRUE);
+       }
+
+       ctf_fs_metadata_fini(&component->metadata);
+       ctf_fs_data_stream_fini(&component->data_stream);
+       BT_PUT(component->current_notification);
+       g_free(component);
+}
+
+static
+void ctf_fs_destroy(struct bt_component *component)
+{
+       void *data = bt_component_get_private_data(component);
+
+       ctf_fs_destroy_data(data);
+}
+
+static
+struct ctf_fs_component *ctf_fs_create(struct bt_value *params)
+{
+       struct ctf_fs_component *ctf_fs;
+       struct bt_value *value = NULL;
+       const char *path;
+       enum bt_value_status ret;
+
+       ctf_fs = g_new0(struct ctf_fs_component, 1);
+       if (!ctf_fs) {
+               goto end;
+       }
+
+       /* FIXME: should probably look for a source URI */
+       value = bt_value_map_get(params, "path");
+       if (!value || bt_value_is_null(value) || !bt_value_is_string(value)) {
+               goto error;
+       }
+
+       ret = bt_value_string_get(value, &path);
+       if (ret != BT_VALUE_STATUS_OK) {
+               goto error;
+       }
+
+       ctf_fs->trace_path = g_string_new(path);
+       if (!ctf_fs->trace_path) {
+               goto error;
+       }
+
+       ctf_fs->error_fp = stderr;
+       ctf_fs->page_size = (size_t) getpagesize();
+       ctf_fs_data_stream_init(ctf_fs, &ctf_fs->data_stream);
+       ctf_fs_metadata_set_trace(ctf_fs);
+       ctf_fs_data_stream_open_streams(ctf_fs);
+       goto end;
+
+error:
+       ctf_fs_destroy_data(ctf_fs);
+end:
+       BT_PUT(value);
+       return ctf_fs;
 }
 
 BT_HIDDEN
@@ -89,6 +245,6 @@ end:
        return ret;
 error:
        (void) bt_component_set_private_data(source, NULL);
-        ctf_fs_destroy(ctf_fs);
+        ctf_fs_destroy_data(ctf_fs);
        return ret;
 }