blob: a2a6492d5214a4e023a180cc925071992c0426b4 [file] [log] [blame]
epoger@google.comec3ed6a2011-07-28 14:26:00 +00001
junov@google.comf93e7172011-03-31 21:26:24 +00002/*
epoger@google.comec3ed6a2011-07-28 14:26:00 +00003 * Copyright 2011 Google Inc.
4 *
5 * Use of this source code is governed by a BSD-style license that can be
6 * found in the LICENSE file.
junov@google.comf93e7172011-03-31 21:26:24 +00007 */
8
epoger@google.comec3ed6a2011-07-28 14:26:00 +00009
junov@google.comf93e7172011-03-31 21:26:24 +000010#include "GrGLProgram.h"
11
junov@google.comf93e7172011-03-31 21:26:24 +000012#include "GrGLConfig.h"
junov@google.comf93e7172011-03-31 21:26:24 +000013
tomhudson@google.com0c8d93a2011-07-01 17:08:26 +000014#include "SkTrace.h"
Scroggo97c88c22011-05-11 14:05:25 +000015#include "SkXfermode.h"
16
junov@google.comf93e7172011-03-31 21:26:24 +000017namespace {
18
bsalomon@google.com0b77d682011-08-19 13:28:54 +000019const char* GrPrecision(const GrGLInterface* gl) {
20 if (gl->supportsES()) {
junov@google.comf93e7172011-03-31 21:26:24 +000021 return "mediump";
22 } else {
bsalomon@google.com91961302011-05-09 18:39:58 +000023 return " ";
junov@google.comf93e7172011-03-31 21:26:24 +000024 }
25}
26
bsalomon@google.com0b77d682011-08-19 13:28:54 +000027const char* GrShaderPrecision(const GrGLInterface* gl) {
28 if (gl->supportsES()) {
junov@google.comf93e7172011-03-31 21:26:24 +000029 return "precision mediump float;\n";
30 } else {
31 return "";
32 }
33}
34
35} // namespace
36
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +000037#define PRINT_SHADERS 0
38
bsalomon@google.com4be283f2011-04-19 21:15:09 +000039#if GR_GL_ATTRIBUTE_MATRICES
junov@google.comf93e7172011-03-31 21:26:24 +000040 #define VIEW_MATRIX_NAME "aViewM"
41#else
42 #define VIEW_MATRIX_NAME "uViewM"
43#endif
44
45#define POS_ATTR_NAME "aPosition"
46#define COL_ATTR_NAME "aColor"
bsalomon@google.comaeb21602011-08-30 18:13:44 +000047#define EDGE_ATTR_NAME "aEdge"
bsalomon@google.com4be283f2011-04-19 21:15:09 +000048#define COL_UNI_NAME "uColor"
senorblanco@chromium.org92e0f222011-05-12 15:49:15 +000049#define EDGES_UNI_NAME "uEdges"
Scroggo97c88c22011-05-11 14:05:25 +000050#define COL_FILTER_UNI_NAME "uColorFilter"
junov@google.comf93e7172011-03-31 21:26:24 +000051
junov@google.comf93e7172011-03-31 21:26:24 +000052static inline void tex_attr_name(int coordIdx, GrStringBuilder* s) {
53 *s = "aTexCoord";
bsalomon@google.comfc296292011-05-06 13:53:47 +000054 s->appendS32(coordIdx);
junov@google.comf93e7172011-03-31 21:26:24 +000055}
56
57static inline const char* float_vector_type(int count) {
58 static const char* FLOAT_VECS[] = {"ERROR", "float", "vec2", "vec3", "vec4"};
59 GrAssert(count >= 1 && count < (int)GR_ARRAY_COUNT(FLOAT_VECS));
60 return FLOAT_VECS[count];
61}
62
63static inline const char* vector_homog_coord(int count) {
64 static const char* HOMOGS[] = {"ERROR", "", ".y", ".z", ".w"};
65 GrAssert(count >= 1 && count < (int)GR_ARRAY_COUNT(HOMOGS));
66 return HOMOGS[count];
67}
68
69static inline const char* vector_nonhomog_coords(int count) {
70 static const char* NONHOMOGS[] = {"ERROR", "", ".x", ".xy", ".xyz"};
71 GrAssert(count >= 1 && count < (int)GR_ARRAY_COUNT(NONHOMOGS));
72 return NONHOMOGS[count];
73}
74
75static inline const char* vector_all_coords(int count) {
76 static const char* ALL[] = {"ERROR", "", ".xy", ".xyz", ".xyzw"};
77 GrAssert(count >= 1 && count < (int)GR_ARRAY_COUNT(ALL));
78 return ALL[count];
79}
80
bsalomon@google.comf2d91552011-05-16 20:56:06 +000081static inline const char* all_ones_vec(int count) {
tomhudson@google.com0d831722011-06-02 15:37:14 +000082 static const char* ONESVEC[] = {"ERROR", "1.0", "vec2(1,1)",
bsalomon@google.comf2d91552011-05-16 20:56:06 +000083 "vec3(1,1,1)", "vec4(1,1,1,1)"};
84 GrAssert(count >= 1 && count < (int)GR_ARRAY_COUNT(ONESVEC));
85 return ONESVEC[count];
86}
87
88static inline const char* all_zeros_vec(int count) {
tomhudson@google.com0d831722011-06-02 15:37:14 +000089 static const char* ZEROSVEC[] = {"ERROR", "0.0", "vec2(0,0)",
bsalomon@google.comf2d91552011-05-16 20:56:06 +000090 "vec3(0,0,0)", "vec4(0,0,0,0)"};
91 GrAssert(count >= 1 && count < (int)GR_ARRAY_COUNT(ZEROSVEC));
92 return ZEROSVEC[count];
93}
94
bsalomon@google.com271cffc2011-05-20 14:13:56 +000095static inline const char* declared_color_output_name() { return "fsColorOut"; }
96static inline const char* dual_source_output_name() { return "dualSourceOut"; }
97
junov@google.comf93e7172011-03-31 21:26:24 +000098static void tex_matrix_name(int stage, GrStringBuilder* s) {
bsalomon@google.com4be283f2011-04-19 21:15:09 +000099#if GR_GL_ATTRIBUTE_MATRICES
junov@google.comf93e7172011-03-31 21:26:24 +0000100 *s = "aTexM";
101#else
102 *s = "uTexM";
103#endif
bsalomon@google.comfc296292011-05-06 13:53:47 +0000104 s->appendS32(stage);
junov@google.comf93e7172011-03-31 21:26:24 +0000105}
106
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +0000107static void normalized_texel_size_name(int stage, GrStringBuilder* s) {
108 *s = "uTexelSize";
bsalomon@google.comfc296292011-05-06 13:53:47 +0000109 s->appendS32(stage);
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +0000110}
111
junov@google.comf93e7172011-03-31 21:26:24 +0000112static void sampler_name(int stage, GrStringBuilder* s) {
113 *s = "uSampler";
bsalomon@google.comfc296292011-05-06 13:53:47 +0000114 s->appendS32(stage);
junov@google.comf93e7172011-03-31 21:26:24 +0000115}
116
117static void stage_varying_name(int stage, GrStringBuilder* s) {
118 *s = "vStage";
bsalomon@google.comfc296292011-05-06 13:53:47 +0000119 s->appendS32(stage);
junov@google.comf93e7172011-03-31 21:26:24 +0000120}
121
122static void radial2_param_name(int stage, GrStringBuilder* s) {
123 *s = "uRadial2Params";
bsalomon@google.comfc296292011-05-06 13:53:47 +0000124 s->appendS32(stage);
junov@google.comf93e7172011-03-31 21:26:24 +0000125}
126
127static void radial2_varying_name(int stage, GrStringBuilder* s) {
128 *s = "vB";
bsalomon@google.comfc296292011-05-06 13:53:47 +0000129 s->appendS32(stage);
junov@google.comf93e7172011-03-31 21:26:24 +0000130}
131
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +0000132static void convolve_param_names(int stage, GrStringBuilder* k, GrStringBuilder* i) {
133 *k = "uKernel";
134 k->appendS32(stage);
135 *i = "uImageIncrement";
136 i->appendS32(stage);
137}
138
junov@google.com6acc9b32011-05-16 18:32:07 +0000139static void tex_domain_name(int stage, GrStringBuilder* s) {
140 *s = "uTexDom";
141 s->appendS32(stage);
142}
143
junov@google.comf93e7172011-03-31 21:26:24 +0000144GrGLProgram::GrGLProgram() {
junov@google.comf93e7172011-03-31 21:26:24 +0000145}
146
147GrGLProgram::~GrGLProgram() {
junov@google.comf93e7172011-03-31 21:26:24 +0000148}
149
tomhudson@google.com0d3f1fb2011-06-01 19:27:31 +0000150void GrGLProgram::overrideBlend(GrBlendCoeff* srcCoeff,
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000151 GrBlendCoeff* dstCoeff) const {
152 switch (fProgramDesc.fDualSrcOutput) {
153 case ProgramDesc::kNone_DualSrcOutput:
154 break;
155 // the prog will write a coverage value to the secondary
156 // output and the dst is blended by one minus that value.
157 case ProgramDesc::kCoverage_DualSrcOutput:
158 case ProgramDesc::kCoverageISA_DualSrcOutput:
159 case ProgramDesc::kCoverageISC_DualSrcOutput:
160 *dstCoeff = (GrBlendCoeff)GrGpu::kIS2C_BlendCoeff;
161 break;
162 default:
163 GrCrash("Unexpected dual source blend output");
164 break;
165 }
166}
167
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000168// assigns modulation of two vars to an output var
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000169// vars can be vec4s or floats (or one of each)
170// result is always vec4
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000171// if either var is "" then assign to the other var
172// if both are "" then assign all ones
173static inline void modulate_helper(const char* outputVar,
174 const char* var0,
175 const char* var1,
176 GrStringBuilder* code) {
177 GrAssert(NULL != outputVar);
178 GrAssert(NULL != var0);
179 GrAssert(NULL != var1);
180 GrAssert(NULL != code);
181
182 bool has0 = '\0' != *var0;
183 bool has1 = '\0' != *var1;
184
185 if (!has0 && !has1) {
186 code->appendf("\t%s = %s;\n", outputVar, all_ones_vec(4));
187 } else if (!has0) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000188 code->appendf("\t%s = vec4(%s);\n", outputVar, var1);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000189 } else if (!has1) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000190 code->appendf("\t%s = vec4(%s);\n", outputVar, var0);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000191 } else {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000192 code->appendf("\t%s = vec4(%s * %s);\n", outputVar, var0, var1);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000193 }
194}
195
196// assigns addition of two vars to an output var
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000197// vars can be vec4s or floats (or one of each)
198// result is always vec4
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000199// if either var is "" then assign to the other var
200// if both are "" then assign all zeros
201static inline void add_helper(const char* outputVar,
202 const char* var0,
203 const char* var1,
204 GrStringBuilder* code) {
205 GrAssert(NULL != outputVar);
206 GrAssert(NULL != var0);
207 GrAssert(NULL != var1);
208 GrAssert(NULL != code);
209
210 bool has0 = '\0' != *var0;
211 bool has1 = '\0' != *var1;
212
213 if (!has0 && !has1) {
214 code->appendf("\t%s = %s;\n", outputVar, all_zeros_vec(4));
215 } else if (!has0) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000216 code->appendf("\t%s = vec4(%s);\n", outputVar, var1);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000217 } else if (!has1) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000218 code->appendf("\t%s = vec4(%s);\n", outputVar, var0);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000219 } else {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000220 code->appendf("\t%s = vec4(%s + %s);\n", outputVar, var0, var1);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000221 }
222}
223
224// given two blend coeffecients determine whether the src
225// and/or dst computation can be omitted.
226static inline void needBlendInputs(SkXfermode::Coeff srcCoeff,
227 SkXfermode::Coeff dstCoeff,
228 bool* needSrcValue,
229 bool* needDstValue) {
230 if (SkXfermode::kZero_Coeff == srcCoeff) {
231 switch (dstCoeff) {
232 // these all read the src
233 case SkXfermode::kSC_Coeff:
234 case SkXfermode::kISC_Coeff:
235 case SkXfermode::kSA_Coeff:
236 case SkXfermode::kISA_Coeff:
237 *needSrcValue = true;
238 break;
239 default:
240 *needSrcValue = false;
241 break;
242 }
243 } else {
244 *needSrcValue = true;
245 }
246 if (SkXfermode::kZero_Coeff == dstCoeff) {
247 switch (srcCoeff) {
248 // these all read the dst
249 case SkXfermode::kDC_Coeff:
250 case SkXfermode::kIDC_Coeff:
251 case SkXfermode::kDA_Coeff:
252 case SkXfermode::kIDA_Coeff:
253 *needDstValue = true;
254 break;
255 default:
256 *needDstValue = false;
257 break;
258 }
259 } else {
260 *needDstValue = true;
Scroggo97c88c22011-05-11 14:05:25 +0000261 }
262}
263
264/**
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000265 * Create a blend_coeff * value string to be used in shader code. Sets empty
266 * string if result is trivially zero.
267 */
268static void blendTermString(GrStringBuilder* str, SkXfermode::Coeff coeff,
269 const char* src, const char* dst,
270 const char* value) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000271 switch (coeff) {
272 case SkXfermode::kZero_Coeff: /** 0 */
273 *str = "";
274 break;
275 case SkXfermode::kOne_Coeff: /** 1 */
276 *str = value;
277 break;
278 case SkXfermode::kSC_Coeff:
279 str->printf("(%s * %s)", src, value);
280 break;
281 case SkXfermode::kISC_Coeff:
282 str->printf("((%s - %s) * %s)", all_ones_vec(4), src, value);
283 break;
284 case SkXfermode::kDC_Coeff:
285 str->printf("(%s * %s)", dst, value);
286 break;
287 case SkXfermode::kIDC_Coeff:
288 str->printf("((%s - %s) * %s)", all_ones_vec(4), dst, value);
289 break;
290 case SkXfermode::kSA_Coeff: /** src alpha */
291 str->printf("(%s.a * %s)", src, value);
292 break;
293 case SkXfermode::kISA_Coeff: /** inverse src alpha (i.e. 1 - sa) */
294 str->printf("((1.0 - %s.a) * %s)", src, value);
295 break;
296 case SkXfermode::kDA_Coeff: /** dst alpha */
297 str->printf("(%s.a * %s)", dst, value);
298 break;
299 case SkXfermode::kIDA_Coeff: /** inverse dst alpha (i.e. 1 - da) */
300 str->printf("((1.0 - %s.a) * %s)", dst, value);
301 break;
302 default:
303 GrCrash("Unexpected xfer coeff.");
304 break;
305 }
306}
307/**
Scroggo97c88c22011-05-11 14:05:25 +0000308 * Adds a line to the fragment shader code which modifies the color by
309 * the specified color filter.
310 */
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000311static void addColorFilter(GrStringBuilder* fsCode, const char * outputVar,
tomhudson@google.com0d3f1fb2011-06-01 19:27:31 +0000312 SkXfermode::Coeff uniformCoeff,
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000313 SkXfermode::Coeff colorCoeff,
314 const char* inColor) {
315 GrStringBuilder colorStr, constStr;
316 blendTermString(&colorStr, colorCoeff, COL_FILTER_UNI_NAME,
317 inColor, inColor);
318 blendTermString(&constStr, uniformCoeff, COL_FILTER_UNI_NAME,
319 inColor, COL_FILTER_UNI_NAME);
320
321 add_helper(outputVar, colorStr.c_str(), constStr.c_str(), fsCode);
Scroggo97c88c22011-05-11 14:05:25 +0000322}
323
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000324bool GrGLProgram::genProgram(const GrGLInterface* gl,
325 GrGLProgram::CachedData* programData) const {
junov@google.comf93e7172011-03-31 21:26:24 +0000326
327 ShaderCodeSegments segments;
328 const uint32_t& layout = fProgramDesc.fVertexLayout;
329
bsalomon@google.com91961302011-05-09 18:39:58 +0000330 programData->fUniLocations.reset();
junov@google.comf93e7172011-03-31 21:26:24 +0000331
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000332 SkXfermode::Coeff colorCoeff, uniformCoeff;
333 // The rest of transfer mode color filters have not been implemented
334 if (fProgramDesc.fColorFilterXfermode < SkXfermode::kCoeffModesCnt) {
335 GR_DEBUGCODE(bool success =)
tomhudson@google.com0d831722011-06-02 15:37:14 +0000336 SkXfermode::ModeAsCoeff(static_cast<SkXfermode::Mode>
337 (fProgramDesc.fColorFilterXfermode),
338 &uniformCoeff, &colorCoeff);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000339 GR_DEBUGASSERT(success);
340 } else {
341 colorCoeff = SkXfermode::kOne_Coeff;
342 uniformCoeff = SkXfermode::kZero_Coeff;
343 }
344
345 bool needColorFilterUniform;
346 bool needComputedColor;
347 needBlendInputs(uniformCoeff, colorCoeff,
348 &needColorFilterUniform, &needComputedColor);
349
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000350 // the dual source output has no canonical var name, have to
351 // declare an output, which is incompatible with gl_FragColor/gl_FragData.
352 const char* fsColorOutput;
353 bool dualSourceOutputWritten = false;
tomhudson@google.com0d3f1fb2011-06-01 19:27:31 +0000354 bool usingDeclaredOutputs = ProgramDesc::kNone_DualSrcOutput !=
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000355 fProgramDesc.fDualSrcOutput;
356 if (usingDeclaredOutputs) {
357 GrAssert(0 == segments.fHeader.size());
358 segments.fHeader.printf("#version 150\n");
359 fsColorOutput = declared_color_output_name();
360 segments.fFSOutputs.appendf("out vec4 %s;\n", fsColorOutput);
361 } else {
362 fsColorOutput = "gl_FragColor";
363 }
364
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000365#if GR_GL_ATTRIBUTE_MATRICES
366 segments.fVSAttrs += "attribute mat3 " VIEW_MATRIX_NAME ";\n";
bsalomon@google.com91961302011-05-09 18:39:58 +0000367 programData->fUniLocations.fViewMatrixUni = kSetAsAttribute;
junov@google.comf93e7172011-03-31 21:26:24 +0000368#else
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000369 segments.fVSUnis += "uniform mat3 " VIEW_MATRIX_NAME ";\n";
bsalomon@google.com91961302011-05-09 18:39:58 +0000370 programData->fUniLocations.fViewMatrixUni = kUseUniform;
junov@google.comf93e7172011-03-31 21:26:24 +0000371#endif
372 segments.fVSAttrs += "attribute vec2 " POS_ATTR_NAME ";\n";
junov@google.comf93e7172011-03-31 21:26:24 +0000373
bsalomon@google.com91961302011-05-09 18:39:58 +0000374 segments.fVSCode.append(
375 "void main() {\n"
376 "\tvec3 pos3 = " VIEW_MATRIX_NAME " * vec3("POS_ATTR_NAME", 1);\n"
377 "\tgl_Position = vec4(pos3.xy, 0, pos3.z);\n");
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000378
bsalomon@google.com91961302011-05-09 18:39:58 +0000379 // incoming color to current stage being processed.
380 GrStringBuilder inColor;
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000381
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000382 if (needComputedColor) {
383 switch (fProgramDesc.fColorType) {
384 case ProgramDesc::kAttribute_ColorType:
385 segments.fVSAttrs.append( "attribute vec4 " COL_ATTR_NAME ";\n");
386 segments.fVaryings.append("varying vec4 vColor;\n");
387 segments.fVSCode.append( "\tvColor = " COL_ATTR_NAME ";\n");
388 inColor = "vColor";
389 break;
390 case ProgramDesc::kUniform_ColorType:
391 segments.fFSUnis.append( "uniform vec4 " COL_UNI_NAME ";\n");
392 programData->fUniLocations.fColorUni = kUseUniform;
393 inColor = COL_UNI_NAME;
394 break;
395 default:
396 GrAssert(ProgramDesc::kNone_ColorType == fProgramDesc.fColorType);
397 break;
398 }
junov@google.comf93e7172011-03-31 21:26:24 +0000399 }
400
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000401 if (fProgramDesc.fEmitsPointSize){
bsalomon@google.com91961302011-05-09 18:39:58 +0000402 segments.fVSCode.append("\tgl_PointSize = 1.0;\n");
junov@google.comf93e7172011-03-31 21:26:24 +0000403 }
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000404
bsalomon@google.com91961302011-05-09 18:39:58 +0000405 segments.fFSCode.append("void main() {\n");
junov@google.comf93e7172011-03-31 21:26:24 +0000406
407 // add texture coordinates that are used to the list of vertex attr decls
bsalomon@google.com91961302011-05-09 18:39:58 +0000408 GrStringBuilder texCoordAttrs[GrDrawTarget::kMaxTexCoords];
junov@google.comf93e7172011-03-31 21:26:24 +0000409 for (int t = 0; t < GrDrawTarget::kMaxTexCoords; ++t) {
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +0000410 if (GrDrawTarget::VertexUsesTexCoordIdx(t, layout)) {
junov@google.comf93e7172011-03-31 21:26:24 +0000411 tex_attr_name(t, texCoordAttrs + t);
bsalomon@google.com91961302011-05-09 18:39:58 +0000412 segments.fVSAttrs.appendf("attribute vec2 %s;\n", texCoordAttrs[t].c_str());
junov@google.comf93e7172011-03-31 21:26:24 +0000413 }
414 }
415
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000416 ///////////////////////////////////////////////////////////////////////////
417 // compute the final color
Scroggo97c88c22011-05-11 14:05:25 +0000418
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000419 // if we have color stages string them together, feeding the output color
420 // of each to the next and generating code for each stage.
421 if (needComputedColor) {
422 GrStringBuilder outColor;
423 for (int s = 0; s < fProgramDesc.fFirstCoverageStage; ++s) {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000424 if (fProgramDesc.fStages[s].isEnabled()) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000425 // create var to hold stage result
426 outColor = "color";
427 outColor.appendS32(s);
428 segments.fFSCode.appendf("\tvec4 %s;\n", outColor.c_str());
429
430 const char* inCoords;
431 // figure out what our input coords are
432 if (GrDrawTarget::StagePosAsTexCoordVertexLayoutBit(s) &
433 layout) {
434 inCoords = POS_ATTR_NAME;
Scroggo97c88c22011-05-11 14:05:25 +0000435 } else {
436 int tcIdx = GrDrawTarget::VertexTexCoordsForStage(s, layout);
437 // we better have input tex coordinates if stage is enabled.
438 GrAssert(tcIdx >= 0);
439 GrAssert(texCoordAttrs[tcIdx].size());
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000440 inCoords = texCoordAttrs[tcIdx].c_str();
junov@google.comf93e7172011-03-31 21:26:24 +0000441 }
442
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000443 genStageCode(gl, s,
junov@google.comf93e7172011-03-31 21:26:24 +0000444 fProgramDesc.fStages[s],
bsalomon@google.comfc296292011-05-06 13:53:47 +0000445 inColor.size() ? inColor.c_str() : NULL,
446 outColor.c_str(),
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000447 inCoords,
junov@google.comf93e7172011-03-31 21:26:24 +0000448 &segments,
449 &programData->fUniLocations.fStages[s]);
junov@google.comf93e7172011-03-31 21:26:24 +0000450 inColor = outColor;
junov@google.comf93e7172011-03-31 21:26:24 +0000451 }
452 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000453 }
Scroggo97c88c22011-05-11 14:05:25 +0000454
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000455 // if have all ones for the "dst" input to the color filter then we can make
456 // additional optimizations.
tomhudson@google.com0d831722011-06-02 15:37:14 +0000457 if (needColorFilterUniform && !inColor.size() &&
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000458 (SkXfermode::kIDC_Coeff == uniformCoeff ||
459 SkXfermode::kIDA_Coeff == uniformCoeff)) {
460 uniformCoeff = SkXfermode::kZero_Coeff;
461 bool bogus;
tomhudson@google.com0d831722011-06-02 15:37:14 +0000462 needBlendInputs(SkXfermode::kZero_Coeff, colorCoeff,
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000463 &needColorFilterUniform, &bogus);
464 }
465 if (needColorFilterUniform) {
466 segments.fFSUnis.append( "uniform vec4 " COL_FILTER_UNI_NAME ";\n");
467 programData->fUniLocations.fColorFilterUni = kUseUniform;
468 }
469
470 bool wroteFragColorZero = false;
tomhudson@google.com0d831722011-06-02 15:37:14 +0000471 if (SkXfermode::kZero_Coeff == uniformCoeff &&
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000472 SkXfermode::kZero_Coeff == colorCoeff) {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000473 segments.fFSCode.appendf("\t%s = %s;\n",
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000474 fsColorOutput,
475 all_zeros_vec(4));
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000476 wroteFragColorZero = true;
477 } else if (SkXfermode::kDst_Mode != fProgramDesc.fColorFilterXfermode) {
478 segments.fFSCode.appendf("\tvec4 filteredColor;\n");
479 const char* color = inColor.size() ? inColor.c_str() : all_ones_vec(4);
tomhudson@google.com0d831722011-06-02 15:37:14 +0000480 addColorFilter(&segments.fFSCode, "filteredColor", uniformCoeff,
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000481 colorCoeff, color);
482 inColor = "filteredColor";
483 }
484
485 ///////////////////////////////////////////////////////////////////////////
486 // compute the partial coverage (coverage stages and edge aa)
487
488 GrStringBuilder inCoverage;
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000489
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000490 // we don't need to compute coverage at all if we know the final shader
491 // output will be zero and we don't have a dual src blend output.
492 if (!wroteFragColorZero ||
493 ProgramDesc::kNone_DualSrcOutput != fProgramDesc.fDualSrcOutput) {
senorblanco@chromium.orgef3913b2011-05-19 17:11:07 +0000494 if (fProgramDesc.fEdgeAANumEdges > 0) {
495 segments.fFSUnis.append("uniform vec3 " EDGES_UNI_NAME "[");
496 segments.fFSUnis.appendS32(fProgramDesc.fEdgeAANumEdges);
497 segments.fFSUnis.append("];\n");
498 programData->fUniLocations.fEdgesUni = kUseUniform;
499 int count = fProgramDesc.fEdgeAANumEdges;
senorblanco@chromium.org92e0f222011-05-12 15:49:15 +0000500 segments.fFSCode.append(
senorblanco@chromium.orgef3913b2011-05-19 17:11:07 +0000501 "\tvec3 pos = vec3(gl_FragCoord.xy, 1);\n");
502 for (int i = 0; i < count; i++) {
503 segments.fFSCode.append("\tfloat a");
504 segments.fFSCode.appendS32(i);
505 segments.fFSCode.append(" = clamp(dot(" EDGES_UNI_NAME "[");
506 segments.fFSCode.appendS32(i);
507 segments.fFSCode.append("], pos), 0.0, 1.0);\n");
508 }
senorblanco@chromium.org129b8e32011-06-15 17:52:09 +0000509 if (fProgramDesc.fEdgeAAConcave && (count & 0x01) == 0) {
510 // For concave polys, we consider the edges in pairs.
511 segments.fFSFunctions.append("float cross2(vec2 a, vec2 b) {\n");
512 segments.fFSFunctions.append("\treturn dot(a, vec2(b.y, -b.x));\n");
513 segments.fFSFunctions.append("}\n");
514 for (int i = 0; i < count; i += 2) {
515 segments.fFSCode.appendf("\tfloat eb%d;\n", i / 2);
516 segments.fFSCode.appendf("\tif (cross2(" EDGES_UNI_NAME "[%d].xy, " EDGES_UNI_NAME "[%d].xy) < 0.0) {\n", i, i + 1);
517 segments.fFSCode.appendf("\t\teb%d = a%d * a%d;\n", i / 2, i, i + 1);
518 segments.fFSCode.append("\t} else {\n");
519 segments.fFSCode.appendf("\t\teb%d = a%d + a%d - a%d * a%d;\n", i / 2, i, i + 1, i, i + 1);
520 segments.fFSCode.append("\t}\n");
521 }
522 segments.fFSCode.append("\tfloat edgeAlpha = ");
523 for (int i = 0; i < count / 2 - 1; i++) {
524 segments.fFSCode.appendf("min(eb%d, ", i);
525 }
526 segments.fFSCode.appendf("eb%d", count / 2 - 1);
527 for (int i = 0; i < count / 2 - 1; i++) {
528 segments.fFSCode.append(")");
529 }
530 segments.fFSCode.append(";\n");
531 } else {
532 segments.fFSCode.append("\tfloat edgeAlpha = ");
533 for (int i = 0; i < count - 1; i++) {
534 segments.fFSCode.appendf("min(a%d * a%d, ", i, i + 1);
535 }
536 segments.fFSCode.appendf("a%d * a0", count - 1);
537 for (int i = 0; i < count - 1; i++) {
538 segments.fFSCode.append(")");
539 }
540 segments.fFSCode.append(";\n");
senorblanco@chromium.orgef3913b2011-05-19 17:11:07 +0000541 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000542 inCoverage = "edgeAlpha";
bsalomon@google.comaeb21602011-08-30 18:13:44 +0000543 } else if (layout & GrDrawTarget::kEdge_VertexLayoutBit) {
544 segments.fVSAttrs.append("attribute vec4 " EDGE_ATTR_NAME ";\n");
545 segments.fVaryings.append("varying vec4 vEdge;\n");
546 segments.fVSCode.append("\tvEdge = " EDGE_ATTR_NAME ";\n");
547 if (GrDrawTarget::kHairLine_EdgeType == fProgramDesc.fVertexEdgeType) {
548 segments.fFSCode.append("\tfloat edgeAlpha = abs(dot(vec3(gl_FragCoord.xy,1), vEdge.xyz));\n");
549 } else {
550 GrAssert(GrDrawTarget::kHairQuad_EdgeType == fProgramDesc.fVertexEdgeType);
551 // for now we know we're not in perspective, so we could compute this
552 // per-quadratic rather than per pixel
553 segments.fFSCode.append("\tvec2 duvdx = dFdx(vEdge.xy);\n");
554 segments.fFSCode.append("\tvec2 duvdy = dFdy(vEdge.xy);\n");
555 segments.fFSCode.append("\tfloat dfdx = 2.0*vEdge.x*duvdx.x - duvdx.y;\n");
556 segments.fFSCode.append("\tfloat dfdy = 2.0*vEdge.x*duvdy.x - duvdy.y;\n");
557 segments.fFSCode.append("\tfloat edgeAlpha = (vEdge.x*vEdge.x - vEdge.y);\n");
558 segments.fFSCode.append("\tedgeAlpha = sqrt(edgeAlpha*edgeAlpha / (dfdx*dfdx + dfdy*dfdy));\n");
559 }
560 segments.fFSCode.append("\tedgeAlpha = max(1.0 - edgeAlpha, 0.0);\n");
561 inCoverage = "edgeAlpha";
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000562 }
563
564 GrStringBuilder outCoverage;
565 const int& startStage = fProgramDesc.fFirstCoverageStage;
566 for (int s = startStage; s < GrDrawTarget::kNumStages; ++s) {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000567 if (fProgramDesc.fStages[s].isEnabled()) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000568 // create var to hold stage output
569 outCoverage = "coverage";
570 outCoverage.appendS32(s);
571 segments.fFSCode.appendf("\tvec4 %s;\n", outCoverage.c_str());
572
573 const char* inCoords;
574 // figure out what our input coords are
575 if (GrDrawTarget::StagePosAsTexCoordVertexLayoutBit(s) & layout) {
576 inCoords = POS_ATTR_NAME;
577 } else {
578 int tcIdx = GrDrawTarget::VertexTexCoordsForStage(s, layout);
579 // we better have input tex coordinates if stage is enabled.
580 GrAssert(tcIdx >= 0);
581 GrAssert(texCoordAttrs[tcIdx].size());
582 inCoords = texCoordAttrs[tcIdx].c_str();
583 }
584
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000585 genStageCode(gl, s,
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000586 fProgramDesc.fStages[s],
587 inCoverage.size() ? inCoverage.c_str() : NULL,
588 outCoverage.c_str(),
589 inCoords,
590 &segments,
591 &programData->fUniLocations.fStages[s]);
592 inCoverage = outCoverage;
senorblanco@chromium.org92e0f222011-05-12 15:49:15 +0000593 }
594 }
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000595 if (ProgramDesc::kNone_DualSrcOutput != fProgramDesc.fDualSrcOutput) {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000596 segments.fFSOutputs.appendf("out vec4 %s;\n",
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000597 dual_source_output_name());
598 bool outputIsZero = false;
599 GrStringBuilder coeff;
tomhudson@google.com0d831722011-06-02 15:37:14 +0000600 if (ProgramDesc::kCoverage_DualSrcOutput !=
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000601 fProgramDesc.fDualSrcOutput && !wroteFragColorZero) {
602 if (!inColor.size()) {
603 outputIsZero = true;
604 } else {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000605 if (fProgramDesc.fDualSrcOutput ==
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000606 ProgramDesc::kCoverageISA_DualSrcOutput) {
607 coeff.printf("(1 - %s.a)", inColor.c_str());
608 } else {
609 coeff.printf("(vec4(1,1,1,1) - %s)", inColor.c_str());
610 }
611 }
612 }
613 if (outputIsZero) {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000614 segments.fFSCode.appendf("\t%s = %s;\n",
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000615 dual_source_output_name(),
616 all_zeros_vec(4));
617 } else {
618 modulate_helper(dual_source_output_name(),
619 coeff.c_str(),
620 inCoverage.c_str(),
621 &segments.fFSCode);
622 }
623 dualSourceOutputWritten = true;
624 }
junov@google.comf93e7172011-03-31 21:26:24 +0000625 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000626
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000627 ///////////////////////////////////////////////////////////////////////////
628 // combine color and coverage as frag color
629
630 if (!wroteFragColorZero) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000631 modulate_helper(fsColorOutput,
632 inColor.c_str(),
633 inCoverage.c_str(),
634 &segments.fFSCode);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000635 }
636
bsalomon@google.com91961302011-05-09 18:39:58 +0000637 segments.fVSCode.append("}\n");
638 segments.fFSCode.append("}\n");
639
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000640 ///////////////////////////////////////////////////////////////////////////
641 // compile and setup attribs and unis
642
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000643 if (!CompileFSAndVS(gl, segments, programData)) {
bsalomon@google.com91961302011-05-09 18:39:58 +0000644 return false;
645 }
646
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000647 if (!this->bindOutputsAttribsAndLinkProgram(gl, texCoordAttrs,
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000648 usingDeclaredOutputs,
649 dualSourceOutputWritten,
650 programData)) {
bsalomon@google.com91961302011-05-09 18:39:58 +0000651 return false;
652 }
653
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000654 this->getUniformLocationsAndInitCache(gl, programData);
bsalomon@google.com91961302011-05-09 18:39:58 +0000655
656 return true;
657}
658
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000659bool GrGLProgram::CompileFSAndVS(const GrGLInterface* gl,
660 const ShaderCodeSegments& segments,
bsalomon@google.com91961302011-05-09 18:39:58 +0000661 CachedData* programData) {
junov@google.comf93e7172011-03-31 21:26:24 +0000662
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000663 static const int MAX_STRINGS = 6;
664 const char* strings[MAX_STRINGS];
665 int lengths[MAX_STRINGS];
junov@google.comf93e7172011-03-31 21:26:24 +0000666 int stringCnt = 0;
667
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000668 if (segments.fHeader.size()) {
669 strings[stringCnt] = segments.fHeader.c_str();
670 lengths[stringCnt] = segments.fHeader.size();
671 ++stringCnt;
672 }
bsalomon@google.comfc296292011-05-06 13:53:47 +0000673 if (segments.fVSUnis.size()) {
674 strings[stringCnt] = segments.fVSUnis.c_str();
675 lengths[stringCnt] = segments.fVSUnis.size();
junov@google.comf93e7172011-03-31 21:26:24 +0000676 ++stringCnt;
677 }
bsalomon@google.comfc296292011-05-06 13:53:47 +0000678 if (segments.fVSAttrs.size()) {
679 strings[stringCnt] = segments.fVSAttrs.c_str();
680 lengths[stringCnt] = segments.fVSAttrs.size();
junov@google.comf93e7172011-03-31 21:26:24 +0000681 ++stringCnt;
682 }
bsalomon@google.comfc296292011-05-06 13:53:47 +0000683 if (segments.fVaryings.size()) {
684 strings[stringCnt] = segments.fVaryings.c_str();
685 lengths[stringCnt] = segments.fVaryings.size();
junov@google.comf93e7172011-03-31 21:26:24 +0000686 ++stringCnt;
687 }
688
bsalomon@google.comfc296292011-05-06 13:53:47 +0000689 GrAssert(segments.fVSCode.size());
690 strings[stringCnt] = segments.fVSCode.c_str();
691 lengths[stringCnt] = segments.fVSCode.size();
junov@google.comf93e7172011-03-31 21:26:24 +0000692 ++stringCnt;
693
694#if PRINT_SHADERS
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000695 GrPrintf(segments.fHeader.c_str());
bsalomon@google.comfc296292011-05-06 13:53:47 +0000696 GrPrintf(segments.fVSUnis.c_str());
697 GrPrintf(segments.fVSAttrs.c_str());
698 GrPrintf(segments.fVaryings.c_str());
699 GrPrintf(segments.fVSCode.c_str());
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +0000700 GrPrintf("\n");
junov@google.comf93e7172011-03-31 21:26:24 +0000701#endif
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000702 GrAssert(stringCnt <= MAX_STRINGS);
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000703 programData->fVShaderID = CompileShader(gl, GR_GL_VERTEX_SHADER,
704 stringCnt, strings, lengths);
junov@google.comf93e7172011-03-31 21:26:24 +0000705
bsalomon@google.com91961302011-05-09 18:39:58 +0000706 if (!programData->fVShaderID) {
707 return false;
708 }
709
junov@google.comf93e7172011-03-31 21:26:24 +0000710 stringCnt = 0;
711
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000712 if (segments.fHeader.size()) {
713 strings[stringCnt] = segments.fHeader.c_str();
714 lengths[stringCnt] = segments.fHeader.size();
715 ++stringCnt;
716 }
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000717 if (strlen(GrShaderPrecision(gl)) > 1) {
718 strings[stringCnt] = GrShaderPrecision(gl);
719 lengths[stringCnt] = strlen(GrShaderPrecision(gl));
junov@google.comf93e7172011-03-31 21:26:24 +0000720 ++stringCnt;
721 }
bsalomon@google.comfc296292011-05-06 13:53:47 +0000722 if (segments.fFSUnis.size()) {
723 strings[stringCnt] = segments.fFSUnis.c_str();
724 lengths[stringCnt] = segments.fFSUnis.size();
junov@google.comf93e7172011-03-31 21:26:24 +0000725 ++stringCnt;
726 }
bsalomon@google.comfc296292011-05-06 13:53:47 +0000727 if (segments.fVaryings.size()) {
728 strings[stringCnt] = segments.fVaryings.c_str();
729 lengths[stringCnt] = segments.fVaryings.size();
junov@google.comf93e7172011-03-31 21:26:24 +0000730 ++stringCnt;
731 }
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000732 if (segments.fFSOutputs.size()) {
733 strings[stringCnt] = segments.fFSOutputs.c_str();
734 lengths[stringCnt] = segments.fFSOutputs.size();
735 ++stringCnt;
736 }
senorblanco@chromium.org129b8e32011-06-15 17:52:09 +0000737 if (segments.fFSFunctions.size()) {
738 strings[stringCnt] = segments.fFSFunctions.c_str();
739 lengths[stringCnt] = segments.fFSFunctions.size();
740 ++stringCnt;
741 }
junov@google.comf93e7172011-03-31 21:26:24 +0000742
bsalomon@google.comfc296292011-05-06 13:53:47 +0000743 GrAssert(segments.fFSCode.size());
744 strings[stringCnt] = segments.fFSCode.c_str();
745 lengths[stringCnt] = segments.fFSCode.size();
junov@google.comf93e7172011-03-31 21:26:24 +0000746 ++stringCnt;
747
748#if PRINT_SHADERS
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000749 GrPrintf(segments.fHeader.c_str());
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000750 GrPrintf(GrShaderPrecision(gl));
bsalomon@google.comfc296292011-05-06 13:53:47 +0000751 GrPrintf(segments.fFSUnis.c_str());
752 GrPrintf(segments.fVaryings.c_str());
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000753 GrPrintf(segments.fFSOutputs.c_str());
senorblanco@chromium.org129b8e32011-06-15 17:52:09 +0000754 GrPrintf(segments.fFSFunctions.c_str());
bsalomon@google.comfc296292011-05-06 13:53:47 +0000755 GrPrintf(segments.fFSCode.c_str());
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +0000756 GrPrintf("\n");
junov@google.comf93e7172011-03-31 21:26:24 +0000757#endif
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000758 GrAssert(stringCnt <= MAX_STRINGS);
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000759 programData->fFShaderID = CompileShader(gl, GR_GL_FRAGMENT_SHADER,
760 stringCnt, strings, lengths);
junov@google.comf93e7172011-03-31 21:26:24 +0000761
bsalomon@google.com91961302011-05-09 18:39:58 +0000762 if (!programData->fFShaderID) {
763 return false;
junov@google.comf93e7172011-03-31 21:26:24 +0000764 }
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000765
bsalomon@google.com91961302011-05-09 18:39:58 +0000766 return true;
junov@google.comf93e7172011-03-31 21:26:24 +0000767}
768
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000769GrGLuint GrGLProgram::CompileShader(const GrGLInterface* gl,
770 GrGLenum type,
771 int stringCnt,
772 const char** strings,
773 int* stringLengths) {
tomhudson@google.com278cbb42011-06-30 19:37:01 +0000774 SK_TRACE_EVENT1("GrGLProgram::CompileShader",
775 "stringCount", SkStringPrintf("%i", stringCnt).c_str());
776
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000777 GrGLuint shader;
778 GR_GL_CALL_RET(gl, shader, CreateShader(type));
junov@google.comf93e7172011-03-31 21:26:24 +0000779 if (0 == shader) {
780 return 0;
781 }
782
783 GrGLint compiled = GR_GL_INIT_ZERO;
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000784 GR_GL_CALL(gl, ShaderSource(shader, stringCnt, strings, stringLengths));
785 GR_GL_CALL(gl, CompileShader(shader));
786 GR_GL_CALL(gl, GetShaderiv(shader, GR_GL_COMPILE_STATUS, &compiled));
junov@google.comf93e7172011-03-31 21:26:24 +0000787
788 if (!compiled) {
789 GrGLint infoLen = GR_GL_INIT_ZERO;
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000790 GR_GL_CALL(gl, GetShaderiv(shader, GR_GL_INFO_LOG_LENGTH, &infoLen));
bsalomon@google.com3582bf92011-06-30 21:32:31 +0000791 SkAutoMalloc log(sizeof(char)*(infoLen+1)); // outside if for debugger
junov@google.comf93e7172011-03-31 21:26:24 +0000792 if (infoLen > 0) {
bsalomon@google.com79afcaa2011-09-14 14:29:18 +0000793 // retrieve length even though we don't need it to workaround
794 // bug in chrome cmd buffer param validation.
795 GrGLsizei length = GR_GL_INIT_ZERO;
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000796 GR_GL_CALL(gl, GetShaderInfoLog(shader, infoLen+1,
bsalomon@google.com79afcaa2011-09-14 14:29:18 +0000797 &length, (char*)log.get()));
junov@google.comf93e7172011-03-31 21:26:24 +0000798 for (int i = 0; i < stringCnt; ++i) {
799 if (NULL == stringLengths || stringLengths[i] < 0) {
800 GrPrintf(strings[i]);
801 } else {
802 GrPrintf("%.*s", stringLengths[i], strings[i]);
803 }
804 }
805 GrPrintf("\n%s", log.get());
806 }
807 GrAssert(!"Shader compilation failed!");
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000808 GR_GL_CALL(gl, DeleteShader(shader));
junov@google.comf93e7172011-03-31 21:26:24 +0000809 return 0;
810 }
811 return shader;
812}
813
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000814bool GrGLProgram::bindOutputsAttribsAndLinkProgram(
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000815 const GrGLInterface* gl,
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000816 GrStringBuilder texCoordAttrNames[],
817 bool bindColorOut,
tomhudson@google.com0d831722011-06-02 15:37:14 +0000818 bool bindDualSrcOut,
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000819 CachedData* programData) const {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000820 GR_GL_CALL_RET(gl, programData->fProgramID, CreateProgram());
bsalomon@google.com91961302011-05-09 18:39:58 +0000821 if (!programData->fProgramID) {
822 return false;
823 }
824 const GrGLint& progID = programData->fProgramID;
825
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000826 GR_GL_CALL(gl, AttachShader(progID, programData->fVShaderID));
827 GR_GL_CALL(gl, AttachShader(progID, programData->fFShaderID));
bsalomon@google.com91961302011-05-09 18:39:58 +0000828
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000829 if (bindColorOut) {
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000830 GR_GL_CALL(gl, BindFragDataLocationIndexed(programData->fProgramID,
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000831 0, 0, declared_color_output_name()));
832 }
833 if (bindDualSrcOut) {
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000834 GR_GL_CALL(gl, BindFragDataLocationIndexed(programData->fProgramID,
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000835 0, 1, dual_source_output_name()));
836 }
837
bsalomon@google.com91961302011-05-09 18:39:58 +0000838 // Bind the attrib locations to same values for all shaders
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000839 GR_GL_CALL(gl, BindAttribLocation(progID, PositionAttributeIdx(),
840 POS_ATTR_NAME));
bsalomon@google.com91961302011-05-09 18:39:58 +0000841 for (int t = 0; t < GrDrawTarget::kMaxTexCoords; ++t) {
842 if (texCoordAttrNames[t].size()) {
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000843 GR_GL_CALL(gl, BindAttribLocation(progID,
844 TexCoordAttributeIdx(t),
845 texCoordAttrNames[t].c_str()));
bsalomon@google.com91961302011-05-09 18:39:58 +0000846 }
847 }
848
bsalomon@google.com91961302011-05-09 18:39:58 +0000849 if (kSetAsAttribute == programData->fUniLocations.fViewMatrixUni) {
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000850 GR_GL_CALL(gl, BindAttribLocation(progID,
851 ViewMatrixAttributeIdx(),
852 VIEW_MATRIX_NAME));
bsalomon@google.com91961302011-05-09 18:39:58 +0000853 }
854
855 for (int s = 0; s < GrDrawTarget::kNumStages; ++s) {
bsalomon@google.com72b4fcb2011-05-09 20:47:34 +0000856 const StageUniLocations& unis = programData->fUniLocations.fStages[s];
857 if (kSetAsAttribute == unis.fTextureMatrixUni) {
bsalomon@google.com91961302011-05-09 18:39:58 +0000858 GrStringBuilder matName;
859 tex_matrix_name(s, &matName);
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000860 GR_GL_CALL(gl, BindAttribLocation(progID,
861 TextureMatrixAttributeIdx(s),
862 matName.c_str()));
bsalomon@google.com91961302011-05-09 18:39:58 +0000863 }
864 }
865
bsalomon@google.comaeb21602011-08-30 18:13:44 +0000866 GR_GL_CALL(gl, BindAttribLocation(progID, ColorAttributeIdx(),
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000867 COL_ATTR_NAME));
bsalomon@google.comaeb21602011-08-30 18:13:44 +0000868 GR_GL_CALL(gl, BindAttribLocation(progID, EdgeAttributeIdx(),
869 EDGE_ATTR_NAME));
bsalomon@google.com91961302011-05-09 18:39:58 +0000870
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000871 GR_GL_CALL(gl, LinkProgram(progID));
bsalomon@google.com91961302011-05-09 18:39:58 +0000872
873 GrGLint linked = GR_GL_INIT_ZERO;
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000874 GR_GL_CALL(gl, GetProgramiv(progID, GR_GL_LINK_STATUS, &linked));
bsalomon@google.com91961302011-05-09 18:39:58 +0000875 if (!linked) {
876 GrGLint infoLen = GR_GL_INIT_ZERO;
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000877 GR_GL_CALL(gl, GetProgramiv(progID, GR_GL_INFO_LOG_LENGTH, &infoLen));
bsalomon@google.com3582bf92011-06-30 21:32:31 +0000878 SkAutoMalloc log(sizeof(char)*(infoLen+1)); // outside if for debugger
bsalomon@google.com91961302011-05-09 18:39:58 +0000879 if (infoLen > 0) {
bsalomon@google.com79afcaa2011-09-14 14:29:18 +0000880 // retrieve length even though we don't need it to workaround
881 // bug in chrome cmd buffer param validation.
882 GrGLsizei length = GR_GL_INIT_ZERO;
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000883 GR_GL_CALL(gl, GetProgramInfoLog(progID, infoLen+1,
bsalomon@google.com79afcaa2011-09-14 14:29:18 +0000884 &length, (char*)log.get()));
bsalomon@google.com91961302011-05-09 18:39:58 +0000885 GrPrintf((char*)log.get());
886 }
887 GrAssert(!"Error linking program");
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000888 GR_GL_CALL(gl, DeleteProgram(progID));
bsalomon@google.com91961302011-05-09 18:39:58 +0000889 programData->fProgramID = 0;
890 return false;
891 }
892 return true;
893}
894
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000895void GrGLProgram::getUniformLocationsAndInitCache(const GrGLInterface* gl,
896 CachedData* programData) const {
bsalomon@google.com91961302011-05-09 18:39:58 +0000897 const GrGLint& progID = programData->fProgramID;
898
899 if (kUseUniform == programData->fUniLocations.fViewMatrixUni) {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000900 GR_GL_CALL_RET(gl, programData->fUniLocations.fViewMatrixUni,
901 GetUniformLocation(progID, VIEW_MATRIX_NAME));
bsalomon@google.com91961302011-05-09 18:39:58 +0000902 GrAssert(kUnusedUniform != programData->fUniLocations.fViewMatrixUni);
903 }
904 if (kUseUniform == programData->fUniLocations.fColorUni) {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000905 GR_GL_CALL_RET(gl, programData->fUniLocations.fColorUni,
906 GetUniformLocation(progID, COL_UNI_NAME));
bsalomon@google.com91961302011-05-09 18:39:58 +0000907 GrAssert(kUnusedUniform != programData->fUniLocations.fColorUni);
908 }
Scroggo97c88c22011-05-11 14:05:25 +0000909 if (kUseUniform == programData->fUniLocations.fColorFilterUni) {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000910 GR_GL_CALL_RET(gl, programData->fUniLocations.fColorFilterUni,
911 GetUniformLocation(progID, COL_FILTER_UNI_NAME));
Scroggo97c88c22011-05-11 14:05:25 +0000912 GrAssert(kUnusedUniform != programData->fUniLocations.fColorFilterUni);
913 }
bsalomon@google.com91961302011-05-09 18:39:58 +0000914
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000915 if (kUseUniform == programData->fUniLocations.fEdgesUni) {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000916 GR_GL_CALL_RET(gl, programData->fUniLocations.fEdgesUni,
917 GetUniformLocation(progID, EDGES_UNI_NAME));
senorblanco@chromium.org92e0f222011-05-12 15:49:15 +0000918 GrAssert(kUnusedUniform != programData->fUniLocations.fEdgesUni);
919 } else {
920 programData->fUniLocations.fEdgesUni = kUnusedUniform;
921 }
922
bsalomon@google.com91961302011-05-09 18:39:58 +0000923 for (int s = 0; s < GrDrawTarget::kNumStages; ++s) {
924 StageUniLocations& locations = programData->fUniLocations.fStages[s];
tomhudson@google.com0d831722011-06-02 15:37:14 +0000925 if (fProgramDesc.fStages[s].isEnabled()) {
bsalomon@google.com91961302011-05-09 18:39:58 +0000926 if (kUseUniform == locations.fTextureMatrixUni) {
927 GrStringBuilder texMName;
928 tex_matrix_name(s, &texMName);
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000929 GR_GL_CALL_RET(gl, locations.fTextureMatrixUni,
930 GetUniformLocation(progID, texMName.c_str()));
bsalomon@google.com91961302011-05-09 18:39:58 +0000931 GrAssert(kUnusedUniform != locations.fTextureMatrixUni);
932 }
933
934 if (kUseUniform == locations.fSamplerUni) {
935 GrStringBuilder samplerName;
936 sampler_name(s, &samplerName);
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000937 GR_GL_CALL_RET(gl, locations.fSamplerUni,
938 GetUniformLocation(progID,samplerName.c_str()));
bsalomon@google.com91961302011-05-09 18:39:58 +0000939 GrAssert(kUnusedUniform != locations.fSamplerUni);
940 }
941
942 if (kUseUniform == locations.fNormalizedTexelSizeUni) {
943 GrStringBuilder texelSizeName;
944 normalized_texel_size_name(s, &texelSizeName);
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000945 GR_GL_CALL_RET(gl, locations.fNormalizedTexelSizeUni,
946 GetUniformLocation(progID, texelSizeName.c_str()));
bsalomon@google.com91961302011-05-09 18:39:58 +0000947 GrAssert(kUnusedUniform != locations.fNormalizedTexelSizeUni);
948 }
949
950 if (kUseUniform == locations.fRadial2Uni) {
951 GrStringBuilder radial2ParamName;
952 radial2_param_name(s, &radial2ParamName);
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000953 GR_GL_CALL_RET(gl, locations.fRadial2Uni,
954 GetUniformLocation(progID, radial2ParamName.c_str()));
bsalomon@google.com91961302011-05-09 18:39:58 +0000955 GrAssert(kUnusedUniform != locations.fRadial2Uni);
956 }
junov@google.com6acc9b32011-05-16 18:32:07 +0000957
958 if (kUseUniform == locations.fTexDomUni) {
959 GrStringBuilder texDomName;
960 tex_domain_name(s, &texDomName);
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000961 GR_GL_CALL_RET(gl, locations.fTexDomUni,
962 GetUniformLocation(progID, texDomName.c_str()));
junov@google.com6acc9b32011-05-16 18:32:07 +0000963 GrAssert(kUnusedUniform != locations.fTexDomUni);
964 }
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +0000965
966 GrStringBuilder kernelName, imageIncrementName;
967 convolve_param_names(s, &kernelName, &imageIncrementName);
968 if (kUseUniform == locations.fKernelUni) {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000969 GR_GL_CALL_RET(gl, locations.fKernelUni,
970 GetUniformLocation(progID, kernelName.c_str()));
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +0000971 GrAssert(kUnusedUniform != locations.fKernelUni);
972 }
973
974 if (kUseUniform == locations.fImageIncrementUni) {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +0000975 GR_GL_CALL_RET(gl, locations.fImageIncrementUni,
976 GetUniformLocation(progID,
977 imageIncrementName.c_str()));
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +0000978 GrAssert(kUnusedUniform != locations.fImageIncrementUni);
979 }
bsalomon@google.com91961302011-05-09 18:39:58 +0000980 }
981 }
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000982 GR_GL_CALL(gl, UseProgram(progID));
bsalomon@google.com91961302011-05-09 18:39:58 +0000983
984 // init sampler unis and set bogus values for state tracking
985 for (int s = 0; s < GrDrawTarget::kNumStages; ++s) {
986 if (kUnusedUniform != programData->fUniLocations.fStages[s].fSamplerUni) {
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000987 GR_GL_CALL(gl, Uniform1i(programData->fUniLocations.fStages[s].fSamplerUni, s));
bsalomon@google.com91961302011-05-09 18:39:58 +0000988 }
989 programData->fTextureMatrices[s] = GrMatrix::InvalidMatrix();
990 programData->fRadial2CenterX1[s] = GR_ScalarMax;
991 programData->fRadial2Radius0[s] = -GR_ScalarMax;
992 programData->fTextureWidth[s] = -1;
993 programData->fTextureHeight[s] = -1;
994 }
995 programData->fViewMatrix = GrMatrix::InvalidMatrix();
996 programData->fColor = GrColor_ILLEGAL;
Scroggo97c88c22011-05-11 14:05:25 +0000997 programData->fColorFilterColor = GrColor_ILLEGAL;
bsalomon@google.com91961302011-05-09 18:39:58 +0000998}
999
junov@google.comf93e7172011-03-31 21:26:24 +00001000//============================================================================
1001// Stage code generation
1002//============================================================================
1003
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001004void GrGLProgram::genStageCode(const GrGLInterface* gl,
1005 int stageNum,
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001006 const GrGLProgram::StageDesc& desc,
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001007 const char* fsInColor, // NULL means no incoming color
1008 const char* fsOutColor,
1009 const char* vsInCoord,
1010 ShaderCodeSegments* segments,
1011 StageUniLocations* locations) const {
junov@google.comf93e7172011-03-31 21:26:24 +00001012
1013 GrAssert(stageNum >= 0 && stageNum <= 9);
1014
bsalomon@google.com91961302011-05-09 18:39:58 +00001015 GrStringBuilder varyingName;
junov@google.comf93e7172011-03-31 21:26:24 +00001016 stage_varying_name(stageNum, &varyingName);
1017
1018 // First decide how many coords are needed to access the texture
1019 // Right now it's always 2 but we could start using 1D textures for
1020 // gradients.
1021 static const int coordDims = 2;
1022 int varyingDims;
1023 /// Vertex Shader Stuff
1024
1025 // decide whether we need a matrix to transform texture coords
1026 // and whether the varying needs a perspective coord.
bsalomon@google.com91961302011-05-09 18:39:58 +00001027 GrStringBuilder texMName;
junov@google.comf93e7172011-03-31 21:26:24 +00001028 tex_matrix_name(stageNum, &texMName);
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001029 if (desc.fOptFlags & StageDesc::kIdentityMatrix_OptFlagBit) {
junov@google.comf93e7172011-03-31 21:26:24 +00001030 varyingDims = coordDims;
1031 } else {
bsalomon@google.com4be283f2011-04-19 21:15:09 +00001032 #if GR_GL_ATTRIBUTE_MATRICES
bsalomon@google.com91961302011-05-09 18:39:58 +00001033 segments->fVSAttrs.appendf("attribute mat3 %s;\n", texMName.c_str());
1034 locations->fTextureMatrixUni = kSetAsAttribute;
junov@google.comf93e7172011-03-31 21:26:24 +00001035 #else
bsalomon@google.com91961302011-05-09 18:39:58 +00001036 segments->fVSUnis.appendf("uniform mat3 %s;\n", texMName.c_str());
1037 locations->fTextureMatrixUni = kUseUniform;
junov@google.comf93e7172011-03-31 21:26:24 +00001038 #endif
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001039 if (desc.fOptFlags & StageDesc::kNoPerspective_OptFlagBit) {
junov@google.comf93e7172011-03-31 21:26:24 +00001040 varyingDims = coordDims;
1041 } else {
1042 varyingDims = coordDims + 1;
1043 }
1044 }
1045
bsalomon@google.com91961302011-05-09 18:39:58 +00001046 GrStringBuilder samplerName;
junov@google.comf93e7172011-03-31 21:26:24 +00001047 sampler_name(stageNum, &samplerName);
bsalomon@google.com91961302011-05-09 18:39:58 +00001048 segments->fFSUnis.appendf("uniform sampler2D %s;\n", samplerName.c_str());
1049 locations->fSamplerUni = kUseUniform;
junov@google.comf93e7172011-03-31 21:26:24 +00001050
bsalomon@google.com91961302011-05-09 18:39:58 +00001051 GrStringBuilder texelSizeName;
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001052 if (StageDesc::k2x2_FetchMode == desc.fFetchMode) {
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001053 normalized_texel_size_name(stageNum, &texelSizeName);
bsalomon@google.com91961302011-05-09 18:39:58 +00001054 segments->fFSUnis.appendf("uniform vec2 %s;\n", texelSizeName.c_str());
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001055 }
1056
tomhudson@google.com0d831722011-06-02 15:37:14 +00001057 segments->fVaryings.appendf("varying %s %s;\n",
bsalomon@google.com91961302011-05-09 18:39:58 +00001058 float_vector_type(varyingDims), varyingName.c_str());
junov@google.comf93e7172011-03-31 21:26:24 +00001059
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001060 if (desc.fOptFlags & StageDesc::kIdentityMatrix_OptFlagBit) {
junov@google.comf93e7172011-03-31 21:26:24 +00001061 GrAssert(varyingDims == coordDims);
bsalomon@google.com91961302011-05-09 18:39:58 +00001062 segments->fVSCode.appendf("\t%s = %s;\n", varyingName.c_str(), vsInCoord);
junov@google.comf93e7172011-03-31 21:26:24 +00001063 } else {
bsalomon@google.com91961302011-05-09 18:39:58 +00001064 // varying = texMatrix * texCoord
1065 segments->fVSCode.appendf("\t%s = (%s * vec3(%s, 1))%s;\n",
1066 varyingName.c_str(), texMName.c_str(),
1067 vsInCoord, vector_all_coords(varyingDims));
junov@google.comf93e7172011-03-31 21:26:24 +00001068 }
1069
bsalomon@google.com91961302011-05-09 18:39:58 +00001070 GrStringBuilder radial2ParamsName;
junov@google.comf93e7172011-03-31 21:26:24 +00001071 radial2_param_name(stageNum, &radial2ParamsName);
1072 // for radial grads without perspective we can pass the linear
1073 // part of the quadratic as a varying.
bsalomon@google.com91961302011-05-09 18:39:58 +00001074 GrStringBuilder radial2VaryingName;
junov@google.comf93e7172011-03-31 21:26:24 +00001075 radial2_varying_name(stageNum, &radial2VaryingName);
1076
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001077 if (StageDesc::kRadial2Gradient_CoordMapping == desc.fCoordMapping ||
1078 StageDesc::kRadial2GradientDegenerate_CoordMapping == desc.fCoordMapping) {
junov@google.comf93e7172011-03-31 21:26:24 +00001079
tomhudson@google.com0d831722011-06-02 15:37:14 +00001080 segments->fVSUnis.appendf("uniform %s float %s[6];\n",
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001081 GrPrecision(gl), radial2ParamsName.c_str());
tomhudson@google.com0d831722011-06-02 15:37:14 +00001082 segments->fFSUnis.appendf("uniform float %s[6];\n",
bsalomon@google.com91961302011-05-09 18:39:58 +00001083 radial2ParamsName.c_str());
1084 locations->fRadial2Uni = kUseUniform;
junov@google.comf93e7172011-03-31 21:26:24 +00001085
1086 // if there is perspective we don't interpolate this
1087 if (varyingDims == coordDims) {
1088 GrAssert(2 == coordDims);
bsalomon@google.com91961302011-05-09 18:39:58 +00001089 segments->fVaryings.appendf("varying float %s;\n", radial2VaryingName.c_str());
junov@google.comf93e7172011-03-31 21:26:24 +00001090
bsalomon@google.com91961302011-05-09 18:39:58 +00001091 // r2Var = 2 * (r2Parm[2] * varCoord.x - r2Param[3])
1092 segments->fVSCode.appendf("\t%s = 2.0 *(%s[2] * %s.x - %s[3]);\n",
1093 radial2VaryingName.c_str(), radial2ParamsName.c_str(),
1094 varyingName.c_str(), radial2ParamsName.c_str());
junov@google.comf93e7172011-03-31 21:26:24 +00001095 }
1096 }
1097
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +00001098 GrStringBuilder kernelName, kernelWidthName, imageIncrementName;
1099 convolve_param_names(stageNum, &kernelName, &imageIncrementName);
1100
1101 if (ProgramDesc::StageDesc::kConvolution_FetchMode == desc.fFetchMode) {
1102 segments->fFSUnis.appendf("uniform float %s[%d];\n",
1103 kernelName.c_str(), desc.fKernelWidth);
1104 segments->fFSUnis.appendf("uniform vec2 %s;\n",
1105 imageIncrementName.c_str());
bsalomon@google.comedc177d2011-08-05 15:46:40 +00001106 segments->fVSUnis.appendf("uniform %s vec2 %s;\n",
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001107 GrPrecision(gl),
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +00001108 imageIncrementName.c_str());
1109 locations->fKernelUni = kUseUniform;
1110 locations->fImageIncrementUni = kUseUniform;
1111 float scale = (desc.fKernelWidth - 1) * 0.5f;
1112 segments->fVSCode.appendf("\t%s -= vec2(%g, %g) * %s;\n",
1113 varyingName.c_str(), scale, scale,
1114 imageIncrementName.c_str());
bsalomon@google.coma8e686e2011-08-16 15:45:58 +00001115 }
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +00001116
junov@google.comf93e7172011-03-31 21:26:24 +00001117 /// Fragment Shader Stuff
bsalomon@google.com91961302011-05-09 18:39:58 +00001118 GrStringBuilder fsCoordName;
junov@google.comf93e7172011-03-31 21:26:24 +00001119 // function used to access the shader, may be made projective
bsalomon@google.com91961302011-05-09 18:39:58 +00001120 GrStringBuilder texFunc("texture2D");
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001121 if (desc.fOptFlags & (StageDesc::kIdentityMatrix_OptFlagBit |
1122 StageDesc::kNoPerspective_OptFlagBit)) {
junov@google.comf93e7172011-03-31 21:26:24 +00001123 GrAssert(varyingDims == coordDims);
1124 fsCoordName = varyingName;
1125 } else {
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001126 // if we have to do some special op on the varyings to get
junov@google.comf93e7172011-03-31 21:26:24 +00001127 // our final tex coords then when in perspective we have to
bsalomon@google.com91961302011-05-09 18:39:58 +00001128 // do an explicit divide. Otherwise, we can use a Proj func.
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001129 if (StageDesc::kIdentity_CoordMapping == desc.fCoordMapping &&
1130 StageDesc::kSingle_FetchMode == desc.fFetchMode) {
bsalomon@google.com91961302011-05-09 18:39:58 +00001131 texFunc.append("Proj");
junov@google.comf93e7172011-03-31 21:26:24 +00001132 fsCoordName = varyingName;
1133 } else {
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001134 fsCoordName = "inCoord";
bsalomon@google.comfc296292011-05-06 13:53:47 +00001135 fsCoordName.appendS32(stageNum);
bsalomon@google.com91961302011-05-09 18:39:58 +00001136 segments->fFSCode.appendf("\t%s %s = %s%s / %s%s;\n",
1137 float_vector_type(coordDims),
1138 fsCoordName.c_str(),
1139 varyingName.c_str(),
1140 vector_nonhomog_coords(varyingDims),
1141 varyingName.c_str(),
1142 vector_homog_coord(varyingDims));
junov@google.comf93e7172011-03-31 21:26:24 +00001143 }
1144 }
1145
bsalomon@google.comfc296292011-05-06 13:53:47 +00001146 GrStringBuilder sampleCoords;
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001147 bool complexCoord = false;
junov@google.comf93e7172011-03-31 21:26:24 +00001148 switch (desc.fCoordMapping) {
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001149 case StageDesc::kIdentity_CoordMapping:
junov@google.comf93e7172011-03-31 21:26:24 +00001150 sampleCoords = fsCoordName;
1151 break;
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001152 case StageDesc::kSweepGradient_CoordMapping:
bsalomon@google.com91961302011-05-09 18:39:58 +00001153 sampleCoords.printf("vec2(atan(- %s.y, - %s.x) * 0.1591549430918 + 0.5, 0.5)", fsCoordName.c_str(), fsCoordName.c_str());
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001154 complexCoord = true;
junov@google.comf93e7172011-03-31 21:26:24 +00001155 break;
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001156 case StageDesc::kRadialGradient_CoordMapping:
bsalomon@google.com91961302011-05-09 18:39:58 +00001157 sampleCoords.printf("vec2(length(%s.xy), 0.5)", fsCoordName.c_str());
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001158 complexCoord = true;
junov@google.comf93e7172011-03-31 21:26:24 +00001159 break;
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001160 case StageDesc::kRadial2Gradient_CoordMapping: {
bsalomon@google.com91961302011-05-09 18:39:58 +00001161 GrStringBuilder cName("c");
1162 GrStringBuilder ac4Name("ac4");
1163 GrStringBuilder rootName("root");
junov@google.comf93e7172011-03-31 21:26:24 +00001164
bsalomon@google.comfc296292011-05-06 13:53:47 +00001165 cName.appendS32(stageNum);
1166 ac4Name.appendS32(stageNum);
1167 rootName.appendS32(stageNum);
junov@google.comf93e7172011-03-31 21:26:24 +00001168
bsalomon@google.com91961302011-05-09 18:39:58 +00001169 // if we were able to interpolate the linear component bVar is the varying
1170 // otherwise compute it
1171 GrStringBuilder bVar;
junov@google.comf93e7172011-03-31 21:26:24 +00001172 if (coordDims == varyingDims) {
1173 bVar = radial2VaryingName;
1174 GrAssert(2 == varyingDims);
1175 } else {
1176 GrAssert(3 == varyingDims);
1177 bVar = "b";
bsalomon@google.comfc296292011-05-06 13:53:47 +00001178 bVar.appendS32(stageNum);
bsalomon@google.com91961302011-05-09 18:39:58 +00001179 segments->fFSCode.appendf("\tfloat %s = 2.0 * (%s[2] * %s.x - %s[3]);\n",
1180 bVar.c_str(), radial2ParamsName.c_str(),
1181 fsCoordName.c_str(), radial2ParamsName.c_str());
junov@google.comf93e7172011-03-31 21:26:24 +00001182 }
1183
bsalomon@google.com91961302011-05-09 18:39:58 +00001184 // c = (x^2)+(y^2) - params[4]
1185 segments->fFSCode.appendf("\tfloat %s = dot(%s, %s) - %s[4];\n",
1186 cName.c_str(), fsCoordName.c_str(),
1187 fsCoordName.c_str(),
1188 radial2ParamsName.c_str());
1189 // ac4 = 4.0 * params[0] * c
1190 segments->fFSCode.appendf("\tfloat %s = %s[0] * 4.0 * %s;\n",
1191 ac4Name.c_str(), radial2ParamsName.c_str(),
1192 cName.c_str());
junov@google.comf93e7172011-03-31 21:26:24 +00001193
bsalomon@google.com91961302011-05-09 18:39:58 +00001194 // root = sqrt(b^2-4ac)
1195 // (abs to avoid exception due to fp precision)
1196 segments->fFSCode.appendf("\tfloat %s = sqrt(abs(%s*%s - %s));\n",
1197 rootName.c_str(), bVar.c_str(), bVar.c_str(),
1198 ac4Name.c_str());
junov@google.comf93e7172011-03-31 21:26:24 +00001199
bsalomon@google.com91961302011-05-09 18:39:58 +00001200 // x coord is: (-b + params[5] * sqrt(b^2-4ac)) * params[1]
1201 // y coord is 0.5 (texture is effectively 1D)
1202 sampleCoords.printf("vec2((-%s + %s[5] * %s) * %s[1], 0.5)",
1203 bVar.c_str(), radial2ParamsName.c_str(),
1204 rootName.c_str(), radial2ParamsName.c_str());
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001205 complexCoord = true;
junov@google.comf93e7172011-03-31 21:26:24 +00001206 break;}
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001207 case StageDesc::kRadial2GradientDegenerate_CoordMapping: {
1208 GrStringBuilder cName("c");
1209
1210 cName.appendS32(stageNum);
1211
1212 // if we were able to interpolate the linear component bVar is the varying
1213 // otherwise compute it
1214 GrStringBuilder bVar;
1215 if (coordDims == varyingDims) {
1216 bVar = radial2VaryingName;
1217 GrAssert(2 == varyingDims);
1218 } else {
1219 GrAssert(3 == varyingDims);
1220 bVar = "b";
1221 bVar.appendS32(stageNum);
1222 segments->fFSCode.appendf("\tfloat %s = 2.0 * (%s[2] * %s.x - %s[3]);\n",
1223 bVar.c_str(), radial2ParamsName.c_str(),
1224 fsCoordName.c_str(), radial2ParamsName.c_str());
1225 }
1226
1227 // c = (x^2)+(y^2) - params[4]
1228 segments->fFSCode.appendf("\tfloat %s = dot(%s, %s) - %s[4];\n",
1229 cName.c_str(), fsCoordName.c_str(),
1230 fsCoordName.c_str(),
1231 radial2ParamsName.c_str());
1232
1233 // x coord is: -c/b
1234 // y coord is 0.5 (texture is effectively 1D)
1235 sampleCoords.printf("vec2((-%s / %s), 0.5)", cName.c_str(), bVar.c_str());
1236 complexCoord = true;
1237 break;}
junov@google.comf93e7172011-03-31 21:26:24 +00001238 };
1239
bsalomon@google.com91961302011-05-09 18:39:58 +00001240 const char* smear;
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001241 if (desc.fModulation == StageDesc::kAlpha_Modulation) {
bsalomon@google.com91961302011-05-09 18:39:58 +00001242 smear = ".aaaa";
1243 } else {
1244 smear = "";
1245 }
1246 GrStringBuilder modulate;
1247 if (NULL != fsInColor) {
1248 modulate.printf(" * %s", fsInColor);
1249 }
1250
tomhudson@google.com0d831722011-06-02 15:37:14 +00001251 if (desc.fOptFlags &
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001252 StageDesc::kCustomTextureDomain_OptFlagBit) {
junov@google.com6acc9b32011-05-16 18:32:07 +00001253 GrStringBuilder texDomainName;
1254 tex_domain_name(stageNum, &texDomainName);
tomhudson@google.com0d831722011-06-02 15:37:14 +00001255 segments->fFSUnis.appendf("uniform %s %s;\n",
junov@google.com6acc9b32011-05-16 18:32:07 +00001256 float_vector_type(4),
1257 texDomainName.c_str());
1258 GrStringBuilder coordVar("clampCoord");
1259 segments->fFSCode.appendf("\t%s %s = clamp(%s, %s.xy, %s.zw);\n",
1260 float_vector_type(coordDims),
1261 coordVar.c_str(),
1262 sampleCoords.c_str(),
1263 texDomainName.c_str(),
1264 texDomainName.c_str());
1265 sampleCoords = coordVar;
1266 locations->fTexDomUni = kUseUniform;
1267 }
1268
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001269 if (StageDesc::k2x2_FetchMode == desc.fFetchMode) {
bsalomon@google.com91961302011-05-09 18:39:58 +00001270 locations->fNormalizedTexelSizeUni = kUseUniform;
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001271 if (complexCoord) {
bsalomon@google.com91961302011-05-09 18:39:58 +00001272 // assign the coord to a var rather than compute 4x.
1273 GrStringBuilder coordVar("tCoord");
bsalomon@google.comfc296292011-05-06 13:53:47 +00001274 coordVar.appendS32(stageNum);
bsalomon@google.com91961302011-05-09 18:39:58 +00001275 segments->fFSCode.appendf("\t%s %s = %s;\n",
1276 float_vector_type(coordDims),
1277 coordVar.c_str(), sampleCoords.c_str());
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001278 sampleCoords = coordVar;
1279 }
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001280 GrAssert(2 == coordDims);
bsalomon@google.com91961302011-05-09 18:39:58 +00001281 GrStringBuilder accumVar("accum");
1282 accumVar.appendS32(stageNum);
1283 segments->fFSCode.appendf("\tvec4 %s = %s(%s, %s + vec2(-%s.x,-%s.y))%s;\n", accumVar.c_str(), texFunc.c_str(), samplerName.c_str(), sampleCoords.c_str(), texelSizeName.c_str(), texelSizeName.c_str(), smear);
1284 segments->fFSCode.appendf("\t%s += %s(%s, %s + vec2(+%s.x,-%s.y))%s;\n", accumVar.c_str(), texFunc.c_str(), samplerName.c_str(), sampleCoords.c_str(), texelSizeName.c_str(), texelSizeName.c_str(), smear);
1285 segments->fFSCode.appendf("\t%s += %s(%s, %s + vec2(-%s.x,+%s.y))%s;\n", accumVar.c_str(), texFunc.c_str(), samplerName.c_str(), sampleCoords.c_str(), texelSizeName.c_str(), texelSizeName.c_str(), smear);
1286 segments->fFSCode.appendf("\t%s += %s(%s, %s + vec2(+%s.x,+%s.y))%s;\n", accumVar.c_str(), texFunc.c_str(), samplerName.c_str(), sampleCoords.c_str(), texelSizeName.c_str(), texelSizeName.c_str(), smear);
1287 segments->fFSCode.appendf("\t%s = .25 * %s%s;\n", fsOutColor, accumVar.c_str(), modulate.c_str());
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +00001288 } else if (ProgramDesc::StageDesc::kConvolution_FetchMode == desc.fFetchMode) {
bsalomon@google.coma8e686e2011-08-16 15:45:58 +00001289 GrStringBuilder sumVar("sum");
1290 sumVar.appendS32(stageNum);
1291 GrStringBuilder coordVar("coord");
1292 coordVar.appendS32(stageNum);
1293
1294 segments->fFSCode.appendf("\tvec4 %s = vec4(0, 0, 0, 0);\n",
1295 sumVar.c_str());
1296 segments->fFSCode.appendf("\tvec2 %s = %s;\n",
1297 coordVar.c_str(),
1298 sampleCoords.c_str());
1299 segments->fFSCode.appendf("\tfor (int i = 0; i < %d; i++) {\n",
1300 desc.fKernelWidth);
1301 segments->fFSCode.appendf("\t\t%s += %s(%s, %s)%s * %s[i];\n",
1302 sumVar.c_str(), texFunc.c_str(),
1303 samplerName.c_str(), coordVar.c_str(), smear,
1304 kernelName.c_str());
1305 segments->fFSCode.appendf("\t\t%s += %s;\n",
1306 coordVar.c_str(),
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +00001307 imageIncrementName.c_str());
1308 segments->fFSCode.appendf("\t}\n");
bsalomon@google.coma8e686e2011-08-16 15:45:58 +00001309 segments->fFSCode.appendf("\t%s = %s%s;\n", fsOutColor,
1310 sumVar.c_str(), modulate.c_str());
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001311 } else {
bsalomon@google.coma8e686e2011-08-16 15:45:58 +00001312 segments->fFSCode.appendf("\t%s = %s(%s, %s)%s%s;\n",
1313 fsOutColor, texFunc.c_str(),
1314 samplerName.c_str(), sampleCoords.c_str(),
1315 smear, modulate.c_str());
junov@google.comf93e7172011-03-31 21:26:24 +00001316 }
junov@google.comf93e7172011-03-31 21:26:24 +00001317}