+/*
+ * BabelTrace - Enumeration Type
+ *
+ * Copyright (c) 2010 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+
+#include <babeltrace/compiler.h>
+#include <babeltrace/types.h>
+#include <stdint.h>
+#include <glib.h>
+
+struct enum_table {
+ GHashTable *value_to_quark; /* Tuples (value, GQuark) */
+ GHashTable *quark_to_value; /* Tuples (GQuark, value) */
+};
+
+#if (__WORDSIZE == 32)
+GQuark enum_uint_to_quark(const struct type_class_enum *enum_class, uint64_t v)
+{
+ gconstpointer q = g_hash_table_lookup(enum_class->table.value_to_quark,
+ &v);
+ return (GQuark) (unsigned long) q;
+}
+
+GQuark enum_int_to_quark(const struct type_class_enum *enum_class, uint64_t v)
+{
+ gconstpointer q = g_hash_table_lookup(enum_class->table.value_to_quark,
+ &v);
+ return (GQuark) (unsigned long) q;
+}
+
+uint64_t enum_quark_to_uint(const struct type_class_enum *enum_class,
+ size_t len, int byte_order, GQuark q)
+{
+ gconstpointer v = g_hash_table_lookup(enum_class->table.quark_to_value,
+ (gconstpointer) q);
+ return *(const uint64_t *) v;
+}
+
+int64_t enum_quark_to_int(const struct type_class_enum *enum_class,
+ size_t len, int byte_order, GQuark q)
+{
+ gconstpointer v = g_hash_table_lookup(enum_class->table.quark_to_value,
+ (gconstpointer) q);
+ return *(const int64_t *) v;
+}
+
+guint enum_val_hash(gconstpointer key)
+{
+ int64_t ukey = *(const int64_t *)key;
+
+ return (guint)ukey ^ (guint)(ukey >> 32);
+}
+
+gboolean enum_val_equal(gconstpointer a, gconstpointer b)
+{
+ int64_t ua = *(const int64_t *)a;
+ int64_t ub = *(const int64_t *)b;
+
+ return ua == ub;
+}
+
+void enum_val_free(void *ptr)
+{
+ g_free(ptr);
+}
+
+void enum_signed_insert(struct type_class_enum *enum_class, int64_t v, GQuark q)
+{
+ int64_t *valuep = g_new(int64_t, 1);
+
+ g_hash_table_insert(enum_class->table>value_to_quark, valuep,
+ (gpointer) (unsigned long) q);
+ g_hash_table_insert(enum_class->table.quark_to_value,
+ (gpointer) (unsigned long) q,
+ valuep);
+}
+
+void enum_unsigned_insert(struct type_class_enum *enum_class, uint64_t v, GQuark q)
+{
+ uint64_t *valuep = g_new(uint64_t, 1);
+
+ g_hash_table_insert(enum_class->table.value_to_quark, valuep,
+ (gpointer) (unsigned long) q);
+ g_hash_table_insert(enum_class->table.quark_to_value,
+ (gpointer) (unsigned long) q,
+ valuep);
+}
+#else /* __WORDSIZE != 32 */
+GQuark enum_uint_to_quark(const struct type_class_enum *enum_class, uint64_t v)
+{
+ gconstpointer q = g_hash_table_lookup(enum_class->table.value_to_quark,
+ (gconstpointer) v);
+ return (GQuark) (unsigned long) q;
+}
+
+GQuark enum_int_to_quark(const struct type_class_enum *enum_class, uint64_t v)
+{
+ gconstpointer q = g_hash_table_lookup(enum_class->table.value_to_quark,
+ (gconstpointer) v);
+ return (GQuark) (unsigned long) q;
+}
+
+uint64_t enum_quark_to_uint(const struct type_class_enum *enum_class,
+ size_t len, int byte_order, GQuark q)
+{
+ gconstpointer v = g_hash_table_lookup(enum_class->table.quark_to_value,
+ (gconstpointer) (unsigned long) q);
+ return *(const uint64_t *) v;
+}
+
+int64_t enum_quark_to_int(const struct type_class_enum *enum_class,
+ size_t len, int byte_order, GQuark q)
+{
+ gconstpointer v = g_hash_table_lookup(enum_class->table.quark_to_value,
+ (gconstpointer) (unsigned long) q);
+ return *(const int64_t *) v;
+}
+
+guint enum_val_hash(gconstpointer key)
+{
+ return g_direct_hash(key);
+}
+
+gboolean enum_val_equal(gconstpointer a, gconstpointer b)
+{
+ return g_direct_equal(a, b);
+}
+
+void enum_val_free(void *ptr)
+{
+}
+
+void enum_signed_insert(struct type_class_enum *enum_class,
+ int64_t v, GQuark q)
+{
+ g_hash_table_insert(enum_class->table.value_to_quark, (gpointer) v,
+ (gpointer) (unsigned long) q);
+ g_hash_table_insert(enum_class->table.quark_to_value,
+ (gpointer) (unsigned long) q,
+ valuep);
+}
+
+void enum_unsigned_insert(struct type_class_enum *enum_class,
+ uint64_t v, GQuark q)
+{
+ g_hash_table_insert(enum_class->table.value_to_quark, (gpointer) v,
+ (gpointer) (unsigned long) q);
+ g_hash_table_insert(enum_class->table.quark_to_value,
+ (gpointer) (unsigned long) q,
+ valuep);
+}
+#endif /* __WORDSIZE != 32 */
+
+size_t enum_copy(unsigned char *dest, const struct format *fdest,
+ const unsigned char *src, const struct format *fsrc,
+ const struct type_class *type_class)
+{
+ struct type_class_enum *enum_class =
+ container_of(type_class, struct type_class_enum, p);
+ struct type_class_bitfield *bitfield_class = &enum_class->p;
+ struct type_class_integer *int_class = &bitfield_class->p;
+ GQuark v;
+
+ v = fsrc->enum_read(src, enum_class)
+ return fdest->enum_write(dest, enum_class, v);
+}
+
+struct type_class_enum *enum_type_new(const char *name,
+ size_t start_offset,
+ size_t len, int byte_order,
+ int signedness,
+ size_t alignment)
+{
+ struct type_class_bitfield *bitfield_class;
+ struct type_class_integer *int_class;
+ int ret;
+
+ enum_class = g_new(struct type_class_bitfield, 1);
+ enum_class->table.value_to_quark = g_hash_table(enum_val_hash,
+ enum_val_equal);
+ enum_class->table.quark_to_value = g_hash_table_new_full(g_direct_hash,
+ g_direct_equal,
+ NULL, enum_val_free);
+ bitfield_class = &enum_class->p;
+ bitfield_class->start_offset = start_offset;
+ int_class = &bitfield_class->p;
+ int_class->p.name = g_quark_from_string(name);
+ int_class->p.alignment = alignment;
+ int_class->len = len;
+ int_class->byte_order = byte_order;
+ int_class->signedness = signedness;
+ bitfield_class->start_offset = start_offset;
+ if (int_class->p.name) {
+ ret = register_type(&int_class->p);
+ if (ret) {
+ g_hash_table_destroy(enum_class->table.value_to_quark);
+ g_hash_table_destroy(enum_class->table.quark_to_value);
+ g_free(enum_class);
+ return NULL;
+ }
+ }
+ return enum_class;
+}
+
+void enum_type_free(struct type_class_enum *enum_class)
+{
+ g_hash_table_destroy(enum_class->table.value_to_quark);
+ g_hash_table_destroy(enum_class->table.quark_to_value);
+ g_free(enum_class);
+}