Index: build/android/apksize.py |
diff --git a/build/android/apksize.py b/build/android/apksize.py |
deleted file mode 100755 |
index ae5462bd222a5c5bd4c72139814f3a36c73c6486..0000000000000000000000000000000000000000 |
--- a/build/android/apksize.py |
+++ /dev/null |
@@ -1,228 +0,0 @@ |
-#!/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()) |