src/codegen/mod.rs


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

pub mod llvm;

use inkwell::execution_engine::JitFunction;
use inkwell::OptimizationLevel;
pub use llvm::*;

use crate::ast::hir::Expr;
use crate::ast::Type;
use crate::common::Result;

pub fn jit_eval<T>(expr: &Expr<Type>) -> Result<T> {
    let context = Context::create();
    let mut codegen = Codegen::new(&context, "eval");
    let execution_engine = codegen
        .module
        .create_jit_execution_engine(OptimizationLevel::None)
        .map_err(Error::from)?;
    codegen.codegen_function("test", &[], &expr)?;

    unsafe {
        let fun: JitFunction<unsafe extern "C" fn() -> T> =
            execution_engine.get_function("eval").unwrap();
        Ok(fun.call())
    }
}