Merge pull request The-OpenROAD-Project#5134 from bingyuew/cad_contest

Adding Timing::equivCells and Timing::makeEquivCells python APIs
andyfox-rushc · May 18, 2024 · dac8235 · dac8235
2 parents a6a172b + 2f7865f
commit dac8235
Show file tree

Hide file tree

Showing 7 changed files with 1,950 additions and 2 deletions.
diff --git a/include/ord/Timing.h b/include/ord/Timing.h
@@ -109,6 +109,9 @@ class Timing
   sta::Corner* cmdCorner();
   sta::Corner* findCorner(const char* name);
 
+  void makeEquivCells();
+  std::vector<odb::dbMaster*> equivCells(odb::dbMaster* master);
+
  private:
   sta::dbSta* getSta();
   sta::MinMax* getMinMax(MinMax type);

diff --git a/src/OpenRoad-py.i b/src/OpenRoad-py.i
@@ -80,6 +80,7 @@ get_db_block();
 
 %template(Corners) std::vector<sta::Corner*>;
 %template(MTerms) std::vector<odb::dbMTerm*>;
+%template(Masters) std::vector<odb::dbMaster*>;
 
 %include "Exception-py.i"
 %include "ord/Tech.h"

diff --git a/src/Timing.cc b/src/Timing.cc
@@ -42,8 +42,9 @@
 #include "odb/db.h"
 #include "ord/OpenRoad.hh"
 #include "sta/Search.hh"
-//#include "ord/Tech.h"
+// #include "ord/Tech.h"
 #include "ord/Design.h"
+#include "rsz/Resizer.hh"
 #include "sta/Corner.hh"
 #include "sta/Liberty.hh"
 #include "sta/TimingArc.hh"
@@ -385,4 +386,31 @@ float Timing::dynamicPower(odb::dbInst* inst, sta::Corner* corner)
   return (power.internal() + power.switching());
 }
 
+void Timing::makeEquivCells()
+{
+  auto app = OpenRoad::openRoad();
+  rsz::Resizer* resizer = app->getResizer();
+  resizer->makeEquivCells();
+}
+
+std::vector<odb::dbMaster*> Timing::equivCells(odb::dbMaster* master)
+{
+  sta::dbSta* sta = getSta();
+  sta::dbNetwork* network = sta->getDbNetwork();
+  sta::Cell* cell = network->dbToSta(master);
+  std::vector<odb::dbMaster*> masterSeq;
+  if (cell) {
+    sta::LibertyCell* libcell = network->libertyCell(cell);
+    sta::LibertyCellSeq* equiv_cells = sta->equivCells(libcell);
+    if (equiv_cells) {
+      for (sta::LibertyCell* equiv_cell : *equiv_cells) {
+        odb::dbMaster* equiv_master = network->staToDb(equiv_cell);
+        masterSeq.emplace_back(equiv_master);
+      }
+    } else {
+      masterSeq.emplace_back(master);
+    }
+  }
+  return masterSeq;
+}
 }  // namespace ord
diff --git a/src/rsz/include/rsz/Resizer.hh b/src/rsz/include/rsz/Resizer.hh
@@ -380,6 +380,7 @@ class Resizer : public dbStaState
   dbBlock* getDbBlock() { return block_; };
   double dbuToMeters(int dist) const;
   int metersToDbu(double dist) const;
+  void makeEquivCells();
 
  protected:
   void init();
@@ -390,7 +391,6 @@ class Resizer : public dbStaState
   void bufferOutput(const Pin* top_pin, LibertyCell* buffer_cell);
   bool hasTristateOrDontTouchDriver(const Net* net);
   bool isTristateDriver(const Pin* pin);
-  void makeEquivCells();
   void checkLibertyForAllCorners();
   void findBuffers();
   bool isLinkCell(LibertyCell* cell);

diff --git a/test/regression_tests.tcl b/test/regression_tests.tcl
@@ -4,6 +4,7 @@ record_tests {
   timing_api
   timing_api_2
   timing_api_3
+  timing_api_4
   upf_test
   upf_aes
   two_designs