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__), os.pardir, 'telemetry')) | 10 sys.path.append(os.path.join(os.path.dirname(__file__), os.pardir, 'telemetry')) |
11 | 11 |
12 from telemetry.core import browser_finder | 12 from telemetry.core import browser_finder |
13 from telemetry.core import browser_options | 13 from telemetry.core import browser_options |
14 | 14 |
15 | 15 |
16 def _RunPrebuilt(options): | 16 def _RunPrebuilt(options): |
17 browser_to_create = browser_finder.FindBrowser(options) | 17 browser_to_create = browser_finder.FindBrowser(options) |
18 with browser_to_create.Create(options) as browser: | 18 with browser_to_create.Create(options) as browser: |
19 output_file = os.path.join(tempfile.mkdtemp(), options.profiler) | 19 output_file = os.path.join(tempfile.mkdtemp(), options.profiler) |
20 raw_input('Press enter to start profiling...') | 20 raw_input('Press enter to start profiling...') |
21 print '>> Starting profiler', options.profiler | 21 print '>> Starting profiler', options.profiler |
22 browser.platform.profiling_controller.Start( | 22 browser.profiling_controller.Start(options.profiler, output_file) |
23 options.profiler, output_file) | |
24 print 'Press enter or CTRL+C to stop' | |
25 try: | 23 try: |
26 raw_input() | 24 raw_input('Press enter or CTRL+C to stop') |
27 except KeyboardInterrupt: | 25 except KeyboardInterrupt: |
28 pass | 26 pass |
29 finally: | 27 finally: |
30 browser.platform.profiling_controller.Stop() | 28 print '<< Stopping ...', |
31 print '<< Stopped profiler ', options.profiler | 29 sys.stdout.flush() |
| 30 browser.profiling_controller.Stop() |
| 31 print 'Stopped profiler ', options.profiler |
32 | 32 |
33 | 33 |
34 if __name__ == '__main__': | 34 if __name__ == '__main__': |
35 browser_finder_options = browser_options.BrowserFinderOptions() | 35 browser_finder_options = browser_options.BrowserFinderOptions() |
36 parser = browser_finder_options.CreateParser('') | 36 parser = browser_finder_options.CreateParser('') |
37 profiler_options, _ = parser.parse_args() | 37 profiler_options, _ = parser.parse_args() |
38 sys.exit(_RunPrebuilt(profiler_options)) | 38 sys.exit(_RunPrebuilt(profiler_options)) |
OLD | NEW |