[SystemZ] Add test missing from r186148

[oota-llvm.git] / include / llvm / Analysis / ProfileDataLoader.h

diff --git a/include/llvm/Analysis/ProfileDataLoader.h b/include/llvm/Analysis/ProfileDataLoader.h
index 3d15bda364d3e9c45a23795839054c9f3cc52fc8..90097f79951d3b0bdee56acccdb36293938aeb7d 100644 (file)
--- a/include/llvm/Analysis/ProfileDataLoader.h
+++ b/include/llvm/Analysis/ProfileDataLoader.h
@@ -16,11 +16,12 @@
 #ifndef LLVM_ANALYSIS_PROFILEDATALOADER_H
 #define LLVM_ANALYSIS_PROFILEDATALOADER_H
 
+#include "llvm/ADT/ArrayRef.h"
+#include "llvm/ADT/DenseMap.h"
+#include "llvm/ADT/SmallVector.h"
 #include "llvm/Support/Debug.h"
 #include "llvm/Support/ErrorHandling.h"
-#include <vector>
 #include <string>
-#include <map>
 
 namespace llvm {
 
@@ -28,11 +29,9 @@ class ModulePass;
 class Function;
 class BasicBlock;
 
-// Helpers for dumping edges to dbgs().
+// Helper for dumping edges to dbgs().
 raw_ostream& operator<<(raw_ostream &O, std::pair<const BasicBlock *,
                                                   const BasicBlock *> E);
-raw_ostream& operator<<(raw_ostream &O, const BasicBlock *BB);
-raw_ostream& operator<<(raw_ostream &O, const Function *F);
 
 /// \brief The ProfileDataT<FType, BType> class is used to store the mapping of
 /// profiling data to control flow edges.
@@ -40,30 +39,26 @@ raw_ostream& operator<<(raw_ostream &O, const Function *F);
 /// An edge is defined by its source and sink basic blocks.
 template<class FType, class BType>
 class ProfileDataT {
-  public:
+public:
   // The profiling information defines an Edge by its source and sink basic
   // blocks.
   typedef std::pair<const BType*, const BType*> Edge;
 
-  private:
-  typedef std::map<Edge, unsigned> EdgeWeights;
+private:
+  typedef DenseMap<Edge, unsigned> EdgeWeights;
 
   /// \brief Count the number of times a transition between two blocks is
   /// executed.
   ///
   /// As a special case, we also hold an edge from the null BasicBlock to the
   /// entry block to indicate how many times the function was entered.
-  std::map<const FType*, EdgeWeights> EdgeInformation;
-
-  public:
-  static char ID; // Class identification, replacement for typeinfo
-  ProfileDataT() {};
-  ~ProfileDataT() {};
+  DenseMap<const FType*, EdgeWeights> EdgeInformation;
 
+public:
   /// getFunction() - Returns the Function for an Edge.
-  static const FType* getFunction(Edge e) {
+  static const FType *getFunction(Edge e) {
     // e.first may be NULL
-    assert(   ((!e.first) || (e.first->getParent() == e.second->getParent()))
+    assert(((!e.first) || (e.first->getParent() == e.second->getParent()))
            && "A ProfileData::Edge can not be between two functions");
     assert(e.second && "A ProfileData::Edge must have a real sink");
     return e.second->getParent();
@@ -71,18 +66,18 @@ class ProfileDataT {
 
   /// getEdge() - Creates an Edge between two BasicBlocks.
   static Edge getEdge(const BType *Src, const BType *Dest) {
-    return std::make_pair(Src, Dest);
+    return Edge(Src, Dest);
   }
 
   /// getEdgeWeight - Return the number of times that a given edge was
   /// executed.
   unsigned getEdgeWeight(Edge e) const {
     const FType *f = getFunction(e);
-    assert(   (EdgeInformation.find(f) != EdgeInformation.end())
+    assert((EdgeInformation.find(f) != EdgeInformation.end())
            && "No profiling information for function");
     EdgeWeights weights = EdgeInformation.find(f)->second;
 
-    assert(   (weights.find(e) != weights.end())
+    assert((weights.find(e) != weights.end())
            && "No profiling information for edge");
     return weights.find(e)->second;
   }
@@ -90,7 +85,7 @@ class ProfileDataT {
   /// addEdgeWeight - Add 'weight' to the already stored execution count for
   /// this edge.
   void addEdgeWeight(Edge e, unsigned weight) {
-      EdgeInformation[getFunction(e)][e] += weight;
+    EdgeInformation[getFunction(e)][e] += weight;
   }
 };
 
@@ -106,11 +101,11 @@ private:
 
   /// A vector of the command line arguments used when the target program was
   /// run to generate profiling data.  One entry per program run.
-  std::vector<std::string> CommandLines;
+  SmallVector<std::string, 1> CommandLines;
 
   /// The raw values for how many times each edge was traversed, values from
   /// multiple program runs are accumulated.
-  std::vector<unsigned> EdgeCounts;
+  SmallVector<unsigned, 32> EdgeCounts;
 
 public:
   /// ProfileDataLoader ctor - Read the specified profiling data file, exiting
@@ -121,22 +116,19 @@ public:
   /// been counted yet.
   static const unsigned Uncounted;
 
-  /// The maximum value that can be stored in a profiling counter.
-  static const unsigned MaxCount;
-
   /// getNumExecutions - Return the number of times the target program was run
   /// to generate this profiling data.
   unsigned getNumExecutions() const { return CommandLines.size(); }
 
   /// getExecution - Return the command line parameters used to generate the
   /// i'th set of profiling data.
-  const std::string& getExecution(unsigned i) const { return CommandLines[i]; }
+  const std::string &getExecution(unsigned i) const { return CommandLines[i]; }
 
-  const std::string& getFileName() const { return Filename; }
+  const std::string &getFileName() const { return Filename; }
 
   /// getRawEdgeCounts - Return the raw profiling data, this is just a list of
   /// numbers with no mappings to edges.
-  const std::vector<unsigned>& getRawEdgeCounts() const { return EdgeCounts; }
+  ArrayRef<unsigned> getRawEdgeCounts() const { return EdgeCounts; }
 };
 
 /// createProfileMetadataLoaderPass - This function returns a Pass that loads