aboutsummaryrefslogtreecommitdiffstats
path: root/activesupport/lib/active_support/core_ext/array/grouping.rb
blob: 6f28e1eb6d946c8bcccc196d88aaacaef82fad33 (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
45
46
module ActiveSupport #:nodoc:
  module CoreExtensions #:nodoc:
    module Array #:nodoc:
      module Grouping
        # 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
    end
  end
end