TypeInference.h

Go to the documentation of this file.

#pragma once
 
#include "ASTDeclNode.h"
#include "ASTProgram.h"
#include "CallGraph.h"
#include "SymbolTable.h"
#include "Unifier.h"
#include <memory>
 
class TypeInference {
  SymbolTable *symbols;
 
  // This unifier stores either the monomorphic or polymorphic type inference
  // results
  std::shared_ptr<Unifier> unifier;
 
public:
  TypeInference(SymbolTable *s, std::shared_ptr<Unifier> u)
      : symbols(s), unifier(u) {}
 
  static std::shared_ptr<TypeInference>
  run(ASTProgram *ast, bool polyInf, CallGraph *cg, SymbolTable *symbols);
 
  std::shared_ptr<TipType> getInferredType(ASTDeclNode *node);
 
  void print(std::ostream &os);
};