28 lines
685 B
Rust
28 lines
685 B
Rust
|
use crate::compiler::value::Value;
|
||
|
|
use std::collections::HashMap;
|
||
|
|
|
||
|
|
#[derive(Default)]
|
||
|
|
pub struct SymTab<'source> {
|
||
|
|
pub locals: HashMap<&'source str, Value>,
|
||
|
|
pub parent: Option<Box<SymTab<'source>>>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<'source> SymTab<'source> {
|
||
|
|
pub fn get(&mut self, symbol: &str) -> &mut Value {
|
||
|
|
if let Some(val) = self.locals.get_mut(symbol) {
|
||
|
|
val
|
||
|
|
} else if let Some(parent) = &mut self.parent {
|
||
|
|
parent.get(symbol)
|
||
|
|
} else {
|
||
|
|
panic!("No symbol {} found!", symbol);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn new_global() -> SymTab<'source> {
|
||
|
|
SymTab {
|
||
|
|
locals: HashMap::new(),
|
||
|
|
parent: None,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|