Enumerator::Lazy

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 30
def initialize(obj, &block)
  super(){|yielder|
    begin
      obj.each{|x|
        if block
          block.call(yielder, x)
        else
          yielder << x
        end
      }
    rescue StopIteration
    end
  }
end
            

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 89
def drop(n)
  dropped = 0
  Lazy.new(self){|yielder, val|
    if dropped < n
      dropped += 1
    else
      yielder << val
    end
  }
end
            

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 100
def drop_while(&block)
  dropping = true
  Lazy.new(self){|yielder, val|
    if dropping
      if not block.call(val)
        yielder << val
        dropping = false
      end
    else
      yielder << val
    end
  }
end
            

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 138
def flat_map(&block)
  Lazy.new(self){|yielder, val|
    ary = block.call(val)
    # TODO: check ary is an Array
    ary.each{|x|
      yielder << x
    }
  }
end
            

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 81
def grep(pattern)
  Lazy.new(self){|yielder, val|
    if pattern === val
      yielder << val
    end
  }
end
            

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 57
def map(&block)
  Lazy.new(self){|yielder, val|
    yielder << block.call(val)
  }
end
            

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 73
def reject(&block)
  Lazy.new(self){|yielder, val|
    unless block.call(val)
      yielder << val
    end
  }
end
            

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 64
def select(&block)
  Lazy.new(self){|yielder, val|
    if block.call(val)
      yielder << val
    end
  }
end
            

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 114
def take(n)
  if n == 0
    return Lazy.new(self){raise StopIteration}
  end
  taken = 0
  Lazy.new(self){|yielder, val|
    yielder << val
    taken += 1
    if taken >= n
      raise StopIteration
    end
  }
end
            

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 128
def take_while(&block)
  Lazy.new(self){|yielder, val|
    if block.call(val)
      yielder << val
    else
      raise StopIteration
    end
  }
end
            

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 45
def to_enum(meth=:each, *args, &block)
  unless self.respond_to?(meth)
    raise ArgumentError, "undefined method #{meth}"
  end
  lz = Lazy.new(self, &block)
  lz.obj = self
  lz.meth = meth
  lz.args = args
  lz
end
            

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 161
def uniq(&block)
  hash = {}
  Lazy.new(self){|yielder, val|
    if block
      v = block.call(val)
    else
      v = val
    end
    unless hash.include?(v)
      yielder << val
      hash[v] = val
    end
  }
end
            

 
               # File mrbgems/mruby-enum-lazy/mrblib/lazy.rb, line 149
def zip(*args, &block)
  enums = [self] + args
  Lazy.new(self){|yielder, val|
    ary = enums.map{|e| e.next}
    if block
      yielder << block.call(ary)
    else
      yielder << ary
    end
  }
end