src/tools/perf/run_measurement

   1 #!/usr/bin/env python
   2 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 import imp
   7 import os
   8 import sys
   9 import urllib2
  10
  11
  12 BASE_URL = 'http://src.chromium.org/chrome/trunk/'
  13 DEPS_FILE = 'bootstrap_deps'
  14
  15 SCRIPT_PATH = os.path.dirname(os.path.realpath(__file__))
  16 # Directory containing src/ in a Chromium checkout.
  17 CHECKOUT_BASE_PATH = os.path.join(SCRIPT_PATH, os.pardir, os.pardir, os.pardir)
  18 # Directory in which to save bootstrap files.
  19 BOOTSTRAP_BASE_PATH = os.path.join(SCRIPT_PATH, 'support', 'bootstrap_files')
  20
  21 PERF_DIR = os.path.join('src', 'tools', 'perf')
  22 TELEMETRY_DIR = os.path.join('src', 'tools', 'telemetry')
  23 TELEMETRY_TOOLS_DIR = os.path.join('src', 'tools', 'telemetry_tools')
  24
  25
  26 def _GetBasePath():
  27   """Find the location of our Chromium or bootstrap checkout.
  28
  29   It tries to import Telemetry.  If the import succeeds,
  30   we assume that's the correct location.
  31
  32   Returns:
  33     The path containing the src/ directory, or None if no checkout is found.
  34   """
  35   module_name = 'telemetry'
  36   module_path = TELEMETRY_DIR
  37
  38   try:
  39     paths = [os.path.join(CHECKOUT_BASE_PATH, module_path)]
  40     imp.find_module(module_name, paths)
  41     return CHECKOUT_BASE_PATH
  42   except ImportError:
  43     pass
  44
  45   try:
  46     paths = [os.path.join(BOOTSTRAP_BASE_PATH, module_path)]
  47     imp.find_module(module_name, paths)
  48     return BOOTSTRAP_BASE_PATH
  49   except ImportError:
  50     pass
  51
  52   return None
  53
  54
  55 def _Bootstrap(bootstrap_deps_url):
  56   """Grab all the deps listed in the file at bootstrap_deps_url."""
  57   bootstrap_txt = urllib2.urlopen(
  58       BASE_URL + 'src/tools/telemetry_tools/telemetry_bootstrap.py').read()
  59   bootstrap = imp.new_module('bootstrap')
  60   exec bootstrap_txt in bootstrap.__dict__
  61   bootstrap.DownloadDeps(BOOTSTRAP_BASE_PATH, bootstrap_deps_url)
  62
  63
  64 def ListBootstrapDeps(base_path):
  65   """List the deps required for telemetry."""
  66   sys.path.append(os.path.join(base_path, TELEMETRY_TOOLS_DIR))
  67   import telemetry_bootstrap
  68
  69   deps_file = os.path.join(base_path, PERF_DIR, DEPS_FILE)
  70   return telemetry_bootstrap.ListAllDepsPaths(deps_file)
  71
  72
  73 def Main():
  74   if not _GetBasePath():
  75     _Bootstrap(BASE_URL + 'src/tools/perf/' + DEPS_FILE)
  76
  77   new_base_path = _GetBasePath()
  78   new_perf_path = os.path.join(new_base_path, PERF_DIR)
  79   new_telemetry_path = os.path.join(new_base_path, TELEMETRY_DIR)
  80
  81   if '--print-bootstrap-deps' in sys.argv:
  82     print ListBootstrapDeps(new_base_path)
  83     return 0
  84
  85   sys.path.append(new_perf_path)
  86   import page_sets
  87   page_set_filenames = page_sets.GetAllPageSetFilenames()
  88
  89   old_benchmark_names = {
  90     "image_decoding_benchmark": "image_decoding",
  91     "image_decoding_measurement": "image_decoding",
  92     "loading_benchmark": "loading",
  93     "loading_measurement": "loading",
  94     "media_measurement": "media",
  95     "memory_benchmark": "memory",
  96     "memory_measurement": "memory",
  97     "rasterize_and_record_benchmark": "rasterize_and_record",
  98     "rasterize_and_record_measurement": "rasterize_and_record",
  99     "robohornetpro": "robohornet_pro",
 100     "scrolling_benchmark": "smoothness",
 101     "smoothness_benchmark": "smoothness",
 102     "smoothness_measurement": "smoothness",
 103     "startup_benchmark": "startup_warm_blank_page",
 104     "startup_measurement": "startup",
 105     "tab_switching_measurement": "tab_switching",
 106   }
 107
 108   sys.path.append(new_telemetry_path)
 109   from telemetry.page import page_measurement_runner
 110   # There are bots that are hard-coded to run some specific named tests.
 111   # Convert these to the current naming conventions by overriding them
 112   # in the runner.
 113   class MeasurementRunner(page_measurement_runner.PageMeasurementRunner):
 114     def GetModernizedTestName(self, arg):
 115       if arg not in old_benchmark_names:
 116         return arg
 117       sys.stderr.write(
 118         'An old name %s was given. Please use %s in the future.\n' % (
 119           arg,
 120           old_benchmark_names.get(arg)))
 121       return old_benchmark_names[arg]
 122
 123   runner = MeasurementRunner()
 124   return runner.Run(new_perf_path, page_set_filenames)
 125
 126
 127 if __name__ == '__main__':
 128   sys.exit(Main())