Lambdas, statements, typechecking lambdas

This commit is contained in:
Krzosa Karol
2022-05-25 14:44:30 +02:00
parent 9dc2e1588d
commit b945f3affd
8 changed files with 229 additions and 118 deletions

View File

@@ -5,7 +5,7 @@ global S32 global_indent;
function void
gen_indent(){
for(S32 i = 0; i < global_indent; i++) gen(" ");
for(S32 i = 0; i < global_indent; i++) gen(" ");
}
// @todo: Gen complicated decl
@@ -19,31 +19,31 @@ gen_indent(){
function void
gen_simple_decl_prefix(Ast_Resolved_Type *ast){
switch(ast->kind){
case TYPE_Int: gen("int "); break;
case TYPE_Bool: gen("bool "); break;
case TYPE_Unsigned: gen("unsigned "); break;
case TYPE_String: gen("String "); break;
case TYPE_Void: gen("void "); break;
case TYPE_Pointer:{gen_simple_decl_prefix(ast->base); gen("*");} break;
case TYPE_Array: gen_simple_decl_prefix(ast->base); break;
case TYPE_Lambda:break;
invalid_default_case;
switch(ast->kind){
case TYPE_Int: gen("int "); break;
case TYPE_Bool: gen("bool "); break;
case TYPE_Unsigned: gen("unsigned "); break;
case TYPE_String: gen("String "); break;
case TYPE_Void: gen("void "); break;
case TYPE_Pointer:{gen_simple_decl_prefix(ast->base); gen("*");} break;
case TYPE_Array: gen_simple_decl_prefix(ast->base); break;
case TYPE_Lambda:break;
invalid_default_case;
}
}
function void
gen_simple_decl_postfix(Ast_Resolved_Type *ast){
switch(ast->kind){
case TYPE_Int: break;
case TYPE_Bool: break;
case TYPE_Unsigned: break;
case TYPE_String: break;
case TYPE_Void: break;
case TYPE_Pointer: gen_simple_decl_postfix(ast->base); break;
case TYPE_Array: gen("[%d]", (int)ast->arr.size); gen_simple_decl_postfix(ast->arr.base); break;
case TYPE_Lambda:break;
invalid_default_case;
switch(ast->kind){
case TYPE_Int: break;
case TYPE_Bool: break;
case TYPE_Unsigned: break;
case TYPE_String: break;
case TYPE_Void: break;
case TYPE_Pointer: gen_simple_decl_postfix(ast->base); break;
case TYPE_Array: gen("[%d]", (int)ast->arr.size); gen_simple_decl_postfix(ast->arr.base); break;
case TYPE_Lambda:break;
invalid_default_case;
}
}
@@ -93,6 +93,15 @@ gen_expr(Ast_Expr *ast){
Ast_End();
}
Ast_Begin(AST_BINARY, Ast_Binary){
gen("(");
gen_expr(node->left);
gen("%s", token_kind_string(node->op).str);
gen_expr(node->right);
gen(")");
Ast_End();
}
Ast_Begin(AST_UNARY, Ast_Unary){
switch(node->op){
case TK_Pointer: {
@@ -157,13 +166,25 @@ gen_ast(Ast *ast){
switch(ast->kind){
Ast_Begin(AST_PACKAGE, Ast_Package){
For(node->decls) gen_ast(*it);
For(node->decls) {
genln("");
gen_ast(*it);
}
Ast_End();
}
Ast_Begin(AST_RETURN, Ast_Return){
gen("return");
if(node->expr){
gen(" ");
gen_expr(node->expr);
}
gen(";");
Ast_End();
}
Ast_Begin(AST_VAR, Ast_Decl){
genln("");
Sym *sym = sym_get(node->name);
Sym *sym = resolved_get(node);
gen_simple_decl(sym->type, node->name);
if(node->var.expr){
gen(" = ");
@@ -179,7 +200,7 @@ gen_ast(Ast *ast){
if(sym->type->kind == TYPE_Lambda){
if(node->var.expr->kind == AST_LAMBDA){
Ast_Lambda *lambda = (Ast_Lambda *)node->var.expr;
genln("");
gen("static ");
gen_simple_decl(lambda->ret->resolved_type, node->name);
gen("(");
For(lambda->args){
@@ -188,9 +209,20 @@ gen_ast(Ast *ast){
if(it != (lambda->args.end() - 1)) gen(", ");
}
gen(")");
if(lambda->block) {
gen("{");
global_indent++;
For(lambda->block->stmts) {
genln("");
gen_ast(it[0]);
}
global_indent--;
genln("}");
}
else gen(";");
}
else{
genln("");
gen_simple_decl(sym->type, node->name);
gen(" = ");
gen_expr(node->var.expr);
@@ -198,10 +230,10 @@ gen_ast(Ast *ast){
}
}
else if(sym->type == type_int){
genln("enum { %s = %lld };", node->name.str, sym->int_val);
gen("enum { %s = %lld };", node->name.str, sym->int_val);
}
else if(sym->type == type_string){
genln("String %s = LIT(\"%s\");", node->name.str, sym->intern_val.str);
gen("String %s = LIT(\"%s\");", node->name.str, sym->intern_val.str);
}
else{
parsing_error(node->pos, "Unhandled type of constant expression");
@@ -217,7 +249,7 @@ gen_ast(Ast *ast){
function void
test_gen(){
TEST_PARSER();
String filename = "test3.kl"_s;
String filename = "globals.kl"_s;
String file_content = os_read_file(scratch, filename);
lex_restream(&ctx, file_content, filename);
Ast_Package *result = parse_file();