mirror of
https://github.com/capstone-engine/llvm-capstone.git
synced 2024-10-10 12:55:00 +00:00
547ebad0b9
Add two options to the code coverage artifact prep script:
* --use-existing-profdata: Use an existing indexed profile instead of
merging the same profiles again.
* --restrict: Restrict the coverage reporting to the given list of
source directories.
With this in place, we can teach the coverage bot how to prepare
separate reports for each of the llvm tools.
llvm-svn: 282204
81 lines
3.5 KiB
Python
81 lines
3.5 KiB
Python
#!/usr/bin/env python
|
|
|
|
'''Prepare a code coverage artifact.
|
|
|
|
- Collate raw profiles into one indexed profile.
|
|
- Generate html reports for the given binaries.
|
|
'''
|
|
|
|
import argparse
|
|
import glob
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
def merge_raw_profiles(host_llvm_profdata, profile_data_dir, preserve_profiles):
|
|
print ':: Merging raw profiles...',
|
|
sys.stdout.flush()
|
|
raw_profiles = glob.glob(os.path.join(profile_data_dir, '*.profraw'))
|
|
manifest_path = os.path.join(profile_data_dir, 'profiles.manifest')
|
|
profdata_path = os.path.join(profile_data_dir, 'Coverage.profdata')
|
|
with open(manifest_path, 'w') as manifest:
|
|
manifest.write('\n'.join(raw_profiles))
|
|
subprocess.check_call([host_llvm_profdata, 'merge', '-sparse', '-f',
|
|
manifest_path, '-o', profdata_path])
|
|
if not preserve_profiles:
|
|
for raw_profile in raw_profiles:
|
|
os.remove(raw_profile)
|
|
os.remove(manifest_path)
|
|
print 'Done!'
|
|
return profdata_path
|
|
|
|
def prepare_html_report(host_llvm_cov, profile, report_dir, binary,
|
|
restricted_dirs):
|
|
print ':: Preparing html report for {0}...'.format(binary),
|
|
sys.stdout.flush()
|
|
binary_report_dir = os.path.join(report_dir, os.path.basename(binary))
|
|
invocation = [host_llvm_cov, 'show', binary, '-format', 'html',
|
|
'-instr-profile', profile, '-o', binary_report_dir,
|
|
'-show-line-counts-or-regions', '-Xdemangler', 'c++filt',
|
|
'-Xdemangler', '-n'] + restricted_dirs
|
|
subprocess.check_call(invocation)
|
|
with open(os.path.join(binary_report_dir, 'summary.txt'), 'wb') as Summary:
|
|
subprocess.check_call([host_llvm_cov, 'report', binary,
|
|
'-instr-profile', profile], stdout=Summary)
|
|
print 'Done!'
|
|
|
|
def prepare_html_reports(host_llvm_cov, profdata_path, report_dir, binaries,
|
|
restricted_dirs):
|
|
for binary in binaries:
|
|
prepare_html_report(host_llvm_cov, profdata_path, report_dir, binary,
|
|
restricted_dirs)
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument('host_llvm_profdata', help='Path to llvm-profdata')
|
|
parser.add_argument('host_llvm_cov', help='Path to llvm-cov')
|
|
parser.add_argument('profile_data_dir',
|
|
help='Path to the directory containing the raw profiles')
|
|
parser.add_argument('report_dir',
|
|
help='Path to the output directory for html reports')
|
|
parser.add_argument('binaries', metavar='B', type=str, nargs='+',
|
|
help='Path to an instrumented binary')
|
|
parser.add_argument('--preserve-profiles',
|
|
help='Do not delete raw profiles', action='store_true')
|
|
parser.add_argument('--use-existing-profdata',
|
|
help='Specify an existing indexed profile to use')
|
|
parser.add_argument('--restrict', metavar='R', type=str, nargs='*',
|
|
default=[],
|
|
help='Restrict the reporting to the given source paths')
|
|
args = parser.parse_args()
|
|
|
|
if args.use_existing_profdata:
|
|
profdata_path = args.use_existing_profdata
|
|
else:
|
|
profdata_path = merge_raw_profiles(args.host_llvm_profdata,
|
|
args.profile_data_dir,
|
|
args.preserve_profiles)
|
|
|
|
prepare_html_reports(args.host_llvm_cov, profdata_path, args.report_dir,
|
|
args.binaries, args.restrict)
|