1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
#!/usr/bin/env python3
# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
""" Merge all the profdata files in PROFDATA_DIR, utilizing the binaries in
BINARIES_DIR, to create a unified coverage report in REPORT_DIR. (If no
REPORT_DIR is provided, defaults to `out/report`.)
* Example usage: merge_all_profdata.py --profdata-dir [PROFDATA_DIR]
--binaries-dir [BINARIES_DIR] [--report-dir [REPORT_DIR]]
"""
import argparse
import logging
import os
import subprocess
def _ParseCommandArguments():
"""Adds and parses relevant arguments for tool commands.
Returns:
A dictionary representing the arguments.
"""
arg_parser = argparse.ArgumentParser()
arg_parser.usage = __doc__
arg_parser.add_argument('--llvm-profdata',
required=True,
type=str,
help='Location of LLVM profdata tool')
arg_parser.add_argument('--profdata-dir',
required=True,
type=str,
help='Directory in which profdata files are stored.')
arg_parser.add_argument(
'--outfile',
type=str,
required=True,
help=('Directory where the coverage report should go. '
'Default is out/report.'))
args = arg_parser.parse_args()
return args
args = _ParseCommandArguments()
targets_to_cover = []
for profdata_file in os.listdir(args.profdata_dir):
targets_to_cover.append(os.path.join(args.profdata_dir, profdata_file))
subprocess_cmd = [args.llvm_profdata, 'merge', '-o', args.outfile]
for target in targets_to_cover:
subprocess_cmd.append(target)
try:
subprocess.check_call(subprocess_cmd)
except:
logging.error("An error occured while merging the profdata.")
exit(1)
|
