123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844 |
- /* Produced by texiweb from libavl.w. */
- /* libavl - library for manipulation of binary trees.
- Copyright (C) 1998, 1999, 2000, 2001, 2002, 2004 Free Software
- Foundation, Inc.
- This library is free software; you can redistribute it and/or
- modify it under the terms of the GNU Lesser General Public
- License as published by the Free Software Foundation; either
- version 3 of the License, or (at your option) any later version.
- This library is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- Lesser General Public License for more details.
- You should have received a copy of the GNU Lesser General Public
- License along with this library; if not, write to the Free Software
- Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
- 02110-1301 USA.
- */
- /* Nov 2016, Markus Metz
- * from libavl-2.0.3
- * added safety checks and speed optimizations
- */
- #include <assert.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include "pavlrc.h"
- #define CMP_RC(a, b) ((a)->row == (b)->row ? (a)->col - (b)->col : (a)->row - (b)->row)
- /* Creates and returns a new table
- with comparison function |compare| using parameter |param|
- and memory allocator |allocator|.
- Returns |NULL| if memory allocation failed. */
- struct pavlrc_table *pavlrc_create(struct libavl_allocator *allocator)
- {
- struct pavlrc_table *tree;
- if (allocator == NULL)
- allocator = &pavlrc_allocator_default;
- tree = allocator->libavl_malloc(allocator, sizeof *tree);
- if (tree == NULL)
- return NULL;
- tree->pavl_root = NULL;
- tree->pavl_alloc = allocator;
- tree->pavl_count = 0;
- return tree;
- }
- /* Search |tree| for an item matching |item|, and return it if found.
- Otherwise return |NULL|. */
- struct pavlrc *pavlrc_find(const struct pavlrc_table *tree, const struct pavlrc *item)
- {
- struct pavlrc_node *p;
- assert(tree != NULL && item != NULL);
- p = tree->pavl_root;
- while (p != NULL) {
- int cmp = CMP_RC(item, &p->pavl_data);
- if (cmp == 0)
- return &p->pavl_data;
- p = p->pavl_link[cmp > 0];
- }
- return NULL;
- }
- /* Inserts |item| into |tree| and returns a pointer to |item|.
- If a duplicate item is found in the tree,
- returns a pointer to the duplicate without inserting |item|.
- Returns |NULL| in case of memory allocation failure. */
- struct pavlrc *pavlrc_probe(struct pavlrc_table *tree, struct pavlrc *item)
- {
- struct pavlrc_node *y; /* Top node to update balance factor, and parent. */
- struct pavlrc_node *p, *q; /* Iterator, and parent. */
- struct pavlrc_node *n; /* Newly inserted node. */
- struct pavlrc_node *w; /* New root of rebalanced subtree. */
- int dir; /* Direction to descend. */
- assert(tree != NULL && item != NULL);
- y = p = tree->pavl_root;
- q = NULL;
- dir = 0;
- while (p != NULL) {
- int cmp = CMP_RC(item, &p->pavl_data);
- if (cmp == 0)
- return &p->pavl_data;
- dir = cmp > 0;
- if (p->pavl_balance != 0)
- y = p;
- q = p, p = p->pavl_link[dir];
- }
- n = tree->pavl_alloc->libavl_malloc(tree->pavl_alloc, sizeof *p);
- if (n == NULL)
- return NULL;
- tree->pavl_count++;
- n->pavl_link[0] = n->pavl_link[1] = NULL;
- n->pavl_parent = q;
- n->pavl_data = *item;
- n->pavl_balance = 0;
- if (q == NULL) {
- tree->pavl_root = n;
- return item /* &n->pavl_data */;
- }
- q->pavl_link[dir] = n;
- p = n;
- while (p != y) {
- q = p->pavl_parent;
- /*
- dir = q->pavl_link[0] != p;
- if (dir == 0)
- q->pavl_balance--;
- else
- q->pavl_balance++;
- */
- if (q->pavl_link[0] != p)
- q->pavl_balance++;
- else
- q->pavl_balance--;
- p = q;
- }
- if (y->pavl_balance == -2) {
- struct pavlrc_node *x = y->pavl_link[0];
- if (x->pavl_balance == -1) {
- w = x;
- y->pavl_link[0] = x->pavl_link[1];
- x->pavl_link[1] = y;
- x->pavl_balance = y->pavl_balance = 0;
- x->pavl_parent = y->pavl_parent;
- y->pavl_parent = x;
- if (y->pavl_link[0] != NULL)
- y->pavl_link[0]->pavl_parent = y;
- }
- else {
- assert(x->pavl_balance == +1);
- w = x->pavl_link[1];
- x->pavl_link[1] = w->pavl_link[0];
- w->pavl_link[0] = x;
- y->pavl_link[0] = w->pavl_link[1];
- w->pavl_link[1] = y;
- if (w->pavl_balance == -1)
- x->pavl_balance = 0, y->pavl_balance = +1;
- else if (w->pavl_balance == 0)
- x->pavl_balance = y->pavl_balance = 0;
- else /* |w->pavl_balance == +1| */
- x->pavl_balance = -1, y->pavl_balance = 0;
- w->pavl_balance = 0;
- w->pavl_parent = y->pavl_parent;
- x->pavl_parent = y->pavl_parent = w;
- if (x->pavl_link[1] != NULL)
- x->pavl_link[1]->pavl_parent = x;
- if (y->pavl_link[0] != NULL)
- y->pavl_link[0]->pavl_parent = y;
- }
- }
- else if (y->pavl_balance == +2) {
- struct pavlrc_node *x = y->pavl_link[1];
- if (x->pavl_balance == +1) {
- w = x;
- y->pavl_link[1] = x->pavl_link[0];
- x->pavl_link[0] = y;
- x->pavl_balance = y->pavl_balance = 0;
- x->pavl_parent = y->pavl_parent;
- y->pavl_parent = x;
- if (y->pavl_link[1] != NULL)
- y->pavl_link[1]->pavl_parent = y;
- }
- else {
- assert(x->pavl_balance == -1);
- w = x->pavl_link[0];
- x->pavl_link[0] = w->pavl_link[1];
- w->pavl_link[1] = x;
- y->pavl_link[1] = w->pavl_link[0];
- w->pavl_link[0] = y;
- if (w->pavl_balance == +1)
- x->pavl_balance = 0, y->pavl_balance = -1;
- else if (w->pavl_balance == 0)
- x->pavl_balance = y->pavl_balance = 0;
- else /* |w->pavl_balance == -1| */
- x->pavl_balance = +1, y->pavl_balance = 0;
- w->pavl_balance = 0;
- w->pavl_parent = y->pavl_parent;
- x->pavl_parent = y->pavl_parent = w;
- if (x->pavl_link[0] != NULL)
- x->pavl_link[0]->pavl_parent = x;
- if (y->pavl_link[1] != NULL)
- y->pavl_link[1]->pavl_parent = y;
- }
- }
- else
- return item /* &n->pavl_data */;
- if (w->pavl_parent != NULL)
- w->pavl_parent->pavl_link[y != w->pavl_parent->pavl_link[0]] = w;
- else
- tree->pavl_root = w;
- return item /* &n->pavl_data */;
- }
- /* Inserts |item| into |table|.
- Returns |NULL| if |item| was successfully inserted
- or if a memory allocation error occurred.
- Otherwise, returns the duplicate item. */
- struct pavlrc *pavlrc_insert(struct pavlrc_table *table, struct pavlrc *item)
- {
- struct pavlrc *p = pavlrc_probe(table, item);
- return p == NULL || p == item ? NULL : p;
- }
- /* Inserts |item| into |table|, replacing any duplicate item.
- Returns |NULL| if |item| was inserted without replacing a duplicate,
- or if a memory allocation error occurred.
- Otherwise, returns the item that was replaced. */
- struct pavlrc *pavlrc_replace(struct pavlrc_table *table, struct pavlrc *item)
- {
- struct pavlrc *p = pavlrc_probe(table, item);
- if (p == NULL || p == item)
- return NULL;
- else {
- struct pavlrc *r = p;
- *p = *item;
- return r;
- }
- }
- /* Deletes from |tree| and returns an item matching |item|.
- Returns a null pointer if no matching item found. */
- struct pavlrc *pavlrc_delete(struct pavlrc_table *tree, struct pavlrc *item)
- {
- struct pavlrc_node *p; /* Traverses tree to find node to delete. */
- struct pavlrc_node *q; /* Parent of |p|. */
- int dir; /* Side of |q| on which |p| is linked. */
- int cmp; /* Result of comparison between |item| and |p|. */
- assert(tree != NULL && item != NULL);
- p = tree->pavl_root;
- dir = 0;
- while (p != NULL) {
- cmp = CMP_RC(item, &p->pavl_data);
- if (cmp == 0)
- break;
- dir = cmp > 0;
- p = p->pavl_link[dir];
- }
- if (p == NULL)
- return NULL;
- /* item = p->pavl_data; */
- q = p->pavl_parent;
- if (q == NULL) {
- q = (struct pavlrc_node *)&tree->pavl_root;
- dir = 0;
- }
- if (p->pavl_link[1] == NULL) {
- q->pavl_link[dir] = p->pavl_link[0];
- if (q->pavl_link[dir] != NULL)
- q->pavl_link[dir]->pavl_parent = p->pavl_parent;
- }
- else {
- struct pavlrc_node *r = p->pavl_link[1];
- if (r->pavl_link[0] == NULL) {
- r->pavl_link[0] = p->pavl_link[0];
- q->pavl_link[dir] = r;
- r->pavl_parent = p->pavl_parent;
- if (r->pavl_link[0] != NULL)
- r->pavl_link[0]->pavl_parent = r;
- r->pavl_balance = p->pavl_balance;
- q = r;
- dir = 1;
- }
- else {
- struct pavlrc_node *s = r->pavl_link[0];
- while (s->pavl_link[0] != NULL)
- s = s->pavl_link[0];
- r = s->pavl_parent;
- r->pavl_link[0] = s->pavl_link[1];
- s->pavl_link[0] = p->pavl_link[0];
- s->pavl_link[1] = p->pavl_link[1];
- q->pavl_link[dir] = s;
- if (s->pavl_link[0] != NULL)
- s->pavl_link[0]->pavl_parent = s;
- s->pavl_link[1]->pavl_parent = s;
- s->pavl_parent = p->pavl_parent;
- if (r->pavl_link[0] != NULL)
- r->pavl_link[0]->pavl_parent = r;
- s->pavl_balance = p->pavl_balance;
- q = r;
- dir = 0;
- }
- }
- tree->pavl_alloc->libavl_free(tree->pavl_alloc, p);
- while (q != (struct pavlrc_node *)&tree->pavl_root) {
- struct pavlrc_node *y = q;
- if (y->pavl_parent != NULL)
- q = y->pavl_parent;
- else
- q = (struct pavlrc_node *)&tree->pavl_root;
- if (dir == 0) {
- dir = q->pavl_link[0] != y;
- y->pavl_balance++;
- if (y->pavl_balance == +1)
- break;
- else if (y->pavl_balance == +2) {
- struct pavlrc_node *x = y->pavl_link[1];
- if (x->pavl_balance == -1) {
- struct pavlrc_node *w;
- assert(x->pavl_balance == -1);
- w = x->pavl_link[0];
- x->pavl_link[0] = w->pavl_link[1];
- w->pavl_link[1] = x;
- y->pavl_link[1] = w->pavl_link[0];
- w->pavl_link[0] = y;
- if (w->pavl_balance == +1)
- x->pavl_balance = 0, y->pavl_balance = -1;
- else if (w->pavl_balance == 0)
- x->pavl_balance = y->pavl_balance = 0;
- else /* |w->pavl_balance == -1| */
- x->pavl_balance = +1, y->pavl_balance = 0;
- w->pavl_balance = 0;
- w->pavl_parent = y->pavl_parent;
- x->pavl_parent = y->pavl_parent = w;
- if (x->pavl_link[0] != NULL)
- x->pavl_link[0]->pavl_parent = x;
- if (y->pavl_link[1] != NULL)
- y->pavl_link[1]->pavl_parent = y;
- q->pavl_link[dir] = w;
- }
- else {
- y->pavl_link[1] = x->pavl_link[0];
- x->pavl_link[0] = y;
- x->pavl_parent = y->pavl_parent;
- y->pavl_parent = x;
- if (y->pavl_link[1] != NULL)
- y->pavl_link[1]->pavl_parent = y;
- q->pavl_link[dir] = x;
- if (x->pavl_balance == 0) {
- x->pavl_balance = -1;
- y->pavl_balance = +1;
- break;
- }
- else {
- x->pavl_balance = y->pavl_balance = 0;
- y = x;
- }
- }
- }
- }
- else {
- dir = q->pavl_link[0] != y;
- y->pavl_balance--;
- if (y->pavl_balance == -1)
- break;
- else if (y->pavl_balance == -2) {
- struct pavlrc_node *x = y->pavl_link[0];
- if (x->pavl_balance == +1) {
- struct pavlrc_node *w;
- assert(x->pavl_balance == +1);
- w = x->pavl_link[1];
- x->pavl_link[1] = w->pavl_link[0];
- w->pavl_link[0] = x;
- y->pavl_link[0] = w->pavl_link[1];
- w->pavl_link[1] = y;
- if (w->pavl_balance == -1)
- x->pavl_balance = 0, y->pavl_balance = +1;
- else if (w->pavl_balance == 0)
- x->pavl_balance = y->pavl_balance = 0;
- else /* |w->pavl_balance == +1| */
- x->pavl_balance = -1, y->pavl_balance = 0;
- w->pavl_balance = 0;
- w->pavl_parent = y->pavl_parent;
- x->pavl_parent = y->pavl_parent = w;
- if (x->pavl_link[1] != NULL)
- x->pavl_link[1]->pavl_parent = x;
- if (y->pavl_link[0] != NULL)
- y->pavl_link[0]->pavl_parent = y;
- q->pavl_link[dir] = w;
- }
- else {
- y->pavl_link[0] = x->pavl_link[1];
- x->pavl_link[1] = y;
- x->pavl_parent = y->pavl_parent;
- y->pavl_parent = x;
- if (y->pavl_link[0] != NULL)
- y->pavl_link[0]->pavl_parent = y;
- q->pavl_link[dir] = x;
- if (x->pavl_balance == 0) {
- x->pavl_balance = +1;
- y->pavl_balance = -1;
- break;
- }
- else {
- x->pavl_balance = y->pavl_balance = 0;
- y = x;
- }
- }
- }
- }
- }
- tree->pavl_count--;
- return item;
- }
- /* Initializes |trav| for use with |tree|
- and selects the null node. */
- void pavlrc_t_init(struct pavlrc_traverser *trav, struct pavlrc_table *tree)
- {
- trav->pavl_table = tree;
- trav->pavl_node = NULL;
- }
- /* Initializes |trav| for |tree|.
- Returns data item in |tree| with the least value,
- or |NULL| if |tree| is empty. */
- struct pavlrc *pavlrc_t_first(struct pavlrc_traverser *trav, struct pavlrc_table *tree)
- {
- assert(tree != NULL && trav != NULL);
- trav->pavl_table = tree;
- trav->pavl_node = tree->pavl_root;
- if (trav->pavl_node != NULL) {
- while (trav->pavl_node->pavl_link[0] != NULL)
- trav->pavl_node = trav->pavl_node->pavl_link[0];
- return &trav->pavl_node->pavl_data;
- }
- else
- return NULL;
- }
- /* Initializes |trav| for |tree|.
- Returns data item in |tree| with the greatest value,
- or |NULL| if |tree| is empty. */
- struct pavlrc *pavlrc_t_last(struct pavlrc_traverser *trav, struct pavlrc_table *tree)
- {
- assert(tree != NULL && trav != NULL);
- trav->pavl_table = tree;
- trav->pavl_node = tree->pavl_root;
- if (trav->pavl_node != NULL) {
- while (trav->pavl_node->pavl_link[1] != NULL)
- trav->pavl_node = trav->pavl_node->pavl_link[1];
- return &trav->pavl_node->pavl_data;
- }
- else
- return NULL;
- }
- /* Searches for |item| in |tree|.
- If found, initializes |trav| to the item found and returns the item
- as well.
- If there is no matching item, initializes |trav| to the null item
- and returns |NULL|. */
- struct pavlrc *pavlrc_t_find(struct pavlrc_traverser *trav, struct pavlrc_table *tree,
- struct pavlrc *item)
- {
- struct pavlrc_node *p;
- assert(trav != NULL && tree != NULL && item != NULL);
- trav->pavl_table = tree;
-
- p = tree->pavl_root;
- while (p != NULL) {
- int cmp = CMP_RC(item, &p->pavl_data);
- if (cmp == 0) {
- trav->pavl_node = p;
- return &p->pavl_data;
- }
- p = p->pavl_link[cmp > 0];
- }
- trav->pavl_node = NULL;
- return NULL;
- }
- /* Attempts to insert |item| into |tree|.
- If |item| is inserted successfully, it is returned and |trav| is
- initialized to its location.
- If a duplicate is found, it is returned and |trav| is initialized to
- its location. No replacement of the item occurs.
- If a memory allocation failure occurs, |NULL| is returned and |trav|
- is initialized to the null item. */
- struct pavlrc *pavlrc_t_insert(struct pavlrc_traverser *trav,
- struct pavlrc_table *tree, struct pavlrc *item)
- {
- struct pavlrc *p;
- assert(trav != NULL && tree != NULL && item != NULL);
- p = pavlrc_probe(tree, item);
- if (p != NULL) {
- trav->pavl_table = tree;
- trav->pavl_node = ((struct pavlrc_node *)((char *)p -
- offsetof(struct pavlrc_node,
- pavl_data)));
- return p;
- }
- else {
- pavlrc_t_init(trav, tree);
- return NULL;
- }
- }
- /* Initializes |trav| to have the same current node as |src|. */
- struct pavlrc *pavlrc_t_copy(struct pavlrc_traverser *trav,
- const struct pavlrc_traverser *src)
- {
- assert(trav != NULL && src != NULL);
- trav->pavl_table = src->pavl_table;
- trav->pavl_node = src->pavl_node;
- return trav->pavl_node != NULL ? &trav->pavl_node->pavl_data : NULL;
- }
- /* Returns the next data item in inorder
- within the tree being traversed with |trav|,
- or if there are no more data items returns |NULL|. */
- struct pavlrc *pavlrc_t_next(struct pavlrc_traverser *trav)
- {
- assert(trav != NULL);
- if (trav->pavl_node == NULL)
- return pavlrc_t_first(trav, trav->pavl_table);
- else if (trav->pavl_node->pavl_link[1] == NULL) {
- struct pavlrc_node *q, *p; /* Current node and its child. */
- for (p = trav->pavl_node, q = p->pavl_parent;;
- p = q, q = q->pavl_parent)
- if (q == NULL || p == q->pavl_link[0]) {
- trav->pavl_node = q;
- return trav->pavl_node != NULL ?
- &trav->pavl_node->pavl_data : NULL;
- }
- }
- else {
- trav->pavl_node = trav->pavl_node->pavl_link[1];
- while (trav->pavl_node->pavl_link[0] != NULL)
- trav->pavl_node = trav->pavl_node->pavl_link[0];
- return &trav->pavl_node->pavl_data;
- }
- }
- /* Returns the previous data item in inorder
- within the tree being traversed with |trav|,
- or if there are no more data items returns |NULL|. */
- struct pavlrc *pavlrc_t_prev(struct pavlrc_traverser *trav)
- {
- assert(trav != NULL);
- if (trav->pavl_node == NULL)
- return pavlrc_t_last(trav, trav->pavl_table);
- else if (trav->pavl_node->pavl_link[0] == NULL) {
- struct pavlrc_node *q, *p; /* Current node and its child. */
- for (p = trav->pavl_node, q = p->pavl_parent;;
- p = q, q = q->pavl_parent)
- if (q == NULL || p == q->pavl_link[1]) {
- trav->pavl_node = q;
- return trav->pavl_node != NULL ?
- &trav->pavl_node->pavl_data : NULL;
- }
- }
- else {
- trav->pavl_node = trav->pavl_node->pavl_link[0];
- while (trav->pavl_node->pavl_link[1] != NULL)
- trav->pavl_node = trav->pavl_node->pavl_link[1];
- return &trav->pavl_node->pavl_data;
- }
- }
- /* Returns |trav|'s current item. */
- struct pavlrc *pavlrc_t_cur(struct pavlrc_traverser *trav)
- {
- assert(trav != NULL);
- return trav->pavl_node != NULL ? &trav->pavl_node->pavl_data : NULL;
- }
- /* Replaces the current item in |trav| by |new| and returns the item replaced.
- |trav| must not have the null item selected.
- The new item must not upset the ordering of the tree. */
- struct pavlrc *pavlrc_t_replace(struct pavlrc_traverser *trav, struct pavlrc *new)
- {
- struct pavlrc *old;
- assert(trav != NULL && trav->pavl_node != NULL && new != NULL);
- old = &trav->pavl_node->pavl_data;
- trav->pavl_node->pavl_data = *new;
- return old;
- }
- /* Destroys |new| with |pavl_destroy (new, destroy)|,
- first initializing right links in |new| that have
- not yet been initialized at time of call. */
- static void
- copy_error_recovery(struct pavlrc_node *q,
- struct pavlrc_table *new)
- {
- assert(q != NULL && new != NULL);
- for (;;) {
- struct pavlrc_node *p = q;
- q = q->pavl_parent;
- if (q == NULL)
- break;
- if (p == q->pavl_link[0])
- q->pavl_link[1] = NULL;
- }
- pavlrc_destroy(new);
- }
- /* Copies |org| to a newly created tree, which is returned.
- If |copy != NULL|, each data item in |org| is first passed to |copy|,
- and the return values are inserted into the tree;
- |NULL| return values are taken as indications of failure.
- On failure, destroys the partially created new tree,
- applying |destroy|, if non-null, to each item in the new tree so far,
- and returns |NULL|.
- If |allocator != NULL|, it is used for allocation in the new tree.
- Otherwise, the same allocator used for |org| is used. */
- struct pavlrc_table *pavlrc_copy(const struct pavlrc_table *org,
- pavlrc_copy_func * copy,
- struct libavl_allocator *allocator)
- {
- struct pavlrc_table *new;
- struct pavlrc_node *x;
- struct pavlrc_node *y;
- assert(org != NULL);
- new = pavlrc_create(allocator != NULL ? allocator : org->pavl_alloc);
- if (new == NULL)
- return NULL;
- new->pavl_count = org->pavl_count;
- if (new->pavl_count == 0)
- return new;
- x = (struct pavlrc_node *)&org->pavl_root;
- y = (struct pavlrc_node *)&new->pavl_root;
- while (x != NULL) {
- while (x->pavl_link[0] != NULL) {
- y->pavl_link[0] =
- new->pavl_alloc->libavl_malloc(new->pavl_alloc,
- sizeof *y->pavl_link[0]);
- if (y->pavl_link[0] == NULL) {
- if (y != (struct pavlrc_node *)&new->pavl_root) {
- /* y->pavl_data = NULL; */
- y->pavl_link[1] = NULL;
- }
- copy_error_recovery(y, new);
- return NULL;
- }
- y->pavl_link[0]->pavl_parent = y;
- x = x->pavl_link[0];
- y = y->pavl_link[0];
- }
- y->pavl_link[0] = NULL;
- for (;;) {
- y->pavl_balance = x->pavl_balance;
- if (copy == NULL)
- y->pavl_data = x->pavl_data;
- else {
- y->pavl_data = *(struct pavlrc *)copy(&x->pavl_data);
- /*
- if (y->pavl_data == NULL) {
- y->pavl_link[1] = NULL;
- copy_error_recovery(y, new);
- return NULL;
- }
- */
- }
- if (x->pavl_link[1] != NULL) {
- y->pavl_link[1] =
- new->pavl_alloc->libavl_malloc(new->pavl_alloc,
- sizeof *y->pavl_link[1]);
- if (y->pavl_link[1] == NULL) {
- copy_error_recovery(y, new);
- return NULL;
- }
- y->pavl_link[1]->pavl_parent = y;
- x = x->pavl_link[1];
- y = y->pavl_link[1];
- break;
- }
- else
- y->pavl_link[1] = NULL;
- for (;;) {
- const struct pavlrc_node *w = x;
- x = x->pavl_parent;
- if (x == NULL) {
- new->pavl_root->pavl_parent = NULL;
- return new;
- }
- y = y->pavl_parent;
- if (w == x->pavl_link[0])
- break;
- }
- }
- }
- return new;
- }
- /* Frees storage allocated for |tree|.
- If |destroy != NULL|, applies it to each data item in inorder. */
- void pavlrc_destroy(struct pavlrc_table *tree)
- {
- struct pavlrc_node *p, *q;
- assert(tree != NULL);
- p = tree->pavl_root;
- while (p != NULL) {
- if (p->pavl_link[0] == NULL) {
- q = p->pavl_link[1];
- tree->pavl_alloc->libavl_free(tree->pavl_alloc, p);
- }
- else {
- q = p->pavl_link[0];
- p->pavl_link[0] = q->pavl_link[1];
- q->pavl_link[1] = p;
- }
- p = q;
- }
- tree->pavl_alloc->libavl_free(tree->pavl_alloc, tree);
- }
- /* Allocates |size| bytes of space using |malloc()|.
- Returns a null pointer if allocation fails. */
- void *pavlrc_malloc(struct libavl_allocator *allocator, size_t size)
- {
- assert(allocator != NULL && size > 0);
- return malloc(size);
- }
- /* Frees |block|. */
- void pavlrc_free(struct libavl_allocator *allocator, void *block)
- {
- assert(allocator != NULL && block != NULL);
- free(block);
- }
- /* Default memory allocator that uses |malloc()| and |free()|. */
- struct libavl_allocator pavlrc_allocator_default = {
- pavlrc_malloc,
- pavlrc_free
- };
- #undef NDEBUG
- #include <assert.h>
- /* Asserts that |pavl_insert()| succeeds at inserting |item| into |table|. */
- void (pavlrc_assert_insert) (struct pavlrc_table * table, struct pavlrc *item)
- {
- struct pavlrc *p = pavlrc_probe(table, item);
- assert(p != NULL && p == item);
- }
- /* Asserts that |pavl_delete()| really removes |item| from |table|,
- and returns the removed item. */
- struct pavlrc *(pavlrc_assert_delete) (struct pavlrc_table * table, struct pavlrc *item)
- {
- struct pavlrc *p = pavlrc_delete(table, item);
- assert(p != NULL);
- return p;
- }
|