diff options
Diffstat (limited to 'systrace/catapult/systrace/profile_chrome/chrome_startup_tracing_agent.py')
-rw-r--r-- | systrace/catapult/systrace/profile_chrome/chrome_startup_tracing_agent.py | 154 |
1 files changed, 0 insertions, 154 deletions
diff --git a/systrace/catapult/systrace/profile_chrome/chrome_startup_tracing_agent.py b/systrace/catapult/systrace/profile_chrome/chrome_startup_tracing_agent.py deleted file mode 100644 index 53be30c..0000000 --- a/systrace/catapult/systrace/profile_chrome/chrome_startup_tracing_agent.py +++ /dev/null @@ -1,154 +0,0 @@ -# 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 re - -import py_utils - -from devil.android import flag_changer -from devil.android.constants import webapk -from devil.android.perf import cache_control -from devil.android.sdk import intent - -from systrace import trace_result -from systrace import tracing_agents - - -class ChromeStartupTracingAgent(tracing_agents.TracingAgent): - def __init__(self, device, package_info, webapk_package, cold, url, - trace_time=None): - tracing_agents.TracingAgent.__init__(self) - self._device = device - self._package_info = package_info - self._webapk_package = webapk_package - self._cold = cold - self._logcat_monitor = self._device.GetLogcatMonitor() - self._url = url - self._trace_time = trace_time - self._trace_file = None - self._trace_finish_re = re.compile(r' Completed startup tracing to (.*)') - self._flag_changer = flag_changer.FlagChanger( - self._device, self._package_info.cmdline_file) - - def __repr__(self): - return 'Browser Startup Trace' - - def _SetupTracing(self): - # TODO(lizeb): Figure out how to clean up the command-line file when - # _TearDownTracing() is not executed in StopTracing(). - flags = ['--trace-startup', '--enable-perfetto'] - if self._trace_time is not None: - flags.append('--trace-startup-duration={}'.format(self._trace_time)) - self._flag_changer.AddFlags(flags) - self._device.ForceStop(self._package_info.package) - if self._webapk_package: - self._device.ForceStop(self._webapk_package) - logging.warning('Forces to stop the WebAPK and the browser provided by ' - '--browser: %s. Please make sure that this browser ' - 'matches the host browser of the WebAPK %s. ', - self._package_info.package, - self._webapk_package) - if self._cold: - self._device.EnableRoot() - cache_control.CacheControl(self._device).DropRamCaches() - launch_intent = None - if self._webapk_package: - launch_intent = intent.Intent( - package=self._webapk_package, - activity=webapk.WEBAPK_MAIN_ACTIVITY, - data=self._url) - elif self._url == '': - launch_intent = intent.Intent( - action='android.intent.action.MAIN', - package=self._package_info.package, - activity=self._package_info.activity) - else: - launch_intent = intent.Intent( - package=self._package_info.package, - activity=self._package_info.activity, - data=self._url, - extras={'create_new_tab': True}) - self._logcat_monitor.Start() - self._device.StartActivity(launch_intent, blocking=True) - - def _TearDownTracing(self): - self._flag_changer.Restore() - - @py_utils.Timeout(tracing_agents.START_STOP_TIMEOUT) - def StartAgentTracing(self, config, timeout=None): - self._SetupTracing() - return True - - @py_utils.Timeout(tracing_agents.START_STOP_TIMEOUT) - def StopAgentTracing(self, timeout=None): - try: - self._trace_file = self._logcat_monitor.WaitFor( - self._trace_finish_re).group(1) - finally: - self._TearDownTracing() - return True - - @py_utils.Timeout(tracing_agents.GET_RESULTS_TIMEOUT) - def GetResults(self, timeout=None): - with open(self._PullTrace(), 'r') as f: - trace_data = f.read() - return trace_result.TraceResult('traceEvents', trace_data) - - def _PullTrace(self): - trace_file = self._trace_file.replace('/storage/emulated/0/', '/sdcard/') - host_file = os.path.join(os.path.curdir, os.path.basename(trace_file)) - self._device.PullFile(trace_file, host_file) - return host_file - - def SupportsExplicitClockSync(self): - return False - - def RecordClockSyncMarker(self, sync_id, did_record_sync_marker_callback): - # pylint: disable=unused-argument - assert self.SupportsExplicitClockSync(), ('Clock sync marker cannot be ' - 'recorded since explicit clock sync is not supported.') - - -class ChromeStartupConfig(tracing_agents.TracingConfig): - def __init__(self, device, package_info, webapk_package, cold, url, - chrome_categories, trace_time): - tracing_agents.TracingConfig.__init__(self) - self.device = device - self.package_info = package_info - self.webapk_package = webapk_package - self.cold = cold - self.url = url - self.chrome_categories = chrome_categories - self.trace_time = trace_time - - -def try_create_agent(config): - return ChromeStartupTracingAgent(config.device, config.package_info, - config.webapk_package, - config.cold, config.url, config.trace_time) - -def add_options(parser): - options = optparse.OptionGroup(parser, 'Chrome startup tracing') - options.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') - options.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') - options.add_option('--webapk-package', help='Specify the package name ' - 'of the WebAPK to launch the given URL. An empty URL ' - 'laucnhes the host browser of the WebAPK with an new ' - 'tab.', default=None) - - return options - -def get_config(options): - return ChromeStartupConfig(options.device, options.package_info, - options.webapk_package, options.cold, - options.url, options.chrome_categories, - options.trace_time) |