Update pretty-print

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

diff --git a/formats/ctf-text/types/variant.c b/formats/ctf-text/types/variant.c
index b4b72e16afeadb33420ec4998416dd8512402a34..e6abd499db9dcc0503468c60dc38f7dfd9a6d284 100644 (file)
--- a/formats/ctf-text/types/variant.c
+++ b/formats/ctf-text/types/variant.c
@@ -19,7 +19,32 @@
 #include <babeltrace/ctf-text/types.h>
 #include <stdio.h>
 
-int ctf_text_variant_write(struct stream_pos *pos, struct definition *definition)
+int ctf_text_variant_write(struct stream_pos *ppos, struct definition *definition)
 {
-       return variant_rw(pos, definition);
+       struct ctf_text_stream_pos *pos = ctf_text_pos(ppos);
+       int ret;
+
+       if (!pos->dummy) {
+               if (pos->depth >= 0) {
+                       //print_pos_tabs(pos);
+                       if (definition->index != 0 && definition->index != INT_MAX)
+                               fprintf(pos->fp, ",");
+                       if (definition->index != INT_MAX)
+                               fprintf(pos->fp, " ");
+                       if (pos->print_names)
+                               fprintf(pos->fp, "%s = ",
+                                       g_quark_to_string(definition->name));
+                       fprintf(pos->fp, "{");
+               }
+               pos->depth++;
+       }
+       ret = variant_rw(ppos, definition);
+       if (!pos->dummy) {
+               pos->depth--;
+               if (pos->depth >= 0) {
+                       //print_pos_tabs(pos);
+                       fprintf(pos->fp, " }");
+               }
+       }
+       return ret;
 }