end_flag

Author: shtaxxx

veriloggen/thread/stream.py

@@ -75,6 +75,10 @@ def __init__(self, m, name, clk, rst,
             '_'.join(['', self.name, 'start_flag']))
         self.start = self.module.Reg(
             '_'.join(['', self.name, 'start']), initval=0)
+
+        self.end_flag = self.module.Reg(
+            '_'.join(['', self.name, 'end_flag']), initval=0)
+
         self.source_busy = self.module.Reg(
             '_'.join(['', self.name, 'source_busy']), initval=0)
         self.sink_busy = self.module.Reg(
@@ -920,6 +924,13 @@ def run(self, fsm):
             self.sink_wait_count.inc()
         )
 
+        self.fsm.seq(
+            self.end_flag(0)
+        )
+        self.fsm.seq.If(self.seq.Prev(end_cond, num_wdelay))(
+            self.end_flag(1)
+        )
+
         self.fsm.goto_init()
 
         fsm.goto_next()