89 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			89 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			C++
		
	
	
	
//===- FunctionPropertiesAnalysis.cpp - Function Properties Analysis ------===//
 | 
						|
//
 | 
						|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 | 
						|
// See https://llvm.org/LICENSE.txt for license information.
 | 
						|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 | 
						|
//
 | 
						|
//===----------------------------------------------------------------------===//
 | 
						|
//
 | 
						|
// This file defines the FunctionPropertiesInfo and FunctionPropertiesAnalysis
 | 
						|
// classes used to extract function properties.
 | 
						|
//
 | 
						|
//===----------------------------------------------------------------------===//
 | 
						|
 | 
						|
#include "llvm/Analysis/FunctionPropertiesAnalysis.h"
 | 
						|
#include "llvm/IR/Instructions.h"
 | 
						|
 | 
						|
using namespace llvm;
 | 
						|
 | 
						|
FunctionPropertiesInfo
 | 
						|
FunctionPropertiesInfo::getFunctionPropertiesInfo(const Function &F,
 | 
						|
                                                  const LoopInfo &LI) {
 | 
						|
 | 
						|
  FunctionPropertiesInfo FPI;
 | 
						|
 | 
						|
  FPI.Uses = ((!F.hasLocalLinkage()) ? 1 : 0) + F.getNumUses();
 | 
						|
 | 
						|
  for (const auto &BB : F) {
 | 
						|
    ++FPI.BasicBlockCount;
 | 
						|
 | 
						|
    if (const auto *BI = dyn_cast<BranchInst>(BB.getTerminator())) {
 | 
						|
      if (BI->isConditional())
 | 
						|
        FPI.BlocksReachedFromConditionalInstruction += BI->getNumSuccessors();
 | 
						|
    } else if (const auto *SI = dyn_cast<SwitchInst>(BB.getTerminator())) {
 | 
						|
      FPI.BlocksReachedFromConditionalInstruction +=
 | 
						|
          (SI->getNumCases() + (nullptr != SI->getDefaultDest()));
 | 
						|
    }
 | 
						|
 | 
						|
    for (const auto &I : BB) {
 | 
						|
      if (auto *CS = dyn_cast<CallBase>(&I)) {
 | 
						|
        const auto *Callee = CS->getCalledFunction();
 | 
						|
        if (Callee && !Callee->isIntrinsic() && !Callee->isDeclaration())
 | 
						|
          ++FPI.DirectCallsToDefinedFunctions;
 | 
						|
      }
 | 
						|
      if (I.getOpcode() == Instruction::Load) {
 | 
						|
        ++FPI.LoadInstCount;
 | 
						|
      } else if (I.getOpcode() == Instruction::Store) {
 | 
						|
        ++FPI.StoreInstCount;
 | 
						|
      }
 | 
						|
    }
 | 
						|
    // Loop Depth of the Basic Block
 | 
						|
    int64_t LoopDepth;
 | 
						|
    LoopDepth = LI.getLoopDepth(&BB);
 | 
						|
    if (FPI.MaxLoopDepth < LoopDepth)
 | 
						|
      FPI.MaxLoopDepth = LoopDepth;
 | 
						|
  }
 | 
						|
  FPI.TopLevelLoopCount += llvm::size(LI);
 | 
						|
  return FPI;
 | 
						|
}
 | 
						|
 | 
						|
void FunctionPropertiesInfo::print(raw_ostream &OS) const {
 | 
						|
  OS << "BasicBlockCount: " << BasicBlockCount << "\n"
 | 
						|
     << "BlocksReachedFromConditionalInstruction: "
 | 
						|
     << BlocksReachedFromConditionalInstruction << "\n"
 | 
						|
     << "Uses: " << Uses << "\n"
 | 
						|
     << "DirectCallsToDefinedFunctions: " << DirectCallsToDefinedFunctions
 | 
						|
     << "\n"
 | 
						|
     << "LoadInstCount: " << LoadInstCount << "\n"
 | 
						|
     << "StoreInstCount: " << StoreInstCount << "\n"
 | 
						|
     << "MaxLoopDepth: " << MaxLoopDepth << "\n"
 | 
						|
     << "TopLevelLoopCount: " << TopLevelLoopCount << "\n\n";
 | 
						|
}
 | 
						|
 | 
						|
AnalysisKey FunctionPropertiesAnalysis::Key;
 | 
						|
 | 
						|
FunctionPropertiesInfo
 | 
						|
FunctionPropertiesAnalysis::run(Function &F, FunctionAnalysisManager &FAM) {
 | 
						|
  return FunctionPropertiesInfo::getFunctionPropertiesInfo(
 | 
						|
      F, FAM.getResult<LoopAnalysis>(F));
 | 
						|
}
 | 
						|
 | 
						|
PreservedAnalyses
 | 
						|
FunctionPropertiesPrinterPass::run(Function &F, FunctionAnalysisManager &AM) {
 | 
						|
  OS << "Printing analysis results of CFA for function "
 | 
						|
     << "'" << F.getName() << "':"
 | 
						|
     << "\n";
 | 
						|
  AM.getResult<FunctionPropertiesAnalysis>(F).print(OS);
 | 
						|
  return PreservedAnalyses::all();
 | 
						|
}
 |