|
|
- #ifndef __KERN_LIBS_RB_TREE_H__
- #define __KERN_LIBS_RB_TREE_H__
-
- #include <defs.h>
-
- typedef struct rb_node {
- bool red; // if red = 0, it's a black node
- struct rb_node *parent;
- struct rb_node *left, *right;
- } rb_node;
-
- typedef struct rb_tree {
- // compare function should return -1 if *node1 < *node2, 1 if *node1 > *node2, and 0 otherwise
- int (*compare)(rb_node *node1, rb_node *node2);
- struct rb_node *nil, *root;
- } rb_tree;
-
- rb_tree *rb_tree_create(int (*compare)(rb_node *node1, rb_node *node2));
- void rb_tree_destroy(rb_tree *tree);
- void rb_insert(rb_tree *tree, rb_node *node);
- void rb_delete(rb_tree *tree, rb_node *node);
- rb_node *rb_search(rb_tree *tree, int (*compare)(rb_node *node, void *key), void *key);
- rb_node *rb_node_prev(rb_tree *tree, rb_node *node);
- rb_node *rb_node_next(rb_tree *tree, rb_node *node);
- rb_node *rb_node_root(rb_tree *tree);
- rb_node *rb_node_left(rb_tree *tree, rb_node *node);
- rb_node *rb_node_right(rb_tree *tree, rb_node *node);
-
- void check_rb_tree(void);
-
- #endif /* !__KERN_LIBS_RBTREE_H__ */
-
|