litex/miscope/com/uart2csr/__init__.py

183 lines
3.1 KiB
Python

from migen.fhdl.structure import *
from migen.fhdl.module import *
from migen.bus import csr
from migen.genlib.fsm import FSM, NextState
from miscope.com.uart2csr.uart import *
WRITE_CMD = 0x01
READ_CMD = 0x02
CLOSE_CMD = 0x03
class Uart2Csr(Module):
def __init__(self, clk_freq, baud):
# Uart interface
self.rx = Signal()
self.tx = Signal()
# Csr interface
self.csr = csr.Interface()
###
self.submodules.uart = UART(clk_freq, baud)
uart = self.uart
#
# In/Out
#
self.comb +=[
uart.rx.eq(self.rx),
self.tx.eq(uart.tx)
]
cmd = Signal(8)
cnt = Signal(3)
sr = Signal(32)
burst_cnt = Signal(8)
addr = Signal(32)
data = Signal(8)
# FSM
self.submodules.fsm = FSM(reset_state="IDLE")
fsm = self.fsm
#
# Global
#
self.sync +=[
If(fsm.ongoing("IDLE"), cnt.eq(0)
).Elif(uart.rx_ev, cnt.eq(cnt + 1)),
If(uart.rx_ev, sr.eq(Cat(uart.rx_dat, sr[0:24])))
]
# State done signals
get_bl_done = Signal()
get_addr_done = Signal()
get_addr_done_d = Signal()
get_data_done = Signal()
send_data_done = Signal()
#
# Idle
#
fsm.act("IDLE",
If(uart.rx_ev & ((uart.rx_dat == WRITE_CMD) | (uart.rx_dat == READ_CMD)),
NextState("GET_BL")
)
)
self.sync += If(fsm.ongoing("IDLE") & uart.rx_ev, cmd.eq(uart.rx_dat))
#
# Get burst length
#
fsm.act("GET_BL",
If(get_bl_done,
NextState("GET_ADDR")
)
)
self.comb += get_bl_done.eq(uart.rx_ev & fsm.ongoing("GET_BL"))
self.sync += If(get_bl_done, burst_cnt.eq(uart.rx_dat))
#
# Get address
#
fsm.act("GET_ADDR",
If(get_addr_done & (cmd == WRITE_CMD),
NextState("GET_DATA")
).Elif(get_addr_done & (cmd == READ_CMD),
NextState("READ_CSR0")
)
)
self.comb += get_addr_done.eq(uart.rx_ev & (cnt == 4) & fsm.ongoing("GET_ADDR"))
self.sync += get_addr_done_d.eq(get_addr_done)
self.sync += [
If(get_addr_done_d,
addr.eq(sr)
).Elif(fsm.leaving("WRITE_CSR") | send_data_done,
addr.eq(addr + 1)
)
]
#
# Get data
#
fsm.act("GET_DATA",
If(get_data_done,
NextState("WRITE_CSR")
)
)
self.comb += get_data_done.eq(uart.rx_ev & fsm.ongoing("GET_DATA"))
self.sync += [
If(get_data_done,
burst_cnt.eq(burst_cnt-1),
data.eq(uart.rx_dat)
)
]
#
# Write Csr
#
fsm.act("WRITE_CSR",
If((burst_cnt==0),
NextState("IDLE")
).Else(NextState("GET_DATA"))
)
#
# Read Csr0
#
fsm.act("READ_CSR0",
NextState("READ_CSR1")
)
self.sync += If(fsm.entering("READ_CSR0"), burst_cnt.eq(burst_cnt-1))
#
# Read Csr1
#
fsm.act("READ_CSR1",
NextState("SEND_DATA")
)
#
# Send Data
#
fsm.act("SEND_DATA",
If(send_data_done & (burst_cnt==0),
NextState("IDLE")
).Elif(send_data_done,
NextState("READ_CSR0")
)
)
self.comb += send_data_done.eq(fsm.ongoing("SEND_DATA") & uart.tx_ev)
self.sync += [
uart.tx_dat.eq(self.csr.dat_r),
uart.tx_we.eq(fsm.entering("SEND_DATA")),
]
#
# Csr access
#
self.comb += self.csr.adr.eq(addr)
self.sync +=[
self.csr.dat_w.eq(data),
If(fsm.ongoing("WRITE_CSR"),
self.csr.we.eq(1)
).Else(
self.csr.we.eq(0)
)
]