fix crash on balance levels graph building

Signed-off-by: arthurjolo <arthurjl@precisioninno.com>

Author: arthurjolo

src/cts/include/cts/TritonCTS.h

@@ -221,6 +221,7 @@ class TritonCTS
   std::set<odb::dbNet*> visitedClockNets_;
   std::map<odb::dbInst*, ClockInst*> inst2clkbuf_;
   std::map<ClockInst*, ClockSubNet*> driver2subnet_;
+  std::map<odb::dbNet*, TreeBuilder*> net2builder_;
 
   // db vars
   odb::dbDatabase* db_ = nullptr;

src/cts/src/LatencyBalancer.cpp

@@ -158,6 +158,9 @@ void LatencyBalancer::buildGraph(odb::dbNet* clkInputNet)
 
     for (odb::dbITerm* sinkIterm : driverNet->getITerms()) {
       if (sinkIterm->getIoType() == odb::dbIoType::INPUT) {
+        if (!propagateClock(sinkIterm)) {
+          continue;
+        }
         int sinkId = graph_.size();
         odb::dbInst* sinkInst = sinkIterm->getInst();
         std::string sinkName = sinkInst->getName();
@@ -197,7 +200,7 @@ void LatencyBalancer::buildGraph(odb::dbNet* clkInputNet)
                 }
               }
             }
-
+            worseDelay_ = std::max(worseDelay_, (arrival + insDelay));
             graph_[sinkId].arrival = arrival + insDelay;
             debugPrint(logger_,
                        CTS,
@@ -288,7 +291,10 @@ float LatencyBalancer::computeAveSinkArrivals(TreeBuilder* builder)
     odb::dbITerm* iterm = sink.getDbInputPin();
     computeSinkArrivalRecur(topInputClockNet, iterm, sumArrivals, numSinks);
   });
-  float aveArrival = sumArrivals / (float) numSinks;
+  float aveArrival = 0.0;
+  if(numSinks) {
+    aveArrival = sumArrivals / (float) numSinks;
+  }
   builder->setAveSinkArrival(aveArrival);
   debugPrint(logger_,
              CTS,

src/cts/src/TritonCTS.cpp

@@ -302,6 +302,9 @@ void TritonCTS::initOneClockTree(odb::dbNet* driverNet,
   } else {
     clockBuilder = initClock(driverNet, clkInputNet, sdcClockName, parent);
   }
+  if(clockBuilder != nullptr && net2builder_[clkInputNet] == nullptr) {
+    net2builder_[clkInputNet] = clockBuilder;
+  }
   // Treat gated clocks as separate clock trees
   // TODO: include sinks from gated clocks together with other sinks and build
   // one clock tree
@@ -317,8 +320,13 @@ void TritonCTS::initOneClockTree(odb::dbNet* driverNet,
           if (visitedClockNets_.find(outputNet) == visitedClockNets_.end()
               && !openSta_->sdc()->isLeafPinClock(
                   network_->dbToSta(outputPin))) {
-            initOneClockTree(
-                outputNet, clkInputNet, sdcClockName, clockBuilder);
+            if(clockBuilder == nullptr && net2builder_[clkInputNet] != nullptr) {
+              initOneClockTree(
+                outputNet, clkInputNet, sdcClockName, net2builder_[clkInputNet]);
+            } else {
+              initOneClockTree(
+                  outputNet, clkInputNet, sdcClockName, clockBuilder);
+            }
           }
         }
       }