diff options
author | Luke Shumaker <lukeshu@lukeshu.com> | 2022-07-13 20:31:51 -0600 |
---|---|---|
committer | Luke Shumaker <lukeshu@lukeshu.com> | 2022-07-13 20:31:51 -0600 |
commit | 09cc146211148a3b2568261c41a804a802c31d4c (patch) | |
tree | e64dc42bb557664333e313e89ddaafa975126e9a /lib/rbtree/rbtree_util.go | |
parent | a0fc940be0ea888c7b380cd95723da65bbd32bcb (diff) |
re-organize some things between files
Diffstat (limited to 'lib/rbtree/rbtree_util.go')
-rw-r--r-- | lib/rbtree/rbtree_util.go | 52 |
1 files changed, 0 insertions, 52 deletions
diff --git a/lib/rbtree/rbtree_util.go b/lib/rbtree/rbtree_util.go deleted file mode 100644 index cee5508..0000000 --- a/lib/rbtree/rbtree_util.go +++ /dev/null @@ -1,52 +0,0 @@ -// Copyright (C) 2022 Luke Shumaker <lukeshu@lukeshu.com> -// -// SPDX-License-Identifier: GPL-2.0-or-later - -package rbtree - -import ( - "reflect" - - "git.lukeshu.com/btrfs-progs-ng/lib/util" -) - -// SearchRange is like Search, but returns all nodes that match the -// function; assuming that they are contiguous. -func (t *Tree[K, V]) SearchRange(fn func(V) int) []V { - middle := t.Search(fn) - if middle == nil { - return nil - } - ret := []V{middle.Value} - for node := t.Prev(middle); node != nil && fn(node.Value) == 0; node = t.Prev(node) { - ret = append(ret, node.Value) - } - util.ReverseSlice(ret) - for node := t.Next(middle); node != nil && fn(node.Value) == 0; node = t.Next(node) { - ret = append(ret, node.Value) - } - return ret -} - -func (t *Tree[K, V]) Equal(u *Tree[K, V]) bool { - if (t == nil) != (u == nil) { - return false - } - if t == nil { - return true - } - - var tSlice []V - _ = t.Walk(func(node *Node[V]) error { - tSlice = append(tSlice, node.Value) - return nil - }) - - var uSlice []V - _ = u.Walk(func(node *Node[V]) error { - uSlice = append(uSlice, node.Value) - return nil - }) - - return reflect.DeepEqual(tSlice, uSlice) -} |