| #!/usr/bin/env python |
| # |
| # Copyright 2015 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import logging |
| import optparse |
| import os |
| import sys |
| import webbrowser |
| |
| from profile_chrome import chrome_startup_controller |
| from profile_chrome import controllers |
| from profile_chrome import flags |
| from profile_chrome import profiler |
| from profile_chrome import systrace_controller |
| from profile_chrome import ui |
| |
| _SRC_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) |
| |
| sys.path.append(os.path.join(_SRC_DIR, 'third_party', 'catapult', 'devil')) |
| from devil.android import device_utils |
| |
| |
| def _CreateOptionParser(): |
| parser = optparse.OptionParser(description='Record about://tracing profiles ' |
| 'from Android browsers startup, combined with ' |
| 'Android systrace. See http://dev.chromium.org' |
| '/developers/how-tos/trace-event-profiling-' |
| 'tool for detailed instructions for ' |
| 'profiling.') |
| parser.add_option('--url', help='URL to visit on startup. Default: ' |
| 'https://www.google.com. An empty URL launches Chrome with' |
| ' a MAIN action instead of VIEW.', |
| default='https://www.google.com', metavar='URL') |
| parser.add_option('--cold', help='Flush the OS page cache before starting the' |
| ' browser. Note that this require a device with root ' |
| 'access.', default=False, action='store_true') |
| parser.add_option_group(flags.SystraceOptions(parser)) |
| parser.add_option_group(flags.OutputOptions(parser)) |
| |
| browsers = sorted(profiler.GetSupportedBrowsers().keys()) |
| parser.add_option('-b', '--browser', help='Select among installed browsers. ' |
| 'One of ' + ', '.join(browsers) + ', "stable" is used by ' |
| 'default.', type='choice', choices=browsers, |
| default='stable') |
| parser.add_option('-v', '--verbose', help='Verbose logging.', |
| action='store_true') |
| parser.add_option('-z', '--compress', help='Compress the resulting trace ' |
| 'with gzip. ', action='store_true') |
| parser.add_option('-t', '--time', help='Stops tracing after N seconds, 0 to ' |
| 'manually stop (startup trace ends after at most 5s).', |
| default=5, metavar='N', type='int') |
| return parser |
| |
| |
| def main(): |
| parser = _CreateOptionParser() |
| options, _ = parser.parse_args() |
| |
| if options.verbose: |
| logging.getLogger().setLevel(logging.DEBUG) |
| |
| devices = device_utils.DeviceUtils.HealthyDevices() |
| if len(devices) != 1: |
| logging.error('Exactly 1 device must be attached.') |
| return 1 |
| device = devices[0] |
| package_info = profiler.GetSupportedBrowsers()[options.browser] |
| |
| if options.systrace_categories in ['list', 'help']: |
| ui.PrintMessage('\n'.join( |
| systrace_controller.SystraceController.GetCategories(device))) |
| return 0 |
| systrace_categories = (options.systrace_categories.split(',') |
| if options.systrace_categories else []) |
| enabled_controllers = [] |
| # Enable the systrace and chrome controller. The systrace controller should go |
| # first because otherwise the resulting traces miss early systrace data. |
| if systrace_categories: |
| enabled_controllers.append(systrace_controller.SystraceController( |
| device, systrace_categories, False)) |
| enabled_controllers.append( |
| chrome_startup_controller.ChromeStartupTracingController( |
| device, package_info, options.cold, options.url)) |
| if options.output: |
| options.output = os.path.expanduser(options.output) |
| result = profiler.CaptureProfile(enabled_controllers, |
| options.time, |
| output=options.output, |
| compress=options.compress, |
| write_json=options.json) |
| if options.view: |
| if sys.platform == 'darwin': |
| os.system('/usr/bin/open %s' % os.path.abspath(result)) |
| else: |
| webbrowser.open(result) |
| |
| |
| if __name__ == '__main__': |
| sys.exit(main()) |