VSinerva/compiler-course
Archived
1
0
Fork 0
Code Activity

Add while-loop to IR Generato

Browse source
This commit is contained in:
Vili Sinervä 2025-02-05 23:00:58 +02:00
parent 7dd60aa9c5
commit 2cf26ec39d
No known key found for this signature in database
GPG key ID: DF8FEAF54EFAC996
1 changed files with 24 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

30
src/compiler/ir_generator.rs
View file

@ -89,10 +89,10 @@ fn add_label(
IrInstruction::new(loc, instruction) IrInstruction::new(loc, instruction)
} }
fn visit_ast_node( fn visit_ast_node<'source>(
ast: &AstNode, ast: &AstNode<'source>,
types: &mut HashMap<IrVar, Type>, types: &mut HashMap<IrVar, Type>,
symbols: &mut SymTab<IrVar>, symbols: &mut SymTab<'source, IrVar>,
instructions: &mut Vec<IrInstruction>, instructions: &mut Vec<IrInstruction>,
labels: &mut HashSet<IrInstructionType>, labels: &mut HashSet<IrInstructionType>,
) -> IrVar { ) -> IrVar {
@ -142,9 +142,10 @@ fn visit_ast_node(
result_var result_var
} }
}, },
VarDeclaration(_, expr, _) => { VarDeclaration(name, expr, _) => {
let expr_var = visit_ast_node(expr, types, symbols, instructions, labels); let expr_var = visit_ast_node(expr, types, symbols, instructions, labels);
let result_var = add_var(&expr.node_type, types); let result_var = add_var(&expr.node_type, types);
symbols.insert(name, result_var.clone());
instructions.push(IrInstruction::new(expr.loc, Copy(expr_var, result_var))); instructions.push(IrInstruction::new(expr.loc, Copy(expr_var, result_var)));
symbols.get("unit").clone() symbols.get("unit").clone()
} }
@ -201,12 +202,29 @@ fn visit_ast_node(
symbols.get("unit").clone() symbols.get("unit").clone()
} }
}, },
While(_, _) => todo!(), While(condition_expr, do_expr) => {
let l_start = add_label("while_start", condition_expr.loc, labels);
let l_body = add_label("while_body", do_expr.loc, labels);
let l_end = add_label("while_end", do_expr.loc, labels);
instructions.push(l_start.clone());
let cond_var = visit_ast_node(condition_expr, types, symbols, instructions, labels);
instructions.push(IrInstruction::new(
condition_expr.loc,
CondJump(cond_var, Box::new(l_body.clone()), Box::new(l_end.clone())),
));
instructions.push(l_body);
visit_ast_node(do_expr, types, symbols, instructions, labels);
instructions.push(IrInstruction::new(do_expr.loc, Jump(Box::new(l_start))));
instructions.push(l_end);
symbols.get("unit").clone()
}
FunCall(_, _) => todo!(), FunCall(_, _) => todo!(),
Block(expressions) => { Block(expressions) => {
let mut result_var = symbols.get("unit").clone(); let mut result_var = symbols.get("unit").clone();
for expression in expressions { for expression in expressions {
result_var = visit_ast_node(expression, types, symbols, instructions, labels) result_var = visit_ast_node(expression, types, symbols, instructions, labels);
} }
result_var result_var
} }