grt,psm: fix stale heatmap pointers on database reload

Override setChip in RoutingCongestionDataSource to reset the cached
layer pointer. Override setChip in RUDYDataSource to reset the cached
rudy pointer, unregister from the old block, and re-register on the
new block. Add GlobalRouter::clearRudy() to delete the cached Rudy
so it is recreated for the new block. Reset layer_, net_, and corner_
in IRDropDataSource::setChip to prevent dangling pointer access.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Signed-off-by: Eder Monteiro <emrmonteiro@precisioninno.com>

Author: eder-matheus

src/grt/include/grt/GlobalRouter.h

@@ -332,6 +332,7 @@ class GlobalRouter
   odb::dbDatabase* db() const { return db_; }
   FastRouteCore* fastroute() const { return fastroute_; }
   Rudy* getRudy();
+  void clearRudy();
 
   void writePinLocations(const char* file_name);
 

src/grt/src/GlobalRouter.cpp

@@ -1138,6 +1138,12 @@ Rudy* GlobalRouter::getRudy()
   return rudy_;
 }
 
+void GlobalRouter::clearRudy()
+{
+  delete rudy_;
+  rudy_ = nullptr;
+}
+
 bool GlobalRouter::findPinAccessPointPositions(
     const Pin& pin,
     std::map<int, std::vector<PointPair>>& ap_positions,

src/grt/src/heatMap.h

@@ -15,6 +15,12 @@ class RoutingCongestionDataSource : public gui::GlobalRoutingDataSource
  public:
   RoutingCongestionDataSource(utl::Logger* logger, odb::dbDatabase* db);
 
+  void setChip(odb::dbChip* chip) override
+  {
+    layer_ = nullptr;
+    HeatMapDataSource::setChip(chip);
+  }
+
  protected:
   bool populateMap() override;
   void combineMapData(bool base_has_value,

src/grt/src/heatMapRudy.cpp

@@ -156,6 +156,17 @@ bool RUDYDataSource::populateMap()
   return true;
 }
 
+void RUDYDataSource::setChip(odb::dbChip* chip)
+{
+  grouter_->clearRudy();
+  rudy_ = nullptr;
+  removeOwner();
+  HeatMapDataSource::setChip(chip);
+  if (chip && chip->getBlock()) {
+    addOwner(chip->getBlock());
+  }
+}
+
 void RUDYDataSource::onShow()
 {
   HeatMapDataSource::onShow();

src/grt/src/heatMapRudy.h

@@ -23,6 +23,7 @@ class RUDYDataSource : public gui::GlobalRoutingDataSource,
                  grt::GlobalRouter* grouter,
                  odb::dbDatabase* db);
 
+  void setChip(odb::dbChip* chip) override;
   void onShow() override;
   void onHide() override;
 

src/psm/src/heatMap.cpp

@@ -96,6 +96,9 @@ IRDropDataSource::IRDropDataSource(PDNSim* psm,
 
 void IRDropDataSource::setChip(odb::dbChip* chip)
 {
+  layer_ = nullptr;
+  net_ = nullptr;
+  corner_ = nullptr;
   gui::HeatMapDataSource::setChip(chip);
   if (chip != nullptr) {
     odb::dbBlock* block = chip->getBlock();