Cleanup: update ifdef wrapper name
[babeltrace.git] / types / array.c
index 7b1d92a5747d54b23b0a1366f0cb5a127159d1a8..047adac21902c7d7f438fc14a1e55099181280f1 100644 (file)
@@ -3,7 +3,9 @@
  *
  * BabelTrace - Array Type Converter
  *
- * Copyright 2010, 2011 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
+ * Copyright 2010-2011 EfficiOS Inc. and Linux Foundation
+ *
+ * Author: Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
  *
  * Permission is hereby granted, free of charge, to any person obtaining a copy
  * of this software and associated documentation files (the "Software"), to deal
 
 #include <babeltrace/compiler.h>
 #include <babeltrace/format.h>
+#include <babeltrace/types.h>
+#include <inttypes.h>
 
 static
 struct definition *_array_definition_new(struct declaration *declaration,
-                       struct definition_scope *parent_scope);
+                       struct definition_scope *parent_scope,
+                       GQuark field_name, int index, const char *root_name);
 static
 void _array_definition_free(struct definition *definition);
 
-void array_copy(struct stream_pos *dest, const struct format *fdest, 
-               struct stream_pos *src, const struct format *fsrc,
-               struct definition *definition)
+int array_rw(struct stream_pos *pos, struct definition *definition)
 {
-       struct definition_array *array =
+       struct definition_array *array_definition =
                container_of(definition, struct definition_array, p);
-       struct declaration_array *array_declaration = array->declaration;
+       const struct declaration_array *array_declaration =
+               array_definition->declaration;
        uint64_t i;
+       int ret;
 
-       fsrc->array_begin(src, array_declaration);
-       fdest->array_begin(dest, array_declaration);
-
+       /* No need to align, because the first field will align itself. */
        for (i = 0; i < array_declaration->len; i++) {
-               struct definition *elem = array->current_element.definition;
-               elem->declaration->copy(dest, fdest, src, fsrc, elem);
+               struct definition *field =
+                       g_ptr_array_index(array_definition->elems, i);
+               ret = generic_rw(pos, field);
+               if (ret)
+                       return ret;
        }
-       fsrc->array_end(src, array_declaration);
-       fdest->array_end(dest, array_declaration);
+       return 0;
 }
 
 static
@@ -57,7 +62,7 @@ void _array_declaration_free(struct declaration *declaration)
 }
 
 struct declaration_array *
-       array_declaration_new(const char *name, size_t len,
+       array_declaration_new(size_t len,
                              struct declaration *elem_declaration,
                              struct declaration_scope *parent_scope)
 {
@@ -71,10 +76,8 @@ struct declaration_array *
        array_declaration->elem = elem_declaration;
        array_declaration->scope = new_declaration_scope(parent_scope);
        declaration->id = CTF_TYPE_ARRAY;
-       declaration->name = g_quark_from_string(name);
        /* No need to align the array, the first element will align itself */
        declaration->alignment = 1;
-       declaration->copy = array_copy;
        declaration->declaration_free = _array_declaration_free;
        declaration->definition_new = _array_definition_new;
        declaration->definition_free = _array_definition_free;
@@ -85,22 +88,84 @@ struct declaration_array *
 static
 struct definition *
        _array_definition_new(struct declaration *declaration,
-                             struct definition_scope *parent_scope)
+                             struct definition_scope *parent_scope,
+                             GQuark field_name, int index, const char *root_name)
 {
        struct declaration_array *array_declaration =
                container_of(declaration, struct declaration_array, p);
        struct definition_array *array;
+       int ret;
+       int i;
 
        array = g_new(struct definition_array, 1);
        declaration_ref(&array_declaration->p);
        array->p.declaration = declaration;
        array->declaration = array_declaration;
        array->p.ref = 1;
-       array->scope = new_definition_scope(parent_scope);
-       array->current_element.definition =
-               array_declaration->elem->definition_new(array_declaration->elem,
-                                                 parent_scope);
+       /*
+        * Use INT_MAX order to ensure that all fields of the parent
+        * scope are seen as being prior to this scope.
+        */
+       array->p.index = root_name ? INT_MAX : index;
+       array->p.name = field_name;
+       array->p.path = new_definition_path(parent_scope, field_name, root_name);
+       array->p.scope = new_definition_scope(parent_scope, field_name, root_name);
+       ret = register_field_definition(field_name, &array->p,
+                                       parent_scope);
+       assert(!ret);
+       array->string = NULL;
+       array->elems = NULL;
+
+       if (array_declaration->elem->id == CTF_TYPE_INTEGER) {
+               struct declaration_integer *integer_declaration =
+                       container_of(array_declaration->elem, struct declaration_integer, p);
+
+               if (integer_declaration->encoding == CTF_STRING_UTF8
+                     || integer_declaration->encoding == CTF_STRING_ASCII) {
+
+                       array->string = g_string_new("");
+
+                       if (integer_declaration->len == CHAR_BIT
+                           && integer_declaration->p.alignment == CHAR_BIT) {
+                               return &array->p;
+                       }
+               }
+       }
+
+       array->elems = g_ptr_array_sized_new(array_declaration->len);
+       g_ptr_array_set_size(array->elems, array_declaration->len);
+       for (i = 0; i < array_declaration->len; i++) {
+               struct definition **field;
+               GString *str;
+               GQuark name;
+
+               str = g_string_new("");
+               g_string_printf(str, "[%u]", (unsigned int) i);
+               name = g_quark_from_string(str->str);
+               (void) g_string_free(str, TRUE);
+
+               field = (struct definition **) &g_ptr_array_index(array->elems, i);
+               *field = array_declaration->elem->definition_new(array_declaration->elem,
+                                         array->p.scope,
+                                         name, i, NULL);
+               if (!*field)
+                       goto error;
+       }
+
        return &array->p;
+
+error:
+       for (i--; i >= 0; i--) {
+               struct definition *field;
+
+               field = g_ptr_array_index(array->elems, i);
+               field->declaration->definition_free(field);
+       }
+       (void) g_ptr_array_free(array->elems, TRUE);
+       free_definition_scope(array->p.scope);
+       declaration_unref(array->p.declaration);
+       g_free(array);
+       return NULL;
 }
 
 static
@@ -108,11 +173,74 @@ void _array_definition_free(struct definition *definition)
 {
        struct definition_array *array =
                container_of(definition, struct definition_array, p);
-       struct definition *elem_definition =
-               array->current_element.definition;
+       uint64_t i;
 
-       elem_definition->declaration->definition_free(elem_definition);
-       free_definition_scope(array->scope);
+       if (array->string)
+               (void) g_string_free(array->string, TRUE);
+       if (array->elems) {
+               for (i = 0; i < array->elems->len; i++) {
+                       struct definition *field;
+
+                       field = g_ptr_array_index(array->elems, i);
+                       field->declaration->definition_free(field);
+               }
+               (void) g_ptr_array_free(array->elems, TRUE);
+       }
+       free_definition_scope(array->p.scope);
        declaration_unref(array->p.declaration);
        g_free(array);
 }
+
+uint64_t array_len(struct definition_array *array)
+{
+       if (!array->elems)
+               return array->string->len;
+       return array->elems->len;
+}
+
+struct definition *array_index(struct definition_array *array, uint64_t i)
+{
+       if (!array->elems)
+               return NULL;
+       if (i >= array->elems->len)
+               return NULL;
+       return g_ptr_array_index(array->elems, i);
+}
+
+int get_array_len(const struct definition *field)
+{
+       struct definition_array *array_definition;
+       struct declaration_array *array_declaration;
+
+       array_definition = container_of(field, struct definition_array, p);
+       array_declaration = array_definition->declaration;
+
+       return array_declaration->len;
+}
+
+GString *get_char_array(const struct definition *field)
+{
+       struct definition_array *array_definition;
+       struct declaration_array *array_declaration;
+       struct declaration *elem;
+
+       array_definition = container_of(field, struct definition_array, p);
+       array_declaration = array_definition->declaration;
+       elem = array_declaration->elem;
+       if (elem->id == CTF_TYPE_INTEGER) {
+               struct declaration_integer *integer_declaration =
+                       container_of(elem, struct declaration_integer, p);
+
+               if (integer_declaration->encoding == CTF_STRING_UTF8
+                               || integer_declaration->encoding == CTF_STRING_ASCII) {
+
+                       if (integer_declaration->len == CHAR_BIT
+                                       && integer_declaration->p.alignment == CHAR_BIT) {
+
+                               return array_definition->string;
+                       }
+               }
+       }
+       fprintf(stderr, "[warning] Extracting string\n");
+       return NULL;
+}
This page took 0.026258 seconds and 4 git commands to generate.