+/*
+ Copyright (c) 2010 Joseph A. Adams
+ All rights reserved.
+
+ Redistribution and use in source and binary forms, with or without
+ modification, are permitted provided that the following conditions
+ are met:
+ 1. Redistributions of source code must retain the above copyright
+ notice, this list of conditions and the following disclaimer.
+ 2. Redistributions in binary form must reproduce the above copyright
+ notice, this list of conditions and the following disclaimer in the
+ documentation and/or other materials provided with the distribution.
+ 3. The name of the author may not be used to endorse or promote products
+ derived from this software without specific prior written permission.
+
+ THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+*/
+
+#include "btree.h"
+
+#include <assert.h>
+#include <stdlib.h>
+#include <stdio.h>
+
+#define MAX (BTREE_ITEM_MAX)
+#define MIN (BTREE_ITEM_MAX >> 1)
+
+static struct btree_node *node_alloc(int internal);
+static void node_delete(struct btree_node *node, struct btree *btree);
+
+static void branch_begin(btree_iterator iter);
+static void branch_end(btree_iterator iter);
+static void begin_end_lr(btree_iterator iter, struct btree_node *node, int lr);
+
+/*
+ * If iter->node has parent, returns 1 and ascends the iterator such that
+ * iter->node->branch[iter->k] will be what iter->node was.
+ *
+ * If iter->node does not have a parent (is a root), returns 0 and leaves the
+ * iterator untouched.
+ */
+#define ascend(iter) ((iter)->node->parent \
+ ? (iter)->k = (iter)->node->k, (iter)->node = (iter)->node->parent, 1 \
+ : 0)
+
+static void node_insert(const void *x, struct btree_node *xr,
+ struct btree_node *p, unsigned int k);
+static void node_split(const void **x, struct btree_node **xr,
+ struct btree_node *p, unsigned int k);
+
+static void node_remove_leaf_item(struct btree_node *node, unsigned int k);
+void node_restore(struct btree_node *node, unsigned int k);
+
+static int node_walk_backward(const struct btree_node *node,
+ btree_action_t action, void *ctx);
+static int node_walk_forward(const struct btree_node *node,
+ btree_action_t action, void *ctx);
+
+
+/************************* Public functions *************************/
+
+struct btree *btree_new(btree_search_t search)
+{
+ struct btree *btree = calloc(1, sizeof(struct btree));
+ struct btree_node *node = node_alloc(0);
+ node->parent = NULL;
+ node->count = 0;
+ node->depth = 0;
+ btree->root = node;
+ btree->search = search;
+ return btree;
+}
+
+void btree_delete(struct btree *btree)
+{
+ node_delete(btree->root, btree);
+ free(btree);
+}
+
+int btree_begin_end_lr(const struct btree *btree, btree_iterator iter, int lr)
+{
+ struct btree_node *node;
+
+ iter->btree = (struct btree *)btree;
+ begin_end_lr(iter, btree->root, lr);
+
+ /* Set iter->item if any items exist. */
+ node = iter->node;
+ if (node->count) {
+ iter->item = (void*)node->item[iter->k - lr];
+ return 1;
+ }
+
+ return 0;
+}
+
+int btree_deref(btree_iterator iter)
+{
+ if (iter->k >= iter->node->count) {
+ struct btree_iterator_s tmp = *iter;
+ do {
+ if (!ascend(iter)) {
+ *iter = tmp;
+ return 0;
+ }
+ } while (iter->k >= iter->node->count);
+ }
+
+ iter->item = (void*)iter->node->item[iter->k];
+ return 1;
+}
+
+int btree_prev(btree_iterator iter)
+{
+ if (iter->node->depth) {
+ branch_end(iter);
+ } else if (iter->k == 0) {
+ struct btree_iterator_s tmp = *iter;
+ do {
+ if (!ascend(iter)) {
+ *iter = tmp;
+ return 0;
+ }
+ } while (iter->k == 0);
+ }
+
+ iter->item = (void*)iter->node->item[--iter->k];
+ return 1;
+}
+
+int btree_next(btree_iterator iter)
+{
+ int ret = btree_deref(iter);
+ if (ret) {
+ iter->k++;
+ if (iter->node->depth)
+ branch_begin(iter);
+ }
+ return ret;
+}
+
+int btree_find_lr(const struct btree *btree, const void *key,
+ btree_iterator iter, int lr)
+{
+ struct btree_node *node = btree->root;
+ unsigned int k;
+ unsigned int depth;
+ int found = 0;
+
+ iter->btree = (struct btree *)btree;
+ iter->item = NULL;
+
+ depth = node->depth;
+ for (;;) {
+ int f = 0;
+ k = btree->search(key, node->item, node->count, lr, &f);
+
+ if (f) {
+ iter->item = (void*)node->item[k - lr];
+ found = 1;
+ }
+ if (!depth--)
+ break;
+
+ node = node->branch[k];
+ }
+
+ iter->node = node;
+ iter->k = k;
+
+ return found;
+}
+
+int btree_walk_backward(const struct btree *btree,
+ btree_action_t action, void *ctx)
+{
+ return node_walk_backward(btree->root, action, ctx);
+}
+
+int btree_walk_forward(const struct btree *btree,
+ btree_action_t action, void *ctx)
+{
+ return node_walk_forward(btree->root, action, ctx);
+}
+
+void btree_insert_at(btree_iterator iter, const void *item)
+{
+ const void *x = item;
+ struct btree_node *xr = NULL;
+ struct btree_node *p;
+ struct btree *btree = iter->btree;
+
+ /* btree_insert_at always sets iter->item to item. */
+ iter->item = (void*)item;
+
+ /*
+ * If node is not a leaf, fall to the end of the left branch of item[k]
+ * so that it will be a leaf. This does not modify the iterator's logical
+ * position.
+ */
+ if (iter->node->depth)
+ branch_end(iter);
+
+ /*
+ * First try inserting item into this node.
+ * If it's too big, split it, and repeat by
+ * trying to insert the median and right subtree into parent.
+ */
+ if (iter->node->count < MAX) {
+ node_insert(x, xr, iter->node, iter->k);
+ goto finished;
+ } else {
+ for (;;) {
+ node_split(&x, &xr, iter->node, iter->k);
+
+ if (!ascend(iter))
+ break;
+
+ if (iter->node->count < MAX) {
+ node_insert(x, xr, iter->node, iter->k);
+ goto finished;
+ }
+ }
+
+ /*
+ * If splitting came all the way up to the root, create a new root whose
+ * left branch is the current root, median is x, and right branch is the
+ * half split off from the root.
+ */
+ assert(iter->node == btree->root);
+ p = node_alloc(1);
+ p->parent = NULL;
+ p->count = 1;
+ p->depth = btree->root->depth + 1;
+ p->item[0] = x;
+ p->branch[0] = btree->root;
+ btree->root->parent = p;
+ btree->root->k = 0;
+ p->branch[1] = xr;
+ xr->parent = p;
+ xr->k = 1;
+ btree->root = p;
+ }
+
+finished:
+ btree->count++;
+ iter->node = NULL;
+}
+
+int btree_remove_at(btree_iterator iter)
+{
+ struct btree *btree = iter->btree;
+ struct btree_node *root;
+
+ if (!btree_deref(iter))
+ return 0;
+
+ if (!iter->node->depth) {
+ node_remove_leaf_item(iter->node, iter->k);
+ if (iter->node->count >= MIN || !iter->node->parent)
+ goto finished;
+ } else {
+ /*
+ * We can't remove an item from an internal node, so we'll replace it
+ * with its successor (which will always be in a leaf), then remove
+ * the original copy of the successor.
+ */
+
+ /* Save pointer to condemned item. */
+ const void **x = &iter->node->item[iter->k];
+
+ /* Descend to successor. */
+ iter->k++;
+ branch_begin(iter);
+
+ /* Replace condemned item with successor. */
+ *x = iter->node->item[0];
+
+ /* Remove successor. */
+ node_remove_leaf_item(iter->node, 0);
+ }
+
+ /*
+ * Restore nodes that fall under their minimum count. This may
+ * propagate all the way up to the root.
+ */
+ for (;;) {
+ if (iter->node->count >= MIN)
+ goto finished;
+ if (!ascend(iter))
+ break;
+ node_restore(iter->node, iter->k);
+ }
+
+ /*
+ * If combining came all the way up to the root, and it has no more
+ * dividers, delete it and make its only branch the root.
+ */
+ root = iter->node;
+ assert(root == btree->root);
+ assert(root->depth > 0);
+ if (root->count == 0) {
+ btree->root = root->branch[0];
+ btree->root->parent = NULL;
+ free(root);
+ }
+
+finished:
+ btree->count--;
+ iter->node = NULL;
+ return 1;
+}
+
+/*
+ * ascends iterator a until it matches iterator b's depth.
+ *
+ * Returns -1 if they end up on the same k (meaning a < b).
+ * Returns 0 otherwise.
+ */
+static int elevate(btree_iterator a, btree_iterator b)
+{
+ while (a->node->depth < b->node->depth)
+ ascend(a);
+
+ if (a->k == b->k)
+ return -1;
+ return 0;
+}
+
+int btree_cmp_iters(const btree_iterator iter_a, const btree_iterator iter_b)
+{
+ btree_iterator a = {*iter_a}, b = {*iter_b};
+ int ad, bd;
+
+ ad = btree_deref(a);
+ bd = btree_deref(b);
+
+ /* Check cases where one or both iterators are at the end. */
+ if (!ad)
+ return bd ? 1 : 0;
+ if (!bd)
+ return ad ? -1 : 0;
+
+ /* Bring iterators to the same depth. */
+ if (a->node->depth < b->node->depth) {
+ if (elevate(a, b))
+ return -1;
+ } else if (a->node->depth > b->node->depth) {
+ if (elevate(b, a))
+ return 1;
+ }
+
+ /* Bring iterators to the same node. */
+ while (a->node != b->node) {
+ ascend(a);
+ ascend(b);
+ }
+
+ /* Now we can compare by k directly. */
+ if (a->k < b->k)
+ return -1;
+ if (a->k > b->k)
+ return 1;
+
+ return 0;
+}
+
+
+/************************* Private functions *************************/
+
+static struct btree_node *node_alloc(int internal)
+{
+ struct btree_node *node;
+ size_t isize = internal
+ ? sizeof(struct btree_node*) * (BTREE_ITEM_MAX+1)
+ : 0;
+ node = malloc(sizeof(struct btree_node) + isize);
+ return node;
+}
+
+static void node_delete(struct btree_node *node, struct btree *btree)
+{
+ unsigned int i, count = node->count;
+
+ if (!node->depth) {
+ if (btree->destroy) {
+ for (i=0; i<count; i++)
+ btree->destroy((void*)node->item[i], btree->destroy_ctx);
+ }
+ } else {
+ for (i=0; i<count; i++) {
+ node_delete(node->branch[i], btree);
+ if (btree->destroy)
+ btree->destroy((void*)node->item[i], btree->destroy_ctx);
+ }
+ node_delete(node->branch[count], btree);
+ }
+
+ free(node);
+}
+
+/* Set iter to beginning of branch pointed to by iter. */
+static void branch_begin(btree_iterator iter)
+{
+ struct btree_node *node = iter->node->branch[iter->k];
+ unsigned int depth = node->depth;
+ while (depth--)
+ node = node->branch[0];
+ iter->node = node;
+ iter->k = 0;
+}
+
+/* Set iter to end of branch pointed to by iter. */
+static void branch_end(btree_iterator iter)
+{
+ struct btree_node *node = iter->node->branch[iter->k];
+ unsigned int depth = node->depth;
+ while (depth--)
+ node = node->branch[node->count];
+ iter->node = node;
+ iter->k = node->count;
+}
+
+/* Traverse to the beginning or end of node, depending on lr. */
+static void begin_end_lr(btree_iterator iter, struct btree_node *node, int lr)
+{
+ iter->node = node;
+ iter->k = lr ? node->count : 0;
+ if (node->depth)
+ (lr ? branch_end : branch_begin)(iter);
+}
+
+/*
+ * Inserts item x and right branch xr into node p at position k.
+ *
+ * Assumes p exists and has enough room.
+ * Ignores xr if p is a leaf.
+ */
+static void node_insert(const void *x, struct btree_node *xr,
+ struct btree_node *p, unsigned int k)
+{
+ unsigned int i;
+
+ for (i = p->count; i-- > k;)
+ p->item[i+1] = p->item[i];
+ p->item[k] = x;
+
+ if (p->depth) {
+ k++;
+ for (i = p->count+1; i-- > k;) {
+ p->branch[i+1] = p->branch[i];
+ p->branch[i+1]->k = i+1;
+ }
+ p->branch[k] = xr;
+ xr->parent = p;
+ xr->k = k;
+ }
+
+ p->count++;
+}
+
+/*
+ * Inserts item *x and subtree *xr into node p at position k, splitting it into
+ * nodes p and *xr with median item *x.
+ *
+ * Assumes p->count == MAX.
+ * Ignores original *xr if p is a leaf, but always sets it.
+ */
+static void node_split(const void **x, struct btree_node **xr,
+ struct btree_node *p, unsigned int k)
+{
+ unsigned int i, split;
+ struct btree_node *l = p, *r;
+
+ /*
+ * If k <= MIN, item will be inserted into left subtree, so give l
+ * fewer items initially.
+ * Otherwise, item will be inserted into right subtree, so give r
+ * fewer items initially.
+ */
+ if (k <= MIN)
+ split = MIN;
+ else
+ split = MIN + 1;
+
+ /*
+ * If l->depth is 0, allocate a leaf node.
+ * Otherwise, allocate an internal node.
+ */
+ r = node_alloc(l->depth);
+
+ /* l and r will be siblings, so they will have the same parent and depth. */
+ r->parent = l->parent;
+ r->depth = l->depth;
+
+ /*
+ * Initialize items/branches of right side.
+ * Do not initialize r's leftmost branch yet because we don't know
+ * whether it will be l's current rightmost branch or if *xr will
+ * take its place.
+ */
+ for (i = split; i < MAX; i++)
+ r->item[i-split] = l->item[i];
+ if (r->depth) {
+ for (i = split+1; i <= MAX; i++) {
+ r->branch[i-split] = l->branch[i];
+ r->branch[i-split]->parent = r;
+ r->branch[i-split]->k = i-split;
+ }
+ }
+
+ /* Update counts. */
+ l->count = split;
+ r->count = MAX - split;
+
+ /*
+ * The nodes are now split, but the key isn't inserted yet.
+ *
+ * Insert key into left or right half,
+ * depending on which side it fell on.
+ */
+ if (k <= MIN)
+ node_insert(*x, *xr, l, k);
+ else
+ node_insert(*x, *xr, r, k - split);
+
+ /*
+ * Give l's rightmost branch to r because l's rightmost item
+ * is going up to become the median.
+ */
+ if (r->depth) {
+ r->branch[0] = l->branch[l->count];
+ r->branch[0]->parent = r;
+ r->branch[0]->k = 0;
+ }
+
+ /*
+ * Take up l's rightmost item to make it the median.
+ * That item's right branch is now r.
+ */
+ *x = l->item[--l->count];
+ *xr = r;
+}
+
+/*
+ * Removes item k from node p, shifting successor items back and
+ * decrementing the count.
+ *
+ * Assumes node p has the item k and is a leaf.
+ */
+static void node_remove_leaf_item(struct btree_node *node, unsigned int k)
+{
+ unsigned int i;
+ for (i = k+1; i < node->count; i++)
+ node->item[i-1] = node->item[i];
+ node->count--;
+}
+
+static void move_left(struct btree_node *node, unsigned int k);
+static void move_right(struct btree_node *node, unsigned int k);
+static void combine(struct btree_node *node, unsigned int k);
+
+/*
+ * Fixes node->branch[k]'s problem of having one less than MIN items.
+ * May or may not cause node to fall below MIN items, depending on whether
+ * two branches are combined or not.
+ */
+void node_restore(struct btree_node *node, unsigned int k)
+{
+ if (k == 0) {
+ if (node->branch[1]->count > MIN)
+ move_left(node, 0);
+ else
+ combine(node, 0);
+ } else if (k == node->count) {
+ if (node->branch[k-1]->count > MIN)
+ move_right(node, k-1);
+ else
+ combine(node, k-1);
+ } else if (node->branch[k-1]->count > MIN) {
+ move_right(node, k-1);
+ } else if (node->branch[k+1]->count > MIN) {
+ move_left(node, k);
+ } else {
+ combine(node, k-1);
+ }
+}
+
+static void move_left(struct btree_node *node, unsigned int k)
+{
+ struct btree_node *l = node->branch[k], *r = node->branch[k+1], *mv;
+ unsigned int i;
+
+ l->item[l->count] = node->item[k];
+ node->item[k] = r->item[0];
+ for (i = 1; i < r->count; i++)
+ r->item[i-1] = r->item[i];
+
+ if (r->depth) {
+ mv = r->branch[0];
+ l->branch[l->count+1] = mv;
+ mv->parent = l;
+ mv->k = l->count+1;
+
+ for (i = 1; i <= r->count; i++) {
+ r->branch[i-1] = r->branch[i];
+ r->branch[i-1]->k = i-1;
+ }
+ }
+
+ l->count++;
+ r->count--;
+}
+
+static void move_right(struct btree_node *node, unsigned int k)
+{
+ struct btree_node *l = node->branch[k], *r = node->branch[k+1];
+ unsigned int i;
+
+ for (i = r->count; i--;)
+ r->item[i+1] = r->item[i];
+ r->item[0] = node->item[k];
+ node->item[k] = l->item[l->count-1];
+
+ if (r->depth) {
+ for (i = r->count+1; i--;) {
+ r->branch[i+1] = r->branch[i];
+ r->branch[i+1]->k = i+1;
+ }
+ r->branch[0] = l->branch[l->count];
+ r->branch[0]->parent = r;
+ r->branch[0]->k = 0;
+ }
+
+ l->count--;
+ r->count++;
+}
+
+/* Combine node->branch[k] and node->branch[k+1]. */
+static void combine(struct btree_node *node, unsigned int k)
+{
+ struct btree_node *l = node->branch[k], *r = node->branch[k+1], *mv;
+ const void **o = &l->item[l->count];
+ unsigned int i;
+
+ //append node->item[k] followed by right node's items to left node
+ *o++ = node->item[k];
+ for (i=0; i<r->count; i++)
+ *o++ = r->item[i];
+
+ //if applicable, append right node's branches to left node
+ if (r->depth) {
+ for (i=0; i<=r->count; i++) {
+ mv = r->branch[i];
+ l->branch[l->count + i + 1] = mv;
+ mv->parent = l;
+ mv->k = l->count + i + 1;
+ }
+ }
+
+ //remove k and its right branch from parent node
+ for (i = k+1; i < node->count; i++) {
+ node->item[i-1] = node->item[i];
+ node->branch[i] = node->branch[i+1];
+ node->branch[i]->k = i;
+ }
+
+ //don't forget to update the left and parent node's counts and to free the right node
+ l->count += r->count + 1;
+ node->count--;
+ free(r);
+}
+
+static int node_walk_backward(const struct btree_node *node,
+ btree_action_t action, void *ctx)
+{
+ unsigned int i, count = node->count;
+
+ if (!node->depth) {
+ for (i=count; i--;)
+ if (!action((void*)node->item[i], ctx))
+ return 0;
+ } else {
+ if (!node_walk_backward(node->branch[count], action, ctx))
+ return 0;
+ for (i=count; i--;) {
+ if (!action((void*)node->item[i], ctx))
+ return 0;
+ if (!node_walk_backward(node->branch[i], action, ctx))
+ return 0;
+ }
+ }
+
+ return 1;
+}
+
+static int node_walk_forward(const struct btree_node *node,
+ btree_action_t action, void *ctx)
+{
+ unsigned int i, count = node->count;
+
+ if (!node->depth) {
+ for (i=0; i<count; i++)
+ if (!action((void*)node->item[i], ctx))
+ return 0;
+ } else {
+ for (i=0; i<count; i++) {
+ if (!node_walk_forward(node->branch[i], action, ctx))
+ return 0;
+ if (!action((void*)node->item[i], ctx))
+ return 0;
+ }
+ if (!node_walk_forward(node->branch[count], action, ctx))
+ return 0;
+ }
+
+ return 1;
+}