Module: Enumerable

Defined in:

lib/lazy_enumerator.rb

Instance Method Summary

• #chunk(init_val = (init_given = true; nil)) ⇒ Object
• #lazy ⇒ Object

Instance Method Details

#chunk(init_val = (init_given = true; nil)) ⇒ Object

# File 'lib/lazy_enumerator.rb', line 252

def chunk(init_val = (init_given = true; nil))
  cls = (self.kind_of? Enumerator) ? self.class : Enumerator
  cls.new do |yielder|
    
    chunk = nil
    each do |elt|
      new_val = init_given ? yield(elt, init_val) : yield(elt)
      
      if chunk.nil?
        chunk = [new_val, [elt]]
      elsif new_val == init_val
        chunk[1] << elt
      else
        yielder.yield chunk
        chunk = [new_val, [elt]]
      end
      
      init_val = new_val
    end
    
    yielder.yield chunk if chunk
  end
end

#lazy ⇒ Object

# File 'lib/lazy_enumerator.rb', line 246

def lazy
  Enumerator::Lazy.new(self) do |output, *values|
    output.yield(*values)
  end
end