| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright 2017 The Chromium Authors. All rights reserved. | 2 # Copyright 2017 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 """Merge the PGC files generated during the profiling step to the PGD database. | 6 """Merge the PGC files generated during the profiling step to the PGD database. |
| 7 | 7 |
| 8 This is required to workaround a flakyness in pgomgr.exe where it can run out | 8 This is required to workaround a flakyness in pgomgr.exe where it can run out |
| 9 of address space while trying to merge all the PGC files at the same time. | 9 of address space while trying to merge all the PGC files at the same time. |
| 10 """ | 10 """ |
| 11 | 11 |
| 12 import glob | 12 import glob |
| 13 import json | 13 import json |
| 14 import optparse | 14 import optparse |
| 15 import os | 15 import os |
| 16 import subprocess | 16 import subprocess |
| 17 import sys | 17 import sys |
| 18 | 18 |
| 19 | 19 |
| 20 script_dir = os.path.dirname(os.path.realpath(__file__)) | |
| 21 sys.path.insert(0, os.path.join(script_dir, os.pardir)) | |
| 22 | |
| 23 import vs_toolchain | |
| 24 | |
| 25 | |
| 26 def find_pgomgr(chrome_checkout_dir): | 20 def find_pgomgr(chrome_checkout_dir): |
| 27 """Find pgomgr.exe.""" | 21 """Find pgomgr.exe.""" |
| 28 win_toolchain_json_file = os.path.join(chrome_checkout_dir, 'build', | 22 win_toolchain_json_file = os.path.join(chrome_checkout_dir, 'build', |
| 29 'win_toolchain.json') | 23 'win_toolchain.json') |
| 30 if not os.path.exists(win_toolchain_json_file): | 24 if not os.path.exists(win_toolchain_json_file): |
| 31 raise Exception('The toolchain JSON file is missing.') | 25 raise Exception('The toolchain JSON file is missing.') |
| 32 with open(win_toolchain_json_file) as temp_f: | 26 with open(win_toolchain_json_file) as temp_f: |
| 33 toolchain_data = json.load(temp_f) | 27 toolchain_data = json.load(temp_f) |
| 34 if not os.path.isdir(toolchain_data['path']): | 28 if not os.path.isdir(toolchain_data['path']): |
| 35 raise Exception('The toolchain JSON file is invalid.') | 29 raise Exception('The toolchain JSON file is invalid.') |
| 36 | 30 |
| 37 # Always use the x64 version of pgomgr (the x86 one doesn't work on the bot's | 31 # Always use the x64 version of pgomgr (the x86 one doesn't work on the bot's |
| 38 # environment). | 32 # environment). |
| 39 if toolchain_data['version'] == '2015': | 33 pgomgr_dir = os.path.join(toolchain_data['path'], 'VC', 'bin', 'amd64') |
| 40 pgomgr_dir = os.path.join(toolchain_data['path'], 'VC', 'bin', 'amd64') | |
| 41 elif toolchain_data['version'] == '2017': | |
| 42 vc_tools_root = vs_toolchain.FindVCToolsRoot() | |
| 43 pgomgr_dir = os.path.join(vc_tools_root, 'HostX64', 'x64') | |
| 44 | 34 |
| 45 pgomgr_path = os.path.join(pgomgr_dir, 'pgomgr.exe') | 35 pgomgr_path = os.path.join(pgomgr_dir, 'pgomgr.exe') |
| 46 if not os.path.exists(pgomgr_path): | 36 if not os.path.exists(pgomgr_path): |
| 47 raise Exception('pgomgr.exe is missing from %s.' % pgomgr_dir) | 37 raise Exception('pgomgr.exe is missing from %s.' % pgomgr_dir) |
| 48 | 38 |
| 49 return pgomgr_path | 39 return pgomgr_path |
| 50 | 40 |
| 51 | 41 |
| 52 def main(): | 42 def main(): |
| 53 parser = optparse.OptionParser(usage='%prog [options]') | 43 parser = optparse.OptionParser(usage='%prog [options]') |
| (...skipping 59 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 113 ]) | 103 ]) |
| 114 proc = subprocess.Popen(merge_command, stdout=subprocess.PIPE) | 104 proc = subprocess.Popen(merge_command, stdout=subprocess.PIPE) |
| 115 stdout, stderr = proc.communicate() | 105 stdout, stderr = proc.communicate() |
| 116 print stdout | 106 print stdout |
| 117 if proc.returncode != 0: | 107 if proc.returncode != 0: |
| 118 raise Exception('Error while trying to merge the PGC files:\n%s' % stderr) | 108 raise Exception('Error while trying to merge the PGC files:\n%s' % stderr) |
| 119 | 109 |
| 120 | 110 |
| 121 if __name__ == '__main__': | 111 if __name__ == '__main__': |
| 122 sys.exit(main()) | 112 sys.exit(main()) |
| OLD | NEW |