CallGraph.h

Go to the documentation of this file.

#pragma once
 
#include "ASTVisitor.h"
#include "CallGraphBuilder.h"
#include "SymbolTable.h"
#include "treetypes/AST.h"
#include <map>
#include <set>
#include <vector>
 
class CallGraph {
 
  std::vector<ASTFunction *> vertices;
  std::vector<std::pair<ASTFunction *, ASTFunction *>> edges;
  int total_edges;
  int total_vertices;
  std::map<ASTFunction *, std::set<ASTFunction *>> callGraph;
  std::map<std::string, ASTFunction *> fromFunNameToASTFuns;
  std::map<ASTFunAppExpr *, std::set<ASTFunction *>> mayCall;
 
public:
  CallGraph(std::map<ASTFunction *, std::set<ASTFunction *>> cGraph,
            std::map<ASTFunAppExpr *, std::set<ASTFunction *>> mc,
            std::vector<ASTFunction *> funs,
            std::map<std::string, ASTFunction *> fmap)
      : callGraph(cGraph), mayCall(mc), vertices(funs),
        total_vertices(vertices.size()), fromFunNameToASTFuns(fmap) {}
 
  static std::shared_ptr<CallGraph> build(ASTProgram *, SymbolTable *st);
 
  int getTotalVertices();
 
  int getTotalEdges();
 
  std::vector<ASTFunction *> getVertices();
 
  std::vector<std::pair<ASTFunction *, ASTFunction *>> getEdges();
 
  std::set<ASTFunction *> getCalledFuns(ASTFunAppExpr *e);
 
  std::set<ASTFunction *> getCallees(ASTFunction *f);
  std::set<ASTFunction *> getCallees(std::string caller);
 
  std::set<ASTFunction *> getCallers(ASTFunction *f);
  std::set<std::string> getCallers(std::string callee);
 
  void print(std::ostream &os);
 
  bool existEdge(std::string caller, std::string callee);
 
  ASTFunction *getASTFun(std::string f_name);
};