aboutsummaryrefslogtreecommitdiffstats
path: root/activesupport/lib/active_support/core_ext/array.rb
blob: 243aeab42311686d570d92dbaec3d40d0e736d51 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
require File.dirname(__FILE__) + '/array/conversions'

class Array #:nodoc:
  include ActiveSupport::CoreExtensions::Array::Conversions

  # Iterate over an array in groups of a certain size, padding any remaining 
  # slots with specified value (<tt>nil</tt> by default).
  # 
  # E.g.
  # 
  #   %w(1 2 3 4 5 6 7).in_groups_of(3) {|g| p g}
  #   ["1", "2", "3"]
  #   ["4", "5", "6"]
  #   ["7", nil, nil]
  def in_groups_of(number, fill_with = nil, &block)
    require 'enumerator'
    collection = dup
    collection << fill_with until collection.size.modulo(number).zero?
    grouped_collection = [] unless block_given?
    collection.each_slice(number) do |group|
      block_given? ? yield(group) : grouped_collection << group
    end
    grouped_collection unless block_given?
  end
  
  # Divide the array into one or more subarrays based on a delimiting +value+
  # or the result of an optional block.
  #
  # ex.
  #
  #   [1, 2, 3, 4, 5].split(3)                # => [[1, 2], [4, 5]]
  #   (1..10).to_a.split { |i| i % 3 == 0 }   # => [[1, 2], [4, 5], [7, 8], [10]]
  def split(value = nil, &block)
    block ||= Proc.new { |e| e == value }
    inject([[]]) do |results, element|
      if block.call(element)
        results << []
      else
        results.last << element
      end
      results
    end
  end
end