From 5ed0ed727be6141553fda8b01d1893f390fefe49 Mon Sep 17 00:00:00 2001
From: greg <greg.shuflin@protonmail.com>
Date: Wed, 10 Feb 2016 03:25:37 -0800
Subject: [PATCH] Starting to add LLVM code

---
 src/compilation.rs | 72 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 72 insertions(+)

diff --git a/src/compilation.rs b/src/compilation.rs
index 0933868..c1d143a 100644
--- a/src/compilation.rs
+++ b/src/compilation.rs
@@ -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;
+}