from nmigen import (Module, Signal, Elaboratable, Const)
from nmigen.cli import rtlil
-from c4m.nmigen.jtag.tap import TAP
-from soc.debug.dmi import DMIInterface
+from c4m.nmigen.jtag.tap import TAP, IOType
+
+from soc.bus.sram import SRAM
+from nmigen import Memory, Signal, Module
+
+from nmigen.back.pysim import Simulator, Delay, Settle, Tick
+from nmutil.util import wrap
# JTAG to DMI interface
class DMITAP(TAP):
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- self._dmis = []
-
- def elaborate(self, platform):
- m = super().elaborate(platform)
- self._elaborate_dmis(m)
- return m
-
- def add_dmi(self, *, ircodes, address_width=8, data_width=64,
- domain="sync", name=None):
- """Add a DMI interface
-
- * writing to DMIADDR will automatically trigger a DMI READ.
- the DMI address does not alter (so writes can be done at that addr)
- * reading from DMIREAD triggers a DMI READ at the current DMI addr
- the address is automatically incremented by 1 after.
- * writing to DMIWRITE triggers a DMI WRITE at the current DMI addr
- the address is automatically incremented by 1 after.
-
- Parameters:
- -----------
- ircodes: sequence of three integer for the JTAG IR codes;
- they represent resp. DMIADDR, DMIREAD and DMIWRITE.
- First code has a shift register of length 'address_width',
- the two other codes share a shift register of length
- data_width.
-
- address_width: width of the address
- data_width: width of the data
-
- Returns:
- dmi: soc.debug.dmi.DMIInterface
- The DMI interface
- """
- if len(ircodes) != 3:
- raise ValueError("3 IR Codes have to be provided")
-
- if name is None:
- name = "dmi" + str(len(self._dmis))
-
- # add 2 shift registers: one for addr, one for data.
- sr_addr = self.add_shiftreg(ircode=ircodes[0], length=address_width,
- domain=domain, name=name+"_addrsr")
- sr_data = self.add_shiftreg(ircode=ircodes[1:], length=data_width,
- domain=domain, name=name+"_datasr")
-
- dmi = DMIInterface(name=name)
- self._dmis.append((sr_addr, sr_data, dmi, domain))
-
- return dmi
-
- def _elaborate_dmis(self, m):
- for sr_addr, sr_data, dmi, domain in self._dmis:
- cd = m.d[domain]
- m.d.comb += sr_addr.i.eq(dmi.addr_i)
-
- with m.FSM(domain=domain) as fsm:
- # detect mode based on whether jtag addr or data read/written
- with m.State("IDLE"):
- with m.If(sr_addr.oe): # DMIADDR code
- cd += dmi.addr_i.eq(sr_addr.o)
- m.next = "READ"
- with m.Elif(sr_data.oe[0]): # DMIREAD code
- # If data is
- cd += dmi.addr_i.eq(dmi.addr_i + 1)
- m.next = "READ"
- with m.Elif(sr_data.oe[1]): # DMIWRITE code
- cd += dmi.dout.eq(sr_data.o)
- m.next = "WRITE"
+ def external_ports(self):
+ return [self.bus.tdo, self.bus.tdi, self.bus.tms, self.bus.tck]
- # req_i raises for 1 clock
- with m.State("READ"):
- m.next = "READACK"
- # wait for read ack
- with m.State("READACK"):
- with m.If(dmi.ack):
- # Store read data in sr_data.i hold till next read
- cd += sr_data.i.eq(dmi.din)
- m.next = "IDLE"
+if __name__ == '__main__':
+ dut = DMITAP(ir_width=4)
+ iotypes = (IOType.In, IOType.Out, IOType.TriOut, IOType.InTriOut)
+ ios = [dut.add_io(iotype=iotype) for iotype in iotypes]
+ dut.sr = dut.add_shiftreg(ircode=4, length=3) # test loopback register
- # req_i raises for 1 clock
- with m.State("WRITE"):
- m.next = "WRITEACK"
+ # create and connect wishbone SRAM (a quick way to do WB test)
+ dut.wb = dut.add_wishbone(ircodes=[5, 6, 7], features={'err'},
+ address_width=16, data_width=16)
- # wait for write ack
- with m.State("WRITEACK"):
- with m.If(dmi.ack):
- cd += dmi.addr_i.eq(dmi.addr_i + 1)
- #m.next = "READ" - for readwrite
+ # create DMI2JTAG (goes through to dmi_sim())
+ dut.dmi = dut.add_dmi(ircodes=[8, 9, 10])
- # set DMI req and write-enable based on ongoing FSM states
- m.d.comb += [
- dmi.req_i.eq(fsm.ongoing("READ") | fsm.ongoing("WRITE")),
- dmi.we_i.eq(fsm.ongoing("WRITE")),
- ]
+ vl = rtlil.convert(dut)
+ with open("test_dmi2jtag.il", "w") as f:
+ f.write(vl)