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

Add initial tests for type checker

Browse source
This commit is contained in:
Vili Sinervä 2025-02-04 16:09:05 +02:00
parent 50bc35753a
commit b0420a9a5f
No known key found for this signature in database
GPG key ID: DF8FEAF54EFAC996
5 changed files with 280 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

82
src/compiler/symtab.rs
View file

@ -1,13 +1,13 @@
use crate::compiler::variable::Value;
use crate::compiler::variable::{Type, Value};
use std::collections::HashMap;
#[derive(Default)]
pub struct SymTab<'source> {
tables: Vec<HashMap<&'source str, Value>>,
pub struct SymTab<'source, T> {
tables: Vec<HashMap<&'source str, T>>,
}
impl<'source> SymTab<'source> {
pub fn get(&mut self, symbol: &str) -> &mut Value {
impl<'source, T> SymTab<'source, T> {
pub fn get(&mut self, symbol: &str) -> &mut T {
for i in (0..self.tables.len()).rev() {
if self.tables[i].contains_key(symbol) {
return self.tables[i].get_mut(symbol).unwrap();
@ -16,28 +16,6 @@ impl<'source> SymTab<'source> {
panic!("No symbol {} found!", symbol);
}
pub fn new() -> SymTab<'source> {
let globals = HashMap::from([
("+", Value::Func(Value::add)),
("*", Value::Func(Value::mul)),
("-", Value::Func(Value::sub)),
("/", Value::Func(Value::div)),
("%", Value::Func(Value::rem)),
("==", Value::Func(Value::eq)),
("!=", Value::Func(Value::neq)),
("<", Value::Func(Value::lt)),
("<=", Value::Func(Value::le)),
(">", Value::Func(Value::gt)),
(">=", Value::Func(Value::ge)),
("not", Value::Func(Value::not)),
("neg", Value::Func(Value::neg)),
]);
SymTab {
tables: vec![globals],
}
}
pub fn push_level(&mut self) {
self.tables.push(HashMap::new());
}
@ -46,7 +24,7 @@ impl<'source> SymTab<'source> {
self.tables.pop();
}
pub fn insert(&mut self, name: &'source str, val: Value) {
pub fn insert(&mut self, name: &'source str, val: T) {
if self
.tables
.last_mut()
@ -58,3 +36,51 @@ impl<'source> SymTab<'source> {
}
}
}
impl<'source> SymTab<'source, Type> {
pub fn new_type_table() -> SymTab<'source, Type> {
use Type::*;
let globals = HashMap::from([
("+", Func(vec![Int, Int], Box::new(Int))),
("*", Func(vec![Int, Int], Box::new(Int))),
("-", Func(vec![Int, Int], Box::new(Int))),
("/", Func(vec![Int, Int], Box::new(Int))),
("%", Func(vec![Int, Int], Box::new(Int))),
("<", Func(vec![Int, Int], Box::new(Bool))),
("<=", Func(vec![Int, Int], Box::new(Bool))),
(">", Func(vec![Int, Int], Box::new(Bool))),
(">=", Func(vec![Int, Int], Box::new(Bool))),
("not", Func(vec![Bool], Box::new(Bool))),
("neg", Func(vec![Bool], Box::new(Bool))),
]);
SymTab {
tables: vec![globals],
}
}
}
impl<'source> SymTab<'source, Value> {
pub fn new_val_table() -> SymTab<'source, Value> {
use Value::*;
let globals = HashMap::from([
("+", Func(Value::add)),
("*", Func(Value::mul)),
("-", Func(Value::sub)),
("/", Func(Value::div)),
("%", Func(Value::rem)),
("==", Func(Value::eq)),
("!=", Func(Value::neq)),
("<", Func(Value::lt)),
("<=", Func(Value::le)),
(">", Func(Value::gt)),
(">=", Func(Value::ge)),
("not", Func(Value::not)),
("neg", Func(Value::neg)),
]);
SymTab {
tables: vec![globals],
}
}
}