Pass in custom options to Systrace agents

systrace/profile_chrome/ddms_tracing_agent.py

 # 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 optparse
 import os
 import py_utils
 import re
 
 from profile_chrome import util
 from systrace import trace_result
 from systrace import tracing_agents
 
 
 _DDMS_SAMPLING_FREQUENCY_US = 100
@@ skipping 10 matching lines @@
   def __repr__(self):
     return 'ddms profile'
 
   def _SupportsSampling(self):
     for line in self._device.RunShellCommand('am --help'):
       if re.match(r'.*am profile start.*--sampling', line):
         return True
     return False
 
   @py_utils.Timeout(tracing_agents.START_STOP_TIMEOUT)
-  def StartAgentTracing(self, options, categories, timeout=None):
+  def StartAgentTracing(self, config, timeout=None):
     self._output_file = (
         '/data/local/tmp/ddms-profile-%s' % util.GetTraceTimestamp())
     cmd = 'am profile start '
     if self._supports_sampling:
       cmd += '--sampling %d ' % _DDMS_SAMPLING_FREQUENCY_US
     cmd += '%s %s' % (self._package, self._output_file)
     self._device.RunShellCommand(cmd)
 
   @py_utils.Timeout(tracing_agents.START_STOP_TIMEOUT)
   def StopAgentTracing(self, timeout=None):
@@ skipping 13 matching lines @@
         os.path.curdir, os.path.basename(self._output_file))
     self._device.PullFile(self._output_file, host_file)
     return host_file
 
   def SupportsExplicitClockSync(self):
     return False
 
   def RecordClockSyncMarker(self, sync_id, did_record_sync_marker_callback):
     assert self.SupportsExplicitClockSync(), ('Clock sync marker cannot be '
         'recorded since explicit clock sync is not supported.')
+
+
+class DdmsConfig(tracing_agents.TracingConfig):
+  def __init__(self):
+    tracing_agents.TracingConfig.__init__(self)
+
+
+def add_options(parser):
+  options = optparse.OptionGroup(parser, 'Java tracing')
+  options.add_option('--ddms', help='Trace Java execution using DDMS '
+                     'sampling.', action='store_true')
+  return options
+
+def get_config(options):
+  # pylint: disable=unused-argument
+  return DdmsConfig()