OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright 2013 The Chromium Authors. All rights reserved. | 2 # Copyright 2013 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 import os | 6 import os |
7 import sys | 7 import sys |
8 import tempfile | 8 import tempfile |
9 | 9 |
10 sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'telemetry')) | 10 from chrome_telemetry_build import chromium_config |
| 11 sys.path.insert(1, chromium_config.GetTelemetryDir()) |
11 | 12 |
12 from telemetry.internal.browser import browser_finder | 13 from telemetry.internal.browser import browser_finder |
13 from telemetry.internal.browser import browser_options | 14 from telemetry.internal.browser import browser_options |
14 | 15 |
15 | 16 |
16 def _RunPrebuilt(options): | 17 def _RunPrebuilt(options): |
17 browser_to_create = browser_finder.FindBrowser(options) | 18 browser_to_create = browser_finder.FindBrowser(options) |
18 with browser_to_create.Create(options) as browser: | 19 with browser_to_create.Create(options) as browser: |
19 output_file = os.path.join(tempfile.mkdtemp(), options.profiler) | 20 output_file = os.path.join(tempfile.mkdtemp(), options.profiler) |
20 raw_input('Press enter to start profiling...') | 21 raw_input('Press enter to start profiling...') |
21 print '>> Starting profiler', options.profiler | 22 print '>> Starting profiler', options.profiler |
22 browser.profiling_controller.Start(options.profiler, output_file) | 23 browser.profiling_controller.Start(options.profiler, output_file) |
23 try: | 24 try: |
24 raw_input('Press enter or CTRL+C to stop') | 25 raw_input('Press enter or CTRL+C to stop') |
25 except KeyboardInterrupt: | 26 except KeyboardInterrupt: |
26 pass | 27 pass |
27 finally: | 28 finally: |
28 print '<< Stopping ...', | 29 print '<< Stopping ...', |
29 sys.stdout.flush() | 30 sys.stdout.flush() |
30 browser.profiling_controller.Stop() | 31 browser.profiling_controller.Stop() |
31 print 'Stopped profiler ', options.profiler | 32 print 'Stopped profiler ', options.profiler |
32 | 33 |
33 | 34 |
34 if __name__ == '__main__': | 35 if __name__ == '__main__': |
35 browser_finder_options = browser_options.BrowserFinderOptions() | 36 browser_finder_options = browser_options.BrowserFinderOptions() |
36 parser = browser_finder_options.CreateParser('') | 37 parser = browser_finder_options.CreateParser('') |
37 profiler_options, _ = parser.parse_args() | 38 profiler_options, _ = parser.parse_args() |
38 sys.exit(_RunPrebuilt(profiler_options)) | 39 sys.exit(_RunPrebuilt(profiler_options)) |
OLD | NEW |