rsProgramFragment.cpp revision 326e0ddf89e8df2837752fbfd7a014814b32082c
1326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams/* 2326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * Copyright (C) 2009 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" 18326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams#include "rsProgramFragment.h" 19326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 20326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsusing namespace android; 21326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsusing namespace android::renderscript; 22326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 23326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 24326e0ddf89e8df2837752fbfd7a014814b32082cJason SamsProgramFragment::ProgramFragment(Element *in, Element *out) : 25326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams Program(in, out) 26326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 27326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams for (uint32_t ct=0; ct < MAX_TEXTURE; ct++) { 28326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mEnvModes[ct] = RS_TEX_ENV_MODE_REPLACE; 29326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mTextureDimensions[ct] = 2; 30326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 31326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mTextureEnableMask = 0; 32326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 33326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 34326e0ddf89e8df2837752fbfd7a014814b32082cJason SamsProgramFragment::~ProgramFragment() 35326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 36326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 37326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 38326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid ProgramFragment::setupGL() 39326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 40326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams for (uint32_t ct=0; ct < MAX_TEXTURE; ct++) { 41326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams glActiveTexture(GL_TEXTURE0 + ct); 42326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams if (!(mTextureEnableMask & (1 << ct)) || 43326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams !mSamplers[ct].get() || 44326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams !mTextures[ct].get()) { 45326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 46326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams glDisable(GL_TEXTURE_2D); 47326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams continue; 48326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 49326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 50326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams glEnable(GL_TEXTURE_2D); 51326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams glBindTexture(GL_TEXTURE_2D, mTextures[ct]->getTextureID()); 52326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 53326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams switch(mEnvModes[ct]) { 54326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams case RS_TEX_ENV_MODE_REPLACE: 55326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams glTexEnvf(GL_TEXTURE_2D, GL_TEXTURE_ENV_MODE, GL_REPLACE); 56326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams break; 57326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams case RS_TEX_ENV_MODE_MODULATE: 58326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams glTexEnvf(GL_TEXTURE_2D, GL_TEXTURE_ENV_MODE, GL_MODULATE); 59326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams break; 60326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams case RS_TEX_ENV_MODE_DECAL: 61326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams glTexEnvf(GL_TEXTURE_2D, GL_TEXTURE_ENV_MODE, GL_DECAL); 62326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams break; 63326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 64326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 65326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mSamplers[ct]->setupGL(); 66326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 67326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams glActiveTexture(GL_TEXTURE0); 68326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 69326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 70326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 71326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid ProgramFragment::bindTexture(uint32_t slot, Allocation *a) 72326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 73326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams if (slot >= MAX_TEXTURE) { 74326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams LOGE("Attempt to bind a texture to a slot > MAX_TEXTURE"); 75326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams return; 76326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 77326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 78326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mTextures[slot].set(a); 79326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 80326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 81326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid ProgramFragment::bindSampler(uint32_t slot, Sampler *s) 82326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 83326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams if (slot >= MAX_TEXTURE) { 84326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams LOGE("Attempt to bind a Sampler to a slot > MAX_TEXTURE"); 85326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams return; 86326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 87326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 88326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mSamplers[slot].set(s); 89326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 90326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 91326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid ProgramFragment::setType(uint32_t slot, const Element *e, uint32_t dim) 92326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 93326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams if (slot >= MAX_TEXTURE) { 94326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams LOGE("Attempt to setType to a slot > MAX_TEXTURE"); 95326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams return; 96326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 97326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 98326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams if (dim >= 4) { 99326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams LOGE("Attempt to setType to a dimension > 3"); 100326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams return; 101326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 102326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 103326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mTextureFormats[slot].set(e); 104326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mTextureDimensions[slot] = dim; 105326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 106326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 107326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid ProgramFragment::setEnvMode(uint32_t slot, RsTexEnvMode env) 108326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 109326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams if (slot >= MAX_TEXTURE) { 110326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams LOGE("Attempt to setEnvMode to a slot > MAX_TEXTURE"); 111326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams return; 112326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 113326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 114326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mEnvModes[slot] = env; 115326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 116326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 117326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid ProgramFragment::setTexEnable(uint32_t slot, bool enable) 118326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 119326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams if (slot >= MAX_TEXTURE) { 120326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams LOGE("Attempt to setEnvMode to a slot > MAX_TEXTURE"); 121326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams return; 122326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 123326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 124326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams uint32_t bit = 1 << slot; 125326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mTextureEnableMask &= ~bit; 126326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams if (enable) { 127326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mTextureEnableMask |= bit; 128326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 129326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 130326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 131326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 132326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 133326e0ddf89e8df2837752fbfd7a014814b32082cJason SamsProgramFragmentState::ProgramFragmentState() 134326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 135326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams mPF = NULL; 136326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 137326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 138326e0ddf89e8df2837752fbfd7a014814b32082cJason SamsProgramFragmentState::~ProgramFragmentState() 139326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 140326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams delete mPF; 141326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 142326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 143326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 144326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 145326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 146326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsnamespace android { 147326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsnamespace renderscript { 148326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 149326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid rsi_ProgramFragmentBegin(Context * rsc, RsElement in, RsElement out) 150326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 151326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams delete rsc->mStateFragment.mPF; 152326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams rsc->mStateFragment.mPF = new ProgramFragment((Element *)in, (Element *)out); 153326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 154326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 155326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid rsi_ProgramFragmentBindTexture(Context *rsc, RsProgramFragment vpf, uint32_t slot, RsAllocation a) 156326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 157326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams ProgramFragment *pf = static_cast<ProgramFragment *>(vpf); 158326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams pf->bindTexture(slot, static_cast<Allocation *>(a)); 159326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 160326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams //LOGE("%p %p", pf, rsc->getFragment()); 161326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams if (pf == rsc->getFragment()) { 162326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams pf->setupGL(); 163326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 164326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 165326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 166326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid rsi_ProgramFragmentBindSampler(Context *rsc, RsProgramFragment vpf, uint32_t slot, RsSampler s) 167326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 168326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams ProgramFragment *pf = static_cast<ProgramFragment *>(vpf); 169326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams pf->bindSampler(slot, static_cast<Sampler *>(s)); 170326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 171326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams if (pf == rsc->getFragment()) { 172326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams pf->setupGL(); 173326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 174326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 175326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 176326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid rsi_ProgramFragmentSetType(Context *rsc, uint32_t slot, RsType vt) 177326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 178326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams const Type *t = static_cast<const Type *>(vt); 179326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams uint32_t dim = 1; 180326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams if (t->getDimY()) { 181326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams dim ++; 182326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams if (t->getDimZ()) { 183326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams dim ++; 184326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 185326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams } 186326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 187326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams rsc->mStateFragment.mPF->setType(slot, t->getElement(), dim); 188326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 189326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 190326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid rsi_ProgramFragmentSetEnvMode(Context *rsc, uint32_t slot, RsTexEnvMode env) 191326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 192326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams rsc->mStateFragment.mPF->setEnvMode(slot, env); 193326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 194326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 195326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid rsi_ProgramFragmentSetTexEnable(Context *rsc, uint32_t slot, bool enable) 196326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 197326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams rsc->mStateFragment.mPF->setTexEnable(slot, enable); 198326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 199326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 200326e0ddf89e8df2837752fbfd7a014814b32082cJason SamsRsProgramFragment rsi_ProgramFragmentCreate(Context *rsc) 201326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 202326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams ProgramFragment *pf = rsc->mStateFragment.mPF; 203326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams pf->incRef(); 204326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams rsc->mStateFragment.mPF = 0; 205326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams return pf; 206326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 207326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 208326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 209326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 210326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 211326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 212326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 213