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 using namespace android; 20 using namespace android::renderscript; 21 22 Type::Type(Context *rsc) : ObjectBase(rsc) { 23 memset(&mHal, 0, sizeof(mHal)); 24 mDimLOD = false; 25 } 26 27 void Type::preDestroy() const { 28 for (uint32_t ct = 0; ct < mRSC->mStateType.mTypes.size(); ct++) { 29 if (mRSC->mStateType.mTypes[ct] == this) { 30 mRSC->mStateType.mTypes.removeAt(ct); 31 break; 32 } 33 } 34 } 35 36 Type::~Type() { 37 clear(); 38 } 39 40 void Type::clear() { 41 if (mHal.state.lodCount) { 42 delete [] mHal.state.lodDimX; 43 delete [] mHal.state.lodDimY; 44 delete [] mHal.state.lodDimZ; 45 delete [] mHal.state.lodOffset; 46 } 47 mElement.clear(); 48 memset(&mHal, 0, sizeof(mHal)); 49 } 50 51 TypeState::TypeState() { 52 } 53 54 TypeState::~TypeState() { 55 rsAssert(!mTypes.size()); 56 } 57 58 size_t Type::getOffsetForFace(uint32_t face) const { 59 rsAssert(mHal.state.faces); 60 return 0; 61 } 62 63 void Type::compute() { 64 uint32_t oldLODCount = mHal.state.lodCount; 65 if (mDimLOD) { 66 uint32_t l2x = rsFindHighBit(mHal.state.dimX) + 1; 67 uint32_t l2y = rsFindHighBit(mHal.state.dimY) + 1; 68 uint32_t l2z = rsFindHighBit(mHal.state.dimZ) + 1; 69 70 mHal.state.lodCount = rsMax(l2x, l2y); 71 mHal.state.lodCount = rsMax(mHal.state.lodCount, l2z); 72 } else { 73 mHal.state.lodCount = 1; 74 } 75 if (mHal.state.lodCount != oldLODCount) { 76 if (oldLODCount) { 77 delete [] mHal.state.lodDimX; 78 delete [] mHal.state.lodDimY; 79 delete [] mHal.state.lodDimZ; 80 delete [] mHal.state.lodOffset; 81 } 82 mHal.state.lodDimX = new uint32_t[mHal.state.lodCount]; 83 mHal.state.lodDimY = new uint32_t[mHal.state.lodCount]; 84 mHal.state.lodDimZ = new uint32_t[mHal.state.lodCount]; 85 mHal.state.lodOffset = new uint32_t[mHal.state.lodCount]; 86 } 87 88 uint32_t tx = mHal.state.dimX; 89 uint32_t ty = mHal.state.dimY; 90 uint32_t tz = mHal.state.dimZ; 91 size_t offset = 0; 92 for (uint32_t lod=0; lod < mHal.state.lodCount; lod++) { 93 mHal.state.lodDimX[lod] = tx; 94 mHal.state.lodDimY[lod] = ty; 95 mHal.state.lodDimZ[lod] = tz; 96 mHal.state.lodOffset[lod] = offset; 97 offset += tx * rsMax(ty, 1u) * rsMax(tz, 1u) * mElement->getSizeBytes(); 98 if (tx > 1) tx >>= 1; 99 if (ty > 1) ty >>= 1; 100 if (tz > 1) tz >>= 1; 101 } 102 103 // At this point the offset is the size of a mipmap chain; 104 mMipChainSizeBytes = offset; 105 106 if (mHal.state.faces) { 107 offset *= 6; 108 } 109 mTotalSizeBytes = offset; 110 mHal.state.element = mElement.get(); 111 } 112 113 uint32_t Type::getLODOffset(uint32_t lod, uint32_t x) const { 114 uint32_t offset = mHal.state.lodOffset[lod]; 115 offset += x * mElement->getSizeBytes(); 116 return offset; 117 } 118 119 uint32_t Type::getLODOffset(uint32_t lod, uint32_t x, uint32_t y) const { 120 uint32_t offset = mHal.state.lodOffset[lod]; 121 offset += (x + y * mHal.state.lodDimX[lod]) * mElement->getSizeBytes(); 122 return offset; 123 } 124 125 uint32_t Type::getLODOffset(uint32_t lod, uint32_t x, uint32_t y, uint32_t z) const { 126 uint32_t offset = mHal.state.lodOffset[lod]; 127 offset += (x + 128 y * mHal.state.lodDimX[lod] + 129 z * mHal.state.lodDimX[lod] * mHal.state.lodDimY[lod]) * mElement->getSizeBytes(); 130 return offset; 131 } 132 133 uint32_t Type::getLODFaceOffset(uint32_t lod, RsAllocationCubemapFace face, 134 uint32_t x, uint32_t y) const { 135 uint32_t offset = mHal.state.lodOffset[lod]; 136 offset += (x + y * mHal.state.lodDimX[lod]) * mElement->getSizeBytes(); 137 138 if (face != 0) { 139 uint32_t faceOffset = getSizeBytes() / 6; 140 offset += faceOffset * face; 141 } 142 return offset; 143 } 144 145 void Type::dumpLOGV(const char *prefix) const { 146 char buf[1024]; 147 ObjectBase::dumpLOGV(prefix); 148 ALOGV("%s Type: x=%u y=%u z=%u mip=%i face=%i", prefix, 149 mHal.state.dimX, 150 mHal.state.dimY, 151 mHal.state.dimZ, 152 mHal.state.lodCount, 153 mHal.state.faces); 154 snprintf(buf, sizeof(buf), "%s element: ", prefix); 155 mElement->dumpLOGV(buf); 156 } 157 158 void Type::serialize(OStream *stream) const { 159 // Need to identify ourselves 160 stream->addU32((uint32_t)getClassId()); 161 162 String8 name(getName()); 163 stream->addString(&name); 164 165 mElement->serialize(stream); 166 167 stream->addU32(mHal.state.dimX); 168 stream->addU32(mHal.state.dimY); 169 stream->addU32(mHal.state.dimZ); 170 171 stream->addU8((uint8_t)(mHal.state.lodCount ? 1 : 0)); 172 stream->addU8((uint8_t)(mHal.state.faces ? 1 : 0)); 173 } 174 175 Type *Type::createFromStream(Context *rsc, IStream *stream) { 176 // First make sure we are reading the correct object 177 RsA3DClassID classID = (RsA3DClassID)stream->loadU32(); 178 if (classID != RS_A3D_CLASS_ID_TYPE) { 179 ALOGE("type loading skipped due to invalid class id\n"); 180 return NULL; 181 } 182 183 String8 name; 184 stream->loadString(&name); 185 186 Element *elem = Element::createFromStream(rsc, stream); 187 if (!elem) { 188 return NULL; 189 } 190 191 uint32_t x = stream->loadU32(); 192 uint32_t y = stream->loadU32(); 193 uint32_t z = stream->loadU32(); 194 uint8_t lod = stream->loadU8(); 195 uint8_t faces = stream->loadU8(); 196 Type *type = Type::getType(rsc, elem, x, y, z, lod != 0, faces !=0 ); 197 elem->decUserRef(); 198 return type; 199 } 200 201 bool Type::getIsNp2() const { 202 uint32_t x = getDimX(); 203 uint32_t y = getDimY(); 204 uint32_t z = getDimZ(); 205 206 if (x && (x & (x-1))) { 207 return true; 208 } 209 if (y && (y & (y-1))) { 210 return true; 211 } 212 if (z && (z & (z-1))) { 213 return true; 214 } 215 return false; 216 } 217 218 ObjectBaseRef<Type> Type::getTypeRef(Context *rsc, const Element *e, 219 uint32_t dimX, uint32_t dimY, uint32_t dimZ, 220 bool dimLOD, bool dimFaces) { 221 ObjectBaseRef<Type> returnRef; 222 223 TypeState * stc = &rsc->mStateType; 224 225 ObjectBase::asyncLock(); 226 for (uint32_t ct=0; ct < stc->mTypes.size(); ct++) { 227 Type *t = stc->mTypes[ct]; 228 if (t->getElement() != e) continue; 229 if (t->getDimX() != dimX) continue; 230 if (t->getDimY() != dimY) continue; 231 if (t->getDimZ() != dimZ) continue; 232 if (t->getDimLOD() != dimLOD) continue; 233 if (t->getDimFaces() != dimFaces) continue; 234 returnRef.set(t); 235 ObjectBase::asyncUnlock(); 236 return returnRef; 237 } 238 ObjectBase::asyncUnlock(); 239 240 241 Type *nt = new Type(rsc); 242 nt->mDimLOD = dimLOD; 243 returnRef.set(nt); 244 nt->mElement.set(e); 245 nt->mHal.state.dimX = dimX; 246 nt->mHal.state.dimY = dimY; 247 nt->mHal.state.dimZ = dimZ; 248 nt->mHal.state.faces = dimFaces; 249 nt->compute(); 250 251 ObjectBase::asyncLock(); 252 stc->mTypes.push(nt); 253 ObjectBase::asyncUnlock(); 254 255 return returnRef; 256 } 257 258 ObjectBaseRef<Type> Type::cloneAndResize1D(Context *rsc, uint32_t dimX) const { 259 return getTypeRef(rsc, mElement.get(), dimX, 260 getDimY(), getDimZ(), getDimLOD(), getDimFaces()); 261 } 262 263 ObjectBaseRef<Type> Type::cloneAndResize2D(Context *rsc, 264 uint32_t dimX, 265 uint32_t dimY) const { 266 return getTypeRef(rsc, mElement.get(), dimX, dimY, 267 getDimZ(), getDimLOD(), getDimFaces()); 268 } 269 270 271 void Type::incRefs(const void *ptr, size_t ct, size_t startOff) const { 272 const uint8_t *p = static_cast<const uint8_t *>(ptr); 273 const Element *e = mHal.state.element; 274 uint32_t stride = e->getSizeBytes(); 275 276 p += stride * startOff; 277 while (ct > 0) { 278 e->incRefs(p); 279 ct--; 280 p += stride; 281 } 282 } 283 284 285 void Type::decRefs(const void *ptr, size_t ct, size_t startOff) const { 286 if (!mHal.state.element->getHasReferences()) { 287 return; 288 } 289 const uint8_t *p = static_cast<const uint8_t *>(ptr); 290 const Element *e = mHal.state.element; 291 uint32_t stride = e->getSizeBytes(); 292 293 p += stride * startOff; 294 while (ct > 0) { 295 e->decRefs(p); 296 ct--; 297 p += stride; 298 } 299 } 300 301 302 ////////////////////////////////////////////////// 303 // 304 namespace android { 305 namespace renderscript { 306 307 RsType rsi_TypeCreate(Context *rsc, RsElement _e, uint32_t dimX, 308 uint32_t dimY, uint32_t dimZ, bool mips, bool faces) { 309 Element *e = static_cast<Element *>(_e); 310 311 return Type::getType(rsc, e, dimX, dimY, dimZ, mips, faces); 312 } 313 314 } 315 } 316 317 void rsaTypeGetNativeData(RsContext con, RsType type, uint32_t *typeData, uint32_t typeDataSize) { 318 rsAssert(typeDataSize == 6); 319 // Pack the data in the follofing way mHal.state.dimX; mHal.state.dimY; mHal.state.dimZ; 320 // mHal.state.lodCount; mHal.state.faces; mElement; into typeData 321 Type *t = static_cast<Type *>(type); 322 323 (*typeData++) = t->getDimX(); 324 (*typeData++) = t->getDimY(); 325 (*typeData++) = t->getDimZ(); 326 (*typeData++) = t->getDimLOD() ? 1 : 0; 327 (*typeData++) = t->getDimFaces() ? 1 : 0; 328 (*typeData++) = (uint32_t)t->getElement(); 329 t->getElement()->incUserRef(); 330 } 331