gh-140677 Add heatmap visualization to Tachyon sampling profiler (#140680)

Co-authored-by: Ivona Stojanovic <stojanovic.i@hotmail.com>

Author: pablogsal

Lib/profiling/sampling/__init__.py

@@ -7,7 +7,8 @@
 from .collector import Collector
 from .pstats_collector import PstatsCollector
 from .stack_collector import CollapsedStackCollector
+from .heatmap_collector import HeatmapCollector
 from .gecko_collector import GeckoCollector
 from .string_table import StringTable
 
-__all__ = ("Collector", "PstatsCollector", "CollapsedStackCollector", "GeckoCollector", "StringTable")
+__all__ = ("Collector", "PstatsCollector", "CollapsedStackCollector", "HeatmapCollector", "GeckoCollector", "StringTable")

Lib/profiling/sampling/_css_utils.py

@@ -0,0 +1,22 @@
+import importlib.resources
+
+
+def get_combined_css(component: str) -> str:
+    template_dir = importlib.resources.files(__package__)
+
+    base_css = (template_dir / "_shared_assets" / "base.css").read_text(encoding="utf-8")
+
+    if component == "flamegraph":
+        component_css = (
+            template_dir / "_flamegraph_assets" / "flamegraph.css"
+        ).read_text(encoding="utf-8")
+    elif component == "heatmap":
+        component_css = (template_dir / "_heatmap_assets" / "heatmap.css").read_text(
+            encoding="utf-8"
+        )
+    else:
+        raise ValueError(
+            f"Unknown component: {component}. Expected 'flamegraph' or 'heatmap'."
+        )
+
+    return f"{base_css}\n\n{component_css}"