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
6import imp
7import os
8import sys
9import urllib2
10
11
12BASE_URL = 'http://src.chromium.org/chrome/trunk/'
13DEPS_FILE = 'bootstrap_deps'
14
15SCRIPT_PATH = os.path.dirname(os.path.abspath(__file__))
16# Directory containing src/ in a Chromium checkout.
17CHECKOUT_BASE_PATH = os.path.join(SCRIPT_PATH, os.pardir, os.pardir, os.pardir)
18# Directory in which to save bootstrap files.
19BOOTSTRAP_BASE_PATH = os.path.join(SCRIPT_PATH, 'support', 'bootstrap_files')
20
21PERF_DIR = os.path.join('src', 'tools', 'perf')
22TELEMETRY_DIR = os.path.join('src', 'tools', 'telemetry')
23TELEMETRY_TOOLS_DIR = os.path.join('src', 'tools', 'telemetry_tools')
24
25
26def _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
55def _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
64def 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
73def 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
127if __name__ == '__main__':
128  sys.exit(Main())
129