rsdShader.cpp revision a04e30dbb5ab11592b03666bb3d102070759c58e
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 24a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk#include "rsdShader.h" 25a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk#include "rsdShaderCache.h" 26a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 27a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukusing namespace android; 28a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukusing namespace android::renderscript; 29a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 30a04e30dbb5ab11592b03666bb3d102070759c58eAlex SakhartchoukRsdShader::RsdShader(const Program *p, uint32_t type, 31a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const char * shaderText, uint32_t shaderLength) { 32a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 33a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUserShader.setTo(shaderText, shaderLength); 34a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mRSProgram = p; 35a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mType = type; 36a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk initMemberVars(); 37a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk initAttribAndUniformArray(); 38a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk init(); 39a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 40a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 41a04e30dbb5ab11592b03666bb3d102070759c58eAlex SakhartchoukRsdShader::~RsdShader() { 42a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mShaderID) { 43a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glDeleteShader(mShaderID); 44a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 45a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 46a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk delete[] mAttribNames; 47a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk delete[] mUniformNames; 48a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk delete[] mUniformArraySizes; 49a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 50a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 51a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::initMemberVars() { 52a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mDirty = true; 53a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShaderID = 0; 54a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mAttribCount = 0; 55a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformCount = 0; 56a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 57a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mAttribNames = NULL; 58a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformNames = NULL; 59a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformArraySizes = NULL; 60a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 61a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mIsValid = false; 62a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 63a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 64a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::init() { 65a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t attribCount = 0; 66a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t uniformCount = 0; 67a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.inputElementsCount; ct++) { 68a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk initAddUserElement(mRSProgram->mHal.state.inputElements[ct].get(), mAttribNames, NULL, &attribCount, RS_SHADER_ATTR); 69a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 70a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.constantsCount; ct++) { 71a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk initAddUserElement(mRSProgram->mHal.state.constantTypes[ct]->getElement(), mUniformNames, mUniformArraySizes, &uniformCount, RS_SHADER_UNI); 72a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 73a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 74a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mTextureUniformIndexStart = uniformCount; 75a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk char buf[256]; 76a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.texturesCount; ct++) { 77a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk snprintf(buf, sizeof(buf), "UNI_Tex%i", ct); 78a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformNames[uniformCount].setTo(buf); 79a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformArraySizes[uniformCount] = 1; 80a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uniformCount++; 81a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 82a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 83a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 84a04e30dbb5ab11592b03666bb3d102070759c58eAlex SakhartchoukString8 RsdShader::getGLSLInputString() const { 85a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk String8 s; 86a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.inputElementsCount; ct++) { 87a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *e = mRSProgram->mHal.state.inputElements[ct].get(); 88a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t field=0; field < e->getFieldCount(); field++) { 89a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *f = e->getField(field); 90a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 91a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk // Cannot be complex 92a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(!f->getFieldCount()); 93a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk switch (f->getComponent().getVectorSize()) { 94a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 1: s.append("attribute float ATTRIB_"); break; 95a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 2: s.append("attribute vec2 ATTRIB_"); break; 96a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 3: s.append("attribute vec3 ATTRIB_"); break; 97a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 4: s.append("attribute vec4 ATTRIB_"); break; 98a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk default: 99a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(0); 100a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 101a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 102a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk s.append(e->getFieldName(field)); 103a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk s.append(";\n"); 104a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 105a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 106a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk return s; 107a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 108a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 109a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::appendAttributes() { 110a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.inputElementsCount; ct++) { 111a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *e = mRSProgram->mHal.state.inputElements[ct].get(); 112a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t field=0; field < e->getFieldCount(); field++) { 113a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *f = e->getField(field); 114a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const char *fn = e->getFieldName(field); 115a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 116a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (fn[0] == '#') { 117a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk continue; 118a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 119a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 120a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk // Cannot be complex 121a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(!f->getFieldCount()); 122a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk switch (f->getComponent().getVectorSize()) { 123a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 1: mShader.append("attribute float ATTRIB_"); break; 124a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 2: mShader.append("attribute vec2 ATTRIB_"); break; 125a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 3: mShader.append("attribute vec3 ATTRIB_"); break; 126a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 4: mShader.append("attribute vec4 ATTRIB_"); break; 127a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk default: 128a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(0); 129a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 130a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 131a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append(fn); 132a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append(";\n"); 133a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 134a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 135a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 136a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 137a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::appendTextures() { 138a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk char buf[256]; 139a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.texturesCount; ct++) { 140a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mRSProgram->mHal.state.textureTargets[ct] == RS_TEXTURE_2D) { 141a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk snprintf(buf, sizeof(buf), "uniform sampler2D UNI_Tex%i;\n", ct); 142a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else { 143a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk snprintf(buf, sizeof(buf), "uniform samplerCube UNI_Tex%i;\n", ct); 144a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 145a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append(buf); 146a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 147a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 148a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 149a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukbool RsdShader::createShader() { 150a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 151a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mType == GL_FRAGMENT_SHADER) { 152a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append("precision mediump float;\n"); 153a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 154a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk appendUserConstants(); 155a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk appendAttributes(); 156a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk appendTextures(); 157a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 158a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append(mUserShader); 159a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 160a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk return true; 161a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 162a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 163a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukbool RsdShader::loadShader(const Context *rsc) { 164a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShaderID = glCreateShader(mType); 165a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(mShaderID); 166a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 167a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (rsc->props.mLogShaders) { 168a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Loading shader type %x, ID %i", mType, mShaderID); 169a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("%s", mShader.string()); 170a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 171a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 172a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mShaderID) { 173a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const char * ss = mShader.string(); 174a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glShaderSource(mShaderID, 1, &ss, NULL); 175a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glCompileShader(mShaderID); 176a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 177a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk GLint compiled = 0; 178a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glGetShaderiv(mShaderID, GL_COMPILE_STATUS, &compiled); 179a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (!compiled) { 180a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk GLint infoLen = 0; 181a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glGetShaderiv(mShaderID, GL_INFO_LOG_LENGTH, &infoLen); 182a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (infoLen) { 183a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk char* buf = (char*) malloc(infoLen); 184a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (buf) { 185a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glGetShaderInfoLog(mShaderID, infoLen, NULL, buf); 186a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("Could not compile shader \n%s\n", buf); 187a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk free(buf); 188a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 189a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glDeleteShader(mShaderID); 190a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShaderID = 0; 191a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "Error returned from GL driver loading shader text,"); 192a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk return false; 193a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 194a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 195a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 196a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 197a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (rsc->props.mLogShaders) { 198a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("--Shader load result %x ", glGetError()); 199a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 200a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mIsValid = true; 201a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk return true; 202a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 203a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 204a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::appendUserConstants() { 205a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.constantsCount; ct++) { 206a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *e = mRSProgram->mHal.state.constantTypes[ct]->getElement(); 207a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t field=0; field < e->getFieldCount(); field++) { 208a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *f = e->getField(field); 209a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const char *fn = e->getFieldName(field); 210a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 211a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (fn[0] == '#') { 212a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk continue; 213a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 214a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 215a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk // Cannot be complex 216a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(!f->getFieldCount()); 217a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (f->getType() == RS_TYPE_MATRIX_4X4) { 218a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append("uniform mat4 UNI_"); 219a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (f->getType() == RS_TYPE_MATRIX_3X3) { 220a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append("uniform mat3 UNI_"); 221a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (f->getType() == RS_TYPE_MATRIX_2X2) { 222a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append("uniform mat2 UNI_"); 223a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else { 224a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk switch (f->getComponent().getVectorSize()) { 225a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 1: mShader.append("uniform float UNI_"); break; 226a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 2: mShader.append("uniform vec2 UNI_"); break; 227a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 3: mShader.append("uniform vec3 UNI_"); break; 228a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 4: mShader.append("uniform vec4 UNI_"); break; 229a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk default: 230a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(0); 231a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 232a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 233a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 234a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append(fn); 235a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (e->getFieldArraySize(field) > 1) { 236a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.appendFormat("[%d]", e->getFieldArraySize(field)); 237a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 238a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mShader.append(";\n"); 239a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 240a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 241a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 242a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 243a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::logUniform(const Element *field, const float *fd, uint32_t arraySize ) { 244a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk RsDataType dataType = field->getType(); 245a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t elementSize = field->getSizeBytes() / sizeof(float); 246a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t i = 0; i < arraySize; i ++) { 247a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (arraySize > 1) { 248a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Array Element [%u]", i); 249a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 250a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (dataType == RS_TYPE_MATRIX_4X4) { 251a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Matrix4x4"); 252a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("{%f, %f, %f, %f", fd[0], fd[4], fd[8], fd[12]); 253a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV(" %f, %f, %f, %f", fd[1], fd[5], fd[9], fd[13]); 254a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV(" %f, %f, %f, %f", fd[2], fd[6], fd[10], fd[14]); 255a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV(" %f, %f, %f, %f}", fd[3], fd[7], fd[11], fd[15]); 256a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (dataType == RS_TYPE_MATRIX_3X3) { 257a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Matrix3x3"); 258a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("{%f, %f, %f", fd[0], fd[3], fd[6]); 259a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV(" %f, %f, %f", fd[1], fd[4], fd[7]); 260a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV(" %f, %f, %f}", fd[2], fd[5], fd[8]); 261a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (dataType == RS_TYPE_MATRIX_2X2) { 262a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Matrix2x2"); 263a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("{%f, %f", fd[0], fd[2]); 264a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV(" %f, %f}", fd[1], fd[3]); 265a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else { 266a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk switch (field->getComponent().getVectorSize()) { 267a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 1: 268a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Uniform 1 = %f", fd[0]); 269a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 270a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 2: 271a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Uniform 2 = %f %f", fd[0], fd[1]); 272a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 273a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 3: 274a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Uniform 3 = %f %f %f", fd[0], fd[1], fd[2]); 275a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 276a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 4: 277a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Uniform 4 = %f %f %f %f", fd[0], fd[1], fd[2], fd[3]); 278a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 279a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk default: 280a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(0); 281a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 282a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 283a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("Element size %u data=%p", elementSize, fd); 284a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk fd += elementSize; 285a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("New data=%p", fd); 286a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 287a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 288a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 289a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::setUniform(const Context *rsc, const Element *field, const float *fd, 290a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk int32_t slot, uint32_t arraySize ) { 291a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk RsDataType dataType = field->getType(); 292a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (dataType == RS_TYPE_MATRIX_4X4) { 293a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniformMatrix4fv(slot, arraySize, GL_FALSE, fd); 294a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (dataType == RS_TYPE_MATRIX_3X3) { 295a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniformMatrix3fv(slot, arraySize, GL_FALSE, fd); 296a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (dataType == RS_TYPE_MATRIX_2X2) { 297a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniformMatrix2fv(slot, arraySize, GL_FALSE, fd); 298a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else { 299a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk switch (field->getComponent().getVectorSize()) { 300a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 1: 301a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniform1fv(slot, arraySize, fd); 302a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 303a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 2: 304a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniform2fv(slot, arraySize, fd); 305a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 306a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 3: 307a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniform3fv(slot, arraySize, fd); 308a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 309a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk case 4: 310a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniform4fv(slot, arraySize, fd); 311a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk break; 312a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk default: 313a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(0); 314a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 315a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 316a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 317a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 318a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::setupTextures(const Context *rsc, RsdShaderCache *sc) { 319a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mRSProgram->mHal.state.texturesCount == 0) { 320a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk return; 321a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 322a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 323a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t numTexturesToBind = mRSProgram->mHal.state.texturesCount; 324a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t numTexturesAvailable = rsc->getMaxFragmentTextures(); 325a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (numTexturesToBind >= numTexturesAvailable) { 326a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("Attempting to bind %u textures on shader id %u, but only %u are available", 327a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mRSProgram->mHal.state.texturesCount, (uint32_t)this, numTexturesAvailable); 328a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "Cannot bind more textuers than available"); 329a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk numTexturesToBind = numTexturesAvailable; 330a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 331a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 332a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < numTexturesToBind; ct++) { 333a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glActiveTexture(GL_TEXTURE0 + ct); 334a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (!mRSProgram->mHal.state.textures[ct].get()) { 335a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("No texture bound for shader id %u, texture unit %u", (uint)this, ct); 336a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "No texture bound"); 337a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk continue; 338a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 339a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 340a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk GLenum target = (GLenum)mRSProgram->mHal.state.textures[ct]->getGLTarget(); 341a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (target != GL_TEXTURE_2D && target != GL_TEXTURE_CUBE_MAP) { 342a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("Attempting to bind unknown texture to shader id %u, texture unit %u", (uint)this, ct); 343a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "Non-texture allocation bound to a shader"); 344a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 345a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glBindTexture(target, mRSProgram->mHal.state.textures[ct]->getTextureID()); 346a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->checkError("ProgramFragment::setupGL2 tex bind"); 347a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mRSProgram->mHal.state.samplers[ct].get()) { 348a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mRSProgram->mHal.state.samplers[ct]->setupGL(rsc, mRSProgram->mHal.state.textures[ct].get()); 349a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else { 350a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_MIN_FILTER, GL_NEAREST); 351a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_MAG_FILTER, GL_NEAREST); 352a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE); 353a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glTexParameteri(target, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE); 354a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->checkError("ProgramFragment::setupGL2 tex env"); 355a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 356a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 357a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glUniform1i(sc->fragUniformSlot(mTextureUniformIndexStart + ct), ct); 358a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->checkError("ProgramFragment::setupGL2 uniforms"); 359a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 360a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 361a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk glActiveTexture(GL_TEXTURE0); 362a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mDirty = false; 363a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->checkError("ProgramFragment::setupGL2"); 364a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 365a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 366a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::setupUserConstants(const Context *rsc, RsdShaderCache *sc, bool isFragment) { 367a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t uidx = 0; 368a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.constantsCount; ct++) { 369a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk Allocation *alloc = mRSProgram->mHal.state.constants[ct].get(); 370a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (!alloc) { 371a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGE("Attempting to set constants on shader id %u, but alloc at slot %u is not set", (uint32_t)this, ct); 372a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsc->setError(RS_ERROR_BAD_SHADER, "No constant allocation bound"); 373a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk continue; 374a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 375a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 376a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const uint8_t *data = static_cast<const uint8_t *>(alloc->getPtr()); 377a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *e = mRSProgram->mHal.state.constantTypes[ct]->getElement(); 378a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t field=0; field < e->getFieldCount(); field++) { 379a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *f = e->getField(field); 380a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const char *fieldName = e->getFieldName(field); 381a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk // If this field is padding, skip it 382a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (fieldName[0] == '#') { 383a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk continue; 384a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 385a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 386a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t offset = e->getFieldOffsetBytes(field); 387a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const float *fd = reinterpret_cast<const float *>(&data[offset]); 388a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 389a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk int32_t slot = -1; 390a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uint32_t arraySize = 1; 391a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (!isFragment) { 392a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk slot = sc->vtxUniformSlot(uidx); 393a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk arraySize = sc->vtxUniformSize(uidx); 394a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else { 395a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk slot = sc->fragUniformSlot(uidx); 396a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk arraySize = sc->fragUniformSize(uidx); 397a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 398a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (rsc->props.mLogShadersUniforms) { 399a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk LOGV("Uniform slot=%i, offset=%i, constant=%i, field=%i, uidx=%i, name=%s", slot, offset, ct, field, uidx, fieldName); 400a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 401a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk uidx ++; 402a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (slot < 0) { 403a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk continue; 404a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 405a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 406a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (rsc->props.mLogShadersUniforms) { 407a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk logUniform(f, fd, arraySize); 408a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 409a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk setUniform(rsc, f, fd, slot, arraySize); 410a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 411a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 412a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 413a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 414a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::setup(const android::renderscript::Context *rsc, RsdShaderCache *sc) { 415a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 416a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk setupUserConstants(rsc, sc, mType == GL_FRAGMENT_SHADER); 417a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk setupTextures(rsc, sc); 418a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 419a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 420a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::initAttribAndUniformArray() { 421a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mAttribCount = 0; 422a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.inputElementsCount; ct++) { 423a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *elem = mRSProgram->mHal.state.inputElements[ct].get(); 424a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t field=0; field < elem->getFieldCount(); field++) { 425a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (elem->getFieldName(field)[0] != '#') { 426a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mAttribCount ++; 427a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 428a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 429a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 430a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 431a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformCount = 0; 432a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < mRSProgram->mHal.state.constantsCount; ct++) { 433a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *elem = mRSProgram->mHal.state.constantTypes[ct]->getElement(); 434a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 435a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t field=0; field < elem->getFieldCount(); field++) { 436a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (elem->getFieldName(field)[0] != '#') { 437a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformCount ++; 438a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 439a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 440a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 441a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformCount += mRSProgram->mHal.state.texturesCount; 442a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 443a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mAttribCount) { 444a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mAttribNames = new String8[mAttribCount]; 445a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 446a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (mUniformCount) { 447a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformNames = new String8[mUniformCount]; 448a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk mUniformArraySizes = new uint32_t[mUniformCount]; 449a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 450a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 451a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk 452a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchoukvoid RsdShader::initAddUserElement(const Element *e, String8 *names, uint32_t *arrayLengths, uint32_t *count, const char *prefix) { 453a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk rsAssert(e->getFieldCount()); 454a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk for (uint32_t ct=0; ct < e->getFieldCount(); ct++) { 455a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk const Element *ce = e->getField(ct); 456a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (ce->getFieldCount()) { 457a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk initAddUserElement(ce, names, arrayLengths, count, prefix); 458a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } else if (e->getFieldName(ct)[0] != '#') { 459a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk String8 tmp(prefix); 460a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk tmp.append(e->getFieldName(ct)); 461a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk names[*count].setTo(tmp.string()); 462a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk if (arrayLengths) { 463a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk arrayLengths[*count] = e->getFieldArraySize(ct); 464a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 465a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk (*count)++; 466a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 467a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk } 468a04e30dbb5ab11592b03666bb3d102070759c58eAlex Sakhartchouk} 469