Types!

inc/AST/Type.h

@@ -1,10 +1,12 @@
 #ifndef TYPE_H
 #define TYPE_H
 
+#include "llvm/Type.h"
 #include <string>
 
 enum class Type { BOOL, INT, STRING, VOID };
 
 std::string typeToString(Type type);
+llvm::Type* typeToLLVMType(Type type);
 
 #endif // TYPE_H

src/AST/CodeGenVisitor.cpp

@@ -88,6 +88,7 @@ void CodeGenVisitor::visit(FunctionCallExpression* e) {
 }
 
 void CodeGenVisitor::visit(FunctionDefinition* e) {
+	//std::vector<llvm::Type*> types;
 }
 
 void CodeGenVisitor::visit(IfStatement* e) {

src/AST/Type.cpp

@@ -1,4 +1,6 @@
 #include "AST/Type.h"
+#include "llvm/DerivedTypes.h"
+#include "llvm/LLVMContext.h"
 
 std::string typeToString(Type type) {
 	switch (type) {
@@ -12,5 +14,21 @@ std::string typeToString(Type type) {
 		return "void";
 	default:
 		return "ERROR";
-  }
+	}
+}
+
+llvm::Type* typeToLLVMType(Type type) {
+	switch (type) {
+	case Type::BOOL:
+		return llvm::Type::getInt1Ty(llvm::getGlobalContext());
+	case Type::INT:
+		return llvm::Type::getInt32Ty(llvm::getGlobalContext());
+	case Type::STRING:
+		return llvm::Type::getInt8PtrTy(llvm::getGlobalContext());
+	case Type::VOID:
+		return llvm::Type::getVoidTy(llvm::getGlobalContext());
+	default:
+		// TODO error
+		return llvm::Type::getVoidTy(llvm::getGlobalContext());
+	}
 }