blog-static/code/compiler/04/main.cpp

57 lines
1.4 KiB
C++
Raw Normal View History

2019-08-26 18:52:25 -07:00
#include "ast.hpp"
#include "parser.hpp"
#include "type.hpp"
2019-08-26 21:05:44 -07:00
#include <iostream>
2019-08-26 18:52:25 -07:00
void yy::parser::error(const std::string& msg) {
std::cout << "An error occured: " << msg << std::endl;
}
extern std::vector<definition_ptr> program;
void typecheck_program(const std::vector<definition_ptr>& prog) {
type_mgr mgr;
type_env env;
type_ptr int_type = type_ptr(new type_base("Int"));
type_ptr binop_type = type_ptr(new type_arr(
int_type,
type_ptr(new type_arr(int_type, int_type))));
env.bind("+", binop_type);
env.bind("-", binop_type);
env.bind("*", binop_type);
env.bind("/", binop_type);
for(auto& def : prog) {
def->typecheck_first(mgr, env);
}
for(auto& def : prog) {
def->typecheck_second(mgr, env);
}
2019-08-26 21:05:44 -07:00
for(auto& pair : env.names) {
std::cout << pair.first << ": ";
pair.second->print(mgr, std::cout);
std::cout << std::endl;
}
2019-08-26 18:52:25 -07:00
}
int main() {
yy::parser parser;
parser.parse();
2019-08-26 21:05:44 -07:00
for(auto& definition : program) {
definition_defn* def = dynamic_cast<definition_defn*>(definition.get());
if(!def) continue;
std::cout << def->name;
for(auto& param : def->params) std::cout << " " << param;
std::cout << ":" << std::endl;
def->body->print(1, std::cout);
}
2019-08-26 18:52:25 -07:00
typecheck_program(program);
std::cout << program.size() << std::endl;
}