summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--MLKLLVMCompiler.mm138
1 files changed, 125 insertions, 13 deletions
diff --git a/MLKLLVMCompiler.mm b/MLKLLVMCompiler.mm
index 6c1f0c3..c376b17 100644
--- a/MLKLLVMCompiler.mm
+++ b/MLKLLVMCompiler.mm
@@ -25,14 +25,19 @@
#include <llvm/Analysis/Verifier.h>
#include <llvm/BasicBlock.h>
+#include <llvm/CallingConv.h>
#include <llvm/DerivedTypes.h>
#include <llvm/ExecutionEngine/ExecutionEngine.h>
+#include <llvm/Instructions.h>
#include <llvm/Module.h>
#include <llvm/ModuleProvider.h>
#include <llvm/PassManager.h>
#include <llvm/Support/IRBuilder.h>
#include <llvm/Value.h>
+#include <deque>
+#include <vector>
+
using namespace llvm;
@@ -40,6 +45,28 @@ static ExecutionEngine *execution_engine;
static llvm::Module *module;
static IRBuilder builder;
static FunctionPassManager *fpm;
+static Type *PointerTy;
+static Function *f_sel_get_uid;
+static Function *f_objc_msg_send;
+static Function *f_objc_get_class;
+
+
+static Constant
+*createGlobalStringPtr (const char *string)
+{
+ Constant *(indices[2]);
+ indices[0] = indices[1] = ConstantInt::get (Type::Int32Ty, 0);
+
+ Constant *str = ConstantArray::get (string);
+ Constant *str2 = new GlobalVariable (str->getType(),
+ true,
+ GlobalValue::InternalLinkage,
+ str,
+ "",
+ module);
+ Constant *ptr = ConstantExpr::getGetElementPtr (str2, indices, 2);
+ return ptr;
+}
@implementation MLKLLVMCompiler
@@ -47,6 +74,48 @@ static FunctionPassManager *fpm;
{
module = new llvm::Module ("MLKLLVMModule");
execution_engine = ExecutionEngine::create (module);
+ PointerTy = PointerType::get(Type::Int8Ty, 0);
+
+ std::vector<const Type*> argtypes1 (1, PointerTy);
+ FunctionType* ftype = FunctionType::get (PointerTy, argtypes1, false);
+ f_sel_get_uid = Function::Create(ftype,
+ GlobalValue::ExternalLinkage,
+#ifdef __NEXT_RUNTIME__
+ "sel_getUid",
+#else
+ "sel_get_uid",
+#endif
+ module);
+ //sel_get_uid->setCallingConv (CallingConv::C);
+
+
+ std::vector<const Type*> argtypes2 (2, PointerTy);
+ ftype = FunctionType::get (PointerTy, argtypes2, true);
+ f_objc_msg_send = Function::Create(ftype,
+ GlobalValue::ExternalLinkage,
+#ifdef __NEXT_RUNTIME__
+ "objc_msgSend",
+#else
+ "objc_msg_send",
+#endif
+ module);
+
+
+#if 0
+ std::vector<const Type*> argtypes3 (1, PointerTy);
+ ftype = FunctionType::get (PointerTy, argtypes3, false);
+ f_objc_get_class = Function::Create(ftype,
+ GlobalValue::ExternalLinkage,
+#ifdef __NEXT_RUNTIME__
+ "objc_getClass",
+#else
+ "objc_get_class",
+#endif
+ module);
+ f_objc_get_class->setCallingConv (CallingConv::C);
+
+ f_objc_get_class->dump();
+#endif
}
+(id) compile:(id)object
@@ -55,7 +124,7 @@ static FunctionPassManager *fpm;
Value *v = NULL;
BasicBlock *block;
std::vector<const Type*> noargs (0, Type::VoidTy);
- FunctionType *function_type = FunctionType::get (PointerType::get (Type::Int8Ty, 0),
+ FunctionType *function_type = FunctionType::get (PointerTy,
noargs,
false);
Function *function = Function::Create (function_type,
@@ -67,11 +136,14 @@ static FunctionPassManager *fpm;
block = BasicBlock::Create ("entry", function);
builder.SetInsertPoint (block);
+ function->dump();
+
v = [self processForm:[MLKForm formWithObject:object
inContext:context
forCompiler:self]];
builder.CreateRet (v);
+ function->dump();
verifyFunction (*function);
fpm->run (*function);
@@ -90,13 +162,22 @@ static FunctionPassManager *fpm;
return [form processForLLVM];
}
++(Value *) insertSelectorLookup:(NSString *)name
+{
+ llvm::Constant *nameval = ConstantArray::get ([name UTF8String]);
+ return builder.CreateCall (f_sel_get_uid, nameval);
+}
+
+(Value *) insertMethodCall:(NSString *)messageName
onObject:(Value *)object
withArgumentVector:(std::vector<Value*> *)argv
{
-#ifdef __NEXT_RUNTIME__
-#else
-#endif
+ Value *sel = [self insertSelectorLookup:messageName];
+
+ std::deque <Value *> argd (argv->begin(), argv->end());
+ argd.push_front (object);
+ argd.push_front (sel);
+ return builder.CreateCall (f_objc_msg_send, argd.begin(), argd.end());
}
+(Value *) insertMethodCall:(NSString *)messageName
@@ -108,8 +189,30 @@ static FunctionPassManager *fpm;
withArgumentVector:&argv];
}
-+(Value *) insertFindClass:(NSString *)className
++(Value *) insertClassLookup:(NSString *)className
{
+// Constant *nameval = ConstantArray::get (std::string ([className UTF8String]));
+// GlobalVariable *namevar = new GlobalVariable (nameval->getType(), //ArrayType::get(IntegerType::get (8), 4),
+// true,
+// GlobalValue::InternalLinkage,
+// nameval,
+// ".blargh");
+// Value *ptr = builder.CreateGEP (namevar, Constant::getNullValue(IntegerType::get(32)));
+
+// nameval->dump();
+// namevar->dump();
+
+ Constant *function =
+ module->getOrInsertFunction ("objc_get_class",
+ PointerTy,
+ PointerTy,
+ NULL);
+
+ // Value *nameptr = builder.CreateGlobalStringPtr ([className UTF8String], ".blargh");
+ Constant *nameptr = createGlobalStringPtr ([className UTF8String]);
+ nameptr->dump();
+ builder.GetInsertBlock()->getParent()->dump();
+ return builder.CreateCall (function, nameptr);
}
@end
@@ -128,7 +231,7 @@ static FunctionPassManager *fpm;
{
NSEnumerator *e = [_bodyForms objectEnumerator];
MLKForm *form;
- Value *value;
+ Value *value = NULL;
if ([_bodyForms count] == 0)
value = ConstantPointerNull::get (PointerType::get(Type::Int8Ty, 0));
@@ -137,7 +240,7 @@ static FunctionPassManager *fpm;
{
value = [form processForLLVM];
}
-
+
return value;
}
@end
@@ -221,6 +324,7 @@ static FunctionPassManager *fpm;
Value *endmarker = builder.CreateIntToPtr (ConstantInt::get(Type::Int64Ty,
(uint64_t)MLKEndOfArgumentsMarker,
false),
+ PointerTy);
args.push_back (endmarker);
CallInst *call = builder.CreateCall (functionPtr,
@@ -236,19 +340,21 @@ static FunctionPassManager *fpm;
@implementation MLKSimpleLambdaForm (MLKLLVMCompilation)
-(Value *) processForLLVM
{
- std::vector <const Type *> argtypes (1, PointerType::get(Type::Int8Ty, 0));
- FunctionType *ftype = FunctionType::get (PointerType::get(Type::Int8Ty, 0),
- argtypes,
- true);
+ std::vector <const Type *> argtypes (1, PointerTy);
+ FunctionType *ftype = FunctionType::get (PointerTy, argtypes, true);
Function *function = Function::Create (ftype,
Function::ExternalLinkage,
"",
module);
+ function->dump();
+
BasicBlock *initBlock = BasicBlock::Create ("init_function", function);
BasicBlock *loopBlock = BasicBlock::Create ("load_args");
BasicBlock *loopInitBlock = BasicBlock::Create ("load_args_init");
BasicBlock *joinBlock = BasicBlock::Create ("after_load_args");
+ function->dump();
+
builder.SetInsertPoint (initBlock);
Value *endmarker = builder.CreateIntToPtr (ConstantInt::get(Type::Int64Ty,
@@ -256,10 +362,15 @@ static FunctionPassManager *fpm;
false),
PointerType::get(Type::Int8Ty, 0));
+ function->dump();
+
Value *ap = builder.CreateAlloca (Type::Int8Ty);
- Value *nsmutablearray = [_compiler insertFindClass:@"NSMutableArray"];
- Value *mlkcons = [_compiler insertFindClass:@"MLKCons"];
+ function->dump();
+
+ Value *nsmutablearray = [_compiler insertClassLookup:@"NSMutableArray"];
+ function->dump();
+ Value *mlkcons = [_compiler insertClassLookup:@"MLKCons"];
Value *lambdaList = builder.CreateAlloca (PointerType::get (Type::Int8Ty, 0));
builder.CreateStore ([_compiler insertMethodCall:@"array"
@@ -292,6 +403,7 @@ static FunctionPassManager *fpm;
withArgumentVector:&argv],
lambdaList);
+ function->dump();
verifyFunction (*function);
fpm->run (*function);
return function;