libs/rs/rsType.cpp - platform/frameworks/base - Git at Google

 /*
  * Copyright (C) 2009 The Android Open Source Project
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */

 #include "rsContext.h"

 using namespace android;
 using namespace android::renderscript;

 Type::Type(Context *rsc) : ObjectBase(rsc) {
     mLODs = 0;
     mLODCount = 0;
     clear();
 }

 void Type::preDestroy() {
     for (uint32_t ct = 0; ct < mRSC->mStateType.mTypes.size(); ct++) {
         if (mRSC->mStateType.mTypes[ct] == this) {
             mRSC->mStateType.mTypes.removeAt(ct);
             break;
         }
     }
 }

 Type::~Type() {
     if (mLODs) {
         delete [] mLODs;
         mLODs = NULL;
     }
 }

 void Type::clear() {
     if (mLODs) {
         delete [] mLODs;
         mLODs = NULL;
     }
     mDimX = 0;
     mDimY = 0;
     mDimZ = 0;
     mDimLOD = 0;
     mFaces = false;
     mElement.clear();
 }

 TypeState::TypeState() {
 }

 TypeState::~TypeState() {
 }

 size_t Type::getOffsetForFace(uint32_t face) const {
     rsAssert(mFaces);
     return 0;
 }

 void Type::compute() {
     uint32_t oldLODCount = mLODCount;
     if (mDimLOD) {
         uint32_t l2x = rsFindHighBit(mDimX) + 1;
         uint32_t l2y = rsFindHighBit(mDimY) + 1;
         uint32_t l2z = rsFindHighBit(mDimZ) + 1;

         mLODCount = rsMax(l2x, l2y);
         mLODCount = rsMax(mLODCount, l2z);
     } else {
         mLODCount = 1;
     }
     if (mLODCount != oldLODCount) {
         if (mLODs){
             delete [] mLODs;
         }
         mLODs = new LOD[mLODCount];
     }

     uint32_t tx = mDimX;
     uint32_t ty = mDimY;
     uint32_t tz = mDimZ;
     size_t offset = 0;
     for (uint32_t lod=0; lod < mLODCount; lod++) {
         mLODs[lod].mX = tx;
         mLODs[lod].mY = ty;
         mLODs[lod].mZ = tz;
         mLODs[lod].mOffset = offset;
         offset += tx * rsMax(ty, 1u) * rsMax(tz, 1u) * mElement->getSizeBytes();
         if (tx > 1) tx >>= 1;
         if (ty > 1) ty >>= 1;
         if (tz > 1) tz >>= 1;
     }

     // At this point the offset is the size of a mipmap chain;
     mMipChainSizeBytes = offset;

     if (mFaces) {
         offset *= 6;
     }
     mTotalSizeBytes = offset;
 }

 uint32_t Type::getLODOffset(uint32_t lod, uint32_t x) const {
     uint32_t offset = mLODs[lod].mOffset;
     offset += x * mElement->getSizeBytes();
     return offset;
 }

 uint32_t Type::getLODOffset(uint32_t lod, uint32_t x, uint32_t y) const {
     uint32_t offset = mLODs[lod].mOffset;
     offset += (x + y * mLODs[lod].mX) * mElement->getSizeBytes();
     return offset;
 }

 uint32_t Type::getLODOffset(uint32_t lod, uint32_t x, uint32_t y, uint32_t z) const {
     uint32_t offset = mLODs[lod].mOffset;
     offset += (x + y*mLODs[lod].mX + z*mLODs[lod].mX*mLODs[lod].mY) * mElement->getSizeBytes();
     return offset;
 }

 uint32_t Type::getLODFaceOffset(uint32_t lod, RsAllocationCubemapFace face, uint32_t x, uint32_t y) const {
     uint32_t offset = mLODs[lod].mOffset;
     offset += (x + y * mLODs[lod].mX) * mElement->getSizeBytes();

     if (face != 0) {
         uint32_t faceOffset = getSizeBytes() / 6;
         offset += faceOffset * face;
     }
     return offset;
 }

 void Type::dumpLOGV(const char *prefix) const {
     char buf[1024];
     ObjectBase::dumpLOGV(prefix);
     LOGV("%s   Type: x=%zu y=%zu z=%zu mip=%i face=%i", prefix, mDimX, mDimY, mDimZ, mDimLOD, mFaces);
     snprintf(buf, sizeof(buf), "%s element: ", prefix);
     mElement->dumpLOGV(buf);
 }

 void Type::serialize(OStream *stream) const {
     // Need to identify ourselves
     stream->addU32((uint32_t)getClassId());

     String8 name(getName());
     stream->addString(&name);

     mElement->serialize(stream);

     stream->addU32(mDimX);
     stream->addU32(mDimY);
     stream->addU32(mDimZ);

     stream->addU8((uint8_t)(mDimLOD ? 1 : 0));
     stream->addU8((uint8_t)(mFaces ? 1 : 0));
 }

 Type *Type::createFromStream(Context *rsc, IStream *stream) {
     // First make sure we are reading the correct object
     RsA3DClassID classID = (RsA3DClassID)stream->loadU32();
     if (classID != RS_A3D_CLASS_ID_TYPE) {
         LOGE("type loading skipped due to invalid class id\n");
         return NULL;
     }

     String8 name;
     stream->loadString(&name);

     Element *elem = Element::createFromStream(rsc, stream);
     if (!elem) {
         return NULL;
     }

     uint32_t x = stream->loadU32();
     uint32_t y = stream->loadU32();
     uint32_t z = stream->loadU32();
     uint8_t lod = stream->loadU8();
     uint8_t faces = stream->loadU8();
     return Type::getType(rsc, elem, x, y, z, lod != 0, faces !=0 );
 }

 bool Type::getIsNp2() const {
     uint32_t x = getDimX();
     uint32_t y = getDimY();
     uint32_t z = getDimZ();

     if (x && (x & (x-1))) {
         return true;
     }
     if (y && (y & (y-1))) {
         return true;
     }
     if (z && (z & (z-1))) {
         return true;
     }
     return false;
 }

 bool Type::isEqual(const Type *other) const {
     if (other == NULL) {
         return false;
     }
     if (other->getElement()->isEqual(getElement()) &&
         other->getDimX() == mDimX &&
         other->getDimY() == mDimY &&
         other->getDimZ() == mDimZ &&
         other->getDimLOD() == mDimLOD &&
         other->getDimFaces() == mFaces) {
         return true;
     }
     return false;
 }

 Type * Type::getType(Context *rsc, const Element *e,
                      uint32_t dimX, uint32_t dimY, uint32_t dimZ,
                      bool dimLOD, bool dimFaces) {
     TypeState * stc = &rsc->mStateType;

     ObjectBase::asyncLock();
     for (uint32_t ct=0; ct < stc->mTypes.size(); ct++) {
         Type *t = stc->mTypes[ct];
         if (t->getElement() != e) continue;
         if (t->getDimX() != dimX) continue;
         if (t->getDimY() != dimY) continue;
         if (t->getDimZ() != dimZ) continue;
         if (t->getDimLOD() != dimLOD) continue;
         if (t->getDimFaces() != dimFaces) continue;
         t->incUserRef();
         ObjectBase::asyncUnlock();
         return t;
     }
     ObjectBase::asyncUnlock();


     Type *nt = new Type(rsc);
     nt->mElement.set(e);
     nt->mDimX = dimX;
     nt->mDimY = dimY;
     nt->mDimZ = dimZ;
     nt->mDimLOD = dimLOD;
     nt->mFaces = dimFaces;
     nt->compute();
     nt->incUserRef();

     ObjectBase::asyncLock();
     stc->mTypes.push(nt);
     ObjectBase::asyncUnlock();

     return nt;
 }

 Type * Type::cloneAndResize1D(Context *rsc, uint32_t dimX) const {
     return getType(rsc, mElement.get(), dimX,
                    mDimY, mDimZ, mDimLOD, mFaces);
 }

 Type * Type::cloneAndResize2D(Context *rsc,
                               uint32_t dimX,
                               uint32_t dimY) const {
     return getType(rsc, mElement.get(), dimX, dimY,
                    mDimZ, mDimLOD, mFaces);
 }


 //////////////////////////////////////////////////
 //
 namespace android {
 namespace renderscript {

 RsType rsi_TypeCreate(Context *rsc, RsElement _e, uint32_t dimX,
                      uint32_t dimY, uint32_t dimZ, bool mips, bool faces) {
     Element *e = static_cast<Element *>(_e);

     return Type::getType(rsc, e, dimX, dimY, dimZ, mips, faces);
 }

 }
 }

 void rsaTypeGetNativeData(RsContext con, RsType type, uint32_t *typeData, uint32_t typeDataSize) {
     rsAssert(typeDataSize == 6);
     // Pack the data in the follofing way mDimX; mDimY; mDimZ;
     // mDimLOD; mDimFaces; mElement; into typeData
     Type *t = static_cast<Type *>(type);

     (*typeData++) = t->getDimX();
     (*typeData++) = t->getDimY();
     (*typeData++) = t->getDimZ();
     (*typeData++) = t->getDimLOD();
     (*typeData++) = t->getDimFaces() ? 1 : 0;
     (*typeData++) = (uint32_t)t->getElement();
     t->getElement()->incUserRef();
 }
	/*
	* Copyright (C) 2009 The Android Open Source Project
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	#include "rsContext.h"

	using namespace android;
	using namespace android::renderscript;

	Type::Type(Context *rsc) : ObjectBase(rsc) {
	mLODs = 0;
	mLODCount = 0;
	clear();
	}

	void Type::preDestroy() {
	for (uint32_t ct = 0; ct < mRSC->mStateType.mTypes.size(); ct++) {
	if (mRSC->mStateType.mTypes[ct] == this) {
	mRSC->mStateType.mTypes.removeAt(ct);
	break;
	}
	}
	}

	Type::~Type() {
	if (mLODs) {
	delete [] mLODs;
	mLODs = NULL;
	}
	}

	void Type::clear() {
	if (mLODs) {
	delete [] mLODs;
	mLODs = NULL;
	}
	mDimX = 0;
	mDimY = 0;
	mDimZ = 0;
	mDimLOD = 0;
	mFaces = false;
	mElement.clear();
	}

	TypeState::TypeState() {
	}

	TypeState::~TypeState() {
	}

	size_t Type::getOffsetForFace(uint32_t face) const {
	rsAssert(mFaces);
	return 0;
	}

	void Type::compute() {
	uint32_t oldLODCount = mLODCount;
	if (mDimLOD) {
	uint32_t l2x = rsFindHighBit(mDimX) + 1;
	uint32_t l2y = rsFindHighBit(mDimY) + 1;
	uint32_t l2z = rsFindHighBit(mDimZ) + 1;

	mLODCount = rsMax(l2x, l2y);
	mLODCount = rsMax(mLODCount, l2z);
	} else {
	mLODCount = 1;
	}
	if (mLODCount != oldLODCount) {
	if (mLODs){
	delete [] mLODs;
	}
	mLODs = new LOD[mLODCount];
	}

	uint32_t tx = mDimX;
	uint32_t ty = mDimY;
	uint32_t tz = mDimZ;
	size_t offset = 0;
	for (uint32_t lod=0; lod < mLODCount; lod++) {
	mLODs[lod].mX = tx;
	mLODs[lod].mY = ty;
	mLODs[lod].mZ = tz;
	mLODs[lod].mOffset = offset;
	offset += tx * rsMax(ty, 1u) * rsMax(tz, 1u) * mElement->getSizeBytes();
	if (tx > 1) tx >>= 1;
	if (ty > 1) ty >>= 1;
	if (tz > 1) tz >>= 1;
	}

	// At this point the offset is the size of a mipmap chain;
	mMipChainSizeBytes = offset;

	if (mFaces) {
	offset *= 6;
	}
	mTotalSizeBytes = offset;
	}

	uint32_t Type::getLODOffset(uint32_t lod, uint32_t x) const {
	uint32_t offset = mLODs[lod].mOffset;
	offset += x * mElement->getSizeBytes();
	return offset;
	}

	uint32_t Type::getLODOffset(uint32_t lod, uint32_t x, uint32_t y) const {
	uint32_t offset = mLODs[lod].mOffset;
	offset += (x + y * mLODs[lod].mX) * mElement->getSizeBytes();
	return offset;
	}

	uint32_t Type::getLODOffset(uint32_t lod, uint32_t x, uint32_t y, uint32_t z) const {
	uint32_t offset = mLODs[lod].mOffset;
	offset += (x + ymLODs[lod].mX + zmLODs[lod].mXmLODs[lod].mY) mElement->getSizeBytes();
	return offset;
	}

	uint32_t Type::getLODFaceOffset(uint32_t lod, RsAllocationCubemapFace face, uint32_t x, uint32_t y) const {
	uint32_t offset = mLODs[lod].mOffset;
	offset += (x + y * mLODs[lod].mX) * mElement->getSizeBytes();

	if (face != 0) {
	uint32_t faceOffset = getSizeBytes() / 6;
	offset += faceOffset * face;
	}
	return offset;
	}

	void Type::dumpLOGV(const char *prefix) const {
	char buf[1024];
	ObjectBase::dumpLOGV(prefix);
	LOGV("%s Type: x=%zu y=%zu z=%zu mip=%i face=%i", prefix, mDimX, mDimY, mDimZ, mDimLOD, mFaces);
	snprintf(buf, sizeof(buf), "%s element: ", prefix);
	mElement->dumpLOGV(buf);
	}

	void Type::serialize(OStream *stream) const {
	// Need to identify ourselves
	stream->addU32((uint32_t)getClassId());

	String8 name(getName());
	stream->addString(&name);

	mElement->serialize(stream);

	stream->addU32(mDimX);
	stream->addU32(mDimY);
	stream->addU32(mDimZ);

	stream->addU8((uint8_t)(mDimLOD ? 1 : 0));
	stream->addU8((uint8_t)(mFaces ? 1 : 0));
	}

	Type Type::createFromStream(Context rsc, IStream *stream) {
	// First make sure we are reading the correct object
	RsA3DClassID classID = (RsA3DClassID)stream->loadU32();
	if (classID != RS_A3D_CLASS_ID_TYPE) {
	LOGE("type loading skipped due to invalid class id\n");
	return NULL;
	}

	String8 name;
	stream->loadString(&name);

	Element *elem = Element::createFromStream(rsc, stream);
	if (!elem) {
	return NULL;
	}

	uint32_t x = stream->loadU32();
	uint32_t y = stream->loadU32();
	uint32_t z = stream->loadU32();
	uint8_t lod = stream->loadU8();
	uint8_t faces = stream->loadU8();
	return Type::getType(rsc, elem, x, y, z, lod != 0, faces !=0 );
	}

	bool Type::getIsNp2() const {
	uint32_t x = getDimX();
	uint32_t y = getDimY();
	uint32_t z = getDimZ();

	if (x && (x & (x-1))) {
	return true;
	}
	if (y && (y & (y-1))) {
	return true;
	}
	if (z && (z & (z-1))) {
	return true;
	}
	return false;
	}

	bool Type::isEqual(const Type *other) const {
	if (other == NULL) {
	return false;
	}
	if (other->getElement()->isEqual(getElement()) &&
	other->getDimX() == mDimX &&
	other->getDimY() == mDimY &&
	other->getDimZ() == mDimZ &&
	other->getDimLOD() == mDimLOD &&
	other->getDimFaces() == mFaces) {
	return true;
	}
	return false;
	}

	Type * Type::getType(Context rsc, const Element e,
	uint32_t dimX, uint32_t dimY, uint32_t dimZ,
	bool dimLOD, bool dimFaces) {
	TypeState * stc = &rsc->mStateType;

	ObjectBase::asyncLock();
	for (uint32_t ct=0; ct < stc->mTypes.size(); ct++) {
	Type *t = stc->mTypes[ct];
	if (t->getElement() != e) continue;
	if (t->getDimX() != dimX) continue;
	if (t->getDimY() != dimY) continue;
	if (t->getDimZ() != dimZ) continue;
	if (t->getDimLOD() != dimLOD) continue;
	if (t->getDimFaces() != dimFaces) continue;
	t->incUserRef();
	ObjectBase::asyncUnlock();
	return t;
	}
	ObjectBase::asyncUnlock();


	Type *nt = new Type(rsc);
	nt->mElement.set(e);
	nt->mDimX = dimX;
	nt->mDimY = dimY;
	nt->mDimZ = dimZ;
	nt->mDimLOD = dimLOD;
	nt->mFaces = dimFaces;
	nt->compute();
	nt->incUserRef();

	ObjectBase::asyncLock();
	stc->mTypes.push(nt);
	ObjectBase::asyncUnlock();

	return nt;
	}

	Type * Type::cloneAndResize1D(Context *rsc, uint32_t dimX) const {
	return getType(rsc, mElement.get(), dimX,
	mDimY, mDimZ, mDimLOD, mFaces);
	}

	Type * Type::cloneAndResize2D(Context *rsc,
	uint32_t dimX,
	uint32_t dimY) const {
	return getType(rsc, mElement.get(), dimX, dimY,
	mDimZ, mDimLOD, mFaces);
	}


	//////////////////////////////////////////////////
	//
	namespace android {
	namespace renderscript {

	RsType rsi_TypeCreate(Context *rsc, RsElement _e, uint32_t dimX,
	uint32_t dimY, uint32_t dimZ, bool mips, bool faces) {
	Element e = static_cast<Element >(_e);

	return Type::getType(rsc, e, dimX, dimY, dimZ, mips, faces);
	}

	}
	}

	void rsaTypeGetNativeData(RsContext con, RsType type, uint32_t *typeData, uint32_t typeDataSize) {
	rsAssert(typeDataSize == 6);
	// Pack the data in the follofing way mDimX; mDimY; mDimZ;
	// mDimLOD; mDimFaces; mElement; into typeData
	Type t = static_cast<Type >(type);

	(*typeData++) = t->getDimX();
	(*typeData++) = t->getDimY();
	(*typeData++) = t->getDimZ();
	(*typeData++) = t->getDimLOD();
	(*typeData++) = t->getDimFaces() ? 1 : 0;
	(*typeData++) = (uint32_t)t->getElement();
	t->getElement()->incUserRef();
	}