You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
3.2 KiB
94 lines
3.2 KiB
import logging
|
|
import random
|
|
import cocotb
|
|
import wavedrom
|
|
from collections import defaultdict
|
|
from Sram import SramRead, SramWrite, SramMonitor
|
|
from cocotb.clock import Clock
|
|
from cocotb.triggers import RisingEdge, Timer
|
|
from cocotbext.wishbone.driver import WishboneMaster, WBOp
|
|
from cocotb.wavedrom import Wavedrom, trace
|
|
|
|
|
|
# Reset coroutine
|
|
async def reset_dut(reset_n, duration_ns):
|
|
reset_n.value = 1
|
|
await Timer(duration_ns, units="ns")
|
|
reset_n.value = 0
|
|
|
|
def wave2svg(wave, file):
|
|
svg = wavedrom.render(wave)
|
|
svg.saveas(file)
|
|
|
|
|
|
@cocotb.test()
|
|
async def test_wishbone(dut):
|
|
""" First simple test """
|
|
|
|
clkedge = RisingEdge(dut.wbclk_i)
|
|
|
|
# Connect reset
|
|
reset = dut.wbrst_i
|
|
|
|
|
|
# Create empty SRAM memory
|
|
memory = defaultdict()
|
|
|
|
mem_read = SramRead(dut.wbclk_i, dut.localren_o,
|
|
dut.localadress_o, dut.localdata_i, memory);
|
|
mem_write = SramWrite(dut.wbclk_i, dut.localwen_o,
|
|
dut.localadress_o, dut.localdata_o, memory);
|
|
sram_monitor = SramMonitor(dut.wbclk_i, dut.localwen_o, dut.localren_o,
|
|
dut.localadress_o, dut.localdata_i, dut.localdata_o);
|
|
|
|
wbmaster = WishboneMaster(dut, "", dut.wbclk_i,
|
|
width=16, # size of data bus
|
|
timeout=10, # in clock cycle number
|
|
signals_dict={"cyc": "wbcyc_i",
|
|
"stb": "wbstb_i",
|
|
"we": "wbwe_i",
|
|
"adr": "wbadr_i",
|
|
"datwr":"wbdat_i",
|
|
"datrd":"wbdat_o",
|
|
"ack": "wback_o" })
|
|
|
|
# Drive input defaults (setimmediatevalue to avoid x asserts)
|
|
dut.wbcyc_i.setimmediatevalue(0)
|
|
dut.wbstb_i.setimmediatevalue(0)
|
|
dut.wbwe_i.setimmediatevalue(0)
|
|
dut.wbadr_i.setimmediatevalue(0)
|
|
dut.wbdat_i.setimmediatevalue(0)
|
|
|
|
clock = Clock(dut.wbclk_i, 10, units="ns") # Create a 10 ns period clock
|
|
cocotb.start_soon(clock.start()) # Start the clock
|
|
|
|
# Execution will block until reset_dut has completed
|
|
await reset_dut(reset, 100)
|
|
dut._log.info("Released reset")
|
|
|
|
# Trace transmissions using wavedrom
|
|
with trace(dut.wbcyc_i, dut.wbstb_i, dut.wbwe_i, dut.wback_o,
|
|
dut.wbadr_i, dut.wbdat_i, dut.wbdat_o, clk=dut.wbclk_i) as waves:
|
|
|
|
# Test 10 Wishbone transmissions
|
|
for i in range(10):
|
|
await clkedge
|
|
adr = random.randint(0, 255)
|
|
data = random.randint(0, 2**16-1)
|
|
await wbmaster.send_cycle([WBOp(adr=adr, dat=data)])
|
|
rec = await wbmaster.send_cycle([WBOp(adr=adr)])
|
|
assert rec[0].datrd == data, \
|
|
f"Read data incorrect, got {hex(rec[0].datrd)}, expected {hex(data)}"
|
|
|
|
# Print out waveforms as json & svg
|
|
_wave = waves.dumpj()
|
|
with open('results/tb_wishbone_wave.json', 'w', encoding='utf-8') as f:
|
|
f.write(_wave)
|
|
wave2svg(_wave, 'results/tb_wishbone_wave.svg')
|
|
|
|
|
|
# Example to print transactions collected by SRAM monitor
|
|
with open('results/tb_wishbone_sram_transactions.log', 'w', encoding='utf-8') as f:
|
|
f.write((f"{'Time':7}{'Type':7}{'Adr':6}{'Data'}\n"))
|
|
for key, value in sram_monitor.transactions.items():
|
|
f.write((f"{key:7}{value['type']:7}{hex(value['adr']):6}{hex(value['data'])}\n"))
|