Contents

module Enumerators
  private
  def reverse_enumerator(e)
    e.reverse_each
  end

  def has_next(e)
    begin
      e.peek
      true
    rescue StopIteration
      false
    end
  end

  def enumerator_of(fn, init)
    Enumerator.new do |y|
      value = init
      y << value
      loop do
        value = fn.(value)
        y << value
      end
    end.lazy
  end

  def repeat_fn_enumerator(fn)
    Enumerator.new do |y|
      loop do
        y << fn.()
      end
    end.lazy
  end

  def repeat_enumerator(value)
    repeat_fn_enumerator(returns(value))
  end

  def character_enumerator(string)
    Enumerator.new do |y|
      index = 0
      loop do
        raise StopIteration.new unless index < string.size
        y << string[index]
        index = index + 1
      end
    end.lazy
  end

  def flatten_enumerator(enumerator)
    enumerator.rewind
    Enumerator.new do |y|
      current_enumerator = empty_enumerator

      get_current_enumerator = ->() {
        until has_next(current_enumerator)
          return empty_enumerator unless has_next(enumerator)
          current_enumerator = enumerator.next.enumerator
          current_enumerator.rewind
        end
        current_enumerator
      }

      loop do
        current = get_current_enumerator.()
        if has_next(current)
          y << current.next
        else
          raise StopIteration.new
        end
      end
    end.lazy
  end

  def cycle_enumerator(e)
    Enumerator.new do |y|
      loop do
        e.rewind unless has_next(e)
        y << e.next
      end
    end.lazy
  end

  def empty_enumerator
    [].lazy
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
totally_lazy-0.1.62	lib/totally_lazy/enumerators.rb
totally_lazy-0.1.61	lib/totally_lazy/enumerators.rb
totally_lazy-0.1.60	lib/totally_lazy/enumerators.rb
totally_lazy-0.1.59	lib/totally_lazy/enumerators.rb