Class: TestProf::Utils::SizedOrderedSet

Inherits:
Object
  • Object
show all
Includes:
Enumerable
Defined in:
lib/test_prof/utils/sized_ordered_set.rb

Overview

Ordered set with capacity

Instance Method Summary collapse

Constructor Details

#initialize(max_size, sort_by: nil) ⇒ SizedOrderedSet

Returns a new instance of SizedOrderedSet.



9
10
11
12
13
14
15
16
17
18
19
20
 
# File 'lib/test_prof/utils/sized_ordered_set.rb', line 9

def initialize(max_size, sort_by: nil)
  @max_size = max_size
  @comparator =
    if block_given?
      Proc.new
    elsif !sort_by.nil?
      ->(x, y) { x[sort_by] >= y[sort_by] }
    else
      ->(x, y) { x >= y }
    end
  @data = []
end

Instance Method Details

#<<(item) ⇒ Object 



22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'lib/test_prof/utils/sized_ordered_set.rb', line 22

def <<(item)
  return if data.size == max_size &&
            comparator.call(data.last, item)

  # Find an index of a smaller element
  index = data.bsearch_index { |x| !comparator.call(x, item) }

  if index.nil?
    data << item
  else
    data.insert(index, item)
  end

  data.pop if data.size > max_size
  data.size
end

#each ⇒ Object 



39
40
41
 
# File 'lib/test_prof/utils/sized_ordered_set.rb', line 39

def each
  data.each
end

#size ⇒ Object 



43
44
45
 
# File 'lib/test_prof/utils/sized_ordered_set.rb', line 43

def size
  data.size
end

#to_a ⇒ Object 



47
48
49
 
# File 'lib/test_prof/utils/sized_ordered_set.rb', line 47

def to_a
  data
end