Starting to add LLVM code

This commit is contained in:
greg 2016-02-10 03:25:37 -08:00
parent 5428810d2c
commit 5ed0ed727b

View File

@ -3,3 +3,75 @@ extern crate iron_llvm;
use std::collections::HashMap;
use llvm_sys::prelude::LLVMValueRef;
use iron_llvm::core;
use iron_llvm::core::types::{RealTypeCtor, RealTypeRef};
use iron_llvm::{LLVMRef, LLVMCtor};
pub struct Context {
context: core::Context,
builder: core::Builder,
named_values: HashMap<String, LLVMValueRef>,
ty: RealTypeRef,
}
impl Context {
pub fn new() -> Context {
let context = core::Context::get_global();
let builder = core::Builder::new();
let named_values = HashMap::new();
let ty = RealTypeRef::get_double();
Context {
context: context,
builder: builder,
named_values: named_values,
ty: ty,
}
}
}
pub trait ModuleProvider {
fn dump(&self);
fn get_module(&mut self) -> &mut core::Module;
fn get_function(&mut self, name: &str) -> Option<(FunctionRef, bool)>;
}
pub struct SimpleModuleProvider {
module: core::Module,
}
impl SimpleModuleProvider {
pub fn new(name: &str) -> SimpleModuleProvider {
let module = core::Module::new(name);
SimpleModuleProvider {
module: module,
}
}
}
impl ModuleProvider for SimpleModuleProvider {
fn dump(&self) {
self.module.dump();
}
fn get_module(&mut self) -> &mut core::Module {
&mut self.module
}
fn get_function(&mut self, name: &str) -> Option<(FunctionRef, bool)> {
match.self.module.get_function_by_name(name) {
Some(f) => Some((f, f.count_basic_block() > 0)),
None => None
}
}
}
pub type IRBuildingResult = Result<LLVMValueRef, bool), String>;
fn error(msg: &str) -> IRBuildingResult {
Err(msg.to_string())
}
pub trait IRBuilder {
fn codegen(&self, context: &mut Context, module_provider: &mut ModuleProvider) -> IRBuildingResult;
}