wash/bluelib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

core: btree: replace legacy iterator interface

Browse Source
This commit is contained in:
max 🍓
2025-10-29 14:30:33 +00:00
parent c2b894ef22
commit 1e4b5bc4e6
2 changed files with 102 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

116
core/btree.c
View File

@@ -77,6 +77,12 @@
#define HEIGHT(x) ((x) ? (x)->b_height : 0)
struct b_btree_iterator_p {
size_t i, depth;
b_btree_node *node;
b_btree *_b;
};
static inline void update_height(struct b_btree_node *x)
{
x->b_height = MAX(HEIGHT(x->b_left), HEIGHT((x->b_right))) + 1;
@@ -413,8 +419,7 @@ void b_btree_insert_fixup(struct b_btree *tree, struct b_btree_node *node)
@param node the node to delete.
*/
static struct b_btree_node *remove_node_with_no_children(
struct b_btree *tree,
struct b_btree_node *node)
struct b_btree *tree, struct b_btree_node *node)
{
struct b_btree_node *w = node->b_parent;
struct b_btree_node *p = node->b_parent;
@@ -447,8 +452,7 @@ static struct b_btree_node *remove_node_with_no_children(
@param node the node to delete.
*/
static struct b_btree_node *replace_node_with_one_subtree(
struct b_btree *tree,
struct b_btree_node *node)
struct b_btree *tree, struct b_btree_node *node)
{
struct b_btree_node *p = node->b_parent;
struct b_btree_node *z = NULL;
@@ -497,8 +501,7 @@ static struct b_btree_node *replace_node_with_one_subtree(
@param z the node to delete.
*/
static struct b_btree_node *replace_node_with_two_subtrees(
struct b_btree *tree,
struct b_btree_node *z)
struct b_btree *tree, struct b_btree_node *z)
{
/* x will replace z */
struct b_btree_node *x = z->b_left;
@@ -733,45 +736,49 @@ b_btree_node *b_btree_prev(const struct b_btree_node *node)
return prev_node(node, &d);
}
static bool btree_iterator_next(struct b_iterator *it)
b_iterator *b_btree_begin(struct b_btree *tree)
{
return b_btree_iterator_next((struct b_btree_iterator *)it);
}
b_iterator *it_obj = b_object_create(B_TYPE_BTREE_ITERATOR);
if (!it_obj) {
return NULL;
}
static b_status btree_iterator_erase(struct b_iterator *it)
{
return b_btree_iterator_erase((struct b_btree_iterator *)it);
}
static bool btree_iterator_is_valid(const struct b_iterator *it)
{
return b_btree_iterator_is_valid((struct b_btree_iterator *)it);
}
static const b_iterator_ops btree_iterator_ops = {
.it_next = btree_iterator_next,
.it_erase = btree_iterator_erase,
.it_close = NULL,
.it_is_valid = btree_iterator_is_valid,
};
int b_btree_iterator_begin(
const struct b_btree *tree,
struct b_btree_iterator *it)
{
struct b_btree_iterator_p *it
= b_object_get_private(it_obj, B_TYPE_BTREE_ITERATOR);
int depth = 0;
it->_b = (struct b_btree *)tree;
it->i = 0;
it->node = first_node(tree, &depth);
it->depth = depth;
it->_base.it_ops = &btree_iterator_ops;
return 0;
return it_obj;
}
bool b_btree_iterator_next(struct b_btree_iterator *it)
const b_iterator *b_btree_cbegin(const struct b_btree *tree)
{
b_iterator *it_obj = b_object_create(B_TYPE_BTREE_ITERATOR);
if (!it_obj) {
return NULL;
}
struct b_btree_iterator_p *it
= b_object_get_private(it_obj, B_TYPE_BTREE_ITERATOR);
int depth = 0;
it->_b = (struct b_btree *)tree;
it->i = 0;
it->node = first_node(tree, &depth);
it->depth = depth;
return it_obj;
}
static enum b_status iterator_move_next(const b_iterator *obj)
{
struct b_btree_iterator_p *it
= b_object_get_private(obj, B_TYPE_BTREE_ITERATOR);
int depth_diff = 0;
struct b_btree_node *next = next_node(it->node, &depth_diff);
@@ -788,8 +795,11 @@ bool b_btree_iterator_next(struct b_btree_iterator *it)
return true;
}
b_status b_btree_iterator_erase(struct b_btree_iterator *it)
static enum b_status iterator_erase(b_iterator *obj)
{
struct b_btree_iterator_p *it
= b_object_get_private(obj, B_TYPE_BTREE_ITERATOR);
if (!it->node) {
return B_ERR_OUT_OF_BOUNDS;
}
@@ -815,7 +825,41 @@ b_status b_btree_iterator_erase(struct b_btree_iterator *it)
return B_SUCCESS;
}
bool b_btree_iterator_is_valid(const struct b_btree_iterator *it)
static b_iterator_value iterator_get_value(b_iterator *obj)
{
return it->node != NULL;
struct b_btree_iterator_p *it
= b_object_get_private(obj, B_TYPE_BTREE_ITERATOR);
return B_ITERATOR_VALUE_PTR(it->node);
}
static const b_iterator_value iterator_get_cvalue(const b_iterator *obj)
{
struct b_btree_iterator_p *it
= b_object_get_private(obj, B_TYPE_BTREE_ITERATOR);
return B_ITERATOR_VALUE_CPTR(it->node);
}
/*** CLASS DEFINITION *********************************************************/
// ---- b_btree_iterator DEFINITION
B_TYPE_CLASS_DEFINITION_BEGIN(b_btree_iterator)
B_TYPE_CLASS_INTERFACE_BEGIN(b_object, B_TYPE_OBJECT)
B_INTERFACE_ENTRY(to_string) = NULL;
B_TYPE_CLASS_INTERFACE_END(b_object, B_TYPE_OBJECT)
B_TYPE_CLASS_INTERFACE_BEGIN(b_iterator, B_TYPE_ITERATOR)
B_INTERFACE_ENTRY(it_move_next) = iterator_move_next;
B_INTERFACE_ENTRY(it_erase) = iterator_erase;
B_INTERFACE_ENTRY(it_get_value) = iterator_get_value;
B_INTERFACE_ENTRY(it_get_cvalue) = iterator_get_cvalue;
B_TYPE_CLASS_INTERFACE_END(b_iterator, B_TYPE_ITERATOR)
B_TYPE_CLASS_DEFINITION_END(b_btree_iterator)
B_TYPE_DEFINITION_BEGIN(b_btree_iterator)
B_TYPE_ID(0x432779d7, 0xc03a, 0x48ea, 0xae8f, 0x12c666c767ae);
B_TYPE_EXTENDS(B_TYPE_ITERATOR);
B_TYPE_CLASS(b_btree_iterator_class);
B_TYPE_INSTANCE_PRIVATE(struct b_btree_iterator_p);
B_TYPE_DEFINITION_END(b_btree_iterator)