memory.py revision 116680a4aac90f2aa7413d9095a592090648e557 
1# Copyright 2012 The Chromium Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5from metrics import memory
6from metrics import power
7from telemetry.page import page_measurement
8
9class Memory(page_measurement.PageMeasurement):
10  def __init__(self):
11    super(Memory, self).__init__('RunStressMemory')
12    self._memory_metric = None
13    self._power_metric = None
14
15  def WillStartBrowser(self, browser):
16    self._power_metric = power.PowerMetric(browser)
17
18  def DidStartBrowser(self, browser):
19    self._memory_metric = memory.MemoryMetric(browser)
20
21  def DidNavigateToPage(self, page, tab):
22    self._memory_metric.Start(page, tab)
23    self._power_metric.Start(page, tab)
24
25  def CustomizeBrowserOptions(self, options):
26    memory.MemoryMetric.CustomizeBrowserOptions(options)
27    # Since this is a memory benchmark, we want to sample memory histograms at
28    # a high frequency.
29    options.AppendExtraBrowserArgs('--memory-metrics')
30
31  def MeasurePage(self, page, tab, results):
32    self._power_metric.Stop(page, tab)
33    self._memory_metric.Stop(page, tab)
34    self._memory_metric.AddResults(tab, results)
35    self._power_metric.AddResults(tab, results)
36
37    if tab.browser.is_profiler_active('tcmalloc-heap'):
38      # The tcmalloc_heap_profiler dumps files at regular
39      # intervals (~20 secs).
40      # This is a minor optimization to ensure it'll dump the last file when
41      # the test completes.
42      tab.ExecuteJavaScript("""
43        if (chrome && chrome.memoryBenchmarking) {
44          chrome.memoryBenchmarking.heapProfilerDump('renderer', 'final');
45          chrome.memoryBenchmarking.heapProfilerDump('browser', 'final');
46        }
47      """)
48