diff options
Diffstat (limited to 'pkg/rbtree/rbtree_util.go')
-rw-r--r-- | pkg/rbtree/rbtree_util.go | 48 |
1 files changed, 0 insertions, 48 deletions
diff --git a/pkg/rbtree/rbtree_util.go b/pkg/rbtree/rbtree_util.go deleted file mode 100644 index 2e7a48d..0000000 --- a/pkg/rbtree/rbtree_util.go +++ /dev/null @@ -1,48 +0,0 @@ -package rbtree - -import ( - "reflect" - - "lukeshu.com/btrfs-tools/pkg/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) -} |