Apply other policies of Deep Memory Profiler in Chrome Endure.

chrome/test/functional/perf_endure.py

 #!/usr/bin/env python
 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """Performance tests for Chrome Endure (long-running perf tests on Chrome).
 
 This module accepts the following environment variable inputs:
   TEST_LENGTH: The number of seconds in which to run each test.
   PERF_STATS_INTERVAL: The number of seconds to wait in-between each sampling
@@ skipping 36 matching lines @@
 
 class DeepMemoryProfiler(object):
   """Controls Deep Memory Profiler (dmprof) for endurance tests."""
   DEEP_MEMORY_PROFILE = False
   DEEP_MEMORY_PROFILE_SAVE = False
 
   _DMPROF_DIR_PATH = os.path.join(
       os.path.dirname(__file__), os.pardir, os.pardir, os.pardir,
       'tools', 'deep_memory_profiler')
   _DMPROF_SCRIPT_PATH = os.path.join(_DMPROF_DIR_PATH, 'dmprof')
+  _POLICIES = ['l0', 'l1', 'l2', 't0']
 
   def __init__(self):
     self._enabled = self.GetEnvironmentVariable(
         'DEEP_MEMORY_PROFILE', bool, self.DEEP_MEMORY_PROFILE)
     self._save = self.GetEnvironmentVariable(
         'DEEP_MEMORY_PROFILE_SAVE', bool, self.DEEP_MEMORY_PROFILE_SAVE)
     self._json_file = None
     self._last_json_filename = ''
     self._proc = None
-    self._last_time = -1.0
+    self._last_time = {}
+    for policy in self._POLICIES:
+      self._last_time[policy] = -1.0
 
   def __nonzero__(self):
     return self._enabled
 
   @staticmethod
   def GetEnvironmentVariable(env_name, converter, default):
     """Returns a converted environment variable for Deep Memory Profiler.
 
     Args:
       env_name: A string name of an environment variable.
@@ skipping 95 matching lines @@
       else:
         logging.info('    No dump files.')
 
   def ParseResultAndOutputPerfGraphValues(
       self, webapp_name, test_description, output_perf_graph_value):
     """Parses Deep Memory Profiler result, and outputs perf graph values."""
     if not self._enabled:
       return
 
     results = {}
-    if self._last_json_filename:
-      json_data = {}
-      with open(self._last_json_filename) as json_f:
-        json_data = json.load(json_f)
-      if json_data['version'] == 'JSON_DEEP_1':
-        results = json_data['snapshots']
-      elif json_data['version'] == 'JSON_DEEP_2':
-        results = json_data['policies']['l2']['snapshots']
-    if results and results[-1]['second'] > self._last_time:
-      started = False
-      for legend in json_data['policies']['l2']['legends']:
-        if legend == 'FROM_HERE_FOR_TOTAL':
-          started = True
-        elif legend == 'UNTIL_HERE_FOR_TOTAL':
-          break
-        elif started:
-          output_perf_graph_value(
-              legend.encode('utf-8'), [
-                  (int(round(snapshot['second'])), snapshot[legend] / 1024)
-                  for snapshot in results
-                  if snapshot['second'] > self._last_time],
-              'KB',
-              graph_name='%s%s-DMP' % (webapp_name, test_description),
-              units_x='seconds', is_stacked=True)
-      self._last_time = results[-1]['second']
+    for policy in self._POLICIES:
+      if self._last_json_filename:
+        json_data = {}
+        with open(self._last_json_filename) as json_f:
+          json_data = json.load(json_f)
+        if json_data['version'] == 'JSON_DEEP_1':
+          results[policy] = json_data['snapshots']
+        elif json_data['version'] == 'JSON_DEEP_2':
+          results[policy] = json_data['policies'][policy]['snapshots']
+    for policy, result in results.iteritems():
+      if result and result[-1]['second'] > self._last_time[policy]:
+        started = False
+        for legend in json_data['policies'][policy]['legends']:
+          if legend == 'FROM_HERE_FOR_TOTAL':
+            started = True
+          elif legend == 'UNTIL_HERE_FOR_TOTAL':
+            break
+          elif started:
+            output_perf_graph_value(
+                legend.encode('utf-8'), [
+                    (int(round(snapshot['second'])), snapshot[legend] / 1024)
+                    for snapshot in result
+                    if snapshot['second'] > self._last_time[policy]],
+                'KB',
+                graph_name='%s%s-%s-DMP' % (
+                    webapp_name, test_description, policy),
+                units_x='seconds', is_stacked=True)
+        self._last_time[policy] = result[-1]['second']
 
   def _WaitForDeepMemoryProfiler(self):
     """Waits for the Deep Memory Profiler to finish if running."""
     if not self._enabled or not self._proc:
       return
 
     self._proc.wait()
     self._proc = None
     if self._json_file:
       self._last_json_filename = self._json_file.name
@@ skipping 1026 matching lines @@
     if not os.path.exists(scripts):
       raise IOError('Injected scripts %s not found.' % scripts)
     replay_options = ['--inject_scripts', scripts]
     if 'WPR_RECORD' in os.environ:
       replay_options.append('--append')
     return webpagereplay.ReplayServer(archive_path, replay_options)
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()