From d55d5053298f6a26bd6ad95cfeed07bd856bbaa9 Mon Sep 17 00:00:00 2001
From: Al <albarrentine@gmail.com>
Date: Mon, 20 Jul 2015 16:29:48 -0400
Subject: [PATCH] [phrases] trie_get_data and trie_set_data interface for
 simpler dictionary-style trie get/set

---
 src/trie.c | 24 ++++++++++++++++++++++++
 src/trie.h |  4 ++++
 2 files changed, 28 insertions(+)

diff --git a/src/trie.c b/src/trie.c
index 51fd68eb..a1b96e5d 100644
--- a/src/trie.c
+++ b/src/trie.c
@@ -711,6 +711,30 @@ inline trie_data_node_t trie_get_data_node(trie_t *self, trie_node_t node) {
 inline bool trie_set_data_node(trie_t *self, uint32_t index, trie_data_node_t data_node) {
     if (self == NULL || self->data == NULL || index >= self->data->n) return false;
     self->data->a[index] = data_node;
+    return true;
+}
+
+inline bool trie_get_data(trie_t *self, char *key, uint32_t *data) {
+     uint32_t node_id = trie_get(self, key);
+     if (node_id == NULL_NODE_ID) return false;
+
+     trie_node_t node = trie_get_node(self, node_id);
+     trie_data_node_t data_node = trie_get_data_node(self, node);
+     *data = data_node.data;
+
+     return true;
+}
+
+inline bool trie_set_data(trie_t *self, char *key, uint32_t data) {
+     uint32_t node_id = trie_get(self, key);
+     if (node_id == NULL_NODE_ID) {
+        return trie_add(self, key, data);
+     }
+
+     trie_node_t node = trie_get_node(self, node_id);
+     trie_data_node_t data_node = trie_get_data_node(self, node);
+     data_node.data = data;
+     return trie_set_data_node(self, node_id, data_node);
 }
 
 trie_prefix_result_t trie_get_prefix_from_index(trie_t *self, char *key, size_t len, uint32_t start_index, size_t tail_pos) {
diff --git a/src/trie.h b/src/trie.h
index 51c7bab7..ac66f0d7 100644
--- a/src/trie.h
+++ b/src/trie.h
@@ -87,6 +87,10 @@ trie_node_t trie_get_free_list(trie_t *self);
 
 trie_data_node_t trie_get_data_node(trie_t *self, trie_node_t node);
 bool trie_set_data_node(trie_t *self, uint32_t index, trie_data_node_t data_node);
+
+bool trie_get_data(trie_t *self, char *key, uint32_t *data);
+bool trie_set_data(trie_t *self, char *key, uint32_t data);
+
 bool trie_tail_match(trie_t *self, char *str, uint32_t tail_index);
 
 uint32_t trie_add_transition(trie_t *self, uint32_t node_id, unsigned char c);