X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/a5338b60e111972364a8bc6f07011c6defd213d2..f89517343bf062d299d54408eea2f9387bfefb6d:/lunaix-os/includes/lunaix/ds/btrie.h?ds=sidebyside

diff --git a/lunaix-os/includes/lunaix/ds/btrie.h b/lunaix-os/includes/lunaix/ds/btrie.h
index ad3d04c..d24cf84 100644
--- a/lunaix-os/includes/lunaix/ds/btrie.h
+++ b/lunaix-os/includes/lunaix/ds/btrie.h
@@ -6,33 +6,49 @@
 
 #define BTRIE_BITS 4
 
+/**
+ * key-sorted prefix tree
+ */
 struct btrie
 {
     struct btrie_node* btrie_root;
-    int truncated;
+    unsigned int order;
 };
 
 struct btrie_node
 {
-    struct llist_header children;
-    struct llist_header siblings;
     struct llist_header nodes;
     struct btrie_node* parent;
-    uint32_t index;
+    unsigned long index;
     void* data;
+
+    struct btrie_node** children;
+    int children_cnt;
 };
 
 void
-btrie_init(struct btrie* btrie, uint32_t trunc_bits);
+btrie_init(struct btrie* btrie, unsigned int order);
 
 void*
-btrie_get(struct btrie* root, uint32_t index);
+btrie_get(struct btrie* root, unsigned long index);
 
+/**
+ * Set an object into btrie tree at given location,
+ * override if another object is already present.
+ */
 void
-btrie_set(struct btrie* root, uint32_t index, void* data);
+btrie_set(struct btrie* root, unsigned long index, void* data);
+
+/**
+ * Map an object into btrie tree, return the index the object
+ * mapped to
+ */
+unsigned long
+btrie_map(struct btrie* root, 
+          unsigned long start, unsigned long end, void* data);
 
 void*
-btrie_remove(struct btrie* root, uint32_t index);
+btrie_remove(struct btrie* root, unsigned long index);
 
 void
 btrie_release(struct btrie* tree);