Cts enable obstruction aware

src/cts/include/cts/TritonCTS.h

@@ -58,6 +58,8 @@ class Clock;
 class dbNetwork;
 class Unit;
 class LibertyCell;
+class Vertex;
+class Graph;
 }  // namespace sta
 
 namespace stt {
@@ -119,23 +121,27 @@ class TritonCTS
   // db functions
   bool masterExists(const std::string& master) const;
   void populateTritonCTS();
-  void writeClockNetsToDb(Clock& clockNet, std::set<odb::dbNet*>& clkLeafNets);
+  void writeClockNetsToDb(TreeBuilder* builder,
+                          std::set<odb::dbNet*>& clkLeafNets);
   void writeClockNDRsToDb(const std::set<odb::dbNet*>& clkLeafNets);
   void incrementNumClocks() { ++numberOfClocks_; }
   void clearNumClocks() { numberOfClocks_ = 0; }
   unsigned getNumClocks() const { return numberOfClocks_; }
   void initOneClockTree(odb::dbNet* driverNet,
+                        odb::dbNet* clkInputNet,
                         const std::string& sdcClockName,
                         TreeBuilder* parent);
-  TreeBuilder* initClock(odb::dbNet* net,
+  TreeBuilder* initClock(odb::dbNet* firstNet,
+                         odb::dbNet* clkInputNet,
                          const std::string& sdcClock,
                          TreeBuilder* parentBuilder);
   void disconnectAllSinksFromNet(odb::dbNet* net);
   void disconnectAllPinsFromNet(odb::dbNet* net);
   void checkUpstreamConnections(odb::dbNet* net);
   void createClockBuffers(Clock& clockNet, odb::dbModule* parent);
   HTreeBuilder* initClockTreeForMacrosAndRegs(
-      odb::dbNet*& net,
+      odb::dbNet*& firstNet,
+      odb::dbNet* clkInputNet,
       const std::unordered_set<odb::dbMaster*>& buffer_masters,
       Clock& ClockNet,
       TreeBuilder* parentBuilder);
@@ -155,7 +161,8 @@ class TritonCTS
       Clock& clockNet,
       const std::vector<std::pair<odb::dbInst*, odb::dbMTerm*>>& registerSinks,
       odb::dbNet*& firstNet,
-      odb::dbNet*& secondNet);
+      odb::dbNet*& secondNet,
+      std::string& topBufferName);
   void computeITermPosition(odb::dbITerm* term, int& x, int& y) const;
   void countSinksPostDbWrite(TreeBuilder* builder,
                              odb::dbNet* net,
@@ -200,11 +207,16 @@ class TritonCTS
                         ClockInst& dummyClock);
   void printClockNetwork(const Clock& clockNet) const;
   void balanceMacroRegisterLatencies();
-  void computeAveSinkArrivals(TreeBuilder* builder);
+  float getVertexClkArrival(sta::Vertex* sinkVertex, odb::dbNet* topNet, odb::dbITerm* iterm);
+  void computeAveSinkArrivals(TreeBuilder* builder, sta::Graph* graph);
+  void computeSinkArrivalRecur(odb::dbNet* topClokcNet,
+                               odb::dbITerm* iterm,
+                               float& sumArrivals,
+                               unsigned& numSinks,
+                               sta::Graph* graph);
   void adjustLatencies(TreeBuilder* macroBuilder, TreeBuilder* registerBuilder);
   void computeTopBufferDelay(TreeBuilder* builder);
   odb::dbInst* insertDelayBuffer(odb::dbInst* driver,
-                                 int index,
                                  const std::string& clockName,
                                  int locX,
                                  int locY);
@@ -232,6 +244,11 @@ class TritonCTS
   // root buffer and sink bufer candidates
   std::vector<std::string> rootBuffers_;
   std::vector<std::string> sinkBuffers_;
+
+  // register tree root buffer indices
+  unsigned regTreeRootBufIndex_ = 0;
+  // index for delay buffer added for latency adjustment
+  unsigned delayBufIndex_ = 0;
 };
 
 }  // namespace cts

src/cts/src/CtsOptions.h

@@ -299,7 +299,7 @@ class CtsOptions
   std::vector<odb::dbNet*> clockNetsObjs_;
   utl::Logger* logger_ = nullptr;
   stt::SteinerTreeBuilder* sttBuilder_ = nullptr;
-  bool obsAware_ = false;
+  bool obsAware_ = true;
   bool applyNDR_ = false;
   bool insertionDelay_ = true;
   bool bufferListInferred_ = false;

src/cts/src/TreeBuilder.h

@@ -247,6 +247,12 @@ class TreeBuilder
   void setTopBufferDelay(float delay) { topBufferDelay_ = delay; }
   odb::dbInst* getTopBuffer() const { return topBuffer_; }
   void setTopBuffer(odb::dbInst* inst) { topBuffer_ = inst; }
+  std::string getTopBufferName() const { return topBufferName_; }
+  void setTopBufferName(std::string name) { topBufferName_ = name; }
+  odb::dbNet* getTopInputNet() const { return topInputNet_; }
+  void setTopInputNet(odb::dbNet* net) { topInputNet_ = net; }
+  odb::dbNet* getDrivingNet() const { return drivingNet_; }
+  void setDrivingNet(odb::dbNet* net) { drivingNet_ = net; }
 
  protected:
   CtsOptions* options_ = nullptr;
@@ -276,6 +282,9 @@ class TreeBuilder
   float aveArrival_ = 0.0;
   float topBufferDelay_ = 0.0;
   odb::dbInst* topBuffer_ = nullptr;
+  std::string topBufferName_;
+  odb::dbNet* drivingNet_ = nullptr;
+  odb::dbNet* topInputNet_ = nullptr;
 };
 
 }  // namespace cts