Add energy monitoring and characterization scripts

1 files changed, 7 insertions, 0 deletions

diff --git a/components/profile/heartbeats.rs b/components/profile/heartbeats.rs
index e782067f8a6..6eaff00e79f 100644
--- a/components/profile/heartbeats.rs
+++ b/components/profile/heartbeats.rs
@@ -51,6 +51,7 @@ pub fn init() {
     maybe_create_heartbeat(&mut hbs, ProfilerCategory::ScriptWebSocketEvent);
     maybe_create_heartbeat(&mut hbs, ProfilerCategory::ScriptWorkerEvent);
     maybe_create_heartbeat(&mut hbs, ProfilerCategory::ScriptXhrEvent);
+    maybe_create_heartbeat(&mut hbs, ProfilerCategory::ApplicationHeartbeat);
     unsafe {
         HBS = Some(mem::transmute(Box::new(hbs)));
     }
@@ -71,6 +72,12 @@ pub fn cleanup() {
     }
 }
 
+pub fn is_heartbeat_enabled(category: &ProfilerCategory) -> bool {
+    unsafe {
+        HBS.map_or(false, |m| (*m).contains_key(category))
+    }
+}
+
 /// Issue a heartbeat (if one exists) for the given category
 pub fn maybe_heartbeat(category: &ProfilerCategory,
                        start_time: u64,