Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2010 The Android Open Source Project |
| 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 | |
| 17 | #define LOG_TAG "OpenGLRenderer" |
| 18 | |
| 19 | #include <utils/String8.h> |
| 20 | |
| 21 | #include "ProgramCache.h" |
| 22 | |
| 23 | namespace android { |
| 24 | namespace uirenderer { |
| 25 | |
| 26 | /////////////////////////////////////////////////////////////////////////////// |
| 27 | // Vertex shaders snippets |
| 28 | /////////////////////////////////////////////////////////////////////////////// |
| 29 | |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 30 | const char* gVS_Header_Attributes = |
| 31 | "attribute vec4 position;\n"; |
| 32 | const char* gVS_Header_Attributes_TexCoords = |
| 33 | "attribute vec2 texCoords;\n"; |
| 34 | const char* gVS_Header_Uniforms = |
| 35 | "uniform mat4 transform;\n"; |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 36 | const char* gVS_Header_Uniforms_HasGradient[3] = { |
| 37 | // Linear |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 38 | "uniform float gradientLength;\n" |
| 39 | "uniform vec2 gradient;\n" |
| 40 | "uniform vec2 gradientStart;\n" |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 41 | "uniform mat4 screenSpace;\n", |
| 42 | // Circular |
| 43 | "", |
| 44 | // Sweep |
| 45 | "uniform vec2 gradientStart;\n" |
| 46 | "uniform mat4 screenSpace;\n" |
| 47 | }; |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 48 | const char* gVS_Header_Uniforms_HasBitmap = |
| 49 | "uniform mat4 textureTransform;\n" |
| 50 | "uniform vec2 textureDimension;\n"; |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 51 | const char* gVS_Header_Varyings_HasTexture = |
| 52 | "varying vec2 outTexCoords;\n"; |
| 53 | const char* gVS_Header_Varyings_HasBitmap = |
| 54 | "varying vec2 outBitmapTexCoords;\n"; |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 55 | const char* gVS_Header_Varyings_HasGradient[3] = { |
| 56 | // Linear |
| 57 | "varying float index;\n", |
| 58 | // Circular |
| 59 | "", |
| 60 | // Sweep |
| 61 | "varying vec2 sweep;\n" |
| 62 | }; |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 63 | const char* gVS_Main = |
| 64 | "\nvoid main(void) {\n"; |
| 65 | const char* gVS_Main_OutTexCoords = |
| 66 | " outTexCoords = texCoords;\n"; |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 67 | const char* gVS_Main_OutGradient[3] = { |
| 68 | // Linear |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 69 | " vec4 location = screenSpace * position;\n" |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 70 | " index = dot(location.xy - gradientStart, gradient) * gradientLength;\n", |
| 71 | // Circular |
| 72 | "", |
| 73 | // Sweep |
| 74 | " vec4 location = screenSpace * position;\n" |
| 75 | " sweep = location.xy - gradientStart;\n" |
| 76 | }; |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 77 | const char* gVS_Main_OutBitmapTexCoords = |
| 78 | " vec4 bitmapCoords = textureTransform * position;\n" |
| 79 | " outBitmapTexCoords = bitmapCoords.xy * textureDimension;\n"; |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 80 | const char* gVS_Main_Position = |
| 81 | " gl_Position = transform * position;\n"; |
| 82 | const char* gVS_Footer = |
| 83 | "}\n\n"; |
| 84 | |
| 85 | /////////////////////////////////////////////////////////////////////////////// |
| 86 | // Fragment shaders snippets |
| 87 | /////////////////////////////////////////////////////////////////////////////// |
| 88 | |
Romain Guy | a5aed0d | 2010-09-09 14:42:43 -0700 | [diff] [blame] | 89 | const char* gFS_Header_Extension_FramebufferFetch = |
| 90 | "#extension GL_NV_shader_framebuffer_fetch : enable\n\n"; |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 91 | const char* gFS_Header = |
| 92 | "precision mediump float;\n\n"; |
| 93 | const char* gFS_Uniforms_Color = |
| 94 | "uniform vec4 color;\n"; |
| 95 | const char* gFS_Uniforms_TextureSampler = |
| 96 | "uniform sampler2D sampler;\n"; |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 97 | const char* gFS_Uniforms_GradientSampler[3] = { |
| 98 | // Linear |
| 99 | "uniform sampler2D gradientSampler;\n", |
| 100 | // Circular |
| 101 | "uniform sampler2D gradientSampler;\n", |
| 102 | // Sweep |
| 103 | "uniform sampler2D gradientSampler;\n" |
| 104 | }; |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 105 | const char* gFS_Uniforms_BitmapSampler = |
| 106 | "uniform sampler2D bitmapSampler;\n"; |
| 107 | const char* gFS_Uniforms_ColorOp[4] = { |
| 108 | // None |
| 109 | "", |
| 110 | // Matrix |
| 111 | "uniform mat4 colorMatrix;\n" |
| 112 | "uniform vec4 colorMatrixVector;\n", |
| 113 | // Lighting |
Romain Guy | db1938e | 2010-08-02 18:50:22 -0700 | [diff] [blame] | 114 | "uniform vec4 lightingMul;\n" |
| 115 | "uniform vec4 lightingAdd;\n", |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 116 | // PorterDuff |
Romain Guy | db1938e | 2010-08-02 18:50:22 -0700 | [diff] [blame] | 117 | "uniform vec4 colorBlend;\n" |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 118 | }; |
| 119 | const char* gFS_Main = |
| 120 | "\nvoid main(void) {\n" |
Romain Guy | 7fbcc04 | 2010-08-04 15:40:07 -0700 | [diff] [blame] | 121 | " lowp vec4 fragColor;\n"; |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 122 | const char* gFS_Main_FetchColor = |
| 123 | " fragColor = color;\n"; |
| 124 | const char* gFS_Main_FetchTexture = |
| 125 | " fragColor = color * texture2D(sampler, outTexCoords);\n"; |
| 126 | const char* gFS_Main_FetchA8Texture = |
| 127 | " fragColor = color * texture2D(sampler, outTexCoords).a;\n"; |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 128 | const char* gFS_Main_FetchGradient[3] = { |
| 129 | // Linear |
| 130 | " vec4 gradientColor = texture2D(gradientSampler, vec2(index, 0.5));\n", |
| 131 | // Circular |
| 132 | "", |
| 133 | // Sweep |
| 134 | " float index = atan(sweep.y, sweep.x) * 0.15915494309; // inv(2 * PI)\n" |
| 135 | " vec4 gradientColor = texture2D(gradientSampler, vec2(index - floor(index), 0.5));\n" |
| 136 | }; |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 137 | const char* gFS_Main_FetchBitmap = |
| 138 | " vec4 bitmapColor = texture2D(bitmapSampler, outBitmapTexCoords);\n"; |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 139 | const char* gFS_Main_FetchBitmapNpot = |
| 140 | " vec4 bitmapColor = texture2D(bitmapSampler, wrap(outBitmapTexCoords));\n"; |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 141 | const char* gFS_Main_BlendShadersBG = |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 142 | " fragColor = blendShaders(gradientColor, bitmapColor)"; |
Romain Guy | 06f96e2 | 2010-07-30 19:18:16 -0700 | [diff] [blame] | 143 | const char* gFS_Main_BlendShadersGB = |
| 144 | " fragColor = blendShaders(bitmapColor, gradientColor)"; |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 145 | const char* gFS_Main_BlendShaders_Modulate = |
| 146 | " * fragColor.a;\n"; |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 147 | const char* gFS_Main_GradientShader_Modulate = |
| 148 | " fragColor = gradientColor * fragColor.a;\n"; |
| 149 | const char* gFS_Main_BitmapShader_Modulate = |
| 150 | " fragColor = bitmapColor * fragColor.a;\n"; |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 151 | const char* gFS_Main_FragColor = |
| 152 | " gl_FragColor = fragColor;\n"; |
Romain Guy | a5aed0d | 2010-09-09 14:42:43 -0700 | [diff] [blame] | 153 | const char* gFS_Main_FragColor_Blend = |
| 154 | " gl_FragColor = blendFramebuffer(fragColor, gl_LastFragColor);\n"; |
Romain Guy | f607bdc | 2010-09-10 19:20:06 -0700 | [diff] [blame] | 155 | const char* gFS_Main_FragColor_Blend_Swap = |
| 156 | " gl_FragColor = blendFramebuffer(gl_LastFragColor, fragColor);\n"; |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 157 | const char* gFS_Main_ApplyColorOp[4] = { |
| 158 | // None |
| 159 | "", |
| 160 | // Matrix |
Romain Guy | db1938e | 2010-08-02 18:50:22 -0700 | [diff] [blame] | 161 | // TODO: Fix premultiplied alpha computations for color matrix |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 162 | " fragColor *= colorMatrix;\n" |
Romain Guy | db1938e | 2010-08-02 18:50:22 -0700 | [diff] [blame] | 163 | " fragColor += colorMatrixVector;\n" |
| 164 | " fragColor.rgb *= fragColor.a;\n", |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 165 | // Lighting |
Romain Guy | db1938e | 2010-08-02 18:50:22 -0700 | [diff] [blame] | 166 | " float lightingAlpha = fragColor.a;\n" |
| 167 | " fragColor = min(fragColor * lightingMul + (lightingAdd * lightingAlpha), lightingAlpha);\n" |
| 168 | " fragColor.a = lightingAlpha;\n", |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 169 | // PorterDuff |
| 170 | " fragColor = blendColors(colorBlend, fragColor);\n" |
| 171 | }; |
| 172 | const char* gFS_Footer = |
| 173 | "}\n\n"; |
| 174 | |
| 175 | /////////////////////////////////////////////////////////////////////////////// |
| 176 | // PorterDuff snippets |
| 177 | /////////////////////////////////////////////////////////////////////////////// |
| 178 | |
Romain Guy | 48daa54 | 2010-08-10 19:21:34 -0700 | [diff] [blame] | 179 | const char* gBlendOps[18] = { |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 180 | // Clear |
| 181 | "return vec4(0.0, 0.0, 0.0, 0.0);\n", |
| 182 | // Src |
| 183 | "return src;\n", |
| 184 | // Dst |
| 185 | "return dst;\n", |
| 186 | // SrcOver |
Romain Guy | 06f96e2 | 2010-07-30 19:18:16 -0700 | [diff] [blame] | 187 | "return src + dst * (1.0 - src.a);\n", |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 188 | // DstOver |
Romain Guy | 06f96e2 | 2010-07-30 19:18:16 -0700 | [diff] [blame] | 189 | "return dst + src * (1.0 - dst.a);\n", |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 190 | // SrcIn |
Romain Guy | 06f96e2 | 2010-07-30 19:18:16 -0700 | [diff] [blame] | 191 | "return src * dst.a;\n", |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 192 | // DstIn |
Romain Guy | 06f96e2 | 2010-07-30 19:18:16 -0700 | [diff] [blame] | 193 | "return dst * src.a;\n", |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 194 | // SrcOut |
Romain Guy | 06f96e2 | 2010-07-30 19:18:16 -0700 | [diff] [blame] | 195 | "return src * (1.0 - dst.a);\n", |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 196 | // DstOut |
Romain Guy | 06f96e2 | 2010-07-30 19:18:16 -0700 | [diff] [blame] | 197 | "return dst * (1.0 - src.a);\n", |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 198 | // SrcAtop |
| 199 | "return vec4(src.rgb * dst.a + (1.0 - src.a) * dst.rgb, dst.a);\n", |
| 200 | // DstAtop |
| 201 | "return vec4(dst.rgb * src.a + (1.0 - dst.a) * src.rgb, src.a);\n", |
| 202 | // Xor |
Romain Guy | 48daa54 | 2010-08-10 19:21:34 -0700 | [diff] [blame] | 203 | "return vec4(src.rgb * (1.0 - dst.a) + (1.0 - src.a) * dst.rgb, " |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 204 | "src.a + dst.a - 2.0 * src.a * dst.a);\n", |
Romain Guy | 48daa54 | 2010-08-10 19:21:34 -0700 | [diff] [blame] | 205 | // Add |
| 206 | "return min(src + dst, 1.0);\n", |
| 207 | // Multiply |
| 208 | "return src * dst;\n", |
| 209 | // Screen |
| 210 | "return src + dst - src * dst;\n", |
| 211 | // Overlay |
| 212 | "return clamp(vec4(mix(" |
| 213 | "2.0 * src.rgb * dst.rgb + src.rgb * (1.0 - dst.a) + dst.rgb * (1.0 - src.a), " |
| 214 | "src.a * dst.a - 2.0 * (dst.a - dst.rgb) * (src.a - src.rgb) + src.rgb * (1.0 - dst.a) + dst.rgb * (1.0 - src.a), " |
| 215 | "step(dst.a, 2.0 * dst.rgb)), " |
| 216 | "src.a + dst.a - src.a * dst.a), 0.0, 1.0);\n", |
| 217 | // Darken |
| 218 | "return vec4(src.rgb * (1.0 - dst.a) + (1.0 - src.a) * dst.rgb + " |
| 219 | "min(src.rgb * dst.a, dst.rgb * src.a), src.a + dst.a - src.a * dst.a);\n", |
| 220 | // Lighten |
| 221 | "return vec4(src.rgb * (1.0 - dst.a) + (1.0 - src.a) * dst.rgb + " |
| 222 | "max(src.rgb * dst.a, dst.rgb * src.a), src.a + dst.a - src.a * dst.a);\n", |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 223 | }; |
| 224 | |
| 225 | /////////////////////////////////////////////////////////////////////////////// |
| 226 | // Constructors/destructors |
| 227 | /////////////////////////////////////////////////////////////////////////////// |
| 228 | |
| 229 | ProgramCache::ProgramCache() { |
| 230 | } |
| 231 | |
| 232 | ProgramCache::~ProgramCache() { |
| 233 | clear(); |
| 234 | } |
| 235 | |
| 236 | /////////////////////////////////////////////////////////////////////////////// |
| 237 | // Cache management |
| 238 | /////////////////////////////////////////////////////////////////////////////// |
| 239 | |
| 240 | void ProgramCache::clear() { |
| 241 | size_t count = mCache.size(); |
| 242 | for (size_t i = 0; i < count; i++) { |
| 243 | delete mCache.valueAt(i); |
| 244 | } |
| 245 | mCache.clear(); |
| 246 | } |
| 247 | |
| 248 | Program* ProgramCache::get(const ProgramDescription& description) { |
| 249 | programid key = description.key(); |
| 250 | ssize_t index = mCache.indexOfKey(key); |
| 251 | Program* program = NULL; |
| 252 | if (index < 0) { |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 253 | description.log("Could not find program"); |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 254 | program = generateProgram(description, key); |
| 255 | mCache.add(key, program); |
| 256 | } else { |
| 257 | program = mCache.valueAt(index); |
| 258 | } |
| 259 | return program; |
| 260 | } |
| 261 | |
| 262 | /////////////////////////////////////////////////////////////////////////////// |
| 263 | // Program generation |
| 264 | /////////////////////////////////////////////////////////////////////////////// |
| 265 | |
| 266 | Program* ProgramCache::generateProgram(const ProgramDescription& description, programid key) { |
| 267 | String8 vertexShader = generateVertexShader(description); |
| 268 | String8 fragmentShader = generateFragmentShader(description); |
| 269 | |
| 270 | Program* program = new Program(vertexShader.string(), fragmentShader.string()); |
| 271 | return program; |
| 272 | } |
| 273 | |
| 274 | String8 ProgramCache::generateVertexShader(const ProgramDescription& description) { |
| 275 | // Add attributes |
| 276 | String8 shader(gVS_Header_Attributes); |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 277 | if (description.hasTexture) { |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 278 | shader.append(gVS_Header_Attributes_TexCoords); |
| 279 | } |
| 280 | // Uniforms |
| 281 | shader.append(gVS_Header_Uniforms); |
| 282 | if (description.hasGradient) { |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 283 | shader.append(gVS_Header_Uniforms_HasGradient[description.gradientType]); |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 284 | } |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 285 | if (description.hasBitmap) { |
| 286 | shader.append(gVS_Header_Uniforms_HasBitmap); |
| 287 | } |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 288 | // Varyings |
| 289 | if (description.hasTexture) { |
| 290 | shader.append(gVS_Header_Varyings_HasTexture); |
| 291 | } |
| 292 | if (description.hasGradient) { |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 293 | shader.append(gVS_Header_Varyings_HasGradient[description.gradientType]); |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 294 | } |
| 295 | if (description.hasBitmap) { |
| 296 | shader.append(gVS_Header_Varyings_HasBitmap); |
| 297 | } |
| 298 | |
| 299 | // Begin the shader |
| 300 | shader.append(gVS_Main); { |
| 301 | if (description.hasTexture) { |
| 302 | shader.append(gVS_Main_OutTexCoords); |
| 303 | } |
| 304 | if (description.hasGradient) { |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 305 | shader.append(gVS_Main_OutGradient[description.gradientType]); |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 306 | } |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 307 | if (description.hasBitmap) { |
| 308 | shader.append(gVS_Main_OutBitmapTexCoords); |
| 309 | } |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 310 | // Output transformed position |
| 311 | shader.append(gVS_Main_Position); |
| 312 | } |
| 313 | // End the shader |
| 314 | shader.append(gVS_Footer); |
| 315 | |
| 316 | PROGRAM_LOGD("*** Generated vertex shader:\n\n%s", shader.string()); |
| 317 | |
| 318 | return shader; |
| 319 | } |
| 320 | |
| 321 | String8 ProgramCache::generateFragmentShader(const ProgramDescription& description) { |
| 322 | // Set the default precision |
Romain Guy | a5aed0d | 2010-09-09 14:42:43 -0700 | [diff] [blame] | 323 | String8 shader; |
| 324 | |
| 325 | bool blendFramebuffer = description.framebufferMode >= SkXfermode::kPlus_Mode; |
| 326 | if (blendFramebuffer) { |
| 327 | shader.append(gFS_Header_Extension_FramebufferFetch); |
| 328 | } |
| 329 | |
| 330 | shader.append(gFS_Header); |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 331 | |
| 332 | // Varyings |
| 333 | if (description.hasTexture) { |
| 334 | shader.append(gVS_Header_Varyings_HasTexture); |
| 335 | } |
| 336 | if (description.hasGradient) { |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 337 | shader.append(gVS_Header_Varyings_HasGradient[description.gradientType]); |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 338 | } |
| 339 | if (description.hasBitmap) { |
| 340 | shader.append(gVS_Header_Varyings_HasBitmap); |
| 341 | } |
| 342 | |
| 343 | |
| 344 | // Uniforms |
| 345 | shader.append(gFS_Uniforms_Color); |
| 346 | if (description.hasTexture) { |
| 347 | shader.append(gFS_Uniforms_TextureSampler); |
| 348 | } |
| 349 | if (description.hasGradient) { |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 350 | shader.append(gFS_Uniforms_GradientSampler[description.gradientType]); |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 351 | } |
| 352 | if (description.hasBitmap) { |
| 353 | shader.append(gFS_Uniforms_BitmapSampler); |
| 354 | } |
| 355 | shader.append(gFS_Uniforms_ColorOp[description.colorOp]); |
| 356 | |
| 357 | // Generate required functions |
| 358 | if (description.hasGradient && description.hasBitmap) { |
Romain Guy | 48daa54 | 2010-08-10 19:21:34 -0700 | [diff] [blame] | 359 | generateBlend(shader, "blendShaders", description.shadersMode); |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 360 | } |
| 361 | if (description.colorOp == ProgramDescription::kColorBlend) { |
Romain Guy | 48daa54 | 2010-08-10 19:21:34 -0700 | [diff] [blame] | 362 | generateBlend(shader, "blendColors", description.colorMode); |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 363 | } |
Romain Guy | a5aed0d | 2010-09-09 14:42:43 -0700 | [diff] [blame] | 364 | if (blendFramebuffer) { |
| 365 | generateBlend(shader, "blendFramebuffer", description.framebufferMode); |
| 366 | } |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 367 | if (description.isBitmapNpot) { |
| 368 | generateTextureWrap(shader, description.bitmapWrapS, description.bitmapWrapT); |
| 369 | } |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 370 | |
| 371 | // Begin the shader |
| 372 | shader.append(gFS_Main); { |
| 373 | // Stores the result in fragColor directly |
| 374 | if (description.hasTexture) { |
| 375 | if (description.hasAlpha8Texture) { |
| 376 | shader.append(gFS_Main_FetchA8Texture); |
| 377 | } else { |
| 378 | shader.append(gFS_Main_FetchTexture); |
| 379 | } |
| 380 | } else { |
| 381 | shader.append(gFS_Main_FetchColor); |
| 382 | } |
| 383 | if (description.hasGradient) { |
Romain Guy | ee916f1 | 2010-09-20 17:53:08 -0700 | [diff] [blame^] | 384 | shader.append(gFS_Main_FetchGradient[description.gradientType]); |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 385 | } |
| 386 | if (description.hasBitmap) { |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 387 | if (!description.isBitmapNpot) { |
| 388 | shader.append(gFS_Main_FetchBitmap); |
| 389 | } else { |
| 390 | shader.append(gFS_Main_FetchBitmapNpot); |
| 391 | } |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 392 | } |
| 393 | // Case when we have two shaders set |
| 394 | if (description.hasGradient && description.hasBitmap) { |
| 395 | if (description.isBitmapFirst) { |
| 396 | shader.append(gFS_Main_BlendShadersBG); |
| 397 | } else { |
| 398 | shader.append(gFS_Main_BlendShadersGB); |
| 399 | } |
| 400 | shader.append(gFS_Main_BlendShaders_Modulate); |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 401 | } else { |
| 402 | if (description.hasGradient) { |
| 403 | shader.append(gFS_Main_GradientShader_Modulate); |
| 404 | } else if (description.hasBitmap) { |
| 405 | shader.append(gFS_Main_BitmapShader_Modulate); |
| 406 | } |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 407 | } |
| 408 | // Apply the color op if needed |
| 409 | shader.append(gFS_Main_ApplyColorOp[description.colorOp]); |
| 410 | // Output the fragment |
Romain Guy | a5aed0d | 2010-09-09 14:42:43 -0700 | [diff] [blame] | 411 | if (!blendFramebuffer) { |
| 412 | shader.append(gFS_Main_FragColor); |
| 413 | } else { |
Romain Guy | f607bdc | 2010-09-10 19:20:06 -0700 | [diff] [blame] | 414 | shader.append(!description.swapSrcDst ? |
| 415 | gFS_Main_FragColor_Blend : gFS_Main_FragColor_Blend_Swap); |
Romain Guy | a5aed0d | 2010-09-09 14:42:43 -0700 | [diff] [blame] | 416 | } |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 417 | } |
| 418 | // End the shader |
| 419 | shader.append(gFS_Footer); |
| 420 | |
Romain Guy | db1938e | 2010-08-02 18:50:22 -0700 | [diff] [blame] | 421 | if (DEBUG_PROGRAM_CACHE) { |
| 422 | PROGRAM_LOGD("*** Generated fragment shader:\n\n"); |
| 423 | printLongString(shader); |
| 424 | } |
| 425 | |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 426 | return shader; |
| 427 | } |
| 428 | |
Romain Guy | 48daa54 | 2010-08-10 19:21:34 -0700 | [diff] [blame] | 429 | void ProgramCache::generateBlend(String8& shader, const char* name, SkXfermode::Mode mode) { |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 430 | shader.append("\nvec4 "); |
| 431 | shader.append(name); |
| 432 | shader.append("(vec4 src, vec4 dst) {\n"); |
| 433 | shader.append(" "); |
Romain Guy | 48daa54 | 2010-08-10 19:21:34 -0700 | [diff] [blame] | 434 | shader.append(gBlendOps[mode]); |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 435 | shader.append("}\n"); |
| 436 | } |
| 437 | |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 438 | void ProgramCache::generateTextureWrap(String8& shader, GLenum wrapS, GLenum wrapT) { |
| 439 | shader.append("\nvec2 wrap(vec2 texCoords) {\n"); |
| 440 | if (wrapS == GL_MIRRORED_REPEAT) { |
| 441 | shader.append(" float xMod2 = mod(texCoords.x, 2.0);\n"); |
| 442 | shader.append(" if (xMod2 > 1.0) xMod2 = 2.0 - xMod2;\n"); |
| 443 | } |
| 444 | if (wrapT == GL_MIRRORED_REPEAT) { |
| 445 | shader.append(" float yMod2 = mod(texCoords.y, 2.0);\n"); |
| 446 | shader.append(" if (yMod2 > 1.0) yMod2 = 2.0 - yMod2;\n"); |
| 447 | } |
| 448 | shader.append(" return vec2("); |
| 449 | switch (wrapS) { |
Romain Guy | 61c8c9c | 2010-08-09 20:48:09 -0700 | [diff] [blame] | 450 | case GL_CLAMP_TO_EDGE: |
| 451 | shader.append("texCoords.x"); |
| 452 | break; |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 453 | case GL_REPEAT: |
| 454 | shader.append("mod(texCoords.x, 1.0)"); |
| 455 | break; |
| 456 | case GL_MIRRORED_REPEAT: |
| 457 | shader.append("xMod2"); |
| 458 | break; |
| 459 | } |
| 460 | shader.append(", "); |
| 461 | switch (wrapT) { |
Romain Guy | 61c8c9c | 2010-08-09 20:48:09 -0700 | [diff] [blame] | 462 | case GL_CLAMP_TO_EDGE: |
| 463 | shader.append("texCoords.y"); |
| 464 | break; |
Romain Guy | 889f8d1 | 2010-07-29 14:37:42 -0700 | [diff] [blame] | 465 | case GL_REPEAT: |
| 466 | shader.append("mod(texCoords.y, 1.0)"); |
| 467 | break; |
| 468 | case GL_MIRRORED_REPEAT: |
| 469 | shader.append("yMod2"); |
| 470 | break; |
| 471 | } |
| 472 | shader.append(");\n"); |
| 473 | shader.append("}\n"); |
| 474 | } |
| 475 | |
Romain Guy | db1938e | 2010-08-02 18:50:22 -0700 | [diff] [blame] | 476 | void ProgramCache::printLongString(const String8& shader) const { |
| 477 | ssize_t index = 0; |
| 478 | ssize_t lastIndex = 0; |
| 479 | const char* str = shader.string(); |
| 480 | while ((index = shader.find("\n", index)) > -1) { |
| 481 | String8 line(str, index - lastIndex); |
| 482 | if (line.length() == 0) line.append("\n"); |
| 483 | PROGRAM_LOGD("%s", line.string()); |
| 484 | index++; |
| 485 | str += (index - lastIndex); |
| 486 | lastIndex = index; |
| 487 | } |
| 488 | } |
| 489 | |
Romain Guy | ac670c0 | 2010-07-27 17:39:27 -0700 | [diff] [blame] | 490 | }; // namespace uirenderer |
| 491 | }; // namespace android |