part of type_check_visitor done

This commit is contained in:
ProgramSnail 2023-04-22 19:30:16 +03:00
parent 6fc91aafa0
commit 656f58bcde
14 changed files with 891 additions and 539 deletions

View file

@ -54,7 +54,7 @@ void LinkSymbolsVisitor::Visit(AliasDefinitionStatement* node) {
for (size_t i = 0; i <node->parameters.size(); ++i) {
abstract_types_.DefineType(node->parameters[i], node->parameter_graph_ids_[i]);
}
Visit(node->value.get());
Visitor::Visit(node->value.get());
abstract_types_.ExitContext();
}
@ -68,10 +68,10 @@ void LinkSymbolsVisitor::Visit(VariableDefinitionStatement* node) {
void LinkSymbolsVisitor::Visit(FunctionDeclaration* node) {
abstract_types_.EnterContext();
for (auto& parameter : node->parameters) {
Visit(parameter.get());
Visitor::Visit(parameter.get());
abstract_types_.DefineType(parameter->type, parameter->type_graph_id_);
}
Visit(node->type.get());
Visitor::Visit(node->type.get());
abstract_types_.ExitContext();
}
@ -90,7 +90,7 @@ void LinkSymbolsVisitor::Visit(TypeDefinitionStatement* node) {
}
void LinkSymbolsVisitor::Visit(AbstractTypeDefinitionStatement* node) {
Visit(node->type.get());
Visitor::Visit(node->type.get());
// TODO: can't be used before definition
abstract_types_.DefineType(node->type->type, node->type->type_graph_id_);
}
@ -108,224 +108,54 @@ void LinkSymbolsVisitor::Visit(TypeclassDefinitionStatement* node) {
void LinkSymbolsVisitor::Visit(FunctionDefinition* node) {
for (auto& parameter : node->parameters) {
Visit(parameter.get());
Visitor::Visit(parameter.get());
abstract_types_.DefineType(parameter->type, parameter->type_graph_id_);
}
}
void LinkSymbolsVisitor::Visit(TypeDefinition* node) {
Visit(node->type.get());
Visitor::Visit(node->type.get());
for (auto& parameter : node->parameters) {
Visit(parameter.get());
Visitor::Visit(parameter.get());
abstract_types_.DefineType(parameter->type, parameter->type_graph_id_);
}
}
void LinkSymbolsVisitor::Visit(AnyAnnotatedType* node) {
// TODO check ??
if (!node->typeclasses.empty() > 0) {
for (auto& typeclass : node->typeclasses) {
Visitor::Visit(typeclass);
}
}
}
// Flow control -----------------
void LinkSymbolsVisitor::Visit(MatchCase* node) {
Visitor::Visit(node->value);
if (node->condition.has_value()) {
Visitor::Visit(node->condition.value());
}
if (node->statement.has_value()) {
Visitor::Visit(node->statement.value());
}
}
void LinkSymbolsVisitor::Visit(Match* node) {
Visitor::Visit(node->value);
for (auto& match_case : node->matches) {
Visit(&match_case);
}
}
void LinkSymbolsVisitor::Visit(Condition* node) {
Visitor::Visit(node->conditions[0]);
Visitor::Visit(node->statements[0]);
for (size_t i = 1; i < node->conditions.size(); ++i) {
Visitor::Visit(node->conditions[i]);
Visitor::Visit(node->statements[i]);
}
if (node->statements.size() > node->conditions.size()) {
Visitor::Visit(node->statements[node->conditions.size()]);
}
}
void LinkSymbolsVisitor::Visit(DoWhileLoop* node) {
Visitor::Visit(node->statement);
Visitor::Visit(node->condition);
}
void LinkSymbolsVisitor::Visit(WhileLoop* node) {
Visitor::Visit(node->condition);
Visitor::Visit(node->statement);
}
void LinkSymbolsVisitor::Visit(ForLoop* node) {
Visitor::Visit(node->variable);
Visitor::Visit(node->interval);
Visitor::Visit(node->statement);
}
void LinkSymbolsVisitor::Visit(LoopLoop* node) {
Visitor::Visit(node->statement);
}
// Statements, expressions, blocks, etc. -----------------
void LinkSymbolsVisitor::Visit(Block* node) {
for (auto& statement : node->statements) {
Visitor::Visit(statement);
}
}
void LinkSymbolsVisitor::Visit(ScopedStatement* node) {
Visitor::Visit(node->statement);
}
void LinkSymbolsVisitor::Visit(LoopControlExpression& node) {}
// Operators
void LinkSymbolsVisitor::Visit(BinaryOperatorExpression* node) {
Visitor::Visit(node->left_expression);
Visitor::Visit(node->right_expression);
}
void LinkSymbolsVisitor::Visit(UnaryOperatorExpression* node) {
Visitor::Visit(node->expression);
}
void LinkSymbolsVisitor::Visit(ReferenceExpression* node) {
Visit(node->expression.get());
}
void LinkSymbolsVisitor::Visit(AccessExpression* node) {
Visitor::Visit(node->name.get());
Visitor::Visit(node->id);
}
// Other Expressions
void LinkSymbolsVisitor::Visit(FunctionCallExpression* node) {
Visit(node->name.get());
for (auto& argument : node->arguments) {
Visitor::Visit(argument);
}
}
void LinkSymbolsVisitor::Visit(TupleExpression* node) {
for (auto& expression : node->expressions) {
Visitor::Visit(expression);
}
}
void LinkSymbolsVisitor::Visit(VariantExpression* node) {
for (auto& expression : node->expressions) {
Visitor::Visit(expression);
}
}
void LinkSymbolsVisitor::Visit(ReturnExpression* node) {
Visitor::Visit(node->expression);
}
void LinkSymbolsVisitor::Visit(TypeConstructor* node) {
Visit(node->type.get());
for (auto& parameter : node->parameters) {
Visitor::Visit(std::get<2>(parameter));
}
}
// TODO: move to find_symbols_visitor
void LinkSymbolsVisitor::Visit(LambdaFunction* node) {
abstract_types_.EnterContext();
for (auto& parameter : node->parameters) {
Visit(parameter.get());
Visitor::Visit(parameter.get());
abstract_types_.DefineType(parameter->type, parameter->type_graph_id_);
}
////////////
node->argument_graph_ids_.resize(node->arguments.size());
for (size_t i = 0; i < node->arguments.size(); ++i) {
node->argument_graph_ids_[i] = namespace_visitor_.GetAbstractTypeGraph()->AddVertex();
}
node->return_type_graph_id_ = namespace_visitor_.GetAbstractTypeGraph()->AddVertex();
///////////
Visitor::Visit(node->expression);
abstract_types_.ExitContext();
}
void LinkSymbolsVisitor::Visit(ArrayExpression* node) {
for (auto& element : node->elements) {
Visitor::Visit(element);
}
}
// Name
void LinkSymbolsVisitor::Visit(NameExpression* node) {
for (auto& variable_namespace : node->namespaces) {
Visitor::Visit(variable_namespace);
}
// for (auto& expression : node->expressions) {
// Visit(expression);
// }
}
void LinkSymbolsVisitor::Visit(TupleName* node) {
for (auto& name : node->names) {
Visitor::Visit(name);
}
}
void LinkSymbolsVisitor::Visit(VariantName* node) {
for (auto& name : node->names) {
Visitor::Visit(name);
}
}
void LinkSymbolsVisitor::Visit(AnnotatedName* node) {
if (node->type.has_value()) {
Visitor::Visit(node->type.value());
}
}
// Type, typeclass, etc. -----------------
// Type
void LinkSymbolsVisitor::Visit(FunctionType* node) {
for (auto& type : node->types) {
Visitor::Visit(type);
}
}
void LinkSymbolsVisitor::Visit(TupleType* node) {
for (auto& entity : node->entities) {
Visit(entity.second.get());
}
}
void LinkSymbolsVisitor::Visit(VariantType* node) {
for (auto& constructor : node->constructors) {
if (std::holds_alternative<Constructor>(constructor)) {
// do nothing
} else if (std::holds_alternative<std::unique_ptr<TupleType>>(constructor)) {
Visit(std::get<std::unique_ptr<TupleType>>(constructor).get());
} else {
// error
}
}
}
void LinkSymbolsVisitor::Visit(ParametrizedType* node) {
Visit(node->type_expression.get());
for (auto& parameter : node->parameters) {
Visitor::Visit(parameter);
}
}
void LinkSymbolsVisitor::Visit(TypeExpression* node) {
std::vector<std::string> path;
path.reserve(node->namespaces.size());
@ -361,19 +191,8 @@ void LinkSymbolsVisitor::Visit(TypeExpression* node) {
}
}
void LinkSymbolsVisitor::Visit(ExtendedScopedAnyType* node) {
Visitor::Visit(node->type);
}
// Typeclass
void LinkSymbolsVisitor::Visit(ParametrizedTypeclass* node) {
Visit(node->typeclass_expression.get());
for (auto& parameter : node->parameters) {
Visitor::Visit(parameter);
}
}
void LinkSymbolsVisitor::Visit(TypeclassExpression* node) {
std::vector<std::string> path;
path.reserve(node->namespaces.size());