X-Git-Url: https://git.efficios.com/?a=blobdiff_plain;f=types%2Ftypes.c;h=4b614f39978e49c6a41d9869e6f60fd251d9bc76;hb=64893f33bdc4bfe20820928b28731277797e41fc;hp=91d6b211d118acaafe50a0a9f29bc5f1b12de329;hpb=ccd7e1c86f36342b0b06651cc52df86bb663c271;p=babeltrace.git

diff --git a/types/types.c b/types/types.c
index 91d6b211..4b614f39 100644
--- a/types/types.c
+++ b/types/types.c
@@ -5,7 +5,7 @@
  *
  * Types registry.
  *
- * 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
@@ -18,49 +18,142 @@
  * all copies or substantial portions of the Software.
  */
 
-#include <ctf/ctf-types.h>
+#include <babeltrace/format.h>
 #include <glib.h>
 #include <errno.h>
 
-/*
- * Type hash table contains the registered types. Type registration is typically
- * performed by a type plugin.
- * TODO: support plugin unload (unregistration of types).
- */
-GHashTable *types;
-
-struct type_class *ctf_lookup_type(GQuark qname)
+static
+struct type *
+	lookup_type_scope(GQuark type_name, struct type_scope *scope)
 {
-	return g_hash_table_lookup(type_classes,
-				   (gconstpointer) (unsigned long) qname)
+	return g_hash_table_lookup(scope->types,
+				   (gconstpointer) (unsigned long) type_name);
 }
 
-static void free_type(struct type_class *type_class)
+struct type *lookup_type(GQuark type_name, struct type_scope *scope)
 {
-	type_class->free(type_class);
+	struct type *type;
+
+	while (scope) {
+		type = lookup_type_scope(type_name, scope);
+		if (type)
+			return type;
+		scope = scope->parent_scope;
+	}
+	return NULL;
 }
 
-int ctf_register_type(struct type_class *type_class)
+int register_type(GQuark name, struct type *type, struct type_scope *scope)
 {
-	if (ctf_lookup_type_class(type_class->name))
+	g_assert(name == type->name);
+
+	if (!name)
+		return -EPERM;
+
+	/* Only lookup in local scope */
+	if (lookup_type_scope(name, scope))
 		return -EEXIST;
 
-	g_hash_table_insert(type_classes,
-			    (gconstpointer) (unsigned long) type_class->name,
-			    type_class);
+	g_hash_table_insert(scope->types,
+			    (gpointer) (unsigned long) name,
+			    type);
+	type_ref(type);
 	return 0;
 }
 
-int ctf_init_types(void)
+static
+struct declaration *
+	lookup_declaration_scope(GQuark field_name, struct declaration_scope *scope)
+{
+	return g_hash_table_lookup(scope->declarations,
+				   (gconstpointer) (unsigned long) field_name);
+}
+
+struct declaration *
+	lookup_declaration(GQuark field_name, struct declaration_scope *scope)
+{
+	struct declaration *declaration;
+
+	while (scope) {
+		declaration = lookup_declaration_scope(field_name, scope);
+		if (declaration)
+			return declaration;
+		scope = scope->parent_scope;
+	}
+	return NULL;
+}
+
+int register_declaration(GQuark field_name, struct declaration *declaration,
+			 struct declaration_scope *scope)
 {
-	type_classes = g_hash_table_new_full(g_direct_hash, g_direct_equal,
-					     NULL, free_type);
-	if (!type_classes)
-		return -ENOMEM;
+	if (!field_name)
+		return -EPERM;
+
+	/* Only lookup in local scope */
+	if (lookup_declaration_scope(field_name, scope))
+		return -EEXIST;
+
+	g_hash_table_insert(scope->declarations,
+			    (gpointer) (unsigned long) field_name,
+			    declaration);
+	declaration_ref(declaration);
 	return 0;
 }
 
-int ctf_finalize_types(void)
+void type_ref(struct type *type)
+{
+	type->ref++;
+}
+
+void type_unref(struct type *type)
+{
+	if (!--type->ref)
+		type->type_free(type);
+}
+
+void declaration_ref(struct declaration *declaration)
+{
+	declaration->ref++;
+}
+
+void declaration_unref(struct declaration *declaration)
+{
+	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->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->types);
+	g_free(scope);
+}
+
+struct declaration_scope *
+	new_declaration_scope(struct declaration_scope *parent_scope)
+{
+	struct declaration_scope *scope = g_new(struct declaration_scope, 1);
+
+	scope->declarations = g_hash_table_new_full(g_direct_hash,
+					g_direct_equal, NULL,
+					(GDestroyNotify) declaration_unref);
+	scope->parent_scope = parent_scope;
+	return scope;
+}
+
+void free_declaration_scope(struct declaration_scope *scope)
 {
-	g_hash_table_destroy(type_classes);
+	g_hash_table_destroy(scope->declarations);
+	g_free(scope);
 }