Initial commit

1 files changed, 26 insertions, 0 deletions

diff --git a/src/codegen/mod.rs b/src/codegen/mod.rs
new file mode 100644
index 000000000000..4620b6b48e84
--- /dev/null
+++ b/src/codegen/mod.rs
@@ -0,0 +1,26 @@
+pub mod llvm;
+
+use inkwell::execution_engine::JitFunction;
+use inkwell::OptimizationLevel;
+pub use llvm::*;
+
+use crate::ast::Expr;
+use crate::common::Result;
+
+pub fn jit_eval<T>(expr: &Expr) -> 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.new_function("eval", context.i64_type().fn_type(&[], false));
+    let res = codegen.codegen_expr(&expr)?;
+    codegen.finish_function(&res);
+
+    unsafe {
+        let fun: JitFunction<unsafe extern "C" fn() -> T> =
+            execution_engine.get_function("eval").unwrap();
+        Ok(fun.call())
+    }
+}