m-labs · Aug 1, 2016 · Aug 1, 2016
Showing with 8 additions and 1 deletion.

+1 −0 migen/__init__.py

+7 −1 migen/fhdl/module2.py
diff --git a/migen/__init__.py b/migen/__init__.py
@@ -1,5 +1,6 @@
 from migen.fhdl.structure import *
 from migen.fhdl.module import *
+from migen.fhdl.module2 import *
 from migen.fhdl.specials import *
 from migen.fhdl.bitcontainer import *
 from migen.fhdl.decorators import *

diff --git a/migen/fhdl/module2.py b/migen/fhdl/module2.py
@@ -8,7 +8,7 @@
 from migen.fhdl.structure import _Assign, _Operator, _Fragment
 
 
-__all__ = ["Module2", "ports", "sync"]
+__all__ = ["Module2", "ports", "comb", "sync"]
 
 
 class PortManager:
@@ -107,6 +107,12 @@ def ast_stmt_to_fhdl(node, globs, locs):
             assert t.attr == "next"
             r.append(_Assign(eval_ast(t.value, globs, locs), value))
         return r
+    elif isinstance(node, ast.If):
+        return If(ast_expr_to_fhdl(node.test, globs, locs),
+                  *ast_stmt_to_fhdl(node.body, globs, locs)).Else(
+                  *ast_stmt_to_fhdl(node.orelse, globs, locs))
+    elif isinstance(node, ast.Expr):
+        return eval_ast(node.value, globs, locs)
     else:
         raise NotImplementedError