Kaydet (Commit) c2dc2692 authored tarafından Andrew M. Kuchling's avatar Andrew M. Kuchling

#687648 from Robert Schuppenies: use classic division. From me: don't use…

#687648 from Robert Schuppenies: use classic division.  From me: don't use string exception; flush stdout after printing
üst 949f71b4
# Generator implementation using threads # Generator implementation using threads
import sys
import thread import thread
Killed = 'Generator.Killed' class Killed(Exception):
pass
class Generator: class Generator:
# Constructor # Constructor
...@@ -16,6 +18,7 @@ class Generator: ...@@ -16,6 +18,7 @@ class Generator:
self.done = 0 self.done = 0
self.killed = 0 self.killed = 0
thread.start_new_thread(self._start, ()) thread.start_new_thread(self._start, ())
# Internal routine # Internal routine
def _start(self): def _start(self):
try: try:
...@@ -29,6 +32,7 @@ class Generator: ...@@ -29,6 +32,7 @@ class Generator:
if not self.killed: if not self.killed:
self.done = 1 self.done = 1
self.getlock.release() self.getlock.release()
# Called by producer for each value; raise Killed if no more needed # Called by producer for each value; raise Killed if no more needed
def put(self, value): def put(self, value):
if self.killed: if self.killed:
...@@ -38,6 +42,7 @@ class Generator: ...@@ -38,6 +42,7 @@ class Generator:
self.putlock.acquire() # Wait for next get() call self.putlock.acquire() # Wait for next get() call
if self.killed: if self.killed:
raise Killed raise Killed
# Called by producer to get next value; raise EOFError if no more # Called by producer to get next value; raise EOFError if no more
def get(self): def get(self):
if self.killed: if self.killed:
...@@ -47,12 +52,14 @@ class Generator: ...@@ -47,12 +52,14 @@ class Generator:
if self.done: if self.done:
raise EOFError # Say there are no more values raise EOFError # Say there are no more values
return self.value return self.value
# Called by consumer if no more values wanted # Called by consumer if no more values wanted
def kill(self): def kill(self):
if self.killed: if self.killed:
raise TypeError, 'kill() called on killed generator' raise TypeError, 'kill() called on killed generator'
self.killed = 1 self.killed = 1
self.putlock.release() self.putlock.release()
# Clone constructor # Clone constructor
def clone(self): def clone(self):
return Generator(self.func, self.args) return Generator(self.func, self.args)
...@@ -64,11 +71,11 @@ def pi(g): ...@@ -64,11 +71,11 @@ def pi(g):
p, q, k = k*k, 2L*k+1L, k+1L p, q, k = k*k, 2L*k+1L, k+1L
a, b, a1, b1 = a1, b1, p*a+q*a1, p*b+q*b1 a, b, a1, b1 = a1, b1, p*a+q*a1, p*b+q*b1
# Print common digits # Print common digits
d, d1 = a/b, a1/b1 d, d1 = a//b, a1//b1
while d == d1: while d == d1:
g.put(int(d)) g.put(int(d))
a, a1 = 10L*(a%b), 10L*(a1%b1) a, a1 = 10L*(a%b), 10L*(a1%b1)
d, d1 = a/b, a1/b1 d, d1 = a//b, a1//b1
def test(): def test():
g = Generator(pi, ()) g = Generator(pi, ())
...@@ -80,5 +87,6 @@ def test(): ...@@ -80,5 +87,6 @@ def test():
g.kill() g.kill()
while 1: while 1:
print h.get(), print h.get(),
sys.stdout.flush()
test() test()
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment