🎋 Remove apksize.py in favor of resource_sizes.py

build/android/apksize.py

-#!/usr/bin/env python
-# Copyright 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 collections
-import json
-import logging
-import os
-import sys
-import zipfile
-
-_BASE_CHART = {
-    'format_version': '0.1',
-    'benchmark_name': 'apk_size',
-    'benchmark_description': 'APK size information.',
-    'trace_rerun_options': [],
-    'charts': {}
-}
-
-
-# TODO(rnephew): Add support for split apks.
-class ApkSizeInfo(object):
-
-  def __init__(self, path):
-    """ApkSizeInfo constructor.
-
-    Args:
-      path: Path to apk.
-    """
-    if not os.path.isfile(path):
-      raise IOError('Not a valid file path for apk.')
-    if not os.access(path, os.R_OK):
-      raise IOError('File is not readable.')
-    if not zipfile.is_zipfile(path):
-      raise TypeError('Not a valid apk')
-    logging.info('APK: %s', path)
-    self._apk_size = os.path.getsize(path)
-    self._zipfile = zipfile.ZipFile(path, 'r')
-    self._processed_files = None
-    self._compressed_size = 0
-    self._total_files = 0
-    self._uncompressed_size = 0
-    self._ProcessFiles()
-
-  def _ProcessFiles(self):
-    """Uses zipinfo to process apk file information."""
-    INITIAL_FILE_EXTENSION_INFO = {
-        'number': 0,
-        'compressed_bytes': 0,
-        'uncompressed_bytes': 0
-    }
-    self._processed_files = collections.defaultdict(
-        lambda: dict(INITIAL_FILE_EXTENSION_INFO))
-
-    for f in self._zipfile.infolist():
-      _, file_ext = os.path.splitext(f.filename)
-      file_ext = file_ext[1:] # Drop . from extension.
-
-      self._compressed_size += f.compress_size
-      self._total_files += 1
-      self._uncompressed_size += f.file_size
-      self._processed_files[file_ext]['number'] += 1
-      self._processed_files[file_ext]['compressed_bytes'] += f.compress_size
-      self._processed_files[file_ext]['uncompressed_bytes'] += f.file_size
-    return self._processed_files
-
-  def Compare(self, other_apk):
-    """Compares size information of two apks.
-
-    Args:
-      other_apk: ApkSizeInfo instance to compare size against.
-
-    Returns:
-      Dictionary of comparision results.
-    """
-    if not isinstance(other_apk, type(self)):
-      raise TypeError('Must pass it an ApkSizeInfo object')
-
-    other_lib_compressed = other_apk.processed_files['so']['compressed_bytes']
-    other_lib_uncompressed = (
-        other_apk.processed_files['so']['uncompressed_bytes'])
-    this_lib_compressed = self._processed_files['so']['compressed_bytes']
-    this_lib_uncompressed = self._processed_files['so']['uncompressed_bytes']
-
-    # TODO(rnephew) This will be made obsolete with modern and legacy apks being
-    # separate, a new method to compare will be required eventually.
-    return collections.OrderedDict([
-        ('APK_size_reduction',
-            other_apk.compressed_size - self.compressed_size),
-        ('ARM32_Legacy_install_or_upgrade_reduction',
-            (other_lib_compressed - this_lib_compressed) +
-            (other_lib_uncompressed - this_lib_uncompressed)),
-        ('ARM32_Legacy_system_image_reduction',
-            other_lib_compressed - this_lib_compressed),
-        ('ARM32_Modern_ARM64_install_or_upgrade_reduction',
-            other_lib_uncompressed - this_lib_uncompressed),
-        ('ARM32_Modern_ARM64_system_image_reduction',
-            other_lib_uncompressed - this_lib_uncompressed),
-    ])
-
-  @property
-  def apk_size(self):
-    return self._apk_size
-
-  @property
-  def compressed_size(self):
-    return self._compressed_size
-
-  @property
-  def total_files(self):
-    return self._total_files
-
-  @property
-  def uncompressed_size(self):
-    return self._uncompressed_size
-
-  @property
-  def processed_files(self):
-    return self._processed_files
-
-def add_value(chart_data, graph_title, trace_title, value, units,
-              improvement_direction='down', important=True):
-  chart_data['charts'].setdefault(graph_title, {})
-  chart_data['charts'][graph_title][trace_title] = {
-      'type': 'scalar',
-      'value': value,
-      'units': units,
-      'imporvement_direction': improvement_direction,
-      'important': important
-  }
-
-def chartjson_size_info(apk, output_dir):
-  """Sends size information to perf dashboard.
-
-  Args:
-    apk: ApkSizeInfo object
-  """
-  data = _BASE_CHART.copy()
-  files = apk.processed_files
-  add_value(data, 'files', 'total', apk.total_files, 'count')
-  add_value(data, 'size', 'total_size_compressed', apk.compressed_size, 'bytes')
-  add_value(data, 'size', 'total_size_uncompressed', apk.uncompressed_size,
-            'bytes')
-  add_value(data, 'size', 'apk_overhead', apk.apk_size - apk.compressed_size,
-           'bytes')
-  for ext in files:
-    add_value(data, 'files', ext, files[ext]['number'], 'count')
-    add_value(data, 'size_compressed', ext, files[ext]['compressed_bytes'],
-              'bytes')
-    add_value(data, 'size_uncompressed', ext, files[ext]['uncompressed_bytes'],
-              'bytes')
-
-  logging.info('Outputing data to json file %s', output_dir)
-  with open(os.path.join(output_dir, 'results-chart.json'), 'w') as json_file:
-    json.dump(data, json_file)
-
-def print_human_readable_size_info(apk):
-  """Prints size information in human readable format.
-
-  Args:
-    apk: ApkSizeInfo object
-  """
-  files = apk.processed_files
-  logging.critical('Stats for files as they exist within the apk:')
-  for ext in files:
-    logging.critical('  %-8s %s bytes in %s files', ext,
-                     files[ext]['compressed_bytes'], files[ext]['number'])
-  logging.critical('--------------------------------------')
-  logging.critical(
-      'All Files: %s bytes in %s files', apk.compressed_size, apk.total_files)
-  logging.critical('APK Size: %s', apk.apk_size)
-  logging.critical('APK overhead: %s', apk.apk_size - apk.compressed_size)
-  logging.critical('--------------------------------------')
-  logging.critical('Stats for files when extracted from the apk:')
-  for ext in files:
-    logging.critical('  %-8s %s bytes in %s files', ext,
-                     files[ext]['uncompressed_bytes'], files[ext]['number'])
-  logging.critical('--------------------------------------')
-  logging.critical(
-      'All Files: %s bytes in %s files', apk.uncompressed_size, apk.total_files)
-
-def chartjson_compare(compare_dict, output_dir):
-  """Sends size comparison between two apks to perf dashboard.
-
-  Args:
-    compare_dict: Dictionary returned from APkSizeInfo.Compare()
-  """
-  data = _BASE_CHART.copy()
-  for key, value in compare_dict.iteritems():
-    add_value(data, 'compare', key, value, 'bytes')
-
-  logging.info('Outputing data to json file %s', output_dir)
-  with open(os.path.join(output_dir, 'results-chart.json'), 'w') as json_file:
-    json.dump(data, json_file)
-
-def print_human_readable_compare(compare_dict):
-  """Prints size comparison between two apks in human readable format.
-
-  Args:
-    compare_dict: Dictionary returned from ApkSizeInfo.Compare()
-  """
-  for key, value in compare_dict.iteritems():
-    logging.critical('  %-50s %s bytes', key, value)
-
-def main():
-  parser = argparse.ArgumentParser()
-  parser.add_argument('file_path')
-  parser.add_argument('-c', '--compare', help='APK to compare against.')
-  parser.add_argument('-o', '--output-dir',
-                      help='Sets it to return data in bot readable format')
-  parser.add_argument('-d', '--device', help='Dummy option for perf runner.')
-  args = parser.parse_args()
-
-  apk = ApkSizeInfo(args.file_path)
-  if args.compare:
-    compare_dict = apk.Compare(ApkSizeInfo(args.compare))
-    print_human_readable_compare(compare_dict)
-    if args.output_dir:
-      chartjson_compare(compare_dict, args.output_dir)
-  else:
-    print_human_readable_size_info(apk)
-    if args.output_dir:
-       chartjson_size_info(apk, args.output_dir)
-
-if __name__ == '__main__':
-  sys.exit(main())