litex/misoc/mem/sdram/test/bankmachine_tb.py

44 lines
1.3 KiB
Python

from migen import *
from migen.sim.generic import run_simulation
from misoc.mem.sdram.code import lasmibus
from misoc.mem.sdram.core.lasmicon.bankmachine import *
from common import sdram_phy, sdram_geom, sdram_timing, CommandLogger
def my_generator():
for x in range(10):
yield True, x
for x in range(10):
yield False, 128*x
class TB(Module):
def __init__(self):
self.req = Interface(32, 32, 1,
sdram_timing.req_queue_size, sdram_phy.read_latency, sdram_phy.write_latency)
self.submodules.dut = BankMachine(sdram_geom, sdram_timing, 2, 0, self.req)
self.submodules.logger = CommandLogger(self.dut.cmd, True)
self.generator = my_generator()
self.dat_ack_cnt = 0
def do_simulation(self, selfp):
if selfp.req.dat_ack:
self.dat_ack_cnt += 1
if selfp.req.req_ack:
try:
we, adr = next(self.generator)
except StopIteration:
selfp.req.stb = 0
if not selfp.req.lock:
print("data ack count: {0}".format(self.dat_ack_cnt))
raise StopSimulation
return
selfp.req.adr = adr
selfp.req.we = we
selfp.req.stb = 1
if __name__ == "__main__":
run_simulation(TB(), vcd_name="my.vcd")