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