litex/examples/sim/dataflow.py

32 lines
828 B
Python
Raw Normal View History

from migen.fhdl.structure import *
from migen.flow.actor import *
2012-12-14 09:55:38 -05:00
from migen.flow.transactions import *
from migen.flow.network import *
from migen.actorlib.sim import *
from migen.sim.generic import Simulator
def source_gen():
for i in range(10):
print("Sending: " + str(i))
yield Token("source", {"value": i})
def sink_gen():
while True:
t = Token("sink")
yield t
print("Received: " + str(t.value["value"]))
def main():
2012-12-12 16:20:48 -05:00
source = SimActor(source_gen(), ("source", Source, [("value", 32)]))
sink = SimActor(sink_gen(), ("sink", Sink, [("value", 32)]))
2012-06-08 16:49:49 -04:00
g = DataFlowGraph()
g.add_connection(source, sink)
comp = CompositeActor(g)
def end_simulation(s):
2012-12-12 16:20:48 -05:00
s.interrupt = source.token_exchanger.done
fragment = comp.get_fragment() + Fragment(sim=[end_simulation])
2013-02-09 11:04:53 -05:00
sim = Simulator(fragment)
sim.run()
main()