Fixed assignment to outputs not getting emitted from Expand When pass

1 files changed, 12 insertions, 1 deletions

diff --git a/src/main/stanza/passes.stanza b/src/main/stanza/passes.stanza
index 5d1848f0..4c706087 100644
--- a/src/main/stanza/passes.stanza
+++ b/src/main/stanza/passes.stanza
@@ -1088,7 +1088,17 @@ defn reduce-or (l:List<Expression>) -> Expression :
 ; kinds:  Used to know the kind of reference, so we know whether we should error if it isn't initialized. We also know how we should declare the refernce.
 ; enables:Calculated off of assigns.
 
-; I think I'm going to restructure this so that not all information is held in the tables, but instead, we walk the graph again, and do stuff on declarations, and delete other stuff
+
+defn expand-whens (ports:List<Port>, table:HashTable<Symbol,SymbolicValue>,cons:Vector<Stmt>) -> False :
+   for p in ports do :
+      if direction(p) == OUTPUT :
+         val ref = WRef(name(p),type(p),PortKind(),FEMALE)
+         add{cons,_} $ 
+            if has-nul?(table[name(p)]) : 
+               println("Uninitialized: ~" % [to-string(name(p))]);TODO actually collect error
+               EmptyStmt()
+            else : Connect(ref,to-exp(table[name(p)]) as Expression)
+
 defn expand-whens (s:Stmt, table:HashTable<Symbol,SymbolicValue>,decs:Vector<Stmt>,cons:Vector<Stmt>) -> Stmt :
    match(map(expand-whens{_,table,decs,cons},s)) : 
       (s:DefNode|DefMemory) : add(decs,s)
@@ -1277,6 +1287,7 @@ defn expand-whens (m:Module) -> Module :
    for x in table do : println-debug(x)
    val decs = Vector<Stmt>()
    val cons = Vector<Stmt>()
+   expand-whens(ports(m),table,cons)
    expand-whens(body(m),table,decs,cons)
    Module(name(m),ports(m),Begin(append(to-list(decs),to-list(cons))))