litex/misoc/cores/sdram_tester.py

120 lines
3.3 KiB
Python
Raw Normal View History

2015-09-22 12:36:47 -04:00
from migen import *
2013-07-10 15:08:57 -04:00
from migen.genlib.misc import optree
2013-07-11 12:31:38 -04:00
from migen.bank.description import *
from migen.actorlib.spi import *
2013-07-10 15:08:57 -04:00
2015-09-22 12:35:02 -04:00
from misoc.mem.sdram.frontend import dma_lasmi
2015-04-13 10:47:22 -04:00
2013-07-25 11:57:17 -04:00
@DecorateModule(InsertReset)
@DecorateModule(InsertCE)
2013-07-10 15:08:57 -04:00
class LFSR(Module):
def __init__(self, n_out, n_state=31, taps=[27, 30]):
self.o = Signal(n_out)
2013-07-10 15:08:57 -04:00
###
2013-07-10 15:08:57 -04:00
state = Signal(n_state)
curval = [state[i] for i in range(n_state)]
curval += [0]*(n_out - n_state)
for i in range(n_out):
nv = ~optree("^", [curval[tap] for tap in taps])
curval.insert(0, nv)
curval.pop()
2013-07-10 15:08:57 -04:00
self.sync += [
state.eq(Cat(*curval[:n_state])),
self.o.eq(Cat(*curval))
]
2013-07-10 15:08:57 -04:00
2013-07-11 12:31:38 -04:00
memtest_magic = 0x361f
2015-04-13 10:47:22 -04:00
2013-07-11 12:31:38 -04:00
class MemtestWriter(Module):
def __init__(self, lasmim):
self._magic = CSRStatus(16)
self._reset = CSR()
self._shoot = CSR()
self.submodules._dma = DMAWriteController(dma_lasmi.Writer(lasmim),
MODE_EXTERNAL)
###
self.comb += self._magic.status.eq(memtest_magic)
lfsr = LFSR(lasmim.dw)
self.submodules += lfsr
self.comb += lfsr.reset.eq(self._reset.re)
en = Signal()
en_counter = Signal(lasmim.aw)
self.comb += en.eq(en_counter != 0)
self.sync += [
If(self._shoot.re,
en_counter.eq(self._dma.length)
).Elif(lfsr.ce,
en_counter.eq(en_counter - 1)
)
]
self.comb += [
self._dma.trigger.eq(self._shoot.re),
self._dma.data.stb.eq(en),
lfsr.ce.eq(en & self._dma.data.ack),
self._dma.data.d.eq(lfsr.o)
]
def get_csrs(self):
return [self._magic, self._reset, self._shoot] + self._dma.get_csrs()
2013-07-11 12:31:38 -04:00
2015-04-13 10:47:22 -04:00
2013-07-11 12:31:38 -04:00
class MemtestReader(Module):
def __init__(self, lasmim):
self._magic = CSRStatus(16)
self._reset = CSR()
self._error_count = CSRStatus(lasmim.aw)
self.submodules._dma = DMAReadController(dma_lasmi.Reader(lasmim),
MODE_SINGLE_SHOT)
###
self.comb += self._magic.status.eq(memtest_magic)
lfsr = LFSR(lasmim.dw)
self.submodules += lfsr
self.comb += lfsr.reset.eq(self._reset.re)
self.comb += [
lfsr.ce.eq(self._dma.data.stb),
self._dma.data.ack.eq(1)
]
err_cnt = self._error_count.status
self.sync += [
If(self._reset.re,
err_cnt.eq(0)
).Elif(self._dma.data.stb,
If(self._dma.data.d != lfsr.o, err_cnt.eq(err_cnt + 1))
)
]
def get_csrs(self):
return [self._magic, self._reset, self._error_count] + self._dma.get_csrs()
2013-07-11 12:31:38 -04:00
2015-04-13 10:47:22 -04:00
class _LFSRTB(Module):
def __init__(self, *args, **kwargs):
self.submodules.dut = LFSR(*args, **kwargs)
self.comb += self.dut.ce.eq(1)
def do_simulation(self, selfp):
print("{0:032x}".format(selfp.dut.o))
2013-07-10 15:08:57 -04:00
if __name__ == "__main__":
from migen.fhdl import verilog
from migen.sim.generic import run_simulation
lfsr = LFSR(3, 4, [3, 2])
print(verilog.convert(lfsr, ios={lfsr.ce, lfsr.reset, lfsr.o}))
run_simulation(_LFSRTB(128), ncycles=20)