Module: Enumerable

Defined in:
lib/lite/ruby/enumerable.rb

Instance Method Summary collapse

Instance Method Details

#cluster ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
 
# File 'lib/lite/ruby/enumerable.rb', line 6

def cluster
  each_with_object([]) do |ele, results|
    last_res = results.last

    if last_res && (yield(ele) == yield(last_res.last))
      last_res << ele
    else
      results << [ele]
    end
  end
end

#cluster_by(&block) ⇒ Object 



18
19
20
 
# File 'lib/lite/ruby/enumerable.rb', line 18

def cluster_by(&block)
  group_by(&block).sort.transpose.pop || []
end

#deduce(identity = 0, &block) ⇒ Object 



22
23
24
25
26
27
28
 
# File 'lib/lite/ruby/enumerable.rb', line 22

def deduce(identity = 0, &block)
  if block_given?
    map(&block).deduce(identity)
  else
    inject { |acc, val| acc - val } || identity
  end
end

#drop_last(num) ⇒ Object 



30
31
32
33
34
35
 
# File 'lib/lite/ruby/enumerable.rb', line 30

def drop_last(num)
  collection_size = to_a.size
  return self if num > collection_size

  self[0...(collection_size - num)]
end

#drop_last_if ⇒ Object 



37
38
39
40
41
42
43
44
 
# File 'lib/lite/ruby/enumerable.rb', line 37

def drop_last_if
  dropping = true
  reverse_each.with_object([]) do |val, arr|
    next if dropping &&= yield(val)

    arr.unshift(val)
  end
end

#exactly?(num) ⇒ Boolean

Returns:

  • (Boolean) 


46
47
48
49
50
51
52
53
54
55
56
 
# File 'lib/lite/ruby/enumerable.rb', line 46

def exactly?(num)
  found_count = 0

  if block_given?
    each { |*opt| found_count += 1 if yield(*opt) }
  else
    each { |opt| found_count += 1 if opt }
  end

  found_count > num ? false : num == found_count
end

#excase?(object) ⇒ Boolean

rubocop:disable Style/CaseEquality

Returns:

  • (Boolean) 


59
60
61
 
# File 'lib/lite/ruby/enumerable.rb', line 59

def excase?(object)
  none? { |val| object === val }
end

#exclude?(object) ⇒ Boolean

rubocop:enable Style/CaseEquality

Returns:

  • (Boolean) 


64
65
66
 
# File 'lib/lite/ruby/enumerable.rb', line 64

def exclude?(object)
  !include?(object)
end

#excluding(*elements) ⇒ Object Also known as: without 



68
69
70
71
 
# File 'lib/lite/ruby/enumerable.rb', line 68

def excluding(*elements)
  elements.flatten!(1)
  reject { |element| elements.include?(element) }
end

#expand ⇒ Object 



75
76
77
 
# File 'lib/lite/ruby/enumerable.rb', line 75

def expand
  map { |val| val.is_a?(Enumerable) ? val.expand : val }
end

#exponential(identity = 0, &block) ⇒ Object 



79
80
81
82
83
84
85
 
# File 'lib/lite/ruby/enumerable.rb', line 79

def exponential(identity = 0, &block)
  if block_given?
    map(&block).exponential(identity)
  else
    inject { |acc, val| acc**val } || identity
  end
end

#frequency ⇒ Object Also known as: occurrences 



87
88
89
 
# File 'lib/lite/ruby/enumerable.rb', line 87

def frequency
  each_with_object(Hash.new(0)) { |val, hash| hash[val] += 1 }
end

#incase?(object) ⇒ Boolean

rubocop:disable Style/CaseEquality

Returns:

  • (Boolean) 


94
95
96
 
# File 'lib/lite/ruby/enumerable.rb', line 94

def incase?(object)
  any? { |val| object === val }
end

#including(*elements) ⇒ Object Also known as: with

rubocop:enable Style/CaseEquality



99
100
101
 
# File 'lib/lite/ruby/enumerable.rb', line 99

def including(*elements)
  to_a.including(*elements)
end

#interpose(sep, &block) ⇒ Object

rubocop:disable Metrics/MethodLength



106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
 
# File 'lib/lite/ruby/enumerable.rb', line 106

