blob: 7d6420ead4d4bcd371022a2aa3be2fe2c34a24cc [file] [log] [blame]
junov@google.comf93e7172011-03-31 21:26:24 +00001/*
epoger@google.comec3ed6a2011-07-28 14:26:00 +00002 * Copyright 2011 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.
junov@google.comf93e7172011-03-31 21:26:24 +00006 */
7
8#include "GrGLProgram.h"
9
tomhudson@google.comd8f856c2012-05-10 12:13:36 +000010#include "GrAllocator.h"
bsalomon@google.coma469c282012-10-24 18:28:34 +000011#include "GrEffect.h"
bsalomon@google.comd698f772012-10-25 13:22:00 +000012#include "GrGLEffect.h"
bsalomon@google.com34cccde2013-01-04 18:34:30 +000013#include "GrGpuGL.h"
bsalomon@google.com4fa66942011-09-20 19:06:12 +000014#include "GrGLShaderVar.h"
bsalomon@google.com396e61f2012-10-25 19:00:29 +000015#include "GrBackendEffectFactory.h"
tomhudson@google.com0c8d93a2011-07-01 17:08:26 +000016#include "SkTrace.h"
Scroggo97c88c22011-05-11 14:05:25 +000017#include "SkXfermode.h"
18
humper@google.com7af56be2013-01-14 18:49:19 +000019#include "SkRTConf.h"
20
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +000021SK_DEFINE_INST_COUNT(GrGLProgram)
22
robertphillips@google.com6177e692013-02-28 20:16:25 +000023#define GL_CALL(X) GR_GL_CALL(fContext.interface(), X)
24#define GL_CALL_RET(R, X) GR_GL_CALL_RET(fContext.interface(), R, X)
bsalomon@google.comecb60aa2012-07-18 13:20:29 +000025
jvanverth@google.com65eb4d52013-03-19 18:51:02 +000026SK_CONF_DECLARE(bool, c_PrintShaders, "gpu.printShaders", false,
27 "Print the source code for all shaders generated.");
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +000028
reed@google.com67e7cde2013-03-20 17:47:16 +000029#define TEX_ATTR_NAME "aTexCoord"
junov@google.comf93e7172011-03-31 21:26:24 +000030#define COL_ATTR_NAME "aColor"
bsalomon@google.coma3108262011-10-10 14:08:47 +000031#define COV_ATTR_NAME "aCoverage"
bsalomon@google.comaeb21602011-08-30 18:13:44 +000032#define EDGE_ATTR_NAME "aEdge"
junov@google.comf93e7172011-03-31 21:26:24 +000033
bsalomon@google.com4fa66942011-09-20 19:06:12 +000034namespace {
bsalomon@google.com4fa66942011-09-20 19:06:12 +000035inline const char* declared_color_output_name() { return "fsColorOut"; }
36inline const char* dual_source_output_name() { return "dualSourceOut"; }
bsalomon@google.com4fa66942011-09-20 19:06:12 +000037}
junov@google.com6acc9b32011-05-16 18:32:07 +000038
jvanverth@google.com9b855c72013-03-01 18:21:22 +000039const GrGLProgram::AttribLayout GrGLProgram::kAttribLayouts[kGrVertexAttribTypeCount] = {
40 {1, GR_GL_FLOAT, false}, // kFloat_GrVertexAttribType
41 {2, GR_GL_FLOAT, false}, // kVec2f_GrVertexAttribType
42 {3, GR_GL_FLOAT, false}, // kVec3f_GrVertexAttribType
43 {4, GR_GL_FLOAT, false}, // kVec4f_GrVertexAttribType
44 {4, GR_GL_UNSIGNED_BYTE, true}, // kVec4ub_GrVertexAttribType
45};
46
bsalomon@google.com91207482013-02-12 21:45:24 +000047void GrGLProgram::BuildDesc(const GrDrawState& drawState,
48 bool isPoints,
49 GrDrawState::BlendOptFlags blendOpts,
50 GrBlendCoeff srcCoeff,
51 GrBlendCoeff dstCoeff,
52 const GrGpuGL* gpu,
53 Desc* desc) {
54
55 // This should already have been caught
56 GrAssert(!(GrDrawState::kSkipDraw_BlendOptFlag & blendOpts));
57
58 bool skipCoverage = SkToBool(blendOpts & GrDrawState::kEmitTransBlack_BlendOptFlag);
59
60 bool skipColor = SkToBool(blendOpts & (GrDrawState::kEmitTransBlack_BlendOptFlag |
61 GrDrawState::kEmitCoverage_BlendOptFlag));
62
63 // The descriptor is used as a cache key. Thus when a field of the
jvanverth@google.com9b855c72013-03-01 18:21:22 +000064 // descriptor will not affect program generation (because of the attribute
65 // bindings in use or other descriptor field settings) it should be set
bsalomon@google.com91207482013-02-12 21:45:24 +000066 // to a canonical value to avoid duplicate programs with different keys.
67
68 // Must initialize all fields or cache will have false negatives!
jvanverth@google.com9b855c72013-03-01 18:21:22 +000069 desc->fAttribBindings = drawState.getAttribBindings();
bsalomon@google.com91207482013-02-12 21:45:24 +000070
71 desc->fEmitsPointSize = isPoints;
72
jvanverth@google.com65eb4d52013-03-19 18:51:02 +000073 bool requiresAttributeColors =
74 !skipColor && SkToBool(desc->fAttribBindings & GrDrawState::kColor_AttribBindingsBit);
75 bool requiresAttributeCoverage =
76 !skipCoverage && SkToBool(desc->fAttribBindings & GrDrawState::kCoverage_AttribBindingsBit);
bsalomon@google.com91207482013-02-12 21:45:24 +000077
78 // fColorInput/fCoverageInput records how colors are specified for the program So we strip the
jvanverth@google.com9b855c72013-03-01 18:21:22 +000079 // bits from the bindings to avoid false negatives when searching for an existing program in the
bsalomon@google.com91207482013-02-12 21:45:24 +000080 // cache.
jvanverth@google.com65eb4d52013-03-19 18:51:02 +000081 desc->fAttribBindings &=
82 ~(GrDrawState::kColor_AttribBindingsBit | GrDrawState::kCoverage_AttribBindingsBit);
bsalomon@google.com91207482013-02-12 21:45:24 +000083
84 desc->fColorFilterXfermode = skipColor ?
85 SkXfermode::kDst_Mode :
86 drawState.getColorFilterMode();
87
88 // no reason to do edge aa or look at per-vertex coverage if coverage is ignored
89 if (skipCoverage) {
jvanverth@google.com9b855c72013-03-01 18:21:22 +000090 desc->fAttribBindings &= ~(GrDrawState::kEdge_AttribBindingsBit |
91 GrDrawState::kCoverage_AttribBindingsBit);
bsalomon@google.com91207482013-02-12 21:45:24 +000092 }
93
94 bool colorIsTransBlack = SkToBool(blendOpts & GrDrawState::kEmitTransBlack_BlendOptFlag);
95 bool colorIsSolidWhite = (blendOpts & GrDrawState::kEmitCoverage_BlendOptFlag) ||
96 (!requiresAttributeColors && 0xffffffff == drawState.getColor());
97 if (colorIsTransBlack) {
98 desc->fColorInput = Desc::kTransBlack_ColorInput;
99 } else if (colorIsSolidWhite) {
100 desc->fColorInput = Desc::kSolidWhite_ColorInput;
101 } else if (GR_GL_NO_CONSTANT_ATTRIBUTES && !requiresAttributeColors) {
102 desc->fColorInput = Desc::kUniform_ColorInput;
103 } else {
104 desc->fColorInput = Desc::kAttribute_ColorInput;
105 }
106
107 bool covIsSolidWhite = !requiresAttributeCoverage && 0xffffffff == drawState.getCoverage();
108
109 if (skipCoverage) {
110 desc->fCoverageInput = Desc::kTransBlack_ColorInput;
111 } else if (covIsSolidWhite) {
112 desc->fCoverageInput = Desc::kSolidWhite_ColorInput;
113 } else if (GR_GL_NO_CONSTANT_ATTRIBUTES && !requiresAttributeCoverage) {
114 desc->fCoverageInput = Desc::kUniform_ColorInput;
115 } else {
116 desc->fCoverageInput = Desc::kAttribute_ColorInput;
117 }
118
119 int lastEnabledStage = -1;
120
jvanverth@google.com9b855c72013-03-01 18:21:22 +0000121 if (!skipCoverage && (desc->fAttribBindings & GrDrawState::kEdge_AttribBindingsBit)) {
bsalomon@google.com91207482013-02-12 21:45:24 +0000122 desc->fVertexEdgeType = drawState.getVertexEdgeType();
123 desc->fDiscardIfOutsideEdge = drawState.getStencil().doesWrite();
124 } else {
125 // Use canonical values when edge-aa is not enabled to avoid program cache misses.
126 desc->fVertexEdgeType = GrDrawState::kHairLine_EdgeType;
127 desc->fDiscardIfOutsideEdge = false;
128 }
129
130 for (int s = 0; s < GrDrawState::kNumStages; ++s) {
131
132 bool skip = s < drawState.getFirstCoverageStage() ? skipColor : skipCoverage;
133 if (!skip && drawState.isStageEnabled(s)) {
134 lastEnabledStage = s;
135 const GrEffectRef& effect = *drawState.getStage(s).getEffect();
136 const GrBackendEffectFactory& factory = effect->getFactory();
reed@google.com67e7cde2013-03-20 17:47:16 +0000137 desc->fEffectKeys[s] = factory.glEffectKey(drawState.getStage(s), gpu->glCaps());
bsalomon@google.com91207482013-02-12 21:45:24 +0000138 } else {
139 desc->fEffectKeys[s] = 0;
140 }
141 }
142
143 desc->fDualSrcOutput = Desc::kNone_DualSrcOutput;
144
145 // Currently the experimental GS will only work with triangle prims (and it doesn't do anything
146 // other than pass through values from the VS to the FS anyway).
147#if GR_GL_EXPERIMENTAL_GS
148#if 0
149 desc->fExperimentalGS = gpu->getCaps().geometryShaderSupport();
150#else
151 desc->fExperimentalGS = false;
152#endif
153#endif
154
155 // We want to avoid generating programs with different "first cov stage" values when they would
156 // compute the same result. We set field in the desc to kNumStages when either there are no
157 // coverage stages or the distinction between coverage and color is immaterial.
158 int firstCoverageStage = GrDrawState::kNumStages;
159 desc->fFirstCoverageStage = GrDrawState::kNumStages;
160 bool hasCoverage = drawState.getFirstCoverageStage() <= lastEnabledStage;
161 if (hasCoverage) {
162 firstCoverageStage = drawState.getFirstCoverageStage();
163 }
164
165 // other coverage inputs
166 if (!hasCoverage) {
167 hasCoverage = requiresAttributeCoverage ||
jvanverth@google.com9b855c72013-03-01 18:21:22 +0000168 (desc->fAttribBindings & GrDrawState::kEdge_AttribBindingsBit);
bsalomon@google.com91207482013-02-12 21:45:24 +0000169 }
170
171 if (hasCoverage) {
172 // color filter is applied between color/coverage computation
173 if (SkXfermode::kDst_Mode != desc->fColorFilterXfermode) {
174 desc->fFirstCoverageStage = firstCoverageStage;
175 }
176
177 if (gpu->getCaps().dualSourceBlendingSupport() &&
178 !(blendOpts & (GrDrawState::kEmitCoverage_BlendOptFlag |
179 GrDrawState::kCoverageAsAlpha_BlendOptFlag))) {
180 if (kZero_GrBlendCoeff == dstCoeff) {
181 // write the coverage value to second color
182 desc->fDualSrcOutput = Desc::kCoverage_DualSrcOutput;
183 desc->fFirstCoverageStage = firstCoverageStage;
184 } else if (kSA_GrBlendCoeff == dstCoeff) {
185 // SA dst coeff becomes 1-(1-SA)*coverage when dst is partially covered.
186 desc->fDualSrcOutput = Desc::kCoverageISA_DualSrcOutput;
187 desc->fFirstCoverageStage = firstCoverageStage;
188 } else if (kSC_GrBlendCoeff == dstCoeff) {
189 // SA dst coeff becomes 1-(1-SA)*coverage when dst is partially covered.
190 desc->fDualSrcOutput = Desc::kCoverageISC_DualSrcOutput;
191 desc->fFirstCoverageStage = firstCoverageStage;
192 }
193 }
194 }
jvanverth@google.com9b855c72013-03-01 18:21:22 +0000195
196 desc->fPositionAttributeIndex = drawState.getAttribIndex(GrDrawState::kPosition_AttribIndex);
197 if (requiresAttributeColors) {
198 desc->fColorAttributeIndex = drawState.getAttribIndex(GrDrawState::kColor_AttribIndex);
199 } else {
200 desc->fColorAttributeIndex = GrDrawState::kColorOverrideAttribIndexValue;
skia.committer@gmail.comf140f182013-03-02 07:01:56 +0000201 }
jvanverth@google.com9b855c72013-03-01 18:21:22 +0000202 if (requiresAttributeCoverage) {
203 desc->fCoverageAttributeIndex = drawState.getAttribIndex(GrDrawState::kCoverage_AttribIndex);
204 } else {
205 desc->fCoverageAttributeIndex = GrDrawState::kCoverageOverrideAttribIndexValue;
206 }
robertphillips@google.com37ebe3f2013-03-04 20:03:44 +0000207 if (desc->fAttribBindings & GrDrawState::kEdge_AttribBindingsBit) {
208 desc->fEdgeAttributeIndex = drawState.getAttribIndex(GrDrawState::kEdge_AttribIndex);
209 }
reed@google.com67e7cde2013-03-20 17:47:16 +0000210 if (GrDrawState::AttributesBindExplicitTexCoords(desc->fAttribBindings)) {
211 desc->fTexCoordAttributeIndex = drawState.getAttribIndex(GrDrawState::kTexCoord_AttribIndex);
robertphillips@google.com37ebe3f2013-03-04 20:03:44 +0000212 }
jvanverth@google.com9b855c72013-03-01 18:21:22 +0000213
214#if GR_DEBUG
215 // verify valid vertex attribute state
216 const GrVertexAttrib* vertexAttribs = drawState.getVertexAttribs();
217 GrAssert(desc->fPositionAttributeIndex < GrDrawState::kVertexAttribCnt);
218 GrAssert(kAttribLayouts[vertexAttribs[desc->fPositionAttributeIndex].fType].fCount == 2);
219 if (requiresAttributeColors) {
220 GrAssert(desc->fColorAttributeIndex < GrDrawState::kVertexAttribCnt);
221 GrAssert(kAttribLayouts[vertexAttribs[desc->fColorAttributeIndex].fType].fCount == 4);
222 }
223 if (requiresAttributeCoverage) {
224 GrAssert(desc->fCoverageAttributeIndex < GrDrawState::kVertexAttribCnt);
225 GrAssert(kAttribLayouts[vertexAttribs[desc->fCoverageAttributeIndex].fType].fCount == 4);
226 }
227 if (desc->fAttribBindings & GrDrawState::kEdge_AttribBindingsBit) {
228 GrAssert(desc->fEdgeAttributeIndex < GrDrawState::kVertexAttribCnt);
229 GrAssert(kAttribLayouts[vertexAttribs[desc->fEdgeAttributeIndex].fType].fCount == 4);
reed@google.com67e7cde2013-03-20 17:47:16 +0000230 }
231 if (GrDrawState::AttributesBindExplicitTexCoords(desc->fAttribBindings)) {
232 GrAssert(desc->fTexCoordAttributeIndex < GrDrawState::kVertexAttribCnt);
233 GrAssert(kAttribLayouts[vertexAttribs[desc->fTexCoordAttributeIndex].fType].fCount == 2);
jvanverth@google.com9b855c72013-03-01 18:21:22 +0000234 }
235#endif
bsalomon@google.com91207482013-02-12 21:45:24 +0000236}
237
robertphillips@google.com6177e692013-02-28 20:16:25 +0000238GrGLProgram* GrGLProgram::Create(const GrGLContext& gl,
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000239 const Desc& desc,
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000240 const GrEffectStage* stages[]) {
241 GrGLProgram* program = SkNEW_ARGS(GrGLProgram, (gl, desc, stages));
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000242 if (!program->succeeded()) {
243 delete program;
244 program = NULL;
245 }
246 return program;
247}
248
robertphillips@google.com6177e692013-02-28 20:16:25 +0000249GrGLProgram::GrGLProgram(const GrGLContext& gl,
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000250 const Desc& desc,
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000251 const GrEffectStage* stages[])
robertphillips@google.com6177e692013-02-28 20:16:25 +0000252: fContext(gl)
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000253, fUniformManager(gl) {
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000254 fDesc = desc;
255 fVShaderID = 0;
256 fGShaderID = 0;
257 fFShaderID = 0;
258 fProgramID = 0;
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000259
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000260 fColor = GrColor_ILLEGAL;
261 fColorFilterColor = GrColor_ILLEGAL;
262
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000263 for (int s = 0; s < GrDrawState::kNumStages; ++s) {
bsalomon@google.com46fba0d2012-10-25 21:42:05 +0000264 fEffects[s] = NULL;
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000265 }
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000266
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000267 this->genProgram(stages);
junov@google.comf93e7172011-03-31 21:26:24 +0000268}
269
270GrGLProgram::~GrGLProgram() {
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000271 if (fVShaderID) {
272 GL_CALL(DeleteShader(fVShaderID));
273 }
274 if (fGShaderID) {
275 GL_CALL(DeleteShader(fGShaderID));
276 }
277 if (fFShaderID) {
278 GL_CALL(DeleteShader(fFShaderID));
279 }
280 if (fProgramID) {
281 GL_CALL(DeleteProgram(fProgramID));
282 }
283
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000284 for (int i = 0; i < GrDrawState::kNumStages; ++i) {
bsalomon@google.com46fba0d2012-10-25 21:42:05 +0000285 delete fEffects[i];
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000286 }
junov@google.comf93e7172011-03-31 21:26:24 +0000287}
288
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000289void GrGLProgram::abandon() {
290 fVShaderID = 0;
291 fGShaderID = 0;
292 fFShaderID = 0;
293 fProgramID = 0;
294}
295
tomhudson@google.com0d3f1fb2011-06-01 19:27:31 +0000296void GrGLProgram::overrideBlend(GrBlendCoeff* srcCoeff,
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000297 GrBlendCoeff* dstCoeff) const {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000298 switch (fDesc.fDualSrcOutput) {
299 case Desc::kNone_DualSrcOutput:
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000300 break;
301 // the prog will write a coverage value to the secondary
302 // output and the dst is blended by one minus that value.
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000303 case Desc::kCoverage_DualSrcOutput:
304 case Desc::kCoverageISA_DualSrcOutput:
305 case Desc::kCoverageISC_DualSrcOutput:
bsalomon@google.com47059542012-06-06 20:51:20 +0000306 *dstCoeff = (GrBlendCoeff)GrGpu::kIS2C_GrBlendCoeff;
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000307 break;
308 default:
309 GrCrash("Unexpected dual source blend output");
310 break;
311 }
312}
313
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000314namespace {
315
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000316// given two blend coeffecients determine whether the src
317// and/or dst computation can be omitted.
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000318inline void need_blend_inputs(SkXfermode::Coeff srcCoeff,
319 SkXfermode::Coeff dstCoeff,
320 bool* needSrcValue,
321 bool* needDstValue) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000322 if (SkXfermode::kZero_Coeff == srcCoeff) {
323 switch (dstCoeff) {
324 // these all read the src
325 case SkXfermode::kSC_Coeff:
326 case SkXfermode::kISC_Coeff:
327 case SkXfermode::kSA_Coeff:
328 case SkXfermode::kISA_Coeff:
329 *needSrcValue = true;
330 break;
331 default:
332 *needSrcValue = false;
333 break;
334 }
335 } else {
336 *needSrcValue = true;
337 }
338 if (SkXfermode::kZero_Coeff == dstCoeff) {
339 switch (srcCoeff) {
340 // these all read the dst
341 case SkXfermode::kDC_Coeff:
342 case SkXfermode::kIDC_Coeff:
343 case SkXfermode::kDA_Coeff:
344 case SkXfermode::kIDA_Coeff:
345 *needDstValue = true;
346 break;
347 default:
348 *needDstValue = false;
349 break;
350 }
351 } else {
352 *needDstValue = true;
Scroggo97c88c22011-05-11 14:05:25 +0000353 }
354}
355
356/**
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000357 * Create a blend_coeff * value string to be used in shader code. Sets empty
358 * string if result is trivially zero.
359 */
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000360inline void blend_term_string(SkString* str, SkXfermode::Coeff coeff,
361 const char* src, const char* dst,
362 const char* value) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000363 switch (coeff) {
364 case SkXfermode::kZero_Coeff: /** 0 */
365 *str = "";
366 break;
367 case SkXfermode::kOne_Coeff: /** 1 */
368 *str = value;
369 break;
370 case SkXfermode::kSC_Coeff:
371 str->printf("(%s * %s)", src, value);
372 break;
373 case SkXfermode::kISC_Coeff:
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000374 str->printf("((%s - %s) * %s)", GrGLSLOnesVecf(4), src, value);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000375 break;
376 case SkXfermode::kDC_Coeff:
377 str->printf("(%s * %s)", dst, value);
378 break;
379 case SkXfermode::kIDC_Coeff:
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000380 str->printf("((%s - %s) * %s)", GrGLSLOnesVecf(4), dst, value);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000381 break;
382 case SkXfermode::kSA_Coeff: /** src alpha */
383 str->printf("(%s.a * %s)", src, value);
384 break;
385 case SkXfermode::kISA_Coeff: /** inverse src alpha (i.e. 1 - sa) */
386 str->printf("((1.0 - %s.a) * %s)", src, value);
387 break;
388 case SkXfermode::kDA_Coeff: /** dst alpha */
389 str->printf("(%s.a * %s)", dst, value);
390 break;
391 case SkXfermode::kIDA_Coeff: /** inverse dst alpha (i.e. 1 - da) */
392 str->printf("((1.0 - %s.a) * %s)", dst, value);
393 break;
394 default:
395 GrCrash("Unexpected xfer coeff.");
396 break;
397 }
398}
399/**
Scroggo97c88c22011-05-11 14:05:25 +0000400 * Adds a line to the fragment shader code which modifies the color by
401 * the specified color filter.
402 */
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000403void add_color_filter(GrGLShaderBuilder* builder,
404 const char * outputVar,
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000405 SkXfermode::Coeff uniformCoeff,
406 SkXfermode::Coeff colorCoeff,
407 const char* filterColor,
408 const char* inColor) {
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000409 SkString colorStr, constStr;
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000410 blend_term_string(&colorStr, colorCoeff, filterColor, inColor, inColor);
411 blend_term_string(&constStr, uniformCoeff, filterColor, inColor, filterColor);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000412
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000413 SkString sum;
414 GrGLSLAdd4f(&sum, colorStr.c_str(), constStr.c_str());
415 builder->fsCodeAppendf("\t%s = %s;\n", outputVar, sum.c_str());
Scroggo97c88c22011-05-11 14:05:25 +0000416}
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000417}
Scroggo97c88c22011-05-11 14:05:25 +0000418
bsalomon@google.comd4726202012-08-03 14:34:46 +0000419bool GrGLProgram::genEdgeCoverage(SkString* coverageVar,
bsalomon@google.com706f6682012-10-23 14:53:55 +0000420 GrGLShaderBuilder* builder) const {
jvanverth@google.com9b855c72013-03-01 18:21:22 +0000421 if (fDesc.fAttribBindings & GrDrawState::kEdge_AttribBindingsBit) {
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000422 const char *vsName, *fsName;
bsalomon@google.com706f6682012-10-23 14:53:55 +0000423 builder->addVarying(kVec4f_GrSLType, "Edge", &vsName, &fsName);
commit-bot@chromium.orgff6ea262013-03-12 12:26:08 +0000424 builder->addAttribute(kVec4f_GrSLType, EDGE_ATTR_NAME);
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000425 builder->vsCodeAppendf("\t%s = " EDGE_ATTR_NAME ";\n", vsName);
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000426 switch (fDesc.fVertexEdgeType) {
bsalomon@google.com93c96602012-04-27 13:05:21 +0000427 case GrDrawState::kHairLine_EdgeType:
jvanverth@google.com65eb4d52013-03-19 18:51:02 +0000428 builder->fsCodeAppendf("\tfloat edgeAlpha = abs(dot(vec3(%s.xy,1), %s.xyz));\n",
429 builder->fragmentPosition(), fsName);
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000430 builder->fsCodeAppendf("\tedgeAlpha = max(1.0 - edgeAlpha, 0.0);\n");
bsalomon@google.com93c96602012-04-27 13:05:21 +0000431 break;
432 case GrDrawState::kQuad_EdgeType:
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000433 builder->fsCodeAppendf("\tfloat edgeAlpha;\n");
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000434 // keep the derivative instructions outside the conditional
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000435 builder->fsCodeAppendf("\tvec2 duvdx = dFdx(%s.xy);\n", fsName);
436 builder->fsCodeAppendf("\tvec2 duvdy = dFdy(%s.xy);\n", fsName);
437 builder->fsCodeAppendf("\tif (%s.z > 0.0 && %s.w > 0.0) {\n", fsName, fsName);
bsalomon@google.com9aed1142012-01-30 14:28:39 +0000438 // today we know z and w are in device space. We could use derivatives
jvanverth@google.com65eb4d52013-03-19 18:51:02 +0000439 builder->fsCodeAppendf("\t\tedgeAlpha = min(min(%s.z, %s.w) + 0.5, 1.0);\n", fsName,
440 fsName);
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000441 builder->fsCodeAppendf ("\t} else {\n");
442 builder->fsCodeAppendf("\t\tvec2 gF = vec2(2.0*%s.x*duvdx.x - duvdx.y,\n"
443 "\t\t 2.0*%s.x*duvdy.x - duvdy.y);\n",
444 fsName, fsName);
445 builder->fsCodeAppendf("\t\tedgeAlpha = (%s.x*%s.x - %s.y);\n", fsName, fsName, fsName);
446 builder->fsCodeAppendf("\t\tedgeAlpha = clamp(0.5 - edgeAlpha / length(gF), 0.0, 1.0);\n"
447 "\t}\n");
robertphillips@google.com6177e692013-02-28 20:16:25 +0000448 if (kES2_GrGLBinding == fContext.info().binding()) {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000449 builder->fHeader.append("#extension GL_OES_standard_derivatives: enable\n");
bsalomon@google.com9aed1142012-01-30 14:28:39 +0000450 }
bsalomon@google.com93c96602012-04-27 13:05:21 +0000451 break;
452 case GrDrawState::kHairQuad_EdgeType:
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000453 builder->fsCodeAppendf("\tvec2 duvdx = dFdx(%s.xy);\n", fsName);
454 builder->fsCodeAppendf("\tvec2 duvdy = dFdy(%s.xy);\n", fsName);
455 builder->fsCodeAppendf("\tvec2 gF = vec2(2.0*%s.x*duvdx.x - duvdx.y,\n"
456 "\t 2.0*%s.x*duvdy.x - duvdy.y);\n",
457 fsName, fsName);
jvanverth@google.com65eb4d52013-03-19 18:51:02 +0000458 builder->fsCodeAppendf("\tfloat edgeAlpha = (%s.x*%s.x - %s.y);\n", fsName, fsName,
459 fsName);
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000460 builder->fsCodeAppend("\tedgeAlpha = sqrt(edgeAlpha*edgeAlpha / dot(gF, gF));\n");
461 builder->fsCodeAppend("\tedgeAlpha = max(1.0 - edgeAlpha, 0.0);\n");
robertphillips@google.com6177e692013-02-28 20:16:25 +0000462 if (kES2_GrGLBinding == fContext.info().binding()) {
bsalomon@google.com706f6682012-10-23 14:53:55 +0000463 builder->fHeader.printf("#extension GL_OES_standard_derivatives: enable\n");
bsalomon@google.com66105672011-09-15 15:12:00 +0000464 }
bsalomon@google.com93c96602012-04-27 13:05:21 +0000465 break;
bsalomon@google.com93c96602012-04-27 13:05:21 +0000466 default:
467 GrCrash("Unknown Edge Type!");
468 break;
bsalomon@google.com66105672011-09-15 15:12:00 +0000469 }
bsalomon@google.com45a15f52012-12-10 19:10:17 +0000470 if (fDesc.fDiscardIfOutsideEdge) {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000471 builder->fsCodeAppend("\tif (edgeAlpha <= 0.0) {\n\t\tdiscard;\n\t}\n");
bsalomon@google.com45a15f52012-12-10 19:10:17 +0000472 }
bsalomon@google.com66105672011-09-15 15:12:00 +0000473 *coverageVar = "edgeAlpha";
bsalomon@google.comd4726202012-08-03 14:34:46 +0000474 return true;
bsalomon@google.com66105672011-09-15 15:12:00 +0000475 } else {
476 coverageVar->reset();
bsalomon@google.comd4726202012-08-03 14:34:46 +0000477 return false;
bsalomon@google.com66105672011-09-15 15:12:00 +0000478 }
479}
480
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000481void GrGLProgram::genInputColor(GrGLShaderBuilder* builder, SkString* inColor) {
482 switch (fDesc.fColorInput) {
483 case GrGLProgram::Desc::kAttribute_ColorInput: {
commit-bot@chromium.orgff6ea262013-03-12 12:26:08 +0000484 builder->addAttribute(kVec4f_GrSLType, COL_ATTR_NAME);
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000485 const char *vsName, *fsName;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000486 builder->addVarying(kVec4f_GrSLType, "Color", &vsName, &fsName);
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000487 builder->vsCodeAppendf("\t%s = " COL_ATTR_NAME ";\n", vsName);
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000488 *inColor = fsName;
489 } break;
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000490 case GrGLProgram::Desc::kUniform_ColorInput: {
491 const char* name;
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000492 fUniformHandles.fColorUni = builder->addUniform(GrGLShaderBuilder::kFragment_ShaderType,
493 kVec4f_GrSLType, "Color", &name);
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000494 *inColor = name;
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000495 break;
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000496 }
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000497 case GrGLProgram::Desc::kTransBlack_ColorInput:
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000498 GrAssert(!"needComputedColor should be false.");
499 break;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000500 case GrGLProgram::Desc::kSolidWhite_ColorInput:
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000501 break;
502 default:
503 GrCrash("Unknown color type.");
504 break;
505 }
506}
507
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000508void GrGLProgram::genUniformCoverage(GrGLShaderBuilder* builder, SkString* inOutCoverage) {
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000509 const char* covUniName;
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000510 fUniformHandles.fCoverageUni = builder->addUniform(GrGLShaderBuilder::kFragment_ShaderType,
511 kVec4f_GrSLType, "Coverage", &covUniName);
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000512 if (inOutCoverage->size()) {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000513 builder->fsCodeAppendf("\tvec4 uniCoverage = %s * %s;\n",
514 covUniName, inOutCoverage->c_str());
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000515 *inOutCoverage = "uniCoverage";
516 } else {
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000517 *inOutCoverage = covUniName;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000518 }
519}
520
521namespace {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000522void gen_attribute_coverage(GrGLShaderBuilder* builder,
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000523 SkString* inOutCoverage) {
commit-bot@chromium.orgff6ea262013-03-12 12:26:08 +0000524 builder->addAttribute(kVec4f_GrSLType, COV_ATTR_NAME);
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000525 const char *vsName, *fsName;
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000526 builder->addVarying(kVec4f_GrSLType, "Coverage", &vsName, &fsName);
527 builder->vsCodeAppendf("\t%s = " COV_ATTR_NAME ";\n", vsName);
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000528 if (inOutCoverage->size()) {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000529 builder->fsCodeAppendf("\tvec4 attrCoverage = %s * %s;\n", fsName, inOutCoverage->c_str());
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000530 *inOutCoverage = "attrCoverage";
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000531 } else {
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000532 *inOutCoverage = fsName;
533 }
534}
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000535}
536
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000537void GrGLProgram::genGeometryShader(GrGLShaderBuilder* builder) const {
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000538#if GR_GL_EXPERIMENTAL_GS
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000539 // TODO: The builder should add all this glue code.
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000540 if (fDesc.fExperimentalGS) {
robertphillips@google.com6177e692013-02-28 20:16:25 +0000541 GrAssert(fContext.info().glslGeneration() >= k150_GrGLSLGeneration);
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000542 builder->fGSHeader.append("layout(triangles) in;\n"
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000543 "layout(triangle_strip, max_vertices = 6) out;\n");
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000544 builder->gsCodeAppend("\tfor (int i = 0; i < 3; ++i) {\n"
545 "\t\tgl_Position = gl_in[i].gl_Position;\n");
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000546 if (fDesc.fEmitsPointSize) {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000547 builder->gsCodeAppend("\t\tgl_PointSize = 1.0;\n");
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000548 }
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000549 GrAssert(builder->fGSInputs.count() == builder->fGSOutputs.count());
550 int count = builder->fGSInputs.count();
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000551 for (int i = 0; i < count; ++i) {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000552 builder->gsCodeAppendf("\t\t%s = %s[i];\n",
553 builder->fGSOutputs[i].getName().c_str(),
554 builder->fGSInputs[i].getName().c_str());
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000555 }
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000556 builder->gsCodeAppend("\t\tEmitVertex();\n"
557 "\t}\n"
558 "\tEndPrimitive();\n");
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000559 }
560#endif
561}
562
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000563const char* GrGLProgram::adjustInColor(const SkString& inColor) const {
senorblanco@chromium.orgb3a39b52012-01-05 18:28:56 +0000564 if (inColor.size()) {
565 return inColor.c_str();
566 } else {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000567 if (Desc::kSolidWhite_ColorInput == fDesc.fColorInput) {
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000568 return GrGLSLOnesVecf(4);
senorblanco@chromium.orgb3a39b52012-01-05 18:28:56 +0000569 } else {
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000570 return GrGLSLZerosVecf(4);
senorblanco@chromium.orgb3a39b52012-01-05 18:28:56 +0000571 }
572 }
573}
574
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000575namespace {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000576// prints a shader using params similar to glShaderSource
577void print_shader(GrGLint stringCnt,
578 const GrGLchar** strings,
579 GrGLint* stringLengths) {
580 for (int i = 0; i < stringCnt; ++i) {
581 if (NULL == stringLengths || stringLengths[i] < 0) {
582 GrPrintf(strings[i]);
583 } else {
584 GrPrintf("%.*s", stringLengths[i], strings[i]);
585 }
586 }
587}
588
589// Compiles a GL shader, returns shader ID or 0 if failed params have same meaning as glShaderSource
robertphillips@google.com6177e692013-02-28 20:16:25 +0000590GrGLuint compile_shader(const GrGLContext& gl,
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000591 GrGLenum type,
592 int stringCnt,
593 const char** strings,
594 int* stringLengths) {
595 SK_TRACE_EVENT1("GrGLProgram::CompileShader",
596 "stringCount", SkStringPrintf("%i", stringCnt).c_str());
597
598 GrGLuint shader;
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000599 GR_GL_CALL_RET(gl.interface(), shader, CreateShader(type));
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000600 if (0 == shader) {
601 return 0;
602 }
603
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000604 const GrGLInterface* gli = gl.interface();
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000605 GrGLint compiled = GR_GL_INIT_ZERO;
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000606 GR_GL_CALL(gli, ShaderSource(shader, stringCnt, strings, stringLengths));
607 GR_GL_CALL(gli, CompileShader(shader));
608 GR_GL_CALL(gli, GetShaderiv(shader, GR_GL_COMPILE_STATUS, &compiled));
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000609
610 if (!compiled) {
611 GrGLint infoLen = GR_GL_INIT_ZERO;
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000612 GR_GL_CALL(gli, GetShaderiv(shader, GR_GL_INFO_LOG_LENGTH, &infoLen));
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000613 SkAutoMalloc log(sizeof(char)*(infoLen+1)); // outside if for debugger
614 if (infoLen > 0) {
615 // retrieve length even though we don't need it to workaround bug in chrome cmd buffer
616 // param validation.
617 GrGLsizei length = GR_GL_INIT_ZERO;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000618 GR_GL_CALL(gli, GetShaderInfoLog(shader, infoLen+1,
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000619 &length, (char*)log.get()));
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000620 print_shader(stringCnt, strings, stringLengths);
621 GrPrintf("\n%s", log.get());
622 }
623 GrAssert(!"Shader compilation failed!");
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000624 GR_GL_CALL(gli, DeleteShader(shader));
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000625 return 0;
626 }
627 return shader;
628}
629
630// helper version of above for when shader is already flattened into a single SkString
robertphillips@google.com6177e692013-02-28 20:16:25 +0000631GrGLuint compile_shader(const GrGLContext& gl, GrGLenum type, const SkString& shader) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000632 const GrGLchar* str = shader.c_str();
633 int length = shader.size();
634 return compile_shader(gl, type, 1, &str, &length);
635}
636
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000637}
638
639// compiles all the shaders from builder and stores the shader IDs
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000640bool GrGLProgram::compileShaders(const GrGLShaderBuilder& builder) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000641
642 SkString shader;
643
644 builder.getShader(GrGLShaderBuilder::kVertex_ShaderType, &shader);
humper@google.com7af56be2013-01-14 18:49:19 +0000645 if (c_PrintShaders) {
646 GrPrintf(shader.c_str());
647 GrPrintf("\n");
648 }
skia.committer@gmail.com7fc0e0a2013-01-15 02:01:40 +0000649
robertphillips@google.com6177e692013-02-28 20:16:25 +0000650 if (!(fVShaderID = compile_shader(fContext, GR_GL_VERTEX_SHADER, shader))) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000651 return false;
652 }
653
654 if (builder.fUsesGS) {
655 builder.getShader(GrGLShaderBuilder::kGeometry_ShaderType, &shader);
humper@google.com7af56be2013-01-14 18:49:19 +0000656 if (c_PrintShaders) {
657 GrPrintf(shader.c_str());
658 GrPrintf("\n");
659 }
robertphillips@google.com6177e692013-02-28 20:16:25 +0000660 if (!(fGShaderID = compile_shader(fContext, GR_GL_GEOMETRY_SHADER, shader))) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000661 return false;
662 }
663 } else {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000664 fGShaderID = 0;
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000665 }
666
667 builder.getShader(GrGLShaderBuilder::kFragment_ShaderType, &shader);
humper@google.com7af56be2013-01-14 18:49:19 +0000668 if (c_PrintShaders) {
669 GrPrintf(shader.c_str());
670 GrPrintf("\n");
671 }
robertphillips@google.com6177e692013-02-28 20:16:25 +0000672 if (!(fFShaderID = compile_shader(fContext, GR_GL_FRAGMENT_SHADER, shader))) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000673 return false;
674 }
675
676 return true;
677}
678
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000679bool GrGLProgram::genProgram(const GrEffectStage* stages[]) {
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000680 GrAssert(0 == fProgramID);
681
reed@google.com67e7cde2013-03-20 17:47:16 +0000682 GrGLShaderBuilder builder(fContext.info(), fUniformManager);
jvanverth@google.com9b855c72013-03-01 18:21:22 +0000683 const GrAttribBindings& attribBindings = fDesc.fAttribBindings;
junov@google.comf93e7172011-03-31 21:26:24 +0000684
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000685#if GR_GL_EXPERIMENTAL_GS
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000686 builder.fUsesGS = fDesc.fExperimentalGS;
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000687#endif
688
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000689 SkXfermode::Coeff colorCoeff, uniformCoeff;
690 // The rest of transfer mode color filters have not been implemented
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000691 if (fDesc.fColorFilterXfermode < SkXfermode::kCoeffModesCnt) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000692 GR_DEBUGCODE(bool success =)
tomhudson@google.com0d831722011-06-02 15:37:14 +0000693 SkXfermode::ModeAsCoeff(static_cast<SkXfermode::Mode>
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000694 (fDesc.fColorFilterXfermode),
tomhudson@google.com0d831722011-06-02 15:37:14 +0000695 &uniformCoeff, &colorCoeff);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000696 GR_DEBUGASSERT(success);
697 } else {
698 colorCoeff = SkXfermode::kOne_Coeff;
699 uniformCoeff = SkXfermode::kZero_Coeff;
700 }
701
bsalomon@google.com67e78c92012-10-17 13:36:14 +0000702 // no need to do the color filter if coverage is 0. The output color is scaled by the coverage.
703 // All the dual source outputs are scaled by the coverage as well.
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000704 if (Desc::kTransBlack_ColorInput == fDesc.fCoverageInput) {
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000705 colorCoeff = SkXfermode::kZero_Coeff;
706 uniformCoeff = SkXfermode::kZero_Coeff;
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000707 }
708
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000709 // If we know the final color is going to be all zeros then we can
bsalomon@google.com67e78c92012-10-17 13:36:14 +0000710 // simplify the color filter coefficients. needComputedColor will then
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000711 // come out false below.
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000712 if (Desc::kTransBlack_ColorInput == fDesc.fColorInput) {
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000713 colorCoeff = SkXfermode::kZero_Coeff;
714 if (SkXfermode::kDC_Coeff == uniformCoeff ||
715 SkXfermode::kDA_Coeff == uniformCoeff) {
716 uniformCoeff = SkXfermode::kZero_Coeff;
717 } else if (SkXfermode::kIDC_Coeff == uniformCoeff ||
718 SkXfermode::kIDA_Coeff == uniformCoeff) {
719 uniformCoeff = SkXfermode::kOne_Coeff;
720 }
721 }
722
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000723 bool needColorFilterUniform;
724 bool needComputedColor;
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000725 need_blend_inputs(uniformCoeff, colorCoeff,
726 &needColorFilterUniform, &needComputedColor);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000727
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000728 // the dual source output has no canonical var name, have to
729 // declare an output, which is incompatible with gl_FragColor/gl_FragData.
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000730 bool dualSourceOutputWritten = false;
robertphillips@google.com6177e692013-02-28 20:16:25 +0000731 builder.fHeader.append(GrGetGLSLVersionDecl(fContext.info().binding(),
732 fContext.info().glslGeneration()));
bsalomon@google.come55fd0f2012-02-10 15:56:06 +0000733
734 GrGLShaderVar colorOutput;
robertphillips@google.com6177e692013-02-28 20:16:25 +0000735 bool isColorDeclared = GrGLSLSetupFSColorOuput(fContext.info().glslGeneration(),
bsalomon@google.come55fd0f2012-02-10 15:56:06 +0000736 declared_color_output_name(),
737 &colorOutput);
738 if (isColorDeclared) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000739 builder.fFSOutputs.push_back(colorOutput);
bsalomon@google.come55fd0f2012-02-10 15:56:06 +0000740 }
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000741
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000742 const char* viewMName;
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000743 fUniformHandles.fViewMatrixUni = builder.addUniform(GrGLShaderBuilder::kVertex_ShaderType,
744 kMat33f_GrSLType, "ViewM", &viewMName);
bsalomon@google.com341767c2012-05-11 20:47:39 +0000745
skia.committer@gmail.come862d162012-10-31 02:01:18 +0000746
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000747 builder.vsCodeAppendf("\tvec3 pos3 = %s * vec3(%s, 1);\n"
748 "\tgl_Position = vec4(pos3.xy, 0, pos3.z);\n",
749 viewMName, builder.positionAttribute().getName().c_str());
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000750
bsalomon@google.com91961302011-05-09 18:39:58 +0000751 // incoming color to current stage being processed.
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000752 SkString inColor;
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000753
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000754 if (needComputedColor) {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000755 this->genInputColor(&builder, &inColor);
junov@google.comf93e7172011-03-31 21:26:24 +0000756 }
757
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000758 // we output point size in the GS if present
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000759 if (fDesc.fEmitsPointSize && !builder.fUsesGS){
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000760 builder.vsCodeAppend("\tgl_PointSize = 1.0;\n");
junov@google.comf93e7172011-03-31 21:26:24 +0000761 }
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000762
reed@google.com67e7cde2013-03-20 17:47:16 +0000763 // add texture coordinates that are used to the list of vertex attr decls
764 if (GrDrawState::AttributesBindExplicitTexCoords(attribBindings)) {
765 builder.addAttribute(kVec2f_GrSLType, TEX_ATTR_NAME);
766 }
767
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000768 ///////////////////////////////////////////////////////////////////////////
769 // compute the final color
Scroggo97c88c22011-05-11 14:05:25 +0000770
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000771 // if we have color stages string them together, feeding the output color
772 // of each to the next and generating code for each stage.
773 if (needComputedColor) {
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000774 SkString outColor;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000775 for (int s = 0; s < fDesc.fFirstCoverageStage; ++s) {
bsalomon@google.comdbe49f72012-11-05 16:36:02 +0000776 if (GrGLEffect::kNoEffectKey != fDesc.fEffectKeys[s]) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000777 // create var to hold stage result
778 outColor = "color";
779 outColor.appendS32(s);
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000780 builder.fsCodeAppendf("\tvec4 %s;\n", outColor.c_str());
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000781
reed@google.com67e7cde2013-03-20 17:47:16 +0000782 const char* inCoords;
783 // figure out what our input coords are
784 if (!GrDrawState::StageBindsExplicitTexCoords(attribBindings, s)) {
785 inCoords = builder.positionAttribute().c_str();
786 } else {
787 // must have input tex coordinates if stage is enabled.
788 inCoords = TEX_ATTR_NAME;
789 }
790
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000791 builder.setCurrentStage(s);
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000792 fEffects[s] = builder.createAndEmitGLEffect(*stages[s],
793 fDesc.fEffectKeys[s],
794 inColor.size() ? inColor.c_str() : NULL,
795 outColor.c_str(),
reed@google.com67e7cde2013-03-20 17:47:16 +0000796 inCoords,
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000797 &fUniformHandles.fSamplerUnis[s]);
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000798 builder.setNonStage();
junov@google.comf93e7172011-03-31 21:26:24 +0000799 inColor = outColor;
junov@google.comf93e7172011-03-31 21:26:24 +0000800 }
801 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000802 }
Scroggo97c88c22011-05-11 14:05:25 +0000803
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000804 // if have all ones or zeros for the "dst" input to the color filter then we
805 // may be able to make additional optimizations.
806 if (needColorFilterUniform && needComputedColor && !inColor.size()) {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000807 GrAssert(Desc::kSolidWhite_ColorInput == fDesc.fColorInput);
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000808 bool uniformCoeffIsZero = SkXfermode::kIDC_Coeff == uniformCoeff ||
809 SkXfermode::kIDA_Coeff == uniformCoeff;
810 if (uniformCoeffIsZero) {
811 uniformCoeff = SkXfermode::kZero_Coeff;
812 bool bogus;
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000813 need_blend_inputs(SkXfermode::kZero_Coeff, colorCoeff,
814 &needColorFilterUniform, &bogus);
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000815 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000816 }
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000817 const char* colorFilterColorUniName = NULL;
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000818 if (needColorFilterUniform) {
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000819 fUniformHandles.fColorFilterUni = builder.addUniform(
820 GrGLShaderBuilder::kFragment_ShaderType,
821 kVec4f_GrSLType, "FilterColor",
822 &colorFilterColorUniName);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000823 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000824 bool wroteFragColorZero = false;
tomhudson@google.com0d831722011-06-02 15:37:14 +0000825 if (SkXfermode::kZero_Coeff == uniformCoeff &&
bsalomon@google.com67e78c92012-10-17 13:36:14 +0000826 SkXfermode::kZero_Coeff == colorCoeff) {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000827 builder.fsCodeAppendf("\t%s = %s;\n", colorOutput.getName().c_str(), GrGLSLZerosVecf(4));
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000828 wroteFragColorZero = true;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000829 } else if (SkXfermode::kDst_Mode != fDesc.fColorFilterXfermode) {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000830 builder.fsCodeAppend("\tvec4 filteredColor;\n");
senorblanco@chromium.orgb3a39b52012-01-05 18:28:56 +0000831 const char* color = adjustInColor(inColor);
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000832 add_color_filter(&builder, "filteredColor", uniformCoeff,
833 colorCoeff, colorFilterColorUniName, color);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000834 inColor = "filteredColor";
835 }
836
837 ///////////////////////////////////////////////////////////////////////////
838 // compute the partial coverage (coverage stages and edge aa)
839
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000840 SkString inCoverage;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000841 bool coverageIsZero = Desc::kTransBlack_ColorInput == fDesc.fCoverageInput;
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000842 // we don't need to compute coverage at all if we know the final shader
843 // output will be zero and we don't have a dual src blend output.
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000844 if (!wroteFragColorZero || Desc::kNone_DualSrcOutput != fDesc.fDualSrcOutput) {
bsalomon@google.com66105672011-09-15 15:12:00 +0000845
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000846 if (!coverageIsZero) {
bsalomon@google.comd4726202012-08-03 14:34:46 +0000847 bool inCoverageIsScalar = this->genEdgeCoverage(&inCoverage, &builder);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000848
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000849 switch (fDesc.fCoverageInput) {
850 case Desc::kSolidWhite_ColorInput:
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000851 // empty string implies solid white
852 break;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000853 case Desc::kAttribute_ColorInput:
854 gen_attribute_coverage(&builder, &inCoverage);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000855 inCoverageIsScalar = false;
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000856 break;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000857 case Desc::kUniform_ColorInput:
858 this->genUniformCoverage(&builder, &inCoverage);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000859 inCoverageIsScalar = false;
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000860 break;
861 default:
862 GrCrash("Unexpected input coverage.");
863 }
bsalomon@google.coma3108262011-10-10 14:08:47 +0000864
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000865 SkString outCoverage;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000866 const int& startStage = fDesc.fFirstCoverageStage;
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000867 for (int s = startStage; s < GrDrawState::kNumStages; ++s) {
bsalomon@google.comdbe49f72012-11-05 16:36:02 +0000868 if (fDesc.fEffectKeys[s]) {
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000869 // create var to hold stage output
870 outCoverage = "coverage";
871 outCoverage.appendS32(s);
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000872 builder.fsCodeAppendf("\tvec4 %s;\n", outCoverage.c_str());
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000873
reed@google.com67e7cde2013-03-20 17:47:16 +0000874 const char* inCoords;
875 // figure out what our input coords are
876 if (!GrDrawState::StageBindsExplicitTexCoords(attribBindings, s)) {
877 inCoords = builder.positionAttribute().c_str();
878 } else {
879 // must have input tex coordinates if stage is
880 // enabled.
881 inCoords = TEX_ATTR_NAME;
882 }
883
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000884 // stages don't know how to deal with a scalar input. (Maybe they should. We
bsalomon@google.comd4726202012-08-03 14:34:46 +0000885 // could pass a GrGLShaderVar)
886 if (inCoverageIsScalar) {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000887 builder.fsCodeAppendf("\tvec4 %s4 = vec4(%s);\n",
888 inCoverage.c_str(), inCoverage.c_str());
bsalomon@google.comd4726202012-08-03 14:34:46 +0000889 inCoverage.append("4");
890 }
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000891 builder.setCurrentStage(s);
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000892 fEffects[s] = builder.createAndEmitGLEffect(
893 *stages[s],
894 fDesc.fEffectKeys[s],
895 inCoverage.size() ? inCoverage.c_str() : NULL,
896 outCoverage.c_str(),
reed@google.com67e7cde2013-03-20 17:47:16 +0000897 inCoords,
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000898 &fUniformHandles.fSamplerUnis[s]);
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000899 builder.setNonStage();
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000900 inCoverage = outCoverage;
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000901 }
senorblanco@chromium.org92e0f222011-05-12 15:49:15 +0000902 }
commit-bot@chromium.orgff6ea262013-03-12 12:26:08 +0000903
904 // discard if coverage is zero
905 if (fDesc.fDiscardIfOutsideEdge && !outCoverage.isEmpty()) {
jvanverth@google.com65eb4d52013-03-19 18:51:02 +0000906 builder.fsCodeAppendf(
907 "\tif (all(lessThanEqual(%s, vec4(0.0)))) {\n\t\tdiscard;\n\t}\n",
908 outCoverage.c_str());
commit-bot@chromium.orgff6ea262013-03-12 12:26:08 +0000909 }
senorblanco@chromium.org92e0f222011-05-12 15:49:15 +0000910 }
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000911
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000912 if (Desc::kNone_DualSrcOutput != fDesc.fDualSrcOutput) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000913 builder.fFSOutputs.push_back().set(kVec4f_GrSLType,
914 GrGLShaderVar::kOut_TypeModifier,
915 dual_source_output_name());
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000916 bool outputIsZero = coverageIsZero;
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000917 SkString coeff;
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000918 if (!outputIsZero &&
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000919 Desc::kCoverage_DualSrcOutput != fDesc.fDualSrcOutput && !wroteFragColorZero) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000920 if (!inColor.size()) {
921 outputIsZero = true;
922 } else {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000923 if (Desc::kCoverageISA_DualSrcOutput == fDesc.fDualSrcOutput) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000924 coeff.printf("(1 - %s.a)", inColor.c_str());
925 } else {
926 coeff.printf("(vec4(1,1,1,1) - %s)", inColor.c_str());
927 }
928 }
929 }
930 if (outputIsZero) {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000931 builder.fsCodeAppendf("\t%s = %s;\n", dual_source_output_name(), GrGLSLZerosVecf(4));
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000932 } else {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000933 SkString modulate;
934 GrGLSLModulate4f(&modulate, coeff.c_str(), inCoverage.c_str());
935 builder.fsCodeAppendf("\t%s = %s;\n", dual_source_output_name(), modulate.c_str());
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000936 }
937 dualSourceOutputWritten = true;
938 }
junov@google.comf93e7172011-03-31 21:26:24 +0000939 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000940
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000941 ///////////////////////////////////////////////////////////////////////////
942 // combine color and coverage as frag color
943
944 if (!wroteFragColorZero) {
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000945 if (coverageIsZero) {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000946 builder.fsCodeAppendf("\t%s = %s;\n", colorOutput.getName().c_str(), GrGLSLZerosVecf(4));
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000947 } else {
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000948 SkString modulate;
949 GrGLSLModulate4f(&modulate, inColor.c_str(), inCoverage.c_str());
950 builder.fsCodeAppendf("\t%s = %s;\n", colorOutput.getName().c_str(), modulate.c_str());
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000951 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000952 }
953
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000954 ///////////////////////////////////////////////////////////////////////////
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000955 // insert GS
956#if GR_DEBUG
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000957 this->genGeometryShader(&builder);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000958#endif
959
960 ///////////////////////////////////////////////////////////////////////////
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000961 // compile and setup attribs and unis
962
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000963 if (!this->compileShaders(builder)) {
bsalomon@google.com91961302011-05-09 18:39:58 +0000964 return false;
965 }
966
bsalomon@google.com17504f52012-10-30 12:34:25 +0000967 if (!this->bindOutputsAttribsAndLinkProgram(builder,
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000968 isColorDeclared,
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000969 dualSourceOutputWritten)) {
bsalomon@google.com91961302011-05-09 18:39:58 +0000970 return false;
971 }
972
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000973 builder.finished(fProgramID);
974 this->initSamplerUniforms();
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000975 fUniformHandles.fRTHeightUni = builder.getRTHeightUniform();
bsalomon@google.com91961302011-05-09 18:39:58 +0000976
977 return true;
978}
979
bsalomon@google.com17504f52012-10-30 12:34:25 +0000980bool GrGLProgram::bindOutputsAttribsAndLinkProgram(const GrGLShaderBuilder& builder,
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000981 bool bindColorOut,
982 bool bindDualSrcOut) {
983 GL_CALL_RET(fProgramID, CreateProgram());
984 if (!fProgramID) {
bsalomon@google.com91961302011-05-09 18:39:58 +0000985 return false;
986 }
bsalomon@google.com91961302011-05-09 18:39:58 +0000987
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000988 GL_CALL(AttachShader(fProgramID, fVShaderID));
989 if (fGShaderID) {
990 GL_CALL(AttachShader(fProgramID, fGShaderID));
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000991 }
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000992 GL_CALL(AttachShader(fProgramID, fFShaderID));
bsalomon@google.com91961302011-05-09 18:39:58 +0000993
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000994 if (bindColorOut) {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000995 GL_CALL(BindFragDataLocation(fProgramID, 0, declared_color_output_name()));
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000996 }
997 if (bindDualSrcOut) {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000998 GL_CALL(BindFragDataLocationIndexed(fProgramID, 0, 1, dual_source_output_name()));
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000999 }
1000
bsalomon@google.com91961302011-05-09 18:39:58 +00001001 // Bind the attrib locations to same values for all shaders
bsalomon@google.com17504f52012-10-30 12:34:25 +00001002 GL_CALL(BindAttribLocation(fProgramID,
jvanverth@google.com9b855c72013-03-01 18:21:22 +00001003 fDesc.fPositionAttributeIndex,
bsalomon@google.com17504f52012-10-30 12:34:25 +00001004 builder.positionAttribute().c_str()));
jvanverth@google.com9b855c72013-03-01 18:21:22 +00001005 GL_CALL(BindAttribLocation(fProgramID, fDesc.fColorAttributeIndex, COL_ATTR_NAME));
1006 GL_CALL(BindAttribLocation(fProgramID, fDesc.fCoverageAttributeIndex, COV_ATTR_NAME));
skia.committer@gmail.comf140f182013-03-02 07:01:56 +00001007
jvanverth@google.com9b855c72013-03-01 18:21:22 +00001008 if (fDesc.fAttribBindings & GrDrawState::kEdge_AttribBindingsBit) {
1009 GL_CALL(BindAttribLocation(fProgramID, fDesc.fEdgeAttributeIndex, EDGE_ATTR_NAME));
1010 }
reed@google.com67e7cde2013-03-20 17:47:16 +00001011 if (GrDrawState::AttributesBindExplicitTexCoords(fDesc.fAttribBindings)) {
1012 GL_CALL(BindAttribLocation(fProgramID, fDesc.fTexCoordAttributeIndex, TEX_ATTR_NAME));
jvanverth@google.com9b855c72013-03-01 18:21:22 +00001013 }
skia.committer@gmail.comf140f182013-03-02 07:01:56 +00001014
commit-bot@chromium.orgff6ea262013-03-12 12:26:08 +00001015 const GrGLShaderBuilder::AttributePair* attribEnd = builder.getEffectAttributes().end();
skia.committer@gmail.com91274b92013-03-13 07:01:04 +00001016 for (const GrGLShaderBuilder::AttributePair* attrib = builder.getEffectAttributes().begin();
commit-bot@chromium.orgff6ea262013-03-12 12:26:08 +00001017 attrib != attribEnd;
1018 ++attrib) {
1019 GL_CALL(BindAttribLocation(fProgramID, attrib->fIndex, attrib->fName.c_str()));
1020 }
1021
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001022 GL_CALL(LinkProgram(fProgramID));
bsalomon@google.com91961302011-05-09 18:39:58 +00001023
1024 GrGLint linked = GR_GL_INIT_ZERO;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001025 GL_CALL(GetProgramiv(fProgramID, GR_GL_LINK_STATUS, &linked));
bsalomon@google.com91961302011-05-09 18:39:58 +00001026 if (!linked) {
1027 GrGLint infoLen = GR_GL_INIT_ZERO;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001028 GL_CALL(GetProgramiv(fProgramID, GR_GL_INFO_LOG_LENGTH, &infoLen));
bsalomon@google.com3582bf92011-06-30 21:32:31 +00001029 SkAutoMalloc log(sizeof(char)*(infoLen+1)); // outside if for debugger
bsalomon@google.com91961302011-05-09 18:39:58 +00001030 if (infoLen > 0) {
bsalomon@google.com79afcaa2011-09-14 14:29:18 +00001031 // retrieve length even though we don't need it to workaround
1032 // bug in chrome cmd buffer param validation.
1033 GrGLsizei length = GR_GL_INIT_ZERO;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001034 GL_CALL(GetProgramInfoLog(fProgramID,
bsalomon@google.com96399942012-02-13 14:39:16 +00001035 infoLen+1,
1036 &length,
1037 (char*)log.get()));
bsalomon@google.com91961302011-05-09 18:39:58 +00001038 GrPrintf((char*)log.get());
1039 }
1040 GrAssert(!"Error linking program");
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001041 GL_CALL(DeleteProgram(fProgramID));
1042 fProgramID = 0;
bsalomon@google.com91961302011-05-09 18:39:58 +00001043 return false;
1044 }
1045 return true;
1046}
1047
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +00001048void GrGLProgram::initSamplerUniforms() {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001049 GL_CALL(UseProgram(fProgramID));
bsalomon@google.com34cccde2013-01-04 18:34:30 +00001050 // We simply bind the uniforms to successive texture units beginning at 0. setData() assumes this
1051 // behavior.
1052 GrGLint texUnitIdx = 0;
tomhudson@google.com93813632011-10-27 20:21:16 +00001053 for (int s = 0; s < GrDrawState::kNumStages; ++s) {
bsalomon@google.com34cccde2013-01-04 18:34:30 +00001054 int numSamplers = fUniformHandles.fSamplerUnis[s].count();
1055 for (int u = 0; u < numSamplers; ++u) {
1056 UniformHandle handle = fUniformHandles.fSamplerUnis[s][u];
1057 if (GrGLUniformManager::kInvalidUniformHandle != handle) {
1058 fUniformManager.setSampler(handle, texUnitIdx);
1059 ++texUnitIdx;
bsalomon@google.com0982d352012-07-31 15:33:25 +00001060 }
bsalomon@google.com91961302011-05-09 18:39:58 +00001061 }
bsalomon@google.com91961302011-05-09 18:39:58 +00001062 }
bsalomon@google.com91961302011-05-09 18:39:58 +00001063}
1064
bsalomon@google.comeb715c82012-07-11 15:03:31 +00001065///////////////////////////////////////////////////////////////////////////////
junov@google.comf93e7172011-03-31 21:26:24 +00001066
bsalomon@google.com91207482013-02-12 21:45:24 +00001067void GrGLProgram::setData(GrGpuGL* gpu,
1068 GrColor color,
1069 GrColor coverage,
1070 SharedGLState* sharedState) {
bsalomon@google.com34cccde2013-01-04 18:34:30 +00001071 const GrDrawState& drawState = gpu->getDrawState();
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001072
bsalomon@google.com91207482013-02-12 21:45:24 +00001073 this->setColor(drawState, color, sharedState);
1074 this->setCoverage(drawState, coverage, sharedState);
bsalomon@google.com6a51dcb2013-02-13 16:03:51 +00001075 this->setMatrixAndRenderTargetHeight(drawState);
bsalomon@google.com91207482013-02-12 21:45:24 +00001076
1077 // Setup the SkXfermode::Mode-based colorfilter uniform if necessary
1078 if (GrGLUniformManager::kInvalidUniformHandle != fUniformHandles.fColorFilterUni &&
1079 fColorFilterColor != drawState.getColorFilterColor()) {
1080 GrGLfloat c[4];
1081 GrColorToRGBAFloat(drawState.getColorFilterColor(), c);
1082 fUniformManager.set4fv(fUniformHandles.fColorFilterUni, 0, 1, c);
1083 fColorFilterColor = drawState.getColorFilterColor();
1084 }
1085
bsalomon@google.com34cccde2013-01-04 18:34:30 +00001086 GrGLint texUnitIdx = 0;
bsalomon@google.com4285acc2012-10-22 14:11:24 +00001087 for (int s = 0; s < GrDrawState::kNumStages; ++s) {
bsalomon@google.com46fba0d2012-10-25 21:42:05 +00001088 if (NULL != fEffects[s]) {
bsalomon@google.com08283af2012-10-26 13:01:20 +00001089 const GrEffectStage& stage = drawState.getStage(s);
1090 GrAssert(NULL != stage.getEffect());
reed@google.com67e7cde2013-03-20 17:47:16 +00001091 fEffects[s]->setData(fUniformManager, stage);
bsalomon@google.com34cccde2013-01-04 18:34:30 +00001092 int numSamplers = fUniformHandles.fSamplerUnis[s].count();
1093 for (int u = 0; u < numSamplers; ++u) {
1094 UniformHandle handle = fUniformHandles.fSamplerUnis[s][u];
1095 if (GrGLUniformManager::kInvalidUniformHandle != handle) {
bsalomon@google.com6340a412013-01-22 19:55:59 +00001096 const GrTextureAccess& access = (*stage.getEffect())->textureAccess(u);
bsalomon@google.com34cccde2013-01-04 18:34:30 +00001097 GrGLTexture* texture = static_cast<GrGLTexture*>(access.getTexture());
1098 gpu->bindTexture(texUnitIdx, access.getParams(), texture);
1099 ++texUnitIdx;
1100 }
1101 }
bsalomon@google.com4285acc2012-10-22 14:11:24 +00001102 }
1103 }
skia.committer@gmail.com8ae714b2013-01-05 02:02:05 +00001104}
bsalomon@google.com91207482013-02-12 21:45:24 +00001105
1106void GrGLProgram::setColor(const GrDrawState& drawState,
1107 GrColor color,
1108 SharedGLState* sharedState) {
jvanverth@google.com9b855c72013-03-01 18:21:22 +00001109 if (!(drawState.getAttribBindings() & GrDrawState::kColor_AttribBindingsBit)) {
bsalomon@google.com91207482013-02-12 21:45:24 +00001110 switch (fDesc.fColorInput) {
1111 case GrGLProgram::Desc::kAttribute_ColorInput:
1112 if (sharedState->fConstAttribColor != color) {
1113 // OpenGL ES only supports the float varieties of glVertexAttrib
1114 GrGLfloat c[4];
1115 GrColorToRGBAFloat(color, c);
jvanverth@google.com9b855c72013-03-01 18:21:22 +00001116 GL_CALL(VertexAttrib4fv(fDesc.fColorAttributeIndex, c));
bsalomon@google.com91207482013-02-12 21:45:24 +00001117 sharedState->fConstAttribColor = color;
1118 }
1119 break;
1120 case GrGLProgram::Desc::kUniform_ColorInput:
1121 if (fColor != color) {
1122 // OpenGL ES doesn't support unsigned byte varieties of glUniform
1123 GrGLfloat c[4];
1124 GrColorToRGBAFloat(color, c);
1125 GrAssert(GrGLUniformManager::kInvalidUniformHandle !=
1126 fUniformHandles.fColorUni);
1127 fUniformManager.set4fv(fUniformHandles.fColorUni, 0, 1, c);
1128 fColor = color;
1129 }
1130 break;
1131 case GrGLProgram::Desc::kSolidWhite_ColorInput:
1132 case GrGLProgram::Desc::kTransBlack_ColorInput:
1133 break;
1134 default:
1135 GrCrash("Unknown color type.");
1136 }
1137 }
1138}
1139
1140void GrGLProgram::setCoverage(const GrDrawState& drawState,
1141 GrColor coverage,
1142 SharedGLState* sharedState) {
jvanverth@google.com9b855c72013-03-01 18:21:22 +00001143 if (!(drawState.getAttribBindings() & GrDrawState::kCoverage_AttribBindingsBit)) {
bsalomon@google.com91207482013-02-12 21:45:24 +00001144 switch (fDesc.fCoverageInput) {
1145 case Desc::kAttribute_ColorInput:
1146 if (sharedState->fConstAttribCoverage != coverage) {
1147 // OpenGL ES only supports the float varieties of glVertexAttrib
1148 GrGLfloat c[4];
1149 GrColorToRGBAFloat(coverage, c);
jvanverth@google.com9b855c72013-03-01 18:21:22 +00001150 GL_CALL(VertexAttrib4fv(fDesc.fCoverageAttributeIndex, c));
bsalomon@google.com91207482013-02-12 21:45:24 +00001151 sharedState->fConstAttribCoverage = coverage;
1152 }
1153 break;
1154 case Desc::kUniform_ColorInput:
1155 if (fCoverage != coverage) {
1156 // OpenGL ES doesn't support unsigned byte varieties of glUniform
1157 GrGLfloat c[4];
1158 GrColorToRGBAFloat(coverage, c);
1159 GrAssert(GrGLUniformManager::kInvalidUniformHandle !=
1160 fUniformHandles.fCoverageUni);
1161 fUniformManager.set4fv(fUniformHandles.fCoverageUni, 0, 1, c);
1162 fCoverage = coverage;
1163 }
1164 break;
1165 case Desc::kSolidWhite_ColorInput:
1166 case Desc::kTransBlack_ColorInput:
1167 break;
1168 default:
1169 GrCrash("Unknown coverage type.");
1170 }
1171 }
1172}
bsalomon@google.com6a51dcb2013-02-13 16:03:51 +00001173
1174void GrGLProgram::setMatrixAndRenderTargetHeight(const GrDrawState& drawState) {
1175 const GrRenderTarget* rt = drawState.getRenderTarget();
1176 SkISize size;
1177 size.set(rt->width(), rt->height());
1178
1179 // Load the RT height uniform if it is needed to y-flip gl_FragCoord.
1180 if (GrGLUniformManager::kInvalidUniformHandle != fUniformHandles.fRTHeightUni &&
1181 fMatrixState.fRenderTargetSize.fHeight != size.fHeight) {
1182 fUniformManager.set1f(fUniformHandles.fRTHeightUni, SkIntToScalar(size.fHeight));
1183 }
1184
1185 if (fMatrixState.fRenderTargetOrigin != rt->origin() ||
1186 !fMatrixState.fViewMatrix.cheapEqualTo(drawState.getViewMatrix()) ||
1187 fMatrixState.fRenderTargetSize != size) {
1188 SkMatrix m;
1189 if (kBottomLeft_GrSurfaceOrigin == rt->origin()) {
1190 m.setAll(
1191 SkIntToScalar(2) / size.fWidth, 0, -SK_Scalar1,
1192 0,-SkIntToScalar(2) / size.fHeight, SK_Scalar1,
1193 0, 0, SkMatrix::I()[8]);
1194 } else {
1195 m.setAll(
1196 SkIntToScalar(2) / size.fWidth, 0, -SK_Scalar1,
1197 0, SkIntToScalar(2) / size.fHeight,-SK_Scalar1,
1198 0, 0, SkMatrix::I()[8]);
1199 }
1200 m.setConcat(m, drawState.getViewMatrix());
1201
1202 // ES doesn't allow you to pass true to the transpose param so we do our own transpose.
1203 GrGLfloat mt[] = {
1204 SkScalarToFloat(m[SkMatrix::kMScaleX]),
1205 SkScalarToFloat(m[SkMatrix::kMSkewY]),
1206 SkScalarToFloat(m[SkMatrix::kMPersp0]),
1207 SkScalarToFloat(m[SkMatrix::kMSkewX]),
1208 SkScalarToFloat(m[SkMatrix::kMScaleY]),
1209 SkScalarToFloat(m[SkMatrix::kMPersp1]),
1210 SkScalarToFloat(m[SkMatrix::kMTransX]),
1211 SkScalarToFloat(m[SkMatrix::kMTransY]),
1212 SkScalarToFloat(m[SkMatrix::kMPersp2])
1213 };
1214 fUniformManager.setMatrix3f(fUniformHandles.fViewMatrixUni, mt);
1215 fMatrixState.fViewMatrix = drawState.getViewMatrix();
1216 fMatrixState.fRenderTargetSize = size;
1217 fMatrixState.fRenderTargetOrigin = rt->origin();
1218 }
1219}