Skip to content

Commit

Permalink
soc/cores/dma/WishboneDMAReader: Split add_csr() in add_ctrl() /add_c…
Browse files Browse the repository at this point in the history
…sr() since in some case just want to control the module from signals/user logic.
  • Loading branch information
enjoy-digital committed Jun 26, 2024
1 parent 06a26b7 commit 23a0d8f
Showing 1 changed file with 35 additions and 13 deletions.
48 changes: 35 additions & 13 deletions litex/soc/cores/dma.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,29 +71,30 @@ def __init__(self, bus, endianness="little", fifo_depth=16, with_csr=False):

# CSRs.
if with_csr:
self.add_ctrl()
self.add_csr()

def add_csr(self, default_base=0, default_length=0, default_enable=0, default_loop=0):
self._base = CSRStorage(64, reset=default_base)
self._length = CSRStorage(32, reset=default_length)
self._enable = CSRStorage(reset=default_enable)
self._done = CSRStatus()
self._loop = CSRStorage(reset=default_loop)
self._offset = CSRStatus(32)
def add_ctrl(self):
self.base = Signal(64)
self.length = Signal(32)
self.enable = Signal()
self.done = Signal()
self.loop = Signal()
self.offset = Signal(32)

# # #

shift = log2_int(self.bus.data_width//8)
base = Signal(self.bus.adr_width)
offset = Signal(self.bus.adr_width)
length = Signal(self.bus.adr_width)
self.comb += base.eq(self._base.storage[shift:])
self.comb += length.eq(self._length.storage[shift:])
self.comb += base.eq(self.base[shift:])
self.comb += length.eq(self.length[shift:])

self.comb += self._offset.status.eq(offset)
self.comb += self.offset.eq(offset)

self.fsm = fsm = ResetInserter()(FSM(reset_state="IDLE"))
self.comb += fsm.reset.eq(~self._enable.storage)
self.comb += fsm.reset.eq(~self.enable)
fsm.act("IDLE",
NextValue(offset, 0),
NextState("RUN"),
Expand All @@ -105,15 +106,36 @@ def add_csr(self, default_base=0, default_length=0, default_enable=0, default_lo
If(self.sink.ready,
NextValue(offset, offset + 1),
If(self.sink.last,
If(self._loop.storage,
If(self.loop,
NextValue(offset, 0)
).Else(
NextState("DONE")
)
)
)
)
fsm.act("DONE", self._done.status.eq(1))
fsm.act("DONE", self.done.eq(1))

def add_csr(self, default_base=0, default_length=0, default_enable=0, default_loop=0):
self._base = CSRStorage(64, reset=default_base)
self._length = CSRStorage(32, reset=default_length)
self._enable = CSRStorage(reset=default_enable)
self._done = CSRStatus()
self._loop = CSRStorage(reset=default_loop)
self._offset = CSRStatus(32)

# # #

self.comb += [
# Control.
self.base.eq(self._base.storage),
self.length.eq(self._length.storage),
self.enable.eq(self._enable.storage),
self.loop.eq(self._loop.storage),
# Status.
self._done.status.eq(self.done),
self._offset.status.eq(self.offset),
]

# WishboneDMAWriter --------------------------------------------------------------------------------

Expand Down

0 comments on commit 23a0d8f

Please sign in to comment.