lemonlang 0.0.3

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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

use super::label::Label;
use crate::{
	ir::{Instr, IrBlock},
	report::throw_error,
};

pub struct Block {
	pub label: Label,
	pub next: Label,
	pub blocks: Vec<IrBlock>,
	pub returned: bool,
}

impl Default for Block {
	fn default() -> Self {
		Self::new()
	}
}

impl Block {
	pub fn new() -> Self {
		let label = Label::default(); // start from 1
		let block = IrBlock::new(label.into());
		Self { label, blocks: vec![block], next: label, returned: false }
	}

	pub fn is_returned(&self) -> bool {
		self.returned
	}

	pub fn as_returned(&mut self) {
		self.returned = true;
	}
	pub fn new_block(&mut self) -> Label {
		let label = self.next.increment();
		self.next = label;
		self.blocks.push(IrBlock::new(label.into()));
		label
	}

	pub fn take_blocks(&mut self) -> Vec<IrBlock> {
		self.returned = false;
		self.label = Label::default();
		self.next = self.label;
		let blocks = std::mem::take(&mut self.blocks);
		self.blocks = vec![IrBlock::new(self.label.into())];
		blocks
	}

	pub fn get_current_block(&mut self) -> &mut IrBlock {
		self.get_block_mut(self.label)
	}

	pub fn add_instr(&mut self, instr: Instr) {
		self.get_current_block().add_instr(instr);
	}

	pub fn switch_to_block(&mut self, label: Label) {
		self.returned = false;
		let value = label.value.wrapping_sub(1);
		if self.blocks.len() <= value {
			let len = self.blocks.len();
			let value = label.value;
			throw_error(format!("block index out of range: {}, blocks: {}", value, len));
		}
		self.label = label;
	}

	pub fn get_block_mut(&mut self, label: Label) -> &mut IrBlock {
		let value = label.value.wrapping_sub(1);
		if self.blocks.len() <= value {
			let len = self.blocks.len();
			let value = label.value;
			throw_error(format!("block index out of range: {}, blocks: {}", value, len));
		}
		match self.blocks.get_mut(value) {
			Some(block) => block,
			None => throw_error(format!("block index out of range: {}", label.value)),
		}
	}
}