include/structures/avl.h View source View on GitHub
struct avl_tree_node {
int height;
struct avl_tree_node *left;
struct avl_tree_node *right;
struct avl_tree_node *parent;
};
struct avl_tree_node_ops {
int (*cmp)(const struct avl_tree_node *a, const struct avl_tree_node *b);
int (*cmp_key)(const struct avl_tree_node *node, const void *key);
};
struct avl_tree {
struct avl_tree_node *root;
const struct avl_tree_node_ops *ops;
};
void avl_tree_init(struct avl_tree *tree, const struct avl_tree_node_ops *ops);
void avl_tree_insert(struct avl_tree *tree, struct avl_tree_node *node);
void avl_tree_remove(struct avl_tree *tree, struct avl_tree_node *node);
struct avl_tree_node * avl_tree_find(const struct avl_tree *tree, const void *key);
struct avl_tree_node * avl_tree_first(const struct avl_tree *tree);
struct avl_tree_node * avl_tree_last(const struct avl_tree *tree);
struct avl_tree_node * avl_tree_next(const struct avl_tree_node *node);
struct avl_tree_node * avl_tree_prev(const struct avl_tree_node *node);
int avl_tree_empty(const struct avl_tree *tree);
#define avl_entry(ptr, type, member) container_of(ptr, type, member)
#define avl_tree_for_each(pos, tree) \
for ((pos) = avl_tree_first(tree); (pos); (pos) = avl_tree_next(pos))
#define avl_tree_for_each_safe(pos, n, tree) \
for ((pos) = avl_tree_first(tree), \
(n) = (pos) ? avl_tree_next(pos) : NULL; \
(pos); (pos) = (n), (n) = (pos) ? avl_tree_next(pos) : NULL)