rsAllocation.cpp revision 0e61af91ecd0d1276ddfc1c43bd61f6587e7026b
1/* 2 * Copyright (C) 2013 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17#include "rsContext.h" 18#include "rsAllocation.h" 19#include "rsAdapter.h" 20#include "rs_hal.h" 21 22#if !defined(RS_SERVER) && !defined(RS_COMPATIBILITY_LIB) 23#include "system/window.h" 24#include "gui/GLConsumer.h" 25#endif 26 27using namespace android; 28using namespace android::renderscript; 29 30Allocation::Allocation(Context *rsc, const Type *type, uint32_t usages, 31 RsAllocationMipmapControl mc, void * ptr) 32 : ObjectBase(rsc) { 33 34 memset(&mHal, 0, sizeof(mHal)); 35 mHal.state.mipmapControl = RS_ALLOCATION_MIPMAP_NONE; 36 mHal.state.usageFlags = usages; 37 mHal.state.mipmapControl = mc; 38 mHal.state.userProvidedPtr = ptr; 39 40 setType(type); 41 updateCache(); 42} 43 44void Allocation::operator delete(void* ptr) { 45 if (ptr) { 46 Allocation *a = (Allocation*) ptr; 47 a->getContext()->mHal.funcs.freeRuntimeMem(ptr); 48 } 49} 50 51Allocation * Allocation::createAllocation(Context *rsc, const Type *type, uint32_t usages, 52 RsAllocationMipmapControl mc, void * ptr) { 53 // Allocation objects must use allocator specified by the driver 54 void* allocMem = rsc->mHal.funcs.allocRuntimeMem(sizeof(Allocation), 0); 55 56 if (!allocMem) { 57 rsc->setError(RS_ERROR_FATAL_DRIVER, "Couldn't allocate memory for Allocation"); 58 return NULL; 59 } 60 61 Allocation *a = new (allocMem) Allocation(rsc, type, usages, mc, ptr); 62 63 if (!rsc->mHal.funcs.allocation.init(rsc, a, type->getElement()->getHasReferences())) { 64 rsc->setError(RS_ERROR_FATAL_DRIVER, "Allocation::Allocation, alloc failure"); 65 delete a; 66 return NULL; 67 } 68 69 return a; 70} 71 72void Allocation::updateCache() { 73 const Type *type = mHal.state.type; 74 mHal.state.yuv = type->getDimYuv(); 75 mHal.state.hasFaces = type->getDimFaces(); 76 mHal.state.hasMipmaps = type->getDimLOD(); 77 mHal.state.elementSizeBytes = type->getElementSizeBytes(); 78 mHal.state.hasReferences = mHal.state.type->getElement()->getHasReferences(); 79} 80 81Allocation::~Allocation() { 82#if !defined(RS_SERVER) && !defined(RS_COMPATIBILITY_LIB) 83 if (mGrallocConsumer.get()) { 84 mGrallocConsumer->unlockBuffer(); 85 mGrallocConsumer = NULL; 86 } 87#endif 88 89 freeChildrenUnlocked(); 90 mRSC->mHal.funcs.allocation.destroy(mRSC, this); 91} 92 93void Allocation::syncAll(Context *rsc, RsAllocationUsageType src) { 94 rsc->mHal.funcs.allocation.syncAll(rsc, this, src); 95} 96 97void * Allocation::getPointer(const Context *rsc, uint32_t lod, RsAllocationCubemapFace face, 98 uint32_t z, uint32_t array, size_t *stride) { 99 100 if ((lod >= mHal.drvState.lodCount) || 101 (z && (z >= mHal.drvState.lod[lod].dimZ)) || 102 ((face != RS_ALLOCATION_CUBEMAP_FACE_POSITIVE_X) && !mHal.state.hasFaces) || 103 (array != 0)) { 104 return NULL; 105 } 106 107 size_t s = 0; 108 //void *ptr = mRSC->mHal.funcs.allocation.lock1D(rsc, this); 109 if ((stride != NULL) && mHal.drvState.lod[0].dimY) { 110 *stride = mHal.drvState.lod[lod].stride; 111 } 112 return mHal.drvState.lod[lod].mallocPtr; 113} 114 115void Allocation::data(Context *rsc, uint32_t xoff, uint32_t lod, 116 uint32_t count, const void *data, size_t sizeBytes) { 117 const size_t eSize = mHal.state.type->getElementSizeBytes(); 118 119 if ((count * eSize) != sizeBytes) { 120 char buf[1024]; 121 sprintf(buf, "Allocation::subData called with mismatched size expected %zu, got %zu", 122 (count * eSize), sizeBytes); 123 rsc->setError(RS_ERROR_BAD_VALUE, buf); 124 mHal.state.type->dumpLOGV("type info"); 125 return; 126 } 127 128 rsc->mHal.funcs.allocation.data1D(rsc, this, xoff, lod, count, data, sizeBytes); 129 sendDirty(rsc); 130} 131 132void Allocation::data(Context *rsc, uint32_t xoff, uint32_t yoff, uint32_t lod, RsAllocationCubemapFace face, 133 uint32_t w, uint32_t h, const void *data, size_t sizeBytes, size_t stride) { 134 rsc->mHal.funcs.allocation.data2D(rsc, this, xoff, yoff, lod, face, w, h, data, sizeBytes, stride); 135 sendDirty(rsc); 136} 137 138void Allocation::data(Context *rsc, uint32_t xoff, uint32_t yoff, uint32_t zoff, 139 uint32_t lod, 140 uint32_t w, uint32_t h, uint32_t d, const void *data, size_t sizeBytes, size_t stride) { 141 rsc->mHal.funcs.allocation.data3D(rsc, this, xoff, yoff, zoff, lod, w, h, d, data, sizeBytes, stride); 142 sendDirty(rsc); 143} 144 145void Allocation::read(Context *rsc, uint32_t xoff, uint32_t lod, 146 uint32_t count, void *data, size_t sizeBytes) { 147 const size_t eSize = mHal.state.type->getElementSizeBytes(); 148 149 if ((count * eSize) != sizeBytes) { 150 char buf[1024]; 151 sprintf(buf, "Allocation::read called with mismatched size expected %zu, got %zu", 152 (count * eSize), sizeBytes); 153 rsc->setError(RS_ERROR_BAD_VALUE, buf); 154 mHal.state.type->dumpLOGV("type info"); 155 return; 156 } 157 158 rsc->mHal.funcs.allocation.read1D(rsc, this, xoff, lod, count, data, sizeBytes); 159} 160 161void Allocation::read(Context *rsc, uint32_t xoff, uint32_t yoff, uint32_t lod, RsAllocationCubemapFace face, 162 uint32_t w, uint32_t h, void *data, size_t sizeBytes, size_t stride) { 163 const size_t eSize = mHal.state.elementSizeBytes; 164 const size_t lineSize = eSize * w; 165 if (!stride) { 166 stride = lineSize; 167 } else { 168 if ((lineSize * h) != sizeBytes) { 169 char buf[1024]; 170 sprintf(buf, "Allocation size mismatch, expected %zu, got %zu", (lineSize * h), sizeBytes); 171 rsc->setError(RS_ERROR_BAD_VALUE, buf); 172 return; 173 } 174 } 175 176 rsc->mHal.funcs.allocation.read2D(rsc, this, xoff, yoff, lod, face, w, h, data, sizeBytes, stride); 177} 178 179void Allocation::read(Context *rsc, uint32_t xoff, uint32_t yoff, uint32_t zoff, uint32_t lod, 180 uint32_t w, uint32_t h, uint32_t d, void *data, size_t sizeBytes, size_t stride) { 181 const size_t eSize = mHal.state.elementSizeBytes; 182 const size_t lineSize = eSize * w; 183 if (!stride) { 184 stride = lineSize; 185 } 186 187 rsc->mHal.funcs.allocation.read3D(rsc, this, xoff, yoff, zoff, lod, w, h, d, data, sizeBytes, stride); 188 189} 190 191void Allocation::elementData(Context *rsc, uint32_t x, const void *data, 192 uint32_t cIdx, size_t sizeBytes) { 193 size_t eSize = mHal.state.elementSizeBytes; 194 195 if (cIdx >= mHal.state.type->getElement()->getFieldCount()) { 196 rsc->setError(RS_ERROR_BAD_VALUE, "subElementData component out of range."); 197 return; 198 } 199 200 if (x >= mHal.drvState.lod[0].dimX) { 201 rsc->setError(RS_ERROR_BAD_VALUE, "subElementData X offset out of range."); 202 return; 203 } 204 205 const Element * e = mHal.state.type->getElement()->getField(cIdx); 206 uint32_t elemArraySize = mHal.state.type->getElement()->getFieldArraySize(cIdx); 207 if (sizeBytes != e->getSizeBytes() * elemArraySize) { 208 rsc->setError(RS_ERROR_BAD_VALUE, "subElementData bad size."); 209 return; 210 } 211 212 rsc->mHal.funcs.allocation.elementData1D(rsc, this, x, data, cIdx, sizeBytes); 213 sendDirty(rsc); 214} 215 216void Allocation::elementData(Context *rsc, uint32_t x, uint32_t y, 217 const void *data, uint32_t cIdx, size_t sizeBytes) { 218 size_t eSize = mHal.state.elementSizeBytes; 219 220 if (x >= mHal.drvState.lod[0].dimX) { 221 rsc->setError(RS_ERROR_BAD_VALUE, "subElementData X offset out of range."); 222 return; 223 } 224 225 if (y >= mHal.drvState.lod[0].dimY) { 226 rsc->setError(RS_ERROR_BAD_VALUE, "subElementData X offset out of range."); 227 return; 228 } 229 230 if (cIdx >= mHal.state.type->getElement()->getFieldCount()) { 231 rsc->setError(RS_ERROR_BAD_VALUE, "subElementData component out of range."); 232 return; 233 } 234 235 const Element * e = mHal.state.type->getElement()->getField(cIdx); 236 uint32_t elemArraySize = mHal.state.type->getElement()->getFieldArraySize(cIdx); 237 if (sizeBytes != e->getSizeBytes() * elemArraySize) { 238 rsc->setError(RS_ERROR_BAD_VALUE, "subElementData bad size."); 239 return; 240 } 241 242 rsc->mHal.funcs.allocation.elementData2D(rsc, this, x, y, data, cIdx, sizeBytes); 243 sendDirty(rsc); 244} 245 246void Allocation::addProgramToDirty(const Program *p) { 247 mToDirtyList.push(p); 248} 249 250void Allocation::removeProgramToDirty(const Program *p) { 251 for (size_t ct=0; ct < mToDirtyList.size(); ct++) { 252 if (mToDirtyList[ct] == p) { 253 mToDirtyList.removeAt(ct); 254 return; 255 } 256 } 257 rsAssert(0); 258} 259 260void Allocation::dumpLOGV(const char *prefix) const { 261 ObjectBase::dumpLOGV(prefix); 262 char buf[1024]; 263 264 if ((strlen(prefix) + 10) < sizeof(buf)) { 265 sprintf(buf, "%s type ", prefix); 266 if (mHal.state.type) { 267 mHal.state.type->dumpLOGV(buf); 268 } 269 } 270 ALOGV("%s allocation ptr=%p mUsageFlags=0x04%x, mMipmapControl=0x%04x", 271 prefix, mHal.drvState.lod[0].mallocPtr, mHal.state.usageFlags, mHal.state.mipmapControl); 272} 273 274uint32_t Allocation::getPackedSize() const { 275 uint32_t numItems = mHal.state.type->getCellCount(); 276 return numItems * mHal.state.type->getElement()->getSizeBytesUnpadded(); 277} 278 279void Allocation::writePackedData(Context *rsc, const Type *type, 280 uint8_t *dst, const uint8_t *src, bool dstPadded) { 281 const Element *elem = type->getElement(); 282 uint32_t unpaddedBytes = elem->getSizeBytesUnpadded(); 283 uint32_t paddedBytes = elem->getSizeBytes(); 284 uint32_t numItems = type->getPackedSizeBytes() / paddedBytes; 285 286 uint32_t srcInc = !dstPadded ? paddedBytes : unpaddedBytes; 287 uint32_t dstInc = dstPadded ? paddedBytes : unpaddedBytes; 288 289 // no sub-elements 290 uint32_t fieldCount = elem->getFieldCount(); 291 if (fieldCount == 0) { 292 for (uint32_t i = 0; i < numItems; i ++) { 293 memcpy(dst, src, unpaddedBytes); 294 src += srcInc; 295 dst += dstInc; 296 } 297 return; 298 } 299 300 // Cache offsets 301 uint32_t *offsetsPadded = new uint32_t[fieldCount]; 302 uint32_t *offsetsUnpadded = new uint32_t[fieldCount]; 303 uint32_t *sizeUnpadded = new uint32_t[fieldCount]; 304 305 for (uint32_t i = 0; i < fieldCount; i++) { 306 offsetsPadded[i] = elem->getFieldOffsetBytes(i); 307 offsetsUnpadded[i] = elem->getFieldOffsetBytesUnpadded(i); 308 sizeUnpadded[i] = elem->getField(i)->getSizeBytesUnpadded(); 309 } 310 311 uint32_t *srcOffsets = !dstPadded ? offsetsPadded : offsetsUnpadded; 312 uint32_t *dstOffsets = dstPadded ? offsetsPadded : offsetsUnpadded; 313 314 // complex elements, need to copy subelem after subelem 315 for (uint32_t i = 0; i < numItems; i ++) { 316 for (uint32_t fI = 0; fI < fieldCount; fI++) { 317 memcpy(dst + dstOffsets[fI], src + srcOffsets[fI], sizeUnpadded[fI]); 318 } 319 src += srcInc; 320 dst += dstInc; 321 } 322 323 delete[] offsetsPadded; 324 delete[] offsetsUnpadded; 325 delete[] sizeUnpadded; 326} 327 328void Allocation::unpackVec3Allocation(Context *rsc, const void *data, size_t dataSize) { 329 const uint8_t *src = (const uint8_t*)data; 330 uint8_t *dst = (uint8_t *)rsc->mHal.funcs.allocation.lock1D(rsc, this); 331 332 writePackedData(rsc, getType(), dst, src, true); 333 rsc->mHal.funcs.allocation.unlock1D(rsc, this); 334} 335 336void Allocation::packVec3Allocation(Context *rsc, OStream *stream) const { 337 uint32_t paddedBytes = getType()->getElement()->getSizeBytes(); 338 uint32_t unpaddedBytes = getType()->getElement()->getSizeBytesUnpadded(); 339 uint32_t numItems = mHal.state.type->getCellCount(); 340 341 const uint8_t *src = (const uint8_t*)rsc->mHal.funcs.allocation.lock1D(rsc, this); 342 uint8_t *dst = new uint8_t[numItems * unpaddedBytes]; 343 344 writePackedData(rsc, getType(), dst, src, false); 345 stream->addByteArray(dst, getPackedSize()); 346 347 delete[] dst; 348 rsc->mHal.funcs.allocation.unlock1D(rsc, this); 349} 350 351void Allocation::serialize(Context *rsc, OStream *stream) const { 352 // Need to identify ourselves 353 stream->addU32((uint32_t)getClassId()); 354 stream->addString(getName()); 355 356 // First thing we need to serialize is the type object since it will be needed 357 // to initialize the class 358 mHal.state.type->serialize(rsc, stream); 359 360 uint32_t dataSize = mHal.state.type->getPackedSizeBytes(); 361 // 3 element vectors are padded to 4 in memory, but padding isn't serialized 362 uint32_t packedSize = getPackedSize(); 363 // Write how much data we are storing 364 stream->addU32(packedSize); 365 if (dataSize == packedSize) { 366 // Now write the data 367 stream->addByteArray(rsc->mHal.funcs.allocation.lock1D(rsc, this), dataSize); 368 rsc->mHal.funcs.allocation.unlock1D(rsc, this); 369 } else { 370 // Now write the data 371 packVec3Allocation(rsc, stream); 372 } 373} 374 375Allocation *Allocation::createFromStream(Context *rsc, IStream *stream) { 376 // First make sure we are reading the correct object 377 RsA3DClassID classID = (RsA3DClassID)stream->loadU32(); 378 if (classID != RS_A3D_CLASS_ID_ALLOCATION) { 379 rsc->setError(RS_ERROR_FATAL_DRIVER, 380 "allocation loading failed due to corrupt file. (invalid id)\n"); 381 return NULL; 382 } 383 384 const char *name = stream->loadString(); 385 386 Type *type = Type::createFromStream(rsc, stream); 387 if (!type) { 388 return NULL; 389 } 390 type->compute(); 391 392 Allocation *alloc = Allocation::createAllocation(rsc, type, RS_ALLOCATION_USAGE_SCRIPT); 393 type->decUserRef(); 394 395 // Number of bytes we wrote out for this allocation 396 uint32_t dataSize = stream->loadU32(); 397 // 3 element vectors are padded to 4 in memory, but padding isn't serialized 398 uint32_t packedSize = alloc->getPackedSize(); 399 if (dataSize != type->getPackedSizeBytes() && 400 dataSize != packedSize) { 401 rsc->setError(RS_ERROR_FATAL_DRIVER, 402 "allocation loading failed due to corrupt file. (invalid size)\n"); 403 ObjectBase::checkDelete(alloc); 404 ObjectBase::checkDelete(type); 405 return NULL; 406 } 407 408 alloc->assignName(name); 409 if (dataSize == type->getPackedSizeBytes()) { 410 uint32_t count = dataSize / type->getElementSizeBytes(); 411 // Read in all of our allocation data 412 alloc->data(rsc, 0, 0, count, stream->getPtr() + stream->getPos(), dataSize); 413 } else { 414 alloc->unpackVec3Allocation(rsc, stream->getPtr() + stream->getPos(), dataSize); 415 } 416 stream->reset(stream->getPos() + dataSize); 417 418 return alloc; 419} 420 421void Allocation::sendDirty(const Context *rsc) const { 422#ifndef RS_COMPATIBILITY_LIB 423 for (size_t ct=0; ct < mToDirtyList.size(); ct++) { 424 mToDirtyList[ct]->forceDirty(); 425 } 426#endif 427 mRSC->mHal.funcs.allocation.markDirty(rsc, this); 428} 429 430void Allocation::incRefs(const void *ptr, size_t ct, size_t startOff) const { 431 mHal.state.type->incRefs(ptr, ct, startOff); 432} 433 434void Allocation::decRefs(const void *ptr, size_t ct, size_t startOff) const { 435 if (!mHal.state.hasReferences || !getIsScript()) { 436 return; 437 } 438 mHal.state.type->decRefs(ptr, ct, startOff); 439} 440 441void Allocation::freeChildrenUnlocked () { 442 void *ptr = mRSC->mHal.funcs.allocation.lock1D(mRSC, this); 443 decRefs(ptr, mHal.state.type->getCellCount(), 0); 444 mRSC->mHal.funcs.allocation.unlock1D(mRSC, this); 445} 446 447bool Allocation::freeChildren() { 448 if (mHal.state.hasReferences) { 449 incSysRef(); 450 freeChildrenUnlocked(); 451 return decSysRef(); 452 } 453 return false; 454} 455 456void Allocation::copyRange1D(Context *rsc, const Allocation *src, int32_t srcOff, int32_t destOff, int32_t len) { 457} 458 459void Allocation::resize1D(Context *rsc, uint32_t dimX) { 460 uint32_t oldDimX = mHal.drvState.lod[0].dimX; 461 if (dimX == oldDimX) { 462 return; 463 } 464 465 ObjectBaseRef<Type> t = mHal.state.type->cloneAndResize1D(rsc, dimX); 466 if (dimX < oldDimX) { 467 decRefs(rsc->mHal.funcs.allocation.lock1D(rsc, this), oldDimX - dimX, dimX); 468 rsc->mHal.funcs.allocation.unlock1D(rsc, this); 469 } 470 rsc->mHal.funcs.allocation.resize(rsc, this, t.get(), mHal.state.hasReferences); 471 setType(t.get()); 472 updateCache(); 473} 474 475void Allocation::resize2D(Context *rsc, uint32_t dimX, uint32_t dimY) { 476 rsc->setError(RS_ERROR_FATAL_DRIVER, "resize2d not implemented"); 477} 478 479#ifndef RS_COMPATIBILITY_LIB 480void Allocation::NewBufferListener::onFrameAvailable() { 481 intptr_t ip = (intptr_t)alloc; 482 rsc->sendMessageToClient(NULL, RS_MESSAGE_TO_CLIENT_NEW_BUFFER, ip, 0, true); 483} 484#endif 485 486void * Allocation::getSurface(const Context *rsc) { 487#ifndef RS_COMPATIBILITY_LIB 488 // Configure GrallocConsumer to be in asynchronous mode 489 sp<BufferQueue> bq = new BufferQueue(); 490 mGrallocConsumer = new GrallocConsumer(this, bq); 491 sp<IGraphicBufferProducer> bp = bq; 492 bp->incStrong(NULL); 493 494 mBufferListener = new NewBufferListener(); 495 mBufferListener->rsc = rsc; 496 mBufferListener->alloc = this; 497 498 mGrallocConsumer->setFrameAvailableListener(mBufferListener); 499 return bp.get(); 500#else 501 return NULL; 502#endif 503 //return rsc->mHal.funcs.allocation.getSurface(rsc, this); 504} 505 506void Allocation::setSurface(const Context *rsc, RsNativeWindow sur) { 507 ANativeWindow *nw = (ANativeWindow *)sur; 508 rsc->mHal.funcs.allocation.setSurface(rsc, this, nw); 509} 510 511void Allocation::ioSend(const Context *rsc) { 512 rsc->mHal.funcs.allocation.ioSend(rsc, this); 513} 514 515void Allocation::ioReceive(const Context *rsc) { 516 void *ptr = NULL; 517 size_t stride = 0; 518#ifndef RS_COMPATIBILITY_LIB 519 if (mHal.state.usageFlags & RS_ALLOCATION_USAGE_SCRIPT) { 520 status_t ret = mGrallocConsumer->lockNextBuffer(); 521 522 if (ret == OK) { 523 rsc->mHal.funcs.allocation.ioReceive(rsc, this); 524 } else if (ret == BAD_VALUE) { 525 // No new frame, don't do anything 526 } else { 527 rsc->setError(RS_ERROR_DRIVER, "Error receiving IO input buffer."); 528 } 529 530 } 531#endif 532} 533 534 535///////////////// 536// 537 538namespace android { 539namespace renderscript { 540 541void rsi_AllocationSyncAll(Context *rsc, RsAllocation va, RsAllocationUsageType src) { 542 Allocation *a = static_cast<Allocation *>(va); 543 a->sendDirty(rsc); 544 a->syncAll(rsc, src); 545} 546 547void rsi_AllocationGenerateMipmaps(Context *rsc, RsAllocation va) { 548 Allocation *alloc = static_cast<Allocation *>(va); 549 rsc->mHal.funcs.allocation.generateMipmaps(rsc, alloc); 550} 551 552void rsi_AllocationCopyToBitmap(Context *rsc, RsAllocation va, void *data, size_t sizeBytes) { 553 Allocation *a = static_cast<Allocation *>(va); 554 const Type * t = a->getType(); 555 a->read(rsc, 0, 0, 0, RS_ALLOCATION_CUBEMAP_FACE_POSITIVE_X, 556 t->getDimX(), t->getDimY(), data, sizeBytes, 0); 557} 558 559void rsi_Allocation1DData(Context *rsc, RsAllocation va, uint32_t xoff, uint32_t lod, 560 uint32_t count, const void *data, size_t sizeBytes) { 561 Allocation *a = static_cast<Allocation *>(va); 562 a->data(rsc, xoff, lod, count, data, sizeBytes); 563} 564 565void rsi_Allocation2DElementData(Context *rsc, RsAllocation va, uint32_t x, uint32_t y, uint32_t lod, RsAllocationCubemapFace face, 566 const void *data, size_t sizeBytes, size_t eoff) { 567 Allocation *a = static_cast<Allocation *>(va); 568 a->elementData(rsc, x, y, data, eoff, sizeBytes); 569} 570 571void rsi_Allocation1DElementData(Context *rsc, RsAllocation va, uint32_t x, uint32_t lod, 572 const void *data, size_t sizeBytes, size_t eoff) { 573 Allocation *a = static_cast<Allocation *>(va); 574 a->elementData(rsc, x, data, eoff, sizeBytes); 575} 576 577void rsi_Allocation2DData(Context *rsc, RsAllocation va, uint32_t xoff, uint32_t yoff, uint32_t lod, RsAllocationCubemapFace face, 578 uint32_t w, uint32_t h, const void *data, size_t sizeBytes, size_t stride) { 579 Allocation *a = static_cast<Allocation *>(va); 580 a->data(rsc, xoff, yoff, lod, face, w, h, data, sizeBytes, stride); 581} 582 583void rsi_Allocation3DData(Context *rsc, RsAllocation va, uint32_t xoff, uint32_t yoff, uint32_t zoff, uint32_t lod, 584 uint32_t w, uint32_t h, uint32_t d, const void *data, size_t sizeBytes, size_t stride) { 585 Allocation *a = static_cast<Allocation *>(va); 586 a->data(rsc, xoff, yoff, zoff, lod, w, h, d, data, sizeBytes, stride); 587} 588 589 590void rsi_AllocationRead(Context *rsc, RsAllocation va, void *data, size_t sizeBytes) { 591 Allocation *a = static_cast<Allocation *>(va); 592 const Type * t = a->getType(); 593 if(t->getDimY()) { 594 a->read(rsc, 0, 0, 0, RS_ALLOCATION_CUBEMAP_FACE_POSITIVE_X, 595 t->getDimX(), t->getDimY(), data, sizeBytes, 0); 596 } else { 597 a->read(rsc, 0, 0, t->getDimX(), data, sizeBytes); 598 } 599 600} 601 602void rsi_AllocationResize1D(Context *rsc, RsAllocation va, uint32_t dimX) { 603 Allocation *a = static_cast<Allocation *>(va); 604 a->resize1D(rsc, dimX); 605} 606 607void rsi_AllocationResize2D(Context *rsc, RsAllocation va, uint32_t dimX, uint32_t dimY) { 608 Allocation *a = static_cast<Allocation *>(va); 609 a->resize2D(rsc, dimX, dimY); 610} 611 612RsAllocation rsi_AllocationCreateTyped(Context *rsc, RsType vtype, 613 RsAllocationMipmapControl mipmaps, 614 uint32_t usages, uintptr_t ptr) { 615 Allocation * alloc = Allocation::createAllocation(rsc, static_cast<Type *>(vtype), usages, mipmaps, (void*)ptr); 616 if (!alloc) { 617 return NULL; 618 } 619 alloc->incUserRef(); 620 return alloc; 621} 622 623RsAllocation rsi_AllocationCreateFromBitmap(Context *rsc, RsType vtype, 624 RsAllocationMipmapControl mipmaps, 625 const void *data, size_t sizeBytes, uint32_t usages) { 626 Type *t = static_cast<Type *>(vtype); 627 628 RsAllocation vTexAlloc = rsi_AllocationCreateTyped(rsc, vtype, mipmaps, usages, 0); 629 Allocation *texAlloc = static_cast<Allocation *>(vTexAlloc); 630 if (texAlloc == NULL) { 631 ALOGE("Memory allocation failure"); 632 return NULL; 633 } 634 635 texAlloc->data(rsc, 0, 0, 0, RS_ALLOCATION_CUBEMAP_FACE_POSITIVE_X, 636 t->getDimX(), t->getDimY(), data, sizeBytes, 0); 637 if (mipmaps == RS_ALLOCATION_MIPMAP_FULL) { 638 rsc->mHal.funcs.allocation.generateMipmaps(rsc, texAlloc); 639 } 640 641 texAlloc->sendDirty(rsc); 642 return texAlloc; 643} 644 645RsAllocation rsi_AllocationCubeCreateFromBitmap(Context *rsc, RsType vtype, 646 RsAllocationMipmapControl mipmaps, 647 const void *data, size_t sizeBytes, uint32_t usages) { 648 Type *t = static_cast<Type *>(vtype); 649 650 // Cubemap allocation's faces should be Width by Width each. 651 // Source data should have 6 * Width by Width pixels 652 // Error checking is done in the java layer 653 RsAllocation vTexAlloc = rsi_AllocationCreateTyped(rsc, vtype, mipmaps, usages, 0); 654 Allocation *texAlloc = static_cast<Allocation *>(vTexAlloc); 655 if (texAlloc == NULL) { 656 ALOGE("Memory allocation failure"); 657 return NULL; 658 } 659 660 uint32_t faceSize = t->getDimX(); 661 uint32_t strideBytes = faceSize * 6 * t->getElementSizeBytes(); 662 uint32_t copySize = faceSize * t->getElementSizeBytes(); 663 664 uint8_t *sourcePtr = (uint8_t*)data; 665 for (uint32_t face = 0; face < 6; face ++) { 666 for (uint32_t dI = 0; dI < faceSize; dI ++) { 667 texAlloc->data(rsc, 0, dI, 0, (RsAllocationCubemapFace)face, 668 t->getDimX(), 1, sourcePtr + strideBytes * dI, copySize, 0); 669 } 670 671 // Move the data pointer to the next cube face 672 sourcePtr += copySize; 673 } 674 675 if (mipmaps == RS_ALLOCATION_MIPMAP_FULL) { 676 rsc->mHal.funcs.allocation.generateMipmaps(rsc, texAlloc); 677 } 678 679 texAlloc->sendDirty(rsc); 680 return texAlloc; 681} 682 683void rsi_AllocationCopy2DRange(Context *rsc, 684 RsAllocation dstAlloc, 685 uint32_t dstXoff, uint32_t dstYoff, 686 uint32_t dstMip, uint32_t dstFace, 687 uint32_t width, uint32_t height, 688 RsAllocation srcAlloc, 689 uint32_t srcXoff, uint32_t srcYoff, 690 uint32_t srcMip, uint32_t srcFace) { 691 Allocation *dst = static_cast<Allocation *>(dstAlloc); 692 Allocation *src= static_cast<Allocation *>(srcAlloc); 693 rsc->mHal.funcs.allocation.allocData2D(rsc, dst, dstXoff, dstYoff, dstMip, 694 (RsAllocationCubemapFace)dstFace, 695 width, height, 696 src, srcXoff, srcYoff,srcMip, 697 (RsAllocationCubemapFace)srcFace); 698} 699 700void rsi_AllocationCopy3DRange(Context *rsc, 701 RsAllocation dstAlloc, 702 uint32_t dstXoff, uint32_t dstYoff, uint32_t dstZoff, 703 uint32_t dstMip, 704 uint32_t width, uint32_t height, uint32_t depth, 705 RsAllocation srcAlloc, 706 uint32_t srcXoff, uint32_t srcYoff, uint32_t srcZoff, 707 uint32_t srcMip) { 708 Allocation *dst = static_cast<Allocation *>(dstAlloc); 709 Allocation *src= static_cast<Allocation *>(srcAlloc); 710 rsc->mHal.funcs.allocation.allocData3D(rsc, dst, dstXoff, dstYoff, dstZoff, dstMip, 711 width, height, depth, 712 src, srcXoff, srcYoff, srcZoff, srcMip); 713} 714 715 716void * rsi_AllocationGetSurface(Context *rsc, RsAllocation valloc) { 717 Allocation *alloc = static_cast<Allocation *>(valloc); 718 void *s = alloc->getSurface(rsc); 719 return s; 720} 721 722void rsi_AllocationSetSurface(Context *rsc, RsAllocation valloc, RsNativeWindow sur) { 723 Allocation *alloc = static_cast<Allocation *>(valloc); 724 alloc->setSurface(rsc, sur); 725} 726 727void rsi_AllocationIoSend(Context *rsc, RsAllocation valloc) { 728 Allocation *alloc = static_cast<Allocation *>(valloc); 729 alloc->ioSend(rsc); 730} 731 732void rsi_AllocationIoReceive(Context *rsc, RsAllocation valloc) { 733 Allocation *alloc = static_cast<Allocation *>(valloc); 734 alloc->ioReceive(rsc); 735} 736 737void rsi_AllocationGetPointer(Context *rsc, RsAllocation valloc, 738 uint32_t lod, RsAllocationCubemapFace face, 739 uint32_t z, uint32_t array, size_t *stride, size_t strideLen) { 740 Allocation *alloc = static_cast<Allocation *>(valloc); 741 rsAssert(strideLen == sizeof(size_t)); 742 743 alloc->getPointer(rsc, lod, face, z, array, stride); 744} 745 746void rsi_Allocation1DRead(Context *rsc, RsAllocation va, uint32_t xoff, uint32_t lod, 747 uint32_t count, void *data, size_t sizeBytes) { 748 Allocation *a = static_cast<Allocation *>(va); 749 rsc->mHal.funcs.allocation.read1D(rsc, a, xoff, lod, count, data, sizeBytes); 750} 751 752void rsi_Allocation2DRead(Context *rsc, RsAllocation va, uint32_t xoff, uint32_t yoff, 753 uint32_t lod, RsAllocationCubemapFace face, uint32_t w, 754 uint32_t h, void *data, size_t sizeBytes, size_t stride) { 755 Allocation *a = static_cast<Allocation *>(va); 756 a->read(rsc, xoff, yoff, lod, face, w, h, data, sizeBytes, stride); 757} 758 759} 760} 761 762extern "C" const void * rsaAllocationGetType(RsContext con, RsAllocation va) { 763 Allocation *a = static_cast<Allocation *>(va); 764 a->getType()->incUserRef(); 765 766 return a->getType(); 767} 768