def interpose(sep, &block)
  enum = Enumerator.new do |val|
    items = each

    loop do
      begin
        val << items.next
      rescue StopIteration
        break
      end

      begin
        items.peek
      rescue StopIteration
        break
      else
        val << sep
      end
    end
  end

  block ? enum.each(&block) : enum
end

#many? ⇒ Boolean

rubocop:enable Metrics/MethodLength

Returns:

  • (Boolean) 


131
132
133
134
135
136
137
138
139
140
141
142
 
# File 'lib/lite/ruby/enumerable.rb', line 131

def many?
  found_count = 0

  if block_given?
    any? do |val|
      found_count += 1 if yield(val)
      found_count > 1
    end
  else
    any? { (found_count += 1) > 1 }
  end
end

#modulate(modulo) ⇒ Object 



144
145
146
147
148
149
150
151
152
153
154
 
# File 'lib/lite/ruby/enumerable.rb', line 144

def modulate(modulo)
  if modulo == 1
    to_a
  elsif size % modulo != 0
    raise ArgumentError, "Invalid modulo: #{modulo.inspect}"
  else
    (0...size).each_with_object(Array.new(modulo, [])) do |i, array|
      array[i % modulo] += [self[i]]
    end
  end
end

#occur(amount = nil) ⇒ Object

rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/MethodLength



157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
 
# File 'lib/lite/ruby/enumerable.rb', line 157

def occur(amount = nil)
  result = Hash.new { |hash, key| hash[key] = [] }

  each do |item|
    key = item
    result[key] << item
  end

  if block_given?
    result.select! { |_key, values| yield(values.size) }
  else
    raise ArgumentError, 'Invalid occur amount' unless amount

    if amount.is_a?(Range)
      result.select! { |_key, values| amount.include?(values.size) }
    else
      result.select! { |_key, values| values.size == amount }
    end
  end

  result.values.flatten.uniq
end

#pluck(*keys) ⇒ Object

rubocop:enable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/MethodLength



181
182
183
184
185
186
187
 
# File 'lib/lite/ruby/enumerable.rb', line 181

def pluck(*keys)
  if keys.many?
    map { |element| keys.map { |key| element[key] } }
  else
    map { |element| element[keys.first] }
  end
end

#produce(identity = 0, &block) ⇒ Object 



189
190
191
192
193
194
195
 
# File 'lib/lite/ruby/enumerable.rb', line 189

def produce(identity = 0, &block)
  if block_given?
    map(&block).produce(identity)
  else
    inject { |acc, val| acc * val } || identity
  end
end

#quotient(identity = 0, &block) ⇒ Object 



197
198
199
200
201
202
203
 
# File 'lib/lite/ruby/enumerable.rb', line 197

def quotient(identity = 0, &block)
  if block_given?
    map(&block).quotient(identity)
  else
    inject { |acc, val| acc / val } || identity
  end
end

#several? ⇒ Boolean

Returns:

  • (Boolean) 


205
206
207
208
209
210
211
212
213
214
215
 
# File 'lib/lite/ruby/enumerable.rb', line 205

def several?
  found_count = 0

  if block_given?
    each { |*opt| found_count += 1 if yield(*opt) }
  else
    each { |opt| found_count += 1 if opt }
  end

  found_count > 1
end

#squeeze(*limited_to) ⇒ Object

rubocop:disable Metrics/MethodLength



218
219
220
221
222
223
224
225
226
227
228
229
230
231
 
# File 'lib/lite/ruby/enumerable.rb', line 218

def squeeze(*limited_to)
  first = true
  current = nil

  each_with_object([]) do |val, array|
    if !limited_to.empty? && !limited_to.include?(val)
      array << val
    elsif first || current != val
      array << val
      first = false
      current = val
    end
  end
end

#take_last(num) ⇒ Object

rubocop:enable Metrics/MethodLength



234
235
236
237
238
239
 
# File 'lib/lite/ruby/enumerable.rb', line 234

def take_last(num)
  collection_size = to_a.size
  return self if num > collection_size

  self[(collection_size - num)..-1]
end

#take_last_if ⇒ Object 



241
242
243
244
245
246
247
 
# File 'lib/lite/ruby/enumerable.rb', line 241

def take_last_if
  reverse_each.with_object([]) do |val, arr|
    break arr unless yield(val)

    arr.unshift(val)
  end
end