SymbolTable.h

Go to the documentation of this file.

#pragma once
 
#include "ASTVisitor.h"
 
#include <map>
#include <vector>
 
class SymbolTable {
  std::map<std::string, std::pair<ASTDeclNode *, bool>> functionNames;
  std::map<ASTDeclNode *, std::map<std::string, ASTDeclNode *>> localNames;
  std::vector<std::string> fieldNames;
 
public:
  SymbolTable(
      std::map<std::string, std::pair<ASTDeclNode *, bool>> fMap,
      std::map<ASTDeclNode *, std::map<std::string, ASTDeclNode *>> lMap,
      std::vector<std::string> fSet)
      : functionNames(fMap), localNames(lMap), fieldNames(fSet) {}
 
  ASTDeclNode *getFunction(std::string s);
 
  bool getPoly(std::string s);
 
  std::vector<ASTDeclNode *> getFunctions();
 
  ASTDeclNode *getLocal(std::string s, ASTDeclNode *f);
 
  std::vector<ASTDeclNode *> getLocals(ASTDeclNode *f);
 
  std::vector<std::string> getFields();
 
  static std::shared_ptr<SymbolTable> build(ASTProgram *p);
 
  void print(std::ostream &os);
};