Fix headless GUI heatmap dumps

Signed-off-by: Naveen Venkat <archgen.guest@nyayanidhi.in>

Author: Naveen Venkat

src/gui/include/gui/gui.h

@@ -1045,7 +1045,7 @@ class Gui
 
   void registerHeatMap(HeatMapDataSource* heatmap);
   void unregisterHeatMap(HeatMapDataSource* heatmap);
-  const std::set<HeatMapDataSource*>& getHeatMaps() { return heat_maps_; }
+  const std::set<HeatMapDataSource*>& getHeatMaps();
   HeatMapDataSource* getHeatMap(const std::string& name);
 
   // returns the Gui singleton
@@ -1082,6 +1082,7 @@ class Gui
 
  private:
   Gui();
+  void syncHeatMapChips();
 
   void registerDescriptor(const std::type_info& type,
                           const Descriptor* descriptor);

src/gui/src/gui.cpp

@@ -30,6 +30,7 @@
 #endif
 #include <cmath>
 #include <optional>
+#include <set>
 #include <stdexcept>
 #include <string>
 #include <vector>
@@ -1171,8 +1172,32 @@ void Gui::unregisterHeatMap(HeatMapDataSource* heatmap)
   heat_maps_.erase(heatmap);
 }
 
+void Gui::syncHeatMapChips()
+{
+  if (hasUI() || db_ == nullptr) {
+    return;
+  }
+
+  // Console and headless sessions do not receive MainWindow::setBlock().
+  auto* chip = db_->getChip();
+  for (auto* heat_map : heat_maps_) {
+    if (heat_map->getChip() != chip) {
+      heat_map->setChip(chip);
+      heat_map->destroyMap();
+    }
+  }
+}
+
+const std::set<HeatMapDataSource*>& Gui::getHeatMaps()
+{
+  syncHeatMapChips();
+  return heat_maps_;
+}
+
 HeatMapDataSource* Gui::getHeatMap(const std::string& name)
 {
+  syncHeatMapChips();
+
   HeatMapDataSource* source = nullptr;
 
   for (auto* heat_map : heat_maps_) {

src/gui/test/CMakeLists.txt

@@ -1,6 +1,22 @@
-or_integration_tests(
-  "gui"
-  TESTS
-    supported
-)
+set(GUI_TESTS supported)
+set(GUI_PASSFAIL_TESTS)
 
+if (Qt5_FOUND AND BUILD_GUI)
+  list(APPEND GUI_PASSFAIL_TESTS dump_heatmap_headless)
+endif()
+
+if (GUI_PASSFAIL_TESTS)
+  or_integration_tests(
+    "gui"
+    TESTS
+      ${GUI_TESTS}
+    PASSFAIL_TESTS
+      ${GUI_PASSFAIL_TESTS}
+  )
+else()
+  or_integration_tests(
+    "gui"
+    TESTS
+      ${GUI_TESTS}
+  )
+endif()

src/gui/test/dump_heatmap_headless.tcl

@@ -0,0 +1,35 @@
+set script_dir [file dirname [info script]]
+set openroad_test_dir [file normalize [file join $script_dir .. .. .. test]]
+
+source [file join $openroad_test_dir helpers.tcl]
+
+if { ![gui::supported] } {
+  puts "fail"
+  exit 1
+}
+
+suppress_message ODB 128
+suppress_message ODB 130
+suppress_message ODB 131
+suppress_message ODB 132
+suppress_message ODB 133
+suppress_message ODB 227
+
+read_lef [file join $openroad_test_dir Nangate45 Nangate45.lef]
+read_def [file join $openroad_test_dir gcd_nangate45.def]
+
+set heatmap_file [make_result_file dump_heatmap_headless.csv]
+gui::dump_heatmap Placement $heatmap_file
+
+set heatmap [open $heatmap_file r]
+set contents [read $heatmap]
+close $heatmap
+
+if {
+  [string first "value (%)" $contents] >= 0
+  && [regexp -line {^[0-9.-]+,[0-9.-]+,[0-9.-]+,[0-9.-]+,} $contents]
+} {
+  puts "pass"
+} else {
+  puts "fail"
+}