Index: tools/android/adb_profile_chrome/systrace_controller.py |
diff --git a/tools/android/adb_profile_chrome/systrace_controller.py b/tools/android/adb_profile_chrome/systrace_controller.py |
deleted file mode 100644 |
index 3ec4cc7d754b3b2df9c060e2e2e8f4e93d9d649b..0000000000000000000000000000000000000000 |
--- a/tools/android/adb_profile_chrome/systrace_controller.py |
+++ /dev/null |
@@ -1,95 +0,0 @@ |
-# Copyright 2014 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 threading |
-import zlib |
- |
-from adb_profile_chrome import controllers |
-from adb_profile_chrome import util |
- |
-from pylib import cmd_helper |
- |
- |
-_SYSTRACE_OPTIONS = [ |
- # Compress the trace before sending it over USB. |
- '-z', |
- # Use a large trace buffer to increase the polling interval. |
- '-b', '16384' |
-] |
- |
-# Interval in seconds for sampling systrace data. |
-_SYSTRACE_INTERVAL = 15 |
- |
- |
-class SystraceController(controllers.BaseController): |
- def __init__(self, device, categories, ring_buffer): |
- controllers.BaseController.__init__(self) |
- self._device = device |
- self._categories = categories |
- self._ring_buffer = ring_buffer |
- self._done = threading.Event() |
- self._thread = None |
- self._trace_data = None |
- |
- def __repr__(self): |
- return 'systrace' |
- |
- @staticmethod |
- def GetCategories(device): |
- return device.RunShellCommand('atrace --list_categories') |
- |
- def StartTracing(self, _): |
- self._thread = threading.Thread(target=self._CollectData) |
- self._thread.start() |
- |
- def StopTracing(self): |
- self._done.set() |
- |
- def PullTrace(self): |
- self._thread.join() |
- self._thread = None |
- if self._trace_data: |
- output_name = 'systrace-%s' % util.GetTraceTimestamp() |
- with open(output_name, 'w') as out: |
- out.write(self._trace_data) |
- 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_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) |
- |
- def _CollectData(self): |
- trace_data = [] |
- self._RunATraceCommand('async_start') |
- try: |
- while not self._done.is_set(): |
- self._done.wait(_SYSTRACE_INTERVAL) |
- if not self._ring_buffer or self._done.is_set(): |
- trace_data.append( |
- self._DecodeTraceData(self._RunATraceCommand('async_dump'))) |
- finally: |
- trace_data.append( |
- self._DecodeTraceData(self._RunATraceCommand('async_stop'))) |
- self._trace_data = ''.join([zlib.decompress(d) for d in trace_data]) |
- |
- @staticmethod |
- def _DecodeTraceData(trace_data): |
- try: |
- trace_start = trace_data.index('TRACE:') |
- except ValueError: |
- raise RuntimeError('Systrace start marker not found') |
- trace_data = trace_data[trace_start + 6:] |
- |
- # Collapse CRLFs that are added by adb shell. |
- if trace_data.startswith('\r\n'): |
- trace_data = trace_data.replace('\r\n', '\n') |
- |
- # Skip the initial newline. |
- return trace_data[1:] |