diff --git a/stdlib/c_generator.krak b/stdlib/c_generator.krak index 348c21c..93ebec1 100644 --- a/stdlib/c_generator.krak +++ b/stdlib/c_generator.krak @@ -57,7 +57,7 @@ obj c_generator (Object) { // add parameters to destructor thingy (for returns)? Or should that be a different pass? // stack-stack thing // this could be a stack of strings too, maybe // start out with one stack on the stack - var defer_stack = stack>(stack<*ast_node>()) + var defer_stack = stack>>(make_pair(false, stack<*ast_node>())) function_definitions += type_to_c(backing.type->return_type) + " " + decorated_name + "(" + parameters + ") {\n" + generate_statement(backing.body_statement, enclosing_object, &defer_stack) // emit parameter destructors? function_definitions += "}\n" @@ -76,7 +76,7 @@ obj c_generator (Object) { top_level_c_passthrough += generate_simple_passthrough(backing.statement->statement.child) } ast_node::simple_passthrough(backing) top_level_c_passthrough += generate_simple_passthrough(child) - ast_node::declaration_statement(backing) variable_declarations += generate_declaration_statement(child, null(), null>>()) + ";\n" + ast_node::declaration_statement(backing) variable_declarations += generate_declaration_statement(child, null(), null>>>()) + ";\n" ast_node::function(backing) { // make sure not a template // or a passthrough @@ -92,7 +92,7 @@ obj c_generator (Object) { type_poset.get_sorted().for_each(fun(vert: *ast_node) { plain_typedefs += string("typedef struct ") + vert->type_def.name + "_dummy " + vert->type_def.name + ";\n" structs += string("struct ") + vert->type_def.name + "_dummy {\n" - vert->type_def.variables.for_each(fun(variable_declaration: *ast_node) structs += generate_declaration_statement(variable_declaration, null(), null>>()) + ";\n";) + vert->type_def.variables.for_each(fun(variable_declaration: *ast_node) structs += generate_declaration_statement(variable_declaration, null(), null>>>()) + ";\n";) structs += "};\n" // generate the methods vert->type_def.methods.for_each(fun(method: *ast_node) generate_function_definition(method, vert);) @@ -102,45 +102,45 @@ obj c_generator (Object) { } fun generate_if_comp(node: *ast_node): string { if (node->if_comp.wanted_generator == "__C__") - return generate(node->if_comp.statement, null(), null>>()) + return generate(node->if_comp.statement, null(), null>>>()) return string("") } fun generate_simple_passthrough(node: *ast_node): string { // deal with all the passthrough params return node->simple_passthrough.passthrough_str } - fun generate_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>): string return generate(node->statement.child, enclosing_object, defer_stack) + ";\n"; - fun generate_declaration_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>): string { + fun generate_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): string return generate(node->statement.child, enclosing_object, defer_stack) + ";\n"; + fun generate_declaration_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): string { // add destruct to defer_stac var identifier = node->declaration_statement.identifier var to_ret = type_to_c(identifier->identifier.type) + " " + identifier->identifier.name - if (node->declaration_statement.expression) to_ret += string(" = ") + generate(node->declaration_statement.expression, enclosing_object, null>>()) + if (node->declaration_statement.expression) to_ret += string(" = ") + generate(node->declaration_statement.expression, enclosing_object, null>>>()) return to_ret } fun generate_assignment_statement(node: *ast_node, enclosing_object: *ast_node): string { - return generate(node->assignment_statement.to, null(), null>>()) + " = " + generate(node->assignment_statement.from, enclosing_object, null>>()) + return generate(node->assignment_statement.to, null(), null>>>()) + " = " + generate(node->assignment_statement.from, enclosing_object, null>>>()) } - fun generate_if_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>): string { - var if_str = string("if (") + generate(node->if_statement.condition, enclosing_object, null>>()) + ") {\n" + generate(node->if_statement.then_part, enclosing_object, defer_stack) + "}" + fun generate_if_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): string { + var if_str = string("if (") + generate(node->if_statement.condition, enclosing_object, null>>>()) + ") {\n" + generate(node->if_statement.then_part, enclosing_object, defer_stack) + "}" if (node->if_statement.else_part) if_str += string(" else {\n") + generate(node->if_statement.else_part, enclosing_object, defer_stack) + "}" return if_str + "\n" } - fun generate_while_loop(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>): string { + fun generate_while_loop(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): string { // stick another stack on - defer_stack->push(stack<*ast_node>()) - var to_ret = string("while (") + generate(node->while_loop.condition, enclosing_object, null>>()) + ") {\n" + generate(node->while_loop.statement, enclosing_object, defer_stack) + defer_stack->push(make_pair(true, stack<*ast_node>())) + var to_ret = string("while (") + generate(node->while_loop.condition, enclosing_object, null>>>()) + ") {\n" + generate(node->while_loop.statement, enclosing_object, defer_stack) to_ret += generate_from_defer_stack(defer_stack, 1, enclosing_object) defer_stack->pop() to_ret += "}\n" return to_ret } - fun generate_for_loop(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>): string { + fun generate_for_loop(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): string { // stick another stack on - defer_stack->push(stack<*ast_node>()) + defer_stack->push(make_pair(true, stack<*ast_node>())) // gotta take off last semicolon - var to_ret = string("for (") + generate(node->for_loop.init, enclosing_object, null>>()) + " " + generate(node->for_loop.condition, enclosing_object, null>>()) + "; " + - generate(node->for_loop.update, enclosing_object, null>>()).slice(0,-3) + ") {\n" + generate(node->for_loop.body, enclosing_object, defer_stack) + var to_ret = string("for (") + generate(node->for_loop.init, enclosing_object, null>>>()) + " " + generate(node->for_loop.condition, enclosing_object, null>>>()) + "; " + + generate(node->for_loop.update, enclosing_object, null>>>()).slice(0,-3) + ") {\n" + generate(node->for_loop.body, enclosing_object, defer_stack) to_ret += generate_from_defer_stack(defer_stack, 1, enclosing_object) defer_stack->pop() to_ret += "}\n" @@ -151,37 +151,47 @@ obj c_generator (Object) { return string("(this->") + node->identifier.name + ")" return node->identifier.name } - fun generate_return_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>): string { + fun generate_return_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): string { // generate all in stac var to_ret = generate_from_defer_stack(defer_stack, -1, enclosing_object) - to_ret += string("return ") + generate(node->return_statement.return_value, enclosing_object, null>>()) + to_ret += string("return ") + generate(node->return_statement.return_value, enclosing_object, null>>>()) return to_ret } - fun generate_branching_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>): string { + fun generate_branching_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): string { // should be up to loop - var to_ret = generate_from_defer_stack(defer_stack, 1, enclosing_object) + var to_ret = generate_from_defer_stack(defer_stack, -2, enclosing_object) match(node->branching_statement.b_type) { branching_type::break_stmt() return to_ret + string("break") branching_type::continue_stmt() return to_ret + string("continue") } } - fun generate_from_defer_stack(defer_stack: *stack>, num: int, enclosing_object: *ast_node): string { + fun generate_from_defer_stack(defer_stack: *stack>>, num: int, enclosing_object: *ast_node): string { var to_ret = string() if (num == -1) num = defer_stack->size() + if (num == -2) { + num = 1 + for (var i = 0; i < defer_stack->size(); i++;) + if (defer_stack->from_top(i).first) + break + else + num++ + print("HERE IS THE THING: "); println(num) + } for (var i = 0; i < num; i++;) - defer_stack->from_top(i).for_each(fun(node: *ast_node) to_ret += generate(node, enclosing_object, null>>());) + defer_stack->from_top(i).second.for_each(fun(node: *ast_node) to_ret += generate(node, enclosing_object, null>>>());) + /*defer_stack->from_top(i).for_each(fun(node: *ast_node) to_ret += generate(node, enclosing_object, null>>());)*/ return to_ret } - fun generate_defer_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>): string { - defer_stack->top().push(node->defer_statement.statement) + fun generate_defer_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): string { + defer_stack->top().second.push(node->defer_statement.statement) return string("/*defer*/") } fun generate_value(node: *ast_node): string return node->value.string_value; - fun generate_code_block(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>): string { + fun generate_code_block(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): string { var to_ret = string("{\n") // stick another stack on - defer_stack->push(stack<*ast_node>()) + defer_stack->push(make_pair(false, stack<*ast_node>())) node->code_block.children.for_each(fun(child: *ast_node) to_ret += generate(child, enclosing_object, defer_stack);) to_ret += generate_from_defer_stack(defer_stack, 1, enclosing_object) defer_stack->pop() @@ -206,13 +216,13 @@ obj c_generator (Object) { is_type_def(get_ast_scope(node->function_call.func->function_call.parameters[1])->get(string("~enclosing_scope"))[0]) if (dot_style_method_call) { - func_name = generate(node->function_call.func->function_call.parameters[1], enclosing_object, null>>()) + func_name = generate(node->function_call.func->function_call.parameters[1], enclosing_object, null>>>()) // don't add & if it was -> if (node->function_call.func->function_call.func->function.name == ".") call_string += string("&") - call_string += generate(node->function_call.func->function_call.parameters[0], enclosing_object, null>>()) + call_string += generate(node->function_call.func->function_call.parameters[0], enclosing_object, null>>>()) } else { // regular style function name - func_name = generate(node->function_call.func, enclosing_object, null>>()) + func_name = generate(node->function_call.func, enclosing_object, null>>>()) } // handle method call from inside method of same object if (!dot_style_method_call && enclosing_object && get_ast_scope(enclosing_object)->contains_key(func_name)) @@ -224,25 +234,25 @@ obj c_generator (Object) { || func_name == "&&" || func_name == "<" || func_name == ">" || func_name == "<=" || func_name == ">=" || func_name == "==" || func_name == "%" ) - return string("(") + generate(parameters[0], enclosing_object, null>>()) + func_name + generate(parameters[1], enclosing_object, null>>()) + string(")") + return string("(") + generate(parameters[0], enclosing_object, null>>>()) + func_name + generate(parameters[1], enclosing_object, null>>>()) + string(")") // don't propegate enclosing function down right of access if (func_name == "." || func_name == "->") - return string("(") + generate(parameters[0], enclosing_object, null>>()) + func_name + generate(parameters[1], null(), null>>()) + string(")") + return string("(") + generate(parameters[0], enclosing_object, null>>>()) + func_name + generate(parameters[1], null(), null>>>()) + string(")") // the post ones need to be post-ed specifically, and take the p off if (func_name == "++p" || func_name == "--p") - return string("(") + generate(parameters[0], enclosing_object, null>>()) + ")" + func_name.slice(0,-2) + return string("(") + generate(parameters[0], enclosing_object, null>>>()) + ")" + func_name.slice(0,-2) // regular parameter generation parameters.for_each(fun(param: *ast_node) { if (call_string != "") call_string += ", " - call_string += generate(param, enclosing_object, null>>()) + call_string += generate(param, enclosing_object, null>>>()) }) return func_name + "(" + call_string + ")" } // for now, anyway - fun generate(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>): string { + fun generate(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): string { if (!node) return string("/*NULL*/") match (*node) { ast_node::if_comp(backing) return generate_if_comp(node) diff --git a/tests/to_import.krak b/tests/to_import.krak index 9b25b2b..88fc46c 100644 --- a/tests/to_import.krak +++ b/tests/to_import.krak @@ -1,4 +1,12 @@ +fun simple_println(to_print: *char) { + simple_print(to_print) + simple_print("\n") +} +fun simple_println(to_print: int) { + simple_print(to_print) + simple_print("\n") +} fun simple_print(to_print: *char) { __if_comp__ __C__ simple_passthrough(to_print::) """ printf("%s", to_print); diff --git a/tests/to_parse.krak b/tests/to_parse.krak index 8234a5d..62657e7 100644 --- a/tests/to_parse.krak +++ b/tests/to_parse.krak @@ -1,4 +1,4 @@ -import to_import: simple_print, a, b, string_id +import to_import: simple_print, simple_println, a, b, string_id obj Something (ObjectTrait) { var member: int @@ -58,9 +58,26 @@ fun main(): int { simple_print(with_ptr->method(2)) simple_print("\n") { + defer simple_print("should_be_third\n") defer simple_print("should_be_second\n") simple_print("should be first\n") } + for (var i = 1; i < 10; i++;) { + defer simple_println("OUTER:happens every time, even when breaking or continueing") + { + defer simple_println("INNER:happens every time, even when breaking or continueing") + if (i % 2 == 0) + continue + if (i == 9) + break + simple_println(i) + } + } + defer simple_println("Here is a return defer outer") + { + defer simple_println("Here is a return defer inner") + return 0 + } return 0 }