Use Systrace tracing controller in profile_chrome

systrace/profile_chrome/atrace_tracing_agent.py

 # 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 optparse
 import py_utils
 import threading
 import zlib
 
 from devil.utils import cmd_helper
@@ skipping 34 matching lines @@
 
   @staticmethod
   def GetCategories(device):
     return device.RunShellCommand('atrace --list_categories')
 
   @py_utils.Timeout(tracing_agents.START_STOP_TIMEOUT)
   def StartAgentTracing(self, config, timeout=None):
     self._categories = _ComputeAtraceCategories(config)
     self._thread = threading.Thread(target=self._CollectData)
     self._thread.start()
+    return True
 
   @py_utils.Timeout(tracing_agents.START_STOP_TIMEOUT)
   def StopAgentTracing(self, timeout=None):
     self._done.set()
+    return True
 
   @py_utils.Timeout(tracing_agents.GET_RESULTS_TIMEOUT)
   def GetResults(self, timeout=None):
     self._thread.join()
     self._thread = None
     return trace_result.TraceResult('systemTraceEvents', self._trace_data)
 
   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.')
 
   def IsTracingOn(self):
     result = self._RunAdbShellCommand(['cat', _TRACING_ON_PATH])
     return result.strip() == '1'
 
   def _RunAdbShellCommand(self, command):
     # We use a separate interface to adb because the one from AndroidCommands
     # isn't re-entrant.
@@ skipping 39 matching lines @@
 
     # 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:]
 
 
 class AtraceConfig(tracing_agents.TracingConfig):
-  def __init__(self, atrace_categories):
+  def __init__(self, atrace_categories, device, ring_buffer):
     tracing_agents.TracingConfig.__init__(self)
     self.atrace_categories = atrace_categories
+    self.device = device
+    self.ring_buffer = ring_buffer
 
 
+def try_create_agent(config):
+  if config.atrace_categories:
+    return AtraceAgent(config.device, config.ring_buffer)
+  return None
+
 def add_options(parser):
   atrace_opts = optparse.OptionGroup(parser, 'Atrace tracing options')
   atrace_opts.add_option('-s', '--systrace', help='Capture a systrace with '
                            'the chosen comma-delimited systrace categories. You'
                            ' can also capture a combined Chrome + systrace by '
                            'enabling both types of categories. Use "list" to '
                            'see the available categories. Systrace is disabled'
                            ' by default. Note that in this case, Systrace is '
                            'synonymous with Atrace.',
                            metavar='ATRACE_CATEGORIES',
                            dest='atrace_categories', default='')
   return atrace_opts
 
 def get_config(options):
-  return AtraceConfig(options.atrace_categories)
+  return AtraceConfig(options.atrace_categories, options.device,
+                      options.ring_buffer)
 
 def _ComputeAtraceCategories(config):
   if not config.atrace_categories:
     return _DEFAULT_CATEGORIES
   return config.atrace_categories.split(',')