task statement is implemented but not tested

Author: shtaxxx

sample/test/function/led.py

@@ -5,12 +5,12 @@
 
 # external function definition
 def get_ext():
-    ext = Function('ext', 32)
+    ext = Function('ext_inc', 32)
     ext_v = ext.Input('v', 32)
     ext_o = ext.Input('o', 32)
     ext_tmp = ext.Reg('tmp', 32)
     ext.Body(
-        ext_tmp(ext_v + 1),
+        ext_tmp(ext_v + ext_o),
         ext(ext_tmp)
     )
     return ext
@@ -28,7 +28,7 @@ def mkLed():
     inc_o = inc.Input('o', width)
     inc_tmp = inc.Reg('tmp', width)
     inc.Body(
-        inc_tmp(inc_v + 1),
+        inc_tmp(inc_v + inc_o),
         inc(inc_tmp)
     )
 
@@ -42,7 +42,8 @@ def mkLed():
             If(count == 1023)(
                 count(0)
             ).Else(
-                count( inc.call(count, 1) )
+                count( FunctionCall(inc.name, count, 1) )
+                #count( inc.call(count, 1) )
             )
         ))
 

sample/test/function/test_led.py

@@ -17,18 +17,18 @@
     input [WIDTH-1:0] o;
     reg [WIDTH-1:0] tmp;
     begin        
-      tmp = v + 1;
+      tmp = v + o;
       inc = tmp;
     end 
   endfunction
 
-  function [32-1:0] ext;
+  function [32-1:0] ext_inc;
     input [32-1:0] v;
     input [32-1:0] o;
     reg [32-1:0] tmp;
     begin        
-      tmp = v + 1;
-      ext = tmp;
+      tmp = v + o;
+      ext_inc = tmp;
     end 
   endfunction
 

veriloggen/task.py

@@ -0,0 +1,45 @@
+import os
+import sys
+import collections
+
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+import vtypes
+
+class Task(vtypes.VeriloggenNode):
+    def __init__(self, name, width=1):
+        self.name = name
+        self.width = width
+        self.io_variable = collections.OrderedDict()
+        self.variable = collections.OrderedDict()
+        self.statement = None
+
+    #---------------------------------------------------------------------------
+    def Input(self, name, width=None, length=None, signed=False, value=None):
+        t = vtypes.Input(name, width, length, signed, value)
+        self.io_variable[name] = t
+        return t
+    
+    def Reg(self, name, width=None, length=None, signed=False, value=None):
+        t = vtypes.Reg(name, width, length, signed, value)
+        self.variable[name] = t
+        return t
+    
+    def Integer(self, name, width=None, length=None, signed=False, value=None):
+        t = vtypes.Integer(name, width, length, signed, value)
+        self.variable[name] = t
+        return t
+    
+    def Body(self, *statement):
+        if self.statement is not None:
+            raise ValueError("Statement is already assigned.")
+        self.statement = tuple(statement)
+        return self
+
+    def call(self, *args):
+        return TaskCall(self.name, *args)
+
+class TaskCall(vtypes.VeriloggenNode):
+    def __init__(self, name, *args):
+        self.name = name
+        self.args = tuple(args)