163 lines
9.0 KiB
C
163 lines
9.0 KiB
C
#ifndef SOCKS_BTREE_H_
|
|
#define SOCKS_BTREE_H_
|
|
|
|
#include <stdint.h>
|
|
|
|
#define BTREE_CONTAINER(t, m, v) ((void *)((v) ? (uintptr_t)(v) - (offsetof(t, m)) : 0))
|
|
|
|
#define BTREE_DEFINE_SIMPLE_INSERT(node_type, container_node_member, container_key_member, function_name) \
|
|
static void function_name(btree_t *tree, node_type *node) \
|
|
{ \
|
|
if (!tree->b_root) { \
|
|
tree->b_root = &node->container_node_member; \
|
|
btree_insert_fixup(tree, &node->container_node_member); \
|
|
return; \
|
|
} \
|
|
\
|
|
btree_node_t *cur = tree->b_root; \
|
|
while (1) { \
|
|
node_type *cur_node = BTREE_CONTAINER(node_type, container_node_member, cur); \
|
|
btree_node_t *next = NULL; \
|
|
\
|
|
if (node->container_key_member > cur_node->container_key_member) { \
|
|
next = btree_right(cur); \
|
|
\
|
|
if (!next) { \
|
|
btree_put_right(cur, &node->container_node_member); \
|
|
break; \
|
|
} \
|
|
} else if (node->container_key_member < cur_node->container_key_member) { \
|
|
next = btree_left(cur); \
|
|
\
|
|
if (!next) { \
|
|
btree_put_left(cur, &node->container_node_member); \
|
|
break; \
|
|
} \
|
|
} else { \
|
|
return; \
|
|
} \
|
|
\
|
|
cur = next; \
|
|
} \
|
|
\
|
|
btree_insert_fixup(tree, &node->container_node_member); \
|
|
}
|
|
|
|
#define BTREE_DEFINE_INSERT(node_type, container_node_member, container_key_member, function_name, comparator) \
|
|
static void function_name(btree_t *tree, node_type *node) \
|
|
{ \
|
|
if (!tree->b_root) { \
|
|
tree->b_root = &node->container_node_member; \
|
|
btree_insert_fixup(tree, &node->container_node_member); \
|
|
return; \
|
|
} \
|
|
\
|
|
btree_node_t *cur = tree->b_root; \
|
|
while (1) { \
|
|
node_type *cur_node = BTREE_CONTAINER(node_type, container_node_member, cur); \
|
|
btree_node_t *next = NULL; \
|
|
int cmp = comparator(node, cur_node); \
|
|
\
|
|
if (cmp == 1) { \
|
|
next = btree_right(cur); \
|
|
\
|
|
if (!next) { \
|
|
btree_put_right(cur, &node->container_node_member); \
|
|
break; \
|
|
} \
|
|
} else if (cmp == -1) { \
|
|
next = btree_left(cur); \
|
|
\
|
|
if (!next) { \
|
|
btree_put_left(cur, &node->container_node_member); \
|
|
break; \
|
|
} \
|
|
} else { \
|
|
return; \
|
|
} \
|
|
\
|
|
cur = next; \
|
|
} \
|
|
\
|
|
btree_insert_fixup(tree, &node->container_node_member); \
|
|
}
|
|
|
|
#define BTREE_DEFINE_SIMPLE_GET(node_type, key_type, container_node_member, container_key_member, function_name) \
|
|
node_type *get(btree_t *tree, key_type key) \
|
|
{ \
|
|
btree_node_t *cur = tree->b_root; \
|
|
while (cur) { \
|
|
node_type *cur_node = BTREE_CONTAINER(node_type, container_node_member, cur); \
|
|
if (key > cur_node->container_key_member) { \
|
|
cur = btree_right(cur); \
|
|
} else if (key < cur_node->container_key_member) { \
|
|
cur = btree_left(cur); \
|
|
} else { \
|
|
return cur_node; \
|
|
} \
|
|
} \
|
|
\
|
|
return NULL; \
|
|
}
|
|
|
|
#define btree_foreach(iter_type, iter_name, tree_name, node_member) \
|
|
for (iter_type *iter_name = BTREE_CONTAINER(iter_type, node_member, btree_first(tree_name)); \
|
|
iter_name; \
|
|
iter_name = BTREE_CONTAINER(iter_type, node_member, btree_next(&((iter_name)->node_member))))
|
|
|
|
#define btree_foreach_r(iter_type, iter_name, tree_name, node_member) \
|
|
for (iter_type *iter_name = BTREE_CONTAINER(iter_type, node_member, btree_last(tree_name)); \
|
|
iter_name; \
|
|
iter_name = BTREE_CONTAINER(iter_type, node_member, btree_prev(&((iter_name)->node_member))))
|
|
|
|
typedef struct btree_node {
|
|
struct btree_node *b_parent, *b_left, *b_right;
|
|
unsigned short b_height;
|
|
} btree_node_t;
|
|
|
|
typedef struct btree {
|
|
struct btree_node *b_root;
|
|
} btree_t;
|
|
|
|
extern void btree_insert_fixup(btree_t *tree, btree_node_t *node);
|
|
extern void btree_delete(btree_t *tree, btree_node_t *node);
|
|
|
|
extern btree_node_t *btree_first(btree_t *tree);
|
|
extern btree_node_t *btree_last(btree_t *tree);
|
|
extern btree_node_t *btree_next(btree_node_t *node);
|
|
extern btree_node_t *btree_prev(btree_node_t *node);
|
|
|
|
static inline void btree_put_left(btree_node_t *parent, btree_node_t *child)
|
|
{
|
|
parent->b_left = child;
|
|
child->b_parent = parent;
|
|
}
|
|
|
|
static inline void btree_put_right(btree_node_t *parent, btree_node_t *child)
|
|
{
|
|
parent->b_right = child;
|
|
child->b_parent = parent;
|
|
}
|
|
|
|
static inline btree_node_t *btree_left(btree_node_t *node)
|
|
{
|
|
return node->b_left;
|
|
}
|
|
|
|
static inline btree_node_t *btree_right(btree_node_t *node)
|
|
{
|
|
return node->b_right;
|
|
}
|
|
|
|
static inline btree_node_t *btree_parent(btree_node_t *node)
|
|
{
|
|
return node->b_parent;
|
|
}
|
|
|
|
static inline unsigned short btree_height(btree_node_t *node)
|
|
{
|
|
return node->b_height;
|
|
}
|
|
|
|
#endif
|