diff options
-rw-r--r-- | lib.c | 2 | ||||
-rw-r--r-- | tree.c | 22 | ||||
-rw-r--r-- | tree.h | 1 | ||||
-rw-r--r-- | txr.1 | 24 |
4 files changed, 49 insertions, 0 deletions
@@ -10081,6 +10081,8 @@ val copy(val seq) return make_random_state(seq, nil); if (seq->co.cls == carray_s) return copy_carray(seq); + if (seq->co.cls == tree_s) + return copy_search_tree(seq); if (obj_struct_p(seq)) return copy_struct(seq); /* fallthrough */ @@ -637,6 +637,27 @@ static val tree_construct(val opts, val keys) return tree(keys, key_fn, less_fn, equal_fn); } +static val deep_copy_tnode(val node) +{ + if (node == nil) + return nil; + return tnode(node->tn.key, + deep_copy_tnode(node->tn.left), + deep_copy_tnode(node->tn.right)); +} + +val copy_search_tree(val tree) +{ + val self = lit("copy-search-tree"); + struct tree *ntr = coerce(struct tree *, malloc(sizeof *ntr)); + struct tree *otr = coerce(struct tree *, cobj_handle(self, tree, tree_s)); + val nroot = deep_copy_tnode(otr->root); + val ntree = cobj(coerce(mem_t *, ntr), tree_s, &tree_ops); + *ntr = *otr; + ntr->root = nroot; + return ntree; +} + val treep(val obj) { return tnil(type(obj) == COBJ && obj->co.cls == tree_s); @@ -711,6 +732,7 @@ void tree_init(void) reg_fun(intern(lit("copy-tnode"), user_package), func_n1(copy_tnode)); reg_fun(tree_s, func_n4o(tree, 0)); reg_fun(tree_construct_s, func_n2(tree_construct)); + reg_fun(intern(lit("copy-search-tree"), user_package), func_n1(copy_search_tree)); reg_fun(intern(lit("treep"), user_package), func_n1(treep)); reg_fun(intern(lit("tree-insert-node"), user_package), func_n2(tree_insert_node)); reg_fun(intern(lit("tree-insert"), user_package), func_n2(tree_insert)); @@ -39,6 +39,7 @@ val set_right(val node, val nright); val set_key(val node, val nkey); val copy_tnode(val node); val tree(val keys, val key_fn, val less_fn, val equal_fn); +val copy_search_tree(val tree); val treep(val obj); val tree_insert_node(val tree, val node); val tree_begin(val tree); @@ -28078,6 +28078,8 @@ the type of the argument, as follows: .meti (make-random-state << object ) .coIP tnode .meti (copy-tnode << object ) +.coIP tree +.meti (copy-search-tree << object ) .RE .IP @@ -45638,6 +45640,28 @@ is already empty, then the function returns otherwise it returns an integer which gives the count of the number of deleted nodes. +.coNP Function @ copy-search-tree +.synb +.mets (copy-search-tree << tree ) +.syne +.desc +The +.code copy-search-tree +returns a new tree object which is a copy of +.metn tree . + +The +.meta tree +argument must be an object of type +.codn tree . + +The returned object has the same key abstraction functions as +.meta tree +and contains the same elements. + +The nodes held inside the new tree are freshly allocated, +but their key objects are shared with the original tree. + .coNP Function @ tree-begin .synb .mets (tree-begin < tree ) |