123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151 |
- #pragma once
- #ifdef __GNUC__
- #pragma GCC diagnostic push
- #pragma GCC diagnostic ignored "-Wunused-parameter"
- #endif
- #ifndef LLVM_ANALYSIS_CALLGRAPHSCCPASS_H
- #define LLVM_ANALYSIS_CALLGRAPHSCCPASS_H
- #include "llvm/ADT/ArrayRef.h"
- #include "llvm/Pass.h"
- #include <vector>
- namespace llvm {
- class CallGraph;
- class CallGraphNode;
- class CallGraphSCC;
- class PMStack;
- class CallGraphSCCPass : public Pass {
- public:
- explicit CallGraphSCCPass(char &pid) : Pass(PT_CallGraphSCC, pid) {}
-
-
- Pass *createPrinterPass(raw_ostream &OS,
- const std::string &Banner) const override;
- using llvm::Pass::doInitialization;
- using llvm::Pass::doFinalization;
-
-
- virtual bool doInitialization(CallGraph &CG) {
- return false;
- }
-
-
-
-
-
-
-
- virtual bool runOnSCC(CallGraphSCC &SCC) = 0;
-
-
- virtual bool doFinalization(CallGraph &CG) {
- return false;
- }
-
- void assignPassManager(PMStack &PMS, PassManagerType PMT) override;
-
- PassManagerType getPotentialPassManagerType() const override {
- return PMT_CallGraphPassManager;
- }
-
-
-
- void getAnalysisUsage(AnalysisUsage &Info) const override;
- protected:
-
-
- bool skipSCC(CallGraphSCC &SCC) const;
- };
- class CallGraphSCC {
- const CallGraph &CG;
- void *Context;
- std::vector<CallGraphNode *> Nodes;
- public:
- CallGraphSCC(CallGraph &cg, void *context) : CG(cg), Context(context) {}
- void initialize(ArrayRef<CallGraphNode *> NewNodes) {
- Nodes.assign(NewNodes.begin(), NewNodes.end());
- }
- bool isSingular() const { return Nodes.size() == 1; }
- unsigned size() const { return Nodes.size(); }
-
-
- void ReplaceNode(CallGraphNode *Old, CallGraphNode *New);
-
-
- void DeleteNode(CallGraphNode *Old);
- using iterator = std::vector<CallGraphNode *>::const_iterator;
- iterator begin() const { return Nodes.begin(); }
- iterator end() const { return Nodes.end(); }
- const CallGraph &getCallGraph() { return CG; }
- };
- void initializeDummyCGSCCPassPass(PassRegistry &);
- class DummyCGSCCPass : public CallGraphSCCPass {
- public:
- static char ID;
- DummyCGSCCPass() : CallGraphSCCPass(ID) {
- PassRegistry &Registry = *PassRegistry::getPassRegistry();
- initializeDummyCGSCCPassPass(Registry);
- }
- bool runOnSCC(CallGraphSCC &SCC) override { return false; }
- void getAnalysisUsage(AnalysisUsage &AU) const override {
- AU.setPreservesAll();
- }
- };
- }
- #endif
- #ifdef __GNUC__
- #pragma GCC diagnostic pop
- #endif
|