import io:* import mem:* import map:* import stack:* import string:* import util:* import tree:* import symbol:* import ast_nodes:* import poset:* // we import ast_transformation for its make_method_call function import ast_transformation:* fun code_triple(): code_triple return code_triple(string(), string(), string()); fun code_triple(only: *char): code_triple return code_triple(string(), string(only), string()); fun code_triple(only: string): code_triple return code_triple(string(), only, string()); fun code_triple(first: string, second: string, third: string): code_triple { var to_ret.construct(first, second, third): code_triple return to_ret } obj code_triple (Object) { var pre: string var value: string var post: string fun construct(first: string, second: string, third: string): *code_triple { pre.copy_construct(&first) value.copy_construct(&second) post.copy_construct(&third) return this } fun copy_construct(old: *code_triple) { pre.copy_construct(&old->pre) value.copy_construct(&old->value) post.copy_construct(&old->post) } fun operator=(other: code_triple) { destruct() copy_construct(&other) } fun destruct() { pre.destruct() value.destruct() post.destruct() } fun operator==(other: code_triple): bool return pre == other.pre && value == other.value && post == other.value; fun operator==(other: string): bool return (pre + value + post) == other; fun operator==(other: *char): bool return (pre + value + post) == other; fun operator!=(other: code_triple): bool return pre != other.pre || value != other.value || post != other.value; fun operator!=(other: string): bool return (pre + value + post) != other; fun operator!=(other: *char): bool return (pre + value + post) != other; fun operator+(other: code_triple): code_triple { return code_triple(pre+other.pre, value+other.value, other.post+post); } fun operator+(other: string): code_triple { return code_triple(pre, value+other, post); } fun operator+(other: *char): code_triple { return code_triple(pre, value + other, post); } fun operator+=(other: string) { value += other } fun operator+=(other: *char) { value += other } fun operator+=(other: code_triple) { pre += other.pre value += other.value post += other.post } fun one_string():string { return pre+value+post } } obj c_generator (Object) { var id_counter: int fun construct(): *c_generator { id_counter = 0 return this } fun copy_construct(old: *c_generator) { id_counter = old->id_counter } fun operator=(other: ref c_generator) { destruct() copy_construct(&other) } fun destruct() { } fun get_id(): string return to_string(id_counter++); fun generate_c(name_ast_map: map,*ast_node>>): pair { var linker_string:string = "" var prequal: string = "#include \n#include \n#include \n" var plain_typedefs: string = "\n/**Plain Typedefs**/\n" var top_level_c_passthrough: string = "" var variable_extern_declarations: string = "" var structs: string = "\n/**Type Structs**/\n" var function_typedef_string_pre: string = "" var function_typedef_string: string = "" var function_prototypes: string = "\n/**Function Prototypes**/\n" var function_definitions: string = "\n/**Function Definitions**/\n" var variable_declarations: string = "\n/**Variable Declarations**/\n" // moved out from below so that it can be used for methods as well as regular functions (and eventually lambdas...) var generate_function_definition = fun(child: *ast_node, enclosing_object: *ast_node) { var backing = child->function var parameter_types = string() var parameters = string() if (enclosing_object) { parameter_types = type_to_c(enclosing_object->type_def.self_type) + "*" parameters = type_to_c(enclosing_object->type_def.self_type) + "* this" } // stack-stack thing // this could be a stack of strings too, maybe // start out with one stack on the stack var defer_stack = stack>>(make_pair(false, stack<*ast_node>())) var decorated_name = generate_function(child).one_string() // also add in name decoration backing.parameters.for_each(fun(parameter: *ast_node) { if (parameter_types != "") { parameter_types += ", "; parameters += ", ";} parameter_types += type_to_c(parameter->identifier.type) parameters += type_to_c(parameter->identifier.type) + " " + parameter->identifier.name // add parameters to destructor thingy (for returns)? Or should that be a different pass? var parameter_type = parameter->identifier.type if (parameter_type->indirection == 0 && parameter_type->is_object() && has_method(parameter_type->type_def, "destruct", vector<*type>())) defer_stack.top().second.push(ast_statement_ptr(make_method_call(parameter, "destruct", vector<*ast_node>()))) }) function_prototypes += type_to_c(backing.type->return_type) + " " + decorated_name + "(" + parameter_types + ");\n" function_definitions += type_to_c(backing.type->return_type) + " " + decorated_name + "(" + parameters + ") {\n" + generate_statement(backing.body_statement, enclosing_object, &defer_stack).one_string() // emit parameter destructors? function_definitions += generate_from_defer_stack(&defer_stack, -1, enclosing_object).one_string() function_definitions += "}\n" } var type_poset = poset<*ast_node>() // iterate through asts name_ast_map.for_each(fun(name: string, tree_pair: pair<*tree,*ast_node>) { // iterate through children for each ast // assert translation_unit? tree_pair.second->translation_unit.children.for_each(fun(child: *ast_node) { match (*child) { // should really check the genrator ast_node::if_comp(backing) { if (is_simple_passthrough(backing.statement->statement.child)) 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>>>(), true).one_string() + ";\n" ast_node::function(backing) { // make sure not a template // or a passthrough // check for and add to parameters if a closure generate_function_definition(child, null()) } ast_node::type_def(backing) { type_poset.add_vertex(child) } } }) }) 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>>>(), true).one_string() + ";\n";) structs += "};\n" // generate the methods vert->type_def.methods.for_each(fun(method: *ast_node) generate_function_definition(method, vert);) }) return make_pair(prequal+plain_typedefs+top_level_c_passthrough+variable_extern_declarations+structs+function_typedef_string_pre+function_typedef_string+function_prototypes+variable_declarations+function_definitions + "\n", linker_string) } fun generate_if_comp(node: *ast_node): code_triple { if (node->if_comp.wanted_generator == "__C__") return generate(node->if_comp.statement, null(), null>>>()) return code_triple() } 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>>): code_triple return generate(node->statement.child, enclosing_object, defer_stack) + ";\n"; fun generate_declaration_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>, add_to_defer: bool): code_triple { // add destruct to defer_stack var identifier = node->declaration_statement.identifier var ident_type = identifier->identifier.type var to_ret = code_triple() + type_to_c(identifier->identifier.type) + " " + identifier->identifier.name if (node->declaration_statement.expression) { if (ident_type->is_object() && has_method(ident_type->type_def, "copy_construct", vector(get_ast_type(node->declaration_statement.expression)->clone_with_increased_indirection()))) { to_ret += ";\n"; to_ret += generate(ast_statement_ptr(make_method_call(identifier, "copy_construct", vector(make_operator_call("&", vector(node->declaration_statement.expression))))), enclosing_object, defer_stack) } else { to_ret += code_triple(" = ") + generate(node->declaration_statement.expression, enclosing_object, null>>>()) } } if (node->declaration_statement.init_method_call) to_ret += code_triple(";\n") + generate(node->declaration_statement.init_method_call, enclosing_object, null>>>()) if (add_to_defer && ident_type->is_object() && has_method(ident_type->type_def, "destruct", vector<*type>())) defer_stack->top().second.push(ast_statement_ptr(make_method_call(identifier, "destruct", vector<*ast_node>()))) return to_ret } fun generate_assignment_statement(node: *ast_node, enclosing_object: *ast_node): code_triple { return generate(node->assignment_statement.to, enclosing_object, null>>>()) + " = " + generate(node->assignment_statement.from, enclosing_object, null>>>()) } fun generate_if_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): code_triple { var if_str = code_triple("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 += code_triple(" 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>>): code_triple { // stick another stack on defer_stack->push(make_pair(true, stack<*ast_node>())) var to_ret = code_triple("while (") + generate(node->while_loop.condition, enclosing_object, null>>>()).one_string() + ") {\n" + generate(node->while_loop.statement, enclosing_object, defer_stack).one_string() 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>>): code_triple { // stick another stack on defer_stack->push(make_pair(true, stack<*ast_node>())) // gotta take off last semicolon var to_ret = code_triple("for (") + generate(node->for_loop.init, enclosing_object, null>>>()).one_string() + " " + generate(node->for_loop.condition, enclosing_object, null>>>()).one_string() + "; " + generate(node->for_loop.update, enclosing_object, null>>>()).one_string().slice(0,-3) + ") {\n" + generate(node->for_loop.body, enclosing_object, defer_stack).one_string() to_ret += generate_from_defer_stack(defer_stack, 1, enclosing_object).one_string() defer_stack->pop() to_ret += "}\n" return to_ret } fun generate_identifier(node: *ast_node, enclosing_object: *ast_node): code_triple { if (enclosing_object && get_ast_scope(enclosing_object)->contains_key(node->identifier.name)) return code_triple("(this->") + node->identifier.name + ")" return code_triple(node->identifier.name) } fun generate_return_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): code_triple { var return_value = node->return_statement.return_value var return_value_type = get_ast_type(return_value) var to_ret = code_triple() // if we're returning an object, copy_construct a new one to return if (return_value_type->is_object() && return_value_type->indirection == 0 && has_method(return_value_type->type_def, "copy_construct", vector(return_value_type->clone_with_indirection(1)))) { var temp_ident = ast_identifier_ptr(string("temporary_return")+get_id(), return_value_type) var declaration = ast_declaration_statement_ptr(temp_ident, null()) // have to pass false to the declaration generator, so can't do it through generate_statement to_ret.pre = generate_declaration_statement(declaration, enclosing_object, defer_stack, false).one_string() + ";\n" to_ret.pre += generate_statement(ast_statement_ptr(make_method_call(temp_ident, "copy_construct", vector(make_operator_call("&", vector(return_value))))), enclosing_object, defer_stack).one_string() // make this new identifier the new return value return_value = temp_ident } // generate all in stack by passing -1 to_ret += generate_from_defer_stack(defer_stack, -1, enclosing_object) to_ret += code_triple("return ") + generate(return_value, enclosing_object, null>>>()) return to_ret } fun generate_branching_statement(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): code_triple { // -2 means generate up through loop 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): code_triple { var to_ret = code_triple() 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++ } for (var i = 0; i < num; i++;) defer_stack->from_top(i).second.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>>): code_triple { defer_stack->top().second.push(node->defer_statement.statement) return code_triple("/*defer wanna know what*/") } fun generate_value(node: *ast_node): code_triple return code_triple(node->value.string_value); fun generate_code_block(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): code_triple { var to_ret = code_triple("{\n") // stick another stack on 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).one_string();) to_ret += generate_from_defer_stack(defer_stack, 1, enclosing_object) defer_stack->pop() return to_ret + "}" } // this generates the function as a value, not the actual function fun generate_function(node: *ast_node): code_triple { var str = code_triple(node->function.name) node->function.parameters.for_each(fun(param: *ast_node) str += string("_") + type_decoration(param->identifier.type);) return str } fun generate_function_call(node: *ast_node, enclosing_object: *ast_node): code_triple { var func_name = string() var call_string = code_triple() var func_return_type = get_ast_type(node) // handle the obj.method() style of method call var dot_style_method_call = is_function_call(node->function_call.func) && is_function(node->function_call.func->function_call.func) && (node->function_call.func->function_call.func->function.name == "->" || node->function_call.func->function_call.func->function.name == ".") && is_function(node->function_call.func->function_call.parameters[1]) && 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>>>()).one_string() // don't add & if it was -> if (node->function_call.func->function_call.func->function.name == ".") call_string += "&" 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>>>()).one_string() } // 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)) call_string += "this" var parameters = node->function_call.parameters if ( parameters.size == 2 && (func_name == "+" || func_name == "-" || func_name == "*" || func_name == "/" || func_name == "||" || func_name == "&&" || func_name == "<" || func_name == ">" || func_name == "<=" || func_name == ">=" || func_name == "==" || func_name == "%" )) return code_triple("(") + 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 code_triple("(") + 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 code_triple("(") + generate(parameters[0], enclosing_object, null>>>()) + ")" + func_name.slice(0,-2) // So we don't end up copy_constructing etc, we just handle the unary operators right here if (func_name == "*" || func_name == "&") return code_triple("(") + func_name + generate(parameters[0], enclosing_object, null>>>()) + ")" // regular parameter generation parameters.for_each(fun(param: *ast_node) { if (call_string != "") call_string += ", " var param_type = get_ast_type(param) if (param_type->is_object() && param_type->indirection == 0 && has_method(param_type->type_def, "copy_construct", vector(param_type->clone_with_indirection(1)))) { var temp_ident = ast_identifier_ptr(string("temporary_param")+get_id(), param_type) var declaration = ast_declaration_statement_ptr(temp_ident, null()) // have to pass false to the declaration generator, so can't do it through generate_statement call_string.pre += generate_declaration_statement(declaration, enclosing_object, null>>>(), false).one_string() + ";\n" call_string.pre += generate_statement(ast_statement_ptr(make_method_call(temp_ident, "copy_construct", vector(make_operator_call("&", vector(param))))), enclosing_object, null>>>()).one_string() call_string += generate(temp_ident, enclosing_object, null>>>()) if (has_method(param_type->type_def, "destruct", vector<*type>())) { call_string.post += generate_statement(ast_statement_ptr(make_method_call(temp_ident, "destruct", vector<*ast_node>())), enclosing_object, null>>>()).one_string() } } else { call_string += generate(param, enclosing_object, null>>>()) } }) if (func_return_type->is_object() && func_return_type->indirection == 0 && has_method(func_return_type->type_def, "destruct", vector<*type>())) { // kind of ugly combo here of var temp_ident = ast_identifier_ptr(string("temporary_return")+get_id(), func_return_type) var declaration = ast_declaration_statement_ptr(temp_ident, null()) // have to pass false to the declaration generator, so can't do it through generate_statement call_string.pre += generate_declaration_statement(declaration, enclosing_object, null>>>(), false).one_string() + ";\n" call_string.pre += generate_identifier(temp_ident, enclosing_object).one_string() + " = " + func_name + "(" + call_string.value + ");\n" call_string.value = generate_identifier(temp_ident, enclosing_object).one_string() call_string.post += generate_statement(ast_statement_ptr(make_method_call(temp_ident, "destruct", vector<*ast_node>())), enclosing_object, null>>>()).one_string() return call_string } return code_triple() + func_name + "(" + call_string + ")" } // for now, anyway fun generate(node: *ast_node, enclosing_object: *ast_node, defer_stack: *stack>>): code_triple { if (!node) return code_triple("/*NULL*/") match (*node) { ast_node::if_comp(backing) return generate_if_comp(node) ast_node::simple_passthrough(backing) return code_triple() + generate_simple_passthrough(node) ast_node::statement(backing) return generate_statement(node, enclosing_object, defer_stack) ast_node::declaration_statement(backing) return generate_declaration_statement(node, enclosing_object, defer_stack, true) ast_node::assignment_statement(backing) return generate_assignment_statement(node, enclosing_object) ast_node::if_statement(backing) return generate_if_statement(node, enclosing_object, defer_stack) ast_node::while_loop(backing) return generate_while_loop(node, enclosing_object, defer_stack) ast_node::for_loop(backing) return generate_for_loop(node, enclosing_object, defer_stack) ast_node::function(backing) return generate_function(node) ast_node::function_call(backing) return generate_function_call(node, enclosing_object) ast_node::code_block(backing) return generate_code_block(node, enclosing_object, defer_stack) ast_node::return_statement(backing) return generate_return_statement(node, enclosing_object, defer_stack) ast_node::branching_statement(backing) return generate_branching_statement(node, enclosing_object, defer_stack) ast_node::defer_statement(backing) return generate_defer_statement(node, enclosing_object, defer_stack) ast_node::value(backing) return generate_value(node) ast_node::identifier(backing) return generate_identifier(node, enclosing_object) } return code_triple("/* COULD NOT GENERATE */") } fun type_decoration(type: *type): string { var indirection = string() for (var i = 0; i < type->indirection; i++;) indirection += "p" if (type->indirection) indirection += "_" match (type->base) { base_type::none() return indirection + string("none") base_type::template() return indirection + string("template") base_type::template_type() return indirection + string("template_type") base_type::void_return() return indirection + string("void") base_type::boolean() return indirection + string("bool") base_type::character() return indirection + string("char") base_type::integer() return indirection + string("int") base_type::floating() return indirection + string("float") base_type::double_precision() return indirection + string("double") base_type::object() { return type->type_def->type_def.name } base_type::function() { var temp = indirection + string("function_") type->parameter_types.for_each(fun(parameter_type: *type) temp += type_decoration(parameter_type) + "_";) return indirection + temp + "_" + type_decoration(type->return_type) } } return string("impossible type") + indirection } fun type_to_c(type: *type): string { var indirection = string() for (var i = 0; i < type->indirection; i++;) indirection += "*" match (type->base) { base_type::none() return string("none") + indirection base_type::template() return string("template") + indirection base_type::template_type() return string("template_type") + indirection base_type::void_return() return string("void") + indirection base_type::boolean() return string("bool") + indirection base_type::character() return string("char") + indirection base_type::integer() return string("int") + indirection base_type::floating() return string("float") + indirection base_type::double_precision() return string("double") + indirection base_type::object() { return type->type_def->type_def.name + indirection } base_type::function() { var temp = indirection + string("function: (") type->parameter_types.for_each(fun(parameter_type: *type) temp += parameter_type->to_string() + " ";) return temp + ")" + type->return_type->to_string() } } return string("impossible type") + indirection } }