blob: 781e15fdde34f7d9418180f48a2fe83d2988060c [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
humper@google.com7af56be2013-01-14 18:49:19 +000026SK_CONF_DECLARE(bool, c_PrintShaders, "gpu.printShaders", false, "Print the source code for all shaders generated.");
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +000027
jvanverth@google.com39768252013-02-14 15:25:44 +000028#define TEX_ATTR_NAME "aTexCoord"
junov@google.comf93e7172011-03-31 21:26:24 +000029#define COL_ATTR_NAME "aColor"
bsalomon@google.coma3108262011-10-10 14:08:47 +000030#define COV_ATTR_NAME "aCoverage"
bsalomon@google.comaeb21602011-08-30 18:13:44 +000031#define EDGE_ATTR_NAME "aEdge"
junov@google.comf93e7172011-03-31 21:26:24 +000032
bsalomon@google.com4fa66942011-09-20 19:06:12 +000033namespace {
bsalomon@google.com4fa66942011-09-20 19:06:12 +000034inline const char* declared_color_output_name() { return "fsColorOut"; }
35inline const char* dual_source_output_name() { return "dualSourceOut"; }
bsalomon@google.com4fa66942011-09-20 19:06:12 +000036}
junov@google.com6acc9b32011-05-16 18:32:07 +000037
jvanverth@google.comb8b705b2013-02-28 16:28:34 +000038const GrGLProgram::AttribLayout GrGLProgram::kAttribLayouts[kGrVertexAttribTypeCount] = {
39 {1, GR_GL_FLOAT, false}, // kFloat_GrVertexAttribType
40 {2, GR_GL_FLOAT, false}, // kVec2f_GrVertexAttribType
41 {3, GR_GL_FLOAT, false}, // kVec3f_GrVertexAttribType
42 {4, GR_GL_FLOAT, false}, // kVec4f_GrVertexAttribType
43 {4, GR_GL_UNSIGNED_BYTE, true}, // kVec4ub_GrVertexAttribType
44};
45
bsalomon@google.com91207482013-02-12 21:45:24 +000046void GrGLProgram::BuildDesc(const GrDrawState& drawState,
47 bool isPoints,
48 GrDrawState::BlendOptFlags blendOpts,
49 GrBlendCoeff srcCoeff,
50 GrBlendCoeff dstCoeff,
51 const GrGpuGL* gpu,
52 Desc* desc) {
53
54 // This should already have been caught
55 GrAssert(!(GrDrawState::kSkipDraw_BlendOptFlag & blendOpts));
56
57 bool skipCoverage = SkToBool(blendOpts & GrDrawState::kEmitTransBlack_BlendOptFlag);
58
59 bool skipColor = SkToBool(blendOpts & (GrDrawState::kEmitTransBlack_BlendOptFlag |
60 GrDrawState::kEmitCoverage_BlendOptFlag));
61
62 // The descriptor is used as a cache key. Thus when a field of the
jvanverth@google.comb8b705b2013-02-28 16:28:34 +000063 // descriptor will not affect program generation (because of the attribute
64 // bindings in use or other descriptor field settings) it should be set
bsalomon@google.com91207482013-02-12 21:45:24 +000065 // to a canonical value to avoid duplicate programs with different keys.
66
67 // Must initialize all fields or cache will have false negatives!
jvanverth@google.comb8b705b2013-02-28 16:28:34 +000068 desc->fAttribBindings = drawState.getAttribBindings();
bsalomon@google.com91207482013-02-12 21:45:24 +000069
70 desc->fEmitsPointSize = isPoints;
71
72 bool requiresAttributeColors = !skipColor &&
jvanverth@google.comb8b705b2013-02-28 16:28:34 +000073 SkToBool(desc->fAttribBindings & GrDrawState::kColor_AttribBindingsBit);
bsalomon@google.com91207482013-02-12 21:45:24 +000074 bool requiresAttributeCoverage = !skipCoverage &&
jvanverth@google.comb8b705b2013-02-28 16:28:34 +000075 SkToBool(desc->fAttribBindings & GrDrawState::kCoverage_AttribBindingsBit);
bsalomon@google.com91207482013-02-12 21:45:24 +000076
77 // fColorInput/fCoverageInput records how colors are specified for the program So we strip the
jvanverth@google.comb8b705b2013-02-28 16:28:34 +000078 // bits from the bindings to avoid false negatives when searching for an existing program in the
bsalomon@google.com91207482013-02-12 21:45:24 +000079 // cache.
jvanverth@google.comb8b705b2013-02-28 16:28:34 +000080 desc->fAttribBindings &= ~(GrDrawState::kColor_AttribBindingsBit | GrDrawState::kCoverage_AttribBindingsBit);
bsalomon@google.com91207482013-02-12 21:45:24 +000081
82 desc->fColorFilterXfermode = skipColor ?
83 SkXfermode::kDst_Mode :
84 drawState.getColorFilterMode();
85
86 // no reason to do edge aa or look at per-vertex coverage if coverage is ignored
87 if (skipCoverage) {
jvanverth@google.comb8b705b2013-02-28 16:28:34 +000088 desc->fAttribBindings &= ~(GrDrawState::kEdge_AttribBindingsBit |
89 GrDrawState::kCoverage_AttribBindingsBit);
bsalomon@google.com91207482013-02-12 21:45:24 +000090 }
91
92 bool colorIsTransBlack = SkToBool(blendOpts & GrDrawState::kEmitTransBlack_BlendOptFlag);
93 bool colorIsSolidWhite = (blendOpts & GrDrawState::kEmitCoverage_BlendOptFlag) ||
94 (!requiresAttributeColors && 0xffffffff == drawState.getColor());
95 if (colorIsTransBlack) {
96 desc->fColorInput = Desc::kTransBlack_ColorInput;
97 } else if (colorIsSolidWhite) {
98 desc->fColorInput = Desc::kSolidWhite_ColorInput;
99 } else if (GR_GL_NO_CONSTANT_ATTRIBUTES && !requiresAttributeColors) {
100 desc->fColorInput = Desc::kUniform_ColorInput;
101 } else {
102 desc->fColorInput = Desc::kAttribute_ColorInput;
103 }
104
105 bool covIsSolidWhite = !requiresAttributeCoverage && 0xffffffff == drawState.getCoverage();
106
107 if (skipCoverage) {
108 desc->fCoverageInput = Desc::kTransBlack_ColorInput;
109 } else if (covIsSolidWhite) {
110 desc->fCoverageInput = Desc::kSolidWhite_ColorInput;
111 } else if (GR_GL_NO_CONSTANT_ATTRIBUTES && !requiresAttributeCoverage) {
112 desc->fCoverageInput = Desc::kUniform_ColorInput;
113 } else {
114 desc->fCoverageInput = Desc::kAttribute_ColorInput;
115 }
116
117 int lastEnabledStage = -1;
118
jvanverth@google.comb8b705b2013-02-28 16:28:34 +0000119 if (!skipCoverage && (desc->fAttribBindings & GrDrawState::kEdge_AttribBindingsBit)) {
bsalomon@google.com91207482013-02-12 21:45:24 +0000120 desc->fVertexEdgeType = drawState.getVertexEdgeType();
121 desc->fDiscardIfOutsideEdge = drawState.getStencil().doesWrite();
122 } else {
123 // Use canonical values when edge-aa is not enabled to avoid program cache misses.
124 desc->fVertexEdgeType = GrDrawState::kHairLine_EdgeType;
125 desc->fDiscardIfOutsideEdge = false;
126 }
127
128 for (int s = 0; s < GrDrawState::kNumStages; ++s) {
129
130 bool skip = s < drawState.getFirstCoverageStage() ? skipColor : skipCoverage;
131 if (!skip && drawState.isStageEnabled(s)) {
132 lastEnabledStage = s;
133 const GrEffectRef& effect = *drawState.getStage(s).getEffect();
134 const GrBackendEffectFactory& factory = effect->getFactory();
135 desc->fEffectKeys[s] = factory.glEffectKey(drawState.getStage(s), gpu->glCaps());
136 } else {
137 desc->fEffectKeys[s] = 0;
138 }
139 }
140
141 desc->fDualSrcOutput = Desc::kNone_DualSrcOutput;
142
143 // Currently the experimental GS will only work with triangle prims (and it doesn't do anything
144 // other than pass through values from the VS to the FS anyway).
145#if GR_GL_EXPERIMENTAL_GS
146#if 0
147 desc->fExperimentalGS = gpu->getCaps().geometryShaderSupport();
148#else
149 desc->fExperimentalGS = false;
150#endif
151#endif
152
153 // We want to avoid generating programs with different "first cov stage" values when they would
154 // compute the same result. We set field in the desc to kNumStages when either there are no
155 // coverage stages or the distinction between coverage and color is immaterial.
156 int firstCoverageStage = GrDrawState::kNumStages;
157 desc->fFirstCoverageStage = GrDrawState::kNumStages;
158 bool hasCoverage = drawState.getFirstCoverageStage() <= lastEnabledStage;
159 if (hasCoverage) {
160 firstCoverageStage = drawState.getFirstCoverageStage();
161 }
162
163 // other coverage inputs
164 if (!hasCoverage) {
165 hasCoverage = requiresAttributeCoverage ||
jvanverth@google.comb8b705b2013-02-28 16:28:34 +0000166 (desc->fAttribBindings & GrDrawState::kEdge_AttribBindingsBit);
bsalomon@google.com91207482013-02-12 21:45:24 +0000167 }
168
169 if (hasCoverage) {
170 // color filter is applied between color/coverage computation
171 if (SkXfermode::kDst_Mode != desc->fColorFilterXfermode) {
172 desc->fFirstCoverageStage = firstCoverageStage;
173 }
174
175 if (gpu->getCaps().dualSourceBlendingSupport() &&
176 !(blendOpts & (GrDrawState::kEmitCoverage_BlendOptFlag |
177 GrDrawState::kCoverageAsAlpha_BlendOptFlag))) {
178 if (kZero_GrBlendCoeff == dstCoeff) {
179 // write the coverage value to second color
180 desc->fDualSrcOutput = Desc::kCoverage_DualSrcOutput;
181 desc->fFirstCoverageStage = firstCoverageStage;
182 } else if (kSA_GrBlendCoeff == dstCoeff) {
183 // SA dst coeff becomes 1-(1-SA)*coverage when dst is partially covered.
184 desc->fDualSrcOutput = Desc::kCoverageISA_DualSrcOutput;
185 desc->fFirstCoverageStage = firstCoverageStage;
186 } else if (kSC_GrBlendCoeff == dstCoeff) {
187 // SA dst coeff becomes 1-(1-SA)*coverage when dst is partially covered.
188 desc->fDualSrcOutput = Desc::kCoverageISC_DualSrcOutput;
189 desc->fFirstCoverageStage = firstCoverageStage;
190 }
191 }
192 }
jvanverth@google.comb8b705b2013-02-28 16:28:34 +0000193
194 desc->fPositionAttributeIndex = drawState.getAttribIndex(GrDrawState::kPosition_AttribIndex);
195 if (requiresAttributeColors) {
196 desc->fColorAttributeIndex = drawState.getAttribIndex(GrDrawState::kColor_AttribIndex);
197 } else {
198 desc->fColorAttributeIndex = GrDrawState::kColorOverrideAttribIndexValue;
199 }
200 if (requiresAttributeCoverage) {
201 desc->fCoverageAttributeIndex = drawState.getAttribIndex(GrDrawState::kCoverage_AttribIndex);
202 } else {
203 desc->fCoverageAttributeIndex = GrDrawState::kCoverageOverrideAttribIndexValue;
204 }
205 desc->fEdgeAttributeIndex = drawState.getAttribIndex(GrDrawState::kEdge_AttribIndex);
206 desc->fTexCoordAttributeIndex = drawState.getAttribIndex(GrDrawState::kTexCoord_AttribIndex);
207
208#if GR_DEBUG
209 // verify valid vertex attribute state
210 const GrVertexAttrib* vertexAttribs = drawState.getVertexAttribs();
211 GrAssert(desc->fPositionAttributeIndex < GrDrawState::kVertexAttribCnt);
212 GrAssert(kAttribLayouts[vertexAttribs[desc->fPositionAttributeIndex].fType].fCount == 2);
213 if (requiresAttributeColors) {
214 GrAssert(desc->fColorAttributeIndex < GrDrawState::kVertexAttribCnt);
215 GrAssert(kAttribLayouts[vertexAttribs[desc->fColorAttributeIndex].fType].fCount == 4);
216 }
217 if (requiresAttributeCoverage) {
218 GrAssert(desc->fCoverageAttributeIndex < GrDrawState::kVertexAttribCnt);
219 GrAssert(kAttribLayouts[vertexAttribs[desc->fCoverageAttributeIndex].fType].fCount == 4);
220 }
221 if (desc->fAttribBindings & GrDrawState::kEdge_AttribBindingsBit) {
222 GrAssert(desc->fEdgeAttributeIndex < GrDrawState::kVertexAttribCnt);
223 GrAssert(kAttribLayouts[vertexAttribs[desc->fEdgeAttributeIndex].fType].fCount == 4);
224 }
225 if (GrDrawState::AttributesBindExplicitTexCoords(desc->fAttribBindings)) {
226 GrAssert(desc->fTexCoordAttributeIndex < GrDrawState::kVertexAttribCnt);
227 GrAssert(kAttribLayouts[vertexAttribs[desc->fTexCoordAttributeIndex].fType].fCount == 2);
228 }
229#endif
bsalomon@google.com91207482013-02-12 21:45:24 +0000230}
231
robertphillips@google.com6177e692013-02-28 20:16:25 +0000232GrGLProgram* GrGLProgram::Create(const GrGLContext& gl,
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000233 const Desc& desc,
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000234 const GrEffectStage* stages[]) {
235 GrGLProgram* program = SkNEW_ARGS(GrGLProgram, (gl, desc, stages));
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000236 if (!program->succeeded()) {
237 delete program;
238 program = NULL;
239 }
240 return program;
241}
242
robertphillips@google.com6177e692013-02-28 20:16:25 +0000243GrGLProgram::GrGLProgram(const GrGLContext& gl,
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000244 const Desc& desc,
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000245 const GrEffectStage* stages[])
robertphillips@google.com6177e692013-02-28 20:16:25 +0000246: fContext(gl)
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000247, fUniformManager(gl) {
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000248 fDesc = desc;
249 fVShaderID = 0;
250 fGShaderID = 0;
251 fFShaderID = 0;
252 fProgramID = 0;
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000253
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000254 fColor = GrColor_ILLEGAL;
255 fColorFilterColor = GrColor_ILLEGAL;
256
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000257 for (int s = 0; s < GrDrawState::kNumStages; ++s) {
bsalomon@google.com46fba0d2012-10-25 21:42:05 +0000258 fEffects[s] = NULL;
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000259 }
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000260
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000261 this->genProgram(stages);
junov@google.comf93e7172011-03-31 21:26:24 +0000262}
263
264GrGLProgram::~GrGLProgram() {
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000265 if (fVShaderID) {
266 GL_CALL(DeleteShader(fVShaderID));
267 }
268 if (fGShaderID) {
269 GL_CALL(DeleteShader(fGShaderID));
270 }
271 if (fFShaderID) {
272 GL_CALL(DeleteShader(fFShaderID));
273 }
274 if (fProgramID) {
275 GL_CALL(DeleteProgram(fProgramID));
276 }
277
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000278 for (int i = 0; i < GrDrawState::kNumStages; ++i) {
bsalomon@google.com46fba0d2012-10-25 21:42:05 +0000279 delete fEffects[i];
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000280 }
junov@google.comf93e7172011-03-31 21:26:24 +0000281}
282
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000283void GrGLProgram::abandon() {
284 fVShaderID = 0;
285 fGShaderID = 0;
286 fFShaderID = 0;
287 fProgramID = 0;
288}
289
tomhudson@google.com0d3f1fb2011-06-01 19:27:31 +0000290void GrGLProgram::overrideBlend(GrBlendCoeff* srcCoeff,
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000291 GrBlendCoeff* dstCoeff) const {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000292 switch (fDesc.fDualSrcOutput) {
293 case Desc::kNone_DualSrcOutput:
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000294 break;
295 // the prog will write a coverage value to the secondary
296 // output and the dst is blended by one minus that value.
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000297 case Desc::kCoverage_DualSrcOutput:
298 case Desc::kCoverageISA_DualSrcOutput:
299 case Desc::kCoverageISC_DualSrcOutput:
bsalomon@google.com47059542012-06-06 20:51:20 +0000300 *dstCoeff = (GrBlendCoeff)GrGpu::kIS2C_GrBlendCoeff;
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000301 break;
302 default:
303 GrCrash("Unexpected dual source blend output");
304 break;
305 }
306}
307
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000308namespace {
309
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000310// given two blend coeffecients determine whether the src
311// and/or dst computation can be omitted.
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000312inline void need_blend_inputs(SkXfermode::Coeff srcCoeff,
313 SkXfermode::Coeff dstCoeff,
314 bool* needSrcValue,
315 bool* needDstValue) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000316 if (SkXfermode::kZero_Coeff == srcCoeff) {
317 switch (dstCoeff) {
318 // these all read the src
319 case SkXfermode::kSC_Coeff:
320 case SkXfermode::kISC_Coeff:
321 case SkXfermode::kSA_Coeff:
322 case SkXfermode::kISA_Coeff:
323 *needSrcValue = true;
324 break;
325 default:
326 *needSrcValue = false;
327 break;
328 }
329 } else {
330 *needSrcValue = true;
331 }
332 if (SkXfermode::kZero_Coeff == dstCoeff) {
333 switch (srcCoeff) {
334 // these all read the dst
335 case SkXfermode::kDC_Coeff:
336 case SkXfermode::kIDC_Coeff:
337 case SkXfermode::kDA_Coeff:
338 case SkXfermode::kIDA_Coeff:
339 *needDstValue = true;
340 break;
341 default:
342 *needDstValue = false;
343 break;
344 }
345 } else {
346 *needDstValue = true;
Scroggo97c88c22011-05-11 14:05:25 +0000347 }
348}
349
350/**
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000351 * Create a blend_coeff * value string to be used in shader code. Sets empty
352 * string if result is trivially zero.
353 */
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000354inline void blend_term_string(SkString* str, SkXfermode::Coeff coeff,
355 const char* src, const char* dst,
356 const char* value) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000357 switch (coeff) {
358 case SkXfermode::kZero_Coeff: /** 0 */
359 *str = "";
360 break;
361 case SkXfermode::kOne_Coeff: /** 1 */
362 *str = value;
363 break;
364 case SkXfermode::kSC_Coeff:
365 str->printf("(%s * %s)", src, value);
366 break;
367 case SkXfermode::kISC_Coeff:
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000368 str->printf("((%s - %s) * %s)", GrGLSLOnesVecf(4), src, value);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000369 break;
370 case SkXfermode::kDC_Coeff:
371 str->printf("(%s * %s)", dst, value);
372 break;
373 case SkXfermode::kIDC_Coeff:
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000374 str->printf("((%s - %s) * %s)", GrGLSLOnesVecf(4), dst, value);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000375 break;
376 case SkXfermode::kSA_Coeff: /** src alpha */
377 str->printf("(%s.a * %s)", src, value);
378 break;
379 case SkXfermode::kISA_Coeff: /** inverse src alpha (i.e. 1 - sa) */
380 str->printf("((1.0 - %s.a) * %s)", src, value);
381 break;
382 case SkXfermode::kDA_Coeff: /** dst alpha */
383 str->printf("(%s.a * %s)", dst, value);
384 break;
385 case SkXfermode::kIDA_Coeff: /** inverse dst alpha (i.e. 1 - da) */
386 str->printf("((1.0 - %s.a) * %s)", dst, value);
387 break;
388 default:
389 GrCrash("Unexpected xfer coeff.");
390 break;
391 }
392}
393/**
Scroggo97c88c22011-05-11 14:05:25 +0000394 * Adds a line to the fragment shader code which modifies the color by
395 * the specified color filter.
396 */
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000397void add_color_filter(SkString* fsCode, const char * outputVar,
398 SkXfermode::Coeff uniformCoeff,
399 SkXfermode::Coeff colorCoeff,
400 const char* filterColor,
401 const char* inColor) {
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000402 SkString colorStr, constStr;
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000403 blend_term_string(&colorStr, colorCoeff, filterColor, inColor, inColor);
404 blend_term_string(&constStr, uniformCoeff, filterColor, inColor, filterColor);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000405
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000406 fsCode->appendf("\t%s = ", outputVar);
407 GrGLSLAdd4f(fsCode, colorStr.c_str(), constStr.c_str());
408 fsCode->append(";\n");
Scroggo97c88c22011-05-11 14:05:25 +0000409}
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000410}
Scroggo97c88c22011-05-11 14:05:25 +0000411
bsalomon@google.comd4726202012-08-03 14:34:46 +0000412bool GrGLProgram::genEdgeCoverage(SkString* coverageVar,
bsalomon@google.com706f6682012-10-23 14:53:55 +0000413 GrGLShaderBuilder* builder) const {
jvanverth@google.comb8b705b2013-02-28 16:28:34 +0000414 if (fDesc.fAttribBindings & GrDrawState::kEdge_AttribBindingsBit) {
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000415 const char *vsName, *fsName;
bsalomon@google.com706f6682012-10-23 14:53:55 +0000416 builder->addVarying(kVec4f_GrSLType, "Edge", &vsName, &fsName);
417 builder->fVSAttrs.push_back().set(kVec4f_GrSLType,
418 GrGLShaderVar::kAttribute_TypeModifier,
419 EDGE_ATTR_NAME);
420 builder->fVSCode.appendf("\t%s = " EDGE_ATTR_NAME ";\n", vsName);
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000421 switch (fDesc.fVertexEdgeType) {
bsalomon@google.com93c96602012-04-27 13:05:21 +0000422 case GrDrawState::kHairLine_EdgeType:
bsalomon@google.com706f6682012-10-23 14:53:55 +0000423 builder->fFSCode.appendf("\tfloat edgeAlpha = abs(dot(vec3(%s.xy,1), %s.xyz));\n", builder->fragmentPosition(), fsName);
424 builder->fFSCode.append("\tedgeAlpha = max(1.0 - edgeAlpha, 0.0);\n");
bsalomon@google.com93c96602012-04-27 13:05:21 +0000425 break;
426 case GrDrawState::kQuad_EdgeType:
bsalomon@google.com706f6682012-10-23 14:53:55 +0000427 builder->fFSCode.append("\tfloat edgeAlpha;\n");
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000428 // keep the derivative instructions outside the conditional
bsalomon@google.com706f6682012-10-23 14:53:55 +0000429 builder->fFSCode.appendf("\tvec2 duvdx = dFdx(%s.xy);\n", fsName);
430 builder->fFSCode.appendf("\tvec2 duvdy = dFdy(%s.xy);\n", fsName);
431 builder->fFSCode.appendf("\tif (%s.z > 0.0 && %s.w > 0.0) {\n", fsName, fsName);
bsalomon@google.com9aed1142012-01-30 14:28:39 +0000432 // today we know z and w are in device space. We could use derivatives
bsalomon@google.com706f6682012-10-23 14:53:55 +0000433 builder->fFSCode.appendf("\t\tedgeAlpha = min(min(%s.z, %s.w) + 0.5, 1.0);\n", fsName, fsName);
434 builder->fFSCode.append ("\t} else {\n");
435 builder->fFSCode.appendf("\t\tvec2 gF = vec2(2.0*%s.x*duvdx.x - duvdx.y,\n"
436 "\t\t 2.0*%s.x*duvdy.x - duvdy.y);\n",
437 fsName, fsName);
438 builder->fFSCode.appendf("\t\tedgeAlpha = (%s.x*%s.x - %s.y);\n", fsName, fsName, fsName);
439 builder->fFSCode.append("\t\tedgeAlpha = clamp(0.5 - edgeAlpha / length(gF), 0.0, 1.0);\n"
440 "\t}\n");
robertphillips@google.com6177e692013-02-28 20:16:25 +0000441 if (kES2_GrGLBinding == fContext.info().binding()) {
bsalomon@google.com706f6682012-10-23 14:53:55 +0000442 builder->fHeader.printf("#extension GL_OES_standard_derivatives: enable\n");
bsalomon@google.com9aed1142012-01-30 14:28:39 +0000443 }
bsalomon@google.com93c96602012-04-27 13:05:21 +0000444 break;
445 case GrDrawState::kHairQuad_EdgeType:
bsalomon@google.com706f6682012-10-23 14:53:55 +0000446 builder->fFSCode.appendf("\tvec2 duvdx = dFdx(%s.xy);\n", fsName);
447 builder->fFSCode.appendf("\tvec2 duvdy = dFdy(%s.xy);\n", fsName);
448 builder->fFSCode.appendf("\tvec2 gF = vec2(2.0*%s.x*duvdx.x - duvdx.y,\n"
449 "\t 2.0*%s.x*duvdy.x - duvdy.y);\n",
450 fsName, fsName);
451 builder->fFSCode.appendf("\tfloat edgeAlpha = (%s.x*%s.x - %s.y);\n", fsName, fsName, fsName);
452 builder->fFSCode.append("\tedgeAlpha = sqrt(edgeAlpha*edgeAlpha / dot(gF, gF));\n");
453 builder->fFSCode.append("\tedgeAlpha = max(1.0 - edgeAlpha, 0.0);\n");
robertphillips@google.com6177e692013-02-28 20:16:25 +0000454 if (kES2_GrGLBinding == fContext.info().binding()) {
bsalomon@google.com706f6682012-10-23 14:53:55 +0000455 builder->fHeader.printf("#extension GL_OES_standard_derivatives: enable\n");
bsalomon@google.com66105672011-09-15 15:12:00 +0000456 }
bsalomon@google.com93c96602012-04-27 13:05:21 +0000457 break;
458 case GrDrawState::kCircle_EdgeType:
bsalomon@google.com706f6682012-10-23 14:53:55 +0000459 builder->fFSCode.append("\tfloat edgeAlpha;\n");
460 builder->fFSCode.appendf("\tfloat d = distance(%s.xy, %s.xy);\n", builder->fragmentPosition(), fsName);
461 builder->fFSCode.appendf("\tfloat outerAlpha = smoothstep(d - 0.5, d + 0.5, %s.z);\n", fsName);
462 builder->fFSCode.appendf("\tfloat innerAlpha = %s.w == 0.0 ? 1.0 : smoothstep(%s.w - 0.5, %s.w + 0.5, d);\n", fsName, fsName, fsName);
463 builder->fFSCode.append("\tedgeAlpha = outerAlpha * innerAlpha;\n");
bsalomon@google.com93c96602012-04-27 13:05:21 +0000464 break;
jvanverth@google.com46d3d392013-01-22 13:34:01 +0000465 case GrDrawState::kEllipse_EdgeType:
466 builder->fFSCode.append("\tfloat edgeAlpha;\n");
467 builder->fFSCode.appendf("\tvec2 offset = (%s.xy - %s.xy);\n", builder->fragmentPosition(), fsName);
468 builder->fFSCode.appendf("\toffset.y *= %s.w;\n", fsName);
469 builder->fFSCode.append("\tfloat d = length(offset);\n");
470 builder->fFSCode.appendf("\tedgeAlpha = smoothstep(d - 0.5, d + 0.5, %s.z);\n", fsName);
471 break;
bsalomon@google.com93c96602012-04-27 13:05:21 +0000472 default:
473 GrCrash("Unknown Edge Type!");
474 break;
bsalomon@google.com66105672011-09-15 15:12:00 +0000475 }
bsalomon@google.com45a15f52012-12-10 19:10:17 +0000476 if (fDesc.fDiscardIfOutsideEdge) {
robertphillips@google.comef818a42012-12-10 21:05:19 +0000477 builder->fFSCode.appendf("\tif (edgeAlpha <= 0.0) {\n\t\tdiscard;\n\t}\n");
bsalomon@google.com45a15f52012-12-10 19:10:17 +0000478 }
bsalomon@google.com66105672011-09-15 15:12:00 +0000479 *coverageVar = "edgeAlpha";
bsalomon@google.comd4726202012-08-03 14:34:46 +0000480 return true;
bsalomon@google.com66105672011-09-15 15:12:00 +0000481 } else {
482 coverageVar->reset();
bsalomon@google.comd4726202012-08-03 14:34:46 +0000483 return false;
bsalomon@google.com66105672011-09-15 15:12:00 +0000484 }
485}
486
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000487void GrGLProgram::genInputColor(GrGLShaderBuilder* builder, SkString* inColor) {
488 switch (fDesc.fColorInput) {
489 case GrGLProgram::Desc::kAttribute_ColorInput: {
490 builder->fVSAttrs.push_back().set(kVec4f_GrSLType,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000491 GrGLShaderVar::kAttribute_TypeModifier,
492 COL_ATTR_NAME);
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000493 const char *vsName, *fsName;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000494 builder->addVarying(kVec4f_GrSLType, "Color", &vsName, &fsName);
495 builder->fVSCode.appendf("\t%s = " COL_ATTR_NAME ";\n", vsName);
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000496 *inColor = fsName;
497 } break;
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000498 case GrGLProgram::Desc::kUniform_ColorInput: {
499 const char* name;
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000500 fUniformHandles.fColorUni = builder->addUniform(GrGLShaderBuilder::kFragment_ShaderType,
501 kVec4f_GrSLType, "Color", &name);
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000502 *inColor = name;
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000503 break;
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000504 }
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000505 case GrGLProgram::Desc::kTransBlack_ColorInput:
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000506 GrAssert(!"needComputedColor should be false.");
507 break;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000508 case GrGLProgram::Desc::kSolidWhite_ColorInput:
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000509 break;
510 default:
511 GrCrash("Unknown color type.");
512 break;
513 }
514}
515
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000516void GrGLProgram::genUniformCoverage(GrGLShaderBuilder* builder, SkString* inOutCoverage) {
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000517 const char* covUniName;
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000518 fUniformHandles.fCoverageUni = builder->addUniform(GrGLShaderBuilder::kFragment_ShaderType,
519 kVec4f_GrSLType, "Coverage", &covUniName);
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000520 if (inOutCoverage->size()) {
521 builder->fFSCode.appendf("\tvec4 uniCoverage = %s * %s;\n",
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000522 covUniName, inOutCoverage->c_str());
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000523 *inOutCoverage = "uniCoverage";
524 } else {
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000525 *inOutCoverage = covUniName;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000526 }
527}
528
529namespace {
530void gen_attribute_coverage(GrGLShaderBuilder* segments,
531 SkString* inOutCoverage) {
tomhudson@google.com168e6342012-04-18 17:49:20 +0000532 segments->fVSAttrs.push_back().set(kVec4f_GrSLType,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000533 GrGLShaderVar::kAttribute_TypeModifier,
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000534 COV_ATTR_NAME);
535 const char *vsName, *fsName;
tomhudson@google.com23cb2292012-05-30 18:26:03 +0000536 segments->addVarying(kVec4f_GrSLType, "Coverage", &vsName, &fsName);
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000537 segments->fVSCode.appendf("\t%s = " COV_ATTR_NAME ";\n", vsName);
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000538 if (inOutCoverage->size()) {
539 segments->fFSCode.appendf("\tvec4 attrCoverage = %s * %s;\n",
540 fsName, inOutCoverage->c_str());
541 *inOutCoverage = "attrCoverage";
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000542 } else {
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000543 *inOutCoverage = fsName;
544 }
545}
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000546}
547
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000548void GrGLProgram::genGeometryShader(GrGLShaderBuilder* segments) const {
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000549#if GR_GL_EXPERIMENTAL_GS
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000550 if (fDesc.fExperimentalGS) {
robertphillips@google.com6177e692013-02-28 20:16:25 +0000551 GrAssert(fContext.info().glslGeneration() >= k150_GrGLSLGeneration);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000552 segments->fGSHeader.append("layout(triangles) in;\n"
553 "layout(triangle_strip, max_vertices = 6) out;\n");
bsalomon@google.comd7bafb72012-10-22 14:30:50 +0000554 segments->fGSCode.append("\tfor (int i = 0; i < 3; ++i) {\n"
555 "\t\tgl_Position = gl_in[i].gl_Position;\n");
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000556 if (fDesc.fEmitsPointSize) {
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000557 segments->fGSCode.append("\t\tgl_PointSize = 1.0;\n");
558 }
559 GrAssert(segments->fGSInputs.count() == segments->fGSOutputs.count());
560 int count = segments->fGSInputs.count();
561 for (int i = 0; i < count; ++i) {
562 segments->fGSCode.appendf("\t\t%s = %s[i];\n",
563 segments->fGSOutputs[i].getName().c_str(),
564 segments->fGSInputs[i].getName().c_str());
565 }
566 segments->fGSCode.append("\t\tEmitVertex();\n"
567 "\t}\n"
bsalomon@google.comd7bafb72012-10-22 14:30:50 +0000568 "\tEndPrimitive();\n");
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000569 }
570#endif
571}
572
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000573const char* GrGLProgram::adjustInColor(const SkString& inColor) const {
senorblanco@chromium.orgb3a39b52012-01-05 18:28:56 +0000574 if (inColor.size()) {
575 return inColor.c_str();
576 } else {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000577 if (Desc::kSolidWhite_ColorInput == fDesc.fColorInput) {
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000578 return GrGLSLOnesVecf(4);
senorblanco@chromium.orgb3a39b52012-01-05 18:28:56 +0000579 } else {
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000580 return GrGLSLZerosVecf(4);
senorblanco@chromium.orgb3a39b52012-01-05 18:28:56 +0000581 }
582 }
583}
584
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000585namespace {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000586// prints a shader using params similar to glShaderSource
587void print_shader(GrGLint stringCnt,
588 const GrGLchar** strings,
589 GrGLint* stringLengths) {
590 for (int i = 0; i < stringCnt; ++i) {
591 if (NULL == stringLengths || stringLengths[i] < 0) {
592 GrPrintf(strings[i]);
593 } else {
594 GrPrintf("%.*s", stringLengths[i], strings[i]);
595 }
596 }
597}
598
599// 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 +0000600GrGLuint compile_shader(const GrGLContext& gl,
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000601 GrGLenum type,
602 int stringCnt,
603 const char** strings,
604 int* stringLengths) {
605 SK_TRACE_EVENT1("GrGLProgram::CompileShader",
606 "stringCount", SkStringPrintf("%i", stringCnt).c_str());
607
608 GrGLuint shader;
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000609 GR_GL_CALL_RET(gl.interface(), shader, CreateShader(type));
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000610 if (0 == shader) {
611 return 0;
612 }
613
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000614 const GrGLInterface* gli = gl.interface();
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000615 GrGLint compiled = GR_GL_INIT_ZERO;
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000616 GR_GL_CALL(gli, ShaderSource(shader, stringCnt, strings, stringLengths));
617 GR_GL_CALL(gli, CompileShader(shader));
618 GR_GL_CALL(gli, GetShaderiv(shader, GR_GL_COMPILE_STATUS, &compiled));
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000619
620 if (!compiled) {
621 GrGLint infoLen = GR_GL_INIT_ZERO;
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000622 GR_GL_CALL(gli, GetShaderiv(shader, GR_GL_INFO_LOG_LENGTH, &infoLen));
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000623 SkAutoMalloc log(sizeof(char)*(infoLen+1)); // outside if for debugger
624 if (infoLen > 0) {
625 // retrieve length even though we don't need it to workaround bug in chrome cmd buffer
626 // param validation.
627 GrGLsizei length = GR_GL_INIT_ZERO;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000628 GR_GL_CALL(gli, GetShaderInfoLog(shader, infoLen+1,
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000629 &length, (char*)log.get()));
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000630 print_shader(stringCnt, strings, stringLengths);
631 GrPrintf("\n%s", log.get());
632 }
633 GrAssert(!"Shader compilation failed!");
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000634 GR_GL_CALL(gli, DeleteShader(shader));
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000635 return 0;
636 }
637 return shader;
638}
639
640// helper version of above for when shader is already flattened into a single SkString
robertphillips@google.com6177e692013-02-28 20:16:25 +0000641GrGLuint compile_shader(const GrGLContext& gl, GrGLenum type, const SkString& shader) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000642 const GrGLchar* str = shader.c_str();
643 int length = shader.size();
644 return compile_shader(gl, type, 1, &str, &length);
645}
646
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000647}
648
649// compiles all the shaders from builder and stores the shader IDs
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000650bool GrGLProgram::compileShaders(const GrGLShaderBuilder& builder) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000651
652 SkString shader;
653
654 builder.getShader(GrGLShaderBuilder::kVertex_ShaderType, &shader);
humper@google.com7af56be2013-01-14 18:49:19 +0000655 if (c_PrintShaders) {
656 GrPrintf(shader.c_str());
657 GrPrintf("\n");
658 }
skia.committer@gmail.com7fc0e0a2013-01-15 02:01:40 +0000659
robertphillips@google.com6177e692013-02-28 20:16:25 +0000660 if (!(fVShaderID = compile_shader(fContext, GR_GL_VERTEX_SHADER, shader))) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000661 return false;
662 }
663
664 if (builder.fUsesGS) {
665 builder.getShader(GrGLShaderBuilder::kGeometry_ShaderType, &shader);
humper@google.com7af56be2013-01-14 18:49:19 +0000666 if (c_PrintShaders) {
667 GrPrintf(shader.c_str());
668 GrPrintf("\n");
669 }
robertphillips@google.com6177e692013-02-28 20:16:25 +0000670 if (!(fGShaderID = compile_shader(fContext, GR_GL_GEOMETRY_SHADER, shader))) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000671 return false;
672 }
673 } else {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000674 fGShaderID = 0;
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000675 }
676
677 builder.getShader(GrGLShaderBuilder::kFragment_ShaderType, &shader);
humper@google.com7af56be2013-01-14 18:49:19 +0000678 if (c_PrintShaders) {
679 GrPrintf(shader.c_str());
680 GrPrintf("\n");
681 }
robertphillips@google.com6177e692013-02-28 20:16:25 +0000682 if (!(fFShaderID = compile_shader(fContext, GR_GL_FRAGMENT_SHADER, shader))) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000683 return false;
684 }
685
686 return true;
687}
688
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000689bool GrGLProgram::genProgram(const GrEffectStage* stages[]) {
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000690 GrAssert(0 == fProgramID);
691
robertphillips@google.com6177e692013-02-28 20:16:25 +0000692 GrGLShaderBuilder builder(fContext.info(), fUniformManager);
jvanverth@google.comb8b705b2013-02-28 16:28:34 +0000693 const GrAttribBindings& attribBindings = fDesc.fAttribBindings;
junov@google.comf93e7172011-03-31 21:26:24 +0000694
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000695#if GR_GL_EXPERIMENTAL_GS
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000696 builder.fUsesGS = fDesc.fExperimentalGS;
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000697#endif
698
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000699 SkXfermode::Coeff colorCoeff, uniformCoeff;
700 // The rest of transfer mode color filters have not been implemented
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000701 if (fDesc.fColorFilterXfermode < SkXfermode::kCoeffModesCnt) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000702 GR_DEBUGCODE(bool success =)
tomhudson@google.com0d831722011-06-02 15:37:14 +0000703 SkXfermode::ModeAsCoeff(static_cast<SkXfermode::Mode>
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000704 (fDesc.fColorFilterXfermode),
tomhudson@google.com0d831722011-06-02 15:37:14 +0000705 &uniformCoeff, &colorCoeff);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000706 GR_DEBUGASSERT(success);
707 } else {
708 colorCoeff = SkXfermode::kOne_Coeff;
709 uniformCoeff = SkXfermode::kZero_Coeff;
710 }
711
bsalomon@google.com67e78c92012-10-17 13:36:14 +0000712 // no need to do the color filter if coverage is 0. The output color is scaled by the coverage.
713 // All the dual source outputs are scaled by the coverage as well.
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000714 if (Desc::kTransBlack_ColorInput == fDesc.fCoverageInput) {
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000715 colorCoeff = SkXfermode::kZero_Coeff;
716 uniformCoeff = SkXfermode::kZero_Coeff;
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000717 }
718
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000719 // If we know the final color is going to be all zeros then we can
bsalomon@google.com67e78c92012-10-17 13:36:14 +0000720 // simplify the color filter coefficients. needComputedColor will then
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000721 // come out false below.
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000722 if (Desc::kTransBlack_ColorInput == fDesc.fColorInput) {
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000723 colorCoeff = SkXfermode::kZero_Coeff;
724 if (SkXfermode::kDC_Coeff == uniformCoeff ||
725 SkXfermode::kDA_Coeff == uniformCoeff) {
726 uniformCoeff = SkXfermode::kZero_Coeff;
727 } else if (SkXfermode::kIDC_Coeff == uniformCoeff ||
728 SkXfermode::kIDA_Coeff == uniformCoeff) {
729 uniformCoeff = SkXfermode::kOne_Coeff;
730 }
731 }
732
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000733 bool needColorFilterUniform;
734 bool needComputedColor;
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000735 need_blend_inputs(uniformCoeff, colorCoeff,
736 &needColorFilterUniform, &needComputedColor);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000737
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000738 // the dual source output has no canonical var name, have to
739 // declare an output, which is incompatible with gl_FragColor/gl_FragData.
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000740 bool dualSourceOutputWritten = false;
robertphillips@google.com6177e692013-02-28 20:16:25 +0000741 builder.fHeader.append(GrGetGLSLVersionDecl(fContext.info().binding(),
742 fContext.info().glslGeneration()));
bsalomon@google.come55fd0f2012-02-10 15:56:06 +0000743
744 GrGLShaderVar colorOutput;
robertphillips@google.com6177e692013-02-28 20:16:25 +0000745 bool isColorDeclared = GrGLSLSetupFSColorOuput(fContext.info().glslGeneration(),
bsalomon@google.come55fd0f2012-02-10 15:56:06 +0000746 declared_color_output_name(),
747 &colorOutput);
748 if (isColorDeclared) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000749 builder.fFSOutputs.push_back(colorOutput);
bsalomon@google.come55fd0f2012-02-10 15:56:06 +0000750 }
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000751
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000752 const char* viewMName;
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000753 fUniformHandles.fViewMatrixUni = builder.addUniform(GrGLShaderBuilder::kVertex_ShaderType,
754 kMat33f_GrSLType, "ViewM", &viewMName);
bsalomon@google.com341767c2012-05-11 20:47:39 +0000755
skia.committer@gmail.come862d162012-10-31 02:01:18 +0000756
bsalomon@google.com17504f52012-10-30 12:34:25 +0000757 builder.fVSCode.appendf("\tvec3 pos3 = %s * vec3(%s, 1);\n"
bsalomon@google.comd7bafb72012-10-22 14:30:50 +0000758 "\tgl_Position = vec4(pos3.xy, 0, pos3.z);\n",
bsalomon@google.com17504f52012-10-30 12:34:25 +0000759 viewMName, builder.positionAttribute().getName().c_str());
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000760
bsalomon@google.com91961302011-05-09 18:39:58 +0000761 // incoming color to current stage being processed.
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000762 SkString inColor;
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000763
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000764 if (needComputedColor) {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000765 this->genInputColor(&builder, &inColor);
junov@google.comf93e7172011-03-31 21:26:24 +0000766 }
767
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000768 // we output point size in the GS if present
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000769 if (fDesc.fEmitsPointSize && !builder.fUsesGS){
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000770 builder.fVSCode.append("\tgl_PointSize = 1.0;\n");
junov@google.comf93e7172011-03-31 21:26:24 +0000771 }
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000772
junov@google.comf93e7172011-03-31 21:26:24 +0000773 // add texture coordinates that are used to the list of vertex attr decls
jvanverth@google.comb8b705b2013-02-28 16:28:34 +0000774 if (GrDrawState::AttributesBindExplicitTexCoords(attribBindings)) {
jvanverth@google.com39768252013-02-14 15:25:44 +0000775 builder.fVSAttrs.push_back().set(kVec2f_GrSLType,
776 GrGLShaderVar::kAttribute_TypeModifier,
777 TEX_ATTR_NAME);
junov@google.comf93e7172011-03-31 21:26:24 +0000778 }
779
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000780 ///////////////////////////////////////////////////////////////////////////
781 // compute the final color
Scroggo97c88c22011-05-11 14:05:25 +0000782
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000783 // if we have color stages string them together, feeding the output color
784 // of each to the next and generating code for each stage.
785 if (needComputedColor) {
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000786 SkString outColor;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000787 for (int s = 0; s < fDesc.fFirstCoverageStage; ++s) {
bsalomon@google.comdbe49f72012-11-05 16:36:02 +0000788 if (GrGLEffect::kNoEffectKey != fDesc.fEffectKeys[s]) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000789 // create var to hold stage result
790 outColor = "color";
791 outColor.appendS32(s);
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000792 builder.fFSCode.appendf("\tvec4 %s;\n", outColor.c_str());
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000793
794 const char* inCoords;
795 // figure out what our input coords are
jvanverth@google.comb8b705b2013-02-28 16:28:34 +0000796 if (!GrDrawState::StageBindsExplicitTexCoords(attribBindings, s)) {
bsalomon@google.com17504f52012-10-30 12:34:25 +0000797 inCoords = builder.positionAttribute().c_str();
Scroggo97c88c22011-05-11 14:05:25 +0000798 } else {
tomhudson@google.comb213ed82012-06-25 15:22:12 +0000799 // must have input tex coordinates if stage is enabled.
jvanverth@google.com39768252013-02-14 15:25:44 +0000800 inCoords = TEX_ATTR_NAME;
junov@google.comf93e7172011-03-31 21:26:24 +0000801 }
802
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000803 builder.setCurrentStage(s);
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000804 fEffects[s] = builder.createAndEmitGLEffect(*stages[s],
805 fDesc.fEffectKeys[s],
806 inColor.size() ? inColor.c_str() : NULL,
807 outColor.c_str(),
808 inCoords,
809 &fUniformHandles.fSamplerUnis[s]);
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000810 builder.setNonStage();
junov@google.comf93e7172011-03-31 21:26:24 +0000811 inColor = outColor;
junov@google.comf93e7172011-03-31 21:26:24 +0000812 }
813 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000814 }
Scroggo97c88c22011-05-11 14:05:25 +0000815
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000816 // if have all ones or zeros for the "dst" input to the color filter then we
817 // may be able to make additional optimizations.
818 if (needColorFilterUniform && needComputedColor && !inColor.size()) {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000819 GrAssert(Desc::kSolidWhite_ColorInput == fDesc.fColorInput);
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000820 bool uniformCoeffIsZero = SkXfermode::kIDC_Coeff == uniformCoeff ||
821 SkXfermode::kIDA_Coeff == uniformCoeff;
822 if (uniformCoeffIsZero) {
823 uniformCoeff = SkXfermode::kZero_Coeff;
824 bool bogus;
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000825 need_blend_inputs(SkXfermode::kZero_Coeff, colorCoeff,
826 &needColorFilterUniform, &bogus);
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000827 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000828 }
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000829 const char* colorFilterColorUniName = NULL;
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000830 if (needColorFilterUniform) {
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000831 fUniformHandles.fColorFilterUni = builder.addUniform(
832 GrGLShaderBuilder::kFragment_ShaderType,
833 kVec4f_GrSLType, "FilterColor",
834 &colorFilterColorUniName);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000835 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000836 bool wroteFragColorZero = false;
tomhudson@google.com0d831722011-06-02 15:37:14 +0000837 if (SkXfermode::kZero_Coeff == uniformCoeff &&
bsalomon@google.com67e78c92012-10-17 13:36:14 +0000838 SkXfermode::kZero_Coeff == colorCoeff) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000839 builder.fFSCode.appendf("\t%s = %s;\n",
840 colorOutput.getName().c_str(),
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000841 GrGLSLZerosVecf(4));
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000842 wroteFragColorZero = true;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000843 } else if (SkXfermode::kDst_Mode != fDesc.fColorFilterXfermode) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000844 builder.fFSCode.append("\tvec4 filteredColor;\n");
senorblanco@chromium.orgb3a39b52012-01-05 18:28:56 +0000845 const char* color = adjustInColor(inColor);
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000846 add_color_filter(&builder.fFSCode, "filteredColor", uniformCoeff,
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000847 colorCoeff, colorFilterColorUniName, color);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000848 inColor = "filteredColor";
849 }
850
851 ///////////////////////////////////////////////////////////////////////////
852 // compute the partial coverage (coverage stages and edge aa)
853
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000854 SkString inCoverage;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000855 bool coverageIsZero = Desc::kTransBlack_ColorInput == fDesc.fCoverageInput;
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000856 // we don't need to compute coverage at all if we know the final shader
857 // output will be zero and we don't have a dual src blend output.
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000858 if (!wroteFragColorZero || Desc::kNone_DualSrcOutput != fDesc.fDualSrcOutput) {
bsalomon@google.com66105672011-09-15 15:12:00 +0000859
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000860 if (!coverageIsZero) {
bsalomon@google.comd4726202012-08-03 14:34:46 +0000861 bool inCoverageIsScalar = this->genEdgeCoverage(&inCoverage, &builder);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000862
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000863 switch (fDesc.fCoverageInput) {
864 case Desc::kSolidWhite_ColorInput:
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000865 // empty string implies solid white
866 break;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000867 case Desc::kAttribute_ColorInput:
868 gen_attribute_coverage(&builder, &inCoverage);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000869 inCoverageIsScalar = false;
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000870 break;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000871 case Desc::kUniform_ColorInput:
872 this->genUniformCoverage(&builder, &inCoverage);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000873 inCoverageIsScalar = false;
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000874 break;
875 default:
876 GrCrash("Unexpected input coverage.");
877 }
bsalomon@google.coma3108262011-10-10 14:08:47 +0000878
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000879 SkString outCoverage;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000880 const int& startStage = fDesc.fFirstCoverageStage;
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000881 for (int s = startStage; s < GrDrawState::kNumStages; ++s) {
bsalomon@google.comdbe49f72012-11-05 16:36:02 +0000882 if (fDesc.fEffectKeys[s]) {
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000883 // create var to hold stage output
884 outCoverage = "coverage";
885 outCoverage.appendS32(s);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000886 builder.fFSCode.appendf("\tvec4 %s;\n", outCoverage.c_str());
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000887
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000888 const char* inCoords;
889 // figure out what our input coords are
jvanverth@google.comb8b705b2013-02-28 16:28:34 +0000890 if (!GrDrawState::StageBindsExplicitTexCoords(attribBindings, s)) {
bsalomon@google.com17504f52012-10-30 12:34:25 +0000891 inCoords = builder.positionAttribute().c_str();
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000892 } else {
tomhudson@google.comb213ed82012-06-25 15:22:12 +0000893 // must have input tex coordinates if stage is
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000894 // enabled.
jvanverth@google.com39768252013-02-14 15:25:44 +0000895 inCoords = TEX_ATTR_NAME;
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000896 }
897
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000898 // stages don't know how to deal with a scalar input. (Maybe they should. We
bsalomon@google.comd4726202012-08-03 14:34:46 +0000899 // could pass a GrGLShaderVar)
900 if (inCoverageIsScalar) {
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000901 builder.fFSCode.appendf("\tvec4 %s4 = vec4(%s);\n",
bsalomon@google.comd4726202012-08-03 14:34:46 +0000902 inCoverage.c_str(), inCoverage.c_str());
903 inCoverage.append("4");
904 }
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000905 builder.setCurrentStage(s);
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000906 fEffects[s] = builder.createAndEmitGLEffect(
907 *stages[s],
908 fDesc.fEffectKeys[s],
909 inCoverage.size() ? inCoverage.c_str() : NULL,
910 outCoverage.c_str(),
911 inCoords,
912 &fUniformHandles.fSamplerUnis[s]);
bsalomon@google.comf06df1b2012-09-06 20:22:31 +0000913 builder.setNonStage();
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000914 inCoverage = outCoverage;
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000915 }
senorblanco@chromium.org92e0f222011-05-12 15:49:15 +0000916 }
917 }
bsalomon@google.com777c3aa2012-07-25 20:58:20 +0000918
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000919 if (Desc::kNone_DualSrcOutput != fDesc.fDualSrcOutput) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000920 builder.fFSOutputs.push_back().set(kVec4f_GrSLType,
921 GrGLShaderVar::kOut_TypeModifier,
922 dual_source_output_name());
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000923 bool outputIsZero = coverageIsZero;
bsalomon@google.comf0a104e2012-07-10 17:51:07 +0000924 SkString coeff;
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000925 if (!outputIsZero &&
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000926 Desc::kCoverage_DualSrcOutput != fDesc.fDualSrcOutput && !wroteFragColorZero) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000927 if (!inColor.size()) {
928 outputIsZero = true;
929 } else {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000930 if (Desc::kCoverageISA_DualSrcOutput == fDesc.fDualSrcOutput) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000931 coeff.printf("(1 - %s.a)", inColor.c_str());
932 } else {
933 coeff.printf("(vec4(1,1,1,1) - %s)", inColor.c_str());
934 }
935 }
936 }
937 if (outputIsZero) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000938 builder.fFSCode.appendf("\t%s = %s;\n",
939 dual_source_output_name(),
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000940 GrGLSLZerosVecf(4));
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000941 } else {
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000942 builder.fFSCode.appendf("\t%s =", dual_source_output_name());
943 GrGLSLModulate4f(&builder.fFSCode, coeff.c_str(), inCoverage.c_str());
944 builder.fFSCode.append(";\n");
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000945 }
946 dualSourceOutputWritten = true;
947 }
junov@google.comf93e7172011-03-31 21:26:24 +0000948 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000949
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000950 ///////////////////////////////////////////////////////////////////////////
951 // combine color and coverage as frag color
952
953 if (!wroteFragColorZero) {
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000954 if (coverageIsZero) {
bsalomon@google.comad5e9372012-07-11 18:11:27 +0000955 builder.fFSCode.appendf("\t%s = %s;\n",
956 colorOutput.getName().c_str(),
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000957 GrGLSLZerosVecf(4));
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000958 } else {
bsalomon@google.com4af0af62012-08-29 12:59:57 +0000959 builder.fFSCode.appendf("\t%s = ", colorOutput.getName().c_str());
960 GrGLSLModulate4f(&builder.fFSCode, inColor.c_str(), inCoverage.c_str());
961 builder.fFSCode.append(";\n");
bsalomon@google.com2401ae82012-01-17 21:03:05 +0000962 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000963 }
964
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000965 ///////////////////////////////////////////////////////////////////////////
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000966 // insert GS
967#if GR_DEBUG
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000968 this->genGeometryShader(&builder);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000969#endif
970
971 ///////////////////////////////////////////////////////////////////////////
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000972 // compile and setup attribs and unis
973
bsalomon@google.comecb60aa2012-07-18 13:20:29 +0000974 if (!this->compileShaders(builder)) {
bsalomon@google.com91961302011-05-09 18:39:58 +0000975 return false;
976 }
977
bsalomon@google.com17504f52012-10-30 12:34:25 +0000978 if (!this->bindOutputsAttribsAndLinkProgram(builder,
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000979 isColorDeclared,
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000980 dualSourceOutputWritten)) {
bsalomon@google.com91961302011-05-09 18:39:58 +0000981 return false;
982 }
983
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +0000984 builder.finished(fProgramID);
985 this->initSamplerUniforms();
bsalomon@google.com34cccde2013-01-04 18:34:30 +0000986 fUniformHandles.fRTHeightUni = builder.getRTHeightUniform();
bsalomon@google.com91961302011-05-09 18:39:58 +0000987
988 return true;
989}
990
bsalomon@google.com17504f52012-10-30 12:34:25 +0000991bool GrGLProgram::bindOutputsAttribsAndLinkProgram(const GrGLShaderBuilder& builder,
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000992 bool bindColorOut,
993 bool bindDualSrcOut) {
994 GL_CALL_RET(fProgramID, CreateProgram());
995 if (!fProgramID) {
bsalomon@google.com91961302011-05-09 18:39:58 +0000996 return false;
997 }
bsalomon@google.com91961302011-05-09 18:39:58 +0000998
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +0000999 GL_CALL(AttachShader(fProgramID, fVShaderID));
1000 if (fGShaderID) {
1001 GL_CALL(AttachShader(fProgramID, fGShaderID));
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001002 }
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001003 GL_CALL(AttachShader(fProgramID, fFShaderID));
bsalomon@google.com91961302011-05-09 18:39:58 +00001004
bsalomon@google.com271cffc2011-05-20 14:13:56 +00001005 if (bindColorOut) {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001006 GL_CALL(BindFragDataLocation(fProgramID, 0, declared_color_output_name()));
bsalomon@google.com271cffc2011-05-20 14:13:56 +00001007 }
1008 if (bindDualSrcOut) {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001009 GL_CALL(BindFragDataLocationIndexed(fProgramID, 0, 1, dual_source_output_name()));
bsalomon@google.com271cffc2011-05-20 14:13:56 +00001010 }
1011
bsalomon@google.com91961302011-05-09 18:39:58 +00001012 // Bind the attrib locations to same values for all shaders
bsalomon@google.com17504f52012-10-30 12:34:25 +00001013 GL_CALL(BindAttribLocation(fProgramID,
jvanverth@google.comb8b705b2013-02-28 16:28:34 +00001014 fDesc.fPositionAttributeIndex,
bsalomon@google.com17504f52012-10-30 12:34:25 +00001015 builder.positionAttribute().c_str()));
jvanverth@google.comb8b705b2013-02-28 16:28:34 +00001016 GL_CALL(BindAttribLocation(fProgramID, fDesc.fColorAttributeIndex, COL_ATTR_NAME));
1017 GL_CALL(BindAttribLocation(fProgramID, fDesc.fCoverageAttributeIndex, COV_ATTR_NAME));
1018
1019 if (fDesc.fAttribBindings & GrDrawState::kEdge_AttribBindingsBit) {
1020 GL_CALL(BindAttribLocation(fProgramID, fDesc.fEdgeAttributeIndex, EDGE_ATTR_NAME));
1021 }
1022 if (GrDrawState::AttributesBindExplicitTexCoords(fDesc.fAttribBindings)) {
1023 GL_CALL(BindAttribLocation(fProgramID, fDesc.fTexCoordAttributeIndex, TEX_ATTR_NAME));
1024 }
1025
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001026 GL_CALL(LinkProgram(fProgramID));
bsalomon@google.com91961302011-05-09 18:39:58 +00001027
1028 GrGLint linked = GR_GL_INIT_ZERO;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001029 GL_CALL(GetProgramiv(fProgramID, GR_GL_LINK_STATUS, &linked));
bsalomon@google.com91961302011-05-09 18:39:58 +00001030 if (!linked) {
1031 GrGLint infoLen = GR_GL_INIT_ZERO;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001032 GL_CALL(GetProgramiv(fProgramID, GR_GL_INFO_LOG_LENGTH, &infoLen));
bsalomon@google.com3582bf92011-06-30 21:32:31 +00001033 SkAutoMalloc log(sizeof(char)*(infoLen+1)); // outside if for debugger
bsalomon@google.com91961302011-05-09 18:39:58 +00001034 if (infoLen > 0) {
bsalomon@google.com79afcaa2011-09-14 14:29:18 +00001035 // retrieve length even though we don't need it to workaround
1036 // bug in chrome cmd buffer param validation.
1037 GrGLsizei length = GR_GL_INIT_ZERO;
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001038 GL_CALL(GetProgramInfoLog(fProgramID,
bsalomon@google.com96399942012-02-13 14:39:16 +00001039 infoLen+1,
1040 &length,
1041 (char*)log.get()));
bsalomon@google.com91961302011-05-09 18:39:58 +00001042 GrPrintf((char*)log.get());
1043 }
1044 GrAssert(!"Error linking program");
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001045 GL_CALL(DeleteProgram(fProgramID));
1046 fProgramID = 0;
bsalomon@google.com91961302011-05-09 18:39:58 +00001047 return false;
1048 }
1049 return true;
1050}
1051
bsalomon@google.comdbbc4e22012-07-25 17:48:39 +00001052void GrGLProgram::initSamplerUniforms() {
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001053 GL_CALL(UseProgram(fProgramID));
bsalomon@google.com34cccde2013-01-04 18:34:30 +00001054 // We simply bind the uniforms to successive texture units beginning at 0. setData() assumes this
1055 // behavior.
1056 GrGLint texUnitIdx = 0;
tomhudson@google.com93813632011-10-27 20:21:16 +00001057 for (int s = 0; s < GrDrawState::kNumStages; ++s) {
bsalomon@google.com34cccde2013-01-04 18:34:30 +00001058 int numSamplers = fUniformHandles.fSamplerUnis[s].count();
1059 for (int u = 0; u < numSamplers; ++u) {
1060 UniformHandle handle = fUniformHandles.fSamplerUnis[s][u];
1061 if (GrGLUniformManager::kInvalidUniformHandle != handle) {
1062 fUniformManager.setSampler(handle, texUnitIdx);
1063 ++texUnitIdx;
bsalomon@google.com0982d352012-07-31 15:33:25 +00001064 }
bsalomon@google.com91961302011-05-09 18:39:58 +00001065 }
bsalomon@google.com91961302011-05-09 18:39:58 +00001066 }
bsalomon@google.com91961302011-05-09 18:39:58 +00001067}
1068
bsalomon@google.comeb715c82012-07-11 15:03:31 +00001069///////////////////////////////////////////////////////////////////////////////
junov@google.comf93e7172011-03-31 21:26:24 +00001070
bsalomon@google.com91207482013-02-12 21:45:24 +00001071void GrGLProgram::setData(GrGpuGL* gpu,
1072 GrColor color,
1073 GrColor coverage,
1074 SharedGLState* sharedState) {
bsalomon@google.com34cccde2013-01-04 18:34:30 +00001075 const GrDrawState& drawState = gpu->getDrawState();
bsalomon@google.com9ba4fa62012-07-16 17:36:28 +00001076
bsalomon@google.com91207482013-02-12 21:45:24 +00001077 this->setColor(drawState, color, sharedState);
1078 this->setCoverage(drawState, coverage, sharedState);
bsalomon@google.com6a51dcb2013-02-13 16:03:51 +00001079 this->setMatrixAndRenderTargetHeight(drawState);
bsalomon@google.com91207482013-02-12 21:45:24 +00001080
1081 // Setup the SkXfermode::Mode-based colorfilter uniform if necessary
1082 if (GrGLUniformManager::kInvalidUniformHandle != fUniformHandles.fColorFilterUni &&
1083 fColorFilterColor != drawState.getColorFilterColor()) {
1084 GrGLfloat c[4];
1085 GrColorToRGBAFloat(drawState.getColorFilterColor(), c);
1086 fUniformManager.set4fv(fUniformHandles.fColorFilterUni, 0, 1, c);
1087 fColorFilterColor = drawState.getColorFilterColor();
1088 }
1089
bsalomon@google.com34cccde2013-01-04 18:34:30 +00001090 GrGLint texUnitIdx = 0;
bsalomon@google.com4285acc2012-10-22 14:11:24 +00001091 for (int s = 0; s < GrDrawState::kNumStages; ++s) {
bsalomon@google.com46fba0d2012-10-25 21:42:05 +00001092 if (NULL != fEffects[s]) {
bsalomon@google.com08283af2012-10-26 13:01:20 +00001093 const GrEffectStage& stage = drawState.getStage(s);
1094 GrAssert(NULL != stage.getEffect());
bsalomon@google.com28a15fb2012-10-26 17:53:18 +00001095 fEffects[s]->setData(fUniformManager, stage);
bsalomon@google.com34cccde2013-01-04 18:34:30 +00001096 int numSamplers = fUniformHandles.fSamplerUnis[s].count();
1097 for (int u = 0; u < numSamplers; ++u) {
1098 UniformHandle handle = fUniformHandles.fSamplerUnis[s][u];
1099 if (GrGLUniformManager::kInvalidUniformHandle != handle) {
bsalomon@google.com6340a412013-01-22 19:55:59 +00001100 const GrTextureAccess& access = (*stage.getEffect())->textureAccess(u);
bsalomon@google.com34cccde2013-01-04 18:34:30 +00001101 GrGLTexture* texture = static_cast<GrGLTexture*>(access.getTexture());
1102 gpu->bindTexture(texUnitIdx, access.getParams(), texture);
1103 ++texUnitIdx;
1104 }
1105 }
bsalomon@google.com4285acc2012-10-22 14:11:24 +00001106 }
1107 }
skia.committer@gmail.com8ae714b2013-01-05 02:02:05 +00001108}
bsalomon@google.com91207482013-02-12 21:45:24 +00001109
1110void GrGLProgram::setColor(const GrDrawState& drawState,
1111 GrColor color,
1112 SharedGLState* sharedState) {
jvanverth@google.comb8b705b2013-02-28 16:28:34 +00001113 if (!(drawState.getAttribBindings() & GrDrawState::kColor_AttribBindingsBit)) {
bsalomon@google.com91207482013-02-12 21:45:24 +00001114 switch (fDesc.fColorInput) {
1115 case GrGLProgram::Desc::kAttribute_ColorInput:
1116 if (sharedState->fConstAttribColor != color) {
1117 // OpenGL ES only supports the float varieties of glVertexAttrib
1118 GrGLfloat c[4];
1119 GrColorToRGBAFloat(color, c);
jvanverth@google.comb8b705b2013-02-28 16:28:34 +00001120 GL_CALL(VertexAttrib4fv(fDesc.fColorAttributeIndex, c));
bsalomon@google.com91207482013-02-12 21:45:24 +00001121 sharedState->fConstAttribColor = color;
1122 }
1123 break;
1124 case GrGLProgram::Desc::kUniform_ColorInput:
1125 if (fColor != color) {
1126 // OpenGL ES doesn't support unsigned byte varieties of glUniform
1127 GrGLfloat c[4];
1128 GrColorToRGBAFloat(color, c);
1129 GrAssert(GrGLUniformManager::kInvalidUniformHandle !=
1130 fUniformHandles.fColorUni);
1131 fUniformManager.set4fv(fUniformHandles.fColorUni, 0, 1, c);
1132 fColor = color;
1133 }
1134 break;
1135 case GrGLProgram::Desc::kSolidWhite_ColorInput:
1136 case GrGLProgram::Desc::kTransBlack_ColorInput:
1137 break;
1138 default:
1139 GrCrash("Unknown color type.");
1140 }
1141 }
1142}
1143
1144void GrGLProgram::setCoverage(const GrDrawState& drawState,
1145 GrColor coverage,
1146 SharedGLState* sharedState) {
jvanverth@google.comb8b705b2013-02-28 16:28:34 +00001147 if (!(drawState.getAttribBindings() & GrDrawState::kCoverage_AttribBindingsBit)) {
bsalomon@google.com91207482013-02-12 21:45:24 +00001148 switch (fDesc.fCoverageInput) {
1149 case Desc::kAttribute_ColorInput:
1150 if (sharedState->fConstAttribCoverage != coverage) {
1151 // OpenGL ES only supports the float varieties of glVertexAttrib
1152 GrGLfloat c[4];
1153 GrColorToRGBAFloat(coverage, c);
jvanverth@google.comb8b705b2013-02-28 16:28:34 +00001154 GL_CALL(VertexAttrib4fv(fDesc.fCoverageAttributeIndex, c));
bsalomon@google.com91207482013-02-12 21:45:24 +00001155 sharedState->fConstAttribCoverage = coverage;
1156 }
1157 break;
1158 case Desc::kUniform_ColorInput:
1159 if (fCoverage != coverage) {
1160 // OpenGL ES doesn't support unsigned byte varieties of glUniform
1161 GrGLfloat c[4];
1162 GrColorToRGBAFloat(coverage, c);
1163 GrAssert(GrGLUniformManager::kInvalidUniformHandle !=
1164 fUniformHandles.fCoverageUni);
1165 fUniformManager.set4fv(fUniformHandles.fCoverageUni, 0, 1, c);
1166 fCoverage = coverage;
1167 }
1168 break;
1169 case Desc::kSolidWhite_ColorInput:
1170 case Desc::kTransBlack_ColorInput:
1171 break;
1172 default:
1173 GrCrash("Unknown coverage type.");
1174 }
1175 }
1176}
bsalomon@google.com6a51dcb2013-02-13 16:03:51 +00001177
1178void GrGLProgram::setMatrixAndRenderTargetHeight(const GrDrawState& drawState) {
1179 const GrRenderTarget* rt = drawState.getRenderTarget();
1180 SkISize size;
1181 size.set(rt->width(), rt->height());
1182
1183 // Load the RT height uniform if it is needed to y-flip gl_FragCoord.
1184 if (GrGLUniformManager::kInvalidUniformHandle != fUniformHandles.fRTHeightUni &&
1185 fMatrixState.fRenderTargetSize.fHeight != size.fHeight) {
1186 fUniformManager.set1f(fUniformHandles.fRTHeightUni, SkIntToScalar(size.fHeight));
1187 }
1188
1189 if (fMatrixState.fRenderTargetOrigin != rt->origin() ||
1190 !fMatrixState.fViewMatrix.cheapEqualTo(drawState.getViewMatrix()) ||
1191 fMatrixState.fRenderTargetSize != size) {
1192 SkMatrix m;
1193 if (kBottomLeft_GrSurfaceOrigin == rt->origin()) {
1194 m.setAll(
1195 SkIntToScalar(2) / size.fWidth, 0, -SK_Scalar1,
1196 0,-SkIntToScalar(2) / size.fHeight, SK_Scalar1,
1197 0, 0, SkMatrix::I()[8]);
1198 } else {
1199 m.setAll(
1200 SkIntToScalar(2) / size.fWidth, 0, -SK_Scalar1,
1201 0, SkIntToScalar(2) / size.fHeight,-SK_Scalar1,
1202 0, 0, SkMatrix::I()[8]);
1203 }
1204 m.setConcat(m, drawState.getViewMatrix());
1205
1206 // ES doesn't allow you to pass true to the transpose param so we do our own transpose.
1207 GrGLfloat mt[] = {
1208 SkScalarToFloat(m[SkMatrix::kMScaleX]),
1209 SkScalarToFloat(m[SkMatrix::kMSkewY]),
1210 SkScalarToFloat(m[SkMatrix::kMPersp0]),
1211 SkScalarToFloat(m[SkMatrix::kMSkewX]),
1212 SkScalarToFloat(m[SkMatrix::kMScaleY]),
1213 SkScalarToFloat(m[SkMatrix::kMPersp1]),
1214 SkScalarToFloat(m[SkMatrix::kMTransX]),
1215 SkScalarToFloat(m[SkMatrix::kMTransY]),
1216 SkScalarToFloat(m[SkMatrix::kMPersp2])
1217 };
1218 fUniformManager.setMatrix3f(fUniformHandles.fViewMatrixUni, mt);
1219 fMatrixState.fViewMatrix = drawState.getViewMatrix();
1220 fMatrixState.fRenderTargetSize = size;
1221 fMatrixState.fRenderTargetOrigin = rt->origin();
1222 }
1223}