Mercurial > projects > ldc
view gen/arrays.cpp @ 650:aa6a0b7968f7
Added test case for bug #100
Removed dubious check for not emitting static private global in other modules without access. This should be handled properly somewhere else, it's causing unresolved global errors for stuff that should work (in MiniD)
author | Tomas Lindquist Olsen <tomas.l.olsen@gmail.com> |
---|---|
date | Sun, 05 Oct 2008 17:28:15 +0200 |
parents | df196c8dea26 |
children | 93433f4b6963 |
line wrap: on
line source
#include "gen/llvm.h" #include "mtype.h" #include "dsymbol.h" #include "aggregate.h" #include "declaration.h" #include "init.h" #include "gen/irstate.h" #include "gen/tollvm.h" #include "gen/llvmhelpers.h" #include "gen/arrays.h" #include "gen/runtime.h" #include "gen/logger.h" #include "gen/dvalue.h" ////////////////////////////////////////////////////////////////////////////////////////// const LLStructType* DtoArrayType(Type* arrayTy) { assert(arrayTy->next); const LLType* elemty = DtoType(arrayTy->next); if (elemty == LLType::VoidTy) elemty = LLType::Int8Ty; return LLStructType::get(DtoSize_t(), getPtrToType(elemty), 0); } const LLStructType* DtoArrayType(const LLType* t) { return LLStructType::get(DtoSize_t(), getPtrToType(t), 0); } ////////////////////////////////////////////////////////////////////////////////////////// const LLArrayType* DtoStaticArrayType(Type* t) { t = t->toBasetype(); assert(t->ty == Tsarray); TypeSArray* tsa = (TypeSArray*)t; Type* tnext = tsa->next; const LLType* elemty = DtoType(tnext); if (elemty == LLType::VoidTy) elemty = LLType::Int8Ty; return LLArrayType::get(elemty, tsa->dim->toUInteger()); } ////////////////////////////////////////////////////////////////////////////////////////// void DtoSetArrayToNull(LLValue* v) { Logger::println("DtoSetArrayToNull"); LOG_SCOPE; LLValue* len = DtoGEPi(v,0,0); LLValue* zerolen = llvm::ConstantInt::get(len->getType()->getContainedType(0), 0, false); DtoStore(zerolen, len); LLValue* ptr = DtoGEPi(v,0,1); const LLPointerType* pty = isaPointer(ptr->getType()->getContainedType(0)); LLValue* nullptr = llvm::ConstantPointerNull::get(pty); DtoStore(nullptr, ptr); } ////////////////////////////////////////////////////////////////////////////////////////// void DtoArrayAssign(LLValue* dst, LLValue* src) { Logger::println("DtoArrayAssign"); LOG_SCOPE; assert(gIR); if (dst->getType() == src->getType()) { LLValue* ptr = DtoGEPi(src,0,0); LLValue* val = DtoLoad(ptr); ptr = DtoGEPi(dst,0,0); DtoStore(val, ptr); ptr = DtoGEPi(src,0,1); val = DtoLoad(ptr); ptr = DtoGEPi(dst,0,1); DtoStore(val, ptr); } else { if (Logger::enabled()) Logger::cout() << "array assignment type dont match: " << *dst->getType() << "\n\n" << *src->getType() << '\n'; const LLArrayType* arrty = isaArray(src->getType()->getContainedType(0)); if (!arrty) { std::cout << "invalid: " << *src << '\n'; assert(0); } const LLType* dstty = getPtrToType(arrty->getElementType()); LLValue* dstlen = DtoGEPi(dst,0,0); LLValue* srclen = DtoConstSize_t(arrty->getNumElements()); DtoStore(srclen, dstlen); LLValue* dstptr = DtoGEPi(dst,0,1); LLValue* srcptr = DtoBitCast(src, dstty); DtoStore(srcptr, dstptr); } } ////////////////////////////////////////////////////////////////////////////////////////// void DtoArrayInit(Loc& loc, DValue* array, DValue* value) { Logger::println("DtoArrayInit"); LOG_SCOPE; LLValue* dim = DtoArrayLen(array); LLValue* ptr = DtoArrayPtr(array); LLValue* val; // give slices and complex values storage (and thus an address to pass) if (value->isSlice()) { val = DtoAlloca(DtoType(value->getType()), ".tmpparam"); DVarValue lval(value->getType(), val); DtoAssign(loc, &lval, value); } else { val = value->getRVal(); } assert(val); // prepare runtime call LLSmallVector<LLValue*, 4> args; args.push_back(ptr); args.push_back(dim); args.push_back(val); // determine the right runtime function to call const char* funcname = NULL; Type* arrayelemty = array->getType()->nextOf()->toBasetype(); Type* valuety = value->getType()->toBasetype(); // lets first optimize all zero initializations down to a memset. // this simplifies codegen later on as llvm null's have no address! if (isaConstant(val) && isaConstant(val)->isNullValue()) { size_t X = getABITypeSize(val->getType()); LLValue* nbytes = gIR->ir->CreateMul(dim, DtoConstSize_t(X), ".nbytes"); DtoMemSetZero(ptr, nbytes); return; } // if not a zero initializer, call the appropriate runtime function! switch (arrayelemty->ty) { case Tbool: funcname = "_d_array_init_i1"; break; case Tvoid: case Tchar: case Tint8: case Tuns8: funcname = "_d_array_init_i8"; break; case Twchar: case Tint16: case Tuns16: funcname = "_d_array_init_i16"; break; case Tdchar: case Tint32: case Tuns32: funcname = "_d_array_init_i32"; break; case Tint64: case Tuns64: funcname = "_d_array_init_i64"; break; case Tfloat32: case Timaginary32: funcname = "_d_array_init_float"; break; case Tfloat64: case Timaginary64: funcname = "_d_array_init_double"; break; case Tfloat80: case Timaginary80: funcname = "_d_array_init_real"; break; case Tcomplex32: funcname = "_d_array_init_cfloat"; break; case Tcomplex64: funcname = "_d_array_init_cdouble"; break; case Tcomplex80: funcname = "_d_array_init_creal"; break; case Tpointer: case Tclass: funcname = "_d_array_init_pointer"; args[0] = DtoBitCast(args[0], getPtrToType(getVoidPtrType())); args[2] = DtoBitCast(args[2], getVoidPtrType()); break; // this currently acts as a kind of fallback for all the bastards... // FIXME: this is probably too slow. case Tstruct: case Tdelegate: case Tarray: case Tsarray: funcname = "_d_array_init_mem"; assert(arrayelemty == valuety && "ArrayInit doesn't work on elem-initialized static arrays"); args[0] = DtoBitCast(args[0], getVoidPtrType()); args[2] = DtoBitCast(args[2], getVoidPtrType()); args.push_back(DtoConstSize_t(getABITypeSize(DtoType(arrayelemty)))); break; default: error("unhandled array init: %s = %s", array->getType()->toChars(), value->getType()->toChars()); assert(0 && "unhandled array init"); } LLFunction* fn = LLVM_D_GetRuntimeFunction(gIR->module, funcname); assert(fn); if (Logger::enabled()) Logger::cout() << "calling array init function: " << *fn <<'\n'; CallOrInvoke* call = gIR->CreateCallOrInvoke(fn, args.begin(), args.end()); call->setCallingConv(llvm::CallingConv::C); } ////////////////////////////////////////////////////////////////////////////////////////// void DtoSetArray(LLValue* arr, LLValue* dim, LLValue* ptr) { Logger::println("SetArray"); assert(isaStruct(arr->getType()->getContainedType(0))); DtoStore(dim, DtoGEPi(arr,0,0)); DtoStore(ptr, DtoGEPi(arr,0,1)); } ////////////////////////////////////////////////////////////////////////////////////////// LLConstant* DtoConstArrayInitializer(ArrayInitializer* arrinit) { Logger::println("DtoConstArrayInitializer: %s | %s", arrinit->toChars(), arrinit->type->toChars()); LOG_SCOPE; Type* arrinittype = arrinit->type->toBasetype(); Type* t; integer_t tdim; if (arrinittype->ty == Tsarray) { Logger::println("static array"); TypeSArray* tsa = (TypeSArray*)arrinittype; tdim = tsa->dim->toInteger(); t = tsa; } else if (arrinittype->ty == Tarray) { Logger::println("dynamic array"); t = arrinittype; tdim = arrinit->dim; } else assert(0); if(arrinit->dim > tdim) error(arrinit->loc, "array initializer for %s is too long (%d)", arrinit->type->toChars(), arrinit->dim); Logger::println("dim = %u", tdim); std::vector<LLConstant*> inits(tdim, NULL); Type* arrnext = arrinittype->next; const LLType* elemty = DtoType(arrinittype->next); assert(arrinit->index.dim == arrinit->value.dim); for (unsigned i=0,j=0; i < tdim; ++i) { Initializer* init = 0; Expression* idx; if (j < arrinit->index.dim) idx = (Expression*)arrinit->index.data[j]; else idx = NULL; LLConstant* v = NULL; if (idx) { Logger::println("%d has idx", i); // this is pretty weird :/ idx->type turned out NULL for the initializer: // const in6_addr IN6ADDR_ANY = { s6_addr8: [0] }; // in std.c.linux.socket if (idx->type) { Logger::println("has idx->type", i); //integer_t k = idx->toInteger(); //Logger::println("getting value for exp: %s | %s", idx->toChars(), arrnext->toChars()); LLConstant* cc = idx->toConstElem(gIR); Logger::println("value gotten"); assert(cc != NULL); LLConstantInt* ci = llvm::dyn_cast<LLConstantInt>(cc); assert(ci != NULL); uint64_t k = ci->getZExtValue(); if (i == k) { init = (Initializer*)arrinit->value.data[j]; assert(init); ++j; } } } else { if (j < arrinit->value.dim) { init = (Initializer*)arrinit->value.data[j]; ++j; } else v = arrnext->defaultInit()->toConstElem(gIR); } if (!v) v = DtoConstInitializer(t->next, init); assert(v); inits[i] = v; if (Logger::enabled()) Logger::cout() << "llval: " << *v << '\n'; } Logger::println("building constant array"); const LLArrayType* arrty = LLArrayType::get(elemty,tdim); LLConstant* constarr = LLConstantArray::get(arrty, inits); if (arrinittype->ty == Tsarray) return constarr; else assert(arrinittype->ty == Tarray); LLGlobalVariable* gvar = new LLGlobalVariable(arrty,true,LLGlobalValue::InternalLinkage,constarr,".constarray",gIR->module); LLConstant* idxs[2] = { DtoConstUint(0), DtoConstUint(0) }; LLConstant* gep = llvm::ConstantExpr::getGetElementPtr(gvar,idxs,2); return DtoConstSlice(DtoConstSize_t(tdim),gep); } ////////////////////////////////////////////////////////////////////////////////////////// static LLValue* get_slice_ptr(DSliceValue* e, LLValue*& sz) { assert(e->len != 0); const LLType* t = e->ptr->getType()->getContainedType(0); sz = gIR->ir->CreateMul(DtoConstSize_t(getABITypeSize(t)), e->len, "tmp"); return e->ptr; } void DtoArrayCopySlices(DSliceValue* dst, DSliceValue* src) { Logger::println("ArrayCopySlices"); LLValue *sz1,*sz2; LLValue* dstarr = get_slice_ptr(dst,sz1); LLValue* srcarr = get_slice_ptr(src,sz2); DtoMemCpy(dstarr, srcarr, sz1); } void DtoArrayCopyToSlice(DSliceValue* dst, DValue* src) { Logger::println("ArrayCopyToSlice"); LLValue* sz1; LLValue* dstarr = get_slice_ptr(dst,sz1); LLValue* srcarr = DtoArrayPtr(src); DtoMemCpy(dstarr, srcarr, sz1); } ////////////////////////////////////////////////////////////////////////////////////////// void DtoStaticArrayCopy(LLValue* dst, LLValue* src) { Logger::println("StaticArrayCopy"); size_t n = getABITypeSize(dst->getType()->getContainedType(0)); DtoMemCpy(dst, src, DtoConstSize_t(n)); } ////////////////////////////////////////////////////////////////////////////////////////// LLConstant* DtoConstSlice(LLConstant* dim, LLConstant* ptr) { LLConstant* values[2] = { dim, ptr }; return llvm::ConstantStruct::get(values, 2); } ////////////////////////////////////////////////////////////////////////////////////////// DSliceValue* DtoNewDynArray(Loc& loc, Type* arrayType, DValue* dim, bool defaultInit) { Logger::println("DtoNewDynArray : %s", arrayType->toChars()); LOG_SCOPE; // typeinfo arg LLValue* arrayTypeInfo = DtoTypeInfoOf(arrayType); // dim arg assert(DtoType(dim->getType()) == DtoSize_t()); LLValue* arrayLen = dim->getRVal(); // get runtime function bool zeroInit = arrayType->toBasetype()->nextOf()->isZeroInit(); const char* fnname = defaultInit ? (zeroInit ? "_d_newarrayT" : "_d_newarrayiT") : "_d_newarrayvT"; LLFunction* fn = LLVM_D_GetRuntimeFunction(gIR->module, fnname); // call allocator LLValue* newptr = gIR->CreateCallOrInvoke2(fn, arrayTypeInfo, arrayLen, ".gc_mem")->get(); // cast to wanted type const LLType* dstType = DtoType(arrayType)->getContainedType(1); if (newptr->getType() != dstType) newptr = DtoBitCast(newptr, dstType, ".gc_mem"); if (Logger::enabled()) Logger::cout() << "final ptr = " << *newptr << '\n'; return new DSliceValue(arrayType, arrayLen, newptr); } ////////////////////////////////////////////////////////////////////////////////////////// DSliceValue* DtoNewMulDimDynArray(Loc& loc, Type* arrayType, DValue** dims, size_t ndims, bool defaultInit) { Logger::println("DtoNewMulDimDynArray : %s", arrayType->toChars()); LOG_SCOPE; // typeinfo arg LLValue* arrayTypeInfo = DtoTypeInfoOf(arrayType); // get value type Type* vtype = arrayType->toBasetype(); for (size_t i=0; i<ndims; ++i) vtype = vtype->nextOf(); // get runtime function bool zeroInit = vtype->isZeroInit(); const char* fnname = defaultInit ? (zeroInit ? "_d_newarraymT" : "_d_newarraymiT") : "_d_newarraymvT"; LLFunction* fn = LLVM_D_GetRuntimeFunction(gIR->module, fnname); // build dims LLValue* dimsArg = DtoAlloca(DtoSize_t(), DtoConstUint(ndims), ".newdims"); LLValue* firstDim = NULL; for (size_t i=0; i<ndims; ++i) { LLValue* dim = dims[i]->getRVal(); if (!firstDim) firstDim = dim; DtoStore(dim, DtoGEPi1(dimsArg, i)); } // call allocator LLValue* newptr = gIR->CreateCallOrInvoke3(fn, arrayTypeInfo, DtoConstSize_t(ndims), dimsArg, ".gc_mem")->get(); // cast to wanted type const LLType* dstType = DtoType(arrayType)->getContainedType(1); if (newptr->getType() != dstType) newptr = DtoBitCast(newptr, dstType, ".gc_mem"); if (Logger::enabled()) Logger::cout() << "final ptr = " << *newptr << '\n'; assert(firstDim); return new DSliceValue(arrayType, firstDim, newptr); } ////////////////////////////////////////////////////////////////////////////////////////// DSliceValue* DtoResizeDynArray(Type* arrayType, DValue* array, DValue* newdim) { Logger::println("DtoResizeDynArray : %s", arrayType->toChars()); LOG_SCOPE; assert(array); assert(newdim); assert(arrayType); assert(arrayType->toBasetype()->ty == Tarray); // decide on what runtime function to call based on whether the type is zero initialized bool zeroInit = arrayType->toBasetype()->next->isZeroInit(); // call runtime LLFunction* fn = LLVM_D_GetRuntimeFunction(gIR->module, zeroInit ? "_d_arraysetlengthT" : "_d_arraysetlengthiT" ); LLSmallVector<LLValue*,4> args; args.push_back(DtoTypeInfoOf(arrayType)); args.push_back(newdim->getRVal()); args.push_back(DtoArrayLen(array)); LLValue* arrPtr = DtoArrayPtr(array); if (Logger::enabled()) Logger::cout() << "arrPtr = " << *arrPtr << '\n'; args.push_back(DtoBitCast(arrPtr, fn->getFunctionType()->getParamType(3), "tmp")); LLValue* newptr = gIR->CreateCallOrInvoke(fn, args.begin(), args.end(), ".gc_mem")->get(); if (newptr->getType() != arrPtr->getType()) newptr = DtoBitCast(newptr, arrPtr->getType(), ".gc_mem"); return new DSliceValue(arrayType, newdim->getRVal(), newptr); } ////////////////////////////////////////////////////////////////////////////////////////// DSliceValue* DtoCatAssignElement(DValue* array, Expression* exp) { Logger::println("DtoCatAssignElement"); LOG_SCOPE; assert(array); LLValue* idx = DtoArrayLen(array); LLValue* one = DtoConstSize_t(1); LLValue* len = gIR->ir->CreateAdd(idx,one,"tmp"); DValue* newdim = new DImValue(Type::tsize_t, len); DSliceValue* slice = DtoResizeDynArray(array->getType(), array, newdim); LLValue* ptr = slice->ptr; ptr = llvm::GetElementPtrInst::Create(ptr, idx, "tmp", gIR->scopebb()); DValue* dptr = new DVarValue(exp->type, ptr); DValue* e = exp->toElem(gIR); DtoAssign(exp->loc, dptr, e); return slice; } ////////////////////////////////////////////////////////////////////////////////////////// DSliceValue* DtoCatAssignArray(DValue* arr, Expression* exp) { Logger::println("DtoCatAssignArray"); LOG_SCOPE; DValue* e = exp->toElem(gIR); llvm::Value *len1, *len2, *src1, *src2, *res; len1 = DtoArrayLen(arr); len2 = DtoArrayLen(e); res = gIR->ir->CreateAdd(len1,len2,"tmp"); DValue* newdim = new DImValue(Type::tsize_t, res); DSliceValue* slice = DtoResizeDynArray(arr->getType(), arr, newdim); src1 = slice->ptr; src2 = DtoArrayPtr(e); // advance ptr src1 = gIR->ir->CreateGEP(src1,len1,"tmp"); // memcpy LLValue* elemSize = DtoConstSize_t(getABITypeSize(src2->getType()->getContainedType(0))); LLValue* bytelen = gIR->ir->CreateMul(len2, elemSize, "tmp"); DtoMemCpy(src1,src2,bytelen); return slice; } ////////////////////////////////////////////////////////////////////////////////////////// DSliceValue* DtoCatArrays(Type* type, Expression* exp1, Expression* exp2) { Logger::println("DtoCatArrays"); LOG_SCOPE; Type* t1 = exp1->type->toBasetype(); Type* t2 = exp2->type->toBasetype(); assert(t1->ty == Tarray || t1->ty == Tsarray); assert(t2->ty == Tarray || t2->ty == Tsarray); DValue* e1 = exp1->toElem(gIR); DValue* e2 = exp2->toElem(gIR); llvm::Value *len1, *len2, *src1, *src2, *res; len1 = DtoArrayLen(e1); len2 = DtoArrayLen(e2); res = gIR->ir->CreateAdd(len1,len2,"tmp"); DValue* lenval = new DImValue(Type::tsize_t, res); DSliceValue* slice = DtoNewDynArray(exp1->loc, type, lenval, false); LLValue* mem = slice->ptr; src1 = DtoArrayPtr(e1); src2 = DtoArrayPtr(e2); // first memcpy LLValue* elemSize = DtoConstSize_t(getABITypeSize(src1->getType()->getContainedType(0))); LLValue* bytelen = gIR->ir->CreateMul(len1, elemSize, "tmp"); DtoMemCpy(mem,src1,bytelen); // second memcpy mem = gIR->ir->CreateGEP(mem,len1,"tmp"); bytelen = gIR->ir->CreateMul(len2, elemSize, "tmp"); DtoMemCpy(mem,src2,bytelen); return slice; } ////////////////////////////////////////////////////////////////////////////////////////// DSliceValue* DtoCatArrayElement(Type* type, Expression* exp1, Expression* exp2) { Logger::println("DtoCatArrayElement"); LOG_SCOPE; Type* t1 = exp1->type->toBasetype(); Type* t2 = exp2->type->toBasetype(); DValue* e1 = exp1->toElem(gIR); DValue* e2 = exp2->toElem(gIR); llvm::Value *len1, *src1, *res; // handle prefix case, eg. int~int[] if (t2->next && t1 == t2->next->toBasetype()) { len1 = DtoArrayLen(e2); res = gIR->ir->CreateAdd(len1,DtoConstSize_t(1),"tmp"); DValue* lenval = new DImValue(Type::tsize_t, res); DSliceValue* slice = DtoNewDynArray(exp1->loc, type, lenval, false); LLValue* mem = slice->ptr; DVarValue* memval = new DVarValue(e1->getType(), mem); DtoAssign(exp1->loc, memval, e1); src1 = DtoArrayPtr(e2); mem = gIR->ir->CreateGEP(mem,DtoConstSize_t(1),"tmp"); LLValue* elemSize = DtoConstSize_t(getABITypeSize(src1->getType()->getContainedType(0))); LLValue* bytelen = gIR->ir->CreateMul(len1, elemSize, "tmp"); DtoMemCpy(mem,src1,bytelen); return slice; } // handle suffix case, eg. int[]~int else { len1 = DtoArrayLen(e1); res = gIR->ir->CreateAdd(len1,DtoConstSize_t(1),"tmp"); DValue* lenval = new DImValue(Type::tsize_t, res); DSliceValue* slice = DtoNewDynArray(exp1->loc, type, lenval, false); LLValue* mem = slice->ptr; src1 = DtoArrayPtr(e1); LLValue* elemSize = DtoConstSize_t(getABITypeSize(src1->getType()->getContainedType(0))); LLValue* bytelen = gIR->ir->CreateMul(len1, elemSize, "tmp"); DtoMemCpy(mem,src1,bytelen); mem = gIR->ir->CreateGEP(mem,len1,"tmp"); DVarValue* memval = new DVarValue(e2->getType(), mem); DtoAssign(exp1->loc, memval, e2); return slice; } } ////////////////////////////////////////////////////////////////////////////////////////// // helper for eq and cmp static LLValue* DtoArrayEqCmp_impl(Loc& loc, const char* func, DValue* l, DValue* r, bool useti) { Logger::println("comparing arrays"); LLFunction* fn = LLVM_D_GetRuntimeFunction(gIR->module, func); assert(fn); LLValue* lmem; LLValue* rmem; // cast static arrays to dynamic ones, this turns them into DSliceValues Logger::println("casting to dynamic arrays"); Type* l_ty = l->getType()->toBasetype(); Type* r_ty = r->getType()->toBasetype(); assert(l_ty->next == r_ty->next); if ((l_ty->ty == Tsarray) || (r_ty->ty == Tsarray)) { Type* a_ty = l_ty->next->arrayOf(); if (l_ty->ty == Tsarray) l = DtoCastArray(loc, l, a_ty); if (r_ty->ty == Tsarray) r = DtoCastArray(loc, r, a_ty); } Logger::println("giving storage"); // we need to give slices storage if (l->isSlice()) { lmem = DtoAlloca(DtoType(l->getType()), "tmpparam"); DtoSetArray(lmem, DtoArrayLen(l), DtoArrayPtr(l)); } // also null else if (l->isNull()) { lmem = DtoAlloca(DtoType(l->getType()), "tmpparam"); DtoSetArray(lmem, llvm::Constant::getNullValue(DtoSize_t()), llvm::Constant::getNullValue(DtoType(l->getType()->next->pointerTo()))); } else lmem = l->getRVal(); // and for the rvalue ... // we need to give slices storage if (r->isSlice()) { rmem = DtoAlloca(DtoType(r->getType()), "tmpparam"); DtoSetArray(rmem, DtoArrayLen(r), DtoArrayPtr(r)); } // also null else if (r->isNull()) { rmem = DtoAlloca(DtoType(r->getType()), "tmpparam"); DtoSetArray(rmem, llvm::Constant::getNullValue(DtoSize_t()), llvm::Constant::getNullValue(DtoType(r->getType()->next->pointerTo()))); } else rmem = r->getRVal(); const LLType* pt = fn->getFunctionType()->getParamType(0); LLSmallVector<LLValue*, 3> args; if (Logger::enabled()) { Logger::cout() << "bitcasting to " << *pt << '\n'; Logger::cout() << *lmem << '\n'; Logger::cout() << *rmem << '\n'; } args.push_back(DtoBitCast(lmem,pt)); args.push_back(DtoBitCast(rmem,pt)); // pass array typeinfo ? if (useti) { Type* t = l->getType(); LLValue* tival = DtoTypeInfoOf(t); // DtoTypeInfoOf only does declare, not enough in this case :/ DtoForceConstInitDsymbol(t->vtinfo); if (Logger::enabled()) Logger::cout() << "typeinfo decl: " << *tival << '\n'; pt = fn->getFunctionType()->getParamType(2); args.push_back(DtoBitCast(tival, pt)); } CallOrInvoke* call = gIR->CreateCallOrInvoke(fn, args.begin(), args.end(), "tmp"); // set param attrs llvm::AttrListPtr palist; palist = palist.addAttr(1, llvm::Attribute::ByVal); palist = palist.addAttr(2, llvm::Attribute::ByVal); call->setAttributes(palist); return call->get(); } ////////////////////////////////////////////////////////////////////////////////////////// LLValue* DtoArrayEquals(Loc& loc, TOK op, DValue* l, DValue* r) { LLValue* res = DtoArrayEqCmp_impl(loc, "_adEq", l, r, true); res = gIR->ir->CreateICmpNE(res, DtoConstInt(0), "tmp"); if (op == TOKnotequal) res = gIR->ir->CreateNot(res, "tmp"); return res; } ////////////////////////////////////////////////////////////////////////////////////////// LLValue* DtoArrayCompare(Loc& loc, TOK op, DValue* l, DValue* r) { LLValue* res = 0; llvm::ICmpInst::Predicate cmpop; bool skip = false; switch(op) { case TOKlt: case TOKul: cmpop = llvm::ICmpInst::ICMP_SLT; break; case TOKle: case TOKule: cmpop = llvm::ICmpInst::ICMP_SLE; break; case TOKgt: case TOKug: cmpop = llvm::ICmpInst::ICMP_SGT; break; case TOKge: case TOKuge: cmpop = llvm::ICmpInst::ICMP_SGE; break; case TOKue: cmpop = llvm::ICmpInst::ICMP_EQ; break; case TOKlg: cmpop = llvm::ICmpInst::ICMP_NE; break; case TOKleg: skip = true; res = llvm::ConstantInt::getTrue(); break; case TOKunord: skip = true; res = llvm::ConstantInt::getFalse(); break; default: assert(0); } if (!skip) { Type* t = l->getType()->toBasetype()->next->toBasetype(); if (t->ty == Tchar) res = DtoArrayEqCmp_impl(loc, "_adCmpChar", l, r, false); else res = DtoArrayEqCmp_impl(loc, "_adCmp", l, r, true); res = gIR->ir->CreateICmp(cmpop, res, DtoConstInt(0), "tmp"); } assert(res); return res; } ////////////////////////////////////////////////////////////////////////////////////////// LLValue* DtoArrayCastLength(LLValue* len, const LLType* elemty, const LLType* newelemty) { Logger::println("DtoArrayCastLength"); LOG_SCOPE; assert(len); assert(elemty); assert(newelemty); size_t esz = getABITypeSize(elemty); size_t nsz = getABITypeSize(newelemty); if (esz == nsz) return len; LLSmallVector<LLValue*, 3> args; args.push_back(len); args.push_back(llvm::ConstantInt::get(DtoSize_t(), esz, false)); args.push_back(llvm::ConstantInt::get(DtoSize_t(), nsz, false)); LLFunction* fn = LLVM_D_GetRuntimeFunction(gIR->module, "_d_array_cast_len"); return gIR->CreateCallOrInvoke(fn, args.begin(), args.end(), "tmp")->get(); } ////////////////////////////////////////////////////////////////////////////////////////// LLValue* DtoDynArrayIs(TOK op, DValue* l, DValue* r) { LLValue *len1, *ptr1, *len2, *ptr2; assert(l); assert(r); // compare lengths len1 = DtoArrayLen(l); len2 = DtoArrayLen(r); LLValue* b1 = gIR->ir->CreateICmpEQ(len1,len2,"tmp"); // compare pointers ptr1 = DtoArrayPtr(l); ptr2 = DtoArrayPtr(r); LLValue* b2 = gIR->ir->CreateICmpEQ(ptr1,ptr2,"tmp"); // combine LLValue* res = gIR->ir->CreateAnd(b1,b2,"tmp"); // return result return (op == TOKnotidentity) ? gIR->ir->CreateNot(res) : res; } ////////////////////////////////////////////////////////////////////////////////////////// LLConstant* DtoConstStaticArray(const LLType* t, LLConstant* c) { const LLArrayType* at = isaArray(t); assert(at); if (isaArray(at->getElementType())) { c = DtoConstStaticArray(at->getElementType(), c); } else { assert(at->getElementType() == c->getType()); } std::vector<LLConstant*> initvals; initvals.resize(at->getNumElements(), c); return llvm::ConstantArray::get(at, initvals); } ////////////////////////////////////////////////////////////////////////////////////////// LLValue* DtoArrayLen(DValue* v) { Logger::println("DtoArrayLen"); LOG_SCOPE; Type* t = v->getType()->toBasetype(); if (t->ty == Tarray) { if (DSliceValue* s = v->isSlice()) return s->len; else if (v->isNull()) return DtoConstSize_t(0); return DtoLoad(DtoGEPi(v->getRVal(), 0,0)); } else if (t->ty == Tsarray) { assert(!v->isSlice()); assert(!v->isNull()); LLValue* rv = v->getRVal(); const LLArrayType* t = isaArray(rv->getType()->getContainedType(0)); assert(t); return DtoConstSize_t(t->getNumElements()); } assert(0 && "unsupported array for len"); return 0; } ////////////////////////////////////////////////////////////////////////////////////////// LLValue* DtoArrayPtr(DValue* v) { Logger::println("DtoArrayPtr"); LOG_SCOPE; Type* t = v->getType()->toBasetype(); if (t->ty == Tarray) { if (DSliceValue* s = v->isSlice()) return s->ptr; else if (v->isNull()) return getNullPtr(getPtrToType(DtoType(t->next))); return DtoLoad(DtoGEPi(v->getRVal(), 0,1)); } else if (t->ty == Tsarray) { assert(!v->isSlice()); assert(!v->isNull()); return DtoGEPi(v->getRVal(), 0,0); } assert(0); return 0; } ////////////////////////////////////////////////////////////////////////////////////////// DValue* DtoCastArray(Loc& loc, DValue* u, Type* to) { Logger::println("DtoCastArray"); LOG_SCOPE; const LLType* tolltype = DtoType(to); Type* totype = to->toBasetype(); Type* fromtype = u->getType()->toBasetype(); assert(fromtype->ty == Tarray || fromtype->ty == Tsarray); LLValue* rval; LLValue* rval2; bool isslice = false; if (Logger::enabled()) Logger::cout() << "from array or sarray" << '\n'; if (totype->ty == Tpointer) { if (Logger::enabled()) Logger::cout() << "to pointer" << '\n'; rval = DtoArrayPtr(u); if (rval->getType() != tolltype) rval = gIR->ir->CreateBitCast(rval, tolltype, "tmp"); } else if (totype->ty == Tarray) { if (Logger::enabled()) Logger::cout() << "to array" << '\n'; const LLType* ptrty = DtoArrayType(totype)->getContainedType(1); const LLType* ety = DtoTypeNotVoid(fromtype->next); if (DSliceValue* usl = u->isSlice()) { if (Logger::enabled()) { Logger::println("from slice"); Logger::cout() << "from: " << *usl->ptr << " to: " << *ptrty << '\n'; } rval = DtoBitCast(usl->ptr, ptrty); if (fromtype->next->size() == totype->next->size()) rval2 = DtoArrayLen(usl); else rval2 = DtoArrayCastLength(DtoArrayLen(usl), ety, ptrty->getContainedType(0)); } else { LLValue* uval = u->getRVal(); if (fromtype->ty == Tsarray) { if (Logger::enabled()) Logger::cout() << "uvalTy = " << *uval->getType() << '\n'; assert(isaPointer(uval->getType())); const LLArrayType* arrty = isaArray(uval->getType()->getContainedType(0)); if(arrty->getNumElements()*fromtype->next->size() % totype->next->size() != 0) { error(loc, "invalid cast from '%s' to '%s', the element sizes don't line up", fromtype->toChars(), totype->toChars()); fatal(); } rval2 = llvm::ConstantInt::get(DtoSize_t(), arrty->getNumElements(), false); rval2 = DtoArrayCastLength(rval2, ety, ptrty->getContainedType(0)); rval = DtoBitCast(uval, ptrty); } else { LLValue* zero = llvm::ConstantInt::get(LLType::Int32Ty, 0, false); LLValue* one = llvm::ConstantInt::get(LLType::Int32Ty, 1, false); rval2 = DtoGEP(uval,zero,zero); rval2 = DtoLoad(rval2); rval2 = DtoArrayCastLength(rval2, ety, ptrty->getContainedType(0)); rval = DtoGEP(uval,zero,one); rval = DtoLoad(rval); //Logger::cout() << *e->mem->getType() << '|' << *ptrty << '\n'; rval = DtoBitCast(rval, ptrty); } } isslice = true; } else if (totype->ty == Tsarray) { if (Logger::enabled()) Logger::cout() << "to sarray" << '\n'; assert(0); } else { assert(0); } if (isslice) { Logger::println("isslice"); return new DSliceValue(to, rval2, rval); } return new DImValue(to, rval); } ////////////////////////////////////////////////////////////////////////////////////////// void DtoArrayBoundsCheck(Loc& loc, DValue* arr, DValue* index, bool isslice) { Type* arrty = arr->getType()->toBasetype(); assert((arrty->ty == Tsarray || arrty->ty == Tarray) && "Can only array bounds check for static or dynamic arrays"); // static arrays could get static checks for static indices // but shouldn't since it might be generic code that's never executed // runtime check llvm::BasicBlock* oldend = gIR->scopeend(); llvm::BasicBlock* failbb = llvm::BasicBlock::Create("arrayboundscheckfail", gIR->topfunc(), oldend); llvm::BasicBlock* okbb = llvm::BasicBlock::Create("arrayboundsok", gIR->topfunc(), oldend); llvm::ICmpInst::Predicate cmpop = isslice ? llvm::ICmpInst::ICMP_ULE : llvm::ICmpInst::ICMP_ULT; LLValue* cond = gIR->ir->CreateICmp(cmpop, index->getRVal(), DtoArrayLen(arr), "boundscheck"); gIR->ir->CreateCondBr(cond, okbb, failbb); // set up failbb to call the array bounds error runtime function gIR->scope() = IRScope(failbb, okbb); std::vector<LLValue*> args; llvm::AttrListPtr palist; // file param // FIXME: every array bounds check creates a global for the filename !!! LLConstant* c = DtoConstString(loc.filename); llvm::AllocaInst* alloc = gIR->func()->srcfileArg; if (!alloc) { alloc = DtoAlloca(c->getType(), ".srcfile"); gIR->func()->srcfileArg = alloc; } LLValue* ptr = DtoGEPi(alloc, 0,0, "tmp"); DtoStore(c->getOperand(0), ptr); ptr = DtoGEPi(alloc, 0,1, "tmp"); DtoStore(c->getOperand(1), ptr); args.push_back(alloc); palist = palist.addAttr(1, llvm::Attribute::ByVal); // line param c = DtoConstUint(loc.linnum); args.push_back(c); // call llvm::Function* errorfn = LLVM_D_GetRuntimeFunction(gIR->module, "_d_array_bounds"); CallOrInvoke* call = gIR->CreateCallOrInvoke(errorfn, args.begin(), args.end()); call->setAttributes(palist); // the function does not return gIR->ir->CreateUnreachable(); // if ok, proceed in okbb gIR->scope() = IRScope(okbb, oldend); }