width_visitor -> rename_visitor

Author: shtaxxx

veriloggen/core/module.py

@@ -9,7 +9,7 @@
 import veriloggen.core.vtypes as vtypes
 import veriloggen.core.function as function
 import veriloggen.core.task as task
-import veriloggen.core.width_visitor as width_visitor
+import veriloggen.core.rename_visitor as rename_visitor
 
 #-------------------------------------------------------------------------------
 class Module(vtypes.VeriloggenNode):
@@ -422,7 +422,7 @@ def copy_params(self, src, prefix=None, postfix=None, exclude=None):
         if postfix is None: postfix = ''
         if exclude is None: exclude = ()
         if isinstance(exclude, str): exclude = [ exclude ] 
-        visitor = width_visitor.WidthVisitor(prefix, postfix)
+        visitor = rename_visitor.RenameVisitor(prefix, postfix)
         ret = collections.OrderedDict()
         for key, obj in src.global_constant.items():
             skip = False
@@ -441,7 +441,7 @@ def copy_localparams(self, src, prefix=None, postfix=None, exclude=None):
         if postfix is None: postfix = ''
         if exclude is None: exclude = ()
         if isinstance(exclude, str): exclude = [ exclude ] 
-        visitor = width_visitor.WidthVisitor(prefix, postfix)
+        visitor = rename_visitor.RenameVisitor(prefix, postfix)
         ret = collections.OrderedDict()
         for key, obj in src.constant.items():
             skip = False
@@ -460,7 +460,7 @@ def copy_ports(self, src, prefix=None, postfix=None, exclude=None):
         if postfix is None: postfix = ''
         if exclude is None: exclude = ()
         if isinstance(exclude, str): exclude = [ exclude ] 
-        visitor = width_visitor.WidthVisitor(prefix, postfix)
+        visitor = rename_visitor.RenameVisitor(prefix, postfix)
         ret = collections.OrderedDict()
         for key, obj in src.io_variable.items():
             skip = False
@@ -479,7 +479,7 @@ def copy_vars(self, src, prefix=None, postfix=None, exclude=None):
         if postfix is None: postfix = ''
         if exclude is None: exclude = ()
         if isinstance(exclude, str): exclude = [ exclude ] 
-        visitor = width_visitor.WidthVisitor(prefix, postfix)
+        visitor = rename_visitor.RenameVisitor(prefix, postfix)
         ret = collections.OrderedDict()
         for key, obj in src.variable.items():
             skip = False
@@ -498,7 +498,7 @@ def copy_sim_ports(self, src, prefix=None, postfix=None, exclude=None):
         if postfix is None: postfix = ''
         if exclude is None: exclude = ()
         if isinstance(exclude, str): exclude = [ exclude ] 
-        visitor = width_visitor.WidthVisitor(prefix, postfix)
+        visitor = rename_visitor.RenameVisitor(prefix, postfix)
         ret = collections.OrderedDict()
         for key, obj in src.io_variable.items():
             skip = False

veriloggen/core/width_visitor.py renamed to veriloggen/core/rename_visitor.py

@@ -7,7 +7,7 @@
 
 import veriloggen.core.vtypes as vtypes
 
-class WidthVisitor(object):
+class RenameVisitor(object):
     def __init__(self, prefix=None, postfix=None):
         self.prefix = prefix if prefix is not None else ''
         self.postfix = postfix if postfix is not None else ''
@@ -27,6 +27,27 @@ def visit(self, node):
         visitor = getattr(self, 'visit_' + node.__class__.__name__, self.generic_visit)
         return visitor(node)
 
+    def visit_Int(self, node):
+        return node
+        
+    def visit_Float(self, node):
+        return node
+    
+    def visit_Str(self, node):
+        return node
+    
+    def visit_bool(self, node):
+        return node
+    
+    def visit_int(self, node):
+        return node
+
+    def visit_str(self, node):
+        return node
+
+    def visit_float(self, node):
+        return node
+
     def visit__Variable(self, node):
         ret = copy.deepcopy(node)
         ret.name = ''.join([self.prefix, ret.name, self.postfix])
@@ -51,6 +72,12 @@ def visit_Repeat(self, node):
         var = self.visit(node.var)
         times = self.visit(node.times)
         return vtypes.Repeat(var, times)
+
+    def visit_Cond(self, node):
+        cond = self.visit(node.condition)
+        true_value = self.visit(node.true_value)
+        false_value = self.visit(node.false_value)
+        return vtype.Cond(cond, true_value, false_value)
 
     def visit__BinaryOperator(self, node):
         op = type(node)