Class: Array

Inherits:
Object show all
Defined in:
lib/extensions.rb

Instance Method Summary collapse

Instance Method Details

#in_groups_of(number, fill_with = nil) ⇒ Object 



22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
# File 'lib/extensions.rb', line 22

def in_groups_of(number, fill_with = nil)
  if fill_with == false
    collection = self
  else
    # size % number gives how many extra we have;
    # subtracting from number gives how many to add;
    # modulo number ensures we don't add group of just fill.
    padding = (number - size % number) % number
    collection = dup.concat([fill_with] * padding)
  end

  if block_given?
    collection.each_slice(number) { |slice| yield(slice) }
  else
    returning [] do |groups|
      collection.each_slice(number) { |group| groups << group }
    end
  end
end