Mercurial > projects > ldc
view gen/aa.cpp @ 109:5ab8e92611f9 trunk
[svn r113] Added initial support for associative arrays (AAs).
Fixed some problems with the string runtime support functions.
Fixed initialization of array of structs.
Fixed slice assignment where LHS is slice but RHS is dynamic array.
Fixed problems with result of assignment expressions.
Fixed foreach problems with key type mismatches.
author | lindquist |
---|---|
date | Wed, 21 Nov 2007 04:13:15 +0100 |
parents | |
children | facc562f5674 |
line wrap: on
line source
#include "gen/llvm.h" #include "mtype.h" #include "declaration.h" #include "aggregate.h" #include "gen/aa.h" #include "gen/runtime.h" #include "gen/tollvm.h" #include "gen/logger.h" #include "gen/irstate.h" #include "gen/dvalue.h" // makes sure the key value lives in memory so it can be passed to the runtime functions without problems // returns the pointer static llvm::Value* to_pkey(DValue* key) { Type* keytype = key->getType(); bool needmem = !DtoIsPassedByRef(keytype); llvm::Value* pkey; if (key->isIm()) { pkey = key->getRVal(); } else if (DVarValue* var = key->isVar()) { if (var->lval) { pkey = key->getLVal(); needmem = false; } else { pkey = key->getRVal(); } } else if (key->isConst()) { needmem = true; pkey = key->getRVal(); } else { assert(0); } // give memory if (needmem) { llvm::Value* tmp = new llvm::AllocaInst(DtoType(keytype), "aatmpkeystorage", gIR->topallocapoint()); DtoStore(pkey, tmp); pkey = tmp; } return pkey; } ///////////////////////////////////////////////////////////////////////////////////// DValue* DtoAAIndex(Type* type, DValue* aa, DValue* key) { // call: // extern(C) void* _aaGet(AA* aa, TypeInfo keyti, void* pkey, size_t valuesize) // first get the runtime function llvm::Function* func = LLVM_D_GetRuntimeFunction(gIR->module, "_aaGet"); const llvm::FunctionType* funcTy = func->getFunctionType(); // aa param llvm::Value* aaval = aa->getLVal(); aaval = DtoBitCast(aaval, funcTy->getParamType(0)); // keyti param Type* keytype = key->getType(); keytype->getTypeInfo(NULL); TypeInfoDeclaration* tid = keytype->getTypeInfoDeclaration(); assert(tid); DtoResolveDsymbol(Type::typeinfo); DtoForceDeclareDsymbol(tid); assert(tid->llvmValue); llvm::Value* keyti = tid->llvmValue; keyti = DtoBitCast(keyti, funcTy->getParamType(1)); // pkey param llvm::Value* pkey = to_pkey(key); pkey = DtoBitCast(pkey, funcTy->getParamType(2)); // valuesize param llvm::Value* valsize = DtoConstSize_t(gTargetData->getTypeSize(DtoType(type))); // build arg vector std::vector<llvm::Value*> args; args.push_back(aaval); args.push_back(keyti); args.push_back(pkey); args.push_back(valsize); // call runtime llvm::Value* ret = gIR->ir->CreateCall(func, args.begin(), args.end(), "aaGet"); // cast return value const llvm::Type* targettype = llvm::PointerType::get(DtoType(type)); if (ret->getType() != targettype) ret = DtoBitCast(ret, targettype); return new DVarValue(type, ret, true); } ///////////////////////////////////////////////////////////////////////////////////// DValue* DtoAAIn(Type* type, DValue* aa, DValue* key) { // call: // extern(C) void* _aaIn(AA aa*, TypeInfo keyti, void* pkey) // first get the runtime function llvm::Function* func = LLVM_D_GetRuntimeFunction(gIR->module, "_aaIn"); const llvm::FunctionType* funcTy = func->getFunctionType(); Logger::cout() << "_aaIn = " << *func << '\n'; // aa param llvm::Value* aaval = aa->getRVal(); Logger::cout() << "aaval: " << *aaval << '\n'; Logger::cout() << "totype: " << *funcTy->getParamType(0) << '\n'; aaval = DtoBitCast(aaval, funcTy->getParamType(0)); // keyti param Type* keytype = key->getType(); keytype->getTypeInfo(NULL); TypeInfoDeclaration* tid = keytype->getTypeInfoDeclaration(); assert(tid); DtoResolveDsymbol(Type::typeinfo); DtoForceDeclareDsymbol(tid); assert(tid->llvmValue); llvm::Value* keyti = tid->llvmValue; keyti = DtoBitCast(keyti, funcTy->getParamType(1)); // pkey param llvm::Value* pkey = to_pkey(key); pkey = DtoBitCast(pkey, funcTy->getParamType(2)); // build arg vector std::vector<llvm::Value*> args; args.push_back(aaval); args.push_back(keyti); args.push_back(pkey); // call runtime llvm::Value* ret = gIR->ir->CreateCall(func, args.begin(), args.end(), "aaIn"); // cast return value const llvm::Type* targettype = DtoType(type); if (ret->getType() != targettype) ret = DtoBitCast(ret, targettype); return new DImValue(type, ret); } /////////////////////////////////////////////////////////////////////////////////////