#include "llvm/ProfileData/SampleProfReader.h"
#include "llvm/ADT/DenseMap.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/IR/Module.h"
#include "llvm/IR/ProfileSummary.h"
#include "llvm/ProfileData/ProfileCommon.h"
#include "llvm/ProfileData/SampleProf.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/Compression.h"
#include "llvm/Support/ErrorOr.h"
#include "llvm/Support/JSON.h"
#include "llvm/Support/LEB128.h"
#include "llvm/Support/LineIterator.h"
#include "llvm/Support/MD5.h"
#include "llvm/Support/MemoryBuffer.h"
#include "llvm/Support/raw_ostream.h"
#include <algorithm>
#include <cstddef>
#include <cstdint>
#include <limits>
#include <memory>
#include <system_error>
#include <vector>

Include dependency graph for SampleProfReader.cpp:

Macros
#define	DEBUG_TYPE "samplepgo-reader"

Enumerations
enum	LineType { LineType::CallSiteProfile, LineType::BodyProfile, LineType::Metadata }

Functions
static void	dumpFunctionProfileJson (const FunctionSamples &S, json::OStream &JOS, bool TopLevel=false)

static bool	ParseHead (const StringRef &Input, StringRef &FName, uint64_t &NumSamples, uint64_t &NumHeadSamples)
	Parse `Input` as function head. More...

static bool	isOffsetLegal (unsigned L)
	Returns true if line offset `L` is legal (only has 16 bits). More...

static bool	parseMetadata (const StringRef &Input, uint64_t &FunctionHash, uint32_t &Attributes)
	Parse `Input` that contains metadata. More...

static bool	ParseLine (const StringRef &Input, LineType &LineTy, uint32_t &Depth, uint64_t &NumSamples, uint32_t &LineOffset, uint32_t &Discriminator, StringRef &CalleeName, DenseMap< StringRef, uint64_t > &TargetCountMap, uint64_t &FunctionHash, uint32_t &Attributes)
	Parse `Input` as line sample. More...

static std::string	getSecFlagsStr (const SecHdrTableEntry &Entry)

static ErrorOr< std::unique_ptr< MemoryBuffer > >	setupMemoryBuffer (const Twine &Filename)
	Prepare a memory buffer for the contents of `Filename`. More...

Variables
static cl::opt< bool >	ProfileIsFSDisciminator ("profile-isfs", cl::Hidden, cl::init(false), cl::desc("Profile uses flow sensitive discriminators"))

Macro Definition Documentation

◆ DEBUG_TYPE

#define DEBUG_TYPE "samplepgo-reader"

Definition at line 50 of file SampleProfReader.cpp.

Enumeration Type Documentation

◆ LineType

enum LineType

strong

Enumerator
CallSiteProfile
BodyProfile
Metadata

Definition at line 195 of file SampleProfReader.cpp.

Function Documentation

◆ dumpFunctionProfileJson()

static void dumpFunctionProfileJson	(	const FunctionSamples &	S,
		json::OStream &	JOS,
		bool	TopLevel = `false`
	)

static

Definition at line 76 of file SampleProfReader.cpp.

References llvm::json::OStream::attribute(), llvm::json::OStream::attributeArray(), llvm::sampleprof::LineLocation::Discriminator, llvm::X86AS::FS, llvm::sampleprof::SampleRecord::getSamples(), llvm::sampleprof::SampleRecord::getSortedCallTargets(), I, llvm::sampleprof::LineLocation::LineOffset, llvm::json::OStream::object(), and S.

Referenced by llvm::sampleprof::SampleProfileReader::dumpJson().

◆ getSecFlagsStr()

static std::string getSecFlagsStr ( const SecHdrTableEntry & Entry )

static

Definition at line 1326 of file SampleProfReader.cpp.

◆ isOffsetLegal()

static bool isOffsetLegal ( unsigned L )

static

Returns true if line offset L is legal (only has 16 bits).

Definition at line 171 of file SampleProfReader.cpp.

Referenced by ParseLine(), llvm::sampleprof::SampleProfileReaderExtBinaryBase::readCSNameTableSec(), and llvm::sampleprof::SampleProfileReaderBinary::readProfile().

◆ ParseHead()

static bool ParseHead	(	const StringRef &	Input,
		StringRef &	FName,
		uint64_t &	NumSamples,
		uint64_t &	NumHeadSamples
	)

static

Parse Input as function head.

Parse one line of Input, and update function name in FName, function's total sample count in NumSamples, function's entry count in NumHeadSamples.

Returns: true if parsing is successful.

Definition at line 156 of file SampleProfReader.cpp.

References llvm::StringRef::getAsInteger(), llvm::StringRef::rfind(), and llvm::StringRef::substr().

Referenced by llvm::sampleprof::SampleProfileReaderText::hasFormat(), and llvm::sampleprof::SampleProfileReaderText::readImpl().

◆ ParseLine()

static bool ParseLine	(	const StringRef &	Input,
		LineType &	LineTy,
		uint32_t &	Depth,
		uint64_t &	NumSamples,
		uint32_t &	LineOffset,
		uint32_t &	Discriminator,
		StringRef &	CalleeName,
		DenseMap< StringRef, uint64_t > &	TargetCountMap,
		uint64_t &	FunctionHash,
		uint32_t &	Attributes
	)

static

Parse Input as line sample.

Parameters

Input	input line.
LineTy	Type of this line.
Depth	the depth of the inline stack.
NumSamples	total samples of the line/inlined callsite.
LineOffset	line offset to the start of the function.
Discriminator	discriminator of the line.
TargetCountMap	map from indirect call target to count.
FunctionHash	the function's CFG hash, used by pseudo probe.

returns true if parsing is successful.

Definition at line 213 of file SampleProfReader.cpp.

References Attributes, BodyProfile, CallSiteProfile, llvm::count(), llvm::Depth, llvm::StringRef::find(), llvm::StringRef::find_first_not_of(), llvm::StringRef::find_first_of(), llvm::StringRef::find_last_of(), llvm::StringRef::getAsInteger(), isDigit(), isOffsetLegal(), Metadata, llvm::StringRef::npos, parseMetadata(), llvm::StringRef::size(), and llvm::StringRef::substr().

Referenced by llvm::sampleprof::SampleProfileReaderText::readImpl().

◆ parseMetadata()

static bool parseMetadata	(	const StringRef &	Input,
		uint64_t &	FunctionHash,
		uint32_t &	Attributes
	)

static

Parse Input that contains metadata.

Possible metadata:

CFG Checksum information: !CFGChecksum: 12345
CFG Checksum information: !Attributes: 1 Stores the FunctionHash (a.k.a. CFG Checksum) into FunctionHash.

Definition at line 180 of file SampleProfReader.cpp.

References Attributes, llvm::StringRef::getAsInteger(), llvm::StringRef::startswith(), llvm::StringRef::substr(), and llvm::StringRef::trim().

Referenced by llvm::MetadataLoader::parseFunctionMetadata(), ParseLine(), and llvm::MetadataLoader::parseModuleMetadata().