Added external modules. Switched lower firrtl back to wire r; r := Register, instead of using nodes. Added a renaming pass for different backends. This will likely get deprecated, as a more robust name mangling scheme could be needed

1 files changed, 36 insertions, 17 deletions

diff --git a/src/main/stanza/verilog.stanza b/src/main/stanza/verilog.stanza
index 19472573..79bccefb 100644
--- a/src/main/stanza/verilog.stanza
+++ b/src/main/stanza/verilog.stanza
@@ -109,7 +109,20 @@ defn emit (e:Expression) -> String :
                v = concat(v, [" ^ " emit(x)])
             v
 
-defn emit-module (m:Module) :
+defn emit-module (m:InModule) :
+   val h = Vector<Symbol>()
+   defn build-table (m:InModule) :
+      defn build-table (s:Stmt) -> Stmt :
+         match(map(build-table,s)) : 
+            (s:Connect) : 
+               match(exp(s)) :
+                  (e:Register) : add(h,name(loc(s) as Ref))
+                  (e) : false
+            (s) : false
+         s
+      build-table(body(m))
+   build-table(m)
+
    val wires = Vector<Streamable>()
    val regs = Vector<Streamable>()
    val inits = Vector<Streamable>()
@@ -120,13 +133,15 @@ defn emit-module (m:Module) :
    
    defn emit-s (s:Stmt) :
       match(map(remove-subfield,s)) :
-         (s:DefWire) : add(wires,["wire " get-width(type(s)) " " name(s) ";"])
+         (s:DefWire) : 
+            if contains?(to-list $ h, name(s)) : 
+               add(regs,["reg  " get-width(type(s)) " " name(s) ";"])
+            else :
+               add(wires,["wire " get-width(type(s)) " " name(s) ";"])
          (s:DefInstance) : 
             inst-ports[name(s)] = Vector<Streamable>()
             insts[name(s)] = name(module(s) as Ref)
             for f in fields(type(module(s)) as BundleType) do :
-               ;val sf = value(s) as Subfield
-               ;val e = exp(sf) as Ref
                val n* = to-symbol $ string-join $ [name(s) "_" name(f)]
                add(wires,["wire " get-width(type(f)) " " n* ";"])
                add(inst-ports[name(s)], ["." name(f) "( " n* " )"])
@@ -137,17 +152,6 @@ defn emit-module (m:Module) :
             add(inits,["for (initvar = 0; initvar < " size(vtype) "; initvar = initvar+1)"])
             add(inits,[name(s) " = {" width!(type(vtype)) "{$random}};"])
          (s:DefNode) :
-            if value(s) typeof Register :
-               val reg = value(s) as Register
-               add(regs,["reg  " get-width(type(reg)) " " name(s) ";"])
-               add(inits,[name(s) " = {" width!(type(reg)) "{$random}};"])
-               add(updates,["if(" emit(enable(reg)) ") begin"])
-               add(updates,["  " name(s) " <= " emit(value(reg)) ";"])
-               add(updates,["end"])
-            else if value(s) typeof ReadPort :
-               val rp = value(s) as ReadPort
-               add(assigns,["assign " name(s) " = " emit(mem(rp)) "[" emit(index(rp)) "];"])
-            else :
                add(wires,["wire " get-width(type(value(s))) " " name(s) ";"])
                add(assigns,["assign " name(s) " = " emit(value(s)) ";"])
          (s:Begin) : do(emit-s, body(s))
@@ -158,7 +162,19 @@ defn emit-module (m:Module) :
                add(updates,["  " emit(mem(wp)) "[" emit(index(wp)) "] <= " emit(exp(s)) ";"])
                add(updates,["end"])
             else :
-               add(assigns,["assign " emit(loc(s)) " = " emit(exp(s)) ";"])
+               if exp(s) typeof Register :
+                  val n = name(loc(s) as Ref)
+                  val reg = exp(s) as Register
+                  add(inits,[n " = {" width!(type(reg)) "{$random}};"])
+                  add(updates,["if(" emit(enable(reg)) ") begin"])
+                  add(updates,["  " n " <= " emit(value(reg)) ";"])
+                  add(updates,["end"])
+               else if exp(s) typeof ReadPort :
+                  val n = name(loc(s) as Ref)
+                  val rp = exp(s) as ReadPort
+                  add(assigns,["assign " n " = " emit(mem(rp)) "[" emit(index(rp)) "];"])
+               else :
+                  add(assigns,["assign " emit(loc(s)) " = " emit(exp(s)) ";"])
          (s) : s 
    
    emit-s(body(m))
@@ -219,5 +235,8 @@ defn emit-module (m:Module) :
 public defn emit-verilog (file:String, c:Circuit) :
    with-output-file{file, _} $ fn () :
       for m in modules(c) do :
-         emit-module(m)
+         match(m) :
+            (m:InModule) : emit-module(m)
+            (m:ExModule) : false
+
    c