[llvm-cov] Factor out logic to remove unmapped inputs (NFC)
llvm-svn: 282286
diff --git a/llvm/tools/llvm-cov/CodeCoverage.cpp b/llvm/tools/llvm-cov/CodeCoverage.cpp
index 9230d01..96543bb 100644
--- a/llvm/tools/llvm-cov/CodeCoverage.cpp
+++ b/llvm/tools/llvm-cov/CodeCoverage.cpp
@@ -92,6 +92,9 @@
/// \brief Load the coverage mapping data. Return nullptr if an error occured.
std::unique_ptr<CoverageMapping> load();
+ /// \brief Remove input source files which aren't mapped by \p Coverage.
+ void removeUnmappedInputs(const CoverageMapping &Coverage);
+
/// \brief If a demangler is available, demangle all symbol names.
void demangleSymbols(const CoverageMapping &Coverage);
@@ -332,7 +335,16 @@
if (Mismatched)
warning(utostr(Mismatched) + " functions have mismatched data");
- std::vector<StringRef> CoveredFiles = Coverage.get()->getUniqueSourceFiles();
+ if (!SourceFiles.empty())
+ removeUnmappedInputs(*Coverage);
+
+ demangleSymbols(*Coverage);
+
+ return Coverage;
+}
+
+void CodeCoverageTool::removeUnmappedInputs(const CoverageMapping &Coverage) {
+ std::vector<StringRef> CoveredFiles = Coverage.getUniqueSourceFiles();
auto UncoveredFilesIt = SourceFiles.end();
if (!CompareFilenamesOnly) {
@@ -360,10 +372,6 @@
}
SourceFiles.erase(UncoveredFilesIt, SourceFiles.end());
-
- demangleSymbols(*Coverage);
-
- return Coverage;
}
void CodeCoverageTool::demangleSymbols(const CoverageMapping &Coverage) {