//tools/binary_size: Consolidate most tools into "supersize" command

tools/binary_size/libsupersize/html_report.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.
 
 """Creates an html report that allows you to view binary size by component."""
 
 import argparse
 import json
 import logging
 import os
 import shutil
 import sys
 
+import archive
 import helpers
-import map2size
 
 
 # Node dictionary keys. These are output in json read by the webapp so
 # keep them short to save file size.
 # Note: If these change, the webapp must also change.
 _NODE_TYPE_KEY = 'k'
 _NODE_TYPE_BUCKET = 'b'
 _NODE_TYPE_PATH = 'p'
 _NODE_TYPE_SYMBOL = 's'
 _NODE_NAME_KEY = 'n'
@@ skipping 129 matching lines @@
   if not os.path.exists(d3_out):
     os.makedirs(d3_out, 0755)
   d3_src = os.path.join(helpers.SRC_ROOT, 'third_party', 'd3', 'src')
   template_src = os.path.join(os.path.dirname(__file__), 'template')
   shutil.copy(os.path.join(d3_src, 'LICENSE'), d3_out)
   shutil.copy(os.path.join(d3_src, 'd3.js'), d3_out)
   shutil.copy(os.path.join(template_src, 'index.html'), dest_dir)
   shutil.copy(os.path.join(template_src, 'D3SymbolTreeMap.js'), dest_dir)
 
 
-def main(argv):
-  parser = argparse.ArgumentParser()
+def AddArguments(parser):
   parser.add_argument('input_file',
                       help='Path to input .size file.')
   parser.add_argument('--report-dir', metavar='PATH', required=True,
                       help='Write output to the specified directory. An HTML '
                             'report is generated here.')
   parser.add_argument('--include-bss', action='store_true',
                       help='Include symbols from .bss (which consume no real '
                            'space)')
   parser.add_argument('--include-symbols', action='store_true',
                       help='Use per-symbol granularity rather than per-file.')
-  args = helpers.AddCommonOptionsAndParseArgs(parser, argv)
 
-  size_info = map2size.LoadAndPostProcessSizeInfo(args.input_file)
+
+def Run(args, parser):
+  if not args.input_file.endswith('.size'):
+    parser.error('Input must end with ".size"')
+
+  logging.info('Reading .size file')
+  size_info = archive.LoadAndPostProcessSizeInfo(args.input_file)
   symbols = size_info.symbols
   if not args.include_bss:
     symbols = symbols.WhereInSection('b').Inverted()
   symbols = symbols.WhereBiggerThan(0)
 
   # Copy report boilerplate into output directory. This also proves that the
   # output directory is safe for writing, so there should be no problems writing
   # the nm.out file later.
   _CopyTemplateFiles(args.report_dir)
 
   logging.info('Creating JSON objects')
   tree_root = _MakeCompactTree(symbols, args.include_symbols)
 
-  logging.info('Serializing')
+  logging.info('Serializing JSON')
   with open(os.path.join(args.report_dir, 'data.js'), 'w') as out_file:
     out_file.write('var tree_data=')
     # Use separators without whitespace to get a smaller file.
     json.dump(tree_root, out_file, ensure_ascii=False, check_circular=False,
               separators=(',', ':'))
 
-  print 'Report saved to ' + args.report_dir + '/index.html'
-
-
-if __name__ == '__main__':
-  sys.exit(main(sys.argv))
+  logging.warning('Report saved to %s/index.html', args.report_dir)