74 lines
1.9 KiB
Python
74 lines
1.9 KiB
Python
|
from migen.flow.ala import *
|
||
|
|
from migen.flow.network import *
|
||
|
|
from migen.flow import plumbing
|
||
|
|
from migen.actorlib.sim import *
|
||
|
|
from migen.sim.generic import Simulator
|
||
|
|
from migen.sim.icarus import Runner
|
||
|
|
|
||
|
|
# Pushes a "1" token and then becomes transparent.
|
||
|
|
class Init(Actor):
|
||
|
|
def __init__(self, nbits):
|
||
|
|
super().__init__(
|
||
|
|
("in", Sink, [("d", BV(nbits))]),
|
||
|
|
("out", Source, [("d", BV(nbits))]))
|
||
|
|
|
||
|
|
def get_fragment(self):
|
||
|
|
done = Signal()
|
||
|
|
comb = [
|
||
|
|
self.busy.eq(~done),
|
||
|
|
If(done,
|
||
|
|
self.endpoints["in"].ack.eq(self.endpoints["out"].ack),
|
||
|
|
self.endpoints["out"].stb.eq(self.endpoints["in"].stb),
|
||
|
|
self.endpoints["out"].token.d.eq(self.endpoints["in"].token.d)
|
||
|
|
).Else(
|
||
|
|
self.endpoints["in"].ack.eq(0),
|
||
|
|
self.endpoints["out"].stb.eq(1),
|
||
|
|
self.endpoints["out"].token.d.eq(1)
|
||
|
|
)
|
||
|
|
]
|
||
|
|
sync = [
|
||
|
|
If(self.endpoints["out"].ack, done.eq(1))
|
||
|
|
]
|
||
|
|
return Fragment(comb, sync)
|
||
|
|
|
||
|
|
class Dumper(SimActor):
|
||
|
|
def __init__(self, nbits):
|
||
|
|
def dumper_gen():
|
||
|
|
while True:
|
||
|
|
t = Token("result")
|
||
|
|
yield t
|
||
|
|
print(t.value["r"])
|
||
|
|
super().__init__(dumper_gen(),
|
||
|
|
("result", Sink, [("r", BV(nbits))]))
|
||
|
|
|
||
|
|
def main():
|
||
|
|
nbits = 32
|
||
|
|
|
||
|
|
# See:
|
||
|
|
# http://www.csse.monash.edu.au/~damian/Idioms/Topics/12.1.DataFlow/html/text.html
|
||
|
|
g = DataFlowGraph()
|
||
|
|
|
||
|
|
adder = ActorNode(Add(BV(nbits)))
|
||
|
|
bufadd = ActorNode(plumbing.Buffer) # TODO FIXME: deadlocks without this buffer
|
||
|
|
init1 = ActorNode(Init(nbits))
|
||
|
|
buf1 = ActorNode(plumbing.Buffer)
|
||
|
|
init2 = ActorNode(Init(nbits))
|
||
|
|
buf2 = ActorNode(plumbing.Buffer)
|
||
|
|
|
||
|
|
g.add_connection(adder, bufadd)
|
||
|
|
g.add_connection(bufadd, init1)
|
||
|
|
g.add_connection(init1, buf1)
|
||
|
|
g.add_connection(buf1, adder, sink_subr="a")
|
||
|
|
g.add_connection(buf1, init2)
|
||
|
|
g.add_connection(init2, buf2)
|
||
|
|
g.add_connection(buf2, adder, sink_subr="b")
|
||
|
|
|
||
|
|
g.add_connection(bufadd, ActorNode(Dumper(nbits)))
|
||
|
|
|
||
|
|
c = CompositeActor(g)
|
||
|
|
fragment = c.get_fragment()
|
||
|
|
sim = Simulator(fragment, Runner())
|
||
|
|
sim.run(100)
|
||
|
|
|
||
|
|
main()
|