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 466deff76601eefca7d9ba8cbf9eda8d6020b945..d7162c46882d9bec025244e6b79bb6286b1a02fd 100644 (file)
--- a/tools/llvm-cov/llvm-cov.cpp
+++ b/tools/llvm-cov/llvm-cov.cpp
@@ -1,4 +1,4 @@
-//===- tools/llvm-cov/llvm-cov.cpp - LLVM coverage tool -------------------===//
+//===- llvm-cov.cpp - LLVM coverage tool ----------------------------------===//
 //
 //                     The LLVM Compiler Infrastructure
 //
@@ -7,31 +7,63 @@
 //
 //===----------------------------------------------------------------------===//
 //
-//
 // llvm-cov is a command line tools to analyze and report coverage information.
 //
-//
 //===----------------------------------------------------------------------===//
 
-#include "GCOVReader.h"
 #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"
 using namespace llvm;
 
-static cl::opt<bool>
-DumpGCOV("dump", cl::init(false), cl::desc("dump gcov file"));
+static cl::opt<std::string> SourceFile(cl::Positional, cl::Required,
+                                       cl::desc("SOURCEFILE"));
+
+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>
-InputGCNO("gcno", cl::desc("<input gcno file>"), cl::init(""));
+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<std::string>
-InputGCDA("gcda", cl::desc("<input gcda file>"), cl::init(""));
+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),
+                              cl::desc("Dump the gcov file to stderr"));
+static cl::opt<std::string> InputGCNO("gcno", cl::cat(DebugCat), cl::init(""),
+                                      cl::desc("Override inferred gcno file"));
+static cl::opt<std::string> InputGCDA("gcda", cl::cat(DebugCat), cl::init(""),
+                                      cl::desc("Override inferred gcda file"));
 
 //===----------------------------------------------------------------------===//
 int main(int argc, char **argv) {
@@ -40,44 +72,54 @@ int main(int argc, char **argv) {
   PrettyStackTraceProgram X(argc, argv);
   llvm_shutdown_obj Y;  // Call llvm_shutdown() on exit.
 
-  cl::ParseCommandLineOptions(argc, argv, "llvm cov\n");
+  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));
 
-  GCOVFile GF;
   if (InputGCNO.empty())
-    errs() << " " << argv[0] << ": No gcov input file!\n";
+    InputGCNO = (CoverageFileStem.str() + ".gcno").str();
+  if (InputGCDA.empty())
+    InputGCDA = (CoverageFileStem.str() + ".gcda").str();
+
+  GCOVFile GF;
 
-  OwningPtr<MemoryBuffer> Buff;
-  if (error_code ec = MemoryBuffer::getFileOrSTDIN(InputGCNO, Buff)) {
+  OwningPtr<MemoryBuffer> GCNO_Buff;
+  if (error_code ec = MemoryBuffer::getFileOrSTDIN(InputGCNO, GCNO_Buff)) {
     errs() << InputGCNO << ": " << ec.message() << "\n";
     return 1;
   }
-  GCOVBuffer GB(Buff.take());
-
-  if (!GF.read(GB)) {
+  GCOVBuffer GCNO_GB(GCNO_Buff.get());
+  if (!GF.readGCNO(GCNO_GB)) {
     errs() << "Invalid .gcno File!\n";
     return 1;
   }
 
-  if (!InputGCDA.empty()) {
-    OwningPtr<MemoryBuffer> Buff2;
-    if (error_code ec = MemoryBuffer::getFileOrSTDIN(InputGCDA, Buff2)) {
+  OwningPtr<MemoryBuffer> GCDA_Buff;
+  if (error_code ec = MemoryBuffer::getFileOrSTDIN(InputGCDA, GCDA_Buff)) {
+    if (ec != errc::no_such_file_or_directory) {
       errs() << InputGCDA << ": " << ec.message() << "\n";
       return 1;
     }
-    GCOVBuffer GB2(Buff2.take());
-    
-    if (!GF.read(GB2)) {
+    // 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();
 
-  FileInfo FI;
+  GCOVOptions Options(AllBlocks, BranchProb, BranchCount, FuncSummary,
+                      PreservePaths, UncondBranch);
+  FileInfo FI(Options);
   GF.collectLineCounts(FI);
+  FI.print(InputGCNO, InputGCDA);
   return 0;
 }