114 lines
2.5 KiB
Python
114 lines
2.5 KiB
Python
from migen.fhdl.std import *
|
|
from migen.bank.description import *
|
|
from migen.genlib.fifo import SyncFIFO
|
|
from migen.genlib.fsm import FSM, NextState
|
|
|
|
from miscope.std import *
|
|
|
|
class RunLengthEncoder(Module, AutoCSR):
|
|
def __init__(self, width, length=1024):
|
|
self.width = width
|
|
self.length = length
|
|
|
|
self.sink = rec_dat(width)
|
|
self.source = rec_dat(width)
|
|
|
|
self._r_enable = CSRStorage()
|
|
|
|
###
|
|
|
|
enable = self._r_enable.storage
|
|
|
|
fsm = FSM(reset_state="BYPASS")
|
|
self.submodules += fsm
|
|
|
|
sink_d = rec_dat(width)
|
|
self.sync += If(self.sink.stb, sink_d.eq(self.sink))
|
|
|
|
cnt = Signal(max=length)
|
|
cnt_inc = Signal()
|
|
cnt_reset = Signal()
|
|
cnt_max = Signal()
|
|
|
|
self.sync += \
|
|
If(cnt_reset,
|
|
cnt.eq(1),
|
|
).Elif(cnt_inc,
|
|
cnt.eq(cnt+1)
|
|
)
|
|
self.comb += cnt_max.eq(cnt == length)
|
|
|
|
change = Signal()
|
|
self.comb += change.eq(self.sink.stb & (self.sink.dat != sink_d.dat))
|
|
|
|
fsm.act("BYPASS",
|
|
sink_d.connect(self.source),
|
|
cnt_reset.eq(1),
|
|
If(enable & ~change & self.sink.stb, NextState("COUNT"))
|
|
)
|
|
|
|
fsm.act("COUNT",
|
|
cnt_inc.eq(self.sink.stb),
|
|
If(change | cnt_max | ~enable,
|
|
self.source.stb.eq(1),
|
|
self.source.dat[width-1].eq(1), # Set RLE bit
|
|
self.source.dat[:flen(cnt)].eq(cnt),
|
|
NextState("BYPASS")
|
|
)
|
|
),
|
|
|
|
class Recorder(Module, AutoCSR):
|
|
def __init__(self, width, depth):
|
|
self.width = width
|
|
|
|
self.trig_sink = rec_hit()
|
|
self.dat_sink = rec_dat(width)
|
|
|
|
self._r_trigger = CSR()
|
|
self._r_length = CSRStorage(bits_for(depth))
|
|
self._r_offset = CSRStorage(bits_for(depth))
|
|
self._r_done = CSRStatus()
|
|
|
|
self._r_read_en = CSR()
|
|
self._r_read_empty = CSRStatus()
|
|
self._r_read_dat = CSRStatus(width)
|
|
|
|
###
|
|
|
|
fifo = SyncFIFO(width, depth)
|
|
self.submodules += fifo
|
|
|
|
fsm = FSM(reset_state="IDLE")
|
|
self.submodules += fsm
|
|
|
|
|
|
self.comb += [
|
|
self._r_read_empty.status.eq(~fifo.readable),
|
|
self._r_read_dat.status.eq(fifo.dout),
|
|
]
|
|
|
|
fsm.act("IDLE",
|
|
If(self._r_trigger.re & self._r_trigger.r,
|
|
NextState("PRE_HIT_RECORDING"),
|
|
fifo.flush.eq(1),
|
|
),
|
|
fifo.re.eq(self._r_read_en.re & self._r_read_en.r),
|
|
self._r_done.status.eq(1)
|
|
)
|
|
|
|
fsm.act("PRE_HIT_RECORDING",
|
|
fifo.we.eq(self.dat_sink.stb),
|
|
fifo.din.eq(self.dat_sink.dat),
|
|
|
|
fifo.re.eq(fifo.level >= self._r_offset.storage),
|
|
|
|
If(self.trig_sink.stb & self.trig_sink.hit, NextState("POST_HIT_RECORDING"))
|
|
)
|
|
|
|
fsm.act("POST_HIT_RECORDING",
|
|
fifo.we.eq(self.dat_sink.stb),
|
|
fifo.din.eq(self.dat_sink.dat),
|
|
|
|
If(~fifo.writable | (fifo.level >= self._r_length.storage), NextState("IDLE"))
|
|
)
|