| Index: build/android/adb_profile_chrome.py
|
| diff --git a/build/android/adb_profile_chrome.py b/build/android/adb_profile_chrome.py
|
| index a219bfe2cecdda2978daaf80349da4ca17262862..ae69cbe772d253e1938abe1c363a29a851dbbd79 100755
|
| --- a/build/android/adb_profile_chrome.py
|
| +++ b/build/android/adb_profile_chrome.py
|
| @@ -22,6 +22,7 @@ from pylib import android_commands
|
| from pylib import cmd_helper
|
| from pylib import constants
|
| from pylib import pexpect
|
| +from pylib.device import device_utils
|
|
|
| _TRACE_VIEWER_ROOT = os.path.join(constants.DIR_SOURCE_ROOT,
|
| 'third_party', 'trace-viewer')
|
| @@ -36,8 +37,8 @@ def _GetTraceTimestamp():
|
|
|
|
|
| class ChromeTracingController(object):
|
| - def __init__(self, adb, package_info, categories, ring_buffer):
|
| - self._adb = adb
|
| + def __init__(self, device, package_info, categories, ring_buffer):
|
| + self._device = device
|
| self._package_info = package_info
|
| self._categories = categories
|
| self._ring_buffer = ring_buffer
|
| @@ -47,17 +48,18 @@ class ChromeTracingController(object):
|
| re.compile(r'Logging performance trace to file: (.*)')
|
| self._trace_finish_re = \
|
| re.compile(r'Profiler finished[.] Results are in (.*)[.]')
|
| - self._adb.StartMonitoringLogcat(clear=False)
|
| + self._device.old_interface.StartMonitoringLogcat(clear=False)
|
|
|
| def __str__(self):
|
| return 'chrome trace'
|
|
|
| def StartTracing(self, interval):
|
| self._trace_interval = interval
|
| - self._adb.SyncLogCat()
|
| - self._adb.BroadcastIntent(self._package_info.package, 'GPU_PROFILER_START',
|
| - '-e categories "%s"' % ','.join(self._categories),
|
| - '-e continuous' if self._ring_buffer else '')
|
| + self._device.old_interface.SyncLogCat()
|
| + self._device.old_interface.BroadcastIntent(
|
| + self._package_info.package, 'GPU_PROFILER_START',
|
| + '-e categories "%s"' % ','.join(self._categories),
|
| + '-e continuous' if self._ring_buffer else '')
|
| # Chrome logs two different messages related to tracing:
|
| #
|
| # 1. "Logging performance trace to file [...]"
|
| @@ -66,9 +68,8 @@ class ChromeTracingController(object):
|
| # The first one is printed when tracing starts and the second one indicates
|
| # that the trace file is ready to be pulled.
|
| try:
|
| - self._trace_file = self._adb.WaitForLogMatch(self._trace_start_re,
|
| - None,
|
| - timeout=5).group(1)
|
| + self._trace_file = self._device.old_interface.WaitForLogMatch(
|
| + self._trace_start_re, None, timeout=5).group(1)
|
| except pexpect.TIMEOUT:
|
| raise RuntimeError('Trace start marker not found. Is the correct version '
|
| 'of the browser running?')
|
| @@ -76,8 +77,10 @@ class ChromeTracingController(object):
|
| def StopTracing(self):
|
| if not self._trace_file:
|
| return
|
| - self._adb.BroadcastIntent(self._package_info.package, 'GPU_PROFILER_STOP')
|
| - self._adb.WaitForLogMatch(self._trace_finish_re, None, timeout=120)
|
| + self._device.old_interface.BroadcastIntent(self._package_info.package,
|
| + 'GPU_PROFILER_STOP')
|
| + self._device.old_interface.WaitForLogMatch(self._trace_finish_re, None,
|
| + timeout=120)
|
|
|
| def PullTrace(self):
|
| # Wait a bit for the browser to finish writing the trace file.
|
| @@ -85,7 +88,7 @@ class ChromeTracingController(object):
|
|
|
| trace_file = self._trace_file.replace('/storage/emulated/0/', '/sdcard/')
|
| host_file = os.path.join(os.path.curdir, os.path.basename(trace_file))
|
| - self._adb.PullFileFromDevice(trace_file, host_file)
|
| + self._device.old_interface.PullFileFromDevice(trace_file, host_file)
|
| return host_file
|
|
|
|
|
| @@ -101,8 +104,8 @@ _SYSTRACE_INTERVAL = 15
|
|
|
|
|
| class SystraceController(object):
|
| - def __init__(self, adb, categories, ring_buffer):
|
| - self._adb = adb
|
| + def __init__(self, device, categories, ring_buffer):
|
| + self._device = device
|
| self._categories = categories
|
| self._ring_buffer = ring_buffer
|
| self._done = threading.Event()
|
| @@ -113,8 +116,8 @@ class SystraceController(object):
|
| return 'systrace'
|
|
|
| @staticmethod
|
| - def GetCategories(adb):
|
| - return adb.RunShellCommand('atrace --list_categories')
|
| + def GetCategories(device):
|
| + return device.old_interface.RunShellCommand('atrace --list_categories')
|
|
|
| def StartTracing(self, _):
|
| self._thread = threading.Thread(target=self._CollectData)
|
| @@ -133,10 +136,12 @@ class SystraceController(object):
|
| return output_name
|
|
|
| def _RunATraceCommand(self, command):
|
| + # TODO(jbudorick) can this be made work with DeviceUtils?
|
| # We use a separate interface to adb because the one from AndroidCommands
|
| # isn't re-entrant.
|
| - device = ['-s', self._adb.GetDevice()] if self._adb.GetDevice() else []
|
| - cmd = ['adb'] + device + ['shell', 'atrace', '--%s' % command] + \
|
| + device_param = (['-s', self._device.old_interface.GetDevice()]
|
| + if self._device.old_interface.GetDevice() else [])
|
| + cmd = ['adb'] + device_param + ['shell', 'atrace', '--%s' % command] + \
|
| _SYSTRACE_OPTIONS + self._categories
|
| return cmd_helper.GetCmdOutput(cmd)
|
|
|
| @@ -375,9 +380,13 @@ When in doubt, just try out --trace-frame-viewer.
|
| if options.verbose:
|
| logging.getLogger().setLevel(logging.DEBUG)
|
|
|
| - adb = android_commands.AndroidCommands()
|
| + devices = android_commands.GetAttachedDevices()
|
| + if len(devices) != 1:
|
| + parser.error('Exactly 1 device much be attached.')
|
| + device = device_utils.DeviceUtils(devices[0])
|
| +
|
| if options.systrace_categories in ['list', 'help']:
|
| - _PrintMessage('\n'.join(SystraceController.GetCategories(adb)))
|
| + _PrintMessage('\n'.join(SystraceController.GetCategories(device)))
|
| return 0
|
|
|
| if not options.time and not options.continuous:
|
| @@ -394,12 +403,12 @@ When in doubt, just try out --trace-frame-viewer.
|
|
|
| controllers = []
|
| if chrome_categories:
|
| - controllers.append(ChromeTracingController(adb,
|
| + controllers.append(ChromeTracingController(device,
|
| package_info,
|
| chrome_categories,
|
| options.ring_buffer))
|
| if systrace_categories:
|
| - controllers.append(SystraceController(adb,
|
| + controllers.append(SystraceController(device,
|
| systrace_categories,
|
| options.ring_buffer))
|
|
|
|
|