/* * Copyright 1995-2015 Andrew Smith * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License as published by the Free * Software Foundation; either version 3 of the License, or (at your option) * any later version. See COPYING for more details. */ #include "ktree.h" static const int dbg = 0; #define DBG if (dbg != 0) printf #define FAIL(fmt, ...) do \ { \ quithere(1, fmt KTREE_FFL, ##__VA_ARGS__, KTREE_FFL_PASS); \ } while (0); #define RED_RED true #define RED_BLACK false #define Yo true #define No false static K_NODE nil[1] = { { Yo, RED_BLACK, NULL, NULL, NULL, NULL, 0 } }; static K_NODE *_new_knode(KTREE_FFL_ARGS) { K_NODE *node = (K_NODE *)malloc(sizeof(*node)); if (node == NULL) FAIL("%s", "node OOM"); node->isNil = Yo; node->red = RED_BLACK; node->parent = nil; node->left = nil; node->right = nil; node->data = NULL; node->test = 0; return node; } K_TREE *_new_ktree(cmp_t (*cmp_funct)(K_ITEM *, K_ITEM *), K_LIST *master, KTREE_FFL_ARGS) { K_TREE *tree = (K_TREE *)malloc(sizeof(*tree)); if (tree == NULL) FAIL("%s", "tree OOM"); tree->root = _new_knode(KTREE_FFL_PASS); tree->cmp_funct = cmp_funct; tree->master = master; return tree; } static K_NODE *new_data(K_ITEM *data, KTREE_FFL_ARGS) { K_NODE *knode = (K_NODE *)malloc(sizeof(*knode)); if (knode == NULL) FAIL("%s", "OOM"); knode->isNil = No; knode->red = RED_RED; knode->parent = nil; knode->left = nil; knode->right = nil; knode->data = data; knode->test = 0; return knode; } static int bCount = 0; static long bTestValue = 0; static long nilTestValue = 0; static long lrpTestValue = 0; static long testValue = 1231230L; static long getTestValue() { return ++testValue; } static void show_ktree(K_NODE *node, char *path, int pos, char *(*dsp_funct)(K_ITEM *)) { char col; if (node->isNil == Yo) return; if (node->left->isNil == No) { path[pos] = 'L'; path[pos+1] = '\0'; show_ktree(node->left, path, pos+1, dsp_funct); } path[pos] = '\0'; if (node->red == RED_RED) col = 'R'; else // if (node->red == RED_BLACK) col = 'B'; printf(" %c %s=%s\n", col, path, dsp_funct(node->data)); if (node->right->isNil == No) { path[pos] = 'R'; path[pos+1] = '\0'; show_ktree(node->right, path, pos+1, dsp_funct); } } void _dump_ktree(K_TREE *tree, char *(*dsp_funct)(K_ITEM *), KTREE_FFL_ARGS) { char buf[42424]; _TREE_READ(tree, true, file, func, line); printf("dump:\n"); if (tree->root->isNil == No) { buf[0] = 'T'; buf[1] = '\0'; show_ktree(tree->root, buf, 1, dsp_funct); } else printf(" Empty tree\n"); } void _dsp_ktree(K_TREE *tree, char *filename, char *msg, KTREE_FFL_ARGS) { K_TREE_CTX ctx[1]; K_ITEM *item; FILE *stream; struct tm tm; time_t now_t; char stamp[128]; if (!(tree->master->dsp_func)) FAIL("NULLDSP NULL dsp_func in %s", tree->master->name); _TREE_READ(tree, true, file, func, line); now_t = time(NULL); localtime_r(&now_t, &tm); snprintf(stamp, sizeof(stamp), "[%d-%02d-%02d %02d:%02d:%02d]", tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday, tm.tm_hour, tm.tm_min, tm.tm_sec); stream = fopen(filename, "ae"); if (!stream) { fprintf(stderr, "%s %s() failed to open '%s' (%d) %s", stamp, __func__, filename, errno, strerror(errno)); return; } if (msg) fprintf(stream, "%s %s\n", stamp, msg); else fprintf(stream, "%s Dump of tree '%s':\n", stamp, tree->master->name); if (tree->root->isNil == No) { item = first_in_ktree(tree, ctx); while (item) { tree->master->dsp_func(item, stream); item = next_in_ktree(ctx); } fprintf(stream, "End\n\n"); } else fprintf(stream, "Empty ktree\n\n"); fclose(stream); } static int nilTest(K_NODE *node, char *msg, int depth, int count, K_NODE *nil2, KTREE_FFL_ARGS) { if (node->isNil == Yo || node == nil2) { if (node == nil2 && node->isNil == No) FAIL("NIL2NOTNIL '%s' depth=%d count=%d", msg, depth, count); if (node != nil && node != nil2) FAIL("NOTNILNIL2 '%s' depth=%d count=%d", msg, depth, count); if (node->red == RED_RED) FAIL("NIRED '%s' depth=%d count=%d", msg, depth, count); if (node != nil2 && node->parent != NULL) FAIL("NILPARENT '%s' depth=%d count=%d", msg, depth, count); if (node != nil2 && node->left != NULL) FAIL("NILLEFT '%s' depth=%d count=%d", msg, depth, count); if (node == nil2 && node->left != nil) FAIL("NIL2LEFT '%s' depth=%d count=%d", msg, depth, count); if (node != nil2 && node->right != NULL) FAIL("NILRIGHT '%s' depth=%d count=%d", msg, depth, count); if (node == nil2 && node->right != nil) FAIL("NIL2RIGHT '%s' depth=%d count=%d", msg, depth, count); if (node->data != NULL) FAIL("NIDATA '%s' depth=%d count=%d", msg, depth, count); } else { count++; if (node->parent == NULL) FAIL("NOPAR '%s' depth=%d count=%d", msg, depth, count); if (node->left == NULL) FAIL("NOLEFT '%s' depth=%d count=%d", msg, depth, count); if (node->right == NULL) FAIL("NORIGHT '%s' depth=%d count=%d", msg, depth, count); if (node->data == NULL) FAIL("NODATA '%s' depth=%d count=%d", msg, depth, count); if (node->test != nilTestValue) node->test = nilTestValue; else FAIL("NILTESTVALUE '%s' depth=%d count=%d", msg, depth, count); count = nilTest(node->left, msg, depth+1, count, nil2, KTREE_FFL_PASS); count = nilTest(node->right, msg, depth+1, count, nil2, KTREE_FFL_PASS); } return(count); } static void bTest(K_NODE *cur, char *msg, int count, KTREE_FFL_ARGS) { if (cur->red != RED_RED) count++; else { if (cur->left->red == RED_RED) FAIL("CURLR '%s' count=%d", msg, count); if (cur->right->red == RED_RED) FAIL("CURRR '%s' count=%d", msg, count); } if (cur->isNil == Yo) { if (bCount == 0) bCount = count; if (count != bCount) FAIL("BCOUNT '%s' count=%d bCount=%d", msg, count, bCount); } else { if (cur->test != bTestValue) cur->test = bTestValue; else FAIL("BTESTVALUE '%s' count=%d", msg, count); bTest(cur->left, msg, count, KTREE_FFL_PASS); bTest(cur->right, msg, count, KTREE_FFL_PASS); } } static void bTestInit(K_TREE *tree, char *msg, KTREE_FFL_ARGS) { bCount = 0; bTestValue = getTestValue(); bTest(tree->root, msg, 0, KTREE_FFL_PASS); } static void lrpTest(K_NODE *node, char *msg, KTREE_FFL_ARGS) { if (node->test != lrpTestValue) node->test = lrpTestValue; else FAIL("LRPTESTVALUE '%s'", msg); if (node->left->isNil == No) { if (node->left->parent != node) FAIL("LRPTESTL '%s'", msg); lrpTest(node->left, msg, KTREE_FFL_PASS); } if (node->right->isNil == No) { if (node->right->parent != node) FAIL("LRPTESTR '%s'", msg); lrpTest(node->right, msg, KTREE_FFL_PASS); } } static __maybe_unused void check_ktree(K_TREE *tree, char *msg, K_NODE *nil2, int debugNil, int debugLRP, int debugColor, KTREE_FFL_ARGS) { if (tree->root->isNil == Yo) return; if (debugNil) { nilTestValue = getTestValue(); nilTest(tree->root, msg, 1, 0, nil2, KTREE_FFL_PASS); } if (debugLRP && tree->root->isNil == No) { lrpTestValue = getTestValue(); lrpTest(tree->root, msg, KTREE_FFL_PASS); } if (debugColor && tree->root->isNil == No) bTestInit(tree, msg, KTREE_FFL_PASS); } static K_ITEM *_first_in_knode(K_NODE *node, K_TREE_CTX *ctx, KTREE_FFL_ARGS) { if (node->isNil == No) { while (node->left->isNil == No) node = node->left; *ctx = node; return(node->data); } *ctx = NULL; return(NULL); } K_ITEM *_first_in_ktree(K_TREE *tree, K_TREE_CTX *ctx, LOCK_MAYBE bool chklock, KTREE_FFL_ARGS) { _TREE_READ(tree, chklock, file, func, line); return _first_in_knode(tree->root, ctx, KTREE_FFL_PASS); } static K_ITEM *_last_in_knode(K_NODE *node, K_TREE_CTX *ctx, KTREE_FFL_ARGS) { if (node->isNil == No) { while (node->right->isNil == No) node = node->right; *ctx = node; return(node->data); } *ctx = NULL; return(NULL); } K_ITEM *_last_in_ktree(K_TREE *tree, K_TREE_CTX *ctx, KTREE_FFL_ARGS) { _TREE_READ(tree, true, file, func, line); return _last_in_knode(tree->root, ctx, KTREE_FFL_PASS); } /* TODO: change ctx to a structure of tree and node then can test _TREE_READ * However, next/prev is never called before a first/last/find so it's less * likely to see an error i.e. if code was missing a lock it would be seen * by first/last/find - here would only see coding errors e.g. looping * outside the lock */ K_ITEM *_next_in_ktree(K_TREE_CTX *ctx, KTREE_FFL_ARGS) { K_NODE *parent; K_NODE *knode = (K_NODE *)(*ctx); if (knode->isNil == No) { if (knode->right->isNil == No) return(_first_in_knode(knode->right, ctx, KTREE_FFL_PASS)); else { parent = knode->parent; while (parent->isNil == No && knode == parent->right) { knode = parent; parent = parent->parent; } if (parent->isNil == No) { *ctx = parent; return(parent->data); } } } *ctx = NULL; return(NULL); } K_ITEM *_prev_in_ktree(K_TREE_CTX *ctx, KTREE_FFL_ARGS) { K_NODE *parent; K_NODE *knode = (K_NODE *)(*ctx); if (knode->isNil == No) { if (knode->left->isNil == No) return(_last_in_knode(knode->left, ctx, KTREE_FFL_PASS)); else { parent = knode->parent; while (parent->isNil == No && knode == parent->left) { knode = parent; parent = parent->parent; } if (parent->isNil == No) { *ctx = parent; return(parent->data); } } } *ctx = NULL; return(NULL); } static K_NODE *left_rotate(K_NODE *root, K_NODE *about) { K_NODE *rotate; rotate = about->right; about->right = rotate->left; if (rotate->left->isNil == No) rotate->left->parent = about; rotate->parent = about->parent; if (about->parent->isNil == Yo) root = rotate; else { if (about == about->parent->left) about->parent->left = rotate; else about->parent->right = rotate; } rotate->left = about; about->parent = rotate; return(root); } static K_NODE *right_rotate(K_NODE *root, K_NODE *about) { K_NODE *rotate; rotate = about->left; about->left = rotate->right; if (rotate->right->isNil == No) rotate->right->parent = about; rotate->parent = about->parent; if (about->parent->isNil == Yo) root = rotate; else if (about == about->parent->right) about->parent->right = rotate; else about->parent->left = rotate; rotate->right = about; about->parent = rotate; return(root); } void _add_to_ktree(K_TREE *tree, K_ITEM *data, LOCK_MAYBE bool chklock, KTREE_FFL_ARGS) { K_NODE *knode; K_NODE *x, *y; K_NODE *pp; cmp_t cmp; if (tree == NULL) FAIL("%s", "ADDNULL add tree is NULL"); //check_ktree(tree, ">add", NULL, 1, 1, 1, KTREE_FFL_PASS); if (tree->root->parent != nil && tree->root->parent != NULL) FAIL("%s", "ADDROOT add tree->root isn't the root"); _TREE_WRITE(tree, chklock, file, func, line); knode = new_data(data, KTREE_FFL_PASS); if (tree->root->isNil == Yo) { if (tree->root != nil) free(tree->root); tree->root = knode; } else { x = tree->root; y = nil; while (x->isNil == No) { y = x; if ((cmp = tree->cmp_funct(knode->data, x->data)) < 0) x = x->left; else x = x->right; } knode->parent = y; if (cmp < 0) y->left = knode; else y->right = knode; x = knode; while (x != tree->root && x->parent->red == RED_RED) { pp = x->parent->parent; if (x->parent == pp->left) { y = pp->right; if (y->red == RED_RED) { x->parent->red = RED_BLACK; y->red = RED_BLACK; pp->red = RED_RED; x = pp; } else { if (x == x->parent->right) { x = x->parent; tree->root = left_rotate(tree->root, x); pp = x->parent->parent; } x->parent->red = RED_BLACK; pp->red = RED_RED; tree->root = right_rotate(tree->root, pp); } } else { y = pp->left; if (y->red == RED_RED) { x->parent->red = RED_BLACK; y->red = RED_BLACK; pp->red = RED_RED; x = pp; } else { if (x == x->parent->left) { x = x->parent; tree->root = right_rotate(tree->root, x); pp = x->parent->parent; } x->parent->red = RED_BLACK; pp->red = RED_RED; tree->root = left_rotate(tree->root, pp); } } } } tree->root->red = RED_BLACK; //check_ktree(tree, "root == NULL) FAIL("%s", "FINDNULL find tree->root is NULL"); if (chklock) { _TREE_READ(tree, true, file, func, line); } knode = tree->root; while (knode->isNil == No && cmp != 0) { if ((cmp = tree->cmp_funct(knode->data, data))) { if (cmp > 0) knode = knode->left; else knode = knode->right; } } if (knode->isNil == No) { *ctx = knode; return(knode->data); } else { *ctx = NULL; return(NULL); } } K_ITEM *_find_after_in_ktree(K_TREE *tree, K_ITEM *data, K_TREE_CTX *ctx, LOCK_MAYBE bool chklock, KTREE_FFL_ARGS) { K_NODE *knode, *old = NULL; cmp_t cmp = -1, oldcmp = -1; if (tree == NULL) FAIL("%s", "FINDNULL find_after tree is NULL"); if (tree->root == NULL) FAIL("%s", "FINDNULL find_after tree->root is NULL"); _TREE_READ(tree, chklock, file, func, line); knode = tree->root; while (knode->isNil == No && cmp != 0) { if ((cmp = tree->cmp_funct(knode->data, data))) { old = knode; oldcmp = cmp; if (cmp > 0) knode = knode->left; else knode = knode->right; } } if (knode->isNil == No) { *ctx = knode; return next_in_ktree(ctx); } else { if (old) { if (oldcmp > 0) { *ctx = old; return(old->data); } *ctx = old; return next_in_ktree(ctx); } *ctx = NULL; return(NULL); } } K_ITEM *_find_before_in_ktree(K_TREE *tree, K_ITEM *data, K_TREE_CTX *ctx, KTREE_FFL_ARGS) { K_NODE *knode, *old = NULL; cmp_t cmp = 1, oldcmp = 1; if (tree == NULL) FAIL("%s", "FINDNULL find_before tree is NULL"); if (tree->root == NULL) FAIL("%s", "FINDNULL find_before tree->root is NULL"); _TREE_READ(tree, true, file, func, line); knode = tree->root; while (knode->isNil == No && cmp != 0) { if ((cmp = tree->cmp_funct(knode->data, data))) { old = knode; oldcmp = cmp; if (cmp > 0) knode = knode->left; else knode = knode->right; } } if (knode->isNil == No) { *ctx = knode; return prev_in_ktree(ctx); } else { if (old) { if (oldcmp < 0) { *ctx = old; return(old->data); } *ctx = old; return prev_in_ktree(ctx); } *ctx = NULL; return(NULL); } } static K_NODE *removeFixup(K_NODE *root, K_NODE *fix) { K_NODE *w = NULL; while (fix != root && fix->red != RED_RED) { if (fix == fix->parent->left) { w = fix->parent->right; if (w->red == RED_RED) { w->red = RED_BLACK; fix->parent->red = RED_RED; root = left_rotate(root, fix->parent); w = fix->parent->right; } if (w->left->red != RED_RED && w->right->red != RED_RED) { w->red = RED_RED; fix = fix->parent; } else { if (w->right->red != RED_RED) { w->left->red = RED_BLACK; w->red = RED_RED; root = right_rotate(root, w); w = fix->parent->right; } w->red = fix->parent->red; fix->parent->red = RED_BLACK; w->right->red = RED_BLACK; root = left_rotate(root, fix->parent); fix = root; } } else { w = fix->parent->left; if (w->red == RED_RED) { w->red = RED_BLACK; fix->parent->red = RED_RED; root = right_rotate(root, fix->parent); w = fix->parent->left; } if (w->right->red != RED_RED && w->left->red != RED_RED) { w->red = RED_RED; fix = fix->parent; } else { if (w->left->red != RED_RED) { w->right->red = RED_BLACK; w->red = RED_RED; root = left_rotate(root, w); w = fix->parent->left; } w->red = fix->parent->red; fix->parent->red = RED_BLACK; w->left->red = RED_BLACK; root = right_rotate(root, fix->parent); fix = root; } } } fix->red = RED_BLACK; return root; } // Does this work OK when you remove the last element in the tree? // It should return the root as 'nil' void _remove_from_ktree(K_TREE *tree, K_ITEM *data, K_TREE_CTX *ctx, KTREE_FFL_ARGS) { K_TREE_CTX tmpctx[1]; K_NODE *found; K_ITEM *fdata; K_NODE *x, *y, *nil2; // cmp_t cmp; int yred; //check_ktree(tree, ">remove", NULL, 1, 1, 1, KTREE_FFL_PASS); if (tree == NULL) FAIL("%s", "REMNULL remove tree is NULL"); if (tree->root == NULL) FAIL("%s", "REMNULL remove tree->root is NULL"); _TREE_WRITE(tree, true, file, func, line); if (tree->root->isNil == Yo) { *ctx = NULL; return; } if (tree->root->parent->isNil == No) FAIL("%s", "REMROOT remove tree->root isn't the root"); fdata = find_in_ktree(tree, data, ctx); if (fdata == NULL) return; if (tree->cmp_funct(fdata, data) != 0) FAIL("%s", "BADFIND cmp(found, remove) != 0"); found = *ctx; x = nil; y = nil; if (found->left->isNil == Yo || found->right->isNil == Yo) y = found; else { *tmpctx = found; next_in_ktree(tmpctx); y = *tmpctx; } yred = y->red; if (y->left->isNil == No && y->right->isNil == No) FAIL("%s", "REMBADY remove error"); if (y->left->isNil == Yo) x = y->right; else x = y->left; if (x != nil) nil2 = NULL; else { nil2 = _new_knode(KTREE_FFL_PASS); x = nil2; } x->parent = y->parent; if (x->parent->isNil == Yo) tree->root = x; else { if (x->parent->left == y) x->parent->left = x; else x->parent->right = x; } if (y != found) { if (tree->root == found) tree->root = y; if (x == found) x = y; y->red = found->red; y->parent = found->parent; if (y->parent->isNil == No) { if (y->parent->left == found) y->parent->left = y; else y->parent->right = y; } y->left = found->left; if (y->left->isNil == No || y->left == nil2) y->left->parent = y; y->right = found->right; if (y->right->isNil == No || y->right == nil2) y->right->parent = y; } if (yred != RED_RED) tree->root = removeFixup(tree->root, x); if (nil2 != NULL) { if (nil2->parent->isNil == No && nil2->parent->left == nil2) nil2->parent->left = nil; if (nil2->parent->isNil == No && nil2->parent->right == nil2) nil2->parent->right = nil; if (tree->root == nil2) tree->root = nil; /* if (dbg != 0) { if (nil2->left != nil) { DBG("@remove nil2->left wasn't nil!!!\n"); } if (nil2->right != nil) { DBG("@remove nil2->right wasn't nil!!!\n"); } cmp = 0; fdata = first_in_ktree(tree, tmpctx);; while (fdata != NULL) { cmp++; x = *tmpctx; if (x == nil2) { DBG("@remove found nil2 in ktree %d!!!\n", (int)cmp); } else if (x->left == nil2) { DBG("@remove found nil2 in ktree(left) %d!!!\n", (int)cmp); } else if (x->right == nil2) { DBG("@remove found nil2 in ktree(right) %d!!!\n", (int)cmp); } fdata = next_in_ktree(tmpctx);; } } */ free(nil2); } /* if (dbg != 0) { cmp = 0; fdata = first_in_ktree(tree, tmpctx);; while (fdata != NULL) { if (tree->cmp_funct(fdata, tree->root->data) < 0) cmp--; else cmp++; fdata = next_in_ktree(tmpctx);; } if (cmp < -10 || cmp > 10) { DBG("@remove after balance=%d :(\n", (int)cmp); } } */ //check_ktree(tree, "data != NULL && free_funct) free_funct(knode->data); free_ktree_sub(knode->left, free_funct); free_ktree_sub(knode->right, free_funct); free(knode); } } void _free_ktree(K_TREE *tree, void (*free_funct)(void *), KTREE_FFL_ARGS) { if (tree == NULL) FAIL("%s", "FREENULL free NULL tree"); if (tree->root->parent != NULL && tree->root->parent != nil) FAIL("%s", "FREENOTROOT free tree->root not root"); free_ktree_sub(tree->root, free_funct); }