from migen.fhdl.std import *
from migen.bus.lasmibus import *
from migen.sim.generic import run_simulation

from misoclib.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")