rsProgram.cpp revision a04e30dbb5ab11592b03666bb3d102070759c58e
1326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams/* 2a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * Copyright (C) 2011 The Android Open Source Project 3326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * 4326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * Licensed under the Apache License, Version 2.0 (the "License"); 5326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * you may not use this file except in compliance with the License. 6326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * You may obtain a copy of the License at 7326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * 8326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * http://www.apache.org/licenses/LICENSE-2.0 9326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * 10326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * Unless required by applicable law or agreed to in writing, software 11326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * distributed under the License is distributed on an "AS IS" BASIS, 12326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * See the License for the specific language governing permissions and 14326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * limitations under the License. 15326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams */ 16326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 17326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams#include "rsContext.h" 18fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk#include "rsProgram.h" 19c460e55d78cbe8bee95c5c947dfe541218142a5bJason Sams 20326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsusing namespace android; 21326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsusing namespace android::renderscript; 22326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 23afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex SakhartchoukProgram::Program(Context *rsc) : ObjectBase(rsc) { 2454929cce0bf44090424b1f91b676529a2422378fAlex Sakhartchouk initMemberVars(); 254815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams} 264815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams 274815c0d121310cfcd6a8aba4eab77a9910af53acJason SamsProgram::Program(Context *rsc, const char * shaderText, uint32_t shaderLength, 28afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk const uint32_t * params, uint32_t paramLength) 29afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk : ObjectBase(rsc) { 304815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams 31afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk initMemberVars(); 324815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams for (uint32_t ct=0; ct < paramLength; ct+=2) { 334815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams if (params[ct] == RS_PROGRAM_PARAM_INPUT) { 34a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.inputElementsCount++; 354815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams } 364815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams if (params[ct] == RS_PROGRAM_PARAM_CONSTANT) { 37a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.constantsCount++; 384815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams } 3984e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk if (params[ct] == RS_PROGRAM_PARAM_TEXTURE_TYPE) { 40a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.texturesCount++; 41f2e4fa215c420cf0f8d226e9a59acb1f312e5e40Jason Sams } 424815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams } 434815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams 44a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.textures = new ObjectBaseRef<Allocation>[mHal.state.texturesCount]; 45a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.samplers = new ObjectBaseRef<Sampler>[mHal.state.texturesCount]; 46a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.textureTargets = new RsTextureTarget[mHal.state.texturesCount]; 47a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.inputElements = new ObjectBaseRef<Element>[mHal.state.inputElementsCount]; 48a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.constantTypes = new ObjectBaseRef<Type>[mHal.state.constantsCount]; 49a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.constants = new ObjectBaseRef<Allocation>[mHal.state.constantsCount]; 504815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams 514815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams uint32_t input = 0; 524815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams uint32_t output = 0; 534815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams uint32_t constant = 0; 5484e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk uint32_t texture = 0; 554815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams for (uint32_t ct=0; ct < paramLength; ct+=2) { 564815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams if (params[ct] == RS_PROGRAM_PARAM_INPUT) { 57a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.inputElements[input++].set(reinterpret_cast<Element *>(params[ct+1])); 584815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams } 594815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams if (params[ct] == RS_PROGRAM_PARAM_CONSTANT) { 60a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.constantTypes[constant++].set(reinterpret_cast<Type *>(params[ct+1])); 614815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams } 6284e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk if (params[ct] == RS_PROGRAM_PARAM_TEXTURE_TYPE) { 63a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.textureTargets[texture++] = (RsTextureTarget)params[ct+1]; 6484e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk } 654815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams } 66e7ae69f4a70f1813cf8086ebd9714192c635300aAlex Sakhartchouk mIsInternal = false; 67e7ae69f4a70f1813cf8086ebd9714192c635300aAlex Sakhartchouk uint32_t internalTokenLen = strlen(RS_SHADER_INTERNAL); 68afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk if (shaderLength > internalTokenLen && 69e7ae69f4a70f1813cf8086ebd9714192c635300aAlex Sakhartchouk strncmp(RS_SHADER_INTERNAL, shaderText, internalTokenLen) == 0) { 70e7ae69f4a70f1813cf8086ebd9714192c635300aAlex Sakhartchouk mIsInternal = true; 71e7ae69f4a70f1813cf8086ebd9714192c635300aAlex Sakhartchouk shaderText += internalTokenLen; 72e7ae69f4a70f1813cf8086ebd9714192c635300aAlex Sakhartchouk shaderLength -= internalTokenLen; 73e7ae69f4a70f1813cf8086ebd9714192c635300aAlex Sakhartchouk } 744815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams mUserShader.setTo(shaderText, shaderLength); 75326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 76326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 77afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex SakhartchoukProgram::~Program() { 78889fe50e7aaebed8cb8284b16a0e51e64e8a3a9cAlex Sakhartchouk 79a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mHal.state.constantsCount; ct++) { 80383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk bindAllocation(NULL, NULL, ct); 819ebb0c44ece47531f3d0a98ba52ae448de42032bJason Sams } 824815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams 83a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mHal.state.texturesCount; ct++) { 84383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk bindTexture(NULL, ct, NULL); 85383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk bindSampler(NULL, ct, NULL); 86383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk } 87a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk delete[] mHal.state.textures; 88a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk delete[] mHal.state.samplers; 89a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk delete[] mHal.state.textureTargets; 90a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk delete[] mHal.state.inputElements; 91a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk delete[] mHal.state.constantTypes; 92a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk delete[] mHal.state.constants; 93a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.inputElementsCount = 0; 94a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.constantsCount = 0; 95a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.texturesCount = 0; 96326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 97326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 9854929cce0bf44090424b1f91b676529a2422378fAlex Sakhartchoukvoid Program::initMemberVars() { 9954929cce0bf44090424b1f91b676529a2422378fAlex Sakhartchouk mDirty = true; 10054929cce0bf44090424b1f91b676529a2422378fAlex Sakhartchouk 101a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.drv = NULL; 102a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.textures = NULL; 103a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.samplers = NULL; 104a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.textureTargets = NULL; 105a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.inputElements = NULL; 106a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.constantTypes = NULL; 107a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.constants = NULL; 108a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 109a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.inputElementsCount = 0; 110a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.constantsCount = 0; 111a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.texturesCount = 0; 112a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 11354929cce0bf44090424b1f91b676529a2422378fAlex Sakhartchouk mIsInternal = false; 11454929cce0bf44090424b1f91b676529a2422378fAlex Sakhartchouk} 115326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 116afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Program::bindAllocation(Context *rsc, Allocation *alloc, uint32_t slot) { 117383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk if (alloc != NULL) { 118a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (slot >= mHal.state.constantsCount) { 119383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk LOGE("Attempt to bind alloc at slot %u, on shader id %u, but const count is %u", 120a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk slot, (uint32_t)this, mHal.state.constantsCount); 121383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "Cannot bind allocation"); 122383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk return; 123383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk } 124a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (!alloc->getType()->isEqual(mHal.state.constantTypes[slot].get())) { 125383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk LOGE("Attempt to bind alloc at slot %u, on shader id %u, but types mismatch", 126383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk slot, (uint32_t)this); 127383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "Cannot bind allocation"); 128383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk return; 129383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk } 130383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk } 131a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mHal.state.constants[slot].get() == alloc) { 1325c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams return; 1335c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams } 134a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mHal.state.constants[slot].get()) { 135a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.constants[slot].get()->removeProgramToDirty(this); 1365c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams } 137a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.constants[slot].set(alloc); 1385c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams if (alloc) { 1395c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams alloc->addProgramToDirty(this); 1405c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams } 141326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mDirty = true; 142326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 143326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 144afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Program::bindTexture(Context *rsc, uint32_t slot, Allocation *a) { 145a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (slot >= mHal.state.texturesCount) { 146a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("Attempt to bind texture to slot %u but tex count is %u", slot, mHal.state.texturesCount); 147383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "Cannot bind texture"); 1487dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams return; 1497dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams } 1507dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams 151a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (a && a->getType()->getDimFaces() && mHal.state.textureTargets[slot] != RS_TEXTURE_CUBE) { 15284e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk LOGE("Attempt to bind cubemap to slot %u but 2d texture needed", slot); 15384e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "Cannot bind cubemap to 2d texture slot"); 15484e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk return; 15584e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk } 15684e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk 1577dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams //LOGE("bindtex %i %p", slot, a); 158a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.textures[slot].set(a); 1597dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams mDirty = true; 1607dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams} 1617dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams 162afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Program::bindSampler(Context *rsc, uint32_t slot, Sampler *s) { 163a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (slot >= mHal.state.texturesCount) { 164a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("Attempt to bind sampler to slot %u but tex count is %u", slot, mHal.state.texturesCount); 165383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "Cannot bind sampler"); 1667dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams return; 1677dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams } 1687dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams 169a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mHal.state.samplers[slot].set(s); 1707dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams mDirty = true; 1717dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams} 1727dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams 173afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Program::setShader(const char *txt, uint32_t len) { 174f2a5d7326b38e5a28b6618f9b1e5a021aef7179fJason Sams mUserShader.setTo(txt, len); 175f2a5d7326b38e5a28b6618f9b1e5a021aef7179fJason Sams} 176f2a5d7326b38e5a28b6618f9b1e5a021aef7179fJason Sams 1774815c0d121310cfcd6a8aba4eab77a9910af53acJason Samsnamespace android { 1784815c0d121310cfcd6a8aba4eab77a9910af53acJason Samsnamespace renderscript { 1794815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams 180afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ProgramBindConstants(Context *rsc, RsProgram vp, uint32_t slot, RsAllocation constants) { 1814815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams Program *p = static_cast<Program *>(vp); 182383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk p->bindAllocation(rsc, static_cast<Allocation *>(constants), slot); 1834815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams} 1844815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams 185afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ProgramBindTexture(Context *rsc, RsProgram vpf, uint32_t slot, RsAllocation a) { 1867dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams Program *p = static_cast<Program *>(vpf); 187383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk p->bindTexture(rsc, slot, static_cast<Allocation *>(a)); 1887dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams} 1897dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams 190afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ProgramBindSampler(Context *rsc, RsProgram vpf, uint32_t slot, RsSampler s) { 1917dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams Program *p = static_cast<Program *>(vpf); 192383e5b1f68c321a77bfd7466fa1171a9bfab4a6fAlex Sakhartchouk p->bindSampler(rsc, slot, static_cast<Sampler *>(s)); 1937dad9c30a59c99b57269e1b498807b6f034d56e9Jason Sams} 1944815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams 1954815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams} 1964815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams} 1974815c0d121310cfcd6a8aba4eab77a9910af53acJason Sams 198