From d00937f981ce0fd5e748b86cc76cdcf65319850b Mon Sep 17 00:00:00 2001
From: James Cherry <cherry@parallaxsw.com>
Date: Tue, 19 Jan 2021 12:40:49 -0700
Subject: [PATCH] write_verilog wire stmts

---
 verilog/VerilogWriter.cc | 46 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/verilog/VerilogWriter.cc b/verilog/VerilogWriter.cc
index 981cd89e..c7456e88 100644
--- a/verilog/VerilogWriter.cc
+++ b/verilog/VerilogWriter.cc
@@ -17,6 +17,7 @@
 #include "VerilogWriter.hh"
 
 #include <stdlib.h>
+#include <algorithm>
 
 #include "Error.hh"
 #include "Liberty.hh"
@@ -24,9 +25,13 @@
 #include "Network.hh"
 #include "NetworkCmp.hh"
 #include "VerilogNamespace.hh"
+#include "ParseBus.hh"
 
 namespace sta {
 
+using std::min;
+using std::max;
+
 class VerilogWriter
 {
 public:
@@ -41,6 +46,7 @@ public:
 protected:
   void writePorts(Cell *cell);
   void writePortDcls(Cell *cell);
+  void writeWireDcls(Instance *inst);
   const char *verilogPortDir(PortDirection *dir);
   void writeChildren(Instance *inst);
   void writeChild(Instance *child);
@@ -114,6 +120,8 @@ VerilogWriter::writeModule(Instance *inst)
   writePorts(cell);
   writePortDcls(cell);
   fprintf(stream_, "\n");
+  writeWireDcls(inst);
+  fprintf(stream_, "\n");
   writeChildren(inst);
   fprintf(stream_, "endmodule\n");
   written_cells_.insert(cell);
@@ -195,6 +203,44 @@ VerilogWriter::verilogPortDir(PortDirection *dir)
   }
 }
 
+typedef std::pair<int, int> BusIndexRange;
+
+void
+VerilogWriter::writeWireDcls(Instance *inst)
+{
+  Cell *cell = network_->cell(inst);
+  char escape = network_->pathEscape();
+  Map<const char*, BusIndexRange, CharPtrLess> bus_ranges;
+  NetIterator *net_iter = network_->netIterator(inst);
+  while (net_iter->hasNext()) {
+    Net *net = net_iter->next();
+    const char *net_name = network_->name(net);
+    if (network_->findPort(cell, net_name) == nullptr) {
+      if (isBusName(net_name, '[', ']', escape)) {
+        char *bus_name;
+        int index;
+        parseBusName(net_name, '[', ']', escape, bus_name, index);
+        BusIndexRange &range = bus_ranges[bus_name];
+        range.first = max(range.first, index);
+        range.second = min(range.second, index);
+      }
+      else
+        fprintf(stream_, " wire %s;\n",
+                netVerilogName(net_name, network_->pathEscape()));;
+    }
+  }
+  delete net_iter;
+
+  for (auto name_range : bus_ranges) {
+    const char *bus_name = name_range.first;
+    const BusIndexRange &range = name_range.second;
+    fprintf(stream_, " wire [%d:%d] %s;\n",
+            range.first,
+            range.second,
+            netVerilogName(bus_name, network_->pathEscape()));;
+  }
+}
+
 void
 VerilogWriter::writeChildren(Instance *inst)			     
 {