Dataflow.acc_custom() supports to append a label for graph drawing.

Author: shtaxxx

tests/lib_dataflow_/acc_custom/Makefile

@@ -26,4 +26,4 @@ check:
 
 .PHONY: clean
 clean:
-	rm -rf *.pyc __pycache__ parsetab.py *.out tmp.v uut.vcd
+	rm -rf *.pyc __pycache__ parsetab.py *.out *.png *.dot tmp.v uut.vcd 

tests/lib_dataflow_/acc_custom/lib_dataflow_acc_custom.py

@@ -28,10 +28,15 @@ def mkLed():
     def op_max(left, right):
         return Cond(left > right, left, right)
 
-    psum = df.acc_custom(px, op_max, initval=0, resetcond=prst)
+    psum = df.acc_custom(px, op_max, initval=0, resetcond=prst, label='custom')
     psum.output(y, valid=vy, ready=ry)
 
     df.make_always()
+    
+    try:
+        df.draw_graph()
+    except:
+        print('Dataflow graph could not be generated.', file=sys.stderr)
 
     return m
 

veriloggen/lib/dataflow.py

@@ -90,10 +90,10 @@ def op(left, right):
             return vtypes.Cond(left < right, left, right)
         return self._accumulate([op], data, width, initval, resetcond, 'min')
 
-    def acc_custom(self, data, ops, initval=0, resetcond=None, width=None):
+    def acc_custom(self, data, ops, initval=0, resetcond=None, width=None, label=None):
         if not isinstance(ops, (tuple, list)):
             ops = [ ops ]
-        return self._accumulate(ops, data, width, initval, resetcond)
+        return self._accumulate(ops, data, width, initval, resetcond, label)
 
     #---------------------------------------------------------------------------
     def make_always(self, reset=(), body=()):