blob: 2e391e3c31be57b0f36153aa1d61bebdee4bb1a6 [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
bsalomon@google.com4fa66942011-09-20 19:06:12 +000012#include "GrAllocator.h"
13#include "GrGLShaderVar.h"
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
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +000019enum {
20 /// Used to mark a StageUniLocation field that should be bound
21 /// to a uniform during getUniformLocationsAndInitCache().
22 kUseUniform = 2000
23};
24
25
bsalomon@google.com0b77d682011-08-19 13:28:54 +000026const char* GrPrecision(const GrGLInterface* gl) {
bsalomon@google.com1dcf5062011-11-14 19:29:53 +000027 if (gl->supportsES2()) {
junov@google.comf93e7172011-03-31 21:26:24 +000028 return "mediump";
29 } else {
bsalomon@google.com91961302011-05-09 18:39:58 +000030 return " ";
junov@google.comf93e7172011-03-31 21:26:24 +000031 }
32}
33
bsalomon@google.com0b77d682011-08-19 13:28:54 +000034const char* GrShaderPrecision(const GrGLInterface* gl) {
bsalomon@google.com1dcf5062011-11-14 19:29:53 +000035 if (gl->supportsES2()) {
junov@google.comf93e7172011-03-31 21:26:24 +000036 return "precision mediump float;\n";
37 } else {
38 return "";
39 }
40}
41
42} // namespace
43
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +000044#define PRINT_SHADERS 0
45
bsalomon@google.com4fa66942011-09-20 19:06:12 +000046typedef GrTAllocator<GrGLShaderVar> VarArray;
47
48// number of each input/output type in a single allocation block
49static const int gVarsPerBlock = 8;
50// except FS outputs where we expect 2 at most.
51static const int gMaxFSOutputs = 2;
52
53struct ShaderCodeSegments {
54 ShaderCodeSegments()
55 : fVSUnis(gVarsPerBlock)
56 , fVSAttrs(gVarsPerBlock)
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +000057 , fVSOutputs(gVarsPerBlock)
58 , fGSInputs(gVarsPerBlock)
59 , fGSOutputs(gVarsPerBlock)
60 , fFSInputs(gVarsPerBlock)
bsalomon@google.com4fa66942011-09-20 19:06:12 +000061 , fFSUnis(gVarsPerBlock)
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +000062 , fFSOutputs(gMaxFSOutputs)
63 , fUsesGS(false) {}
bsalomon@google.com4fa66942011-09-20 19:06:12 +000064 GrStringBuilder fHeader; // VS+FS, GLSL version, etc
65 VarArray fVSUnis;
66 VarArray fVSAttrs;
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +000067 VarArray fVSOutputs;
68 VarArray fGSInputs;
69 VarArray fGSOutputs;
70 VarArray fFSInputs;
71 GrStringBuilder fGSHeader; // layout qualifiers specific to GS
bsalomon@google.com4fa66942011-09-20 19:06:12 +000072 VarArray fFSUnis;
73 VarArray fFSOutputs;
74 GrStringBuilder fFSFunctions;
75 GrStringBuilder fVSCode;
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +000076 GrStringBuilder fGSCode;
bsalomon@google.com4fa66942011-09-20 19:06:12 +000077 GrStringBuilder fFSCode;
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +000078
79 bool fUsesGS;
bsalomon@google.com4fa66942011-09-20 19:06:12 +000080};
81
bsalomon@google.com85b505b2011-11-07 14:56:51 +000082typedef GrGLProgram::ProgramDesc::StageDesc StageDesc;
bsalomon@google.com4fa66942011-09-20 19:06:12 +000083
bsalomon@google.com4be283f2011-04-19 21:15:09 +000084#if GR_GL_ATTRIBUTE_MATRICES
junov@google.comf93e7172011-03-31 21:26:24 +000085 #define VIEW_MATRIX_NAME "aViewM"
86#else
87 #define VIEW_MATRIX_NAME "uViewM"
88#endif
89
90#define POS_ATTR_NAME "aPosition"
91#define COL_ATTR_NAME "aColor"
bsalomon@google.coma3108262011-10-10 14:08:47 +000092#define COV_ATTR_NAME "aCoverage"
bsalomon@google.comaeb21602011-08-30 18:13:44 +000093#define EDGE_ATTR_NAME "aEdge"
bsalomon@google.com4be283f2011-04-19 21:15:09 +000094#define COL_UNI_NAME "uColor"
senorblanco@chromium.org92e0f222011-05-12 15:49:15 +000095#define EDGES_UNI_NAME "uEdges"
Scroggo97c88c22011-05-11 14:05:25 +000096#define COL_FILTER_UNI_NAME "uColorFilter"
senorblanco@chromium.org50bdad82012-01-03 20:51:57 +000097#define COL_MATRIX_UNI_NAME "uColorMatrix"
98#define COL_MATRIX_VEC_UNI_NAME "uColorMatrixVec"
junov@google.comf93e7172011-03-31 21:26:24 +000099
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000100namespace {
101inline void tex_attr_name(int coordIdx, GrStringBuilder* s) {
junov@google.comf93e7172011-03-31 21:26:24 +0000102 *s = "aTexCoord";
bsalomon@google.comfc296292011-05-06 13:53:47 +0000103 s->appendS32(coordIdx);
junov@google.comf93e7172011-03-31 21:26:24 +0000104}
105
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000106inline GrGLShaderVar::Type float_vector_type(int count) {
107 GR_STATIC_ASSERT(GrGLShaderVar::kFloat_Type == 0);
108 GR_STATIC_ASSERT(GrGLShaderVar::kVec2f_Type == 1);
109 GR_STATIC_ASSERT(GrGLShaderVar::kVec3f_Type == 2);
110 GR_STATIC_ASSERT(GrGLShaderVar::kVec4f_Type == 3);
111 GrAssert(count > 0 && count <= 4);
112 return (GrGLShaderVar::Type)(count - 1);
junov@google.comf93e7172011-03-31 21:26:24 +0000113}
114
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000115inline const char* float_vector_type_str(int count) {
116 return GrGLShaderVar::TypeString(float_vector_type(count));
117}
118
119inline const char* vector_homog_coord(int count) {
junov@google.comf93e7172011-03-31 21:26:24 +0000120 static const char* HOMOGS[] = {"ERROR", "", ".y", ".z", ".w"};
121 GrAssert(count >= 1 && count < (int)GR_ARRAY_COUNT(HOMOGS));
122 return HOMOGS[count];
123}
124
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000125inline const char* vector_nonhomog_coords(int count) {
junov@google.comf93e7172011-03-31 21:26:24 +0000126 static const char* NONHOMOGS[] = {"ERROR", "", ".x", ".xy", ".xyz"};
127 GrAssert(count >= 1 && count < (int)GR_ARRAY_COUNT(NONHOMOGS));
128 return NONHOMOGS[count];
129}
130
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000131inline const char* vector_all_coords(int count) {
junov@google.comf93e7172011-03-31 21:26:24 +0000132 static const char* ALL[] = {"ERROR", "", ".xy", ".xyz", ".xyzw"};
133 GrAssert(count >= 1 && count < (int)GR_ARRAY_COUNT(ALL));
134 return ALL[count];
135}
136
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000137inline const char* all_ones_vec(int count) {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000138 static const char* ONESVEC[] = {"ERROR", "1.0", "vec2(1,1)",
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000139 "vec3(1,1,1)", "vec4(1,1,1,1)"};
140 GrAssert(count >= 1 && count < (int)GR_ARRAY_COUNT(ONESVEC));
141 return ONESVEC[count];
142}
143
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000144inline const char* all_zeros_vec(int count) {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000145 static const char* ZEROSVEC[] = {"ERROR", "0.0", "vec2(0,0)",
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000146 "vec3(0,0,0)", "vec4(0,0,0,0)"};
147 GrAssert(count >= 1 && count < (int)GR_ARRAY_COUNT(ZEROSVEC));
148 return ZEROSVEC[count];
149}
150
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000151inline const char* declared_color_output_name() { return "fsColorOut"; }
152inline const char* dual_source_output_name() { return "dualSourceOut"; }
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000153
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000154inline void tex_matrix_name(int stage, GrStringBuilder* s) {
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000155#if GR_GL_ATTRIBUTE_MATRICES
junov@google.comf93e7172011-03-31 21:26:24 +0000156 *s = "aTexM";
157#else
158 *s = "uTexM";
159#endif
bsalomon@google.comfc296292011-05-06 13:53:47 +0000160 s->appendS32(stage);
junov@google.comf93e7172011-03-31 21:26:24 +0000161}
162
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000163inline void normalized_texel_size_name(int stage, GrStringBuilder* s) {
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +0000164 *s = "uTexelSize";
bsalomon@google.comfc296292011-05-06 13:53:47 +0000165 s->appendS32(stage);
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +0000166}
167
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000168inline void sampler_name(int stage, GrStringBuilder* s) {
junov@google.comf93e7172011-03-31 21:26:24 +0000169 *s = "uSampler";
bsalomon@google.comfc296292011-05-06 13:53:47 +0000170 s->appendS32(stage);
junov@google.comf93e7172011-03-31 21:26:24 +0000171}
172
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000173inline void radial2_param_name(int stage, GrStringBuilder* s) {
junov@google.comf93e7172011-03-31 21:26:24 +0000174 *s = "uRadial2Params";
bsalomon@google.comfc296292011-05-06 13:53:47 +0000175 s->appendS32(stage);
junov@google.comf93e7172011-03-31 21:26:24 +0000176}
177
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000178inline void convolve_param_names(int stage, GrStringBuilder* k, GrStringBuilder* i) {
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +0000179 *k = "uKernel";
180 k->appendS32(stage);
181 *i = "uImageIncrement";
182 i->appendS32(stage);
183}
184
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000185inline void tex_domain_name(int stage, GrStringBuilder* s) {
junov@google.com6acc9b32011-05-16 18:32:07 +0000186 *s = "uTexDom";
187 s->appendS32(stage);
188}
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000189}
junov@google.com6acc9b32011-05-16 18:32:07 +0000190
junov@google.comf93e7172011-03-31 21:26:24 +0000191GrGLProgram::GrGLProgram() {
junov@google.comf93e7172011-03-31 21:26:24 +0000192}
193
194GrGLProgram::~GrGLProgram() {
junov@google.comf93e7172011-03-31 21:26:24 +0000195}
196
tomhudson@google.com0d3f1fb2011-06-01 19:27:31 +0000197void GrGLProgram::overrideBlend(GrBlendCoeff* srcCoeff,
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000198 GrBlendCoeff* dstCoeff) const {
199 switch (fProgramDesc.fDualSrcOutput) {
200 case ProgramDesc::kNone_DualSrcOutput:
201 break;
202 // the prog will write a coverage value to the secondary
203 // output and the dst is blended by one minus that value.
204 case ProgramDesc::kCoverage_DualSrcOutput:
205 case ProgramDesc::kCoverageISA_DualSrcOutput:
206 case ProgramDesc::kCoverageISC_DualSrcOutput:
207 *dstCoeff = (GrBlendCoeff)GrGpu::kIS2C_BlendCoeff;
208 break;
209 default:
210 GrCrash("Unexpected dual source blend output");
211 break;
212 }
213}
214
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000215// assigns modulation of two vars to an output var
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000216// vars can be vec4s or floats (or one of each)
217// result is always vec4
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000218// if either var is "" then assign to the other var
219// if both are "" then assign all ones
220static inline void modulate_helper(const char* outputVar,
221 const char* var0,
222 const char* var1,
223 GrStringBuilder* code) {
224 GrAssert(NULL != outputVar);
225 GrAssert(NULL != var0);
226 GrAssert(NULL != var1);
227 GrAssert(NULL != code);
228
229 bool has0 = '\0' != *var0;
230 bool has1 = '\0' != *var1;
231
232 if (!has0 && !has1) {
233 code->appendf("\t%s = %s;\n", outputVar, all_ones_vec(4));
234 } else if (!has0) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000235 code->appendf("\t%s = vec4(%s);\n", outputVar, var1);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000236 } else if (!has1) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000237 code->appendf("\t%s = vec4(%s);\n", outputVar, var0);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000238 } else {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000239 code->appendf("\t%s = vec4(%s * %s);\n", outputVar, var0, var1);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000240 }
241}
242
243// assigns addition of two vars to an output var
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000244// vars can be vec4s or floats (or one of each)
245// result is always vec4
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000246// if either var is "" then assign to the other var
247// if both are "" then assign all zeros
248static inline void add_helper(const char* outputVar,
249 const char* var0,
250 const char* var1,
251 GrStringBuilder* code) {
252 GrAssert(NULL != outputVar);
253 GrAssert(NULL != var0);
254 GrAssert(NULL != var1);
255 GrAssert(NULL != code);
256
257 bool has0 = '\0' != *var0;
258 bool has1 = '\0' != *var1;
259
260 if (!has0 && !has1) {
261 code->appendf("\t%s = %s;\n", outputVar, all_zeros_vec(4));
262 } else if (!has0) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000263 code->appendf("\t%s = vec4(%s);\n", outputVar, var1);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000264 } else if (!has1) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000265 code->appendf("\t%s = vec4(%s);\n", outputVar, var0);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000266 } else {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000267 code->appendf("\t%s = vec4(%s + %s);\n", outputVar, var0, var1);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000268 }
269}
270
271// given two blend coeffecients determine whether the src
272// and/or dst computation can be omitted.
273static inline void needBlendInputs(SkXfermode::Coeff srcCoeff,
274 SkXfermode::Coeff dstCoeff,
275 bool* needSrcValue,
276 bool* needDstValue) {
277 if (SkXfermode::kZero_Coeff == srcCoeff) {
278 switch (dstCoeff) {
279 // these all read the src
280 case SkXfermode::kSC_Coeff:
281 case SkXfermode::kISC_Coeff:
282 case SkXfermode::kSA_Coeff:
283 case SkXfermode::kISA_Coeff:
284 *needSrcValue = true;
285 break;
286 default:
287 *needSrcValue = false;
288 break;
289 }
290 } else {
291 *needSrcValue = true;
292 }
293 if (SkXfermode::kZero_Coeff == dstCoeff) {
294 switch (srcCoeff) {
295 // these all read the dst
296 case SkXfermode::kDC_Coeff:
297 case SkXfermode::kIDC_Coeff:
298 case SkXfermode::kDA_Coeff:
299 case SkXfermode::kIDA_Coeff:
300 *needDstValue = true;
301 break;
302 default:
303 *needDstValue = false;
304 break;
305 }
306 } else {
307 *needDstValue = true;
Scroggo97c88c22011-05-11 14:05:25 +0000308 }
309}
310
311/**
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000312 * Create a blend_coeff * value string to be used in shader code. Sets empty
313 * string if result is trivially zero.
314 */
315static void blendTermString(GrStringBuilder* str, SkXfermode::Coeff coeff,
316 const char* src, const char* dst,
317 const char* value) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000318 switch (coeff) {
319 case SkXfermode::kZero_Coeff: /** 0 */
320 *str = "";
321 break;
322 case SkXfermode::kOne_Coeff: /** 1 */
323 *str = value;
324 break;
325 case SkXfermode::kSC_Coeff:
326 str->printf("(%s * %s)", src, value);
327 break;
328 case SkXfermode::kISC_Coeff:
329 str->printf("((%s - %s) * %s)", all_ones_vec(4), src, value);
330 break;
331 case SkXfermode::kDC_Coeff:
332 str->printf("(%s * %s)", dst, value);
333 break;
334 case SkXfermode::kIDC_Coeff:
335 str->printf("((%s - %s) * %s)", all_ones_vec(4), dst, value);
336 break;
337 case SkXfermode::kSA_Coeff: /** src alpha */
338 str->printf("(%s.a * %s)", src, value);
339 break;
340 case SkXfermode::kISA_Coeff: /** inverse src alpha (i.e. 1 - sa) */
341 str->printf("((1.0 - %s.a) * %s)", src, value);
342 break;
343 case SkXfermode::kDA_Coeff: /** dst alpha */
344 str->printf("(%s.a * %s)", dst, value);
345 break;
346 case SkXfermode::kIDA_Coeff: /** inverse dst alpha (i.e. 1 - da) */
347 str->printf("((1.0 - %s.a) * %s)", dst, value);
348 break;
349 default:
350 GrCrash("Unexpected xfer coeff.");
351 break;
352 }
353}
354/**
Scroggo97c88c22011-05-11 14:05:25 +0000355 * Adds a line to the fragment shader code which modifies the color by
356 * the specified color filter.
357 */
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000358static void addColorFilter(GrStringBuilder* fsCode, const char * outputVar,
tomhudson@google.com0d3f1fb2011-06-01 19:27:31 +0000359 SkXfermode::Coeff uniformCoeff,
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000360 SkXfermode::Coeff colorCoeff,
361 const char* inColor) {
362 GrStringBuilder colorStr, constStr;
363 blendTermString(&colorStr, colorCoeff, COL_FILTER_UNI_NAME,
364 inColor, inColor);
365 blendTermString(&constStr, uniformCoeff, COL_FILTER_UNI_NAME,
366 inColor, COL_FILTER_UNI_NAME);
367
368 add_helper(outputVar, colorStr.c_str(), constStr.c_str(), fsCode);
Scroggo97c88c22011-05-11 14:05:25 +0000369}
senorblanco@chromium.org50bdad82012-01-03 20:51:57 +0000370/**
371 * Adds code to the fragment shader code which modifies the color by
372 * the specified color matrix.
373 */
374static void addColorMatrix(GrStringBuilder* fsCode, const char * outputVar,
375 const char* inColor) {
senorblanco@chromium.orgb3a39b52012-01-05 18:28:56 +0000376 fsCode->appendf("\t%s = %s * vec4(%s.rgb / %s.a, %s.a) + %s;\n", outputVar, COL_MATRIX_UNI_NAME, inColor, inColor, inColor, COL_MATRIX_VEC_UNI_NAME);
377 fsCode->appendf("\t%s.rgb *= %s.a;\n", outputVar, outputVar);
senorblanco@chromium.org50bdad82012-01-03 20:51:57 +0000378}
Scroggo97c88c22011-05-11 14:05:25 +0000379
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000380namespace {
381
382const char* glsl_version_string(const GrGLInterface* gl,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000383 GrGLSLGeneration v) {
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000384 switch (v) {
tomhudson@google.com086e5352011-12-08 14:44:10 +0000385 case k110_GLSLGeneration:
bsalomon@google.com1dcf5062011-11-14 19:29:53 +0000386 if (gl->supportsES2()) {
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000387 // ES2s shader language is based on version 1.20 but is version
388 // 1.00 of the ES language.
389 return "#version 100\n";
390 } else {
bsalomon@google.com373a6632011-10-19 20:43:20 +0000391 return "#version 110\n";
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000392 }
tomhudson@google.com086e5352011-12-08 14:44:10 +0000393 case k130_GLSLGeneration:
bsalomon@google.com4c5de582011-11-14 19:35:08 +0000394 GrAssert(!gl->supportsES2());
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000395 return "#version 130\n";
tomhudson@google.com086e5352011-12-08 14:44:10 +0000396 case k150_GLSLGeneration:
bsalomon@google.com4c5de582011-11-14 19:35:08 +0000397 GrAssert(!gl->supportsES2());
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000398 return "#version 150\n";
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000399 default:
400 GrCrash("Unknown GL version.");
401 return ""; // suppress warning
402 }
403}
404
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000405// Adds a var that is computed in the VS and read in FS.
406// If there is a GS it will just pass it through.
407void append_varying(GrGLShaderVar::Type type,
408 const char* name,
409 ShaderCodeSegments* segments,
410 const char** vsOutName = NULL,
411 const char** fsInName = NULL) {
412 segments->fVSOutputs.push_back();
413 segments->fVSOutputs.back().setType(type);
tomhudson@google.com086e5352011-12-08 14:44:10 +0000414 segments->fVSOutputs.back().setTypeModifier(
415 GrGLShaderVar::kOut_TypeModifier);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000416 segments->fVSOutputs.back().accessName()->printf("v%s", name);
417 if (vsOutName) {
418 *vsOutName = segments->fVSOutputs.back().getName().c_str();
419 }
420 // input to FS comes either from VS or GS
421 const GrStringBuilder* fsName;
422 if (segments->fUsesGS) {
423 // if we have a GS take each varying in as an array
424 // and output as non-array.
425 segments->fGSInputs.push_back();
426 segments->fGSInputs.back().setType(type);
tomhudson@google.com086e5352011-12-08 14:44:10 +0000427 segments->fGSInputs.back().setTypeModifier(
428 GrGLShaderVar::kIn_TypeModifier);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000429 segments->fGSInputs.back().setUnsizedArray();
430 *segments->fGSInputs.back().accessName() =
431 segments->fVSOutputs.back().getName();
432 segments->fGSOutputs.push_back();
433 segments->fGSOutputs.back().setType(type);
tomhudson@google.com086e5352011-12-08 14:44:10 +0000434 segments->fGSOutputs.back().setTypeModifier(
435 GrGLShaderVar::kOut_TypeModifier);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000436 segments->fGSOutputs.back().accessName()->printf("g%s", name);
437 fsName = segments->fGSOutputs.back().accessName();
438 } else {
439 fsName = segments->fVSOutputs.back().accessName();
440 }
441 segments->fFSInputs.push_back();
442 segments->fFSInputs.back().setType(type);
tomhudson@google.com086e5352011-12-08 14:44:10 +0000443 segments->fFSInputs.back().setTypeModifier(
444 GrGLShaderVar::kIn_TypeModifier);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000445 segments->fFSInputs.back().setName(*fsName);
446 if (fsInName) {
447 *fsInName = fsName->c_str();
448 }
449}
450
451// version of above that adds a stage number to the
452// the var name (for uniqueness)
453void append_varying(GrGLShaderVar::Type type,
454 const char* name,
455 int stageNum,
456 ShaderCodeSegments* segments,
457 const char** vsOutName = NULL,
458 const char** fsInName = NULL) {
459 GrStringBuilder nameWithStage(name);
460 nameWithStage.appendS32(stageNum);
461 append_varying(type, nameWithStage.c_str(), segments, vsOutName, fsInName);
462}
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000463}
464
bsalomon@google.com66105672011-09-15 15:12:00 +0000465void GrGLProgram::genEdgeCoverage(const GrGLInterface* gl,
466 GrVertexLayout layout,
467 CachedData* programData,
468 GrStringBuilder* coverageVar,
469 ShaderCodeSegments* segments) const {
470 if (fProgramDesc.fEdgeAANumEdges > 0) {
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000471 segments->fFSUnis.push_back().set(GrGLShaderVar::kVec3f_Type,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000472 GrGLShaderVar::kUniform_TypeModifier,
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000473 EDGES_UNI_NAME,
474 fProgramDesc.fEdgeAANumEdges);
bsalomon@google.com66105672011-09-15 15:12:00 +0000475 programData->fUniLocations.fEdgesUni = kUseUniform;
476 int count = fProgramDesc.fEdgeAANumEdges;
477 segments->fFSCode.append(
478 "\tvec3 pos = vec3(gl_FragCoord.xy, 1);\n");
479 for (int i = 0; i < count; i++) {
480 segments->fFSCode.append("\tfloat a");
481 segments->fFSCode.appendS32(i);
482 segments->fFSCode.append(" = clamp(dot(" EDGES_UNI_NAME "[");
483 segments->fFSCode.appendS32(i);
484 segments->fFSCode.append("], pos), 0.0, 1.0);\n");
485 }
486 if (fProgramDesc.fEdgeAAConcave && (count & 0x01) == 0) {
487 // For concave polys, we consider the edges in pairs.
488 segments->fFSFunctions.append("float cross2(vec2 a, vec2 b) {\n");
489 segments->fFSFunctions.append("\treturn dot(a, vec2(b.y, -b.x));\n");
490 segments->fFSFunctions.append("}\n");
491 for (int i = 0; i < count; i += 2) {
492 segments->fFSCode.appendf("\tfloat eb%d;\n", i / 2);
493 segments->fFSCode.appendf("\tif (cross2(" EDGES_UNI_NAME "[%d].xy, " EDGES_UNI_NAME "[%d].xy) < 0.0) {\n", i, i + 1);
494 segments->fFSCode.appendf("\t\teb%d = a%d * a%d;\n", i / 2, i, i + 1);
495 segments->fFSCode.append("\t} else {\n");
496 segments->fFSCode.appendf("\t\teb%d = a%d + a%d - a%d * a%d;\n", i / 2, i, i + 1, i, i + 1);
497 segments->fFSCode.append("\t}\n");
498 }
499 segments->fFSCode.append("\tfloat edgeAlpha = ");
500 for (int i = 0; i < count / 2 - 1; i++) {
501 segments->fFSCode.appendf("min(eb%d, ", i);
502 }
503 segments->fFSCode.appendf("eb%d", count / 2 - 1);
504 for (int i = 0; i < count / 2 - 1; i++) {
505 segments->fFSCode.append(")");
506 }
507 segments->fFSCode.append(";\n");
508 } else {
509 segments->fFSCode.append("\tfloat edgeAlpha = ");
510 for (int i = 0; i < count - 1; i++) {
511 segments->fFSCode.appendf("min(a%d * a%d, ", i, i + 1);
512 }
513 segments->fFSCode.appendf("a%d * a0", count - 1);
514 for (int i = 0; i < count - 1; i++) {
515 segments->fFSCode.append(")");
516 }
517 segments->fFSCode.append(";\n");
518 }
519 *coverageVar = "edgeAlpha";
520 } else if (layout & GrDrawTarget::kEdge_VertexLayoutBit) {
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000521 const char *vsName, *fsName;
tomhudson@google.com086e5352011-12-08 14:44:10 +0000522 append_varying(GrGLShaderVar::kVec4f_Type, "Edge", segments,
523 &vsName, &fsName);
524 segments->fVSAttrs.push_back().set(GrGLShaderVar::kVec4f_Type,
525 GrGLShaderVar::kAttribute_TypeModifier, EDGE_ATTR_NAME);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000526 segments->fVSCode.appendf("\t%s = " EDGE_ATTR_NAME ";\n", vsName);
tomhudson@google.com93813632011-10-27 20:21:16 +0000527 if (GrDrawState::kHairLine_EdgeType == fProgramDesc.fVertexEdgeType) {
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000528 segments->fFSCode.appendf("\tfloat edgeAlpha = abs(dot(vec3(gl_FragCoord.xy,1), %s.xyz));\n", fsName);
bsalomon@google.com66105672011-09-15 15:12:00 +0000529 } else {
tomhudson@google.com93813632011-10-27 20:21:16 +0000530 GrAssert(GrDrawState::kHairQuad_EdgeType == fProgramDesc.fVertexEdgeType);
bsalomon@google.com66105672011-09-15 15:12:00 +0000531 // for now we know we're not in perspective, so we could compute this
532 // per-quadratic rather than per pixel
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000533 segments->fFSCode.appendf("\tvec2 duvdx = dFdx(%s.xy);\n", fsName);
534 segments->fFSCode.appendf("\tvec2 duvdy = dFdy(%s.xy);\n", fsName);
535 segments->fFSCode.appendf("\tfloat dfdx = 2.0*%s.x*duvdx.x - duvdx.y;\n", fsName);
536 segments->fFSCode.appendf("\tfloat dfdy = 2.0*%s.x*duvdy.x - duvdy.y;\n", fsName);
537 segments->fFSCode.appendf("\tfloat edgeAlpha = (%s.x*%s.x - %s.y);\n", fsName, fsName, fsName);
bsalomon@google.com66105672011-09-15 15:12:00 +0000538 segments->fFSCode.append("\tedgeAlpha = sqrt(edgeAlpha*edgeAlpha / (dfdx*dfdx + dfdy*dfdy));\n");
bsalomon@google.com1dcf5062011-11-14 19:29:53 +0000539 if (gl->supportsES2()) {
bsalomon@google.com66105672011-09-15 15:12:00 +0000540 segments->fHeader.printf("#extension GL_OES_standard_derivatives: enable\n");
541 }
542 }
543 segments->fFSCode.append("\tedgeAlpha = max(1.0 - edgeAlpha, 0.0);\n");
544 *coverageVar = "edgeAlpha";
545 } else {
546 coverageVar->reset();
547 }
548}
549
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000550namespace {
551
552// returns true if the color output was explicitly declared or not.
tomhudson@google.com086e5352011-12-08 14:44:10 +0000553bool decl_and_get_fs_color_output(GrGLSLGeneration v,
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000554 VarArray* fsOutputs,
555 const char** name) {
556 switch (v) {
tomhudson@google.com086e5352011-12-08 14:44:10 +0000557 case k110_GLSLGeneration:
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000558 *name = "gl_FragColor";
559 return false;
560 break;
tomhudson@google.com086e5352011-12-08 14:44:10 +0000561 case k130_GLSLGeneration: // fallthru
562 case k150_GLSLGeneration:
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000563 *name = declared_color_output_name();
564 fsOutputs->push_back().set(GrGLShaderVar::kVec4f_Type,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000565 GrGLShaderVar::kOut_TypeModifier,
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000566 declared_color_output_name());
567 return true;
568 break;
569 default:
570 GrCrash("Unknown GLSL version.");
571 return false; // suppress warning
572 }
573}
574
bsalomon@google.com85b505b2011-11-07 14:56:51 +0000575void genInputColor(GrGLProgram::ProgramDesc::ColorInput colorInput,
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000576 GrGLProgram::CachedData* programData,
577 ShaderCodeSegments* segments,
578 GrStringBuilder* inColor) {
bsalomon@google.com85b505b2011-11-07 14:56:51 +0000579 switch (colorInput) {
580 case GrGLProgram::ProgramDesc::kAttribute_ColorInput: {
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000581 segments->fVSAttrs.push_back().set(GrGLShaderVar::kVec4f_Type,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000582 GrGLShaderVar::kAttribute_TypeModifier,
583 COL_ATTR_NAME);
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000584 const char *vsName, *fsName;
585 append_varying(GrGLShaderVar::kVec4f_Type, "Color", segments, &vsName, &fsName);
586 segments->fVSCode.appendf("\t%s = " COL_ATTR_NAME ";\n", vsName);
587 *inColor = fsName;
588 } break;
bsalomon@google.com85b505b2011-11-07 14:56:51 +0000589 case GrGLProgram::ProgramDesc::kUniform_ColorInput:
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000590 segments->fFSUnis.push_back().set(GrGLShaderVar::kVec4f_Type,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000591 GrGLShaderVar::kUniform_TypeModifier,
592 COL_UNI_NAME);
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000593 programData->fUniLocations.fColorUni = kUseUniform;
594 *inColor = COL_UNI_NAME;
595 break;
bsalomon@google.com85b505b2011-11-07 14:56:51 +0000596 case GrGLProgram::ProgramDesc::kTransBlack_ColorInput:
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000597 GrAssert(!"needComputedColor should be false.");
598 break;
bsalomon@google.com85b505b2011-11-07 14:56:51 +0000599 case GrGLProgram::ProgramDesc::kSolidWhite_ColorInput:
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000600 break;
601 default:
602 GrCrash("Unknown color type.");
603 break;
604 }
605}
606
607void genPerVertexCoverage(ShaderCodeSegments* segments,
608 GrStringBuilder* inCoverage) {
609 segments->fVSAttrs.push_back().set(GrGLShaderVar::kFloat_Type,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000610 GrGLShaderVar::kAttribute_TypeModifier,
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000611 COV_ATTR_NAME);
612 const char *vsName, *fsName;
613 append_varying(GrGLShaderVar::kFloat_Type, "Coverage",
614 segments, &vsName, &fsName);
615 segments->fVSCode.appendf("\t%s = " COV_ATTR_NAME ";\n", vsName);
616 if (inCoverage->size()) {
617 segments->fFSCode.appendf("\tfloat edgeAndAttrCov = %s * %s;\n",
618 fsName, inCoverage->c_str());
619 *inCoverage = "edgeAndAttrCov";
620 } else {
621 *inCoverage = fsName;
622 }
623}
624
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000625}
626
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000627void GrGLProgram::genGeometryShader(const GrGLInterface* gl,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000628 GrGLSLGeneration glslGeneration,
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000629 ShaderCodeSegments* segments) const {
630#if GR_GL_EXPERIMENTAL_GS
631 if (fProgramDesc.fExperimentalGS) {
tomhudson@google.com086e5352011-12-08 14:44:10 +0000632 GrAssert(glslGeneration >= k150_GLSLGeneration);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000633 segments->fGSHeader.append("layout(triangles) in;\n"
634 "layout(triangle_strip, max_vertices = 6) out;\n");
635 segments->fGSCode.append("void main() {\n"
636 "\tfor (int i = 0; i < 3; ++i) {\n"
637 "\t\tgl_Position = gl_in[i].gl_Position;\n");
638 if (this->fProgramDesc.fEmitsPointSize) {
639 segments->fGSCode.append("\t\tgl_PointSize = 1.0;\n");
640 }
641 GrAssert(segments->fGSInputs.count() == segments->fGSOutputs.count());
642 int count = segments->fGSInputs.count();
643 for (int i = 0; i < count; ++i) {
644 segments->fGSCode.appendf("\t\t%s = %s[i];\n",
645 segments->fGSOutputs[i].getName().c_str(),
646 segments->fGSInputs[i].getName().c_str());
647 }
648 segments->fGSCode.append("\t\tEmitVertex();\n"
649 "\t}\n"
650 "\tEndPrimitive();\n"
651 "}\n");
652 }
653#endif
654}
655
senorblanco@chromium.orgb3a39b52012-01-05 18:28:56 +0000656const char* GrGLProgram::adjustInColor(const GrStringBuilder& inColor) const {
657 const char* color;
658 if (inColor.size()) {
659 return inColor.c_str();
660 } else {
661 if (ProgramDesc::kSolidWhite_ColorInput == fProgramDesc.fColorInput) {
662 return all_ones_vec(4);
663 } else {
664 return all_zeros_vec(4);
665 }
666 }
667}
668
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000669bool GrGLProgram::genProgram(const GrGLInterface* gl,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000670 GrGLSLGeneration glslGeneration,
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000671 GrGLProgram::CachedData* programData) const {
junov@google.comf93e7172011-03-31 21:26:24 +0000672
673 ShaderCodeSegments segments;
674 const uint32_t& layout = fProgramDesc.fVertexLayout;
675
bsalomon@google.com91961302011-05-09 18:39:58 +0000676 programData->fUniLocations.reset();
junov@google.comf93e7172011-03-31 21:26:24 +0000677
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000678#if GR_GL_EXPERIMENTAL_GS
679 segments.fUsesGS = fProgramDesc.fExperimentalGS;
680#endif
681
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000682 SkXfermode::Coeff colorCoeff, uniformCoeff;
683 // The rest of transfer mode color filters have not been implemented
684 if (fProgramDesc.fColorFilterXfermode < SkXfermode::kCoeffModesCnt) {
685 GR_DEBUGCODE(bool success =)
tomhudson@google.com0d831722011-06-02 15:37:14 +0000686 SkXfermode::ModeAsCoeff(static_cast<SkXfermode::Mode>
687 (fProgramDesc.fColorFilterXfermode),
688 &uniformCoeff, &colorCoeff);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000689 GR_DEBUGASSERT(success);
690 } else {
691 colorCoeff = SkXfermode::kOne_Coeff;
692 uniformCoeff = SkXfermode::kZero_Coeff;
693 }
694
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000695 // If we know the final color is going to be all zeros then we can
696 // simplify the color filter coeffecients. needComputedColor will then
697 // come out false below.
bsalomon@google.com85b505b2011-11-07 14:56:51 +0000698 if (ProgramDesc::kTransBlack_ColorInput == fProgramDesc.fColorInput) {
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000699 colorCoeff = SkXfermode::kZero_Coeff;
700 if (SkXfermode::kDC_Coeff == uniformCoeff ||
701 SkXfermode::kDA_Coeff == uniformCoeff) {
702 uniformCoeff = SkXfermode::kZero_Coeff;
703 } else if (SkXfermode::kIDC_Coeff == uniformCoeff ||
704 SkXfermode::kIDA_Coeff == uniformCoeff) {
705 uniformCoeff = SkXfermode::kOne_Coeff;
706 }
707 }
708
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000709 bool needColorFilterUniform;
710 bool needComputedColor;
711 needBlendInputs(uniformCoeff, colorCoeff,
712 &needColorFilterUniform, &needComputedColor);
713
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000714 // the dual source output has no canonical var name, have to
715 // declare an output, which is incompatible with gl_FragColor/gl_FragData.
bsalomon@google.com2ec72802011-09-21 21:46:03 +0000716 const char* fsColorOutput = NULL;
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000717 bool dualSourceOutputWritten = false;
tomhudson@google.com086e5352011-12-08 14:44:10 +0000718 segments.fHeader.printf(glsl_version_string(gl, glslGeneration));
719 bool isColorDeclared = decl_and_get_fs_color_output(glslGeneration,
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000720 &segments.fFSOutputs,
721 &fsColorOutput);
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000722
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000723#if GR_GL_ATTRIBUTE_MATRICES
tomhudson@google.com086e5352011-12-08 14:44:10 +0000724 segments.fVSAttrs.push_back().set(GrGLShaderVar::kMat33f_Type,
725 GrGLShaderVar::kAttribute_TypeModifier, VIEW_MATRIX_NAME);
bsalomon@google.com91961302011-05-09 18:39:58 +0000726 programData->fUniLocations.fViewMatrixUni = kSetAsAttribute;
junov@google.comf93e7172011-03-31 21:26:24 +0000727#else
tomhudson@google.com086e5352011-12-08 14:44:10 +0000728 segments.fVSUnis.push_back().set(GrGLShaderVar::kMat33f_Type,
729 GrGLShaderVar::kUniform_TypeModifier, VIEW_MATRIX_NAME);
bsalomon@google.com91961302011-05-09 18:39:58 +0000730 programData->fUniLocations.fViewMatrixUni = kUseUniform;
junov@google.comf93e7172011-03-31 21:26:24 +0000731#endif
tomhudson@google.com086e5352011-12-08 14:44:10 +0000732 segments.fVSAttrs.push_back().set(GrGLShaderVar::kVec2f_Type,
733 GrGLShaderVar::kAttribute_TypeModifier, POS_ATTR_NAME);
junov@google.comf93e7172011-03-31 21:26:24 +0000734
bsalomon@google.com91961302011-05-09 18:39:58 +0000735 segments.fVSCode.append(
736 "void main() {\n"
737 "\tvec3 pos3 = " VIEW_MATRIX_NAME " * vec3("POS_ATTR_NAME", 1);\n"
738 "\tgl_Position = vec4(pos3.xy, 0, pos3.z);\n");
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000739
bsalomon@google.com91961302011-05-09 18:39:58 +0000740 // incoming color to current stage being processed.
741 GrStringBuilder inColor;
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000742
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000743 if (needComputedColor) {
bsalomon@google.com85b505b2011-11-07 14:56:51 +0000744 genInputColor((ProgramDesc::ColorInput) fProgramDesc.fColorInput,
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000745 programData, &segments, &inColor);
junov@google.comf93e7172011-03-31 21:26:24 +0000746 }
747
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000748 // we output point size in the GS if present
749 if (fProgramDesc.fEmitsPointSize && !segments.fUsesGS){
bsalomon@google.com91961302011-05-09 18:39:58 +0000750 segments.fVSCode.append("\tgl_PointSize = 1.0;\n");
junov@google.comf93e7172011-03-31 21:26:24 +0000751 }
bsalomon@google.com4be283f2011-04-19 21:15:09 +0000752
bsalomon@google.com91961302011-05-09 18:39:58 +0000753 segments.fFSCode.append("void main() {\n");
junov@google.comf93e7172011-03-31 21:26:24 +0000754
755 // add texture coordinates that are used to the list of vertex attr decls
tomhudson@google.com93813632011-10-27 20:21:16 +0000756 GrStringBuilder texCoordAttrs[GrDrawState::kMaxTexCoords];
757 for (int t = 0; t < GrDrawState::kMaxTexCoords; ++t) {
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +0000758 if (GrDrawTarget::VertexUsesTexCoordIdx(t, layout)) {
junov@google.comf93e7172011-03-31 21:26:24 +0000759 tex_attr_name(t, texCoordAttrs + t);
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000760 segments.fVSAttrs.push_back().set(GrGLShaderVar::kVec2f_Type,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000761 GrGLShaderVar::kAttribute_TypeModifier,
762 texCoordAttrs[t].c_str());
junov@google.comf93e7172011-03-31 21:26:24 +0000763 }
764 }
765
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000766 ///////////////////////////////////////////////////////////////////////////
767 // compute the final color
Scroggo97c88c22011-05-11 14:05:25 +0000768
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000769 // if we have color stages string them together, feeding the output color
770 // of each to the next and generating code for each stage.
771 if (needComputedColor) {
772 GrStringBuilder outColor;
773 for (int s = 0; s < fProgramDesc.fFirstCoverageStage; ++s) {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000774 if (fProgramDesc.fStages[s].isEnabled()) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000775 // create var to hold stage result
776 outColor = "color";
777 outColor.appendS32(s);
778 segments.fFSCode.appendf("\tvec4 %s;\n", outColor.c_str());
779
780 const char* inCoords;
781 // figure out what our input coords are
782 if (GrDrawTarget::StagePosAsTexCoordVertexLayoutBit(s) &
783 layout) {
784 inCoords = POS_ATTR_NAME;
Scroggo97c88c22011-05-11 14:05:25 +0000785 } else {
786 int tcIdx = GrDrawTarget::VertexTexCoordsForStage(s, layout);
787 // we better have input tex coordinates if stage is enabled.
788 GrAssert(tcIdx >= 0);
789 GrAssert(texCoordAttrs[tcIdx].size());
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000790 inCoords = texCoordAttrs[tcIdx].c_str();
junov@google.comf93e7172011-03-31 21:26:24 +0000791 }
792
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000793 this->genStageCode(gl,
794 s,
795 fProgramDesc.fStages[s],
796 inColor.size() ? inColor.c_str() : NULL,
797 outColor.c_str(),
798 inCoords,
799 &segments,
800 &programData->fUniLocations.fStages[s]);
junov@google.comf93e7172011-03-31 21:26:24 +0000801 inColor = outColor;
junov@google.comf93e7172011-03-31 21:26:24 +0000802 }
803 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000804 }
Scroggo97c88c22011-05-11 14:05:25 +0000805
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000806 // if have all ones or zeros for the "dst" input to the color filter then we
807 // may be able to make additional optimizations.
808 if (needColorFilterUniform && needComputedColor && !inColor.size()) {
bsalomon@google.com85b505b2011-11-07 14:56:51 +0000809 GrAssert(ProgramDesc::kSolidWhite_ColorInput == fProgramDesc.fColorInput);
bsalomon@google.com86c1f712011-10-12 14:54:26 +0000810 bool uniformCoeffIsZero = SkXfermode::kIDC_Coeff == uniformCoeff ||
811 SkXfermode::kIDA_Coeff == uniformCoeff;
812 if (uniformCoeffIsZero) {
813 uniformCoeff = SkXfermode::kZero_Coeff;
814 bool bogus;
815 needBlendInputs(SkXfermode::kZero_Coeff, colorCoeff,
816 &needColorFilterUniform, &bogus);
817 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000818 }
819 if (needColorFilterUniform) {
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000820 segments.fFSUnis.push_back().set(GrGLShaderVar::kVec4f_Type,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000821 GrGLShaderVar::kUniform_TypeModifier,
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000822 COL_FILTER_UNI_NAME);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000823 programData->fUniLocations.fColorFilterUni = kUseUniform;
824 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000825 bool wroteFragColorZero = false;
tomhudson@google.com0d831722011-06-02 15:37:14 +0000826 if (SkXfermode::kZero_Coeff == uniformCoeff &&
senorblanco@chromium.org50bdad82012-01-03 20:51:57 +0000827 SkXfermode::kZero_Coeff == colorCoeff &&
828 !fProgramDesc.fColorMatrixEnabled) {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000829 segments.fFSCode.appendf("\t%s = %s;\n",
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000830 fsColorOutput,
831 all_zeros_vec(4));
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000832 wroteFragColorZero = true;
833 } else if (SkXfermode::kDst_Mode != fProgramDesc.fColorFilterXfermode) {
834 segments.fFSCode.appendf("\tvec4 filteredColor;\n");
senorblanco@chromium.orgb3a39b52012-01-05 18:28:56 +0000835 const char* color = adjustInColor(inColor);
tomhudson@google.com0d831722011-06-02 15:37:14 +0000836 addColorFilter(&segments.fFSCode, "filteredColor", uniformCoeff,
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000837 colorCoeff, color);
838 inColor = "filteredColor";
839 }
senorblanco@chromium.org50bdad82012-01-03 20:51:57 +0000840 if (fProgramDesc.fColorMatrixEnabled) {
841 segments.fFSUnis.push_back().set(GrGLShaderVar::kMat44f_Type,
842 GrGLShaderVar::kUniform_TypeModifier,
843 COL_MATRIX_UNI_NAME);
844 segments.fFSUnis.push_back().set(GrGLShaderVar::kVec4f_Type,
845 GrGLShaderVar::kUniform_TypeModifier,
846 COL_MATRIX_VEC_UNI_NAME);
847 programData->fUniLocations.fColorMatrixUni = kUseUniform;
848 programData->fUniLocations.fColorMatrixVecUni = kUseUniform;
849 segments.fFSCode.appendf("\tvec4 matrixedColor;\n");
senorblanco@chromium.orgb3a39b52012-01-05 18:28:56 +0000850 const char* color = adjustInColor(inColor);
851 addColorMatrix(&segments.fFSCode, "matrixedColor", color);
senorblanco@chromium.org50bdad82012-01-03 20:51:57 +0000852 inColor = "matrixedColor";
853 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000854
855 ///////////////////////////////////////////////////////////////////////////
856 // compute the partial coverage (coverage stages and edge aa)
857
858 GrStringBuilder inCoverage;
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000859
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000860 // we don't need to compute coverage at all if we know the final shader
861 // output will be zero and we don't have a dual src blend output.
862 if (!wroteFragColorZero ||
863 ProgramDesc::kNone_DualSrcOutput != fProgramDesc.fDualSrcOutput) {
bsalomon@google.com66105672011-09-15 15:12:00 +0000864
865 // get edge AA coverage and use it as inCoverage to first coverage stage
866 this->genEdgeCoverage(gl, layout, programData, &inCoverage, &segments);
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000867
bsalomon@google.coma3108262011-10-10 14:08:47 +0000868 // include explicit per-vertex coverage if we have it
869 if (GrDrawTarget::kCoverage_VertexLayoutBit & layout) {
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +0000870 genPerVertexCoverage(&segments, &inCoverage);
bsalomon@google.coma3108262011-10-10 14:08:47 +0000871 }
872
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000873 GrStringBuilder outCoverage;
874 const int& startStage = fProgramDesc.fFirstCoverageStage;
tomhudson@google.com93813632011-10-27 20:21:16 +0000875 for (int s = startStage; s < GrDrawState::kNumStages; ++s) {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000876 if (fProgramDesc.fStages[s].isEnabled()) {
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000877 // create var to hold stage output
878 outCoverage = "coverage";
879 outCoverage.appendS32(s);
880 segments.fFSCode.appendf("\tvec4 %s;\n", outCoverage.c_str());
881
882 const char* inCoords;
883 // figure out what our input coords are
884 if (GrDrawTarget::StagePosAsTexCoordVertexLayoutBit(s) & layout) {
885 inCoords = POS_ATTR_NAME;
886 } else {
887 int tcIdx = GrDrawTarget::VertexTexCoordsForStage(s, layout);
888 // we better have input tex coordinates if stage is enabled.
889 GrAssert(tcIdx >= 0);
890 GrAssert(texCoordAttrs[tcIdx].size());
891 inCoords = texCoordAttrs[tcIdx].c_str();
892 }
893
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000894 genStageCode(gl, s,
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000895 fProgramDesc.fStages[s],
896 inCoverage.size() ? inCoverage.c_str() : NULL,
897 outCoverage.c_str(),
898 inCoords,
899 &segments,
900 &programData->fUniLocations.fStages[s]);
901 inCoverage = outCoverage;
senorblanco@chromium.org92e0f222011-05-12 15:49:15 +0000902 }
903 }
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000904 if (ProgramDesc::kNone_DualSrcOutput != fProgramDesc.fDualSrcOutput) {
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000905 segments.fFSOutputs.push_back().set(GrGLShaderVar::kVec4f_Type,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000906 GrGLShaderVar::kOut_TypeModifier,
907 dual_source_output_name());
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000908 bool outputIsZero = false;
909 GrStringBuilder coeff;
tomhudson@google.com0d831722011-06-02 15:37:14 +0000910 if (ProgramDesc::kCoverage_DualSrcOutput !=
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000911 fProgramDesc.fDualSrcOutput && !wroteFragColorZero) {
912 if (!inColor.size()) {
913 outputIsZero = true;
914 } else {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000915 if (fProgramDesc.fDualSrcOutput ==
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000916 ProgramDesc::kCoverageISA_DualSrcOutput) {
917 coeff.printf("(1 - %s.a)", inColor.c_str());
918 } else {
919 coeff.printf("(vec4(1,1,1,1) - %s)", inColor.c_str());
920 }
921 }
922 }
923 if (outputIsZero) {
tomhudson@google.com0d831722011-06-02 15:37:14 +0000924 segments.fFSCode.appendf("\t%s = %s;\n",
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000925 dual_source_output_name(),
926 all_zeros_vec(4));
927 } else {
928 modulate_helper(dual_source_output_name(),
929 coeff.c_str(),
930 inCoverage.c_str(),
931 &segments.fFSCode);
932 }
933 dualSourceOutputWritten = true;
934 }
junov@google.comf93e7172011-03-31 21:26:24 +0000935 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000936
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000937 ///////////////////////////////////////////////////////////////////////////
938 // combine color and coverage as frag color
939
940 if (!wroteFragColorZero) {
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000941 modulate_helper(fsColorOutput,
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000942 inColor.c_str(),
943 inCoverage.c_str(),
944 &segments.fFSCode);
bsalomon@google.comc4364992011-11-07 15:54:49 +0000945 if (ProgramDesc::kNo_OutputPM == fProgramDesc.fOutputPM) {
bsalomon@google.come63799d2011-11-09 14:14:38 +0000946 segments.fFSCode.appendf("\t%s = %s.a <= 0.0 ? vec4(0,0,0,0) : vec4(%s.rgb / %s.a, %s.a);\n",
947 fsColorOutput,
bsalomon@google.comc4364992011-11-07 15:54:49 +0000948 fsColorOutput,
949 fsColorOutput,
950 fsColorOutput,
951 fsColorOutput);
952 }
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000953 }
954
bsalomon@google.com91961302011-05-09 18:39:58 +0000955 segments.fVSCode.append("}\n");
956 segments.fFSCode.append("}\n");
957
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000958 ///////////////////////////////////////////////////////////////////////////
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000959 // insert GS
960#if GR_DEBUG
tomhudson@google.com086e5352011-12-08 14:44:10 +0000961 this->genGeometryShader(gl, glslGeneration, &segments);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000962#endif
963
964 ///////////////////////////////////////////////////////////////////////////
bsalomon@google.comf2d91552011-05-16 20:56:06 +0000965 // compile and setup attribs and unis
966
tomhudson@google.com086e5352011-12-08 14:44:10 +0000967 if (!CompileShaders(gl, glslGeneration, segments, programData)) {
bsalomon@google.com91961302011-05-09 18:39:58 +0000968 return false;
969 }
970
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000971 if (!this->bindOutputsAttribsAndLinkProgram(gl, texCoordAttrs,
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000972 isColorDeclared,
bsalomon@google.com271cffc2011-05-20 14:13:56 +0000973 dualSourceOutputWritten,
974 programData)) {
bsalomon@google.com91961302011-05-09 18:39:58 +0000975 return false;
976 }
977
bsalomon@google.com0b77d682011-08-19 13:28:54 +0000978 this->getUniformLocationsAndInitCache(gl, programData);
bsalomon@google.com91961302011-05-09 18:39:58 +0000979
980 return true;
981}
982
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000983namespace {
984
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000985inline void expand_decls(const VarArray& vars,
986 const GrGLInterface* gl,
tomhudson@google.com086e5352011-12-08 14:44:10 +0000987 GrStringBuilder* string,
988 GrGLSLGeneration gen) {
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000989 const int count = vars.count();
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000990 for (int i = 0; i < count; ++i) {
tomhudson@google.com086e5352011-12-08 14:44:10 +0000991 vars[i].appendDecl(gl, string, gen);
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000992 }
993}
994
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +0000995inline void print_shader(int stringCnt,
996 const char** strings,
997 int* stringLengths) {
bsalomon@google.com4fa66942011-09-20 19:06:12 +0000998 for (int i = 0; i < stringCnt; ++i) {
999 if (NULL == stringLengths || stringLengths[i] < 0) {
1000 GrPrintf(strings[i]);
1001 } else {
1002 GrPrintf("%.*s", stringLengths[i], strings[i]);
1003 }
1004 }
1005}
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001006
1007typedef SkTArray<const char*, true> StrArray;
1008#define PREALLOC_STR_ARRAY(N) SkSTArray<(N), const char*, true>
1009
1010typedef SkTArray<int, true> LengthArray;
1011#define PREALLOC_LENGTH_ARRAY(N) SkSTArray<(N), int, true>
1012
1013// these shouldn't relocate
1014typedef GrTAllocator<GrStringBuilder> TempArray;
1015#define PREALLOC_TEMP_ARRAY(N) GrSTAllocator<(N), GrStringBuilder>
1016
1017inline void append_string(const GrStringBuilder& str,
1018 StrArray* strings,
1019 LengthArray* lengths) {
1020 int length = (int) str.size();
1021 if (length) {
1022 strings->push_back(str.c_str());
1023 lengths->push_back(length);
1024 }
1025 GrAssert(strings->count() == lengths->count());
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001026}
1027
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001028inline void append_decls(const VarArray& vars,
1029 const GrGLInterface* gl,
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001030 StrArray* strings,
1031 LengthArray* lengths,
tomhudson@google.com086e5352011-12-08 14:44:10 +00001032 TempArray* temp,
1033 GrGLSLGeneration gen) {
1034 expand_decls(vars, gl, &temp->push_back(), gen);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001035 append_string(temp->back(), strings, lengths);
1036}
1037
1038}
1039
1040bool GrGLProgram::CompileShaders(const GrGLInterface* gl,
tomhudson@google.com086e5352011-12-08 14:44:10 +00001041 GrGLSLGeneration glslGeneration,
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001042 const ShaderCodeSegments& segments,
bsalomon@google.com91961302011-05-09 18:39:58 +00001043 CachedData* programData) {
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001044 enum { kPreAllocStringCnt = 8 };
junov@google.comf93e7172011-03-31 21:26:24 +00001045
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001046 PREALLOC_STR_ARRAY(kPreAllocStringCnt) strs;
1047 PREALLOC_LENGTH_ARRAY(kPreAllocStringCnt) lengths;
1048 PREALLOC_TEMP_ARRAY(kPreAllocStringCnt) temps;
1049
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001050 GrStringBuilder unis;
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001051 GrStringBuilder inputs;
1052 GrStringBuilder outputs;
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001053
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001054 append_string(segments.fHeader, &strs, &lengths);
tomhudson@google.com086e5352011-12-08 14:44:10 +00001055 append_decls(segments.fVSUnis, gl, &strs, &lengths, &temps, glslGeneration);
1056 append_decls(segments.fVSAttrs, gl, &strs, &lengths,
1057 &temps, glslGeneration);
1058 append_decls(segments.fVSOutputs, gl, &strs, &lengths,
1059 &temps, glslGeneration);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001060 append_string(segments.fVSCode, &strs, &lengths);
junov@google.comf93e7172011-03-31 21:26:24 +00001061
1062#if PRINT_SHADERS
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001063 print_shader(strs.count(), &strs[0], &lengths[0]);
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001064 GrPrintf("\n");
junov@google.comf93e7172011-03-31 21:26:24 +00001065#endif
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001066
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001067 programData->fVShaderID =
1068 CompileShader(gl, GR_GL_VERTEX_SHADER, strs.count(),
1069 &strs[0], &lengths[0]);
junov@google.comf93e7172011-03-31 21:26:24 +00001070
bsalomon@google.com91961302011-05-09 18:39:58 +00001071 if (!programData->fVShaderID) {
1072 return false;
1073 }
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001074 if (segments.fUsesGS) {
1075 strs.reset();
1076 lengths.reset();
1077 temps.reset();
1078 append_string(segments.fHeader, &strs, &lengths);
1079 append_string(segments.fGSHeader, &strs, &lengths);
tomhudson@google.com086e5352011-12-08 14:44:10 +00001080 append_decls(segments.fGSInputs, gl, &strs, &lengths,
1081 &temps, glslGeneration);
1082 append_decls(segments.fGSOutputs, gl, &strs, &lengths,
1083 &temps, glslGeneration);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001084 append_string(segments.fGSCode, &strs, &lengths);
1085#if PRINT_SHADERS
1086 print_shader(strs.count(), &strs[0], &lengths[0]);
1087 GrPrintf("\n");
1088#endif
1089 programData->fGShaderID =
1090 CompileShader(gl, GR_GL_GEOMETRY_SHADER, strs.count(),
1091 &strs[0], &lengths[0]);
1092 } else {
1093 programData->fGShaderID = 0;
senorblanco@chromium.org129b8e32011-06-15 17:52:09 +00001094 }
junov@google.comf93e7172011-03-31 21:26:24 +00001095
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001096 strs.reset();
1097 lengths.reset();
1098 temps.reset();
1099
1100 append_string(segments.fHeader, &strs, &lengths);
1101 GrStringBuilder precisionStr(GrShaderPrecision(gl));
1102 append_string(precisionStr, &strs, &lengths);
tomhudson@google.com086e5352011-12-08 14:44:10 +00001103 append_decls(segments.fFSUnis, gl, &strs, &lengths, &temps, glslGeneration);
1104 append_decls(segments.fFSInputs, gl, &strs, &lengths,
1105 &temps, glslGeneration);
bsalomon@google.com373a6632011-10-19 20:43:20 +00001106 // We shouldn't have declared outputs on 1.10
tomhudson@google.com086e5352011-12-08 14:44:10 +00001107 GrAssert(k110_GLSLGeneration != glslGeneration ||
1108 segments.fFSOutputs.empty());
1109 append_decls(segments.fFSOutputs, gl, &strs, &lengths,
1110 &temps, glslGeneration);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001111 append_string(segments.fFSFunctions, &strs, &lengths);
1112 append_string(segments.fFSCode, &strs, &lengths);
junov@google.comf93e7172011-03-31 21:26:24 +00001113
1114#if PRINT_SHADERS
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001115 print_shader(strs.count(), &strs[0], &lengths[0]);
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001116 GrPrintf("\n");
junov@google.comf93e7172011-03-31 21:26:24 +00001117#endif
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001118
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001119 programData->fFShaderID =
1120 CompileShader(gl, GR_GL_FRAGMENT_SHADER, strs.count(),
1121 &strs[0], &lengths[0]);
junov@google.comf93e7172011-03-31 21:26:24 +00001122
bsalomon@google.com91961302011-05-09 18:39:58 +00001123 if (!programData->fFShaderID) {
1124 return false;
junov@google.comf93e7172011-03-31 21:26:24 +00001125 }
bsalomon@google.com271cffc2011-05-20 14:13:56 +00001126
bsalomon@google.com91961302011-05-09 18:39:58 +00001127 return true;
junov@google.comf93e7172011-03-31 21:26:24 +00001128}
1129
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001130GrGLuint GrGLProgram::CompileShader(const GrGLInterface* gl,
1131 GrGLenum type,
1132 int stringCnt,
1133 const char** strings,
1134 int* stringLengths) {
tomhudson@google.com278cbb42011-06-30 19:37:01 +00001135 SK_TRACE_EVENT1("GrGLProgram::CompileShader",
1136 "stringCount", SkStringPrintf("%i", stringCnt).c_str());
1137
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001138 GrGLuint shader;
1139 GR_GL_CALL_RET(gl, shader, CreateShader(type));
junov@google.comf93e7172011-03-31 21:26:24 +00001140 if (0 == shader) {
1141 return 0;
1142 }
1143
1144 GrGLint compiled = GR_GL_INIT_ZERO;
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001145 GR_GL_CALL(gl, ShaderSource(shader, stringCnt, strings, stringLengths));
1146 GR_GL_CALL(gl, CompileShader(shader));
1147 GR_GL_CALL(gl, GetShaderiv(shader, GR_GL_COMPILE_STATUS, &compiled));
junov@google.comf93e7172011-03-31 21:26:24 +00001148
1149 if (!compiled) {
1150 GrGLint infoLen = GR_GL_INIT_ZERO;
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001151 GR_GL_CALL(gl, GetShaderiv(shader, GR_GL_INFO_LOG_LENGTH, &infoLen));
bsalomon@google.com3582bf92011-06-30 21:32:31 +00001152 SkAutoMalloc log(sizeof(char)*(infoLen+1)); // outside if for debugger
junov@google.comf93e7172011-03-31 21:26:24 +00001153 if (infoLen > 0) {
bsalomon@google.com79afcaa2011-09-14 14:29:18 +00001154 // retrieve length even though we don't need it to workaround
1155 // bug in chrome cmd buffer param validation.
1156 GrGLsizei length = GR_GL_INIT_ZERO;
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001157 GR_GL_CALL(gl, GetShaderInfoLog(shader, infoLen+1,
bsalomon@google.com79afcaa2011-09-14 14:29:18 +00001158 &length, (char*)log.get()));
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001159 print_shader(stringCnt, strings, stringLengths);
junov@google.comf93e7172011-03-31 21:26:24 +00001160 GrPrintf("\n%s", log.get());
1161 }
1162 GrAssert(!"Shader compilation failed!");
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001163 GR_GL_CALL(gl, DeleteShader(shader));
junov@google.comf93e7172011-03-31 21:26:24 +00001164 return 0;
1165 }
1166 return shader;
1167}
1168
bsalomon@google.com271cffc2011-05-20 14:13:56 +00001169bool GrGLProgram::bindOutputsAttribsAndLinkProgram(
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001170 const GrGLInterface* gl,
bsalomon@google.com271cffc2011-05-20 14:13:56 +00001171 GrStringBuilder texCoordAttrNames[],
1172 bool bindColorOut,
tomhudson@google.com0d831722011-06-02 15:37:14 +00001173 bool bindDualSrcOut,
bsalomon@google.com271cffc2011-05-20 14:13:56 +00001174 CachedData* programData) const {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001175 GR_GL_CALL_RET(gl, programData->fProgramID, CreateProgram());
bsalomon@google.com91961302011-05-09 18:39:58 +00001176 if (!programData->fProgramID) {
1177 return false;
1178 }
1179 const GrGLint& progID = programData->fProgramID;
1180
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001181 GR_GL_CALL(gl, AttachShader(progID, programData->fVShaderID));
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001182 if (programData->fGShaderID) {
1183 GR_GL_CALL(gl, AttachShader(progID, programData->fGShaderID));
1184 }
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001185 GR_GL_CALL(gl, AttachShader(progID, programData->fFShaderID));
bsalomon@google.com91961302011-05-09 18:39:58 +00001186
bsalomon@google.com271cffc2011-05-20 14:13:56 +00001187 if (bindColorOut) {
bsalomon@google.combc5cf512011-09-21 16:21:07 +00001188 GR_GL_CALL(gl, BindFragDataLocation(programData->fProgramID,
1189 0, declared_color_output_name()));
bsalomon@google.com271cffc2011-05-20 14:13:56 +00001190 }
1191 if (bindDualSrcOut) {
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001192 GR_GL_CALL(gl, BindFragDataLocationIndexed(programData->fProgramID,
bsalomon@google.com271cffc2011-05-20 14:13:56 +00001193 0, 1, dual_source_output_name()));
1194 }
1195
bsalomon@google.com91961302011-05-09 18:39:58 +00001196 // Bind the attrib locations to same values for all shaders
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001197 GR_GL_CALL(gl, BindAttribLocation(progID, PositionAttributeIdx(),
1198 POS_ATTR_NAME));
tomhudson@google.com93813632011-10-27 20:21:16 +00001199 for (int t = 0; t < GrDrawState::kMaxTexCoords; ++t) {
bsalomon@google.com91961302011-05-09 18:39:58 +00001200 if (texCoordAttrNames[t].size()) {
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001201 GR_GL_CALL(gl, BindAttribLocation(progID,
1202 TexCoordAttributeIdx(t),
1203 texCoordAttrNames[t].c_str()));
bsalomon@google.com91961302011-05-09 18:39:58 +00001204 }
1205 }
1206
bsalomon@google.com91961302011-05-09 18:39:58 +00001207 if (kSetAsAttribute == programData->fUniLocations.fViewMatrixUni) {
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001208 GR_GL_CALL(gl, BindAttribLocation(progID,
1209 ViewMatrixAttributeIdx(),
1210 VIEW_MATRIX_NAME));
bsalomon@google.com91961302011-05-09 18:39:58 +00001211 }
1212
tomhudson@google.com93813632011-10-27 20:21:16 +00001213 for (int s = 0; s < GrDrawState::kNumStages; ++s) {
bsalomon@google.com72b4fcb2011-05-09 20:47:34 +00001214 const StageUniLocations& unis = programData->fUniLocations.fStages[s];
1215 if (kSetAsAttribute == unis.fTextureMatrixUni) {
bsalomon@google.com91961302011-05-09 18:39:58 +00001216 GrStringBuilder matName;
1217 tex_matrix_name(s, &matName);
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001218 GR_GL_CALL(gl, BindAttribLocation(progID,
1219 TextureMatrixAttributeIdx(s),
1220 matName.c_str()));
bsalomon@google.com91961302011-05-09 18:39:58 +00001221 }
1222 }
1223
bsalomon@google.comaeb21602011-08-30 18:13:44 +00001224 GR_GL_CALL(gl, BindAttribLocation(progID, ColorAttributeIdx(),
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001225 COL_ATTR_NAME));
bsalomon@google.coma3108262011-10-10 14:08:47 +00001226 GR_GL_CALL(gl, BindAttribLocation(progID, CoverageAttributeIdx(),
1227 COV_ATTR_NAME));
bsalomon@google.comaeb21602011-08-30 18:13:44 +00001228 GR_GL_CALL(gl, BindAttribLocation(progID, EdgeAttributeIdx(),
1229 EDGE_ATTR_NAME));
bsalomon@google.com91961302011-05-09 18:39:58 +00001230
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001231 GR_GL_CALL(gl, LinkProgram(progID));
bsalomon@google.com91961302011-05-09 18:39:58 +00001232
1233 GrGLint linked = GR_GL_INIT_ZERO;
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001234 GR_GL_CALL(gl, GetProgramiv(progID, GR_GL_LINK_STATUS, &linked));
bsalomon@google.com91961302011-05-09 18:39:58 +00001235 if (!linked) {
1236 GrGLint infoLen = GR_GL_INIT_ZERO;
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001237 GR_GL_CALL(gl, GetProgramiv(progID, GR_GL_INFO_LOG_LENGTH, &infoLen));
bsalomon@google.com3582bf92011-06-30 21:32:31 +00001238 SkAutoMalloc log(sizeof(char)*(infoLen+1)); // outside if for debugger
bsalomon@google.com91961302011-05-09 18:39:58 +00001239 if (infoLen > 0) {
bsalomon@google.com79afcaa2011-09-14 14:29:18 +00001240 // retrieve length even though we don't need it to workaround
1241 // bug in chrome cmd buffer param validation.
1242 GrGLsizei length = GR_GL_INIT_ZERO;
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001243 GR_GL_CALL(gl, GetProgramInfoLog(progID, infoLen+1,
bsalomon@google.com79afcaa2011-09-14 14:29:18 +00001244 &length, (char*)log.get()));
bsalomon@google.com91961302011-05-09 18:39:58 +00001245 GrPrintf((char*)log.get());
1246 }
1247 GrAssert(!"Error linking program");
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001248 GR_GL_CALL(gl, DeleteProgram(progID));
bsalomon@google.com91961302011-05-09 18:39:58 +00001249 programData->fProgramID = 0;
1250 return false;
1251 }
1252 return true;
1253}
1254
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001255void GrGLProgram::getUniformLocationsAndInitCache(const GrGLInterface* gl,
1256 CachedData* programData) const {
bsalomon@google.com91961302011-05-09 18:39:58 +00001257 const GrGLint& progID = programData->fProgramID;
1258
1259 if (kUseUniform == programData->fUniLocations.fViewMatrixUni) {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001260 GR_GL_CALL_RET(gl, programData->fUniLocations.fViewMatrixUni,
1261 GetUniformLocation(progID, VIEW_MATRIX_NAME));
bsalomon@google.com91961302011-05-09 18:39:58 +00001262 GrAssert(kUnusedUniform != programData->fUniLocations.fViewMatrixUni);
1263 }
1264 if (kUseUniform == programData->fUniLocations.fColorUni) {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001265 GR_GL_CALL_RET(gl, programData->fUniLocations.fColorUni,
1266 GetUniformLocation(progID, COL_UNI_NAME));
bsalomon@google.com91961302011-05-09 18:39:58 +00001267 GrAssert(kUnusedUniform != programData->fUniLocations.fColorUni);
1268 }
Scroggo97c88c22011-05-11 14:05:25 +00001269 if (kUseUniform == programData->fUniLocations.fColorFilterUni) {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001270 GR_GL_CALL_RET(gl, programData->fUniLocations.fColorFilterUni,
1271 GetUniformLocation(progID, COL_FILTER_UNI_NAME));
Scroggo97c88c22011-05-11 14:05:25 +00001272 GrAssert(kUnusedUniform != programData->fUniLocations.fColorFilterUni);
1273 }
bsalomon@google.com91961302011-05-09 18:39:58 +00001274
senorblanco@chromium.org50bdad82012-01-03 20:51:57 +00001275 if (kUseUniform == programData->fUniLocations.fColorMatrixUni) {
1276 GR_GL_CALL_RET(gl, programData->fUniLocations.fColorMatrixUni,
1277 GetUniformLocation(progID, COL_MATRIX_UNI_NAME));
1278 }
1279
1280 if (kUseUniform == programData->fUniLocations.fColorMatrixVecUni) {
1281 GR_GL_CALL_RET(gl, programData->fUniLocations.fColorMatrixVecUni,
1282 GetUniformLocation(progID, COL_MATRIX_VEC_UNI_NAME));
1283 }
1284
bsalomon@google.comf2d91552011-05-16 20:56:06 +00001285 if (kUseUniform == programData->fUniLocations.fEdgesUni) {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001286 GR_GL_CALL_RET(gl, programData->fUniLocations.fEdgesUni,
1287 GetUniformLocation(progID, EDGES_UNI_NAME));
senorblanco@chromium.org92e0f222011-05-12 15:49:15 +00001288 GrAssert(kUnusedUniform != programData->fUniLocations.fEdgesUni);
1289 } else {
1290 programData->fUniLocations.fEdgesUni = kUnusedUniform;
1291 }
1292
tomhudson@google.com93813632011-10-27 20:21:16 +00001293 for (int s = 0; s < GrDrawState::kNumStages; ++s) {
bsalomon@google.com91961302011-05-09 18:39:58 +00001294 StageUniLocations& locations = programData->fUniLocations.fStages[s];
tomhudson@google.com0d831722011-06-02 15:37:14 +00001295 if (fProgramDesc.fStages[s].isEnabled()) {
bsalomon@google.com91961302011-05-09 18:39:58 +00001296 if (kUseUniform == locations.fTextureMatrixUni) {
1297 GrStringBuilder texMName;
1298 tex_matrix_name(s, &texMName);
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001299 GR_GL_CALL_RET(gl, locations.fTextureMatrixUni,
1300 GetUniformLocation(progID, texMName.c_str()));
bsalomon@google.com91961302011-05-09 18:39:58 +00001301 GrAssert(kUnusedUniform != locations.fTextureMatrixUni);
1302 }
1303
1304 if (kUseUniform == locations.fSamplerUni) {
1305 GrStringBuilder samplerName;
1306 sampler_name(s, &samplerName);
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001307 GR_GL_CALL_RET(gl, locations.fSamplerUni,
1308 GetUniformLocation(progID,samplerName.c_str()));
bsalomon@google.com91961302011-05-09 18:39:58 +00001309 GrAssert(kUnusedUniform != locations.fSamplerUni);
1310 }
1311
1312 if (kUseUniform == locations.fNormalizedTexelSizeUni) {
1313 GrStringBuilder texelSizeName;
1314 normalized_texel_size_name(s, &texelSizeName);
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001315 GR_GL_CALL_RET(gl, locations.fNormalizedTexelSizeUni,
1316 GetUniformLocation(progID, texelSizeName.c_str()));
bsalomon@google.com91961302011-05-09 18:39:58 +00001317 GrAssert(kUnusedUniform != locations.fNormalizedTexelSizeUni);
1318 }
1319
1320 if (kUseUniform == locations.fRadial2Uni) {
1321 GrStringBuilder radial2ParamName;
1322 radial2_param_name(s, &radial2ParamName);
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001323 GR_GL_CALL_RET(gl, locations.fRadial2Uni,
1324 GetUniformLocation(progID, radial2ParamName.c_str()));
bsalomon@google.com91961302011-05-09 18:39:58 +00001325 GrAssert(kUnusedUniform != locations.fRadial2Uni);
1326 }
junov@google.com6acc9b32011-05-16 18:32:07 +00001327
1328 if (kUseUniform == locations.fTexDomUni) {
1329 GrStringBuilder texDomName;
1330 tex_domain_name(s, &texDomName);
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001331 GR_GL_CALL_RET(gl, locations.fTexDomUni,
1332 GetUniformLocation(progID, texDomName.c_str()));
junov@google.com6acc9b32011-05-16 18:32:07 +00001333 GrAssert(kUnusedUniform != locations.fTexDomUni);
1334 }
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +00001335
1336 GrStringBuilder kernelName, imageIncrementName;
1337 convolve_param_names(s, &kernelName, &imageIncrementName);
1338 if (kUseUniform == locations.fKernelUni) {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001339 GR_GL_CALL_RET(gl, locations.fKernelUni,
1340 GetUniformLocation(progID, kernelName.c_str()));
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +00001341 GrAssert(kUnusedUniform != locations.fKernelUni);
1342 }
1343
1344 if (kUseUniform == locations.fImageIncrementUni) {
bsalomon@google.com56bfc5a2011-09-01 13:28:16 +00001345 GR_GL_CALL_RET(gl, locations.fImageIncrementUni,
1346 GetUniformLocation(progID,
1347 imageIncrementName.c_str()));
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +00001348 GrAssert(kUnusedUniform != locations.fImageIncrementUni);
1349 }
bsalomon@google.com91961302011-05-09 18:39:58 +00001350 }
1351 }
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001352 GR_GL_CALL(gl, UseProgram(progID));
bsalomon@google.com91961302011-05-09 18:39:58 +00001353
1354 // init sampler unis and set bogus values for state tracking
tomhudson@google.com93813632011-10-27 20:21:16 +00001355 for (int s = 0; s < GrDrawState::kNumStages; ++s) {
bsalomon@google.com91961302011-05-09 18:39:58 +00001356 if (kUnusedUniform != programData->fUniLocations.fStages[s].fSamplerUni) {
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001357 GR_GL_CALL(gl, Uniform1i(programData->fUniLocations.fStages[s].fSamplerUni, s));
bsalomon@google.com91961302011-05-09 18:39:58 +00001358 }
1359 programData->fTextureMatrices[s] = GrMatrix::InvalidMatrix();
1360 programData->fRadial2CenterX1[s] = GR_ScalarMax;
1361 programData->fRadial2Radius0[s] = -GR_ScalarMax;
1362 programData->fTextureWidth[s] = -1;
1363 programData->fTextureHeight[s] = -1;
1364 }
1365 programData->fViewMatrix = GrMatrix::InvalidMatrix();
1366 programData->fColor = GrColor_ILLEGAL;
Scroggo97c88c22011-05-11 14:05:25 +00001367 programData->fColorFilterColor = GrColor_ILLEGAL;
bsalomon@google.com91961302011-05-09 18:39:58 +00001368}
1369
junov@google.comf93e7172011-03-31 21:26:24 +00001370//============================================================================
1371// Stage code generation
1372//============================================================================
1373
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001374namespace {
1375
1376bool isRadialMapping(GrGLProgram::StageDesc::CoordMapping mapping) {
1377 return
1378 (GrGLProgram::StageDesc::kRadial2Gradient_CoordMapping == mapping ||
1379 GrGLProgram::StageDesc::kRadial2GradientDegenerate_CoordMapping == mapping);
1380}
1381
tomhudson@google.comda668982011-12-07 15:06:29 +00001382GrGLShaderVar* genRadialVS(int stageNum,
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001383 ShaderCodeSegments* segments,
1384 GrGLProgram::StageUniLocations* locations,
1385 const char** radial2VaryingVSName,
1386 const char** radial2VaryingFSName,
1387 const char* varyingVSName,
1388 int varyingDims, int coordDims) {
1389
1390 GrGLShaderVar* radial2FSParams = &segments->fFSUnis.push_back();
1391 radial2FSParams->setType(GrGLShaderVar::kFloat_Type);
tomhudson@google.com086e5352011-12-08 14:44:10 +00001392 radial2FSParams->setTypeModifier(GrGLShaderVar::kUniform_TypeModifier);
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001393 radial2FSParams->setArrayCount(6);
1394 radial2_param_name(stageNum, radial2FSParams->accessName());
1395 segments->fVSUnis.push_back(*radial2FSParams).setEmitPrecision(true);
1396
1397 locations->fRadial2Uni = kUseUniform;
1398
1399 // for radial grads without perspective we can pass the linear
1400 // part of the quadratic as a varying.
1401 if (varyingDims == coordDims) {
1402 GrAssert(2 == coordDims);
1403 append_varying(GrGLShaderVar::kFloat_Type,
1404 "Radial2BCoeff",
1405 stageNum,
1406 segments,
1407 radial2VaryingVSName,
1408 radial2VaryingFSName);
1409
tomhudson@google.comda668982011-12-07 15:06:29 +00001410 GrStringBuilder radial2p2;
1411 GrStringBuilder radial2p3;
1412 radial2FSParams->appendArrayAccess(2, &radial2p2);
1413 radial2FSParams->appendArrayAccess(3, &radial2p3);
1414
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001415 // r2Var = 2 * (r2Parm[2] * varCoord.x - r2Param[3])
1416 const char* r2ParamName = radial2FSParams->getName().c_str();
tomhudson@google.comda668982011-12-07 15:06:29 +00001417 segments->fVSCode.appendf("\t%s = 2.0 *(%s * %s.x - %s);\n",
1418 *radial2VaryingVSName, radial2p2.c_str(),
1419 varyingVSName, radial2p3.c_str());
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001420 }
1421
tomhudson@google.comda668982011-12-07 15:06:29 +00001422 return radial2FSParams;
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001423}
1424
1425bool genRadial2GradientCoordMapping(int stageNum,
1426 ShaderCodeSegments* segments,
1427 const char* radial2VaryingFSName,
tomhudson@google.comda668982011-12-07 15:06:29 +00001428 GrGLShaderVar* radial2Params,
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001429 GrStringBuilder& sampleCoords,
1430 GrStringBuilder& fsCoordName,
1431 int varyingDims,
1432 int coordDims) {
tomhudson@google.com5960d002011-10-19 20:21:48 +00001433 GrStringBuilder cName("c");
1434 GrStringBuilder ac4Name("ac4");
1435 GrStringBuilder rootName("root");
1436
1437 cName.appendS32(stageNum);
1438 ac4Name.appendS32(stageNum);
1439 rootName.appendS32(stageNum);
1440
tomhudson@google.comda668982011-12-07 15:06:29 +00001441 GrStringBuilder radial2p0;
1442 GrStringBuilder radial2p1;
1443 GrStringBuilder radial2p2;
1444 GrStringBuilder radial2p3;
1445 GrStringBuilder radial2p4;
1446 GrStringBuilder radial2p5;
1447 radial2Params->appendArrayAccess(0, &radial2p0);
1448 radial2Params->appendArrayAccess(1, &radial2p1);
1449 radial2Params->appendArrayAccess(2, &radial2p2);
1450 radial2Params->appendArrayAccess(3, &radial2p3);
1451 radial2Params->appendArrayAccess(4, &radial2p4);
1452 radial2Params->appendArrayAccess(5, &radial2p5);
1453
tomhudson@google.com5960d002011-10-19 20:21:48 +00001454 // if we were able to interpolate the linear component bVar is the varying
1455 // otherwise compute it
1456 GrStringBuilder bVar;
1457 if (coordDims == varyingDims) {
1458 bVar = radial2VaryingFSName;
1459 GrAssert(2 == varyingDims);
1460 } else {
1461 GrAssert(3 == varyingDims);
1462 bVar = "b";
1463 bVar.appendS32(stageNum);
tomhudson@google.comda668982011-12-07 15:06:29 +00001464 segments->fFSCode.appendf("\tfloat %s = 2.0 * (%s * %s.x - %s);\n",
1465 bVar.c_str(), radial2p2.c_str(),
1466 fsCoordName.c_str(), radial2p3.c_str());
tomhudson@google.com5960d002011-10-19 20:21:48 +00001467 }
1468
1469 // c = (x^2)+(y^2) - params[4]
tomhudson@google.comda668982011-12-07 15:06:29 +00001470 segments->fFSCode.appendf("\tfloat %s = dot(%s, %s) - %s;\n",
tomhudson@google.com5960d002011-10-19 20:21:48 +00001471 cName.c_str(), fsCoordName.c_str(),
1472 fsCoordName.c_str(),
tomhudson@google.comda668982011-12-07 15:06:29 +00001473 radial2p4.c_str());
tomhudson@google.com5960d002011-10-19 20:21:48 +00001474 // ac4 = 4.0 * params[0] * c
tomhudson@google.comda668982011-12-07 15:06:29 +00001475 segments->fFSCode.appendf("\tfloat %s = %s * 4.0 * %s;\n",
1476 ac4Name.c_str(), radial2p0.c_str(),
tomhudson@google.com5960d002011-10-19 20:21:48 +00001477 cName.c_str());
1478
1479 // root = sqrt(b^2-4ac)
1480 // (abs to avoid exception due to fp precision)
1481 segments->fFSCode.appendf("\tfloat %s = sqrt(abs(%s*%s - %s));\n",
1482 rootName.c_str(), bVar.c_str(), bVar.c_str(),
1483 ac4Name.c_str());
1484
1485 // x coord is: (-b + params[5] * sqrt(b^2-4ac)) * params[1]
1486 // y coord is 0.5 (texture is effectively 1D)
tomhudson@google.comda668982011-12-07 15:06:29 +00001487 sampleCoords.printf("vec2((-%s + %s * %s) * %s, 0.5)",
1488 bVar.c_str(), radial2p5.c_str(),
1489 rootName.c_str(), radial2p1.c_str());
tomhudson@google.com5960d002011-10-19 20:21:48 +00001490 return true;
1491}
1492
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001493bool genRadial2GradientDegenerateCoordMapping(int stageNum,
1494 ShaderCodeSegments* segments,
1495 const char* radial2VaryingFSName,
tomhudson@google.comda668982011-12-07 15:06:29 +00001496 GrGLShaderVar* radial2Params,
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001497 GrStringBuilder& sampleCoords,
1498 GrStringBuilder& fsCoordName,
1499 int varyingDims,
1500 int coordDims) {
tomhudson@google.com5960d002011-10-19 20:21:48 +00001501 GrStringBuilder cName("c");
1502
1503 cName.appendS32(stageNum);
1504
tomhudson@google.comda668982011-12-07 15:06:29 +00001505 GrStringBuilder radial2p2;
1506 GrStringBuilder radial2p3;
1507 GrStringBuilder radial2p4;
1508 radial2Params->appendArrayAccess(2, &radial2p2);
1509 radial2Params->appendArrayAccess(3, &radial2p3);
1510 radial2Params->appendArrayAccess(4, &radial2p4);
1511
tomhudson@google.com5960d002011-10-19 20:21:48 +00001512 // if we were able to interpolate the linear component bVar is the varying
1513 // otherwise compute it
1514 GrStringBuilder bVar;
1515 if (coordDims == varyingDims) {
1516 bVar = radial2VaryingFSName;
1517 GrAssert(2 == varyingDims);
1518 } else {
1519 GrAssert(3 == varyingDims);
1520 bVar = "b";
1521 bVar.appendS32(stageNum);
tomhudson@google.comda668982011-12-07 15:06:29 +00001522 segments->fFSCode.appendf("\tfloat %s = 2.0 * (%s * %s.x - %s);\n",
1523 bVar.c_str(), radial2p2.c_str(),
1524 fsCoordName.c_str(), radial2p3.c_str());
tomhudson@google.com5960d002011-10-19 20:21:48 +00001525 }
1526
1527 // c = (x^2)+(y^2) - params[4]
tomhudson@google.comda668982011-12-07 15:06:29 +00001528 segments->fFSCode.appendf("\tfloat %s = dot(%s, %s) - %s;\n",
tomhudson@google.com5960d002011-10-19 20:21:48 +00001529 cName.c_str(), fsCoordName.c_str(),
1530 fsCoordName.c_str(),
tomhudson@google.comda668982011-12-07 15:06:29 +00001531 radial2p4.c_str());
tomhudson@google.com5960d002011-10-19 20:21:48 +00001532
1533 // x coord is: -c/b
1534 // y coord is 0.5 (texture is effectively 1D)
1535 sampleCoords.printf("vec2((-%s / %s), 0.5)", cName.c_str(), bVar.c_str());
1536 return true;
1537}
1538
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001539void gen2x2FS(int stageNum,
1540 ShaderCodeSegments* segments,
1541 GrGLProgram::StageUniLocations* locations,
1542 GrStringBuilder* sampleCoords,
1543 const char* samplerName,
1544 const char* texelSizeName,
bsalomon@google.com0a97be22011-11-08 19:20:57 +00001545 const char* swizzle,
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001546 const char* fsOutColor,
1547 GrStringBuilder& texFunc,
1548 GrStringBuilder& modulate,
1549 bool complexCoord,
1550 int coordDims) {
1551 locations->fNormalizedTexelSizeUni = kUseUniform;
1552 if (complexCoord) {
1553 // assign the coord to a var rather than compute 4x.
1554 GrStringBuilder coordVar("tCoord");
1555 coordVar.appendS32(stageNum);
1556 segments->fFSCode.appendf("\t%s %s = %s;\n",
1557 float_vector_type_str(coordDims),
1558 coordVar.c_str(), sampleCoords->c_str());
1559 *sampleCoords = coordVar;
1560 }
1561 GrAssert(2 == coordDims);
1562 GrStringBuilder accumVar("accum");
1563 accumVar.appendS32(stageNum);
bsalomon@google.com0a97be22011-11-08 19:20:57 +00001564 segments->fFSCode.appendf("\tvec4 %s = %s(%s, %s + vec2(-%s.x,-%s.y))%s;\n", accumVar.c_str(), texFunc.c_str(), samplerName, sampleCoords->c_str(), texelSizeName, texelSizeName, swizzle);
1565 segments->fFSCode.appendf("\t%s += %s(%s, %s + vec2(+%s.x,-%s.y))%s;\n", accumVar.c_str(), texFunc.c_str(), samplerName, sampleCoords->c_str(), texelSizeName, texelSizeName, swizzle);
1566 segments->fFSCode.appendf("\t%s += %s(%s, %s + vec2(-%s.x,+%s.y))%s;\n", accumVar.c_str(), texFunc.c_str(), samplerName, sampleCoords->c_str(), texelSizeName, texelSizeName, swizzle);
1567 segments->fFSCode.appendf("\t%s += %s(%s, %s + vec2(+%s.x,+%s.y))%s;\n", accumVar.c_str(), texFunc.c_str(), samplerName, sampleCoords->c_str(), texelSizeName, texelSizeName, swizzle);
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001568 segments->fFSCode.appendf("\t%s = .25 * %s%s;\n", fsOutColor, accumVar.c_str(), modulate.c_str());
1569
1570}
1571
1572void genConvolutionVS(int stageNum,
bsalomon@google.com85b505b2011-11-07 14:56:51 +00001573 const StageDesc& desc,
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001574 ShaderCodeSegments* segments,
1575 GrGLProgram::StageUniLocations* locations,
tomhudson@google.comda668982011-12-07 15:06:29 +00001576 GrGLShaderVar** kernel,
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001577 const char** imageIncrementName,
1578 const char* varyingVSName) {
tomhudson@google.comda668982011-12-07 15:06:29 +00001579 //GrGLShaderVar* kernel = &segments->fFSUnis.push_back();
1580 *kernel = &segments->fFSUnis.push_back();
1581 (*kernel)->setType(GrGLShaderVar::kFloat_Type);
tomhudson@google.com086e5352011-12-08 14:44:10 +00001582 (*kernel)->setTypeModifier(GrGLShaderVar::kUniform_TypeModifier);
tomhudson@google.comda668982011-12-07 15:06:29 +00001583 (*kernel)->setArrayCount(desc.fKernelWidth);
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001584 GrGLShaderVar* imgInc = &segments->fFSUnis.push_back();
1585 imgInc->setType(GrGLShaderVar::kVec2f_Type);
tomhudson@google.com086e5352011-12-08 14:44:10 +00001586 imgInc->setTypeModifier(GrGLShaderVar::kUniform_TypeModifier);
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001587
1588 convolve_param_names(stageNum,
tomhudson@google.comda668982011-12-07 15:06:29 +00001589 (*kernel)->accessName(),
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001590 imgInc->accessName());
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001591 *imageIncrementName = imgInc->getName().c_str();
1592
1593 // need image increment in both VS and FS
1594 segments->fVSUnis.push_back(*imgInc).setEmitPrecision(true);
1595
1596 locations->fKernelUni = kUseUniform;
1597 locations->fImageIncrementUni = kUseUniform;
1598 float scale = (desc.fKernelWidth - 1) * 0.5f;
1599 segments->fVSCode.appendf("\t%s -= vec2(%g, %g) * %s;\n",
1600 varyingVSName, scale, scale,
1601 *imageIncrementName);
1602}
1603
1604void genConvolutionFS(int stageNum,
bsalomon@google.com85b505b2011-11-07 14:56:51 +00001605 const StageDesc& desc,
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001606 ShaderCodeSegments* segments,
1607 const char* samplerName,
tomhudson@google.comda668982011-12-07 15:06:29 +00001608 GrGLShaderVar* kernel,
bsalomon@google.com0a97be22011-11-08 19:20:57 +00001609 const char* swizzle,
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001610 const char* imageIncrementName,
1611 const char* fsOutColor,
1612 GrStringBuilder& sampleCoords,
1613 GrStringBuilder& texFunc,
1614 GrStringBuilder& modulate) {
1615 GrStringBuilder sumVar("sum");
1616 sumVar.appendS32(stageNum);
1617 GrStringBuilder coordVar("coord");
1618 coordVar.appendS32(stageNum);
1619
tomhudson@google.comda668982011-12-07 15:06:29 +00001620 GrStringBuilder kernelIndex;
1621 kernel->appendArrayAccess("i", &kernelIndex);
1622
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001623 segments->fFSCode.appendf("\tvec4 %s = vec4(0, 0, 0, 0);\n",
1624 sumVar.c_str());
1625 segments->fFSCode.appendf("\tvec2 %s = %s;\n",
1626 coordVar.c_str(),
1627 sampleCoords.c_str());
1628 segments->fFSCode.appendf("\tfor (int i = 0; i < %d; i++) {\n",
1629 desc.fKernelWidth);
tomhudson@google.comda668982011-12-07 15:06:29 +00001630 segments->fFSCode.appendf("\t\t%s += %s(%s, %s)%s * %s;\n",
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001631 sumVar.c_str(), texFunc.c_str(),
bsalomon@google.com0a97be22011-11-08 19:20:57 +00001632 samplerName, coordVar.c_str(), swizzle,
tomhudson@google.comda668982011-12-07 15:06:29 +00001633 kernelIndex.c_str());
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001634 segments->fFSCode.appendf("\t\t%s += %s;\n",
1635 coordVar.c_str(),
1636 imageIncrementName);
1637 segments->fFSCode.appendf("\t}\n");
1638 segments->fFSCode.appendf("\t%s = %s%s;\n", fsOutColor,
1639 sumVar.c_str(), modulate.c_str());
1640}
1641
1642}
1643
bsalomon@google.com0b77d682011-08-19 13:28:54 +00001644void GrGLProgram::genStageCode(const GrGLInterface* gl,
1645 int stageNum,
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001646 const GrGLProgram::StageDesc& desc,
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001647 const char* fsInColor, // NULL means no incoming color
1648 const char* fsOutColor,
1649 const char* vsInCoord,
1650 ShaderCodeSegments* segments,
1651 StageUniLocations* locations) const {
junov@google.comf93e7172011-03-31 21:26:24 +00001652
bsalomon@google.com74b98712011-11-11 19:46:16 +00001653 GrAssert(stageNum >= 0 && stageNum <= GrDrawState::kNumStages);
1654 GrAssert((desc.fInConfigFlags & StageDesc::kInConfigBitMask) ==
1655 desc.fInConfigFlags);
junov@google.comf93e7172011-03-31 21:26:24 +00001656
junov@google.comf93e7172011-03-31 21:26:24 +00001657 // First decide how many coords are needed to access the texture
1658 // Right now it's always 2 but we could start using 1D textures for
1659 // gradients.
1660 static const int coordDims = 2;
1661 int varyingDims;
1662 /// Vertex Shader Stuff
1663
1664 // decide whether we need a matrix to transform texture coords
1665 // and whether the varying needs a perspective coord.
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001666 const char* matName = NULL;
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001667 if (desc.fOptFlags & StageDesc::kIdentityMatrix_OptFlagBit) {
junov@google.comf93e7172011-03-31 21:26:24 +00001668 varyingDims = coordDims;
1669 } else {
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001670 GrGLShaderVar* mat;
bsalomon@google.com4be283f2011-04-19 21:15:09 +00001671 #if GR_GL_ATTRIBUTE_MATRICES
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001672 mat = &segments->fVSAttrs.push_back();
tomhudson@google.com086e5352011-12-08 14:44:10 +00001673 mat->setTypeModifier(GrGLShaderVar::kAttribute_TypeModifier);
bsalomon@google.com91961302011-05-09 18:39:58 +00001674 locations->fTextureMatrixUni = kSetAsAttribute;
junov@google.comf93e7172011-03-31 21:26:24 +00001675 #else
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001676 mat = &segments->fVSUnis.push_back();
tomhudson@google.com086e5352011-12-08 14:44:10 +00001677 mat->setTypeModifier(GrGLShaderVar::kUniform_TypeModifier);
bsalomon@google.com91961302011-05-09 18:39:58 +00001678 locations->fTextureMatrixUni = kUseUniform;
junov@google.comf93e7172011-03-31 21:26:24 +00001679 #endif
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001680 tex_matrix_name(stageNum, mat->accessName());
1681 mat->setType(GrGLShaderVar::kMat33f_Type);
1682 matName = mat->getName().c_str();
1683
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001684 if (desc.fOptFlags & StageDesc::kNoPerspective_OptFlagBit) {
junov@google.comf93e7172011-03-31 21:26:24 +00001685 varyingDims = coordDims;
1686 } else {
1687 varyingDims = coordDims + 1;
1688 }
1689 }
1690
tomhudson@google.com086e5352011-12-08 14:44:10 +00001691 segments->fFSUnis.push_back().set(GrGLShaderVar::kSampler2D_Type,
1692 GrGLShaderVar::kUniform_TypeModifier, "");
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001693 sampler_name(stageNum, segments->fFSUnis.back().accessName());
bsalomon@google.com91961302011-05-09 18:39:58 +00001694 locations->fSamplerUni = kUseUniform;
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001695 const char* samplerName = segments->fFSUnis.back().getName().c_str();
junov@google.comf93e7172011-03-31 21:26:24 +00001696
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001697 const char* texelSizeName = NULL;
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001698 if (StageDesc::k2x2_FetchMode == desc.fFetchMode) {
tomhudson@google.com086e5352011-12-08 14:44:10 +00001699 segments->fFSUnis.push_back().set(GrGLShaderVar::kVec2f_Type,
1700 GrGLShaderVar::kUniform_TypeModifier, "");
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001701 normalized_texel_size_name(stageNum, segments->fFSUnis.back().accessName());
1702 texelSizeName = segments->fFSUnis.back().getName().c_str();
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001703 }
1704
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001705 const char *varyingVSName, *varyingFSName;
1706 append_varying(float_vector_type(varyingDims),
1707 "Stage",
1708 stageNum,
1709 segments,
1710 &varyingVSName,
1711 &varyingFSName);
junov@google.comf93e7172011-03-31 21:26:24 +00001712
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001713 if (!matName) {
junov@google.comf93e7172011-03-31 21:26:24 +00001714 GrAssert(varyingDims == coordDims);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001715 segments->fVSCode.appendf("\t%s = %s;\n", varyingVSName, vsInCoord);
junov@google.comf93e7172011-03-31 21:26:24 +00001716 } else {
bsalomon@google.com91961302011-05-09 18:39:58 +00001717 // varying = texMatrix * texCoord
1718 segments->fVSCode.appendf("\t%s = (%s * vec3(%s, 1))%s;\n",
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001719 varyingVSName, matName, vsInCoord,
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001720 vector_all_coords(varyingDims));
junov@google.comf93e7172011-03-31 21:26:24 +00001721 }
1722
tomhudson@google.comda668982011-12-07 15:06:29 +00001723 GrGLShaderVar* radial2Params = NULL;
1724 const char* radial2VaryingVSName = NULL;
1725 const char* radial2VaryingFSName = NULL;
junov@google.comf93e7172011-03-31 21:26:24 +00001726
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001727 if (isRadialMapping((StageDesc::CoordMapping) desc.fCoordMapping)) {
tomhudson@google.comda668982011-12-07 15:06:29 +00001728 radial2Params = genRadialVS(stageNum, segments,
1729 locations,
1730 &radial2VaryingVSName,
1731 &radial2VaryingFSName,
1732 varyingVSName,
1733 varyingDims, coordDims);
junov@google.comf93e7172011-03-31 21:26:24 +00001734 }
1735
tomhudson@google.comda668982011-12-07 15:06:29 +00001736 GrGLShaderVar* kernel = NULL;
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001737 const char* imageIncrementName = NULL;
bsalomon@google.com85b505b2011-11-07 14:56:51 +00001738 if (StageDesc::kConvolution_FetchMode == desc.fFetchMode) {
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001739 genConvolutionVS(stageNum, desc, segments, locations,
tomhudson@google.comda668982011-12-07 15:06:29 +00001740 &kernel, &imageIncrementName, varyingVSName);
bsalomon@google.coma8e686e2011-08-16 15:45:58 +00001741 }
senorblanco@chromium.org027de5f2011-07-08 18:03:33 +00001742
junov@google.comf93e7172011-03-31 21:26:24 +00001743 /// Fragment Shader Stuff
bsalomon@google.com91961302011-05-09 18:39:58 +00001744 GrStringBuilder fsCoordName;
junov@google.comf93e7172011-03-31 21:26:24 +00001745 // function used to access the shader, may be made projective
bsalomon@google.com91961302011-05-09 18:39:58 +00001746 GrStringBuilder texFunc("texture2D");
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001747 if (desc.fOptFlags & (StageDesc::kIdentityMatrix_OptFlagBit |
1748 StageDesc::kNoPerspective_OptFlagBit)) {
junov@google.comf93e7172011-03-31 21:26:24 +00001749 GrAssert(varyingDims == coordDims);
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001750 fsCoordName = varyingFSName;
junov@google.comf93e7172011-03-31 21:26:24 +00001751 } else {
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001752 // if we have to do some special op on the varyings to get
junov@google.comf93e7172011-03-31 21:26:24 +00001753 // our final tex coords then when in perspective we have to
bsalomon@google.com91961302011-05-09 18:39:58 +00001754 // do an explicit divide. Otherwise, we can use a Proj func.
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001755 if (StageDesc::kIdentity_CoordMapping == desc.fCoordMapping &&
1756 StageDesc::kSingle_FetchMode == desc.fFetchMode) {
bsalomon@google.com91961302011-05-09 18:39:58 +00001757 texFunc.append("Proj");
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001758 fsCoordName = varyingFSName;
junov@google.comf93e7172011-03-31 21:26:24 +00001759 } else {
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001760 fsCoordName = "inCoord";
bsalomon@google.comfc296292011-05-06 13:53:47 +00001761 fsCoordName.appendS32(stageNum);
bsalomon@google.com91961302011-05-09 18:39:58 +00001762 segments->fFSCode.appendf("\t%s %s = %s%s / %s%s;\n",
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001763 GrGLShaderVar::TypeString(float_vector_type(coordDims)),
1764 fsCoordName.c_str(),
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001765 varyingFSName,
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001766 vector_nonhomog_coords(varyingDims),
bsalomon@google.comedfe1aa2011-09-29 14:40:26 +00001767 varyingFSName,
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001768 vector_homog_coord(varyingDims));
junov@google.comf93e7172011-03-31 21:26:24 +00001769 }
1770 }
1771
bsalomon@google.comfc296292011-05-06 13:53:47 +00001772 GrStringBuilder sampleCoords;
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001773 bool complexCoord = false;
junov@google.comf93e7172011-03-31 21:26:24 +00001774 switch (desc.fCoordMapping) {
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001775 case StageDesc::kIdentity_CoordMapping:
junov@google.comf93e7172011-03-31 21:26:24 +00001776 sampleCoords = fsCoordName;
1777 break;
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001778 case StageDesc::kSweepGradient_CoordMapping:
bsalomon@google.com91961302011-05-09 18:39:58 +00001779 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 +00001780 complexCoord = true;
junov@google.comf93e7172011-03-31 21:26:24 +00001781 break;
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001782 case StageDesc::kRadialGradient_CoordMapping:
bsalomon@google.com91961302011-05-09 18:39:58 +00001783 sampleCoords.printf("vec2(length(%s.xy), 0.5)", fsCoordName.c_str());
bsalomon@google.com6aef1fb2011-05-05 12:33:22 +00001784 complexCoord = true;
junov@google.comf93e7172011-03-31 21:26:24 +00001785 break;
tomhudson@google.com5960d002011-10-19 20:21:48 +00001786 case StageDesc::kRadial2Gradient_CoordMapping:
1787 complexCoord = genRadial2GradientCoordMapping(
1788 stageNum, segments,
tomhudson@google.comda668982011-12-07 15:06:29 +00001789 radial2VaryingFSName, radial2Params,
tomhudson@google.com5960d002011-10-19 20:21:48 +00001790 sampleCoords, fsCoordName,
1791 varyingDims, coordDims);
junov@google.comf93e7172011-03-31 21:26:24 +00001792
tomhudson@google.com5960d002011-10-19 20:21:48 +00001793 break;
1794 case StageDesc::kRadial2GradientDegenerate_CoordMapping:
1795 complexCoord = genRadial2GradientDegenerateCoordMapping(
1796 stageNum, segments,
tomhudson@google.comda668982011-12-07 15:06:29 +00001797 radial2VaryingFSName, radial2Params,
tomhudson@google.com5960d002011-10-19 20:21:48 +00001798 sampleCoords, fsCoordName,
1799 varyingDims, coordDims);
1800 break;
junov@google.comf93e7172011-03-31 21:26:24 +00001801
junov@google.comf93e7172011-03-31 21:26:24 +00001802 };
1803
tomhudson@google.comf74ad8c2011-11-09 22:15:08 +00001804 const char* swizzle = "";
bsalomon@google.com74b98712011-11-11 19:46:16 +00001805 if (desc.fInConfigFlags & StageDesc::kSwapRAndB_InConfigFlag) {
1806 GrAssert(!(desc.fInConfigFlags & StageDesc::kSmearAlpha_InConfigFlag));
1807 swizzle = ".bgra";
1808 } else if (desc.fInConfigFlags & StageDesc::kSmearAlpha_InConfigFlag) {
1809 GrAssert(!(desc.fInConfigFlags &
1810 StageDesc::kMulRGBByAlpha_InConfigFlag));
1811 swizzle = ".aaaa";
1812 }
bsalomon@google.com0a97be22011-11-08 19:20:57 +00001813
bsalomon@google.com91961302011-05-09 18:39:58 +00001814 GrStringBuilder modulate;
1815 if (NULL != fsInColor) {
1816 modulate.printf(" * %s", fsInColor);
1817 }
1818
tomhudson@google.com0d831722011-06-02 15:37:14 +00001819 if (desc.fOptFlags &
bsalomon@google.com22c5dea2011-07-07 14:38:03 +00001820 StageDesc::kCustomTextureDomain_OptFlagBit) {
junov@google.com6acc9b32011-05-16 18:32:07 +00001821 GrStringBuilder texDomainName;
1822 tex_domain_name(stageNum, &texDomainName);
tomhudson@google.com086e5352011-12-08 14:44:10 +00001823 segments->fFSUnis.push_back().set(GrGLShaderVar::kVec4f_Type,
1824 GrGLShaderVar::kUniform_TypeModifier, texDomainName);
junov@google.com6acc9b32011-05-16 18:32:07 +00001825 GrStringBuilder coordVar("clampCoord");
1826 segments->fFSCode.appendf("\t%s %s = clamp(%s, %s.xy, %s.zw);\n",
bsalomon@google.com4fa66942011-09-20 19:06:12 +00001827 float_vector_type_str(coordDims),
junov@google.com6acc9b32011-05-16 18:32:07 +00001828 coordVar.c_str(),
1829 sampleCoords.c_str(),
1830 texDomainName.c_str(),
1831 texDomainName.c_str());
1832 sampleCoords = coordVar;
1833 locations->fTexDomUni = kUseUniform;
1834 }
1835
tomhudson@google.com5960d002011-10-19 20:21:48 +00001836 switch (desc.fFetchMode) {
1837 case StageDesc::k2x2_FetchMode:
bsalomon@google.com74b98712011-11-11 19:46:16 +00001838 GrAssert(!(desc.fInConfigFlags &
1839 StageDesc::kMulRGBByAlpha_InConfigFlag));
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001840 gen2x2FS(stageNum, segments, locations, &sampleCoords,
bsalomon@google.com0a97be22011-11-08 19:20:57 +00001841 samplerName, texelSizeName, swizzle, fsOutColor,
tomhudson@google.com5960d002011-10-19 20:21:48 +00001842 texFunc, modulate, complexCoord, coordDims);
1843 break;
bsalomon@google.com85b505b2011-11-07 14:56:51 +00001844 case StageDesc::kConvolution_FetchMode:
bsalomon@google.com74b98712011-11-11 19:46:16 +00001845 GrAssert(!(desc.fInConfigFlags &
1846 StageDesc::kMulRGBByAlpha_InConfigFlag));
tomhudson@google.com2a2e3ef2011-10-25 19:51:09 +00001847 genConvolutionFS(stageNum, desc, segments,
tomhudson@google.comda668982011-12-07 15:06:29 +00001848 samplerName, kernel, swizzle, imageIncrementName, fsOutColor,
tomhudson@google.com5960d002011-10-19 20:21:48 +00001849 sampleCoords, texFunc, modulate);
1850 break;
1851 default:
bsalomon@google.com74b98712011-11-11 19:46:16 +00001852 if (desc.fInConfigFlags & StageDesc::kMulRGBByAlpha_InConfigFlag) {
1853 GrAssert(!(desc.fInConfigFlags &
1854 StageDesc::kSmearAlpha_InConfigFlag));
1855 segments->fFSCode.appendf("\t%s = %s(%s, %s)%s;\n",
1856 fsOutColor, texFunc.c_str(),
1857 samplerName, sampleCoords.c_str(),
1858 swizzle);
1859 segments->fFSCode.appendf("\t%s = vec4(%s.rgb*%s.a,%s.a)%s;\n",
1860 fsOutColor, fsOutColor, fsOutColor,
1861 fsOutColor, modulate.c_str());
1862 } else {
1863 segments->fFSCode.appendf("\t%s = %s(%s, %s)%s%s;\n",
1864 fsOutColor, texFunc.c_str(),
1865 samplerName, sampleCoords.c_str(),
1866 swizzle, modulate.c_str());
1867 }
junov@google.comf93e7172011-03-31 21:26:24 +00001868 }
junov@google.comf93e7172011-03-31 21:26:24 +00001869}
tomhudson@google.com5960d002011-10-19 20:21:48 +00001870
tomhudson@google.com5960d002011-10-19 20:21:48 +00001871