| Index: tools/profile_chrome/chrome_controller.py
|
| diff --git a/tools/profile_chrome/chrome_controller.py b/tools/profile_chrome/chrome_controller.py
|
| index 9f522631b1e6e6591c49d595a8985f5657e78180..d8cff46b7457f0899a377fee4dd8a51febda35f1 100644
|
| --- a/tools/profile_chrome/chrome_controller.py
|
| +++ b/tools/profile_chrome/chrome_controller.py
|
| @@ -9,7 +9,7 @@ import time
|
|
|
| from profile_chrome import controllers
|
|
|
| -from pylib import pexpect
|
| +from pylib.device import device_errors
|
| from pylib.device import intent
|
|
|
|
|
| @@ -23,6 +23,7 @@ class ChromeTracingController(controllers.BaseController):
|
| self._package_info = package_info
|
| self._categories = categories
|
| self._ring_buffer = ring_buffer
|
| + self._logcat_monitor = self._device.GetLogcatMonitor()
|
| self._trace_file = None
|
| self._trace_interval = None
|
| self._trace_memory = trace_memory
|
| @@ -31,21 +32,21 @@ class ChromeTracingController(controllers.BaseController):
|
| re.compile(r'Logging performance trace to file')
|
| self._trace_finish_re = \
|
| re.compile(r'Profiler finished[.] Results are in (.*)[.]')
|
| - self._device.old_interface.StartMonitoringLogcat(clear=False)
|
|
|
| def __repr__(self):
|
| return 'chrome trace'
|
|
|
| @staticmethod
|
| def GetCategories(device, package_info):
|
| - device.BroadcastIntent(intent.Intent(
|
| - action='%s.GPU_PROFILER_LIST_CATEGORIES' % package_info.package))
|
| - try:
|
| - json_category_list = device.old_interface.WaitForLogMatch(
|
| - re.compile(r'{"traceCategoriesList(.*)'), None, timeout=5).group(0)
|
| - except pexpect.TIMEOUT:
|
| - raise RuntimeError('Performance trace category list marker not found. '
|
| - 'Is the correct version of the browser running?')
|
| + with device.GetLogcatMonitor() as logmon:
|
| + device.BroadcastIntent(intent.Intent(
|
| + action='%s.GPU_PROFILER_LIST_CATEGORIES' % package_info.package))
|
| + try:
|
| + json_category_list = logmon.WaitFor(
|
| + re.compile(r'{"traceCategoriesList(.*)'), timeout=5).group(0)
|
| + except device_errors.CommandTimeoutError:
|
| + raise RuntimeError('Performance trace category list marker not found. '
|
| + 'Is the correct version of the browser running?')
|
|
|
| record_categories = set()
|
| disabled_by_default_categories = set()
|
| @@ -61,7 +62,7 @@ class ChromeTracingController(controllers.BaseController):
|
|
|
| def StartTracing(self, interval):
|
| self._trace_interval = interval
|
| - self._device.old_interface.SyncLogCat()
|
| + self._logcat_monitor.Start()
|
| start_extras = {'categories': ','.join(self._categories)}
|
| if self._ring_buffer:
|
| start_extras['continuous'] = None
|
| @@ -70,7 +71,7 @@ class ChromeTracingController(controllers.BaseController):
|
| extras=start_extras))
|
|
|
| if self._trace_memory:
|
| - self._device.old_interface.EnableAdbRoot()
|
| + self._device.EnableRoot()
|
| self._device.SetProp(_HEAP_PROFILE_MMAP_PROPERTY, 1)
|
|
|
| # Chrome logs two different messages related to tracing:
|
| @@ -81,10 +82,9 @@ class ChromeTracingController(controllers.BaseController):
|
| # The first one is printed when tracing starts and the second one indicates
|
| # that the trace file is ready to be pulled.
|
| try:
|
| - self._device.old_interface.WaitForLogMatch(
|
| - self._trace_start_re, None, timeout=5)
|
| + self._logcat_monitor.WaitFor(self._trace_start_re, timeout=5)
|
| self._is_tracing = True
|
| - except pexpect.TIMEOUT:
|
| + except device_errors.CommandTimeoutError:
|
| raise RuntimeError('Trace start marker not found. Is the correct version '
|
| 'of the browser running?')
|
|
|
| @@ -92,8 +92,8 @@ class ChromeTracingController(controllers.BaseController):
|
| if self._is_tracing:
|
| self._device.BroadcastIntent(intent.Intent(
|
| action='%s.GPU_PROFILER_STOP' % self._package_info.package))
|
| - self._trace_file = self._device.old_interface.WaitForLogMatch(
|
| - self._trace_finish_re, None, timeout=120).group(1)
|
| + self._trace_file = self._logcat_monitor.WaitFor(
|
| + self._trace_finish_re, timeout=120).group(1)
|
| self._is_tracing = False
|
| if self._trace_memory:
|
| self._device.SetProp(_HEAP_PROFILE_MMAP_PROPERTY, 0)
|
|
|