lemonlang 0.0.4

an experimental, modern, purely safe, programming language.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use crate::ast;
use crate::ir::{self, IrBasicValue};

use super::Builder;

impl Builder<'_> {
	pub fn build_if_expr(&mut self, if_expr: &mut ast::IfExpr) -> IrBasicValue {
		let cond = self.build_expr(&mut if_expr.cond);
		let then_block = self.ctx.block.new_block();
		let otherwise_block = if_expr.otherwise.as_ref().map(|_| self.ctx.block.new_block());
		let merge_block = self.ctx.block.new_block();

		let otherwise = otherwise_block.unwrap_or(merge_block);

		let instr = ir::JmpIfInstr::new(cond, then_block.into(), otherwise.into());
		self.ctx.block.add_instr(instr.into());

		self.ctx.block.switch_to_block(then_block);
		self.build_stmt(&mut if_expr.then);
		if !self.ctx.block.is_returned() {
			let jump = ir::JmpInstr::new(merge_block.into());
			self.ctx.block.add_instr(jump.into());
		}

		if let Some(otherwise) = &mut if_expr.otherwise {
			self.ctx.block.switch_to_block(otherwise_block.unwrap());
			self.build_stmt(otherwise);

			if !self.ctx.block.is_returned() {
				let jump = ir::JmpInstr::new(merge_block.into());
				self.ctx.block.add_instr(jump.into());
			}
		}
		self.ctx.block.switch_to_block(merge_block);
		IrBasicValue::default()
	}
}