X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=types%2Farray.c;h=047adac21902c7d7f438fc14a1e55099181280f1;hp=01483b520017e4fbfb84fd5800fbb029babdae46;hb=f133896d405417ba90ac250d243d4b8e539e98f1;hpb=be85c1c7633af07e35db7b415d6ee8447c30e80a

diff --git a/types/array.c b/types/array.c
index 01483b52..047adac2 100644
--- a/types/array.c
+++ b/types/array.c
@@ -3,7 +3,9 @@
  *
  * BabelTrace - Array Type Converter
  *
- * Copyright 2010 - 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
@@ -18,68 +20,227 @@
 
 #include <babeltrace/compiler.h>
 #include <babeltrace/format.h>
+#include <babeltrace/types.h>
+#include <inttypes.h>
 
-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_class_array *array_class =
-		container_of(type_class, struct type_class_array, p);
-	unsigned int i;
+static
+struct definition *_array_definition_new(struct declaration *declaration,
+			struct definition_scope *parent_scope,
+			GQuark field_name, int index, const char *root_name);
+static
+void _array_definition_free(struct definition *definition);
 
-	fsrc->array_begin(src, array_class);
-	fdest->array_begin(dest, array_class);
+int array_rw(struct stream_pos *pos, struct definition *definition)
+{
+	struct definition_array *array_definition =
+		container_of(definition, struct definition_array, p);
+	const struct declaration_array *array_declaration =
+		array_definition->declaration;
+	uint64_t i;
+	int ret;
 
-	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);
+	/* No need to align, because the first field will align itself. */
+	for (i = 0; i < array_declaration->len; i++) {
+		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_class);
-	fdest->array_end(dest, array_class);
+	return 0;
 }
 
-void array_type_free(struct type_class_array *array_class)
+static
+void _array_declaration_free(struct declaration *declaration)
 {
-	type_unref(array_class->elem);
-	g_free(array_class);
+	struct declaration_array *array_declaration =
+		container_of(declaration, struct declaration_array, p);
+
+	free_declaration_scope(array_declaration->scope);
+	declaration_unref(array_declaration->elem);
+	g_free(array_declaration);
 }
 
-static void _array_type_free(struct type_class *type_class)
+struct declaration_array *
+	array_declaration_new(size_t len,
+			      struct declaration *elem_declaration,
+			      struct declaration_scope *parent_scope)
 {
-	struct type_class_array *array_class =
-		container_of(type_class, struct type_class_array, p);
-	array_type_free(array_class);
+	struct declaration_array *array_declaration;
+	struct declaration *declaration;
+
+	array_declaration = g_new(struct declaration_array, 1);
+	declaration = &array_declaration->p;
+	array_declaration->len = len;
+	declaration_ref(elem_declaration);
+	array_declaration->elem = elem_declaration;
+	array_declaration->scope = new_declaration_scope(parent_scope);
+	declaration->id = CTF_TYPE_ARRAY;
+	/* No need to align the array, the first element will align itself */
+	declaration->alignment = 1;
+	declaration->declaration_free = _array_declaration_free;
+	declaration->definition_new = _array_definition_new;
+	declaration->definition_free = _array_definition_free;
+	declaration->ref = 1;
+	return array_declaration;
 }
 
-struct type_class_array *array_type_new(const char *name, size_t len,
-					struct type_class *elem)
+static
+struct definition *
+	_array_definition_new(struct declaration *declaration,
+			      struct definition_scope *parent_scope,
+			      GQuark field_name, int index, const char *root_name)
 {
-	struct type_class_array *array_class;
-	struct type_class *type_class;
+	struct declaration_array *array_declaration =
+		container_of(declaration, struct declaration_array, p);
+	struct definition_array *array;
 	int ret;
+	int i;
 
-	array_class = g_new(struct type_class_array, 1);
-	type_class = &array_class->p;
+	array = g_new(struct definition_array, 1);
+	declaration_ref(&array_declaration->p);
+	array->p.declaration = declaration;
+	array->declaration = array_declaration;
+	array->p.ref = 1;
+	/*
+	 * 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;
 
-	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->ref = 1;
+	if (array_declaration->elem->id == CTF_TYPE_INTEGER) {
+		struct declaration_integer *integer_declaration =
+			container_of(array_declaration->elem, struct declaration_integer, p);
 
-	if (type_class->name) {
-		ret = register_type(type_class);
-		if (ret)
-			goto error_register;
+		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;
+			}
+		}
 	}
-	return array_class;
 
-error_register:
-	type_unref(array_class->elem);
-	g_free(array_class);
+	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
+void _array_definition_free(struct definition *definition)
+{
+	struct definition_array *array =
+		container_of(definition, struct definition_array, p);
+	uint64_t i;
+
+	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;
 }