Permalink
Browse files

examples/dataflow: structuring test

  • Loading branch information...
1 parent 7d0e179 commit 6fac3f027fc9c5bd779ee47e566b8065f8659356 @sbourdeauducq sbourdeauducq committed Jun 20, 2012
Showing with 46 additions and 10 deletions.
  1. +38 −0 examples/dataflow/structuring.py
  2. +8 −10 migen/actorlib/structuring.py
@@ -0,0 +1,38 @@
+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 @@ def get_process_fragment(self):
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):
self.n = n
- layout_from = [("chunk{0}".format(i), layout_to)
- for i in range(self.n)]
super().__init__(
- ("sink", Sink, layout_from),
+ ("sink", Sink, pack_layout(layout_to, n)),
("source", Source, layout_to))
def get_fragment(self):
@@ -43,21 +44,18 @@ def get_fragment(self):
)
)
]
- 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()))
for i in range(self.n)]
- cases[-1][0] = Default()
comb.append(Case(mux, *cases))
return Fragment(comb, sync)
class Pack(Actor):
def __init__(self, layout_from, n):
self.n = n
- layout_to = [("chunk{0}".format(i), layout_from)
- for i in range(self.n)]
super().__init__(
("sink", Sink, layout_from),
- ("source", Source, layout_to))
+ ("source", Source, pack_layout(layout_from, n)))
def get_fragment(self):
demuxbits = bits_for(self.n-1)
@@ -66,7 +64,7 @@ def get_fragment(self):
load_part = Signal()
strobe_all = Signal()
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)]
comb = [
self.endpoints["sink"].ack.eq(~strobe_all | self.endpoints["source"].ack),

0 comments on commit 6fac3f0

Please sign in to comment.