Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 1 | /* |
Jason Sams | 3522f40 | 2012-03-23 11:47:26 -0700 | [diff] [blame] | 2 | * Copyright (C) 2011-2012 The Android Open Source Project |
Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
Alex Sakhartchouk | a04e30d | 2011-04-29 16:49:08 -0700 | [diff] [blame] | 17 | #ifndef ANDROID_RSD_SHADER_CACHE_H |
| 18 | #define ANDROID_RSD_SHADER_CACHE_H |
Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 19 | |
Chris Wailes | 93d6bc8 | 2014-07-28 16:54:38 -0700 | [diff] [blame] | 20 | #include <string> |
| 21 | #include <vector> |
| 22 | |
Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 23 | namespace android { |
| 24 | namespace renderscript { |
| 25 | |
Alex Sakhartchouk | a04e30d | 2011-04-29 16:49:08 -0700 | [diff] [blame] | 26 | class Context; |
| 27 | |
| 28 | } |
| 29 | } |
| 30 | |
Chris Wailes | 6847e73 | 2014-08-11 17:30:51 -0700 | [diff] [blame] | 31 | #if defined(RS_SERVER) || defined(RS_COMPATIBILITY_LIB) |
Tim Murray | 0b575de | 2013-03-15 15:56:43 -0700 | [diff] [blame] | 32 | #include "rsUtils.h" |
| 33 | #endif |
Alex Sakhartchouk | a04e30d | 2011-04-29 16:49:08 -0700 | [diff] [blame] | 34 | class RsdShader; |
| 35 | |
| 36 | // --------------------------------------------------------------------------- |
Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 37 | |
| 38 | // An element is a group of Components that occupies one cell in a structure. |
Alex Sakhartchouk | a04e30d | 2011-04-29 16:49:08 -0700 | [diff] [blame] | 39 | class RsdShaderCache { |
Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 40 | public: |
Alex Sakhartchouk | a04e30d | 2011-04-29 16:49:08 -0700 | [diff] [blame] | 41 | RsdShaderCache(); |
| 42 | virtual ~RsdShaderCache(); |
Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 43 | |
Alex Sakhartchouk | a04e30d | 2011-04-29 16:49:08 -0700 | [diff] [blame] | 44 | void setActiveVertex(RsdShader *pv) { |
| 45 | mVertexDirty = true; |
| 46 | mVertex = pv; |
| 47 | } |
| 48 | |
| 49 | void setActiveFragment(RsdShader *pf) { |
| 50 | mFragmentDirty = true; |
| 51 | mFragment = pf; |
| 52 | } |
| 53 | |
| 54 | bool setup(const android::renderscript::Context *rsc); |
Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 55 | |
Alex Sakhartchouk | 6d6e114 | 2012-03-26 13:52:24 -0700 | [diff] [blame] | 56 | void cleanupVertex(RsdShader *s); |
| 57 | void cleanupFragment(RsdShader *s); |
Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 58 | |
| 59 | void cleanupAll(); |
| 60 | |
Chris Wailes | 93d6bc8 | 2014-07-28 16:54:38 -0700 | [diff] [blame] | 61 | int32_t vtxAttribSlot(const std::string &attrName) const; |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 62 | int32_t vtxUniformSlot(uint32_t a) const {return mCurrent->vtxUniforms[a].slot;} |
| 63 | uint32_t vtxUniformSize(uint32_t a) const {return mCurrent->vtxUniforms[a].arraySize;} |
| 64 | int32_t fragUniformSlot(uint32_t a) const {return mCurrent->fragUniforms[a].slot;} |
| 65 | uint32_t fragUniformSize(uint32_t a) const {return mCurrent->fragUniforms[a].arraySize;} |
Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 66 | |
| 67 | protected: |
Alex Sakhartchouk | a04e30d | 2011-04-29 16:49:08 -0700 | [diff] [blame] | 68 | bool link(const android::renderscript::Context *rsc); |
| 69 | bool mFragmentDirty; |
| 70 | bool mVertexDirty; |
| 71 | RsdShader *mVertex; |
| 72 | RsdShader *mFragment; |
| 73 | |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 74 | struct UniformQueryData { |
| 75 | char *name; |
| 76 | uint32_t nameLength; |
| 77 | int32_t writtenLength; |
| 78 | int32_t arraySize; |
| 79 | uint32_t type; |
| 80 | UniformQueryData(uint32_t maxName) { |
Chris Wailes | 44bef6f | 2014-08-12 13:51:10 -0700 | [diff] [blame^] | 81 | name = nullptr; |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 82 | nameLength = maxName; |
Alex Sakhartchouk | afb743a | 2010-11-09 17:00:54 -0800 | [diff] [blame] | 83 | if (nameLength > 0 ) { |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 84 | name = new char[nameLength]; |
| 85 | } |
| 86 | } |
| 87 | ~UniformQueryData() { |
Chris Wailes | 44bef6f | 2014-08-12 13:51:10 -0700 | [diff] [blame^] | 88 | if (name != nullptr) { |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 89 | delete[] name; |
Chris Wailes | 44bef6f | 2014-08-12 13:51:10 -0700 | [diff] [blame^] | 90 | name = nullptr; |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 91 | } |
| 92 | } |
| 93 | }; |
| 94 | struct UniformData { |
| 95 | int32_t slot; |
| 96 | uint32_t arraySize; |
| 97 | }; |
| 98 | struct AttrData { |
| 99 | int32_t slot; |
| 100 | const char* name; |
| 101 | }; |
| 102 | struct ProgramEntry { |
| 103 | ProgramEntry(uint32_t numVtxAttr, uint32_t numVtxUnis, |
| 104 | uint32_t numFragUnis) : vtx(0), frag(0), program(0), vtxAttrCount(0), |
Stephen Hines | ebf4a14 | 2012-02-13 17:49:20 -0800 | [diff] [blame] | 105 | vtxAttrs(0), vtxUniforms(0), fragUniforms(0), |
| 106 | fragUniformIsSTO(0) { |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 107 | vtxAttrCount = numVtxAttr; |
Alex Sakhartchouk | afb743a | 2010-11-09 17:00:54 -0800 | [diff] [blame] | 108 | if (numVtxAttr) { |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 109 | vtxAttrs = new AttrData[numVtxAttr]; |
| 110 | } |
Alex Sakhartchouk | afb743a | 2010-11-09 17:00:54 -0800 | [diff] [blame] | 111 | if (numVtxUnis) { |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 112 | vtxUniforms = new UniformData[numVtxUnis]; |
| 113 | } |
Alex Sakhartchouk | afb743a | 2010-11-09 17:00:54 -0800 | [diff] [blame] | 114 | if (numFragUnis) { |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 115 | fragUniforms = new UniformData[numFragUnis]; |
Jason Sams | cf27eb4 | 2012-02-10 13:24:18 -0800 | [diff] [blame] | 116 | fragUniformIsSTO = new bool[numFragUnis]; |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 117 | } |
| 118 | } |
| 119 | ~ProgramEntry() { |
Alex Sakhartchouk | afb743a | 2010-11-09 17:00:54 -0800 | [diff] [blame] | 120 | if (vtxAttrs) { |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 121 | delete[] vtxAttrs; |
Chris Wailes | 44bef6f | 2014-08-12 13:51:10 -0700 | [diff] [blame^] | 122 | vtxAttrs = nullptr; |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 123 | } |
Alex Sakhartchouk | afb743a | 2010-11-09 17:00:54 -0800 | [diff] [blame] | 124 | if (vtxUniforms) { |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 125 | delete[] vtxUniforms; |
Chris Wailes | 44bef6f | 2014-08-12 13:51:10 -0700 | [diff] [blame^] | 126 | vtxUniforms = nullptr; |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 127 | } |
Alex Sakhartchouk | afb743a | 2010-11-09 17:00:54 -0800 | [diff] [blame] | 128 | if (fragUniforms) { |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 129 | delete[] fragUniforms; |
Chris Wailes | 44bef6f | 2014-08-12 13:51:10 -0700 | [diff] [blame^] | 130 | fragUniforms = nullptr; |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 131 | } |
Jason Sams | cf27eb4 | 2012-02-10 13:24:18 -0800 | [diff] [blame] | 132 | if (fragUniformIsSTO) { |
| 133 | delete[] fragUniformIsSTO; |
Chris Wailes | 44bef6f | 2014-08-12 13:51:10 -0700 | [diff] [blame^] | 134 | fragUniformIsSTO = nullptr; |
Jason Sams | cf27eb4 | 2012-02-10 13:24:18 -0800 | [diff] [blame] | 135 | } |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 136 | } |
Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 137 | uint32_t vtx; |
| 138 | uint32_t frag; |
| 139 | uint32_t program; |
Alex Sakhartchouk | 886f11a | 2010-09-29 09:49:13 -0700 | [diff] [blame] | 140 | uint32_t vtxAttrCount; |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 141 | AttrData *vtxAttrs; |
| 142 | UniformData *vtxUniforms; |
| 143 | UniformData *fragUniforms; |
Jason Sams | cf27eb4 | 2012-02-10 13:24:18 -0800 | [diff] [blame] | 144 | bool *fragUniformIsSTO; |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 145 | }; |
Chris Wailes | 93d6bc8 | 2014-07-28 16:54:38 -0700 | [diff] [blame] | 146 | std::vector<ProgramEntry*> mEntries; |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 147 | ProgramEntry *mCurrent; |
Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 148 | |
Alex Sakhartchouk | a04e30d | 2011-04-29 16:49:08 -0700 | [diff] [blame] | 149 | bool hasArrayUniforms(RsdShader *vtx, RsdShader *frag); |
| 150 | void populateUniformData(RsdShader *prog, uint32_t linkedID, UniformData *data); |
| 151 | void updateUniformArrayData(const android::renderscript::Context *rsc, |
| 152 | RsdShader *prog, uint32_t linkedID, |
Alex Sakhartchouk | 54929cc | 2010-11-08 15:10:52 -0800 | [diff] [blame] | 153 | UniformData *data, const char* logTag, |
| 154 | UniformQueryData **uniformList, uint32_t uniListSize); |
Jason Sams | c460e55 | 2009-11-25 13:22:07 -0800 | [diff] [blame] | 155 | }; |
| 156 | |
| 157 | |
Alex Sakhartchouk | a04e30d | 2011-04-29 16:49:08 -0700 | [diff] [blame] | 158 | #endif //ANDROID_RSD_SHADER_CACHE_H |