tommy/libpostal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[phrases] trie_add_suffix_at_index method

Browse Source
This commit is contained in:
Al
2015-06-04 02:30:53 -04:00
parent 9bdf118423
commit fa784677f2
2 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/trie.c
View File

@@ -663,20 +663,27 @@ bool trie_add(trie_t *self, char *key, uint32_t data) {
return trie_add_at_index(self, ROOT_NODE_ID, key, data); return trie_add_at_index(self, ROOT_NODE_ID, key, data);
} }
bool trie_add_suffix(trie_t *self, char *key, uint32_t data) { bool trie_add_suffix_at_index(trie_t *self, char *key, uint32_t start_node_id, uint32_t data) {
if (strlen(key) == 0) return false; if (start_node_id == NULL_NODE_ID || strlen(key) == 0) return false;
trie_node_t root = trie_get_root(self);
uint32_t node_id = trie_get_transition_index(self, root, '\0'); trie_node_t start_node = trie_get_node(self, start_node_id);
uint32_t node_id = trie_get_transition_index(self, start_node, '\0');
trie_node_t node = trie_get_node(self, node_id); trie_node_t node = trie_get_node(self, node_id);
if (node.check != ROOT_NODE_ID) { if (node.check != start_node_id) {
node_id = trie_add_transition(self, ROOT_NODE_ID, '\0'); node_id = trie_add_transition(self, start_node_id, '\0');
} }
char *suffix = utf8_reversed_string(key); char *suffix = utf8_reversed_string(key);
bool success = trie_add_at_index(self, node_id, suffix, data); bool success = trie_add_at_index(self, node_id, suffix, data);
free(suffix); free(suffix);
return success; return success;
}
inline bool trie_add_suffix(trie_t *self, char *key, uint32_t data) {
return trie_add_suffix_at_index(self, key, ROOT_NODE_ID, data);
} }
bool trie_compare_tail(trie_t *self, char *str, size_t len, size_t tail_index) { bool trie_compare_tail(trie_t *self, char *str, size_t len, size_t tail_index) {

1
src/trie.h
View File

@@ -100,6 +100,7 @@ void trie_tail_merge(trie_t *self, uint32_t old_node_id, unsigned char *suffix,
bool trie_add_at_index(trie_t *self, uint32_t node_id, char *key, uint32_t data); bool trie_add_at_index(trie_t *self, uint32_t node_id, char *key, uint32_t data);
bool trie_add(trie_t *self, char *key, uint32_t data); bool trie_add(trie_t *self, char *key, uint32_t data);
bool trie_add_suffix(trie_t *self, char *key, uint32_t data); bool trie_add_suffix(trie_t *self, char *key, uint32_t data);
bool trie_add_suffix_at_index(trie_t *self, char *key, uint32_t start_node_id, uint32_t data);
uint32_t trie_get_from_index(trie_t *self, char *word, size_t len, uint32_t i); uint32_t trie_get_from_index(trie_t *self, char *word, size_t len, uint32_t i);
uint32_t trie_get_len(trie_t *self, char *word, size_t len); uint32_t trie_get_len(trie_t *self, char *word, size_t len);