fips-md 0.3.0

A framework for simulating arbitrary particle dynamics
Documentation 
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

mod const_value;
mod binop;
mod identifier;

use std::collections::HashMap;

use super::llwrap as ll;

use crate::parsing::{ASTNode};

pub struct CodegenContext {
    pub ir_builder: ll::IRBuilder,
    variables: HashMap<String, Vec<ll::Value>>
}

impl CodegenContext {
    pub fn new(builder: ll::IRBuilder) -> CodegenContext {
        CodegenContext {
            ir_builder: builder,
            variables: HashMap::new()
        }
    }

    pub fn push_var(&mut self, name: &str, val: ll::Value) {
        match self.variables.get_mut(name) {
            Some(vstack) => vstack.push(val),
            None => { 
                let v = vec!(val);
                self.variables.insert(String::from(name), v); 
            }
        }
    }

    // None option means that variable has never existed or is no longer valid
    // TODO: Differentiate these cases for better debug messages?
    pub fn pop_var(&mut self, name: &str) -> Option<ll::Value> {
        match self.variables.get_mut(name) {
            Some(v) => v.pop(),
            None => None
        }
    }
}

pub trait Codegen {
    fn codegen(&self, cctx: &mut CodegenContext) -> ll::Value;
}

impl Codegen for ASTNode {
    fn codegen(&self, cctx: &mut CodegenContext) -> ll::Value {
        match self {
            ASTNode::BinOp(binop) => binop.codegen(cctx),
            ASTNode::Literal(cv) => cv.codegen(cctx),
            _ => panic!("JITing not implemented for this ASTNode")
        }
    }
}