| OLD | NEW |
| (Empty) |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 # Use of this source code is governed by a BSD-style license that can be | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 import os | |
| 6 | |
| 7 from devil.android.constants import chrome | |
| 8 | |
| 9 from profile_chrome import trace_packager | |
| 10 from profile_chrome import ui | |
| 11 | |
| 12 | |
| 13 def _StartTracing(controllers, interval): | |
| 14 for controller in controllers: | |
| 15 controller.StartTracing(interval) | |
| 16 | |
| 17 | |
| 18 def _StopTracing(controllers): | |
| 19 for controller in controllers: | |
| 20 controller.StopTracing() | |
| 21 | |
| 22 | |
| 23 def _PullTraces(controllers, output, compress, write_json): | |
| 24 ui.PrintMessage('Downloading...', eol='') | |
| 25 trace_files = [controller.PullTrace() for controller in controllers] | |
| 26 trace_files = [trace for trace in trace_files if trace] | |
| 27 if not trace_files: | |
| 28 ui.PrintMessage('No results') | |
| 29 return [] | |
| 30 result = trace_packager.PackageTraces(trace_files, | |
| 31 output=output, | |
| 32 compress=compress, | |
| 33 write_json=write_json) | |
| 34 ui.PrintMessage('done') | |
| 35 ui.PrintMessage('Trace written to file://%s' % os.path.abspath(result)) | |
| 36 return result | |
| 37 | |
| 38 | |
| 39 def GetSupportedBrowsers(): | |
| 40 """Returns the package names of all supported browsers.""" | |
| 41 # Add aliases for backwards compatibility. | |
| 42 supported_browsers = { | |
| 43 'stable': chrome.PACKAGE_INFO['chrome_stable'], | |
| 44 'beta': chrome.PACKAGE_INFO['chrome_beta'], | |
| 45 'dev': chrome.PACKAGE_INFO['chrome_dev'], | |
| 46 'build': chrome.PACKAGE_INFO['chrome'], | |
| 47 } | |
| 48 supported_browsers.update(chrome.PACKAGE_INFO) | |
| 49 unsupported_browsers = ['content_browsertests', 'gtest', 'legacy_browser'] | |
| 50 for browser in unsupported_browsers: | |
| 51 if browser in supported_browsers: | |
| 52 del supported_browsers[browser] | |
| 53 return supported_browsers | |
| 54 | |
| 55 | |
| 56 def CaptureProfile(controllers, interval, output=None, compress=False, | |
| 57 write_json=False): | |
| 58 """Records a profiling trace saves the result to a file. | |
| 59 | |
| 60 Args: | |
| 61 controllers: List of tracing controllers. | |
| 62 interval: Time interval to capture in seconds. An interval of None (or 0) | |
| 63 continues tracing until stopped by the user. | |
| 64 output: Output file name or None to use an automatically generated name. | |
| 65 compress: If True, the result will be compressed either with gzip or zip | |
| 66 depending on the number of captured subtraces. | |
| 67 write_json: If True, prefer JSON output over HTML. | |
| 68 | |
| 69 Returns: | |
| 70 Path to saved profile. | |
| 71 """ | |
| 72 trace_type = ' + '.join(map(str, controllers)) | |
| 73 try: | |
| 74 _StartTracing(controllers, interval) | |
| 75 if interval: | |
| 76 ui.PrintMessage('Capturing %d-second %s. Press Enter to stop early...' % \ | |
| 77 (interval, trace_type), eol='') | |
| 78 ui.WaitForEnter(interval) | |
| 79 else: | |
| 80 ui.PrintMessage('Capturing %s. Press Enter to stop...' % \ | |
| 81 trace_type, eol='') | |
| 82 raw_input() | |
| 83 finally: | |
| 84 _StopTracing(controllers) | |
| 85 if interval: | |
| 86 ui.PrintMessage('done') | |
| 87 | |
| 88 return _PullTraces(controllers, output, compress, write_json) | |
| OLD | NEW |