added dict iterators

[swftools.git] / lib / q.h

diff --git a/lib/q.h b/lib/q.h
index 942bf28..3cf8cbe 100644 (file)
--- a/lib/q.h
+++ b/lib/q.h
@@ -120,6 +120,12 @@ typedef struct _heap
     int(*compare)(const void *, const void *);
 } heap_t;
 
+typedef struct _trie {
+    struct _trie*row[256];
+    unsigned char*rest;
+} trie_t;
+
+
 char* strdup_n(const char*str, int size);
 
 unsigned int crc32_add_byte(unsigned int crc32, unsigned char b);
@@ -184,6 +190,9 @@ void dict_free_all(dict_t*h, char free_keys, void (*free_data_function)(void*));
 void dict_clear(dict_t*h);
 void dict_destroy_shallow(dict_t*dict);
 void dict_destroy(dict_t*dict);
+#define DICT_ITERATE_DATA(d,t,v) int v##_i;dictentry_t*v##_e;t v;for(v##_i=0;v##_i<(d)->hashsize;v##_i++) for(v##_e=(d)->slots[v##_i]; v##_e && ((v=(t)v##_e->data)||1); v##_e=v##_e->next)
+#define DICT_ITERATE_KEY(d,t,v)  int v##_i;dictentry_t*v##_e;t v;for(v##_i=0;v##_i<(d)->hashsize;v##_i++) for(v##_e=(d)->slots[v##_i];v##_e && ((v=(t)v##_e->key)||1);v##_e=v##_e->next)
+#define DICT_ITERATE_ITEMS(d,t1,v1,t2,v2) int v1##_i;dictentry_t*v1##_e;t1 v1;t2 v2;for(v1##_i=0;v1##_i<(d)->hashsize;v1##_i++) for(v1##_e=(d)->slots[v1##_i]; v1##_e && (((v1=(t1)v1##_e->key)&&(v2=(t2)v1##_e->data))||1); v1##_e=v1##_e->next)
 
 void map_init(map_t*map);
 void map_put(map_t*map, string_t t1, string_t t2);
@@ -201,6 +210,9 @@ void* heap_chopmax(heap_t*h);
 void heap_dump(heap_t*h, FILE*fi);
 void** heap_flatten(heap_t*h);
 
+void trie_put(trie_t**t, unsigned const char*id);
+int trie_lookup(trie_t*t, unsigned const char*id);
+
 array_t* array_new();
 array_t* array_new2(type_t*type);
 void array_free(array_t*array);