tb/asmicon/bankmachine: selector test bench

diff --git a/tb/asmicon/common.py b/tb/asmicon/common.py
index f5ff896436009e46894166dbc023f7fb71da9f07..01d761b50e4b43699e98269263235a42be9ff8d9 100644 (file)
--- a/tb/asmicon/common.py
+++ b/tb/asmicon/common.py
@@ -39,3 +39,28 @@ class CommandLogger:
        
        def get_fragment(self):
                return Fragment(sim=[self.do_simulation])
+
+class SlotsLogger:
+       def __init__(self, slicer, slots):
+               self.slicer = slicer
+               self.slots = slots
+               
+       def do_simulation(self, sim):
+               state_strs = ["EMPTY", "PEND", "PRCESS"]
+               rw_strs = ["RD", "WR"]
+               print("\t" + "\t".join([str(x) for x in range(len(self.slots))]))
+               print("State:\t" + "\t".join([state_strs[sim.rd(s.state)] for s in self.slots]))
+               print("RW:\t" + "\t".join([rw_strs[sim.rd(s.we)] for s in self.slots]))
+               print("Row:\t" + "\t".join([str(self.slicer.row(sim.rd(s.adr))) for s in self.slots]))
+               print("Bank:\t" + "\t".join([str(self.slicer.bank(sim.rd(s.adr))) for s in self.slots]))
+               print("Col:\t" + "\t".join([str(self.slicer.col(sim.rd(s.adr))) for s in self.slots]))
+               times = []
+               for s in self.slots:
+                       if s.time:
+                               times.append(str(sim.rd(s._counter)) + "/" + str(s.time))
+                       else:
+                               times.append("N/A")
+               print("Time:\t" + "\t".join(times))
+
+       def get_fragment(self):
+               return Fragment(sim=[self.do_simulation])

diff --git a/tb/asmicon/selector.py b/tb/asmicon/selector.py
new file mode 100644 (file)
index 0000000..b817c71
--- /dev/null
+++ b/tb/asmicon/selector.py
@@ -0,0 +1,77 @@
+from migen.fhdl.structure import *
+from migen.bus.asmibus import *
+from migen.sim.generic import Simulator, TopLevel
+from migen.sim.icarus import Runner
+
+from milkymist import asmicon
+from milkymist.asmicon.bankmachine import _AddressSlicer, _Selector, _Buffer
+
+from common import SlotsLogger
+
+sdram_geom = asmicon.GeomSettings(
+       bank_a=2,
+       row_a=13,
+       col_a=10
+)
+
+def my_generator(dt, offset):
+       for t in range(dt):
+               yield None
+       for x in range(10):
+               t = TRead(x + offset)
+               yield t
+
+class Selector:
+       def __init__(self, slicer, bankn, slots):
+               self.selector = _Selector(slicer, bankn, slots)
+               self.queue = []
+       
+       def do_simulation(self, s):
+               if s.rd(self.selector.stb):
+                       tag = s.rd(self.selector.tag)
+                       self.queue.append(tag)
+                       print("==> SELECTED: " + str(tag))
+               print("")
+       
+       def get_fragment(self):
+               comb = [self.selector.ack.eq(1)]
+               return self.selector.get_fragment() + Fragment(comb, sim=[self.do_simulation])
+
+class Completer:
+       def __init__(self, hub, queue):
+               self.hub = hub
+               self.queue = queue
+       
+       def do_simulation(self, s):
+               if self.queue:
+                       tag = self.queue.pop()
+                       s.wr(self.hub.call, 1)
+                       s.wr(self.hub.tag_call, tag)
+               else:
+                       s.wr(self.hub.call, 0)
+               
+       def get_fragment(self):
+               return Fragment(sim=[self.do_simulation])
+
+def main():
+       hub = Hub(12, 128, 8)
+       initiators = [Initiator(hub.get_port(), my_generator(0, 2200*(i//6)+i*10))
+               for i in range(8)]
+       hub.finalize()
+       
+       slots = hub.get_slots()
+       slicer = _AddressSlicer(sdram_geom, 2)
+       logger = SlotsLogger(slicer, slots)
+       selector = Selector(slicer, 0, slots)
+       completer = Completer(hub, selector.queue)
+       
+       def end_simulation(s):
+               s.interrupt = all([i.done for i in initiators])
+       
+       fragment = hub.get_fragment() + sum([i.get_fragment() for i in initiators], Fragment()) + \
+               logger.get_fragment() + selector.get_fragment() + completer.get_fragment() + \
+               Fragment(sim=[end_simulation])
+       sim = Simulator(fragment, Runner(), TopLevel("my.vcd"))
+       sim.run()
+
+main()