92 lines
2.3 KiB
C++
92 lines
2.3 KiB
C++
#pragma once
|
|
#include <memory>
|
|
#include <vector>
|
|
#include <map>
|
|
#include <set>
|
|
#include "instruction.hpp"
|
|
#include "llvm_context.hpp"
|
|
#include "parsed_type.hpp"
|
|
#include "type_env.hpp"
|
|
#include "location.hh"
|
|
#include "global_scope.hpp"
|
|
|
|
struct ast;
|
|
using ast_ptr = std::unique_ptr<ast>;
|
|
|
|
struct constructor {
|
|
std::string name;
|
|
std::vector<parsed_type_ptr> types;
|
|
int8_t tag;
|
|
|
|
constructor(std::string n, std::vector<parsed_type_ptr> ts)
|
|
: name(std::move(n)), types(std::move(ts)) {}
|
|
};
|
|
|
|
using constructor_ptr = std::unique_ptr<constructor>;
|
|
|
|
struct definition_defn {
|
|
std::string name;
|
|
std::vector<std::string> params;
|
|
ast_ptr body;
|
|
yy::location loc;
|
|
|
|
type_env_ptr env;
|
|
type_env_ptr var_env;
|
|
std::set<std::string> free_variables;
|
|
std::set<std::string> nearby_variables;
|
|
type_ptr full_type;
|
|
type_ptr return_type;
|
|
|
|
definition_defn(
|
|
std::string n,
|
|
std::vector<std::string> p,
|
|
ast_ptr b,
|
|
yy::location l = yy::location())
|
|
: name(std::move(n)), params(std::move(p)), body(std::move(b)), loc(std::move(l)) {
|
|
|
|
}
|
|
|
|
void find_free();
|
|
void insert_types(type_mgr& mgr, type_env_ptr& env, visibility v);
|
|
void typecheck(type_mgr& mgr);
|
|
|
|
global_function& into_global(global_scope& scope);
|
|
};
|
|
|
|
using definition_defn_ptr = std::unique_ptr<definition_defn>;
|
|
|
|
struct definition_data {
|
|
std::string name;
|
|
std::vector<std::string> vars;
|
|
std::vector<constructor_ptr> constructors;
|
|
yy::location loc;
|
|
|
|
type_env_ptr env;
|
|
|
|
definition_data(
|
|
std::string n,
|
|
std::vector<std::string> vs,
|
|
std::vector<constructor_ptr> cs,
|
|
yy::location l = yy::location())
|
|
: name(std::move(n)), vars(std::move(vs)), constructors(std::move(cs)), loc(std::move(l)) {}
|
|
|
|
void insert_types(type_env_ptr& env);
|
|
void insert_constructors() const;
|
|
|
|
void into_globals(global_scope& scope);
|
|
};
|
|
|
|
using definition_data_ptr = std::unique_ptr<definition_data>;
|
|
|
|
struct definition_group {
|
|
std::map<std::string, definition_data_ptr> defs_data;
|
|
std::map<std::string, definition_defn_ptr> defs_defn;
|
|
visibility vis;
|
|
type_env_ptr env;
|
|
|
|
definition_group(visibility v = visibility::local) : vis(v) {}
|
|
|
|
void find_free(std::set<std::string>& into);
|
|
void typecheck(type_mgr& mgr, type_env_ptr& env);
|
|
};
|