* lib/generator.rb: removed obsolete library. [ruby-core:16233]
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@15947 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
parent
eb18a827fb
commit
db4a767d25
@ -1,3 +1,7 @@
|
||||
Thu Apr 10 15:03:47 2008 Yukihiro Matsumoto <matz@ruby-lang.org>
|
||||
|
||||
* lib/generator.rb: removed obsolete library. [ruby-core:16233]
|
||||
|
||||
Thu Apr 10 10:27:24 2008 Nobuyoshi Nakada <nobu@ruby-lang.org>
|
||||
|
||||
* thread_pthread.c (native_sleep): sleep_cond is initialized at
|
||||
|
385
lib/generator.rb
385
lib/generator.rb
@ -1,385 +0,0 @@
|
||||
#!/usr/bin/env ruby
|
||||
#--
|
||||
# $Idaemons: /home/cvs/rb/generator.rb,v 1.8 2001/10/03 08:54:32 knu Exp $
|
||||
# $RoughId: generator.rb,v 1.10 2003/10/14 19:36:58 knu Exp $
|
||||
# $Id$
|
||||
#++
|
||||
#
|
||||
# = generator.rb: convert an internal iterator to an external one
|
||||
#
|
||||
# Copyright (c) 2001,2003 Akinori MUSHA <knu@iDaemons.org>
|
||||
#
|
||||
# All rights reserved. You can redistribute and/or modify it under
|
||||
# the same terms as Ruby.
|
||||
#
|
||||
# == Overview
|
||||
#
|
||||
# This library provides the Generator class, which converts an
|
||||
# internal iterator (i.e. an Enumerable object) to an external
|
||||
# iterator. In that form, you can roll many iterators independently.
|
||||
#
|
||||
# The SyncEnumerator class, which is implemented using Generator,
|
||||
# makes it easy to roll many Enumerable objects synchronously.
|
||||
#
|
||||
# See the respective classes for examples of usage.
|
||||
|
||||
|
||||
#
|
||||
# Generator converts an internal iterator (i.e. an Enumerable object)
|
||||
# to an external iterator.
|
||||
#
|
||||
# == Example
|
||||
#
|
||||
# require 'generator'
|
||||
#
|
||||
# # Generator from an Enumerable object
|
||||
# g = Generator.new(['A', 'B', 'C', 'Z'])
|
||||
#
|
||||
# while g.next?
|
||||
# puts g.next
|
||||
# end
|
||||
#
|
||||
# # Generator from a block
|
||||
# g = Generator.new { |g|
|
||||
# for i in 'A'..'C'
|
||||
# g.yield i
|
||||
# end
|
||||
#
|
||||
# g.yield 'Z'
|
||||
# }
|
||||
#
|
||||
# # The same result as above
|
||||
# while g.next?
|
||||
# puts g.next
|
||||
# end
|
||||
#
|
||||
class Generator
|
||||
include Enumerable
|
||||
|
||||
# Creates a new generator either from an Enumerable object or from a
|
||||
# block.
|
||||
#
|
||||
# In the former, block is ignored even if given.
|
||||
#
|
||||
# In the latter, the given block is called with the generator
|
||||
# itself, and expected to call the +yield+ method for each element.
|
||||
def initialize(enum = nil, &block)
|
||||
if enum
|
||||
@block = proc{|g| enum.each{|value| g.yield value}}
|
||||
else
|
||||
@block = block
|
||||
end
|
||||
@index = 0
|
||||
@queue = []
|
||||
@main_thread = nil
|
||||
@loop_thread.kill if defined?(@loop_thread)
|
||||
@loop_thread = Thread.new do
|
||||
Thread.stop
|
||||
begin
|
||||
@block.call(self)
|
||||
rescue
|
||||
@main_thread.raise $!
|
||||
ensure
|
||||
@main_thread.wakeup
|
||||
end
|
||||
end
|
||||
Thread.pass until @loop_thread.stop?
|
||||
self
|
||||
end
|
||||
|
||||
# Yields an element to the generator.
|
||||
def yield(value)
|
||||
if Thread.current != @loop_thread
|
||||
raise "should be called in Generator.new{|g| ... }"
|
||||
end
|
||||
Thread.critical = true
|
||||
begin
|
||||
@queue << value
|
||||
@main_thread.wakeup
|
||||
Thread.stop
|
||||
ensure
|
||||
Thread.critical = false
|
||||
end
|
||||
self
|
||||
end
|
||||
|
||||
# Returns true if the generator has reached the end.
|
||||
def end?
|
||||
if @queue.empty?
|
||||
if @main_thread
|
||||
raise "should not be called in Generator.new{|g| ... }"
|
||||
end
|
||||
Thread.critical = true
|
||||
begin
|
||||
@main_thread = Thread.current
|
||||
@loop_thread.wakeup
|
||||
Thread.stop
|
||||
rescue ThreadError
|
||||
# ignore
|
||||
ensure
|
||||
@main_thread = nil
|
||||
Thread.critical = false
|
||||
end
|
||||
end
|
||||
@queue.empty?
|
||||
end
|
||||
|
||||
# Returns true if the generator has not reached the end yet.
|
||||
def next?
|
||||
!end?
|
||||
end
|
||||
|
||||
# Returns the current index (position) counting from zero.
|
||||
def index
|
||||
@index
|
||||
end
|
||||
|
||||
# Returns the current index (position) counting from zero.
|
||||
def pos
|
||||
@index
|
||||
end
|
||||
|
||||
# Returns the element at the current position and moves forward.
|
||||
def next
|
||||
raise EOFError.new("no more elements available") if end?
|
||||
@index += 1
|
||||
@queue.shift
|
||||
end
|
||||
|
||||
# Returns the element at the current position.
|
||||
def current
|
||||
raise EOFError.new("no more elements available") if end?
|
||||
@queue.first
|
||||
end
|
||||
|
||||
# Rewinds the generator.
|
||||
def rewind
|
||||
initialize(nil, &@block) if @index.nonzero?
|
||||
self
|
||||
end
|
||||
|
||||
# Rewinds the generator and enumerates the elements.
|
||||
def each
|
||||
rewind
|
||||
until end?
|
||||
yield self.next
|
||||
end
|
||||
self
|
||||
end
|
||||
end
|
||||
|
||||
#
|
||||
# SyncEnumerator creates an Enumerable object from multiple Enumerable
|
||||
# objects and enumerates them synchronously.
|
||||
#
|
||||
# == Example
|
||||
#
|
||||
# require 'generator'
|
||||
#
|
||||
# s = SyncEnumerator.new([1,2,3], ['a', 'b', 'c'])
|
||||
#
|
||||
# # Yields [1, 'a'], [2, 'b'], and [3,'c']
|
||||
# s.each { |row| puts row.join(', ') }
|
||||
#
|
||||
class SyncEnumerator
|
||||
include Enumerable
|
||||
|
||||
# Creates a new SyncEnumerator which enumerates rows of given
|
||||
# Enumerable objects.
|
||||
def initialize(*enums)
|
||||
@gens = enums.map { |e| Generator.new(e) }
|
||||
end
|
||||
|
||||
# Returns the number of enumerated Enumerable objects, i.e. the size
|
||||
# of each row.
|
||||
def size
|
||||
@gens.size
|
||||
end
|
||||
|
||||
# Returns the number of enumerated Enumerable objects, i.e. the size
|
||||
# of each row.
|
||||
def length
|
||||
@gens.length
|
||||
end
|
||||
|
||||
# Returns true if the given nth Enumerable object has reached the
|
||||
# end. If no argument is given, returns true if any of the
|
||||
# Enumerable objects has reached the end.
|
||||
def end?(i = nil)
|
||||
if i.nil?
|
||||
@gens.detect { |g| g.end? } ? true : false
|
||||
else
|
||||
@gens[i].end?
|
||||
end
|
||||
end
|
||||
|
||||
# Enumerates rows of the Enumerable objects.
|
||||
def each
|
||||
@gens.each { |g| g.rewind }
|
||||
|
||||
loop do
|
||||
count = 0
|
||||
|
||||
ret = @gens.map { |g|
|
||||
if g.end?
|
||||
count += 1
|
||||
nil
|
||||
else
|
||||
g.next
|
||||
end
|
||||
}
|
||||
|
||||
if count == @gens.size
|
||||
break
|
||||
end
|
||||
|
||||
yield ret
|
||||
end
|
||||
|
||||
self
|
||||
end
|
||||
end
|
||||
|
||||
if $0 == __FILE__
|
||||
eval DATA.read, nil, $0, __LINE__+4
|
||||
end
|
||||
|
||||
__END__
|
||||
|
||||
__END__
|
||||
|
||||
require 'test/unit'
|
||||
|
||||
class TC_Generator < Test::Unit::TestCase
|
||||
def test_block1
|
||||
g = Generator.new { |g|
|
||||
# no yield's
|
||||
}
|
||||
|
||||
assert_equal(0, g.pos)
|
||||
assert_raises(EOFError) { g.current }
|
||||
end
|
||||
|
||||
def test_block2
|
||||
g = Generator.new { |g|
|
||||
for i in 'A'..'C'
|
||||
g.yield i
|
||||
end
|
||||
|
||||
g.yield 'Z'
|
||||
}
|
||||
|
||||
assert_equal(0, g.pos)
|
||||
assert_equal('A', g.current)
|
||||
|
||||
assert_equal(true, g.next?)
|
||||
assert_equal(0, g.pos)
|
||||
assert_equal('A', g.current)
|
||||
assert_equal(0, g.pos)
|
||||
assert_equal('A', g.next)
|
||||
|
||||
assert_equal(1, g.pos)
|
||||
assert_equal(true, g.next?)
|
||||
assert_equal(1, g.pos)
|
||||
assert_equal('B', g.current)
|
||||
assert_equal(1, g.pos)
|
||||
assert_equal('B', g.next)
|
||||
|
||||
assert_equal(g, g.rewind)
|
||||
|
||||
assert_equal(0, g.pos)
|
||||
assert_equal('A', g.current)
|
||||
|
||||
assert_equal(true, g.next?)
|
||||
assert_equal(0, g.pos)
|
||||
assert_equal('A', g.current)
|
||||
assert_equal(0, g.pos)
|
||||
assert_equal('A', g.next)
|
||||
|
||||
assert_equal(1, g.pos)
|
||||
assert_equal(true, g.next?)
|
||||
assert_equal(1, g.pos)
|
||||
assert_equal('B', g.current)
|
||||
assert_equal(1, g.pos)
|
||||
assert_equal('B', g.next)
|
||||
|
||||
assert_equal(2, g.pos)
|
||||
assert_equal(true, g.next?)
|
||||
assert_equal(2, g.pos)
|
||||
assert_equal('C', g.current)
|
||||
assert_equal(2, g.pos)
|
||||
assert_equal('C', g.next)
|
||||
|
||||
assert_equal(3, g.pos)
|
||||
assert_equal(true, g.next?)
|
||||
assert_equal(3, g.pos)
|
||||
assert_equal('Z', g.current)
|
||||
assert_equal(3, g.pos)
|
||||
assert_equal('Z', g.next)
|
||||
|
||||
assert_equal(4, g.pos)
|
||||
assert_equal(false, g.next?)
|
||||
assert_raises(EOFError) { g.next }
|
||||
end
|
||||
|
||||
def test_each
|
||||
a = [5, 6, 7, 8, 9]
|
||||
|
||||
g = Generator.new(a)
|
||||
|
||||
i = 0
|
||||
|
||||
g.each { |x|
|
||||
assert_equal(a[i], x)
|
||||
|
||||
i += 1
|
||||
|
||||
break if i == 3
|
||||
}
|
||||
|
||||
assert_equal(3, i)
|
||||
|
||||
i = 0
|
||||
|
||||
g.each { |x|
|
||||
assert_equal(a[i], x)
|
||||
|
||||
i += 1
|
||||
}
|
||||
|
||||
assert_equal(5, i)
|
||||
end
|
||||
end
|
||||
|
||||
class TC_SyncEnumerator < Test::Unit::TestCase
|
||||
def test_each
|
||||
r = ['a'..'f', 1..10, 10..20]
|
||||
ra = r.map { |x| x.to_a }
|
||||
|
||||
a = (0...(ra.map {|x| x.size}.max)).map { |i| ra.map { |x| x[i] } }
|
||||
|
||||
s = SyncEnumerator.new(*r)
|
||||
|
||||
i = 0
|
||||
|
||||
s.each { |x|
|
||||
assert_equal(a[i], x)
|
||||
|
||||
i += 1
|
||||
|
||||
break if i == 3
|
||||
}
|
||||
|
||||
assert_equal(3, i)
|
||||
|
||||
i = 0
|
||||
|
||||
s.each { |x|
|
||||
assert_equal(a[i], x)
|
||||
|
||||
i += 1
|
||||
}
|
||||
|
||||
assert_equal(a.size, i)
|
||||
end
|
||||
end
|
Loading…
x
Reference in New Issue
Block a user