From: Sebastien Bourdeauducq <sebastien@milkymist.org>
Date: Wed, 3 Jul 2013 14:35:24 +0000 (+0200)
Subject: examples/pytholite/basic: demonstrate generator arguments
X-Git-Tag: 24jan2021_ls180~2099^2~537
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=4096a785f9a839316dfeda3e66b9a507d01964e9;p=litex.git

examples/pytholite/basic: demonstrate generator arguments
---

diff --git a/examples/pytholite/basic.py b/examples/pytholite/basic.py
index 857b22f2..425cfe7f 100644
--- a/examples/pytholite/basic.py
+++ b/examples/pytholite/basic.py
@@ -7,14 +7,14 @@ from migen.fhdl import verilog
 
 layout = [("r", 32)]
 
-def number_gen():
-	for i in range(10):
+def number_gen(n):
+	for i in range(n):
 		yield Token("result", {"r": i})
 
 class SimNumberGen(SimActor):
 	def __init__(self):
 		self.result = Source(layout)
-		SimActor.__init__(self, number_gen())
+		SimActor.__init__(self, number_gen(5))
 
 def run_sim(ng):
 	g = DataFlowGraph()
@@ -23,11 +23,11 @@ def run_sim(ng):
 	
 	c = CompositeActor(g)
 	sim = Simulator(c)
-	sim.run(30)
+	sim.run(20)
 	del sim
 
 def make_ng_pytholite():
-	ng_pytholite = Pytholite(number_gen)
+	ng_pytholite = Pytholite(number_gen, 5)
 	ng_pytholite.result = Source(layout)
 	ng_pytholite.finalize()
 	return ng_pytholite