Add type class/type structure management

[babeltrace.git] / types / array.c

diff --git a/types/array.c b/types/array.c
index d6a0c8b2bc552f622473e8babe165d51be659ad1..6ac4a1a1cbfab26b06ef5648380b48ceb1264d95 100644 (file)
--- a/types/array.c
+++ b/types/array.c
@@ -3,7 +3,7 @@
  *
  * BabelTrace - Array Type Converter
  *
- * Copyright 2010 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
+ * Copyright 2010, 2011 - 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
@@ -17,65 +17,105 @@
  */
 
 #include <babeltrace/compiler.h>
-#include <babeltrace/types.h>
+#include <babeltrace/format.h>
+
+static
+struct type *_array_type_new(struct type_class *type_class,
+                            struct declaration_scope *parent_scope);
+static
+void _array_type_free(struct type *type);
 
 void array_copy(struct stream_pos *dest, const struct format *fdest, 
                struct stream_pos *src, const struct format *fsrc,
-               const struct type_class *type_class)
+               struct type *type)
 {
-       struct type_class_array *array_class =
-               container_of(type_class, struct type_class_array, p);
-       unsigned int i;
+       struct type_sequence *array = container_of(type, struct type_array, p);
+       struct type_class_array *array_class = array->_class;
+       uint64_t i;
 
        fsrc->array_begin(src, array_class);
        fdest->array_begin(dest, array_class);
 
        for (i = 0; i < array_class->len; i++) {
-               struct type_class *elem_class = array_class->elem;
-               elem_class->copy(dest, fdest, src, fsrc, &elem_class->p);
+               struct type_class *elem_class = array->current_element.type;
+               elem_type->p._class->copy(dest, fdest, src, fsrc, elem_type);
        }
        fsrc->array_end(src, array_class);
        fdest->array_end(dest, array_class);
 }
 
-void array_type_free(struct type_class_array *array_class)
+static
+void _array_type_class_free(struct type_class *type_class)
 {
-       array_class->elem->free(&array_class->elem->p);
-       g_free(array_class);
-}
-
-static void _array_type_free(struct type_class *type_class)
-{
-       struct type_class_struct *array_class =
+       struct type_class_array *array_class =
                container_of(type_class, struct type_class_array, p);
-       array_type_free(array_class);
+
+       type_class_unref(array_class->elem);
+       g_free(array_class);
 }
 
-struct type_class_array *array_type_new(const char *name, size_t len,
-                                       struct type_class *elem)
+struct type_class_array *
+array_type_class_new(const char *name, size_t len, struct type_class *elem)
 {
        struct type_class_array *array_class;
+       struct type_class *type_class;
        int ret;
 
        array_class = g_new(struct type_class_array, 1);
-       type_class = &float_class->p;
-
+       type_class = &array_class->p;
        array_class->len = len;
+       type_ref(elem);
+       array_class->elem = elem;
        type_class->name = g_quark_from_string(name);
        /* No need to align the array, the first element will align itself */
        type_class->alignment = 1;
        type_class->copy = array_copy;
-       type_class->free = _array_type_free;
+       type_class->class_free = _array_type_class_free;
+       type_class->type_new = _array_type_new;
+       type_class->type_free = _array_type_free;
+       type_class->ref = 1;
 
        if (type_class->name) {
-               ret = ctf_register_type(type_class);
+               ret = register_type(type_class);
                if (ret)
                        goto error_register;
        }
        return array_class;
 
 error_register:
-       array_class->elem->free(&array_class->elem->p);
+       type_class_unref(array_class->elem);
        g_free(array_class);
        return NULL;
 }
+
+static
+struct type_array *_array_type_new(struct type_class *type_class,
+                                  struct declaration_scope *parent_scope)
+{
+       struct type_class_array *array_class =
+               container_of(type_class, struct type_class_array, p);
+       struct type_array *array;
+
+       array = g_new(struct type_array, 1);
+       type_class_ref(&array_class->p);
+       array->p._class = array_class;
+       array->p.ref = 1;
+       array->scope = new_declaration_scope(parent_scope);
+       array->current_element.type =
+               array_class->elem.p->type_new(&array_class->elem.p,
+                                                parent_scope);
+       return &array->p;
+}
+
+static
+void _array_type_free(struct type *type)
+{
+       struct type_array *array =
+               container_of(type, struct type_array, p);
+       struct type *elem_type = array->current_element.type;
+
+       elem_type->p._class->type_free(elem_type);
+       free_declaration_scope(array->scope);
+       type_class_unref(array->p._class);
+       g_free(array);
+}