blob: 60e9188252a08a76cf6a43ff6f35763213cdfa29 [file] [log] [blame]
tomhudson@google.comf9ad8862012-05-11 20:38:48 +00001/*
2 * Copyright 2012 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
8#include "gl/GrGLShaderBuilder.h"
tomhudson@google.com52598142012-05-24 17:44:30 +00009#include "gl/GrGLProgram.h"
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +000010#include "gl/GrGLUniformHandle.h"
bsalomon@google.comc7818882013-03-20 19:19:53 +000011#include "GrDrawEffect.h"
twiz@google.coma5e65ec2012-08-02 15:15:16 +000012#include "GrTexture.h"
tomhudson@google.comf9ad8862012-05-11 20:38:48 +000013
tomhudson@google.comf9ad8862012-05-11 20:38:48 +000014// number of each input/output type in a single allocation block
bsalomon@google.comeb715c82012-07-11 15:03:31 +000015static const int kVarsPerBlock = 8;
tomhudson@google.comf9ad8862012-05-11 20:38:48 +000016
17// except FS outputs where we expect 2 at most.
bsalomon@google.comeb715c82012-07-11 15:03:31 +000018static const int kMaxFSOutputs = 2;
tomhudson@google.comf9ad8862012-05-11 20:38:48 +000019
bsalomon@google.comd7727ce2012-07-12 16:40:03 +000020// ES2 FS only guarantees mediump and lowp support
21static const GrGLShaderVar::Precision kDefaultFragmentPrecision = GrGLShaderVar::kMedium_Precision;
22
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +000023typedef GrGLUniformManager::UniformHandle UniformHandle;
24///////////////////////////////////////////////////////////////////////////////
25
bsalomon@google.com34bcb9f2012-08-28 18:20:18 +000026namespace {
twiz@google.coma5e65ec2012-08-02 15:15:16 +000027
bsalomon@google.com2b1b8c02013-02-28 22:06:02 +000028inline const char* sample_function_name(GrSLType type, GrGLSLGeneration glslGen) {
bsalomon@google.com34bcb9f2012-08-28 18:20:18 +000029 if (kVec2f_GrSLType == type) {
bsalomon@google.com2b1b8c02013-02-28 22:06:02 +000030 return glslGen >= k130_GrGLSLGeneration ? "texture" : "texture2D";
bsalomon@google.com34bcb9f2012-08-28 18:20:18 +000031 } else {
32 GrAssert(kVec3f_GrSLType == type);
bsalomon@google.com2b1b8c02013-02-28 22:06:02 +000033 return glslGen >= k130_GrGLSLGeneration ? "textureProj" : "texture2DProj";
bsalomon@google.com34bcb9f2012-08-28 18:20:18 +000034 }
twiz@google.coma5e65ec2012-08-02 15:15:16 +000035}
36
bsalomon@google.com6d003d12012-09-11 15:45:20 +000037/**
bsalomon@google.com26e18b52013-03-29 19:22:36 +000038 * Do we need to either map r,g,b->a or a->r. configComponentMask indicates which channels are
39 * present in the texture's config. swizzleComponentMask indicates the channels present in the
40 * shader swizzle.
bsalomon@google.com6d003d12012-09-11 15:45:20 +000041 */
42inline bool swizzle_requires_alpha_remapping(const GrGLCaps& caps,
bsalomon@google.com26e18b52013-03-29 19:22:36 +000043 uint32_t configComponentMask,
44 uint32_t swizzleComponentMask) {
45 if (caps.textureSwizzleSupport()) {
46 // Any remapping is handled using texture swizzling not shader modifications.
47 return false;
48 }
49 // check if the texture is alpha-only
50 if (kA_GrColorComponentFlag == configComponentMask) {
51 if (caps.textureRedSupport() && (kA_GrColorComponentFlag & swizzleComponentMask)) {
52 // we must map the swizzle 'a's to 'r'.
bsalomon@google.com6d003d12012-09-11 15:45:20 +000053 return true;
54 }
bsalomon@google.com26e18b52013-03-29 19:22:36 +000055 if (kRGB_GrColorComponentFlags & swizzleComponentMask) {
56 // The 'r', 'g', and/or 'b's must be mapped to 'a' according to our semantics that
57 // alpha-only textures smear alpha across all four channels when read.
bsalomon@google.com6d003d12012-09-11 15:45:20 +000058 return true;
twiz@google.coma5e65ec2012-08-02 15:15:16 +000059 }
60 }
bsalomon@google.com6d003d12012-09-11 15:45:20 +000061 return false;
62}
twiz@google.coma5e65ec2012-08-02 15:15:16 +000063
bsalomon@google.com6d003d12012-09-11 15:45:20 +000064void append_swizzle(SkString* outAppend,
bsalomon@google.comb8eb2e82013-03-28 13:46:42 +000065 const GrGLShaderBuilder::TextureSampler& texSampler,
bsalomon@google.com6d003d12012-09-11 15:45:20 +000066 const GrGLCaps& caps) {
bsalomon@google.comb8eb2e82013-03-28 13:46:42 +000067 const char* swizzle = texSampler.swizzle();
bsalomon@google.com6d003d12012-09-11 15:45:20 +000068 char mangledSwizzle[5];
69
70 // The swizzling occurs using texture params instead of shader-mangling if ARB_texture_swizzle
71 // is available.
bsalomon@google.comb8eb2e82013-03-28 13:46:42 +000072 if (!caps.textureSwizzleSupport() &&
73 (kA_GrColorComponentFlag == texSampler.configComponentMask())) {
bsalomon@google.com6d003d12012-09-11 15:45:20 +000074 char alphaChar = caps.textureRedSupport() ? 'r' : 'a';
75 int i;
76 for (i = 0; '\0' != swizzle[i]; ++i) {
77 mangledSwizzle[i] = alphaChar;
78 }
79 mangledSwizzle[i] ='\0';
80 swizzle = mangledSwizzle;
81 }
bsalomon@google.com73d5b2f2012-10-04 13:26:32 +000082 // For shader prettiness we omit the swizzle rather than appending ".rgba".
83 if (memcmp(swizzle, "rgba", 4)) {
84 outAppend->appendf(".%s", swizzle);
85 }
twiz@google.coma5e65ec2012-08-02 15:15:16 +000086}
87
bsalomon@google.com34bcb9f2012-08-28 18:20:18 +000088}
89
bsalomon@google.com26e18b52013-03-29 19:22:36 +000090static const char kDstColorName[] = "_dstColor";
twiz@google.coma5e65ec2012-08-02 15:15:16 +000091
bsalomon@google.com26e18b52013-03-29 19:22:36 +000092///////////////////////////////////////////////////////////////////////////////
tomhudson@google.com9c639a42012-05-14 19:58:06 +000093
skia.committer@gmail.com631cdcb2013-03-01 12:12:55 +000094GrGLShaderBuilder::GrGLShaderBuilder(const GrGLContextInfo& ctxInfo,
bsalomon@google.comc7818882013-03-20 19:19:53 +000095 GrGLUniformManager& uniformManager,
bsalomon@google.com26e18b52013-03-29 19:22:36 +000096 const GrGLProgramDesc& desc)
bsalomon@google.com032b2212012-07-16 13:36:18 +000097 : fUniforms(kVarsPerBlock)
bsalomon@google.comeb715c82012-07-11 15:03:31 +000098 , fVSAttrs(kVarsPerBlock)
99 , fVSOutputs(kVarsPerBlock)
100 , fGSInputs(kVarsPerBlock)
101 , fGSOutputs(kVarsPerBlock)
102 , fFSInputs(kVarsPerBlock)
bsalomon@google.comeb715c82012-07-11 15:03:31 +0000103 , fFSOutputs(kMaxFSOutputs)
robertphillips@google.com6177e692013-02-28 20:16:25 +0000104 , fCtxInfo(ctxInfo)
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000105 , fUniformManager(uniformManager)
bsalomon@google.com08283af2012-10-26 13:01:20 +0000106 , fCurrentStageIdx(kNonStageIdx)
bsalomon@google.com42eff162013-04-02 12:50:49 +0000107 , fFSFeaturesAddedMask(0)
bsalomon@google.com26e18b52013-03-29 19:22:36 +0000108#if GR_GL_EXPERIMENTAL_GS
109 , fUsesGS(desc.fExperimentalGS)
110#else
111 , fUsesGS(false)
112#endif
bsalomon@google.com706f6682012-10-23 14:53:55 +0000113 , fSetupFragPosition(false)
bsalomon@google.com26e18b52013-03-29 19:22:36 +0000114 , fRTHeightUniform(GrGLUniformManager::kInvalidUniformHandle)
115 , fDstCopyTopLeftUniform (GrGLUniformManager::kInvalidUniformHandle)
116 , fDstCopyScaleUniform (GrGLUniformManager::kInvalidUniformHandle) {
skia.committer@gmail.come862d162012-10-31 02:01:18 +0000117
bsalomon@google.com17504f52012-10-30 12:34:25 +0000118 fPositionVar = &fVSAttrs.push_back();
119 fPositionVar->set(kVec2f_GrSLType, GrGLShaderVar::kAttribute_TypeModifier, "aPosition");
jvanverth@google.com054ae992013-04-01 20:06:51 +0000120 if (-1 != desc.fLocalCoordAttributeIndex) {
bsalomon@google.comc7818882013-03-20 19:19:53 +0000121 fLocalCoordsVar = &fVSAttrs.push_back();
122 fLocalCoordsVar->set(kVec2f_GrSLType,
123 GrGLShaderVar::kAttribute_TypeModifier,
124 "aLocalCoords");
125 } else {
126 fLocalCoordsVar = fPositionVar;
127 }
bsalomon@google.com26e18b52013-03-29 19:22:36 +0000128 if (kNoDstRead_DstReadKey != desc.fDstRead) {
129 bool topDown = SkToBool(kTopLeftOrigin_DstReadKeyBit & desc.fDstRead);
130 const char* dstCopyTopLeftName;
131 const char* dstCopyCoordScaleName;
132 uint32_t configMask;
133 if (SkToBool(kUseAlphaConfig_DstReadKeyBit & desc.fDstRead)) {
134 configMask = kA_GrColorComponentFlag;
135 } else {
136 configMask = kRGBA_GrColorComponentFlags;
137 }
138 fDstCopySampler.init(this, configMask, "rgba", 0);
139
140 fDstCopyTopLeftUniform = this->addUniform(kFragment_ShaderType,
141 kVec2f_GrSLType,
142 "DstCopyUpperLeft",
143 &dstCopyTopLeftName);
144 fDstCopyScaleUniform = this->addUniform(kFragment_ShaderType,
145 kVec2f_GrSLType,
146 "DstCopyCoordScale",
147 &dstCopyCoordScaleName);
148 const char* fragPos = this->fragmentPosition();
149 this->fsCodeAppend("\t// Read color from copy of the destination.\n");
150 this->fsCodeAppendf("\tvec2 _dstTexCoord = (%s.xy - %s) * %s;\n",
151 fragPos, dstCopyTopLeftName, dstCopyCoordScaleName);
152 if (!topDown) {
153 this->fsCodeAppend("\t_dstTexCoord.y = 1.0 - _dstTexCoord.y;\n");
154 }
155 this->fsCodeAppendf("\tvec4 %s = ", kDstColorName);
156 this->appendTextureLookup(kFragment_ShaderType, fDstCopySampler, "_dstTexCoord");
157 this->fsCodeAppend(";\n\n");
158 }
159}
160
bsalomon@google.com42eff162013-04-02 12:50:49 +0000161bool GrGLShaderBuilder::enableFeature(GLSLFeature feature) {
162 switch (feature) {
163 case kStandardDerivatives_GLSLFeature:
164 if (!fCtxInfo.caps()->shaderDerivativeSupport()) {
165 return false;
166 }
167 if (kES2_GrGLBinding == fCtxInfo.binding()) {
168 this->addFSFeature(1 << kStandardDerivatives_GLSLFeature,
169 "GL_OES_standard_derivatives");
170 }
171 return true;
172 default:
173 GrCrash("Unexpected GLSLFeature requested.");
174 return false;
175 }
176}
177
178bool GrGLShaderBuilder::enablePrivateFeature(GLSLPrivateFeature feature) {
179 switch (feature) {
180 case kFragCoordConventions_GLSLPrivateFeature:
181 if (!fCtxInfo.caps()->fragCoordConventionsSupport()) {
182 return false;
183 }
184 if (fCtxInfo.glslGeneration() < k150_GrGLSLGeneration) {
185 this->addFSFeature(1 << kFragCoordConventions_GLSLPrivateFeature,
186 "GL_ARB_fragment_coord_conventions");
187 }
188 return true;
189 default:
190 GrCrash("Unexpected GLSLPrivateFeature requested.");
191 return false;
192 }
193}
194
195void GrGLShaderBuilder::addFSFeature(uint32_t featureBit, const char* extensionName) {
196 if (!(featureBit & fFSFeaturesAddedMask)) {
197 fFSExtensions.appendf("#extension %s: require\n", extensionName);
198 fFSFeaturesAddedMask |= featureBit;
199 }
200}
201
bsalomon@google.com26e18b52013-03-29 19:22:36 +0000202const char* GrGLShaderBuilder::dstColor() const {
203 if (fDstCopySampler.isInitialized()) {
204 return kDstColorName;
205 } else {
206 return NULL;
207 }
tomhudson@google.comf9ad8862012-05-11 20:38:48 +0000208}
209
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000210void GrGLShaderBuilder::codeAppendf(ShaderType type, const char format[], va_list args) {
sugoi@google.com9c55f802013-03-07 20:52:59 +0000211 SkString* string = NULL;
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000212 switch (type) {
213 case kVertex_ShaderType:
214 string = &fVSCode;
215 break;
216 case kGeometry_ShaderType:
217 string = &fGSCode;
218 break;
219 case kFragment_ShaderType:
220 string = &fFSCode;
221 break;
222 default:
223 GrCrash("Invalid shader type");
224 }
225 string->appendf(format, args);
226}
227
228void GrGLShaderBuilder::codeAppend(ShaderType type, const char* str) {
sugoi@google.com9c55f802013-03-07 20:52:59 +0000229 SkString* string = NULL;
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000230 switch (type) {
231 case kVertex_ShaderType:
232 string = &fVSCode;
233 break;
234 case kGeometry_ShaderType:
235 string = &fGSCode;
236 break;
237 case kFragment_ShaderType:
238 string = &fFSCode;
239 break;
240 default:
241 GrCrash("Invalid shader type");
242 }
243 string->append(str);
244}
245
bsalomon@google.com868a8e72012-08-30 19:11:34 +0000246void GrGLShaderBuilder::appendTextureLookup(SkString* out,
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000247 const GrGLShaderBuilder::TextureSampler& sampler,
bsalomon@google.com868a8e72012-08-30 19:11:34 +0000248 const char* coordName,
249 GrSLType varyingType) const {
bsalomon@google.comdbe49f72012-11-05 16:36:02 +0000250 GrAssert(NULL != coordName);
bsalomon@google.com2d8edaf2012-09-07 14:47:31 +0000251
bsalomon@google.com6d003d12012-09-11 15:45:20 +0000252 out->appendf("%s(%s, %s)",
bsalomon@google.com2b1b8c02013-02-28 22:06:02 +0000253 sample_function_name(varyingType, fCtxInfo.glslGeneration()),
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000254 this->getUniformCStr(sampler.fSamplerUniform),
bsalomon@google.com6d003d12012-09-11 15:45:20 +0000255 coordName);
bsalomon@google.comb8eb2e82013-03-28 13:46:42 +0000256 append_swizzle(out, sampler, *fCtxInfo.caps());
tomhudson@google.com52598142012-05-24 17:44:30 +0000257}
258
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000259void GrGLShaderBuilder::appendTextureLookup(ShaderType type,
260 const GrGLShaderBuilder::TextureSampler& sampler,
261 const char* coordName,
262 GrSLType varyingType) {
263 GrAssert(kFragment_ShaderType == type);
264 this->appendTextureLookup(&fFSCode, sampler, coordName, varyingType);
265}
266
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000267void GrGLShaderBuilder::appendTextureLookupAndModulate(
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000268 ShaderType type,
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000269 const char* modulation,
270 const GrGLShaderBuilder::TextureSampler& sampler,
271 const char* coordName,
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000272 GrSLType varyingType) {
273 GrAssert(kFragment_ShaderType == type);
bsalomon@google.com868a8e72012-08-30 19:11:34 +0000274 SkString lookup;
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000275 this->appendTextureLookup(&lookup, sampler, coordName, varyingType);
bsalomon@google.com018f1792013-04-18 19:36:09 +0000276 GrGLSLModulatef<4>(&fFSCode, modulation, lookup.c_str());
tomhudson@google.com52598142012-05-24 17:44:30 +0000277}
278
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000279GrBackendEffectFactory::EffectKey GrGLShaderBuilder::KeyForTextureAccess(
280 const GrTextureAccess& access,
281 const GrGLCaps& caps) {
bsalomon@google.com26e18b52013-03-29 19:22:36 +0000282 uint32_t configComponentMask = GrPixelConfigComponentMask(access.getTexture()->config());
283 if (swizzle_requires_alpha_remapping(caps, configComponentMask, access.swizzleMask())) {
284 return 1;
285 } else {
286 return 0;
twiz@google.coma5e65ec2012-08-02 15:15:16 +0000287 }
bsalomon@google.com26e18b52013-03-29 19:22:36 +0000288}
289
290GrGLShaderBuilder::DstReadKey GrGLShaderBuilder::KeyForDstRead(const GrTexture* dstCopy,
291 const GrGLCaps& caps) {
292 uint32_t key = kYesDstRead_DstReadKeyBit;
293 if (!caps.textureSwizzleSupport() && GrPixelConfigIsAlphaOnly(dstCopy->config())) {
294 // The fact that the config is alpha-only must be considered when generating code.
295 key |= kUseAlphaConfig_DstReadKeyBit;
296 }
297 if (kTopLeft_GrSurfaceOrigin == dstCopy->origin()) {
298 key |= kTopLeftOrigin_DstReadKeyBit;
299 }
300 GrAssert(static_cast<DstReadKey>(key) == key);
301 return static_cast<DstReadKey>(key);
twiz@google.coma5e65ec2012-08-02 15:15:16 +0000302}
303
bsalomon@google.com6d003d12012-09-11 15:45:20 +0000304const GrGLenum* GrGLShaderBuilder::GetTexParamSwizzle(GrPixelConfig config, const GrGLCaps& caps) {
305 if (caps.textureSwizzleSupport() && GrPixelConfigIsAlphaOnly(config)) {
306 if (caps.textureRedSupport()) {
307 static const GrGLenum gRedSmear[] = { GR_GL_RED, GR_GL_RED, GR_GL_RED, GR_GL_RED };
308 return gRedSmear;
309 } else {
310 static const GrGLenum gAlphaSmear[] = { GR_GL_ALPHA, GR_GL_ALPHA,
311 GR_GL_ALPHA, GR_GL_ALPHA };
312 return gAlphaSmear;
313 }
314 } else {
315 static const GrGLenum gStraight[] = { GR_GL_RED, GR_GL_GREEN, GR_GL_BLUE, GR_GL_ALPHA };
316 return gStraight;
317 }
318}
319
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000320GrGLUniformManager::UniformHandle GrGLShaderBuilder::addUniformArray(uint32_t visibility,
321 GrSLType type,
322 const char* name,
323 int count,
324 const char** outName) {
tomhudson@google.com242ed6f2012-05-30 17:38:57 +0000325 GrAssert(name && strlen(name));
humper@google.com0e515772013-01-07 19:54:40 +0000326 SkDEBUGCODE(static const uint32_t kVisibilityMask = kVertex_ShaderType | kFragment_ShaderType);
bsalomon@google.comeb715c82012-07-11 15:03:31 +0000327 GrAssert(0 == (~kVisibilityMask & visibility));
328 GrAssert(0 != visibility);
tomhudson@google.com242ed6f2012-05-30 17:38:57 +0000329
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000330 BuilderUniform& uni = fUniforms.push_back();
bsalomon@google.com032b2212012-07-16 13:36:18 +0000331 UniformHandle h = index_to_handle(fUniforms.count() - 1);
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000332 GR_DEBUGCODE(UniformHandle h2 =)
333 fUniformManager.appendUniform(type, count);
334 // We expect the uniform manager to initially have no uniforms and that all uniforms are added
335 // by this function. Therefore, the handles should match.
336 GrAssert(h2 == h);
bsalomon@google.com032b2212012-07-16 13:36:18 +0000337 uni.fVariable.setType(type);
338 uni.fVariable.setTypeModifier(GrGLShaderVar::kUniform_TypeModifier);
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000339 SkString* uniName = uni.fVariable.accessName();
bsalomon@google.com08283af2012-10-26 13:01:20 +0000340 if (kNonStageIdx == fCurrentStageIdx) {
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000341 uniName->printf("u%s", name);
342 } else {
bsalomon@google.com08283af2012-10-26 13:01:20 +0000343 uniName->printf("u%s%d", name, fCurrentStageIdx);
tomhudson@google.com242ed6f2012-05-30 17:38:57 +0000344 }
bsalomon@google.com032b2212012-07-16 13:36:18 +0000345 uni.fVariable.setArrayCount(count);
346 uni.fVisibility = visibility;
tomhudson@google.com242ed6f2012-05-30 17:38:57 +0000347
bsalomon@google.com032b2212012-07-16 13:36:18 +0000348 // If it is visible in both the VS and FS, the precision must match.
349 // We declare a default FS precision, but not a default VS. So set the var
350 // to use the default FS precision.
bsalomon@google.comeb715c82012-07-11 15:03:31 +0000351 if ((kVertex_ShaderType | kFragment_ShaderType) == visibility) {
bsalomon@google.comd7727ce2012-07-12 16:40:03 +0000352 // the fragment and vertex precisions must match
bsalomon@google.com032b2212012-07-16 13:36:18 +0000353 uni.fVariable.setPrecision(kDefaultFragmentPrecision);
tomhudson@google.com242ed6f2012-05-30 17:38:57 +0000354 }
355
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000356 if (NULL != outName) {
357 *outName = uni.fVariable.c_str();
358 }
359
bsalomon@google.com032b2212012-07-16 13:36:18 +0000360 return h;
361}
362
363const GrGLShaderVar& GrGLShaderBuilder::getUniformVariable(UniformHandle u) const {
364 return fUniforms[handle_to_index(u)].fVariable;
tomhudson@google.com242ed6f2012-05-30 17:38:57 +0000365}
tomhudson@google.com23cb2292012-05-30 18:26:03 +0000366
commit-bot@chromium.orgff6ea262013-03-12 12:26:08 +0000367bool GrGLShaderBuilder::addAttribute(GrSLType type,
368 const char* name) {
369 for (int i = 0; i < fVSAttrs.count(); ++i) {
370 const GrGLShaderVar& attr = fVSAttrs[i];
371 // if attribute already added, don't add it again
372 if (attr.getName().equals(name)) {
373 GrAssert(attr.getType() == type);
374 return false;
375 }
376 }
377 fVSAttrs.push_back().set(type,
378 GrGLShaderVar::kAttribute_TypeModifier,
379 name);
380 return true;
381}
382
tomhudson@google.com23cb2292012-05-30 18:26:03 +0000383void GrGLShaderBuilder::addVarying(GrSLType type,
384 const char* name,
385 const char** vsOutName,
386 const char** fsInName) {
387 fVSOutputs.push_back();
388 fVSOutputs.back().setType(type);
389 fVSOutputs.back().setTypeModifier(GrGLShaderVar::kOut_TypeModifier);
bsalomon@google.com08283af2012-10-26 13:01:20 +0000390 if (kNonStageIdx == fCurrentStageIdx) {
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000391 fVSOutputs.back().accessName()->printf("v%s", name);
392 } else {
bsalomon@google.com08283af2012-10-26 13:01:20 +0000393 fVSOutputs.back().accessName()->printf("v%s%d", name, fCurrentStageIdx);
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000394 }
tomhudson@google.com23cb2292012-05-30 18:26:03 +0000395 if (vsOutName) {
396 *vsOutName = fVSOutputs.back().getName().c_str();
397 }
398 // input to FS comes either from VS or GS
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000399 const SkString* fsName;
tomhudson@google.com23cb2292012-05-30 18:26:03 +0000400 if (fUsesGS) {
401 // if we have a GS take each varying in as an array
402 // and output as non-array.
403 fGSInputs.push_back();
404 fGSInputs.back().setType(type);
405 fGSInputs.back().setTypeModifier(GrGLShaderVar::kIn_TypeModifier);
406 fGSInputs.back().setUnsizedArray();
407 *fGSInputs.back().accessName() = fVSOutputs.back().getName();
408 fGSOutputs.push_back();
409 fGSOutputs.back().setType(type);
410 fGSOutputs.back().setTypeModifier(GrGLShaderVar::kOut_TypeModifier);
bsalomon@google.com08283af2012-10-26 13:01:20 +0000411 if (kNonStageIdx == fCurrentStageIdx) {
bsalomon@google.comd4726202012-08-03 14:34:46 +0000412 fGSOutputs.back().accessName()->printf("g%s", name);
413 } else {
bsalomon@google.com08283af2012-10-26 13:01:20 +0000414 fGSOutputs.back().accessName()->printf("g%s%d", name, fCurrentStageIdx);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000415 }
tomhudson@google.com23cb2292012-05-30 18:26:03 +0000416 fsName = fGSOutputs.back().accessName();
417 } else {
418 fsName = fVSOutputs.back().accessName();
419 }
420 fFSInputs.push_back();
421 fFSInputs.back().setType(type);
422 fFSInputs.back().setTypeModifier(GrGLShaderVar::kIn_TypeModifier);
423 fFSInputs.back().setName(*fsName);
424 if (fsInName) {
425 *fsInName = fsName->c_str();
426 }
427}
428
bsalomon@google.com706f6682012-10-23 14:53:55 +0000429const char* GrGLShaderBuilder::fragmentPosition() {
senorblanco@chromium.org3cb406b2013-02-05 19:50:46 +0000430#if 1
bsalomon@google.combcce8922013-03-25 15:38:39 +0000431 if (fCtxInfo.caps()->fragCoordConventionsSupport()) {
bsalomon@google.com706f6682012-10-23 14:53:55 +0000432 if (!fSetupFragPosition) {
bsalomon@google.com42eff162013-04-02 12:50:49 +0000433 SkAssertResult(this->enablePrivateFeature(kFragCoordConventions_GLSLPrivateFeature));
bsalomon@google.com5fa21072012-10-25 14:57:46 +0000434 fFSInputs.push_back().set(kVec4f_GrSLType,
435 GrGLShaderVar::kIn_TypeModifier,
436 "gl_FragCoord",
437 GrGLShaderVar::kDefault_Precision,
438 GrGLShaderVar::kUpperLeft_Origin);
bsalomon@google.com706f6682012-10-23 14:53:55 +0000439 fSetupFragPosition = true;
440 }
skia.committer@gmail.com1e34ff72012-10-24 02:01:24 +0000441 return "gl_FragCoord";
bsalomon@google.com706f6682012-10-23 14:53:55 +0000442 } else {
443 static const char* kCoordName = "fragCoordYDown";
444 if (!fSetupFragPosition) {
445 GrAssert(GrGLUniformManager::kInvalidUniformHandle == fRTHeightUniform);
446 const char* rtHeightName;
skia.committer@gmail.com1e34ff72012-10-24 02:01:24 +0000447
bsalomon@google.com706f6682012-10-23 14:53:55 +0000448 // temporarily change the stage index because we're inserting a uniform whose name
449 // shouldn't be mangled to be stage-specific.
bsalomon@google.com08283af2012-10-26 13:01:20 +0000450 int oldStageIdx = fCurrentStageIdx;
451 fCurrentStageIdx = kNonStageIdx;
bsalomon@google.com706f6682012-10-23 14:53:55 +0000452 fRTHeightUniform = this->addUniform(kFragment_ShaderType,
453 kFloat_GrSLType,
454 "RTHeight",
455 &rtHeightName);
bsalomon@google.com08283af2012-10-26 13:01:20 +0000456 fCurrentStageIdx = oldStageIdx;
skia.committer@gmail.com1e34ff72012-10-24 02:01:24 +0000457
bsalomon@google.com706f6682012-10-23 14:53:55 +0000458 this->fFSCode.prependf("\tvec4 %s = vec4(gl_FragCoord.x, %s - gl_FragCoord.y, gl_FragCoord.zw);\n",
459 kCoordName, rtHeightName);
460 fSetupFragPosition = true;
461 }
462 GrAssert(GrGLUniformManager::kInvalidUniformHandle != fRTHeightUniform);
463 return kCoordName;
464 }
senorblanco@chromium.org3cb406b2013-02-05 19:50:46 +0000465#else
466 // This is the path we'll need to use once we have support for TopLeft
467 // render targets.
468 if (!fSetupFragPosition) {
469 fFSInputs.push_back().set(kVec4f_GrSLType,
470 GrGLShaderVar::kIn_TypeModifier,
471 "gl_FragCoord",
472 GrGLShaderVar::kDefault_Precision);
473 fSetupFragPosition = true;
474 }
475 return "gl_FragCoord";
476#endif
bsalomon@google.com706f6682012-10-23 14:53:55 +0000477}
478
479
bsalomon@google.coma1bf0ff2012-08-07 17:36:29 +0000480void GrGLShaderBuilder::emitFunction(ShaderType shader,
481 GrSLType returnType,
482 const char* name,
483 int argCnt,
484 const GrGLShaderVar* args,
485 const char* body,
486 SkString* outName) {
487 GrAssert(kFragment_ShaderType == shader);
bsalomon@google.com018f1792013-04-18 19:36:09 +0000488 fFSFunctions.append(GrGLSLTypeString(returnType));
bsalomon@google.com08283af2012-10-26 13:01:20 +0000489 if (kNonStageIdx != fCurrentStageIdx) {
490 outName->printf(" %s_%d", name, fCurrentStageIdx);
bsalomon@google.coma1bf0ff2012-08-07 17:36:29 +0000491 } else {
492 *outName = name;
493 }
494 fFSFunctions.append(*outName);
495 fFSFunctions.append("(");
496 for (int i = 0; i < argCnt; ++i) {
robertphillips@google.com6177e692013-02-28 20:16:25 +0000497 args[i].appendDecl(fCtxInfo, &fFSFunctions);
bsalomon@google.coma1bf0ff2012-08-07 17:36:29 +0000498 if (i < argCnt - 1) {
499 fFSFunctions.append(", ");
500 }
501 }
502 fFSFunctions.append(") {\n");
503 fFSFunctions.append(body);
504 fFSFunctions.append("}\n\n");
505}
tomhudson@google.com23cb2292012-05-30 18:26:03 +0000506
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000507namespace {
bsalomon@google.comd7727ce2012-07-12 16:40:03 +0000508
509inline void append_default_precision_qualifier(GrGLShaderVar::Precision p,
510 GrGLBinding binding,
511 SkString* str) {
512 // Desktop GLSL has added precision qualifiers but they don't do anything.
513 if (kES2_GrGLBinding == binding) {
514 switch (p) {
515 case GrGLShaderVar::kHigh_Precision:
516 str->append("precision highp float;\n");
517 break;
518 case GrGLShaderVar::kMedium_Precision:
519 str->append("precision mediump float;\n");
520 break;
521 case GrGLShaderVar::kLow_Precision:
522 str->append("precision lowp float;\n");
523 break;
524 case GrGLShaderVar::kDefault_Precision:
525 GrCrash("Default precision now allowed.");
526 default:
527 GrCrash("Unknown precision value.");
528 }
529 }
530}
bsalomon@google.com032b2212012-07-16 13:36:18 +0000531}
bsalomon@google.comd7727ce2012-07-12 16:40:03 +0000532
bsalomon@google.com032b2212012-07-16 13:36:18 +0000533void GrGLShaderBuilder::appendDecls(const VarArray& vars, SkString* out) const {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000534 for (int i = 0; i < vars.count(); ++i) {
robertphillips@google.com6177e692013-02-28 20:16:25 +0000535 vars[i].appendDecl(fCtxInfo, out);
bsalomon@google.coma1bf0ff2012-08-07 17:36:29 +0000536 out->append(";\n");
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000537 }
538}
bsalomon@google.com032b2212012-07-16 13:36:18 +0000539
540void GrGLShaderBuilder::appendUniformDecls(ShaderType stype, SkString* out) const {
541 for (int i = 0; i < fUniforms.count(); ++i) {
542 if (fUniforms[i].fVisibility & stype) {
robertphillips@google.com6177e692013-02-28 20:16:25 +0000543 fUniforms[i].fVariable.appendDecl(fCtxInfo, out);
bsalomon@google.coma1bf0ff2012-08-07 17:36:29 +0000544 out->append(";\n");
bsalomon@google.com032b2212012-07-16 13:36:18 +0000545 }
546 }
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000547}
548
549void GrGLShaderBuilder::getShader(ShaderType type, SkString* shaderStr) const {
bsalomon@google.com42eff162013-04-02 12:50:49 +0000550 const char* version = GrGetGLSLVersionDecl(fCtxInfo.binding(), fCtxInfo.glslGeneration());
551
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000552 switch (type) {
553 case kVertex_ShaderType:
bsalomon@google.com42eff162013-04-02 12:50:49 +0000554 *shaderStr = version;
bsalomon@google.com032b2212012-07-16 13:36:18 +0000555 this->appendUniformDecls(kVertex_ShaderType, shaderStr);
556 this->appendDecls(fVSAttrs, shaderStr);
557 this->appendDecls(fVSOutputs, shaderStr);
bsalomon@google.comd7bafb72012-10-22 14:30:50 +0000558 shaderStr->append("void main() {\n");
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000559 shaderStr->append(fVSCode);
bsalomon@google.comd7bafb72012-10-22 14:30:50 +0000560 shaderStr->append("}\n");
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000561 break;
562 case kGeometry_ShaderType:
563 if (fUsesGS) {
bsalomon@google.com42eff162013-04-02 12:50:49 +0000564 *shaderStr = version;
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000565 shaderStr->append(fGSHeader);
bsalomon@google.com032b2212012-07-16 13:36:18 +0000566 this->appendDecls(fGSInputs, shaderStr);
567 this->appendDecls(fGSOutputs, shaderStr);
bsalomon@google.comd7bafb72012-10-22 14:30:50 +0000568 shaderStr->append("void main() {\n");
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000569 shaderStr->append(fGSCode);
bsalomon@google.comd7bafb72012-10-22 14:30:50 +0000570 shaderStr->append("}\n");
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000571 } else {
572 shaderStr->reset();
573 }
574 break;
575 case kFragment_ShaderType:
bsalomon@google.com42eff162013-04-02 12:50:49 +0000576 *shaderStr = version;
577 shaderStr->append(fFSExtensions);
bsalomon@google.comd7727ce2012-07-12 16:40:03 +0000578 append_default_precision_qualifier(kDefaultFragmentPrecision,
robertphillips@google.com6177e692013-02-28 20:16:25 +0000579 fCtxInfo.binding(),
bsalomon@google.comd7727ce2012-07-12 16:40:03 +0000580 shaderStr);
bsalomon@google.com032b2212012-07-16 13:36:18 +0000581 this->appendUniformDecls(kFragment_ShaderType, shaderStr);
582 this->appendDecls(fFSInputs, shaderStr);
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000583 // We shouldn't have declared outputs on 1.10
robertphillips@google.com6177e692013-02-28 20:16:25 +0000584 GrAssert(k110_GrGLSLGeneration != fCtxInfo.glslGeneration() || fFSOutputs.empty());
bsalomon@google.com032b2212012-07-16 13:36:18 +0000585 this->appendDecls(fFSOutputs, shaderStr);
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000586 shaderStr->append(fFSFunctions);
bsalomon@google.comd7bafb72012-10-22 14:30:50 +0000587 shaderStr->append("void main() {\n");
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000588 shaderStr->append(fFSCode);
bsalomon@google.comd7bafb72012-10-22 14:30:50 +0000589 shaderStr->append("}\n");
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000590 break;
591 }
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000592 }
bsalomon@google.comd7727ce2012-07-12 16:40:03 +0000593
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000594void GrGLShaderBuilder::finished(GrGLuint programID) {
595 fUniformManager.getUniformLocations(programID, fUniforms);
596}
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000597
598GrGLEffect* GrGLShaderBuilder::createAndEmitGLEffect(
599 const GrEffectStage& stage,
600 GrGLEffect::EffectKey key,
601 const char* fsInColor,
602 const char* fsOutColor,
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000603 SkTArray<GrGLUniformManager::UniformHandle, true>* samplerHandles) {
604 GrAssert(NULL != stage.getEffect());
605
bsalomon@google.com6340a412013-01-22 19:55:59 +0000606 const GrEffectRef& effect = *stage.getEffect();
607 int numTextures = effect->numTextures();
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000608 SkSTArray<8, GrGLShaderBuilder::TextureSampler> textureSamplers;
609 textureSamplers.push_back_n(numTextures);
610 for (int i = 0; i < numTextures; ++i) {
bsalomon@google.com6340a412013-01-22 19:55:59 +0000611 textureSamplers[i].init(this, &effect->textureAccess(i), i);
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000612 samplerHandles->push_back(textureSamplers[i].fSamplerUniform);
613 }
bsalomon@google.comc7818882013-03-20 19:19:53 +0000614 GrDrawEffect drawEffect(stage, this->hasExplicitLocalCoords());
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000615
commit-bot@chromium.orgff6ea262013-03-12 12:26:08 +0000616 int numAttributes = stage.getVertexAttribIndexCount();
617 const int* attributeIndices = stage.getVertexAttribIndices();
618 SkSTArray<GrEffect::kMaxVertexAttribs, SkString> attributeNames;
619 for (int i = 0; i < numAttributes; ++i) {
620 SkString attributeName("aAttr");
621 attributeName.appendS32(attributeIndices[i]);
622
623 if (this->addAttribute(effect->vertexAttribType(i), attributeName.c_str())) {
624 fEffectAttributes.push_back().set(attributeIndices[i], attributeName);
625 }
626 }
627
bsalomon@google.comc7818882013-03-20 19:19:53 +0000628 GrGLEffect* glEffect = effect->getFactory().createGLInstance(drawEffect);
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000629
630 // Enclose custom code in a block to avoid namespace conflicts
631 this->fVSCode.appendf("\t{ // %s\n", glEffect->name());
632 this->fFSCode.appendf("\t{ // %s \n", glEffect->name());
bsalomon@google.comc7818882013-03-20 19:19:53 +0000633
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000634 glEffect->emitCode(this,
bsalomon@google.comc7818882013-03-20 19:19:53 +0000635 drawEffect,
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000636 key,
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000637 fsOutColor,
638 fsInColor,
639 textureSamplers);
640 this->fVSCode.appendf("\t}\n");
641 this->fFSCode.appendf("\t}\n");
642
643 return glEffect;
644}
commit-bot@chromium.orgff6ea262013-03-12 12:26:08 +0000645
646const SkString* GrGLShaderBuilder::getEffectAttributeName(int attributeIndex) const {
647 const AttributePair* attribEnd = this->getEffectAttributes().end();
skia.committer@gmail.com91274b92013-03-13 07:01:04 +0000648 for (const AttributePair* attrib = this->getEffectAttributes().begin();
commit-bot@chromium.orgff6ea262013-03-12 12:26:08 +0000649 attrib != attribEnd;
650 ++attrib) {
651 if (attrib->fIndex == attributeIndex) {
652 return &attrib->fName;
653 }
skia.committer@gmail.com91274b92013-03-13 07:01:04 +0000654 }
commit-bot@chromium.orgff6ea262013-03-12 12:26:08 +0000655
656 return NULL;
657}