libjuno 0.1.0

Library for JunoLang
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 std::collections::HashMap;

use super::*;
use crate::{ast::{Function, Item, Program}, *};
impl<'a> MetaIRGen<'a> {
    pub fn push_scope(&mut self) {
        self.locals.push(HashMap::new());
    }

    pub fn pop_scope(&mut self) {
        self.locals.pop();
    }

    pub fn insert_local(&mut self, id: SymbolId, ty: MetaType) {
        self.locals.last_mut().unwrap().insert(id, ty);
    }

    pub fn lookup_local(&self, id: SymbolId) -> MetaType {
        for scope in self.locals.iter().rev() {
            if let Some(ty) = scope.get(&id) {
                return ty.clone();
            }
        }

        panic!("unknown variable {}", id);
    }
    pub fn lookup_local_type(&self, id: SymbolId) -> MetaType {
        for scope in self.locals.iter().rev() {
            if let Some(ty) = scope.get(&id) {
                return ty.clone();
            }
        }

        panic!("unknown local variable {}", id);
    }
    
}