rsAllocation.cpp revision c9d43db4d216b01b13aebfdb31d5615909591b33
1/* 2 * Copyright (C) 2009 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 19#include <GLES/gl.h> 20#include <GLES/glext.h> 21 22using namespace android; 23using namespace android::renderscript; 24 25Allocation::Allocation(const Type *type) 26{ 27 mPtr = NULL; 28 29 mCpuWrite = false; 30 mCpuRead = false; 31 mGpuWrite = false; 32 mGpuRead = false; 33 34 mReadWriteRatio = 0; 35 mUpdateSize = 0; 36 37 mIsTexture = false; 38 mTextureID = 0; 39 40 mIsVertexBuffer = false; 41 mBufferID = 0; 42 43 mType.set(type); 44 mPtr = malloc(mType->getSizeBytes()); 45 if (!mPtr) { 46 LOGE("Allocation::Allocation, alloc failure"); 47 } 48 49} 50 51Allocation::~Allocation() 52{ 53} 54 55void Allocation::setCpuWritable(bool) 56{ 57} 58 59void Allocation::setGpuWritable(bool) 60{ 61} 62 63void Allocation::setCpuReadable(bool) 64{ 65} 66 67void Allocation::setGpuReadable(bool) 68{ 69} 70 71bool Allocation::fixAllocation() 72{ 73 return false; 74} 75 76void Allocation::uploadToTexture(uint32_t lodOffset) 77{ 78 //rsAssert(!mTextureId); 79 rsAssert(lodOffset < mType->getLODCount()); 80 81 GLenum type = mType->getElement()->getGLType(); 82 GLenum format = mType->getElement()->getGLFormat(); 83 84 if (!type || !format) { 85 return; 86 } 87 88 if (!mTextureID) { 89 glGenTextures(1, &mTextureID); 90 } 91 glBindTexture(GL_TEXTURE_2D, mTextureID); 92 93 Adapter2D adapt(this); 94 for(uint32_t lod = 0; (lod + lodOffset) < mType->getLODCount(); lod++) { 95 adapt.setLOD(lod+lodOffset); 96 97 uint16_t * ptr = static_cast<uint16_t *>(adapt.getElement(0,0)); 98 glTexImage2D(GL_TEXTURE_2D, lod, format, 99 adapt.getDimX(), adapt.getDimY(), 100 0, format, type, ptr); 101 } 102} 103 104void Allocation::uploadToBufferObject() 105{ 106 rsAssert(!mType->getDimY()); 107 rsAssert(!mType->getDimZ()); 108 109 if (!mBufferID) { 110 glGenBuffers(1, &mBufferID); 111 } 112 glBindBuffer(GL_ARRAY_BUFFER, mBufferID); 113 glBufferData(GL_ARRAY_BUFFER, mType->getSizeBytes(), getPtr(), GL_DYNAMIC_DRAW); 114 glBindBuffer(GL_ARRAY_BUFFER, 0); 115} 116 117void Allocation::data(const void *data) 118{ 119 memcpy(mPtr, data, mType->getSizeBytes()); 120} 121 122void Allocation::subData(uint32_t xoff, uint32_t count, const void *data) 123{ 124 uint32_t eSize = mType->getElementSizeBytes(); 125 uint8_t * ptr = static_cast<uint8_t *>(mPtr); 126 ptr += eSize * xoff; 127 memcpy(ptr, data, count * eSize); 128} 129 130void Allocation::subData(uint32_t xoff, uint32_t yoff, 131 uint32_t w, uint32_t h, const void *data) 132{ 133 uint32_t eSize = mType->getElementSizeBytes(); 134 uint32_t lineSize = eSize * w; 135 uint32_t destW = mType->getDimX(); 136 137 const uint8_t *src = static_cast<const uint8_t *>(data); 138 uint8_t *dst = static_cast<uint8_t *>(mPtr); 139 dst += eSize * (xoff + yoff * destW); 140 for (uint32_t line=yoff; line < (yoff+h); line++) { 141 uint8_t * ptr = static_cast<uint8_t *>(mPtr); 142 memcpy(dst, src, lineSize); 143 src += lineSize; 144 dst += destW * eSize; 145 } 146} 147 148void Allocation::subData(uint32_t xoff, uint32_t yoff, uint32_t zoff, 149 uint32_t w, uint32_t h, uint32_t d, const void *data) 150{ 151} 152 153 154 155///////////////// 156// 157 158 159namespace android { 160namespace renderscript { 161 162RsAllocation rsi_AllocationCreateTyped(Context *rsc, RsType vtype) 163{ 164 const Type * type = static_cast<const Type *>(vtype); 165 166 Allocation * alloc = new Allocation(type); 167 alloc->incRef(); 168 return alloc; 169} 170 171RsAllocation rsi_AllocationCreatePredefSized(Context *rsc, RsElementPredefined t, size_t count) 172{ 173 RsElement e = rsi_ElementGetPredefined(rsc, t); 174 return rsi_AllocationCreateSized(rsc, e, count); 175} 176 177RsAllocation rsi_AllocationCreateSized(Context *rsc, RsElement e, size_t count) 178{ 179 Type * type = new Type(); 180 type->setDimX(count); 181 type->setElement(static_cast<Element *>(e)); 182 type->compute(); 183 return rsi_AllocationCreateTyped(rsc, type); 184} 185 186void rsi_AllocationUploadToTexture(Context *rsc, RsAllocation va, uint32_t baseMipLevel) 187{ 188 Allocation *alloc = static_cast<Allocation *>(va); 189 alloc->uploadToTexture(baseMipLevel); 190} 191 192void rsi_AllocationUploadToBufferObject(Context *rsc, RsAllocation va) 193{ 194 Allocation *alloc = static_cast<Allocation *>(va); 195 alloc->uploadToBufferObject(); 196} 197 198void rsi_AllocationDestroy(Context *rsc, RsAllocation) 199{ 200} 201 202static void mip565(const Adapter2D &out, const Adapter2D &in) 203{ 204 uint32_t w = out.getDimX(); 205 uint32_t h = out.getDimY(); 206 207 for (uint32_t y=0; y < w; y++) { 208 uint16_t *oPtr = static_cast<uint16_t *>(out.getElement(0, y)); 209 const uint16_t *i1 = static_cast<uint16_t *>(in.getElement(0, y*2)); 210 const uint16_t *i2 = static_cast<uint16_t *>(in.getElement(0, y*2+1)); 211 212 for (uint32_t x=0; x < h; x++) { 213 *oPtr = rsBoxFilter565(i1[0], i1[1], i2[0], i2[1]); 214 oPtr ++; 215 i1 += 2; 216 i2 += 2; 217 } 218 } 219} 220 221static void mip8888(const Adapter2D &out, const Adapter2D &in) 222{ 223 uint32_t w = out.getDimX(); 224 uint32_t h = out.getDimY(); 225 226 for (uint32_t y=0; y < w; y++) { 227 uint32_t *oPtr = static_cast<uint32_t *>(out.getElement(0, y)); 228 const uint32_t *i1 = static_cast<uint32_t *>(in.getElement(0, y*2)); 229 const uint32_t *i2 = static_cast<uint32_t *>(in.getElement(0, y*2+1)); 230 231 for (uint32_t x=0; x < h; x++) { 232 *oPtr = rsBoxFilter8888(i1[0], i1[1], i2[0], i2[1]); 233 oPtr ++; 234 i1 += 2; 235 i2 += 2; 236 } 237 } 238 239} 240 241 242typedef void (*ElementConverter_t)(void *dst, const void *src, uint32_t count); 243 244static void elementConverter_cpy_16(void *dst, const void *src, uint32_t count) 245{ 246 memcpy(dst, src, count * 2); 247} 248static void elementConverter_cpy_8(void *dst, const void *src, uint32_t count) 249{ 250 memcpy(dst, src, count); 251} 252static void elementConverter_cpy_32(void *dst, const void *src, uint32_t count) 253{ 254 memcpy(dst, src, count * 4); 255} 256 257 258static void elementConverter_888_to_565(void *dst, const void *src, uint32_t count) 259{ 260 uint16_t *d = static_cast<uint16_t *>(dst); 261 const uint8_t *s = static_cast<const uint8_t *>(src); 262 263 while(count--) { 264 *d = rs888to565(s[0], s[1], s[2]); 265 d++; 266 s+= 3; 267 } 268} 269 270static void elementConverter_8888_to_565(void *dst, const void *src, uint32_t count) 271{ 272 uint16_t *d = static_cast<uint16_t *>(dst); 273 const uint8_t *s = static_cast<const uint8_t *>(src); 274 275 while(count--) { 276 *d = rs888to565(s[0], s[1], s[2]); 277 d++; 278 s+= 4; 279 } 280} 281 282static ElementConverter_t pickConverter(RsElementPredefined dstFmt, RsElementPredefined srcFmt) 283{ 284 if ((dstFmt == RS_ELEMENT_RGB_565) && 285 (srcFmt == RS_ELEMENT_RGB_565)) { 286 return elementConverter_cpy_16; 287 } 288 289 if ((dstFmt == RS_ELEMENT_RGB_565) && 290 (srcFmt == RS_ELEMENT_RGB_888)) { 291 return elementConverter_888_to_565; 292 } 293 294 if ((dstFmt == RS_ELEMENT_RGB_565) && 295 (srcFmt == RS_ELEMENT_RGBA_8888)) { 296 return elementConverter_8888_to_565; 297 } 298 299 if ((dstFmt == RS_ELEMENT_RGBA_8888) && 300 (srcFmt == RS_ELEMENT_RGBA_8888)) { 301 return elementConverter_cpy_32; 302 } 303 304 LOGE("pickConverter, unsuported combo, src %i, dst %i", srcFmt, dstFmt); 305 return 0; 306} 307 308 309RsAllocation rsi_AllocationCreateFromBitmap(Context *rsc, uint32_t w, uint32_t h, RsElementPredefined dstFmt, RsElementPredefined srcFmt, bool genMips, const void *data) 310{ 311 rsAssert(!(w & (w-1))); 312 rsAssert(!(h & (h-1))); 313 314 //LOGE("rsi_AllocationCreateFromBitmap %i %i %i %i %i", w, h, dstFmt, srcFmt, genMips); 315 rsi_TypeBegin(rsc, rsi_ElementGetPredefined(rsc, dstFmt)); 316 rsi_TypeAdd(rsc, RS_DIMENSION_X, w); 317 rsi_TypeAdd(rsc, RS_DIMENSION_Y, h); 318 if (genMips) { 319 rsi_TypeAdd(rsc, RS_DIMENSION_LOD, 1); 320 } 321 RsType type = rsi_TypeCreate(rsc); 322 323 RsAllocation vTexAlloc = rsi_AllocationCreateTyped(rsc, type); 324 Allocation *texAlloc = static_cast<Allocation *>(vTexAlloc); 325 if (texAlloc == NULL) { 326 LOGE("Memory allocation failure"); 327 return NULL; 328 } 329 texAlloc->incRef(); 330 331 ElementConverter_t cvt = pickConverter(dstFmt, srcFmt); 332 cvt(texAlloc->getPtr(), data, w * h); 333 334 if (genMips) { 335 Adapter2D adapt(texAlloc); 336 Adapter2D adapt2(texAlloc); 337 for(uint32_t lod=0; lod < (texAlloc->getType()->getLODCount() -1); lod++) { 338 adapt.setLOD(lod); 339 adapt2.setLOD(lod + 1); 340 mip565(adapt2, adapt); 341 } 342 } 343 344 return texAlloc; 345} 346 347static uint32_t fmtToBits(RsElementPredefined fmt) 348{ 349 return 16; 350} 351 352RsAllocation rsi_AllocationCreateFromBitmapBoxed(Context *rsc, uint32_t w, uint32_t h, RsElementPredefined dstFmt, RsElementPredefined srcFmt, bool genMips, const void *data) 353{ 354 uint32_t w2 = rsHigherPow2(w); 355 uint32_t h2 = rsHigherPow2(h); 356 357 if ((w2 == w) && (h2 == h)) { 358 return rsi_AllocationCreateFromBitmap(rsc, w, h, dstFmt, srcFmt, genMips, data); 359 } 360 361 uint32_t bpp = fmtToBits(srcFmt) >> 3; 362 size_t size = w2 * h2 * bpp; 363 uint8_t *tmp = static_cast<uint8_t *>(malloc(size)); 364 memset(tmp, 0, size); 365 366 const uint8_t * src = static_cast<const uint8_t *>(data); 367 for (uint32_t y = 0; y < h; y++) { 368 uint8_t * ydst = &tmp[y + ((h2 - h) >> 1)]; 369 memcpy(&ydst[(w2 - w) >> 1], src, w * bpp); 370 src += h * bpp; 371 } 372 373 RsAllocation ret = rsi_AllocationCreateFromBitmap(rsc, w2, h2, dstFmt, srcFmt, genMips, tmp); 374 free(tmp); 375 return ret; 376 377 378 379 380} 381 382 383RsAllocation rsi_AllocationCreateFromFile(Context *rsc, const char *file, bool genMips) 384{ 385 bool use32bpp = false; 386 387 typedef struct _Win3xBitmapHeader 388 { 389 uint16_t type; 390 uint32_t totalSize; 391 uint32_t reserved; 392 uint32_t offset; 393 int32_t hdrSize; /* Size of this header in bytes */ 394 int32_t width; /* Image width in pixels */ 395 int32_t height; /* Image height in pixels */ 396 int16_t planes; /* Number of color planes */ 397 int16_t bpp; /* Number of bits per pixel */ 398 /* Fields added for Windows 3.x follow this line */ 399 int32_t compression; /* Compression methods used */ 400 int32_t sizeOfBitmap; /* Size of bitmap in bytes */ 401 int32_t horzResolution; /* Horizontal resolution in pixels per meter */ 402 int32_t vertResolution; /* Vertical resolution in pixels per meter */ 403 int32_t colorsUsed; /* Number of colors in the image */ 404 int32_t colorsImportant; /* Minimum number of important colors */ 405 } __attribute__((__packed__)) WIN3XBITMAPHEADER; 406 407 _Win3xBitmapHeader hdr; 408 409 FILE *f = fopen(file, "rb"); 410 if (f == NULL) { 411 LOGE("rsAllocationCreateFromBitmap failed to open file %s", file); 412 return NULL; 413 } 414 memset(&hdr, 0, sizeof(hdr)); 415 fread(&hdr, sizeof(hdr), 1, f); 416 417 if (hdr.bpp != 24) { 418 LOGE("Unsuported BMP type"); 419 fclose(f); 420 return NULL; 421 } 422 423 int32_t texWidth = rsHigherPow2(hdr.width); 424 int32_t texHeight = rsHigherPow2(hdr.height); 425 426 if (use32bpp) { 427 rsi_TypeBegin(rsc, rsi_ElementGetPredefined(rsc, RS_ELEMENT_RGBA_8888)); 428 } else { 429 rsi_TypeBegin(rsc, rsi_ElementGetPredefined(rsc, RS_ELEMENT_RGB_565)); 430 } 431 rsi_TypeAdd(rsc, RS_DIMENSION_X, texWidth); 432 rsi_TypeAdd(rsc, RS_DIMENSION_Y, texHeight); 433 if (genMips) { 434 rsi_TypeAdd(rsc, RS_DIMENSION_LOD, 1); 435 } 436 RsType type = rsi_TypeCreate(rsc); 437 438 RsAllocation vTexAlloc = rsi_AllocationCreateTyped(rsc, type); 439 Allocation *texAlloc = static_cast<Allocation *>(vTexAlloc); 440 texAlloc->incRef(); 441 if (texAlloc == NULL) { 442 LOGE("Memory allocation failure"); 443 fclose(f); 444 return NULL; 445 } 446 447 // offset to letterbox if height is not pow2 448 Adapter2D adapt(texAlloc); 449 uint8_t * fileInBuf = new uint8_t[texWidth * 3]; 450 uint32_t yOffset = (hdr.width - hdr.height) / 2; 451 452 if (use32bpp) { 453 uint8_t *tmp = static_cast<uint8_t *>(adapt.getElement(0, yOffset)); 454 for (int y=0; y < hdr.height; y++) { 455 fseek(f, hdr.offset + (y*hdr.width*3), SEEK_SET); 456 fread(fileInBuf, 1, hdr.width * 3, f); 457 for(int x=0; x < hdr.width; x++) { 458 tmp[0] = fileInBuf[x*3 + 2]; 459 tmp[1] = fileInBuf[x*3 + 1]; 460 tmp[2] = fileInBuf[x*3]; 461 tmp[3] = 0xff; 462 tmp += 4; 463 } 464 } 465 } else { 466 uint16_t *tmp = static_cast<uint16_t *>(adapt.getElement(0, yOffset)); 467 for (int y=0; y < hdr.height; y++) { 468 fseek(f, hdr.offset + (y*hdr.width*3), SEEK_SET); 469 fread(fileInBuf, 1, hdr.width * 3, f); 470 for(int x=0; x < hdr.width; x++) { 471 *tmp = rs888to565(fileInBuf[x*3 + 2], fileInBuf[x*3 + 1], fileInBuf[x*3]); 472 tmp++; 473 } 474 } 475 } 476 477 fclose(f); 478 delete [] fileInBuf; 479 480 if (genMips) { 481 Adapter2D adapt2(texAlloc); 482 for(uint32_t lod=0; lod < (texAlloc->getType()->getLODCount() -1); lod++) { 483 adapt.setLOD(lod); 484 adapt2.setLOD(lod + 1); 485 if (use32bpp) { 486 mip8888(adapt2, adapt); 487 } else { 488 mip565(adapt2, adapt); 489 } 490 } 491 } 492 493 return texAlloc; 494} 495 496 497void rsi_AllocationData(Context *rsc, RsAllocation va, const void *data) 498{ 499 Allocation *a = static_cast<Allocation *>(va); 500 a->data(data); 501} 502 503void rsi_Allocation1DSubData(Context *rsc, RsAllocation va, uint32_t xoff, uint32_t count, const void *data) 504{ 505 Allocation *a = static_cast<Allocation *>(va); 506 a->subData(xoff, count, data); 507} 508 509void rsi_Allocation2DSubData(Context *rsc, RsAllocation va, uint32_t xoff, uint32_t yoff, uint32_t w, uint32_t h, const void *data) 510{ 511 Allocation *a = static_cast<Allocation *>(va); 512 a->subData(xoff, yoff, w, h, data); 513} 514 515 516} 517} 518