aboutsummaryrefslogtreecommitdiff
path: root/utils/prepare-code-coverage-artifact.py
diff options
context:
space:
mode:
Diffstat (limited to 'utils/prepare-code-coverage-artifact.py')
-rw-r--r--utils/prepare-code-coverage-artifact.py55
1 files changed, 55 insertions, 0 deletions
diff --git a/utils/prepare-code-coverage-artifact.py b/utils/prepare-code-coverage-artifact.py
new file mode 100644
index 000000000000..e233c2c6efec
--- /dev/null
+++ b/utils/prepare-code-coverage-artifact.py
@@ -0,0 +1,55 @@
+#!/usr/bin/env python
+
+'''Prepare a code coverage artifact.
+
+- Collate raw profiles into one indexed profile.
+- Delete the raw profiles.
+- Copy the coverage mappings in the binaries directory.
+'''
+
+import argparse
+import glob
+import os
+import subprocess
+import sys
+
+def merge_raw_profiles(host_llvm_profdata, profile_data_dir):
+ 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])
+ for raw_profile in raw_profiles:
+ os.remove(raw_profile)
+ print 'Done!'
+
+def extract_covmappings(host_llvm_cov, profile_data_dir, llvm_bin_dir):
+ print ':: Extracting covmappings...',
+ sys.stdout.flush()
+ for prog in os.listdir(llvm_bin_dir):
+ if prog == 'llvm-lit':
+ continue
+ covmapping_path = os.path.join(profile_data_dir,
+ os.path.basename(prog) + '.covmapping')
+ subprocess.check_call([host_llvm_cov, 'convert-for-testing',
+ os.path.join(llvm_bin_dir, prog), '-o',
+ covmapping_path])
+ print 'Done!'
+
+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('llvm_bin_dir',
+ help='Path to the directory containing llvm binaries')
+ args = parser.parse_args()
+
+ merge_raw_profiles(args.host_llvm_profdata, args.profile_data_dir)
+ extract_covmappings(args.host_llvm_cov, args.profile_data_dir,
+ args.llvm_bin_dir)