X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=types%2Ftypes.c;h=2f262987f4d3a2b4997887ee96dc4d3be6cdda80;hp=961337800a1e45472a2abd175de21448726884d5;hb=de47353a173cf134d0bd50673520e243ebc29054;hpb=c054553dac076f91196b372fa19efaf2adc4e4f9

diff --git a/types/types.c b/types/types.c
index 96133780..2f262987 100644
--- a/types/types.c
+++ b/types/types.c
@@ -23,86 +23,266 @@
 #include <errno.h>
 
 static
-struct type_class *lookup_type_class_scope(GQuark qname,
-					   struct declaration_scope *scope)
+struct declaration *
+	lookup_type_declaration_scope(GQuark type_name, struct type_scope *scope)
 {
-	return g_hash_table_lookup(scope->type_classes,
-				   (gconstpointer) (unsigned long) qname);
+	return g_hash_table_lookup(scope->type_declarations,
+				   (gconstpointer) (unsigned long) type_name);
 }
 
-struct type_class *lookup_type_class(GQuark qname,
-				     struct declaration_scope *scope)
+struct declaration *lookup_type_declaration(GQuark type_name, struct type_scope *scope)
 {
-	struct type_class *tc;
+	struct declaration *declaration;
 
 	while (scope) {
-		tc = lookup_type_class_scope(qname, scope);
-		if (tc)
-			return tc;
+		declaration = lookup_type_declaration_scope(type_name, scope);
+		if (declaration)
+			return declaration;
 		scope = scope->parent_scope;
 	}
 	return NULL;
 }
 
-static void free_type_class(struct type_class *type_class)
+int register_type_declaration(GQuark name, struct declaration *declaration,
+			      struct type_scope *scope)
 {
-	type_class->class_free(type_class);
+	if (!name)
+		return -EPERM;
+
+	/* Only lookup in local scope */
+	if (lookup_type_declaration_scope(name, scope))
+		return -EEXIST;
+
+	g_hash_table_insert(scope->type_declarations,
+			    (gpointer) (unsigned long) name,
+			    declaration);
+	declaration_ref(declaration);
+	return 0;
 }
 
-static void free_type(struct type *type)
+static
+struct declaration *
+	lookup_field_declaration_scope(GQuark field_name, struct declaration_scope *scope)
 {
-	type->p.type_free(type);
+	return g_hash_table_lookup(scope->declarations,
+				   (gconstpointer) (unsigned long) field_name);
 }
 
-int register_type_class(struct type_class *type_class,
-			struct declaration_scope *scope)
+struct declaration *
+	lookup_field_declaration(GQuark field_name, struct declaration_scope *scope)
 {
+	struct declaration *declaration;
+
+	while (scope) {
+		declaration = lookup_field_declaration_scope(field_name, scope);
+		if (declaration)
+			return declaration;
+		scope = scope->parent_scope;
+	}
+	return NULL;
+}
+
+int register_field_declaration(GQuark field_name, struct declaration *declaration,
+			       struct declaration_scope *scope)
+{
+	if (!field_name)
+		return -EPERM;
+
 	/* Only lookup in local scope */
-	if (lookup_type_class_scope(type_class->name, scope))
+	if (lookup_field_declaration_scope(field_name, scope))
 		return -EEXIST;
 
-	g_hash_table_insert(scope->type_classes,
-			    (gpointer) (unsigned long) type_class->name,
-			    type_class);
+	g_hash_table_insert(scope->declarations,
+			    (gpointer) (unsigned long) field_name,
+			    declaration);
+	declaration_ref(declaration);
 	return 0;
 }
 
-void type_class_ref(struct type_class *type_class)
+void type_ref(struct type *type)
 {
-	type_class->ref++;
+	type->ref++;
 }
 
-void type_class_unref(struct type_class *type_class)
+void type_unref(struct type *type)
 {
-	if (!--type_class->ref)
-		free_type_class(type_class);
+	if (!--type->ref)
+		type->type_free(type);
 }
 
-void type_ref(struct type *type)
+void declaration_ref(struct declaration *declaration)
 {
-	type->ref++;
+	declaration->ref++;
 }
 
-void type_unref(struct type *type)
+void declaration_unref(struct declaration *declaration)
 {
-	if (!--type->ref)
-		free_type(type);
+	if (!--declaration->ref)
+		declaration->type->declaration_free(declaration);
+}
+
+struct type_scope *
+	new_type_scope(struct type_scope *parent_scope)
+{
+	struct type_scope *scope = g_new(struct type_scope, 1);
+
+	scope->type_declarations = g_hash_table_new_full(g_direct_hash,
+					g_direct_equal, NULL,
+					(GDestroyNotify) declaration_unref);
+	scope->struct_types = g_hash_table_new_full(g_direct_hash,
+					g_direct_equal, NULL,
+					(GDestroyNotify) type_unref);
+	scope->variant_types = g_hash_table_new_full(g_direct_hash,
+					g_direct_equal, NULL,
+					(GDestroyNotify) type_unref);
+	scope->enum_types = g_hash_table_new_full(g_direct_hash,
+					g_direct_equal, NULL,
+					(GDestroyNotify) type_unref);
+	scope->parent_scope = parent_scope;
+	return scope;
+}
+
+void free_type_scope(struct type_scope *scope)
+{
+	g_hash_table_destroy(scope->enum_types);
+	g_hash_table_destroy(scope->variant_types);
+	g_hash_table_destroy(scope->struct_types);
+	g_hash_table_destroy(scope->type_declarations);
+	g_free(scope);
+}
+
+static
+struct type_struct *lookup_struct_type_scope(GQuark struct_name,
+					     struct type_scope *scope)
+{
+	return g_hash_table_lookup(scope->struct_types,
+				   (gconstpointer) (unsigned long) struct_name);
+}
+
+struct type_struct *lookup_struct_type(GQuark struct_name,
+				       struct type_scope *scope)
+{
+	struct type_struct *type;
+
+	while (scope) {
+		type = lookup_struct_type_scope(struct_name, scope);
+		if (type)
+			return type;
+		scope = scope->parent_scope;
+	}
+	return NULL;
+}
+
+int register_struct_type(GQuark struct_name, struct type_struct *struct_type,
+			 struct type_scope *scope)
+{
+	if (!struct_name)
+		return -EPERM;
+
+	/* Only lookup in local scope */
+	if (lookup_struct_type_scope(struct_name, scope))
+		return -EEXIST;
+
+	g_hash_table_insert(scope->struct_types,
+			    (gpointer) (unsigned long) struct_name,
+			    struct_type);
+	type_ref(&struct_type->p);
+	return 0;
+}
+
+static
+struct type_variant *lookup_variant_type_scope(GQuark variant_name,
+					       struct type_scope *scope)
+{
+	return g_hash_table_lookup(scope->variant_types,
+				   (gconstpointer) (unsigned long) variant_name);
+}
+
+struct type_variant *lookup_variant_type(GQuark variant_name,
+					 struct type_scope *scope)
+{
+	struct type_variant *type;
+
+	while (scope) {
+		type = lookup_variant_type_scope(variant_name, scope);
+		if (type)
+			return type;
+		scope = scope->parent_scope;
+	}
+	return NULL;
+}
+
+int register_variant_type(GQuark variant_name,
+			  struct type_variant *variant_type,
+		          struct type_scope *scope)
+{
+	if (!variant_name)
+		return -EPERM;
+
+	/* Only lookup in local scope */
+	if (lookup_variant_type_scope(variant_name, scope))
+		return -EEXIST;
+
+	g_hash_table_insert(scope->variant_types,
+			    (gpointer) (unsigned long) variant_name,
+			    variant_type);
+	type_ref(&variant_type->p);
+	return 0;
+}
+
+static
+struct type_enum *lookup_enum_type_scope(GQuark enum_name,
+					 struct type_scope *scope)
+{
+	return g_hash_table_lookup(scope->enum_types,
+				   (gconstpointer) (unsigned long) enum_name);
+}
+
+struct type_enum *lookup_enum_type(GQuark enum_name,
+				   struct type_scope *scope)
+{
+	struct type_enum *type;
+
+	while (scope) {
+		type = lookup_enum_type_scope(enum_name, scope);
+		if (type)
+			return type;
+		scope = scope->parent_scope;
+	}
+	return NULL;
+}
+
+int register_enum_type(GQuark enum_name, struct type_enum *enum_type,
+		       struct type_scope *scope)
+{
+	if (!enum_name)
+		return -EPERM;
+
+	/* Only lookup in local scope */
+	if (lookup_enum_type_scope(enum_name, scope))
+		return -EEXIST;
+
+	g_hash_table_insert(scope->enum_types,
+			    (gpointer) (unsigned long) enum_name,
+			    enum_type);
+	type_ref(&enum_type->p);
+	return 0;
 }
 
 struct declaration_scope *
-new_declaration_scope(struct declaration_scope *parent_scope)
+	new_declaration_scope(struct declaration_scope *parent_scope)
 {
 	struct declaration_scope *scope = g_new(struct declaration_scope, 1);
 
-	scope->type_classes = g_hash_table_new_full(g_direct_hash,
+	scope->declarations = g_hash_table_new_full(g_direct_hash,
 					g_direct_equal, NULL,
-					(GDestroyNotify) type_class_unref);
+					(GDestroyNotify) declaration_unref);
 	scope->parent_scope = parent_scope;
 	return scope;
 }
 
 void free_declaration_scope(struct declaration_scope *scope)
 {
-	g_hash_table_destroy(scope->type_classes);
+	g_hash_table_destroy(scope->declarations);
 	g_free(scope);
 }