104 lines
3.0 KiB
Rust
104 lines
3.0 KiB
Rust
extern crate llvm_sys;
|
|
|
|
use self::llvm_sys::prelude::*;
|
|
use self::llvm_sys::core;
|
|
use std::ptr;
|
|
|
|
use parser::{ParseResult, AST, ASTNode, Prototype, Function, Expression};
|
|
|
|
mod LLVMWrap {
|
|
extern crate llvm_sys;
|
|
use self::llvm_sys::prelude::*;
|
|
use self::llvm_sys::core;
|
|
use std::ptr;
|
|
use std::ffi::CString;
|
|
|
|
pub fn create_context() -> LLVMContextRef {
|
|
unsafe {
|
|
core::LLVMContextCreate()
|
|
}
|
|
}
|
|
pub fn module_create_with_name(name: &str) -> LLVMModuleRef {
|
|
unsafe {
|
|
let n = name.as_ptr() as *const _;
|
|
core::LLVMModuleCreateWithName(n)
|
|
}
|
|
}
|
|
pub fn CreateBuilderInContext(context: LLVMContextRef) -> LLVMBuilderRef {
|
|
unsafe {
|
|
core::LLVMCreateBuilderInContext(context)
|
|
}
|
|
}
|
|
|
|
pub fn AppendBasicBlockInContext(context: LLVMContextRef, function: LLVMValueRef, name: &str) -> LLVMBasicBlockRef {
|
|
let c_name = CString::new(name).unwrap();
|
|
unsafe {
|
|
core::LLVMAppendBasicBlockInContext(context, function, c_name.as_ptr())
|
|
}
|
|
}
|
|
|
|
pub fn AddFunction(module: LLVMModuleRef, name: &str, function_type: LLVMTypeRef) -> LLVMValueRef {
|
|
let c_name = CString::new(name).unwrap();
|
|
unsafe {
|
|
core::LLVMAddFunction(module, c_name.as_ptr(), function_type)
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn compile_ast(ast: AST) {
|
|
println!("Compiling!");
|
|
let context = LLVMWrap::create_context();
|
|
let module = LLVMWrap::module_create_with_name("schala");
|
|
let builder = LLVMWrap::CreateBuilderInContext(context);
|
|
unsafe {
|
|
|
|
let void = core::LLVMVoidTypeInContext(context);
|
|
let function_type = core::LLVMFunctionType(void, ptr::null_mut(), 0, 0);
|
|
let function = LLVMWrap::AddFunction(module, "nop", function_type);
|
|
|
|
let bb = LLVMWrap::AppendBasicBlockInContext(context, function, "entry");
|
|
|
|
core::LLVMPositionBuilderAtEnd(builder, bb);
|
|
|
|
// Emit a `ret void` into the function
|
|
core::LLVMBuildRetVoid(builder);
|
|
|
|
// Dump the module as IR to stdout.
|
|
core::LLVMDumpModule(module);
|
|
|
|
// Clean up. Values created in the context mostly get cleaned up there.
|
|
core::LLVMDisposeBuilder(builder);
|
|
core::LLVMDisposeModule(module);
|
|
core::LLVMContextDispose(context);
|
|
}
|
|
}
|
|
|
|
trait CodeGen {
|
|
fn codegen(&self, LLVMContextRef) -> LLVMValueRef;
|
|
}
|
|
|
|
impl CodeGen for ASTNode {
|
|
fn codegen(&self, context: LLVMContextRef) -> LLVMValueRef {
|
|
use self::ASTNode::*;
|
|
match self {
|
|
&ExprNode(ref expr) => expr.codegen(context),
|
|
&FuncNode(ref func) => unimplemented!(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl CodeGen for Expression {
|
|
fn codegen(&self, context: LLVMContextRef) -> LLVMValueRef {
|
|
use self::Expression::*;
|
|
match self {
|
|
&BinExp(ref op, ref left, ref right) => {
|
|
unimplemented!()
|
|
},
|
|
&Number(ref n) => {
|
|
unimplemented!()
|
|
},
|
|
_ => unimplemented!(),
|
|
}
|
|
}
|
|
}
|