Index: tools/profile_chrome/chrome_startup_controller.py |
diff --git a/tools/profile_chrome/chrome_startup_controller.py b/tools/profile_chrome/chrome_startup_controller.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..5a946ad1f894ef19ea82c813f582eb6da078a774 |
--- /dev/null |
+++ b/tools/profile_chrome/chrome_startup_controller.py |
@@ -0,0 +1,63 @@ |
+# 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 os |
+import re |
+import time |
+ |
+from pylib import flag_changer |
+from pylib.perf import cache_control |
+ |
+from profile_chrome import controllers |
+ |
+class ChromeStartupTracingController(controllers.BaseController): |
+ def __init__(self, device, package_info, cold, url): |
+ self._device = device |
+ self._package_info = package_info |
+ self._cold = cold |
+ self._url = url |
+ self._trace_file = None |
+ self._trace_finish_re = re.compile(r' Completed startup tracing to (.*)') |
+ self._device.old_interface.StartMonitoringLogcat(clear=False) |
+ |
+ def __repr__(self): |
+ return 'Browser Startup Trace' |
+ |
+ def _SetupTracing(self): |
+ changer = flag_changer.FlagChanger( |
+ self._device, self._package_info.cmdline_file) |
+ changer.AddFlags(['--trace-startup']) |
+ self._device.old_interface.CloseApplication(self._package_info.package) |
+ if self._cold: |
+ self._device.old_interface.EnableAdbRoot() |
+ cache_control.CacheControl(self._device).DropRamCaches() |
+ self._device.old_interface.StartActivity( |
+ package=self._package_info.package, |
+ activity=self._package_info.activity, |
+ data=self._url, |
+ extras={'create_new_tab' : True}) |
+ |
+ def _TearDownTracing(self): |
+ changer = flag_changer.FlagChanger( |
+ self._device, self._package_info.cmdline_file) |
+ changer.RemoveFlags(['--trace-startup']) |
+ |
+ def StartTracing(self, interval): |
+ self._SetupTracing() |
+ self._device.old_interface.SyncLogCat() |
+ |
+ def StopTracing(self): |
+ try: |
+ self._trace_file = self._device.old_interface.WaitForLogMatch( |
+ self._trace_finish_re, None, timeout=10).group(1) |
+ finally: |
+ self._TearDownTracing() |
+ |
+ def PullTrace(self): |
+ # Wait a bit for the browser to finish writing the trace file. |
+ time.sleep(3) |
+ 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 |