| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright 2013 The LUCI Authors. All rights reserved. | 2 # Copyright 2013 The LUCI Authors. All rights reserved. |
| 3 # Use of this source code is governed under the Apache License, Version 2.0 | 3 # Use of this source code is governed under the Apache License, Version 2.0 |
| 4 # that can be found in the LICENSE file. | 4 # that can be found in the LICENSE file. |
| 5 | 5 |
| 6 """Profiler to compare various compression levels with regards to speed | 6 """Profiler to compare various compression levels with regards to speed |
| 7 and final size when compressing the full set of files from a given | 7 and final size when compressing the full set of files from a given |
| 8 isolated file. | 8 isolated file. |
| 9 """ | 9 """ |
| 10 | 10 |
| 11 import bz2 | 11 import bz2 |
| 12 import optparse | 12 import optparse |
| 13 import os | 13 import os |
| 14 import subprocess | 14 import subprocess |
| 15 import sys | 15 import sys |
| 16 import tempfile | 16 import tempfile |
| 17 import time | 17 import time |
| 18 import zlib | 18 import zlib |
| 19 | 19 |
| 20 ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) | 20 ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath( |
| 21 __file__.decode(sys.getfilesystemencoding())))) |
| 21 sys.path.insert(0, ROOT_DIR) | 22 sys.path.insert(0, ROOT_DIR) |
| 22 | 23 |
| 23 from third_party.depot_tools import fix_encoding | 24 from third_party.depot_tools import fix_encoding |
| 24 from utils import file_path | 25 from utils import file_path |
| 25 from utils import tools | 26 from utils import tools |
| 26 | 27 |
| 27 | 28 |
| 28 def zip_file(compressor_constructor, compression_level, filename): | 29 def zip_file(compressor_constructor, compression_level, filename): |
| 29 compressed_size = 0 | 30 compressed_size = 0 |
| 30 compressor = compressor_constructor(compression_level) | 31 compressor = compressor_constructor(compression_level) |
| (...skipping 85 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 116 temp_dir) | 117 temp_dir) |
| 117 profile_compress('bz2', bz2.BZ2Compressor, range(1, 10), zip_directory, | 118 profile_compress('bz2', bz2.BZ2Compressor, range(1, 10), zip_directory, |
| 118 temp_dir) | 119 temp_dir) |
| 119 finally: | 120 finally: |
| 120 file_path.rmtree(temp_dir) | 121 file_path.rmtree(temp_dir) |
| 121 | 122 |
| 122 | 123 |
| 123 if __name__ == '__main__': | 124 if __name__ == '__main__': |
| 124 fix_encoding.fix_encoding() | 125 fix_encoding.fix_encoding() |
| 125 sys.exit(main()) | 126 sys.exit(main()) |
| OLD | NEW |