litex/examples/dataflow/misc.py

32 lines
749 B
Python

from migen.flow.network import *
from migen.actorlib import misc
from migen.actorlib.sim import *
from migen.sim.generic import Simulator
from migen.sim.icarus import Runner
def source_gen():
for i in range(10):
v = i + 5
print("==> " + str(v))
yield Token("source", {"value": v})
def sink_gen():
while True:
t = Token("sink")
yield t
print(t.value["value"])
def main():
source = SimActor(source_gen(), ("source", Source, [("value", 32)]))
loop = misc.IntSequence(32)
sink = SimActor(sink_gen(), ("sink", Sink, [("value", 32)]))
g = DataFlowGraph()
g.add_connection(source, loop)
g.add_connection(loop, sink)
comp = CompositeActor(g)
fragment = comp.get_fragment()
sim = Simulator(fragment, Runner())
sim.run(500)
main()