Print pak sizes on a per-header basis in resource_sizes.py

build/android/resource_sizes.py

 #!/usr/bin/python
 # Copyright (c) 2011 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.
 
 """Prints the size of each given file and optionally computes the size of
    libchrome.so without the dependencies added for building with android NDK.
    Also breaks down the contents of the APK to determine the installed size
    and assign size contributions to different classes of file.
 """
@@ skipping 502 matching lines @@
 
   # Output the overall resource summary.
   total_resource_size = sum(resource_size_map.values())
   total_resource_count = len(resource_count_map)
   assert total_resource_size <= total_file_size
   print 'Total pak resources: %s' % total_resource_count
   print 'Total uncompressed resource size: %s' % _FormatBytes(
       total_resource_size)
   print
 
-  resource_id_name_map = _GetResourceIdNameMap()
+  resource_id_name_map, resources_id_header_map = _AnnotatePakResources()
 
   # Output the table of details about all resources across pak files.
   print
   print '%56s %5s %17s' % ('RESOURCE', 'COUNT', 'UNCOMPRESSED SIZE')
   for i in sorted(resource_size_map, key=resource_size_map.get,
                   reverse=True):
-    if resource_size_map[i] >= min_pak_resource_size:
-      print '%56s %5s %9s %6.2f%%' % (
-          resource_id_name_map.get(i, i),
-          resource_count_map[i],
-          _FormatBytes(resource_size_map[i]),
-          100.0 * resource_size_map[i] / total_resource_size)
+    if resource_size_map[i] < min_pak_resource_size:
+      break
+
+    print '%56s %5s %9s %6.2f%%' % (
+        resource_id_name_map.get(i, i),
+        resource_count_map[i],
+        _FormatBytes(resource_size_map[i]),
+        100.0 * resource_size_map[i] / total_resource_size)
+
+  # Print breakdown on a per-grd file basis.
+  size_by_header = collections.defaultdict(int)
+  for resid, size in resource_size_map.iteritems():
+    size_by_header[resources_id_header_map.get(resid, 'unknown')] += size
+
+  print
+  print '%80s %17s' % ('HEADER', 'UNCOMPRESSED SIZE')
+  for header in sorted(size_by_header, key=size_by_header.get, reverse=True):
+    if size_by_header[header] < min_pak_resource_size:
+      break
+
+    print '%80s %9s %6.2f%%' % (
+        header,
+        _FormatBytes(size_by_header[header]),
+        100.0 * size_by_header[header] / total_resource_size)
 
 
-def _GetResourceIdNameMap():
-  """Returns a map of {resource_id: resource_name}."""
+def _AnnotatePakResources():
+  """Returns a pair of maps: id_name_map, id_header_map."""
   out_dir = constants.GetOutDirectory()
   assert os.path.isdir(out_dir), 'Failed to locate out dir at %s' % out_dir
   print 'Looking at resources in: %s' % out_dir
 
   grit_headers = []
   for root, _, files in os.walk(out_dir):
     if root.endswith('grit'):
       grit_headers += [os.path.join(root, f) for f in files if f.endswith('.h')]
   assert grit_headers, 'Failed to find grit headers in %s' % out_dir
 
   id_name_map = {}
+  id_header_map = {}
   for header in grit_headers:
     with open(header, 'r') as f:
       for line in f.readlines():
         m = _RC_HEADER_RE.match(line.strip())
         if m:
           i = int(m.group('id'))
           name = m.group('name')
           if i in id_name_map and name != id_name_map[i]:
             print 'WARNING: Resource ID conflict %s (%s vs %s)' % (
                 i, id_name_map[i], name)
           id_name_map[i] = name
-  return id_name_map
+          id_header_map[i] = os.path.relpath(header, out_dir)
+  return id_name_map, id_header_map
 
 
 def _PrintStaticInitializersCountFromApk(apk_filename, chartjson=None):
   print 'Finding static initializers (can take a minute)'
   with zipfile.ZipFile(apk_filename) as z:
     infolist = z.infolist()
   out_dir = constants.GetOutDirectory()
   si_count = 0
   for zip_info in infolist:
     # Check file size to account for placeholder libraries.
@@ skipping 137 matching lines @@
 
   if chartjson:
     results_path = os.path.join(options.output_dir, 'results-chart.json')
     logging.critical('Dumping json to %s', results_path)
     with open(results_path, 'w') as json_file:
       json.dump(chartjson, json_file)
 
 
 if __name__ == '__main__':
   sys.exit(main(sys.argv))