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

use inkwell::{basic_block::BasicBlock, values::PointerValue};
use rustc_hash::FxHashMap;

#[derive(Debug)]
pub struct Scope<'ll> {
	block_store: FxHashMap<String, BasicBlock<'ll>>,
	value_store: FxHashMap<String, inkwell::values::BasicValueEnum<'ll>>,
	temp_count: u32,
}

impl<'ll> Scope<'ll> {
	pub fn new() -> Self {
		let block_store = FxHashMap::default();
		let value_store = FxHashMap::default();
		Self { block_store, value_store, temp_count: 0 }
	}

	pub fn get_temp(&mut self) -> String {
		let name = format!("t{}", self.temp_count);
		self.temp_count += 1;
		name
	}

	pub fn get_block(&self, name: &str) -> Option<&BasicBlock<'ll>> {
		self.block_store.get(name)
	}

	pub fn set_block(&mut self, name: &str, block: BasicBlock<'ll>) {
		self.block_store.insert(name.to_string(), block);
	}

	pub fn get_value(&self, name: &str) -> Option<&inkwell::values::BasicValueEnum<'ll>> {
		self.value_store.get(name)
	}

	pub fn set_value(&mut self, name: &str, value: inkwell::values::BasicValueEnum<'ll>) {
		self.value_store.insert(name.to_string(), value);
	}

	pub fn get_ptr_value(&self, name: &str) -> Option<PointerValue<'ll>> {
		let value = self.get_value(name)?;
		if value.is_pointer_value() {
			return Some(value.into_pointer_value());
		}
		None
	}
}