Enable vulcanizing perf_insights mapper

perf_insights/perf_insights/map_traces.py

 # Copyright (c) 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 argparse
 import os
 import sys
 import traceback
 
 import perf_insights
 from perf_insights import local_directory_corpus_driver
@@ skipping 38 matching lines @@
 
 
 def Main(argv):
   parser = argparse.ArgumentParser(
       description='Bulk trace processing')
   parser.add_argument(
       '-c', '--corpus',
       choices=_CORPUS_DRIVERS.keys(),
       default=_CORPUS_DRIVER_DEFAULT)
   parser.add_argument('--query')
-  parser.add_argument('map_file')
+  parser.add_argument('map_function_name')
+  parser.add_argument('--map-file')
 
+  parser.add_argument('--no-vulcanize', dest='vulcanize',
+                      action='store_false', default=True)
   parser.add_argument('-j', '--jobs', type=int, default=1)
   parser.add_argument('-o', '--output-file')
   parser.add_argument('-s', '--stop-on-error',
                       action='store_true')
 
   for k, v in _CORPUS_DRIVERS.iteritems():
     if not v:
       continue
     parser_group = parser.add_argument_group(k)
     driver_cls = v['class']
@@ skipping 11 matching lines @@
     corpus_descriptions = '\n'.join(
         ['%s: %s' % (k, v['description'])
             for k, v in _CORPUS_DRIVERS.iteritems() if v]
       )
     parser.exit('Valid drivers:\n\n%s\n' % corpus_descriptions)
 
   cls = _CORPUS_DRIVERS[corpus]['class']
   init_args = cls.CheckAndCreateInitArguments(parser, args)
   corpus_driver = cls(**init_args)
 
-  if not os.path.exists(args.map_file):
+  if args.map_file and not os.path.exists(args.map_file):
     parser.error('Map does not exist.')
 
   if args.query == 'help':
     parser.exit(_CORPUS_QUERY_HELP)
   elif args.query is None:
     query = corpus_query.CorpusQuery.FromString('True')
   else:
     query = corpus_query.CorpusQuery.FromString(args.query)
 
   if args.output_file:
     ofile = open(args.output_file, 'w')
   else:
     ofile = sys.stdout
 
   output_formatter = json_output_formatter.JSONOutputFormatter(ofile)
 
+  map_file_path = None
+  if args.map_file:
+    map_file_path = os.path.abspath(args.map_file)
   map_function_handle = map_function_handle_module.MapFunctionHandle(
-      filename=os.path.abspath(args.map_file))
+      filename=map_file_path, map_function_name=args.map_function_name)
   try:
     trace_handles = corpus_driver.GetTraceHandlesMatchingQuery(query)
     runner = map_runner.MapRunner(trace_handles, map_function_handle,
-                                  stop_on_error=args.stop_on_error)
+                                  stop_on_error=args.stop_on_error,
+                                  vulcanize=args.vulcanize)
     results = runner.Run(jobs=args.jobs, output_formatters=[output_formatter])
     if not results.had_failures:
       return 0
     else:
       return 255
   finally:
     if ofile != sys.stdout:
       ofile.close()