Don't try to build ExceptionDemo on ARM.

[oota-llvm.git] / tools / llvm-cov / llvm-cov.cpp

diff --git a/tools/llvm-cov/llvm-cov.cpp b/tools/llvm-cov/llvm-cov.cpp
index 3f52f94010cbb2cb5adedb670d5fa8a84b01da8e..d7162c46882d9bec025244e6b79bb6286b1a02fd 100644 (file)
--- a/tools/llvm-cov/llvm-cov.cpp
+++ b/tools/llvm-cov/llvm-cov.cpp
@@ -12,10 +12,12 @@
 //===----------------------------------------------------------------------===//
 
 #include "llvm/ADT/OwningPtr.h"
+#include "llvm/ADT/SmallString.h"
 #include "llvm/Support/CommandLine.h"
 #include "llvm/Support/GCOV.h"
 #include "llvm/Support/ManagedStatic.h"
 #include "llvm/Support/MemoryObject.h"
+#include "llvm/Support/Path.h"
 #include "llvm/Support/PrettyStackTrace.h"
 #include "llvm/Support/Signals.h"
 #include "llvm/Support/system_error.h"
@@ -26,20 +28,34 @@ static cl::opt<std::string> SourceFile(cl::Positional, cl::Required,
 
 static cl::opt<bool> AllBlocks("a", cl::init(false),
                                cl::desc("Display all basic blocks"));
+static cl::alias AllBlocksA("all-blocks", cl::aliasopt(AllBlocks));
 
 static cl::opt<bool> BranchProb("b", cl::init(false),
                                 cl::desc("Display branch probabilities"));
+static cl::alias BranchProbA("branch-probabilities", cl::aliasopt(BranchProb));
 
 static cl::opt<bool> BranchCount("c", cl::init(false),
                                  cl::desc("Display branch counts instead "
                                            "of percentages (requires -b)"));
+static cl::alias BranchCountA("branch-counts", cl::aliasopt(BranchCount));
 
 static cl::opt<bool> FuncSummary("f", cl::init(false),
                                  cl::desc("Show coverage for each function"));
+static cl::alias FuncSummaryA("function-summaries", cl::aliasopt(FuncSummary));
+
+static cl::opt<std::string> ObjectDir("o", cl::value_desc("DIR"), cl::init(""),
+                                      cl::desc("Search for objects in DIR"));
+static cl::alias ObjectDirA("object-directory", cl::aliasopt(ObjectDir));
+
+static cl::opt<bool> PreservePaths("p", cl::init(false),
+                                   cl::desc("Preserve path components"));
+static cl::alias PreservePathsA("preserve-paths", cl::aliasopt(PreservePaths));
 
 static cl::opt<bool> UncondBranch("u", cl::init(false),
                                   cl::desc("Display unconditional branch info "
                                            "(requires -b)"));
+static cl::alias UncondBranchA("unconditional-branches",
+                               cl::aliasopt(UncondBranch));
 
 static cl::OptionCategory DebugCat("Internal and debugging options");
 static cl::opt<bool> DumpGCOV("dump", cl::init(false), cl::cat(DebugCat),
@@ -58,10 +74,15 @@ int main(int argc, char **argv) {
 
   cl::ParseCommandLineOptions(argc, argv, "LLVM code coverage tool\n");
 
+  SmallString<128> CoverageFileStem(ObjectDir);
+  if (CoverageFileStem.empty())
+    CoverageFileStem = sys::path::parent_path(SourceFile);
+  sys::path::append(CoverageFileStem, sys::path::stem(SourceFile));
+
   if (InputGCNO.empty())
-    InputGCNO = SourceFile.substr(0, SourceFile.rfind(".")) + ".gcno";
+    InputGCNO = (CoverageFileStem.str() + ".gcno").str();
   if (InputGCDA.empty())
-    InputGCDA = SourceFile.substr(0, SourceFile.rfind(".")) + ".gcda";
+    InputGCDA = (CoverageFileStem.str() + ".gcda").str();
 
   GCOVFile GF;
 
@@ -78,20 +99,25 @@ int main(int argc, char **argv) {
 
   OwningPtr<MemoryBuffer> GCDA_Buff;
   if (error_code ec = MemoryBuffer::getFileOrSTDIN(InputGCDA, GCDA_Buff)) {
-    errs() << InputGCDA << ": " << ec.message() << "\n";
-    return 1;
-  }
-  GCOVBuffer GCDA_GB(GCDA_Buff.get());
-  if (!GF.readGCDA(GCDA_GB)) {
-    errs() << "Invalid .gcda File!\n";
-    return 1;
+    if (ec != errc::no_such_file_or_directory) {
+      errs() << InputGCDA << ": " << ec.message() << "\n";
+      return 1;
+    }
+    // Clear the filename to make it clear we didn't read anything.
+    InputGCDA = "-";
+  } else {
+    GCOVBuffer GCDA_GB(GCDA_Buff.get());
+    if (!GF.readGCDA(GCDA_GB)) {
+      errs() << "Invalid .gcda File!\n";
+      return 1;
+    }
   }
 
   if (DumpGCOV)
     GF.dump();
 
   GCOVOptions Options(AllBlocks, BranchProb, BranchCount, FuncSummary,
-                      UncondBranch);
+                      PreservePaths, UncondBranch);
   FileInfo FI(Options);
   GF.collectLineCounts(FI);
   FI.print(InputGCNO, InputGCDA);