Commit 6fac3f02 authored by Sebastien Bourdeauducq's avatar Sebastien Bourdeauducq
Browse files

examples/dataflow: structuring test

parent 7d0e179a
from migen.flow.network import *
from migen.actorlib import structuring
from migen.actorlib.sim import *
from migen.sim.generic import Simulator
from migen.sim.icarus import Runner
pack_factor = 5
def source_gen():
for i in range(80):
#print("==> " + str(i))
yield Token("source", {"value": i})
def sink_gen():
while True:
t = Token("sink")
yield t
print(t.value["value"])
def main():
base_layout = [("value", BV(32))]
source = ActorNode(SimActor(source_gen(), ("source", Source, base_layout)))
packer = ActorNode(structuring.Pack(base_layout, pack_factor))
unpacker = ActorNode(structuring.Unpack(pack_factor, base_layout))
sink = ActorNode(SimActor(sink_gen(), ("sink", Sink, base_layout)))
g = DataFlowGraph()
g.add_connection(source, packer)
g.add_connection(packer, unpacker)
g.add_connection(unpacker, sink)
comp = CompositeActor(g)
fragment = comp.get_fragment()
sim = Simulator(fragment, Runner())
sim.run(100)
main()
...@@ -16,13 +16,14 @@ class Cast(CombinatorialActor): ...@@ -16,13 +16,14 @@ class Cast(CombinatorialActor):
Cat(*sigs_to).eq(Cat(*sigs_from)) Cat(*sigs_to).eq(Cat(*sigs_from))
]) ])
class Chop(Actor): def pack_layout(l, n):
return [("chunk{0}".format(i), l) for i in range(n)]
class Unpack(Actor):
def __init__(self, n, layout_to): def __init__(self, n, layout_to):
self.n = n self.n = n
layout_from = [("chunk{0}".format(i), layout_to)
for i in range(self.n)]
super().__init__( super().__init__(
("sink", Sink, layout_from), ("sink", Sink, pack_layout(layout_to, n)),
("source", Source, layout_to)) ("source", Source, layout_to))
def get_fragment(self): def get_fragment(self):
...@@ -43,21 +44,18 @@ class Chop(Actor): ...@@ -43,21 +44,18 @@ class Chop(Actor):
) )
) )
] ]
cases = [(Constant(i, BV(muxbits)), cases = [(Constant(i, BV(muxbits)) if i else Default(),
Cat(*self.token("source").flatten()).eq(*self.token("sink").subrecord("chunk{0}".format(i)).flatten())) Cat(*self.token("source").flatten()).eq(*self.token("sink").subrecord("chunk{0}".format(i)).flatten()))
for i in range(self.n)] for i in range(self.n)]
cases[-1][0] = Default()
comb.append(Case(mux, *cases)) comb.append(Case(mux, *cases))
return Fragment(comb, sync) return Fragment(comb, sync)
class Pack(Actor): class Pack(Actor):
def __init__(self, layout_from, n): def __init__(self, layout_from, n):
self.n = n self.n = n
layout_to = [("chunk{0}".format(i), layout_from)
for i in range(self.n)]
super().__init__( super().__init__(
("sink", Sink, layout_from), ("sink", Sink, layout_from),
("source", Source, layout_to)) ("source", Source, pack_layout(layout_from, n)))
def get_fragment(self): def get_fragment(self):
demuxbits = bits_for(self.n-1) demuxbits = bits_for(self.n-1)
...@@ -66,7 +64,7 @@ class Pack(Actor): ...@@ -66,7 +64,7 @@ class Pack(Actor):
load_part = Signal() load_part = Signal()
strobe_all = Signal() strobe_all = Signal()
cases = [(Constant(i, BV(demuxbits)), cases = [(Constant(i, BV(demuxbits)),
Cat(*self.token("source").subrecord("chunk{0}".format(i)).flatten()).eq(*self.token("sink").flatten()) Cat(*self.token("source").subrecord("chunk{0}".format(i)).flatten()).eq(*self.token("sink").flatten()))
for i in range(self.n)] for i in range(self.n)]
comb = [ comb = [
self.endpoints["sink"].ack.eq(~strobe_all | self.endpoints["source"].ack), self.endpoints["sink"].ack.eq(~strobe_all | self.endpoints["source"].ack),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment