rsz: add some gui debugging for repair_design

Draw the Steiner tree and the elements being added along it.

Signed-off-by: Matt Liberty <[email protected]>

Author: maliberty

src/rsz/BUILD

@@ -77,6 +77,7 @@ cc_library(
 cc_library(
     name = "ui",
     srcs = [
+        "src/BufferedNet.hh",
         "src/Graphics.cc",
         "src/Graphics.hh",
         "src/MakeResizer.cc",

src/rsz/src/Graphics.cc

@@ -54,12 +54,79 @@ void Graphics::subdivideDone()
   }
 }
 
+void Graphics::repairNetStart(const BufferedNetPtr& bnet, odb::dbNet* net)
+{
+  lines_.clear();
+  if (net_) {
+    repair_net_ignore_ = (net != net_);
+  }
+  if (!repair_net_ignore_) {
+    bnet_ = bnet;
+    gui::Gui::get()->redraw();
+    gui::Gui::get()->pause();
+  }
+}
+
+void Graphics::makeBuffer(odb::dbInst* inst)
+{
+  if (!repair_net_ignore_) {
+    buffers_.push_back(inst);
+    gui::Gui::get()->redraw();
+    gui::Gui::get()->pause();
+  }
+}
+
+void Graphics::repairNetDone()
+{
+  if (!repair_net_ignore_) {
+    gui::Gui::get()->redraw();
+    gui::Gui::get()->pause();
+  }
+  bnet_.reset();
+  buffers_.clear();
+}
+
+void Graphics::drawBNet(const BufferedNetPtr& bnet, gui::Painter& painter)
+{
+  switch (bnet->type()) {
+    case BufferedNetType::wire: {
+      painter.drawLine(bnet->location(), bnet->ref()->location());
+      drawBNet(bnet->ref(), painter);
+      break;
+    }
+    case BufferedNetType::junction:
+      painter.drawLine(bnet->location(), bnet->ref()->location());
+      painter.drawLine(bnet->location(), bnet->ref2()->location());
+      drawBNet(bnet->ref(), painter);
+      drawBNet(bnet->ref2(), painter);
+      break;
+    case BufferedNetType::load: {
+      Point loc = bnet->location();
+      painter.drawCircle(loc.x(), loc.y(), 1000);
+      break;
+    }
+    case BufferedNetType::buffer:
+      painter.drawLine(bnet->location(), bnet->ref()->location());
+      drawBNet(bnet->ref(), painter);
+      break;
+  }
+}
+
 void Graphics::drawObjects(gui::Painter& painter)
 {
   painter.setPen(gui::Painter::kRed, true);
   for (const odb::Line& line : lines_) {
     painter.drawLine(line.pt0(), line.pt1());
   }
+
+  if (bnet_) {
+    painter.setPenAndBrush(gui::Painter::kPink, true);
+    drawBNet(bnet_, painter);
+    painter.setPenAndBrush(gui::Painter::kGreen, true);
+    for (odb::dbInst* buffer : buffers_) {
+      painter.drawRect(buffer->getBBox()->getBox());
+    }
+  }
 }
 
 }  // namespace rsz

src/rsz/src/Graphics.hh

@@ -20,14 +20,23 @@ class Graphics : public gui::Renderer, public ResizerObserver
   void subdivide(const odb::Line& line) override;
   void subdivideDone() override;
 
+  void repairNetStart(const BufferedNetPtr& bnet, odb::dbNet* net) override;
+  void makeBuffer(odb::dbInst* inst) override;
+  void repairNetDone() override;
+
   // Renderer
   void drawObjects(gui::Painter& painter) override;
 
  private:
+  void drawBNet(const BufferedNetPtr& bnet, gui::Painter& painter);
+
+  BufferedNetPtr bnet_;
+  std::vector<odb::dbInst*> buffers_;
   odb::dbNet* net_{nullptr};
   std::vector<odb::Line> lines_;
-  // Ingore this net if true
+  // Ignore this net if true
   bool subdivide_ignore_{false};
+  bool repair_net_ignore_{false};
   // stop at each step?
   bool stop_on_subdivide_step_{false};
 };

src/rsz/src/RepairDesign.cc

@@ -777,6 +777,10 @@ bool RepairDesign::performGainBuffering(Net* net,
 
     repaired_net = true;
     inserted_buffer_count_++;
+    if (graphics_) {
+      dbInst* db_inst = db_network_->staToDb(inst);
+      graphics_->makeBuffer(db_inst);
+    }
 
     int max_level = 0;
     for (auto it = sinks.begin(); it != group_end; it++) {
@@ -1226,9 +1230,19 @@ void RepairDesign::repairNet(const BufferedNetPtr& bnet,
   max_length_ = max_length;
   corner_ = corner;
 
+  if (graphics_) {
+    Net* net = db_network_->net(drvr_pin);
+    odb::dbNet* db_net = db_network_->staToDb(net);
+    graphics_->repairNetStart(bnet, db_net);
+  }
+
   int wire_length;
   PinSeq load_pins;
   repairNet(bnet, 0, wire_length, load_pins);
+
+  if (graphics_) {
+    graphics_->repairNetDone();
+  }
 }
 
 void RepairDesign::repairNet(const BufferedNetPtr& bnet,
@@ -2575,6 +2589,11 @@ bool RepairDesign::makeRepeater(
     }
   }
 
+  if (graphics_) {
+    dbInst* db_inst = db_network_->staToDb(buffer);
+    graphics_->makeBuffer(db_inst);
+  }
+
   // Resize repeater as we back up by levels.
   if (resize) {
     Pin* buffer_out_pin = network_->findPin(buffer, buffer_output_port);

src/rsz/src/ResizerObserver.hh

@@ -3,6 +3,7 @@
 
 #pragma once
 
+#include "BufferedNet.hh"
 #include "odb/db.h"
 #include "odb/geom.h"
 
@@ -19,6 +20,10 @@ class ResizerObserver
   virtual void subdivideStart(odb::dbNet* net) {}
   virtual void subdivide(const odb::Line& line) {}
   virtual void subdivideDone() {}
+
+  virtual void repairNetStart(const BufferedNetPtr& bnet, odb::dbNet* net) {}
+  virtual void makeBuffer(odb::dbInst* inst) {}
+  virtual void repairNetDone() {}
 };
 
 }  // namespace rsz