[Android] Add an option to view the tracing record categories when running from python script.

build/android/adb_profile_chrome.py

 #!/usr/bin/env python
 #
 # Copyright 2013 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 gzip
 import logging
 import optparse
 import os
@@ skipping 35 matching lines @@
     self._trace_interval = None
     self._trace_start_re = \
        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 __str__(self):
     return 'chrome trace'
 
+  @staticmethod
+  def GetCategories(device, package_info):
+    device.old_interface.BroadcastIntent(
+        package_info.package, 'GPU_PROFILER_LIST_CATEGORY')
+    try:
+      trace_list_category_re = re.compile(r'{"traceCategoriesList": \[(.*)\]}')
+      category_list = device.old_interface.WaitForLogMatch(

[Xianzhu, 2014/05/05 16:32:21]

You may want to use 'import json'.

[r.kasibhatla, 2014/05/06 11:04:06]

Done.

+          trace_list_category_re, None, timeout=5).group(1).split(',')
+
+      record_categories = []
+      disabled_by_default_categories = []
+      # Divide the list into default and debug lists.
+      for item in category_list:
+        item = item.strip("\"")
+        if item.startswith('disabled-by-default'):
+          disabled_by_default_categories.append(item)
+        else:
+          record_categories.append(item)
+
+      # Sort the individual lists.
+      record_categories = list(set(record_categories))
+      record_categories.sort()
+      disabled_by_default_categories = list(set(disabled_by_default_categories))
+      disabled_by_default_categories.sort()
+
+      return record_categories, disabled_by_default_categories
+
+    except pexpect.TIMEOUT:
+      raise RuntimeError('Performance trace category list marker not found. '
+                         'Is the correct version of the browser running?')
+
   def StartTracing(self, interval):
     self._trace_interval = interval
     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"
@@ skipping 251 matching lines @@
                           'ring buffer and save its contents when stopping '
                           'instead of appending events into one long trace.',
                           action='store_true')
   parser.add_option_group(cont_options)
 
   categories = optparse.OptionGroup(parser, 'Trace categories')
   categories.add_option('-c', '--categories', help='Select Chrome tracing '
                         'categories with comma-delimited wildcards, '
                         'e.g., "*", "cat1*,-cat1a". Omit this option to trace '
                         'Chrome\'s default categories. Chrome tracing can be '
-                        'disabled with "--categories=\'\'".',
+                        'disabled with "--categories=\'\'". Use "list" to see '
+                        'the available categories.',
                         metavar='CHROME_CATEGORIES', dest='chrome_categories',
                         default=_DEFAULT_CHROME_CATEGORIES)
   categories.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.', metavar='SYS_CATEGORIES',
                         dest='systrace_categories', default='')
   categories.add_option('--trace-cc',
@@ skipping 38 matching lines @@
 When in doubt, just try out --trace-frame-viewer.
 """)
 
   if options.verbose:
     logging.getLogger().setLevel(logging.DEBUG)
 
   devices = android_commands.GetAttachedDevices()
   if len(devices) != 1:
     parser.error('Exactly 1 device much be attached.')
   device = device_utils.DeviceUtils(devices[0])
+  package_info = _GetSupportedBrowsers()[options.browser]
+
+  if options.chrome_categories in ['list', 'help']:
+    _PrintMessage('Fetching the record categories list ...')
+    record_categories = []
+    disabled_by_default_categories = []
+    record_categories, disabled_by_default_categories = \
+        ChromeTracingController.GetCategories(device, package_info)
+
+    _PrintMessage('\nRecord Categories -\n')
+    for item in record_categories:
+      _PrintMessage(item)
+
+    _PrintMessage('\nDisabled by Default Categories -\n')
+    for item in disabled_by_default_categories:
+      _PrintMessage(item)
+
+    return 0
 
   if options.systrace_categories in ['list', 'help']:
     _PrintMessage('\n'.join(SystraceController.GetCategories(device)))
     return 0
 
   if not options.time and not options.continuous:
     _PrintMessage('Time interval or continuous tracing should be specified.')
     return 1
 
   chrome_categories = _ComputeChromeCategories(options)
   systrace_categories = _ComputeSystraceCategories(options)
-  package_info = _GetSupportedBrowsers()[options.browser]
 
   if chrome_categories and 'webview' in systrace_categories:
     logging.warning('Using the "webview" category in systrace together with '
                     'Chrome tracing results in duplicate trace events.')
 
   controllers = []
   if chrome_categories:
     controllers.append(ChromeTracingController(device,
                                                package_info,
                                                chrome_categories,
@@ skipping 16 matching lines @@
                                 options.json)
   if options.view:
     if sys.platform == 'darwin':
       os.system('/usr/bin/open %s' % os.path.abspath(result))
     else:
       webbrowser.open(result)
 
 
 if __name__ == '__main__':
   sys.exit(main())