CallGraphBuilder.h

Go to the documentation of this file.

#include "ASTVisitor.h"
#include "CFAnalyzer.h"
#include "treetypes/AST.h"
#include <map>
#include <ostream>
#include <set>
 
class CallGraphBuilder : ASTVisitor {
public:
  static CallGraphBuilder build(ASTProgram *ast, CFAnalyzer cfa);
  bool visit(ASTFunction *element) override;
  bool visit(ASTFunAppExpr *element) override;
 
  std::map<ASTFunction *, std::set<ASTFunction *>> getCallGraph();
 
  std::map<ASTFunAppExpr *, std::set<ASTFunction *>> getMayCall();
 
  std::map<std::string, ASTFunction *> getFunMap();
 
private:
  CallGraphBuilder(CFAnalyzer pass);
  ASTNode *getCanonical(ASTNode *n);
  ASTFunction *cfun;
  CFAnalyzer cfa;
  std::map<ASTFunction *, std::set<ASTFunction *>> graph;
  std::map<ASTFunAppExpr *, std::set<ASTFunction *>> mayCall;
  std::map<std::string, ASTFunction *> fromFunNameToASTFun;
};