Class: Tins::Generator

Inherits:
Object show all
Includes:
Enumerable
Defined in:
lib/tins/generator.rb

Overview

This class can create generator objects, that can produce all tuples, that would be created by as many for-loops as dimensions were given.

The generator

g = Tins::Generator[1..2, %w[a b c]]

produces

g.to_a # => [[1, "a"], [1, "b"], [1, "c"], [2, "a"], [2, "b"], [2, "c"]]

The ‘each’ method can be used to iterate over the tuples

g.each { |a, b| puts "#{a} #{b}" }

and Tins::Generator includes the Enumerable module, so Enumerable.instance_methods can be used as well:

g.select { |a, b| %w[a c].include? b  } # => [[1, "a"], [1, "c"], [2, "a"], [2, "c"]]

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(enums) ⇒ Generator

Create a new Generator instance. Use the objects in the Array enums as dimensions. The should all respond to the :each method (see module Enumerable in the core ruby library).



27
28
29
30
 
# File 'lib/tins/generator.rb', line 27

def initialize(enums)
  @enums, @iterators, @n = [], [], 0
  enums.each { |e| add_dimension(e) }
end

Class Method Details

.[](*enums) ⇒ Object

Create a new Generator object from the enumberables enums.



20
21
22
 
# File 'lib/tins/generator.rb', line 20

def self.[](*enums)
  new(enums)
end

Instance Method Details

#add_dimension(enum, iterator = :each) ⇒ Object

Add another dimension to this generator. enum is an object, that ought to respond to the iterator method (defaults to :each).



55
56
57
58
59
 
# File 'lib/tins/generator.rb', line 55

def add_dimension(enum, iterator = :each)
  @enums << enum
  @iterators << iterator
  @n += 1
end

#each(&block) ⇒ Object

Iterate over all tuples produced by this generator and yield to them.



33
34
35
36
 
# File 'lib/tins/generator.rb', line 33

def each(&block) # :yield: tuple
  recurse(&block)
  self
end

#size ⇒ Object

Return the size of this generator, that is the number of its dimensions.



62
63
64
 
# File 'lib/tins/generator.rb', line 62

def size
  @enums.size
end