Update pretty-print

[babeltrace.git] / formats / ctf-text / ctf-text.c

diff --git a/formats/ctf-text/ctf-text.c b/formats/ctf-text/ctf-text.c
index 2e9509e71cf281672f11c1a2afcbbbbfd001d8f7..ceca3e4ea6c2b579bd63f44e3f992a5a3052cb9b 100644 (file)
--- a/formats/ctf-text/ctf-text.c
+++ b/formats/ctf-text/ctf-text.c
@@ -18,6 +18,7 @@
 
 #include <babeltrace/format.h>
 #include <babeltrace/ctf-text/types.h>
+#include <babeltrace/ctf/metadata.h>
 #include <babeltrace/babeltrace.h>
 #include <inttypes.h>
 #include <uuid/uuid.h>
@@ -34,11 +35,6 @@
 struct trace_descriptor *ctf_text_open_trace(const char *path, int flags);
 void ctf_text_close_trace(struct trace_descriptor *descriptor);
 
-static
-rw_dispatch read_dispatch_table[] = {
-       /* All unimplemented */
-};
-
 static
 rw_dispatch write_dispatch_table[] = {
        [ CTF_TYPE_INTEGER ] = ctf_text_integer_write,
@@ -57,6 +53,96 @@ struct format ctf_text_format = {
        .close_trace = ctf_text_close_trace,
 };
 
+static
+int ctf_text_write_event(struct stream_pos *ppos,
+                        struct ctf_stream *stream_class)
+{
+       struct ctf_text_stream_pos *pos =
+               container_of(ppos, struct ctf_text_stream_pos, parent);
+       struct ctf_event *event_class;
+       uint64_t id = 0;
+       int len_index;
+       int ret;
+
+       /* print event header */
+       if (stream_class->event_header) {
+               /* lookup event id */
+               len_index = struct_declaration_lookup_field_index(stream_class->event_header_decl,
+                               g_quark_from_static_string("id"));
+               if (len_index >= 0) {
+                       struct definition_integer *defint;
+                       struct definition *field;
+
+                       field = struct_definition_get_field_from_index(stream_class->event_header, len_index);
+                       assert(field->declaration->id == CTF_TYPE_INTEGER);
+                       defint = container_of(field, struct definition_integer, p);
+                       assert(defint->declaration->signedness == FALSE);
+                       id = defint->value._unsigned;   /* set id */
+               }
+       }
+
+       if (id >= stream_class->events_by_id->len) {
+               fprintf(stdout, "[error] Event id %" PRIu64 " is outside range.\n", id);
+               return -EINVAL;
+       }
+       event_class = g_ptr_array_index(stream_class->events_by_id, id);
+       if (!event_class) {
+               fprintf(stdout, "[error] Event id %" PRIu64 " is unknown.\n", id);
+               return -EINVAL;
+       }
+
+       if (pos->print_names)
+               fprintf(pos->fp, "name = ");
+       fprintf(pos->fp, "%s", g_quark_to_string(event_class->name));
+
+       if (stream_class->event_header) {
+               if (pos->print_names)
+                       fprintf(pos->fp, ", stream.event.header =");
+               fprintf(pos->fp, " ");
+               ret = generic_rw(ppos, &stream_class->event_header->p);
+               if (ret)
+                       goto error;
+       }
+
+       /* print stream-declared event context */
+       if (stream_class->event_context) {
+               if (pos->print_names)
+                       fprintf(pos->fp, ", stream.event.context =");
+               fprintf(pos->fp, " ");
+               ret = generic_rw(ppos, &stream_class->event_context->p);
+               if (ret)
+                       goto error;
+       }
+
+       /* print event-declared event context */
+       if (event_class->context) {
+               if (pos->print_names)
+                       fprintf(pos->fp, ", event.context =");
+               fprintf(pos->fp, " ");
+               ret = generic_rw(ppos, &event_class->context->p);
+               if (ret)
+                       goto error;
+       }
+
+       /* Read and print event payload */
+       if (event_class->fields) {
+               if (pos->print_names)
+                       fprintf(pos->fp, ", event.fields =");
+               fprintf(pos->fp, " ");
+               ret = generic_rw(ppos, &event_class->fields->p);
+               if (ret)
+                       goto error;
+       }
+       fprintf(pos->fp, "\n");
+
+       return 0;
+
+error:
+       fprintf(stdout, "[error] Unexpected end of stream. Either the trace data stream is corrupted or metadata description does not match data layout.\n");
+       return ret;
+}
+
+
 struct trace_descriptor *ctf_text_open_trace(const char *path, int flags)
 {
        struct ctf_text_stream_pos *pos;
@@ -65,12 +151,17 @@ struct trace_descriptor *ctf_text_open_trace(const char *path, int flags)
        pos = g_new0(struct ctf_text_stream_pos, 1);
 
        switch (flags & O_ACCMODE) {
-       case O_WRONLY:
-               fp = fopen(path, "w");
+       case O_RDWR:
+               if (!path)
+                       fp = stdout;
+               else
+                       fp = fopen(path, "w");
                if (!fp)
                        goto error;
                pos->fp = fp;
                pos->parent.rw_table = write_dispatch_table;
+               pos->parent.event_cb = ctf_text_write_event;
+               pos->print_names = 1;
                break;
        case O_RDONLY:
        default: