X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=formats%2Fctf%2Ftypes%2Fenum.c;h=e4e7ac85bb72f094d1ee197e7276cfbe95aa5493;hp=a1d034c38b79f39903e2a593e9aed3e1e68b31ee;hb=c462e188f3e7819c7bc74f671038cdbf36e8c3c0;hpb=ccd7e1c86f36342b0b06651cc52df86bb663c271

diff --git a/formats/ctf/types/enum.c b/formats/ctf/types/enum.c
index a1d034c3..e4e7ac85 100644
--- a/formats/ctf/types/enum.c
+++ b/formats/ctf/types/enum.c
@@ -3,7 +3,9 @@
  *
  * Enumeration mapping strings (quarks) from/to integers.
  *
- * 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
@@ -14,45 +16,65 @@
  *
  * The above copyright notice and this permission notice shall be included in
  * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
  */
 
 #include <babeltrace/ctf/types.h>
+#include <inttypes.h>
 #include <stdint.h>
 #include <glib.h>
 
-GQuark ctf_enum_read(struct stream_pos *pos,
-		     const struct type_class_enum *src)
+int ctf_enum_read(struct stream_pos *ppos, struct definition *definition)
 {
-	struct type_class_integer *int_class = &src->p;
+	struct definition_enum *enum_definition =
+		container_of(definition, struct definition_enum, p);
+	const struct declaration_enum *enum_declaration =
+		enum_definition->declaration;
+	struct definition_integer *integer_definition =
+		enum_definition->integer;
+	const struct declaration_integer *integer_declaration =
+		integer_definition->declaration;
+	GArray *qs;
+	int ret;
 
-	if (!int_class->signedness) {
-		uint64_t v;
-
-		v = ctf_uint_read(pos, int_class);
-		return enum_uint_to_quark(src, v);
+	ret = ctf_integer_read(ppos, &integer_definition->p);
+	if (ret)
+		return ret;
+	if (!integer_declaration->signedness) {
+		qs = enum_uint_to_quark_set(enum_declaration,
+			integer_definition->value._unsigned);
+		if (!qs) {
+			fprintf(stderr, "[warning] Unknown value %" PRIu64 " in enum.\n",
+				integer_definition->value._unsigned);
+		}
 	} else {
-		int64_t v;
-
-		v = fsrc->ctf_int_read(pos, int_class);
-		return enum_int_to_quark(src, v);
+		qs = enum_int_to_quark_set(enum_declaration,
+			integer_definition->value._signed);
+		if (!qs) {
+			fprintf(stderr, "[warning] Unknown value %" PRId64 " in enum.\n",
+				integer_definition->value._signed);
+		}
 	}
+	/* unref previous quark set */
+	if (enum_definition->value)
+		g_array_unref(enum_definition->value);
+	enum_definition->value = qs;
+	return 0;
 }
 
-size_t ctf_enum_write(struct stream_pos *pos,
-		      const struct type_class_enum *dest,
-		      GQuark q)
+int ctf_enum_write(struct stream_pos *pos, struct definition *definition)
 {
-	struct type_class_integer *int_class = &dest->p;
+	struct definition_enum *enum_definition =
+		container_of(definition, struct definition_enum, p);
+	struct definition_integer *integer_definition =
+		enum_definition->integer;
 
-	if (!int_class->signedness) {
-		uint64_t v;
-
-		v = enum_quark_to_uint(dest, q);
-		return ctf_uint_write(pos, int_class, v);
-	} else {
-		int64_t v;
-
-		v = enum_quark_to_int(dest, q);
-		return ctf_int_write(pos, int_class, v);
-	}
+	return ctf_integer_write(pos, &integer_definition->p);
 }