diff --git a/include/tr/tree.h b/include/tr/tree.h index 1fbf230..8c832a6 100644 --- a/include/tr/tree.h +++ b/include/tr/tree.h @@ -23,104 +23,9 @@ #ifndef __TR_TREE_H__ #define __TR_TREE_H__ +#include "tr/tree_macros.h" #include "trbase.h" -#define TR_TREE_RIGHT(node) (NULL!=(node)?(node)->right:NULL) -#define TR_TREE_LEFT(node) (NULL!=(node)?(node)->left:NULL) -#define TR_TREE_PARENT(node) (NULL!=(node)?(node)->parent:NULL) - -/* - * Find data in a tree. - * Attention: This will change node, so normally you need to copy - * it before using this macro. - * Also be aware that found needs to be a valid lvalue and an integer. - */ -#define TR_TREE_FIND(node, search, found, comp) \ - (found) = -1; \ - while((node) && ((node)->left || (node)->right)) { \ - (found) = (comp)((node)->data, (search)); \ - if (0 != (found)) { \ - if (0 < (found)) { \ - (node) = (node)->left; \ - } else { \ - (node) = (node)->right; \ - } \ - } else { \ - break; \ - } \ - } - -#define TR_TREE_CHILD(node) \ - (NULL==TR_TREE_RIGHT((node))?TR_TREE_LEFT((node)):TR_TREE_RIGHT((node))) - -#define TR_TREE_SIBLING(node) \ - (NULL!=TR_TREE_PARENT((node))? \ - ((node)==TR_TREE_PARENT((node))->left? \ - TR_TREE_PARENT((node))->right: \ - TR_TREE_PARENT((node))->left): \ - NULL) - -#define TR_TREE_GRANDPARENT(node) (TR_TREE_PARENT((node))->parent) - -#define TR_TREE_UNCLE(node) \ - ((node)->parent == (node)->parent->parent->left? \ - (node)->parent->parent->right: \ - (node)->parent->parent->left) - -#define TR_TREE_REPLACE_NODE(root, node1, node2) \ - if (NULL != (node1)->parent) { \ - if ((node1) == (node1)->parent->left) { \ - (node1)->parent->left = (node2); \ - } else { \ - (node1)->parent->right = (node2); \ - } \ - } else { \ - *(root) = (node2); \ - } \ - if (NULL != (node2)) { \ - (node2)->parent = (node1)->parent; \ - } - -#define TR_TREE_ROT_RCLD_right(node) ((node)->left) -#define TR_TREE_ROT_RCLD_left(node) ((node)->right) - -#define TR_TREE_ROTATE(lr, root, node) \ - if (NULL != (node)) { \ - void * stPar = node->parent; \ - void * relCld = TR_TREE_ROT_RCLD_##lr(node); \ - void * relCCld = TR_TREE_ROT_RCLD_##lr(node)->lr; \ - void * nLeft_p = &TR_TREE_ROT_RCLD_##lr(node); \ - if (NULL != relCCld) { \ - TR_TREE_ROT_RCLD_##lr(node)->lr->parent = node; \ - } \ - TR_TREE_ROT_RCLD_##lr(node)->lr = node; \ - if (NULL != node->parent) { \ - if (node->parent->left == node) { \ - node->parent->left = relCld; \ - } else { \ - node->parent->right = relCld; \ - } \ - } else { \ - *(root) = relCld; \ - } \ - node->parent = relCld; \ - TR_TREE_ROT_RCLD_##lr(node)->parent = stPar; \ - *(void**)nLeft_p = relCCld; \ - } - -typedef enum {rbBlack=1, rbRed=2} TR_rbColor; - -#define TR_TREE_NODE_BLACK(node) (NULL == (node) || rbBlack == (node)->color) -#define TR_TREE_NODE_RED(node) (NULL == (node) || rbRed == (node)->color) -#define TR_TREE_NODE_STRICT_BLACK(node) (NULL != (node) && rbBlack == (node)->color) -#define TR_TREE_NODE_STRICT_RED(node) (NULL != (node) && rbRed == (node)->color) - -#define TR_TREE_INORDER_SUCC(node, succ) \ - succ = (node)->right; \ - while (NULL != succ->left) { \ - succ = succ->left; \ - } - TR_CLASS(TR_Tree) { void * data; diff --git a/src/tree/delete.c b/src/tree/delete.c index 56628a0..2a5bf61 100644 --- a/src/tree/delete.c +++ b/src/tree/delete.c @@ -26,9 +26,9 @@ void * TR_treeDelete(TR_Tree * this, const void * search, TR_TreeComp comp) { - TR_Tree node = *this; - TR_Tree del_node; - int found; + TR_Tree node = *this; + TR_Tree del_node; + int found; void * data; diff --git a/src/tree/find.c b/src/tree/find.c index fbe7b7c..d55aaa4 100644 --- a/src/tree/find.c +++ b/src/tree/find.c @@ -25,12 +25,11 @@ void * TR_treeFind(TR_Tree this, const void * search, TR_TreeComp comp) { - TR_Tree node = this; - int found; + int found; - TR_TREE_FIND(node, search, found, comp); + TR_TREE_FIND(this, search, found, comp); - return found == 0 ? node->data : NULL; + return found == 0 ? this->data : NULL; } // vim: set ts=4 sw=4: