Add --trace-memory option for tracing heap memory

build/android/chrome_profiler/main.py

 #!/usr/bin/env python
 #
 # 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 logging
 import optparse
 import os
 import sys
@@ skipping 14 matching lines @@
 def _ComputeChromeCategories(options):
   categories = []
   if options.trace_frame_viewer:
     categories.append('disabled-by-default-cc.debug')
   if options.trace_ubercompositor:
     categories.append('disabled-by-default-cc.debug*')
   if options.trace_gpu:
     categories.append('disabled-by-default-gpu.debug*')
   if options.trace_flow:
     categories.append('disabled-by-default-toplevel.flow')
+  if options.trace_memory:
+    categories.append('disabled-by-default-memory')
   if options.chrome_categories:
     categories += options.chrome_categories.split(',')
   return categories
 
 
 def _ComputeSystraceCategories(options):
   if not options.systrace_categories:
     return []
   return options.systrace_categories.split(',')
 
@@ skipping 35 matching lines @@
   chrome_opts.add_option('--trace-frame-viewer',
                          help='Enable enough trace categories for '
                          'compositor frame viewing.', action='store_true')
   chrome_opts.add_option('--trace-ubercompositor',
                          help='Enable enough trace categories for '
                          'ubercompositor frame data.', action='store_true')
   chrome_opts.add_option('--trace-gpu', help='Enable extra trace categories '
                          'for GPU data.', action='store_true')
   chrome_opts.add_option('--trace-flow', help='Enable extra trace categories '
                          'for IPC message flows.', action='store_true')
+  chrome_opts.add_option('--trace-memory', help='Enable extra trace categories '
+                         'for memory profile. (tcmalloc required)',
+                         action='store_true')
   parser.add_option_group(chrome_opts)
 
   systrace_opts = optparse.OptionGroup(parser, 'Systrace tracing options')
   systrace_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 '
                         'enable both types of categories. Use "list" to see '
                         'the available categories. Systrace is disabled by '
                         'default.', metavar='SYS_CATEGORIES',
                         dest='systrace_categories', default='')
@@ skipping 74 matching lines @@
   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.')
 
   enabled_controllers = []
   if chrome_categories:
     enabled_controllers.append(
         chrome_controller.ChromeTracingController(device,
                                                   package_info,
                                                   chrome_categories,
-                                                  options.ring_buffer))
+                                                  options.ring_buffer,
+                                                  options.trace_memory))
   if systrace_categories:
     enabled_controllers.append(
         systrace_controller.SystraceController(device,
                                                systrace_categories,
                                                options.ring_buffer))
 
   if not enabled_controllers:
     ui.PrintMessage('No trace categories enabled.')
     return 1
 
   if options.output:
     options.output = os.path.expanduser(options.output)
   result = profiler.CaptureProfile(
       enabled_controllers,
       options.time if not options.continuous else 0,
       output=options.output,
       compress=options.compress,
       write_json=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())