1d26200423ee818e54d4088bd0c499caf840d866dManman Ren//===- ProfileDataLoader.h - Load & convert profile info ----*- C++ -*-===// 2d26200423ee818e54d4088bd0c499caf840d866dManman Ren// 3d26200423ee818e54d4088bd0c499caf840d866dManman Ren// The LLVM Compiler Infrastructure 4d26200423ee818e54d4088bd0c499caf840d866dManman Ren// 5d26200423ee818e54d4088bd0c499caf840d866dManman Ren// This file is distributed under the University of Illinois Open Source 6d26200423ee818e54d4088bd0c499caf840d866dManman Ren// License. See LICENSE.TXT for details. 7d26200423ee818e54d4088bd0c499caf840d866dManman Ren// 8d26200423ee818e54d4088bd0c499caf840d866dManman Ren//===----------------------------------------------------------------------===// 9d26200423ee818e54d4088bd0c499caf840d866dManman Ren// 10d26200423ee818e54d4088bd0c499caf840d866dManman Ren// The ProfileDataLoader class is used to load profiling data from a dump file. 11d26200423ee818e54d4088bd0c499caf840d866dManman Ren// The ProfileDataT<FType, BType> class is used to store the mapping of this 12d26200423ee818e54d4088bd0c499caf840d866dManman Ren// data to control flow edges. 13d26200423ee818e54d4088bd0c499caf840d866dManman Ren// 14d26200423ee818e54d4088bd0c499caf840d866dManman Ren//===----------------------------------------------------------------------===// 15d26200423ee818e54d4088bd0c499caf840d866dManman Ren 16d26200423ee818e54d4088bd0c499caf840d866dManman Ren#ifndef LLVM_ANALYSIS_PROFILEDATALOADER_H 17d26200423ee818e54d4088bd0c499caf840d866dManman Ren#define LLVM_ANALYSIS_PROFILEDATALOADER_H 18d26200423ee818e54d4088bd0c499caf840d866dManman Ren 19d04a8d4b33ff316ca4cf961e06c9e312eff8e64fChandler Carruth#include "llvm/ADT/ArrayRef.h" 20f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling#include "llvm/ADT/DenseMap.h" 21f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling#include "llvm/ADT/SmallVector.h" 22d26200423ee818e54d4088bd0c499caf840d866dManman Ren#include "llvm/Support/Debug.h" 23d26200423ee818e54d4088bd0c499caf840d866dManman Ren#include "llvm/Support/ErrorHandling.h" 24d26200423ee818e54d4088bd0c499caf840d866dManman Ren#include <string> 25d26200423ee818e54d4088bd0c499caf840d866dManman Ren 26d26200423ee818e54d4088bd0c499caf840d866dManman Rennamespace llvm { 27d26200423ee818e54d4088bd0c499caf840d866dManman Ren 28d26200423ee818e54d4088bd0c499caf840d866dManman Renclass ModulePass; 29d26200423ee818e54d4088bd0c499caf840d866dManman Renclass Function; 30d26200423ee818e54d4088bd0c499caf840d866dManman Renclass BasicBlock; 31d26200423ee818e54d4088bd0c499caf840d866dManman Ren 32cb5f63d7fa717b67a666712a3a0d7eebd4d8fb8bBenjamin Kramer// Helper for dumping edges to dbgs(). 33d26200423ee818e54d4088bd0c499caf840d866dManman Renraw_ostream& operator<<(raw_ostream &O, std::pair<const BasicBlock *, 34d26200423ee818e54d4088bd0c499caf840d866dManman Ren const BasicBlock *> E); 35d26200423ee818e54d4088bd0c499caf840d866dManman Ren 36d26200423ee818e54d4088bd0c499caf840d866dManman Ren/// \brief The ProfileDataT<FType, BType> class is used to store the mapping of 37d26200423ee818e54d4088bd0c499caf840d866dManman Ren/// profiling data to control flow edges. 38d26200423ee818e54d4088bd0c499caf840d866dManman Ren/// 39d26200423ee818e54d4088bd0c499caf840d866dManman Ren/// An edge is defined by its source and sink basic blocks. 40d26200423ee818e54d4088bd0c499caf840d866dManman Rentemplate<class FType, class BType> 41d26200423ee818e54d4088bd0c499caf840d866dManman Renclass ProfileDataT { 42cb5f63d7fa717b67a666712a3a0d7eebd4d8fb8bBenjamin Kramerpublic: 43d26200423ee818e54d4088bd0c499caf840d866dManman Ren // The profiling information defines an Edge by its source and sink basic 44d26200423ee818e54d4088bd0c499caf840d866dManman Ren // blocks. 45d26200423ee818e54d4088bd0c499caf840d866dManman Ren typedef std::pair<const BType*, const BType*> Edge; 46d26200423ee818e54d4088bd0c499caf840d866dManman Ren 47cb5f63d7fa717b67a666712a3a0d7eebd4d8fb8bBenjamin Kramerprivate: 48f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling typedef DenseMap<Edge, unsigned> EdgeWeights; 49d26200423ee818e54d4088bd0c499caf840d866dManman Ren 50d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// \brief Count the number of times a transition between two blocks is 51d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// executed. 52d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// 53d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// As a special case, we also hold an edge from the null BasicBlock to the 54d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// entry block to indicate how many times the function was entered. 55f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling DenseMap<const FType*, EdgeWeights> EdgeInformation; 56d26200423ee818e54d4088bd0c499caf840d866dManman Ren 57cb5f63d7fa717b67a666712a3a0d7eebd4d8fb8bBenjamin Kramerpublic: 58d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// getFunction() - Returns the Function for an Edge. 59f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling static const FType *getFunction(Edge e) { 60d26200423ee818e54d4088bd0c499caf840d866dManman Ren // e.first may be NULL 61f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling assert(((!e.first) || (e.first->getParent() == e.second->getParent())) 62d26200423ee818e54d4088bd0c499caf840d866dManman Ren && "A ProfileData::Edge can not be between two functions"); 63d26200423ee818e54d4088bd0c499caf840d866dManman Ren assert(e.second && "A ProfileData::Edge must have a real sink"); 64d26200423ee818e54d4088bd0c499caf840d866dManman Ren return e.second->getParent(); 65d26200423ee818e54d4088bd0c499caf840d866dManman Ren } 66d26200423ee818e54d4088bd0c499caf840d866dManman Ren 67d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// getEdge() - Creates an Edge between two BasicBlocks. 68d26200423ee818e54d4088bd0c499caf840d866dManman Ren static Edge getEdge(const BType *Src, const BType *Dest) { 69f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling return Edge(Src, Dest); 70d26200423ee818e54d4088bd0c499caf840d866dManman Ren } 71d26200423ee818e54d4088bd0c499caf840d866dManman Ren 72d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// getEdgeWeight - Return the number of times that a given edge was 73d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// executed. 74d26200423ee818e54d4088bd0c499caf840d866dManman Ren unsigned getEdgeWeight(Edge e) const { 75d26200423ee818e54d4088bd0c499caf840d866dManman Ren const FType *f = getFunction(e); 76f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling assert((EdgeInformation.find(f) != EdgeInformation.end()) 77d26200423ee818e54d4088bd0c499caf840d866dManman Ren && "No profiling information for function"); 78d26200423ee818e54d4088bd0c499caf840d866dManman Ren EdgeWeights weights = EdgeInformation.find(f)->second; 79d26200423ee818e54d4088bd0c499caf840d866dManman Ren 80f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling assert((weights.find(e) != weights.end()) 81d26200423ee818e54d4088bd0c499caf840d866dManman Ren && "No profiling information for edge"); 82d26200423ee818e54d4088bd0c499caf840d866dManman Ren return weights.find(e)->second; 83d26200423ee818e54d4088bd0c499caf840d866dManman Ren } 84d26200423ee818e54d4088bd0c499caf840d866dManman Ren 85d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// addEdgeWeight - Add 'weight' to the already stored execution count for 86d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// this edge. 87d26200423ee818e54d4088bd0c499caf840d866dManman Ren void addEdgeWeight(Edge e, unsigned weight) { 88cb5f63d7fa717b67a666712a3a0d7eebd4d8fb8bBenjamin Kramer EdgeInformation[getFunction(e)][e] += weight; 89d26200423ee818e54d4088bd0c499caf840d866dManman Ren } 90d26200423ee818e54d4088bd0c499caf840d866dManman Ren}; 91d26200423ee818e54d4088bd0c499caf840d866dManman Ren 92d26200423ee818e54d4088bd0c499caf840d866dManman Rentypedef ProfileDataT<Function, BasicBlock> ProfileData; 93d26200423ee818e54d4088bd0c499caf840d866dManman Ren//typedef ProfileDataT<MachineFunction, MachineBasicBlock> MachineProfileData; 94d26200423ee818e54d4088bd0c499caf840d866dManman Ren 95d26200423ee818e54d4088bd0c499caf840d866dManman Ren/// The ProfileDataLoader class is used to load raw profiling data from the 96d26200423ee818e54d4088bd0c499caf840d866dManman Ren/// dump file. 97d26200423ee818e54d4088bd0c499caf840d866dManman Renclass ProfileDataLoader { 98d26200423ee818e54d4088bd0c499caf840d866dManman Renprivate: 99d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// The name of the file where the raw profiling data is stored. 100d26200423ee818e54d4088bd0c499caf840d866dManman Ren const std::string &Filename; 101d26200423ee818e54d4088bd0c499caf840d866dManman Ren 102d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// A vector of the command line arguments used when the target program was 103d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// run to generate profiling data. One entry per program run. 104f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling SmallVector<std::string, 1> CommandLines; 105d26200423ee818e54d4088bd0c499caf840d866dManman Ren 106d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// The raw values for how many times each edge was traversed, values from 107d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// multiple program runs are accumulated. 108f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling SmallVector<unsigned, 32> EdgeCounts; 109d26200423ee818e54d4088bd0c499caf840d866dManman Ren 110d26200423ee818e54d4088bd0c499caf840d866dManman Renpublic: 111d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// ProfileDataLoader ctor - Read the specified profiling data file, exiting 112d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// the program if the file is invalid or broken. 113d26200423ee818e54d4088bd0c499caf840d866dManman Ren ProfileDataLoader(const char *ToolName, const std::string &Filename); 114d26200423ee818e54d4088bd0c499caf840d866dManman Ren 115d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// A special value used to represent the weight of an edge which has not 116d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// been counted yet. 117d26200423ee818e54d4088bd0c499caf840d866dManman Ren static const unsigned Uncounted; 118d26200423ee818e54d4088bd0c499caf840d866dManman Ren 119d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// getNumExecutions - Return the number of times the target program was run 120d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// to generate this profiling data. 121d26200423ee818e54d4088bd0c499caf840d866dManman Ren unsigned getNumExecutions() const { return CommandLines.size(); } 122d26200423ee818e54d4088bd0c499caf840d866dManman Ren 123d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// getExecution - Return the command line parameters used to generate the 124d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// i'th set of profiling data. 125f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling const std::string &getExecution(unsigned i) const { return CommandLines[i]; } 126d26200423ee818e54d4088bd0c499caf840d866dManman Ren 127f91e400b2125067e75e0f2159bca7ffd0cf109a0Bill Wendling const std::string &getFileName() const { return Filename; } 128d26200423ee818e54d4088bd0c499caf840d866dManman Ren 129d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// getRawEdgeCounts - Return the raw profiling data, this is just a list of 130d26200423ee818e54d4088bd0c499caf840d866dManman Ren /// numbers with no mappings to edges. 131cb5f63d7fa717b67a666712a3a0d7eebd4d8fb8bBenjamin Kramer ArrayRef<unsigned> getRawEdgeCounts() const { return EdgeCounts; } 132d26200423ee818e54d4088bd0c499caf840d866dManman Ren}; 133d26200423ee818e54d4088bd0c499caf840d866dManman Ren 134d26200423ee818e54d4088bd0c499caf840d866dManman Ren/// createProfileMetadataLoaderPass - This function returns a Pass that loads 135d26200423ee818e54d4088bd0c499caf840d866dManman Ren/// the profiling information for the module from the specified filename. 136d26200423ee818e54d4088bd0c499caf840d866dManman RenModulePass *createProfileMetadataLoaderPass(const std::string &Filename); 137d26200423ee818e54d4088bd0c499caf840d866dManman Ren 138d26200423ee818e54d4088bd0c499caf840d866dManman Ren} // End llvm namespace 139d26200423ee818e54d4088bd0c499caf840d866dManman Ren 140d26200423ee818e54d4088bd0c499caf840d866dManman Ren#endif 141