examples/dataflow/control.py

   1 from migen.flow.network import *
   2 from migen.actorlib import control
   3 from migen.actorlib.sim import *
   4 from migen.sim.generic import Simulator
   5 from migen.sim.icarus import Runner
   6
   7 def source_gen():
   8         for i in range(10):
   9                 v = i + 5
  10                 print("==> " + str(v))
  11                 yield Token("source", {"value": v})
  12
  13 def sink_gen():
  14         while True:
  15                 t = Token("sink")
  16                 yield t
  17                 print(t.value["value"])
  18
  19 def main():
  20         source = ActorNode(SimActor(source_gen(), ("source", Source, [("value", BV(32))])))
  21         loop = ActorNode(control.For(32))
  22         sink = ActorNode(SimActor(sink_gen(), ("sink", Sink, [("value", BV(32))])))
  23         g = DataFlowGraph()
  24         g.add_connection(source, loop)
  25         g.add_connection(loop, sink)
  26         comp = CompositeActor(g)
  27         fragment = comp.get_fragment()
  28         sim = Simulator(fragment, Runner())
  29         sim.run(500)
  30
  31 main()