TypeConstraintVisitor.h

Go to the documentation of this file.

#pragma once
 
#include "ASTVisitor.h"
#include "ConstraintHandler.h"
#include "SymbolTable.h"
#include "TipType.h"
#include <memory>
#include <set>
#include <stack>
#include <string>
#include <vector>
 
class TypeConstraintVisitor : public ASTVisitor {
public:
  TypeConstraintVisitor() = delete;
 
  TypeConstraintVisitor(SymbolTable *st,
                        std::shared_ptr<ConstraintHandler> handler);
 
  bool visit(ASTFunction *element) override;
  void endVisit(ASTAccessExpr *element) override;
  void endVisit(ASTAllocExpr *element) override;
  void endVisit(ASTAssignStmt *element) override;
  void endVisit(ASTBinaryExpr *element) override;
  void endVisit(ASTDeRefExpr *element) override;
  void endVisit(ASTErrorStmt *element) override;
  void endVisit(ASTFunAppExpr *element) override;
  void endVisit(ASTFunction *element) override;
  void endVisit(ASTIfStmt *element) override;
  void endVisit(ASTInputExpr *element) override;
  void endVisit(ASTNullExpr *element) override;
  void endVisit(ASTNumberExpr *element) override;
  void endVisit(ASTOutputStmt *element) override;
  void endVisit(ASTRecordExpr *element) override;
  void endVisit(ASTRefExpr *element) override;
  void endVisit(ASTWhileStmt *element) override;
 
protected:
  std::shared_ptr<ConstraintHandler> constraintHandler;
  SymbolTable *symbolTable;
  std::shared_ptr<TipType> astToVar(ASTNode *n);
 
private:
  std::stack<ASTDeclNode *> scope;
};