Class: Immutable::SortedSet
- Inherits:
-
Object
- Object
- Immutable::SortedSet
- Includes:
- Enumerable
- Defined in:
- lib/immutable/sorted_set.rb
Overview
A ‘SortedSet` is a collection of ordered values with no duplicates. Unlike a Vector, in which items can appear in any arbitrary order, a `SortedSet` always keeps items either in their natural order, or in an order defined by a comparator block which is provided at initialization time.
‘SortedSet` uses `#<=>` (or its comparator block) to determine which items are equivalent. If the comparator indicates that an existing item and a new item are equal, any attempt to insert the new item will have no effect.
This means that all the items inserted into any one ‘SortedSet` must all be comparable. For example, you cannot put `String`s and `Integer`s in the same `SortedSet`. This is unlike Set, which can store items of any type, as long as they all support `#hash` and `#eql?`.
A ‘SortedSet` can be created in either of the following ways:
Immutable::SortedSet.new([1, 2, 3]) # any Enumerable can be used to initialize
Immutable::SortedSet['A', 'B', 'C', 'D']
Or if you want to use a custom ordering:
Immutable::SortedSet.new([1,2,3]) { |a, b| -a <=> -b }
Immutable::SortedSet.new([1, 2, 3]) { |num| -num }
‘SortedSet` can use a 2-parameter block which returns 0, 1, or -1 as a comparator (like `Array#sort`), or use a 1-parameter block to derive sort keys (like `Array#sort_by`) which will be compared using `#<=>`.
Like all ‘immutable-ruby` collections, `SortedSet`s are immutable. Any operation which you might expect to “modify” a `SortedSet` will actually return a new collection and leave the existing one unchanged.
‘SortedSet` supports the same basic set-theoretic operations as Set, including #union, #intersection, #difference, and #exclusion, as well as #subset?, #superset?, and so on. Unlike Set, it does not define comparison operators like `#>` or `#<` as aliases for the superset/subset predicates. Instead, these comparison operators do a item-by-item comparison between the `SortedSet` and another sequential collection. (See `Array#<=>` for details.)
Additionally, since ‘SortedSet`s are ordered, they also support indexed retrieval of items using #at or #[]. Like Vector, negative indices count back from the end of the `SortedSet`.
Getting the #max or #min item from a ‘SortedSet`, as defined by its comparator, is a constant time operation.
Defined Under Namespace
Classes: AVLNode, PlainAVLNode
Class Method Summary collapse
-
.[](*items) ⇒ SortedSet
Create a new ‘SortedSet` populated with the given items.
-
.alloc(node) ⇒ Set
“Raw” allocation of a new ‘SortedSet`.
-
.empty ⇒ SortedSet
Return an empty ‘SortedSet`.
-
.uniq_by_comparator!(array, comparator) ⇒ Object
Unfortunately, Ruby’s stdlib doesn’t do this for us array must be sorted.
Instance Method Summary collapse
-
#above(item, &block) ⇒ Object
Select elements greater than a value.
-
#add(item) ⇒ SortedSet
(also: #<<)
Return a new ‘SortedSet` with `item` added.
-
#add?(item) ⇒ SortedSet, false
If ‘item` is not a member of this `SortedSet`, return a new `SortedSet` with `item` added.
-
#at(index) ⇒ Object
Retrieve the item at ‘index`.
-
#below(item, &block) ⇒ Object
Select elements less than a value.
-
#between(from, to, &block) ⇒ Object
Select elements between two values.
-
#clear ⇒ SortedSet
Return an empty ‘SortedSet` instance, of the same class as this one.
-
#delete(item) ⇒ SortedSet
Return a new ‘SortedSet` with `item` removed.
-
#delete?(item) ⇒ SortedSet, false
If ‘item` is a member of this `SortedSet`, return a new `SortedSet` with `item` removed.
-
#delete_at(index) ⇒ SortedSet
Return a new ‘SortedSet` with the item at `index` removed.
-
#difference(other) ⇒ SortedSet
(also: #subtract, #-)
Return a new ‘SortedSet` with all the items in `other` removed.
-
#disjoint?(other) ⇒ Boolean
Return ‘true` if this set and `other` do not share any items.
-
#drop(n) ⇒ SortedSet
Drop the first ‘n` elements and return the rest in a new `SortedSet`.
-
#drop_while {|item| ... } ⇒ SortedSet, Enumerator
Drop elements up to, but not including, the first element for which the block returns ‘nil` or `false`.
-
#dup ⇒ SortedSet
(also: #clone)
Return ‘self`.
-
#each {|item| ... } ⇒ self, Enumerator
Call the given block once for each item in the set, passing each item from first to last successively to the block.
-
#empty? ⇒ Boolean
Return ‘true` if this `SortedSet` contains no items.
-
#eql?(other) ⇒ Boolean
Return true if ‘other` has the same type and contents as this `SortedSet`.
-
#exclusion(other) ⇒ SortedSet
(also: #^)
Return a new ‘SortedSet` with all the items which are members of this set or of `other`, but not both.
-
#fetch(index, default = (missing_default = true)) ⇒ Object
Retrieve the value at ‘index` with optional default.
-
#find_index(obj = (missing_obj = true), &block) ⇒ Integer
(also: #index)
Find the index of a given object or an element that satisfies the given block.
-
#first ⇒ Object
Return the “lowest” element in this set, as determined by its sort order.
-
#from(item, &block) ⇒ Object
Select elements greater than or equal to a value.
-
#hash ⇒ Integer
See ‘Object#hash`.
-
#include?(item) ⇒ Boolean
(also: #member?)
Return ‘true` if the given item is present in this `SortedSet`.
-
#initialize(items = [], &block) ⇒ SortedSet
constructor
A new instance of SortedSet.
-
#intersect?(other) ⇒ Boolean
Return ‘true` if this set and `other` have at least one item in common.
-
#intersection(other) ⇒ SortedSet
(also: #&)
Return a new ‘SortedSet` which contains all the items which are members of both this set and `other`.
-
#last ⇒ Object
Return the “highest” element in this set, as determined by its sort order.
-
#map {|item| ... } ⇒ SortedSet, Enumerator
(also: #collect)
Invoke the given block once for each item in the set, and return a new ‘SortedSet` containing the values returned by the block.
- #marshal_dump ⇒ ::Array
- #marshal_load(array) ⇒ Object
-
#max {|a, b| ... } ⇒ Object
Return the “highest” element in this set, as determined by its sort order.
-
#min {|a, b| ... } ⇒ Object
Return the “lowest” element in this set, as determined by its sort order.
-
#proper_subset?(other) ⇒ Boolean
Returns ‘true` if `other` contains all the items in this set, plus at least one item which is not in this set.
-
#proper_superset?(other) ⇒ Boolean
Returns ‘true` if this set contains all the items in `other`, plus at least one item which is not in `other`.
-
#reverse_each(&block) ⇒ self
Call the given block once for each item in the set, passing each item starting from the last, and counting back to the first, successively to the block.
-
#sample ⇒ Object
Return a randomly chosen item from this set.
-
#select {|item| ... } ⇒ SortedSet
(also: #find_all, #keep_if)
Return a new ‘SortedSet` containing all elements for which the given block returns true.
-
#size ⇒ Integer
(also: #length)
Return the number of items in this ‘SortedSet`.
-
#slice(arg, length = (missing_length = true)) ⇒ Object
(also: #[])
Return specific objects from the ‘Vector`.
-
#sort(&block) ⇒ SortedSet
(also: #sort_by)
Return a new ‘SortedSet` with the same items, but a sort order determined by the given block.
-
#subset?(other) ⇒ Boolean
Return ‘true` if all items in this set are also in `other`.
-
#superset?(other) ⇒ Boolean
Return ‘true` if all items in `other` are also in this set.
-
#take(n) ⇒ SortedSet
Return only the first ‘n` elements in a new `SortedSet`.
-
#take_while {|item| ... } ⇒ SortedSet, Enumerator
Gather elements up to, but not including, the first element for which the block returns ‘nil` or `false`, and return them in a new `SortedSet`.
-
#union(other) ⇒ SortedSet
(also: #|, #+, #merge)
Return a new ‘SortedSet` which contains all the members of both this set and `other`.
-
#up_to(item, &block) ⇒ Object
Select elements less than or equal to a value.
-
#values_at(*indices) ⇒ SortedSet
Return a new ‘SortedSet` with only the elements at the given `indices`.
Methods included from Enumerable
#<=>, #==, #compact, #each_index, #grep, #grep_v, #group_by, #inspect, #join, #partition, #pretty_print, #product, #reject, #sum, #to_set
Methods included from Enumerable
Constructor Details
#initialize(items = [], &block) ⇒ SortedSet
Returns a new instance of SortedSet.
101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 |
# File 'lib/immutable/sorted_set.rb', line 101 def initialize(items=[], &block) items = items.to_a if block if block.arity == 1 || block.arity == -1 items = items.uniq(&block) items.sort_by!(&block) comparator = lambda { |a,b| block.call(a) <=> block.call(b) } else items = items.sort(&block) SortedSet.uniq_by_comparator!(items, block) comparator = block end @node = AVLNode.from_items(items, comparator) else @node = PlainAVLNode.from_items(items.uniq.sort!) end freeze end |
Class Method Details
.[](*items) ⇒ SortedSet
Create a new ‘SortedSet` populated with the given items. This method does not accept a comparator block.
59 60 61 |
# File 'lib/immutable/sorted_set.rb', line 59 def [](*items) new(items) end |
.alloc(node) ⇒ Set
“Raw” allocation of a new ‘SortedSet`. Used internally to create a new instance quickly after obtaining a modified binary tree.
76 77 78 |
# File 'lib/immutable/sorted_set.rb', line 76 def alloc(node) allocate.tap { |s| s.instance_variable_set(:@node, node) }.freeze end |
.empty ⇒ SortedSet
Return an empty ‘SortedSet`. If used on a subclass, returns an empty instance of that class.
67 68 69 |
# File 'lib/immutable/sorted_set.rb', line 67 def empty @empty ||= self.alloc(PlainAVLNode::EmptyNode) end |
.uniq_by_comparator!(array, comparator) ⇒ Object
Unfortunately, Ruby’s stdlib doesn’t do this for us array must be sorted
83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
# File 'lib/immutable/sorted_set.rb', line 83 def uniq_by_comparator!(array, comparator) to_check, shift, sz, prev_obj = 1, 0, array.size, array[0] while to_check < sz next_obj = array[to_check] if comparator.call(prev_obj, next_obj) == 0 shift += 1 else if shift > 0 array[to_check - shift] = next_obj end prev_obj = next_obj end to_check += 1 end array.pop(shift) if shift > 0 end |
Instance Method Details
#above(item) ⇒ SortedSet #above(item) {|item| ... } ⇒ nil
Select elements greater than a value.
781 782 783 784 785 786 787 |
# File 'lib/immutable/sorted_set.rb', line 781 def above(item, &block) if block_given? @node.each_greater(item, false, &block) else self.class.alloc(@node.suffix(item, false)) end end |
#add(item) ⇒ SortedSet Also known as: <<
Return a new ‘SortedSet` with `item` added. If `item` is already in the set, return `self`.
147 148 149 150 151 152 153 |
# File 'lib/immutable/sorted_set.rb', line 147 def add(item) catch :present do node = @node.insert(item) return self.class.alloc(node) end self end |
#add?(item) ⇒ SortedSet, false
If ‘item` is not a member of this `SortedSet`, return a new `SortedSet` with `item` added. Otherwise, return `false`.
167 168 169 |
# File 'lib/immutable/sorted_set.rb', line 167 def add?(item) !include?(item) && add(item) end |
#at(index) ⇒ Object
Retrieve the item at ‘index`. If there is none (either the provided index is too high or too low), return `nil`.
231 232 233 234 235 |
# File 'lib/immutable/sorted_set.rb', line 231 def at(index) index += @node.size if index < 0 return nil if index >= @node.size || index < 0 @node.at(index) end |
#below(item) ⇒ SortedSet #below(item) {|item| ... } ⇒ nil
Select elements less than a value.
812 813 814 815 816 817 818 |
# File 'lib/immutable/sorted_set.rb', line 812 def below(item, &block) if block_given? @node.each_less(item, false, &block) else self.class.alloc(@node.prefix(item, false)) end end |
#between(from, to) ⇒ SortedSet #between(item) {|item| ... } ⇒ nil
Select elements between two values.
912 913 914 915 916 917 918 |
# File 'lib/immutable/sorted_set.rb', line 912 def between(from, to, &block) if block_given? @node.each_between(from, to, &block) else self.class.alloc(@node.between(from, to)) end end |
#clear ⇒ SortedSet
Return an empty ‘SortedSet` instance, of the same class as this one. Useful if you have multiple subclasses of `SortedSet` and want to treat them polymorphically.
934 935 936 937 938 939 940 |
# File 'lib/immutable/sorted_set.rb', line 934 def clear if @node.natural_order? self.class.empty else self.class.alloc(@node.clear) end end |
#delete(item) ⇒ SortedSet
Return a new ‘SortedSet` with `item` removed. If `item` is not a member of the set, return `self`.
180 181 182 183 184 185 186 187 188 189 190 |
# File 'lib/immutable/sorted_set.rb', line 180 def delete(item) catch :not_present do node = @node.delete(item) if node.empty? && node.natural_order? return self.class.empty else return self.class.alloc(node) end end self end |
#delete?(item) ⇒ SortedSet, false
If ‘item` is a member of this `SortedSet`, return a new `SortedSet` with `item` removed. Otherwise, return `false`.
203 204 205 |
# File 'lib/immutable/sorted_set.rb', line 203 def delete?(item) include?(item) && delete(item) end |
#delete_at(index) ⇒ SortedSet
Return a new ‘SortedSet` with the item at `index` removed. If the given `index` does not exist (if it is too high or too low), return `self`.
216 217 218 |
# File 'lib/immutable/sorted_set.rb', line 216 def delete_at(index) (item = at(index)) ? delete(item) : self end |
#difference(other) ⇒ SortedSet Also known as: subtract, -
Return a new ‘SortedSet` with all the items in `other` removed. `other` can be any `Enumerable` object.
658 659 660 |
# File 'lib/immutable/sorted_set.rb', line 658 def difference(other) self.class.alloc(@node.bulk_delete(other)) end |
#disjoint?(other) ⇒ Boolean
Return ‘true` if this set and `other` do not share any items.
735 736 737 738 739 740 741 742 |
# File 'lib/immutable/sorted_set.rb', line 735 def disjoint?(other) if size < other.size each { |item| return false if other.include?(item) } else other.each { |item| return false if include?(item) } end true end |
#drop(n) ⇒ SortedSet
Drop the first ‘n` elements and return the rest in a new `SortedSet`.
563 564 565 |
# File 'lib/immutable/sorted_set.rb', line 563 def drop(n) derive_new_sorted_set(@node.drop(n)) end |
#drop_while {|item| ... } ⇒ SortedSet, Enumerator
Drop elements up to, but not including, the first element for which the block returns ‘nil` or `false`. Gather the remaining elements into a new `SortedSet`. If no block is given, an `Enumerator` is returned instead.
589 590 591 592 593 594 595 596 597 |
# File 'lib/immutable/sorted_set.rb', line 589 def drop_while return enum_for(:drop_while) if not block_given? n = 0 each do |item| break unless yield item n += 1 end drop(n) end |
#dup ⇒ SortedSet Also known as: clone
Return ‘self`. Since this is an immutable object duplicates are equivalent.
967 968 969 |
# File 'lib/immutable/sorted_set.rb', line 967 def dup self end |
#each {|item| ... } ⇒ self, Enumerator
Call the given block once for each item in the set, passing each item from first to last successively to the block. If no block is provided, returns an ‘Enumerator`.
376 377 378 379 380 |
# File 'lib/immutable/sorted_set.rb', line 376 def each(&block) return @node.to_enum if not block_given? @node.each(&block) self end |
#empty? ⇒ Boolean
Return ‘true` if this `SortedSet` contains no items.
123 124 125 |
# File 'lib/immutable/sorted_set.rb', line 123 def empty? @node.empty? end |
#eql?(other) ⇒ Boolean
Return true if ‘other` has the same type and contents as this `SortedSet`.
946 947 948 949 950 951 952 953 954 955 956 |
# File 'lib/immutable/sorted_set.rb', line 946 def eql?(other) return true if other.equal?(self) return false if not instance_of?(other.class) return false if size != other.size a, b = self.to_enum, other.to_enum while true return false if !a.next.eql?(b.next) end rescue StopIteration true end |
#exclusion(other) ⇒ SortedSet Also known as: ^
Return a new ‘SortedSet` with all the items which are members of this set or of `other`, but not both. `other` can be any `Enumerable` object.
673 674 675 |
# File 'lib/immutable/sorted_set.rb', line 673 def exclusion(other) ((self | other) - (self & other)) end |
#fetch(index) ⇒ Object #fetch(index) {|index| ... } ⇒ Object #fetch(index, default) ⇒ Object
Retrieve the value at ‘index` with optional default.
276 277 278 279 280 281 282 283 284 285 286 |
# File 'lib/immutable/sorted_set.rb', line 276 def fetch(index, default = (missing_default = true)) if index >= -@node.size && index < @node.size at(index) elsif block_given? yield(index) elsif !missing_default default else raise IndexError, "index #{index} outside of sorted set bounds" end end |
#find_index(obj) ⇒ Integer #find_index {|element| ... } ⇒ Integer Also known as: index
Find the index of a given object or an element that satisfies the given block.
531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 |
# File 'lib/immutable/sorted_set.rb', line 531 def find_index(obj = (missing_obj = true), &block) if !missing_obj # Enumerable provides a default implementation, but this is more efficient node = @node index = node.left.size while !node.empty? direction = node.direction(obj) if direction > 0 node = node.right index += (node.left.size + 1) elsif direction < 0 node = node.left index -= (node.right.size + 1) else return index end end nil else super(&block) end end |
#first ⇒ Object
Return the “lowest” element in this set, as determined by its sort order.
416 417 418 |
# File 'lib/immutable/sorted_set.rb', line 416 def first @node.min end |
#from(item) ⇒ SortedSet #from(item) {|item| ... } ⇒ nil
Select elements greater than or equal to a value.
844 845 846 847 848 849 850 |
# File 'lib/immutable/sorted_set.rb', line 844 def from(item, &block) if block_given? @node.each_greater(item, true, &block) else self.class.alloc(@node.suffix(item, true)) end end |
#hash ⇒ Integer
See ‘Object#hash`.
960 961 962 |
# File 'lib/immutable/sorted_set.rb', line 960 def hash reduce(0) { |hash, item| (hash << 5) - hash + item.hash } end |
#include?(item) ⇒ Boolean Also known as: member?
Return ‘true` if the given item is present in this `SortedSet`. More precisely, return `true` if an object which compares as “equal” using this set’s comparator is present.
483 484 485 |
# File 'lib/immutable/sorted_set.rb', line 483 def include?(item) @node.include?(item) end |
#intersect?(other) ⇒ Boolean
Return ‘true` if this set and `other` have at least one item in common.
751 752 753 |
# File 'lib/immutable/sorted_set.rb', line 751 def intersect?(other) !disjoint?(other) end |
#intersection(other) ⇒ SortedSet Also known as: &
Return a new ‘SortedSet` which contains all the items which are members of both this set and `other`. `other` can be any `Enumerable` object.
644 645 646 |
# File 'lib/immutable/sorted_set.rb', line 644 def intersection(other) self.class.alloc(@node.keep_only(other)) end |
#last ⇒ Object
Return the “highest” element in this set, as determined by its sort order.
435 436 437 |
# File 'lib/immutable/sorted_set.rb', line 435 def last @node.max end |
#map {|item| ... } ⇒ SortedSet, Enumerator Also known as: collect
Invoke the given block once for each item in the set, and return a new ‘SortedSet` containing the values returned by the block. If no block is given, returns an `Enumerator`.
467 468 469 470 471 |
# File 'lib/immutable/sorted_set.rb', line 467 def map return enum_for(:map) if not block_given? return self if empty? self.class.alloc(@node.from_items(super)) end |
#marshal_dump ⇒ ::Array
974 975 976 977 978 979 980 |
# File 'lib/immutable/sorted_set.rb', line 974 def marshal_dump if @node.natural_order? to_a else raise TypeError, "can't dump SortedSet with custom sort order" end end |
#marshal_load(array) ⇒ Object
983 984 985 |
# File 'lib/immutable/sorted_set.rb', line 983 def marshal_load(array) initialize(array) end |
#max {|a, b| ... } ⇒ Object
Return the “highest” element in this set, as determined by its sort order. Or, if a block is provided, use the block as a comparator to find the “highest” element. (See ‘Enumerable#max`.)
429 430 431 |
# File 'lib/immutable/sorted_set.rb', line 429 def max block_given? ? super : @node.max end |
#min {|a, b| ... } ⇒ Object
Return the “lowest” element in this set, as determined by its sort order. Or, if a block is provided, use the block as a comparator to find the “lowest” element. (See ‘Enumerable#min`.)
410 411 412 |
# File 'lib/immutable/sorted_set.rb', line 410 def min block_given? ? super : @node.min end |
#proper_subset?(other) ⇒ Boolean
Returns ‘true` if `other` contains all the items in this set, plus at least one item which is not in this set.
710 711 712 713 |
# File 'lib/immutable/sorted_set.rb', line 710 def proper_subset?(other) return false if other.size <= size all? { |item| other.include?(item) } end |
#proper_superset?(other) ⇒ Boolean
Returns ‘true` if this set contains all the items in `other`, plus at least one item which is not in `other`.
724 725 726 |
# File 'lib/immutable/sorted_set.rb', line 724 def proper_superset?(other) other.proper_subset?(self) end |
#reverse_each(&block) ⇒ self
Call the given block once for each item in the set, passing each item starting from the last, and counting back to the first, successively to the block.
395 396 397 398 399 |
# File 'lib/immutable/sorted_set.rb', line 395 def reverse_each(&block) return @node.enum_for(:reverse_each) if not block_given? @node.reverse_each(&block) self end |
#sample ⇒ Object
Return a randomly chosen item from this set. If the set is empty, return ‘nil`.
926 927 928 |
# File 'lib/immutable/sorted_set.rb', line 926 def sample @node.at(rand(@node.size)) end |
#select {|item| ... } ⇒ SortedSet Also known as: find_all, keep_if
Return a new ‘SortedSet` containing all elements for which the given block returns true.
448 449 450 451 452 453 |
# File 'lib/immutable/sorted_set.rb', line 448 def select return enum_for(:select) unless block_given? items_to_delete = [] each { |item| items_to_delete << item unless yield(item) } derive_new_sorted_set(@node.bulk_delete(items_to_delete)) end |
#size ⇒ Integer Also known as: length
Return the number of items in this ‘SortedSet`.
133 134 135 |
# File 'lib/immutable/sorted_set.rb', line 133 def size @node.size end |
#set.slice(index) ⇒ Object #set.slice(index, length) ⇒ SortedSet #set.slice(index..end) ⇒ SortedSet Also known as: []
Return specific objects from the ‘Vector`. All overloads return `nil` if the starting index is out of range.
328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 |
# File 'lib/immutable/sorted_set.rb', line 328 def slice(arg, length = (missing_length = true)) if missing_length if arg.is_a?(Range) from, to = arg.begin, arg.end from += @node.size if from < 0 to += @node.size if to < 0 to += 1 if !arg.exclude_end? length = to - from length = 0 if length < 0 subsequence(from, length) else at(arg) end else arg += @node.size if arg < 0 subsequence(arg, length) end end |
#sort(&block) ⇒ SortedSet Also known as: sort_by
Return a new ‘SortedSet` with the same items, but a sort order determined by the given block.
498 499 500 501 502 503 504 505 506 |
# File 'lib/immutable/sorted_set.rb', line 498 def sort(&block) if block self.class.new(self.to_a, &block) elsif @node.natural_order? self else self.class.new(self) end end |
#subset?(other) ⇒ Boolean
Return ‘true` if all items in this set are also in `other`.
685 686 687 688 |
# File 'lib/immutable/sorted_set.rb', line 685 def subset?(other) return false if other.size < size all? { |item| other.include?(item) } end |
#superset?(other) ⇒ Boolean
Return ‘true` if all items in `other` are also in this set.
697 698 699 |
# File 'lib/immutable/sorted_set.rb', line 697 def superset?(other) other.subset?(self) end |
#take(n) ⇒ SortedSet
Return only the first ‘n` elements in a new `SortedSet`.
575 576 577 |
# File 'lib/immutable/sorted_set.rb', line 575 def take(n) derive_new_sorted_set(@node.take(n)) end |
#take_while {|item| ... } ⇒ SortedSet, Enumerator
Gather elements up to, but not including, the first element for which the block returns ‘nil` or `false`, and return them in a new `SortedSet`. If no block is given, an `Enumerator` is returned instead.
609 610 611 612 613 614 615 616 617 |
# File 'lib/immutable/sorted_set.rb', line 609 def take_while return enum_for(:take_while) if not block_given? n = 0 each do |item| break unless yield item n += 1 end take(n) end |
#union(other) ⇒ SortedSet Also known as: |, +, merge
Return a new ‘SortedSet` which contains all the members of both this set and `other`. `other` can be any `Enumerable` object.
628 629 630 |
# File 'lib/immutable/sorted_set.rb', line 628 def union(other) self.class.alloc(@node.bulk_insert(other)) end |
#up_to(item) ⇒ SortedSet #up_to(item) {|item| ... } ⇒ nil
Select elements less than or equal to a value.
878 879 880 881 882 883 884 |
# File 'lib/immutable/sorted_set.rb', line 878 def up_to(item, &block) if block_given? @node.each_less(item, true, &block) else self.class.alloc(@node.prefix(item, true)) end end |
#values_at(*indices) ⇒ SortedSet
Return a new ‘SortedSet` with only the elements at the given `indices`. If any of the `indices` do not exist, they will be skipped.
357 358 359 360 |
# File 'lib/immutable/sorted_set.rb', line 357 def values_at(*indices) indices.select! { |i| i >= -@node.size && i < @node.size } self.class.new(indices.map! { |i| at(i) }) end |