SamplePGO - Add dump routines for LineLocation, SampleRecord and FunctionSamples

[oota-llvm.git] / include / llvm / ProfileData / SampleProf.h

diff --git a/include/llvm/ProfileData/SampleProf.h b/include/llvm/ProfileData/SampleProf.h
index 7b7e6c46cd84bdbd4853c0900802bb3aa12ab8d0..52cbba396a3382c5071700aea56824236dca529f 100644 (file)
--- a/include/llvm/ProfileData/SampleProf.h
+++ b/include/llvm/ProfileData/SampleProf.h
@@ -75,10 +75,22 @@ static inline uint64_t SPVersion() { return 102; }
 /// (e.g., the two post-increment instructions in "if (p) x++; else y++;").
 struct LineLocation {
   LineLocation(uint32_t L, uint32_t D) : LineOffset(L), Discriminator(D) {}
+  void print(raw_ostream &OS) const {
+    OS << LineOffset;
+    if (Discriminator > 0)
+      OS << "." << Discriminator;
+  }
+  void dump() const { print(dbgs()); }
+
   uint32_t LineOffset;
   uint32_t Discriminator;
 };
 
+inline raw_ostream &operator<<(raw_ostream &OS, const LineLocation &Loc) {
+  Loc.print(OS);
+  return OS;
+}
+
 /// Represents the relative location of a callsite.
 ///
 /// Callsite locations are specified by the line offset from the
@@ -89,8 +101,19 @@ struct CallsiteLocation : public LineLocation {
   CallsiteLocation(uint32_t L, uint32_t D, StringRef N)
       : LineLocation(L, D), CalleeName(N) {}
   StringRef CalleeName;
+
+  void print(raw_ostream &OS) const {
+    LineLocation::print(OS);
+    OS << ": inlined callee: " << CalleeName;
+  }
+  void dump() const { print(dbgs()); }
 };
 
+inline raw_ostream &operator<<(raw_ostream &OS, const CallsiteLocation &Loc) {
+  Loc.print(OS);
+  return OS;
+}
+
 } // End namespace sampleprof
 
 template <> struct DenseMapInfo<sampleprof::LineLocation> {
@@ -194,11 +217,19 @@ public:
       addCalledTarget(I.first(), I.second);
   }
 
+  void print(raw_ostream &OS, unsigned Indent) const;
+  void dump() const { print(dbgs(), 0); }
+
 private:
   uint64_t NumSamples;
   CallTargetMap CallTargets;
 };
 
+inline raw_ostream &operator<<(raw_ostream &OS, const SampleRecord &Sample) {
+  Sample.print(OS, 0);
+  return OS;
+}
+
 typedef DenseMap<LineLocation, SampleRecord> BodySampleMap;
 class FunctionSamples;
 typedef DenseMap<CallsiteLocation, FunctionSamples> CallsiteSampleMap;
@@ -212,6 +243,7 @@ class FunctionSamples {
 public:
   FunctionSamples() : TotalSamples(0), TotalHeadSamples(0) {}
   void print(raw_ostream &OS = dbgs(), unsigned Indent = 0) const;
+  void dump(void) const { print(); }
   void addTotalSamples(uint64_t Num) { TotalSamples += Num; }
   void addHeadSamples(uint64_t Num) { TotalHeadSamples += Num; }
   void addBodySamples(uint32_t LineOffset, uint32_t Discriminator,
@@ -323,6 +355,11 @@ private:
   CallsiteSampleMap CallsiteSamples;
 };
 
+inline raw_ostream &operator<<(raw_ostream &OS, const FunctionSamples &FS) {
+  FS.print(OS);
+  return OS;
+}
+
 } // end namespace sampleprof
 
 } // end namespace llvm