rsdShader.cpp revision ba157304ce99d212403898b055998e6da7a20e9b
1a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk/* 2a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * Copyright (C) 2011 The Android Open Source Project 3a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * 4a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * Licensed under the Apache License, Version 2.0 (the "License"); 5a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * you may not use this file except in compliance with the License. 6a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * You may obtain a copy of the License at 7a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * 8a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * http://www.apache.org/licenses/LICENSE-2.0 9a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * 10a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * Unless required by applicable law or agreed to in writing, software 11a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * distributed under the License is distributed on an "AS IS" BASIS, 12a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * See the License for the specific language governing permissions and 14a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk * limitations under the License. 15a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk */ 16a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 17a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk#include <GLES2/gl2.h> 18a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk#include <GLES2/gl2ext.h> 19a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 20a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk#include <rs_hal.h> 21a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk#include <rsContext.h> 22a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk#include <rsProgram.h> 23a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 247f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk#include "rsdCore.h" 25eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams#include "rsdAllocation.h" 26a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk#include "rsdShader.h" 27a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk#include "rsdShaderCache.h" 28a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 29a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukusing namespace android; 30a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukusing namespace android::renderscript; 31a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 32a04e30dbb5ab11592b03666bb3d102070759c58eAlex SakhartchoukRsdShader::RsdShader(const Program *p, uint32_t type, 33a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const char * shaderText, uint32_t shaderLength) { 34a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 35a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUserShader.setTo(shaderText, shaderLength); 36a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mRSProgram = p; 37a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mType = type; 38a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk initMemberVars(); 39a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk initAttribAndUniformArray(); 40a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk init(); 41a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 42a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 43a04e30dbb5ab11592b03666bb3d102070759c58eAlex SakhartchoukRsdShader::~RsdShader() { 44a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mShaderID) { 45a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glDeleteShader(mShaderID); 46a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 47a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 48a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk delete[] mAttribNames; 49a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk delete[] mUniformNames; 50a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk delete[] mUniformArraySizes; 51ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk delete[] mTextureTargets; 52a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 53a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 54a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::initMemberVars() { 55a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mDirty = true; 56a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShaderID = 0; 57a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mAttribCount = 0; 58a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformCount = 0; 59a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 60a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mAttribNames = NULL; 61a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformNames = NULL; 62a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformArraySizes = NULL; 63ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk mTextureTargets = NULL; 64a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 65a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mIsValid = false; 66a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 67a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 68a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::init() { 69a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t attribCount = 0; 70a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t uniformCount = 0; 71a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.inputElementsCount; ct++) { 72a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk initAddUserElement(mRSProgram->mHal.state.inputElements[ct].get(), mAttribNames, NULL, &attribCount, RS_SHADER_ATTR); 73a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 74a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.constantsCount; ct++) { 75a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk initAddUserElement(mRSProgram->mHal.state.constantTypes[ct]->getElement(), mUniformNames, mUniformArraySizes, &uniformCount, RS_SHADER_UNI); 76a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 77a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 78a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mTextureUniformIndexStart = uniformCount; 79a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk char buf[256]; 80a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.texturesCount; ct++) { 81a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk snprintf(buf, sizeof(buf), "UNI_Tex%i", ct); 82a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformNames[uniformCount].setTo(buf); 83a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformArraySizes[uniformCount] = 1; 84a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uniformCount++; 85a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 86ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk 87a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 88a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 89a04e30dbb5ab11592b03666bb3d102070759c58eAlex SakhartchoukString8 RsdShader::getGLSLInputString() const { 90a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk String8 s; 91a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.inputElementsCount; ct++) { 92a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *e = mRSProgram->mHal.state.inputElements[ct].get(); 93a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t field=0; field < e->getFieldCount(); field++) { 94a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *f = e->getField(field); 95a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 96a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk // Cannot be complex 97a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(!f->getFieldCount()); 98a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk switch (f->getComponent().getVectorSize()) { 99a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 1: s.append("attribute float ATTRIB_"); break; 100a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 2: s.append("attribute vec2 ATTRIB_"); break; 101a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 3: s.append("attribute vec3 ATTRIB_"); break; 102a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 4: s.append("attribute vec4 ATTRIB_"); break; 103a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk default: 104a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(0); 105a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 106a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 107a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk s.append(e->getFieldName(field)); 108a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk s.append(";\n"); 109a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 110a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 111a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk return s; 112a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 113a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 114a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::appendAttributes() { 115a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.inputElementsCount; ct++) { 116a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *e = mRSProgram->mHal.state.inputElements[ct].get(); 117a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t field=0; field < e->getFieldCount(); field++) { 118a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *f = e->getField(field); 119a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const char *fn = e->getFieldName(field); 120a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 121a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (fn[0] == '#') { 122a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk continue; 123a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 124a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 125a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk // Cannot be complex 126a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(!f->getFieldCount()); 127a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk switch (f->getComponent().getVectorSize()) { 128a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 1: mShader.append("attribute float ATTRIB_"); break; 129a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 2: mShader.append("attribute vec2 ATTRIB_"); break; 130a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 3: mShader.append("attribute vec3 ATTRIB_"); break; 131a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 4: mShader.append("attribute vec4 ATTRIB_"); break; 132a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk default: 133a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(0); 134a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 135a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 136a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append(fn); 137a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append(";\n"); 138a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 139a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 140a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 141a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 142a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::appendTextures() { 143a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk char buf[256]; 144a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.texturesCount; ct++) { 145a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mRSProgram->mHal.state.textureTargets[ct] == RS_TEXTURE_2D) { 146a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk snprintf(buf, sizeof(buf), "uniform sampler2D UNI_Tex%i;\n", ct); 147ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk mTextureTargets[ct] = GL_TEXTURE_2D; 148a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else { 149a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk snprintf(buf, sizeof(buf), "uniform samplerCube UNI_Tex%i;\n", ct); 150ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk mTextureTargets[ct] = GL_TEXTURE_CUBE_MAP; 151a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 152a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append(buf); 153a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 154a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 155a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 156a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukbool RsdShader::createShader() { 157a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 158a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mType == GL_FRAGMENT_SHADER) { 159a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append("precision mediump float;\n"); 160a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 161a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk appendUserConstants(); 162a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk appendAttributes(); 163a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk appendTextures(); 164a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 165a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append(mUserShader); 166a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 167a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk return true; 168a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 169a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 170a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukbool RsdShader::loadShader(const Context *rsc) { 171a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShaderID = glCreateShader(mType); 172a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(mShaderID); 173a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 174a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (rsc->props.mLogShaders) { 175a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Loading shader type %x, ID %i", mType, mShaderID); 176a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("%s", mShader.string()); 177a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 178a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 179a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mShaderID) { 180a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const char * ss = mShader.string(); 181a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glShaderSource(mShaderID, 1, &ss, NULL); 182a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glCompileShader(mShaderID); 183a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 184a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk GLint compiled = 0; 185a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glGetShaderiv(mShaderID, GL_COMPILE_STATUS, &compiled); 186a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (!compiled) { 187a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk GLint infoLen = 0; 188a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glGetShaderiv(mShaderID, GL_INFO_LOG_LENGTH, &infoLen); 189a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (infoLen) { 190a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk char* buf = (char*) malloc(infoLen); 191a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (buf) { 192a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glGetShaderInfoLog(mShaderID, infoLen, NULL, buf); 193a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("Could not compile shader \n%s\n", buf); 194a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk free(buf); 195a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 196a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glDeleteShader(mShaderID); 197a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShaderID = 0; 198a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "Error returned from GL driver loading shader text,"); 199a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk return false; 200a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 201a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 202a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 203a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 204a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (rsc->props.mLogShaders) { 205a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("--Shader load result %x ", glGetError()); 206a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 207a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mIsValid = true; 208a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk return true; 209a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 210a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 211a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::appendUserConstants() { 212a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.constantsCount; ct++) { 213a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *e = mRSProgram->mHal.state.constantTypes[ct]->getElement(); 214a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t field=0; field < e->getFieldCount(); field++) { 215a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *f = e->getField(field); 216a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const char *fn = e->getFieldName(field); 217a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 218a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (fn[0] == '#') { 219a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk continue; 220a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 221a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 222a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk // Cannot be complex 223a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(!f->getFieldCount()); 224a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (f->getType() == RS_TYPE_MATRIX_4X4) { 225a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append("uniform mat4 UNI_"); 226a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (f->getType() == RS_TYPE_MATRIX_3X3) { 227a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append("uniform mat3 UNI_"); 228a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (f->getType() == RS_TYPE_MATRIX_2X2) { 229a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append("uniform mat2 UNI_"); 230a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else { 231a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk switch (f->getComponent().getVectorSize()) { 232a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 1: mShader.append("uniform float UNI_"); break; 233a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 2: mShader.append("uniform vec2 UNI_"); break; 234a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 3: mShader.append("uniform vec3 UNI_"); break; 235a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 4: mShader.append("uniform vec4 UNI_"); break; 236a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk default: 237a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(0); 238a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 239a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 240a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 241a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append(fn); 242a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (e->getFieldArraySize(field) > 1) { 243a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.appendFormat("[%d]", e->getFieldArraySize(field)); 244a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 245a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append(";\n"); 246a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 247a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 248a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 249a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 250a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::logUniform(const Element *field, const float *fd, uint32_t arraySize ) { 251a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk RsDataType dataType = field->getType(); 252a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t elementSize = field->getSizeBytes() / sizeof(float); 253a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t i = 0; i < arraySize; i ++) { 254a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (arraySize > 1) { 255a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Array Element [%u]", i); 256a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 257a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (dataType == RS_TYPE_MATRIX_4X4) { 258a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Matrix4x4"); 259a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("{%f, %f, %f, %f", fd[0], fd[4], fd[8], fd[12]); 260a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV(" %f, %f, %f, %f", fd[1], fd[5], fd[9], fd[13]); 261a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV(" %f, %f, %f, %f", fd[2], fd[6], fd[10], fd[14]); 262a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV(" %f, %f, %f, %f}", fd[3], fd[7], fd[11], fd[15]); 263a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (dataType == RS_TYPE_MATRIX_3X3) { 264a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Matrix3x3"); 265a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("{%f, %f, %f", fd[0], fd[3], fd[6]); 266a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV(" %f, %f, %f", fd[1], fd[4], fd[7]); 267a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV(" %f, %f, %f}", fd[2], fd[5], fd[8]); 268a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (dataType == RS_TYPE_MATRIX_2X2) { 269a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Matrix2x2"); 270a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("{%f, %f", fd[0], fd[2]); 271a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV(" %f, %f}", fd[1], fd[3]); 272a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else { 273a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk switch (field->getComponent().getVectorSize()) { 274a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 1: 275a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Uniform 1 = %f", fd[0]); 276a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 277a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 2: 278a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Uniform 2 = %f %f", fd[0], fd[1]); 279a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 280a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 3: 281a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Uniform 3 = %f %f %f", fd[0], fd[1], fd[2]); 282a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 283a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 4: 284a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Uniform 4 = %f %f %f %f", fd[0], fd[1], fd[2], fd[3]); 285a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 286a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk default: 287a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(0); 288a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 289a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 290a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("Element size %u data=%p", elementSize, fd); 291a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk fd += elementSize; 292a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("New data=%p", fd); 293a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 294a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 295a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 296a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::setUniform(const Context *rsc, const Element *field, const float *fd, 297a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk int32_t slot, uint32_t arraySize ) { 298a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk RsDataType dataType = field->getType(); 299a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (dataType == RS_TYPE_MATRIX_4X4) { 300a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniformMatrix4fv(slot, arraySize, GL_FALSE, fd); 301a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (dataType == RS_TYPE_MATRIX_3X3) { 302a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniformMatrix3fv(slot, arraySize, GL_FALSE, fd); 303a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (dataType == RS_TYPE_MATRIX_2X2) { 304a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniformMatrix2fv(slot, arraySize, GL_FALSE, fd); 305a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else { 306a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk switch (field->getComponent().getVectorSize()) { 307a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 1: 308a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniform1fv(slot, arraySize, fd); 309a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 310a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 2: 311a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniform2fv(slot, arraySize, fd); 312a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 313a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 3: 314a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniform3fv(slot, arraySize, fd); 315a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 316a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 4: 317a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniform4fv(slot, arraySize, fd); 318a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 319a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk default: 320a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(0); 321a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 322a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 323a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 324a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 3257f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchoukvoid RsdShader::setupSampler(const Context *rsc, const Sampler *s, const Allocation *tex) { 3267f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk RsdHal *dc = (RsdHal *)rsc->mHal.drv; 3277f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk 3287f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GLenum trans[] = { 3297f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GL_NEAREST, //RS_SAMPLER_NEAREST, 3307f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GL_LINEAR, //RS_SAMPLER_LINEAR, 3317f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GL_LINEAR_MIPMAP_LINEAR, //RS_SAMPLER_LINEAR_MIP_LINEAR, 3327f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GL_REPEAT, //RS_SAMPLER_WRAP, 3337f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GL_CLAMP_TO_EDGE, //RS_SAMPLER_CLAMP 3347f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GL_LINEAR_MIPMAP_NEAREST, //RS_SAMPLER_LINEAR_MIP_NEAREST 3357f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk }; 3367f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk 3377f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GLenum transNP[] = { 3387f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GL_NEAREST, //RS_SAMPLER_NEAREST, 3397f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GL_LINEAR, //RS_SAMPLER_LINEAR, 3407f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GL_LINEAR, //RS_SAMPLER_LINEAR_MIP_LINEAR, 3417f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GL_CLAMP_TO_EDGE, //RS_SAMPLER_WRAP, 3427f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GL_CLAMP_TO_EDGE, //RS_SAMPLER_CLAMP 3437f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk GL_LINEAR, //RS_SAMPLER_LINEAR_MIP_NEAREST, 3447f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk }; 3457f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk 3467f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk // This tells us the correct texture type 347eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams DrvAllocation *drvTex = (DrvAllocation *)tex->mHal.drv; 348eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams const GLenum target = drvTex->glTarget; 3497f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk 3507f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk if (!dc->gl.gl.OES_texture_npot && tex->getType()->getIsNp2()) { 3517f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk if (tex->getHasGraphicsMipmaps() && 3527f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk (dc->gl.gl.GL_NV_texture_npot_2D_mipmap || dc->gl.gl.GL_IMG_texture_npot)) { 3537f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk if (dc->gl.gl.GL_NV_texture_npot_2D_mipmap) { 3547f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_MIN_FILTER, trans[s->mHal.state.minFilter]); 3557f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk } else { 3567f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk switch (trans[s->mHal.state.minFilter]) { 3577f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk case GL_LINEAR_MIPMAP_LINEAR: 3587f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_MIN_FILTER, GL_LINEAR_MIPMAP_NEAREST); 3597f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk break; 3607f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk default: 3617f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_MIN_FILTER, trans[s->mHal.state.minFilter]); 3627f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk break; 3637f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk } 3647f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk } 3657f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk } else { 3667f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_MIN_FILTER, transNP[s->mHal.state.minFilter]); 3677f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk } 3687f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_MAG_FILTER, transNP[s->mHal.state.magFilter]); 3697f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_WRAP_S, transNP[s->mHal.state.wrapS]); 3707f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_WRAP_T, transNP[s->mHal.state.wrapT]); 3717f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk } else { 3727f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk if (tex->getHasGraphicsMipmaps()) { 3737f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_MIN_FILTER, trans[s->mHal.state.minFilter]); 3747f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk } else { 3757f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_MIN_FILTER, transNP[s->mHal.state.minFilter]); 3767f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk } 3777f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_MAG_FILTER, trans[s->mHal.state.magFilter]); 3787f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_WRAP_S, trans[s->mHal.state.wrapS]); 3797f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_WRAP_T, trans[s->mHal.state.wrapT]); 3807f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk } 3817f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk 3827f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk float anisoValue = rsMin(dc->gl.gl.EXT_texture_max_aniso, s->mHal.state.aniso); 3837f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk if (dc->gl.gl.EXT_texture_max_aniso > 1.0f) { 3847f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk glTexParameterf(target, GL_TEXTURE_MAX_ANISOTROPY_EXT, anisoValue); 3857f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk } 3867f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk 387c19ff0177a7a0dadfc01b1990f822354fdc95991Alex Sakhartchouk rsdGLCheckError(rsc, "Sampler::setup tex env"); 3887f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk} 3897f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk 390a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::setupTextures(const Context *rsc, RsdShaderCache *sc) { 391a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mRSProgram->mHal.state.texturesCount == 0) { 392a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk return; 393a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 394a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 395c19ff0177a7a0dadfc01b1990f822354fdc95991Alex Sakhartchouk RsdHal *dc = (RsdHal *)rsc->mHal.drv; 396c19ff0177a7a0dadfc01b1990f822354fdc95991Alex Sakhartchouk 397a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t numTexturesToBind = mRSProgram->mHal.state.texturesCount; 398c19ff0177a7a0dadfc01b1990f822354fdc95991Alex Sakhartchouk uint32_t numTexturesAvailable = dc->gl.gl.maxFragmentTextureImageUnits; 399a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (numTexturesToBind >= numTexturesAvailable) { 400a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("Attempting to bind %u textures on shader id %u, but only %u are available", 401a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mRSProgram->mHal.state.texturesCount, (uint32_t)this, numTexturesAvailable); 402a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "Cannot bind more textuers than available"); 403a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk numTexturesToBind = numTexturesAvailable; 404a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 405a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 406a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < numTexturesToBind; ct++) { 407a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glActiveTexture(GL_TEXTURE0 + ct); 408ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk glUniform1i(sc->fragUniformSlot(mTextureUniformIndexStart + ct), ct); 409ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk 410a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (!mRSProgram->mHal.state.textures[ct].get()) { 411ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk // if nothing is bound, reset to default GL texture 412ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk glBindTexture(mTextureTargets[ct], 0); 413a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk continue; 414a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 415a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 416eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams DrvAllocation *drvTex = (DrvAllocation *)mRSProgram->mHal.state.textures[ct]->mHal.drv; 417eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams if (drvTex->glTarget != GL_TEXTURE_2D && drvTex->glTarget != GL_TEXTURE_CUBE_MAP) { 418a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("Attempting to bind unknown texture to shader id %u, texture unit %u", (uint)this, ct); 419a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "Non-texture allocation bound to a shader"); 420a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 421eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams glBindTexture(drvTex->glTarget, drvTex->textureID); 422c19ff0177a7a0dadfc01b1990f822354fdc95991Alex Sakhartchouk rsdGLCheckError(rsc, "ProgramFragment::setup tex bind"); 423a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mRSProgram->mHal.state.samplers[ct].get()) { 4247f126c78a107257090c6675ea40ffac41516a9dcAlex Sakhartchouk setupSampler(rsc, mRSProgram->mHal.state.samplers[ct].get(), mRSProgram->mHal.state.textures[ct].get()); 425a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else { 426eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams glTexParameteri(drvTex->glTarget, GL_TEXTURE_MIN_FILTER, GL_NEAREST); 427eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams glTexParameteri(drvTex->glTarget, GL_TEXTURE_MAG_FILTER, GL_NEAREST); 428eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams glTexParameteri(drvTex->glTarget, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE); 429eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams glTexParameteri(drvTex->glTarget, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE); 430c19ff0177a7a0dadfc01b1990f822354fdc95991Alex Sakhartchouk rsdGLCheckError(rsc, "ProgramFragment::setup tex env"); 431a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 432c19ff0177a7a0dadfc01b1990f822354fdc95991Alex Sakhartchouk rsdGLCheckError(rsc, "ProgramFragment::setup uniforms"); 433a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 434a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 435a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glActiveTexture(GL_TEXTURE0); 436a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mDirty = false; 437c19ff0177a7a0dadfc01b1990f822354fdc95991Alex Sakhartchouk rsdGLCheckError(rsc, "ProgramFragment::setup"); 438a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 439a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 440a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::setupUserConstants(const Context *rsc, RsdShaderCache *sc, bool isFragment) { 441a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t uidx = 0; 442a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.constantsCount; ct++) { 443a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk Allocation *alloc = mRSProgram->mHal.state.constants[ct].get(); 444a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (!alloc) { 445a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("Attempting to set constants on shader id %u, but alloc at slot %u is not set", (uint32_t)this, ct); 446a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "No constant allocation bound"); 447a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk continue; 448a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 449a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 450a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const uint8_t *data = static_cast<const uint8_t *>(alloc->getPtr()); 451a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *e = mRSProgram->mHal.state.constantTypes[ct]->getElement(); 452a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t field=0; field < e->getFieldCount(); field++) { 453a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *f = e->getField(field); 454a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const char *fieldName = e->getFieldName(field); 455a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk // If this field is padding, skip it 456a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (fieldName[0] == '#') { 457a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk continue; 458a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 459a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 460a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t offset = e->getFieldOffsetBytes(field); 461a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const float *fd = reinterpret_cast<const float *>(&data[offset]); 462a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 463a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk int32_t slot = -1; 464a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t arraySize = 1; 465a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (!isFragment) { 466a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk slot = sc->vtxUniformSlot(uidx); 467a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk arraySize = sc->vtxUniformSize(uidx); 468a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else { 469a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk slot = sc->fragUniformSlot(uidx); 470a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk arraySize = sc->fragUniformSize(uidx); 471a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 472a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (rsc->props.mLogShadersUniforms) { 473a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Uniform slot=%i, offset=%i, constant=%i, field=%i, uidx=%i, name=%s", slot, offset, ct, field, uidx, fieldName); 474a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 475a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uidx ++; 476a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (slot < 0) { 477a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk continue; 478a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 479a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 480a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (rsc->props.mLogShadersUniforms) { 481a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk logUniform(f, fd, arraySize); 482a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 483a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk setUniform(rsc, f, fd, slot, arraySize); 484a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 485a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 486a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 487a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 488a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::setup(const android::renderscript::Context *rsc, RsdShaderCache *sc) { 489a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 490a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk setupUserConstants(rsc, sc, mType == GL_FRAGMENT_SHADER); 491a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk setupTextures(rsc, sc); 492a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 493a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 494a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::initAttribAndUniformArray() { 495a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mAttribCount = 0; 496a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.inputElementsCount; ct++) { 497a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *elem = mRSProgram->mHal.state.inputElements[ct].get(); 498a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t field=0; field < elem->getFieldCount(); field++) { 499a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (elem->getFieldName(field)[0] != '#') { 500a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mAttribCount ++; 501a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 502a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 503a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 504a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 505a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformCount = 0; 506a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.constantsCount; ct++) { 507a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *elem = mRSProgram->mHal.state.constantTypes[ct]->getElement(); 508a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 509a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t field=0; field < elem->getFieldCount(); field++) { 510a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (elem->getFieldName(field)[0] != '#') { 511a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformCount ++; 512a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 513a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 514a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 515a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformCount += mRSProgram->mHal.state.texturesCount; 516a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 517a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mAttribCount) { 518a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mAttribNames = new String8[mAttribCount]; 519a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 520a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mUniformCount) { 521a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformNames = new String8[mUniformCount]; 522a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformArraySizes = new uint32_t[mUniformCount]; 523a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 524ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk 525ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk mTextureCount = mRSProgram->mHal.state.texturesCount; 526ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk if (mTextureCount) { 527ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk mTextureTargets = new uint32_t[mTextureCount]; 528ba157304ce99d212403898b055998e6da7a20e9bAlex Sakhartchouk } 529a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 530a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 531a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::initAddUserElement(const Element *e, String8 *names, uint32_t *arrayLengths, uint32_t *count, const char *prefix) { 532a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(e->getFieldCount()); 533a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < e->getFieldCount(); ct++) { 534a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *ce = e->getField(ct); 535a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (ce->getFieldCount()) { 536a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk initAddUserElement(ce, names, arrayLengths, count, prefix); 537a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (e->getFieldName(ct)[0] != '#') { 538a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk String8 tmp(prefix); 539a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk tmp.append(e->getFieldName(ct)); 540a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk names[*count].setTo(tmp.string()); 541a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (arrayLengths) { 542a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk arrayLengths[*count] = e->getFieldArraySize(ct); 543a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 544a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk (*count)++; 545a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 546a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 547a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 548