blob: 9875f0a5544d75dc5e46e5cb591e2bd1c67f07d5 [file] [log] [blame]
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001//
2// Copyright (c) 2002-2010 The ANGLE Project Authors. All rights reserved.
3// Use of this source code is governed by a BSD-style license that can be
4// found in the LICENSE file.
5//
6
7#include "OutputHLSL.h"
8
alokp@chromium.orgea0e1af2010-03-22 19:33:14 +00009#include "common/debug.h"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000010#include "InfoSink.h"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000011
12namespace sh
13{
14OutputHLSL::OutputHLSL(TParseContext &context) : TIntermTraverser(true, true, true), context(context)
15{
16}
17
18void OutputHLSL::header()
19{
20 EShLanguage language = context.language;
21 TInfoSinkBase &out = context.infoSink.obj;
22
23 if (language == EShLangFragment)
24 {
25 TString uniforms;
26 TString varyingInput;
27 TString varyingGlobals;
28
29 TSymbolTableLevel *symbols = context.symbolTable.getGlobalLevel();
30 int semanticIndex = 0;
31
32 for (TSymbolTableLevel::const_iterator namedSymbol = symbols->begin(); namedSymbol != symbols->end(); namedSymbol++)
33 {
34 const TSymbol *symbol = (*namedSymbol).second;
35 const TString &name = symbol->getName();
36
37 if (symbol->isVariable())
38 {
39 const TVariable *variable = static_cast<const TVariable*>(symbol);
40 const TType &type = variable->getType();
41 TQualifier qualifier = type.getQualifier();
42
43 if (qualifier == EvqUniform)
44 {
45 uniforms += "uniform " + typeString(type) + " " + name + arrayString(type) + ";\n";
46 }
47 else if (qualifier == EvqVaryingIn || qualifier == EvqInvariantVaryingIn)
48 {
49 char semantic[100];
50 sprintf(semantic, " : TEXCOORD%d", semanticIndex);
51 semanticIndex += type.isArray() ? type.getArraySize() : 1;
52
daniel@transgaming.com0e3358a2010-04-05 20:32:42 +000053 // Program linking depends on this exact format
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000054 varyingInput += " " + typeString(type) + " " + name + arrayString(type) + semantic + ";\n";
55 varyingGlobals += "static " + typeString(type) + " " + name + arrayString(type) + " = " + initializer(type) + ";\n";
56 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +000057 else if (qualifier == EvqGlobal || qualifier == EvqTemporary)
daniel@transgaming.comd25ab252010-03-30 03:36:26 +000058 {
59 // Globals are declared and intialized as an aggregate node
60 }
daniel@transgaming.com49bce7e2010-03-17 03:58:51 +000061 else if (qualifier == EvqConst)
62 {
63 // Constants are repeated as literals where used
64 }
65 else UNREACHABLE();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000066 }
67 }
68
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +000069 out << "uniform float4 gl_Window;\n"
70 "uniform float2 gl_Depth;\n"
daniel@transgaming.com79b820b2010-03-16 05:48:57 +000071 "uniform bool __frontCCW;\n"
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +000072 "\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000073 out << uniforms;
daniel@transgaming.com86487c22010-03-11 19:41:43 +000074 out << "\n"
75 "struct PS_INPUT\n" // FIXME: Prevent name clashes
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000076 "{\n";
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +000077 out << varyingInput;
daniel@transgaming.com0e3358a2010-04-05 20:32:42 +000078 out << " float4 gl_FragCoord : TEXCOORD" << semanticIndex << ";\n";
daniel@transgaming.com79b820b2010-03-16 05:48:57 +000079 out << " float __vFace : VFACE;\n"
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +000080 "};\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000081 "\n";
82 out << varyingGlobals;
83 out << "\n"
84 "struct PS_OUTPUT\n" // FIXME: Prevent name clashes
85 "{\n"
86 " float4 gl_Color[1] : COLOR;\n"
87 "};\n"
88 "\n"
89 "static float4 gl_Color[1] = {float4(0, 0, 0, 0)};\n"
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +000090 "static float4 gl_FragCoord = float4(0, 0, 0, 0);\n"
daniel@transgaming.comccad59f2010-03-26 04:08:39 +000091 "static float2 gl_PointCoord = float2(0.5, 0.5);\n"
daniel@transgaming.com79b820b2010-03-16 05:48:57 +000092 "static bool gl_FrontFacing = false;\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000093 "\n"
94 "float4 gl_texture2D(sampler2D s, float2 t)\n"
95 "{\n"
96 " return tex2D(s, t);\n"
97 "}\n"
98 "\n"
99 "float4 gl_texture2DProj(sampler2D s, float3 t)\n"
100 "{\n"
101 " return tex2Dproj(s, float4(t.x, t.y, 0, t.z));\n"
102 "}\n"
103 "float4 gl_texture2DBias(sampler2D s, float2 t, float bias)\n"
104 "{\n"
105 " return tex2Dbias(s, float4(t.x, t.y, 0, bias));\n"
106 "}\n"
107 "\n"
108 "float4 gl_textureCube(samplerCUBE s, float3 t)\n"
109 "{\n"
110 " return texCUBE(s, t);\n"
111 "}\n"
112 "\n";
113 }
daniel@transgaming.comd25ab252010-03-30 03:36:26 +0000114 else // Vertex shader
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000115 {
116 TString uniforms;
117 TString attributeInput;
118 TString attributeGlobals;
119 TString varyingOutput;
120 TString varyingGlobals;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000121
122 TSymbolTableLevel *symbols = context.symbolTable.getGlobalLevel();
123 int semanticIndex = 0;
124
125 for (TSymbolTableLevel::const_iterator namedSymbol = symbols->begin(); namedSymbol != symbols->end(); namedSymbol++)
126 {
127 const TSymbol *symbol = (*namedSymbol).second;
128 const TString &name = symbol->getName();
129
130 if (symbol->isVariable())
131 {
132 const TVariable *variable = static_cast<const TVariable*>(symbol);
133 const TType &type = variable->getType();
134 TQualifier qualifier = type.getQualifier();
135
136 if (qualifier == EvqUniform)
137 {
138 uniforms += "uniform " + typeString(type) + " " + name + arrayString(type) + ";\n";
139 }
140 else if (qualifier == EvqAttribute)
141 {
142 char semantic[100];
143 sprintf(semantic, " : TEXCOORD%d", semanticIndex);
144 semanticIndex += type.isArray() ? type.getArraySize() : 1;
145
146 attributeInput += " " + typeString(type) + " " + name + arrayString(type) + semantic + ";\n";
147 attributeGlobals += "static " + typeString(type) + " " + name + arrayString(type) + " = " + initializer(type) + ";\n";
148 }
149 else if (qualifier == EvqVaryingOut || qualifier == EvqInvariantVaryingOut)
150 {
daniel@transgaming.com0e3358a2010-04-05 20:32:42 +0000151 // Program linking depends on this exact format
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000152 varyingOutput += " " + typeString(type) + " " + name + arrayString(type) + " : TEXCOORD0;\n"; // Actual semantic index assigned during link
153 varyingGlobals += "static " + typeString(type) + " " + name + arrayString(type) + " = " + initializer(type) + ";\n";
154 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000155 else if (qualifier == EvqGlobal || qualifier == EvqTemporary)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000156 {
daniel@transgaming.comd25ab252010-03-30 03:36:26 +0000157 // Globals are declared and intialized as an aggregate node
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000158 }
daniel@transgaming.com49bce7e2010-03-17 03:58:51 +0000159 else if (qualifier == EvqConst)
160 {
161 // Constants are repeated as literals where used
162 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000163 else UNREACHABLE();
164 }
165 }
166
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +0000167 out << "uniform float2 gl_HalfPixelSize;\n"
168 "\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000169 out << uniforms;
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +0000170 out << "\n"
171 "struct VS_INPUT\n" // FIXME: Prevent name clashes
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000172 "{\n";
173 out << attributeInput;
174 out << "};\n"
175 "\n";
176 out << attributeGlobals;
177 out << "\n"
178 "struct VS_OUTPUT\n" // FIXME: Prevent name clashes
179 "{\n"
180 " float4 gl_Position : POSITION;\n"
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +0000181 " float gl_PointSize : PSIZE;\n"
daniel@transgaming.com0e3358a2010-04-05 20:32:42 +0000182 " float4 gl_FragCoord : TEXCOORD0;\n"; // Actual semantic index assigned during link
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +0000183 out << varyingOutput;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000184 out << "};\n"
185 "\n"
186 "static float4 gl_Position = float4(0, 0, 0, 0);\n"
daniel@transgaming.comccad59f2010-03-26 04:08:39 +0000187 "static float gl_PointSize = float(1);\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000188 out << varyingGlobals;
189 out << "\n";
190 }
191
daniel@transgaming.com86487c22010-03-11 19:41:43 +0000192 out << "struct gl_DepthRangeParameters\n"
193 "{\n"
194 " float near;\n"
195 " float far;\n"
196 " float diff;\n"
197 "};\n"
198 "\n"
199 "uniform gl_DepthRangeParameters gl_DepthRange;\n"
200 "\n"
201 "float vec1(float x)\n" // FIXME: Prevent name clashes
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000202 "{\n"
203 " return x;\n"
204 "}\n"
205 "\n"
206 "float vec1(float2 xy)\n" // FIXME: Prevent name clashes
207 "{\n"
208 " return xy[0];\n"
209 "}\n"
210 "\n"
211 "float vec1(float3 xyz)\n" // FIXME: Prevent name clashes
212 "{\n"
213 " return xyz[0];\n"
214 "}\n"
215 "\n"
216 "float vec1(float4 xyzw)\n" // FIXME: Prevent name clashes
217 "{\n"
218 " return xyzw[0];\n"
219 "}\n"
220 "\n"
221 "float2 vec2(float x)\n" // FIXME: Prevent name clashes
222 "{\n"
223 " return float2(x, x);\n"
224 "}\n"
225 "\n"
226 "float2 vec2(float x, float y)\n" // FIXME: Prevent name clashes
227 "{\n"
228 " return float2(x, y);\n"
229 "}\n"
230 "\n"
231 "float2 vec2(float2 xy)\n" // FIXME: Prevent name clashes
232 "{\n"
233 " return xy;\n"
234 "}\n"
235 "\n"
236 "float2 vec2(float3 xyz)\n" // FIXME: Prevent name clashes
237 "{\n"
238 " return float2(xyz[0], xyz[1]);\n"
239 "}\n"
240 "\n"
241 "float2 vec2(float4 xyzw)\n" // FIXME: Prevent name clashes
242 "{\n"
243 " return float2(xyzw[0], xyzw[1]);\n"
244 "}\n"
245 "\n"
246 "float3 vec3(float x)\n" // FIXME: Prevent name clashes
247 "{\n"
248 " return float3(x, x, x);\n"
249 "}\n"
250 "\n"
251 "float3 vec3(float x, float y, float z)\n" // FIXME: Prevent name clashes
252 "{\n"
253 " return float3(x, y, z);\n"
254 "}\n"
255 "\n"
256 "float3 vec3(float2 xy, float z)\n" // FIXME: Prevent name clashes
257 "{\n"
258 " return float3(xy[0], xy[1], z);\n"
259 "}\n"
260 "\n"
261 "float3 vec3(float x, float2 yz)\n" // FIXME: Prevent name clashes
262 "{\n"
263 " return float3(x, yz[0], yz[1]);\n"
264 "}\n"
265 "\n"
266 "float3 vec3(float3 xyz)\n" // FIXME: Prevent name clashes
267 "{\n"
268 " return xyz;\n"
269 "}\n"
270 "\n"
271 "float3 vec3(float4 xyzw)\n" // FIXME: Prevent name clashes
272 "{\n"
273 " return float3(xyzw[0], xyzw[1], xyzw[2]);\n"
274 "}\n"
275 "\n"
276 "float4 vec4(float x)\n" // FIXME: Prevent name clashes
277 "{\n"
278 " return float4(x, x, x, x);\n"
279 "}\n"
280 "\n"
281 "float4 vec4(float x, float y, float z, float w)\n" // FIXME: Prevent name clashes
282 "{\n"
283 " return float4(x, y, z, w);\n"
284 "}\n"
285 "\n"
286 "float4 vec4(float2 xy, float z, float w)\n" // FIXME: Prevent name clashes
287 "{\n"
288 " return float4(xy[0], xy[1], z, w);\n"
289 "}\n"
290 "\n"
291 "float4 vec4(float x, float2 yz, float w)\n" // FIXME: Prevent name clashes
292 "{\n"
293 " return float4(x, yz[0], yz[1], w);\n"
294 "}\n"
295 "\n"
296 "float4 vec4(float x, float y, float2 zw)\n" // FIXME: Prevent name clashes
297 "{\n"
298 " return float4(x, y, zw[0], zw[1]);\n"
299 "}\n"
300 "\n"
301 "float4 vec4(float2 xy, float2 zw)\n" // FIXME: Prevent name clashes
302 "{\n"
303 " return float4(xy[0], xy[1], zw[0], zw[1]);\n"
304 "}\n"
305 "\n"
306 "float4 vec4(float3 xyz, float w)\n" // FIXME: Prevent name clashes
307 "{\n"
308 " return float4(xyz[0], xyz[1], xyz[2], w);\n"
309 "}\n"
310 "\n"
311 "float4 vec4(float x, float3 yzw)\n" // FIXME: Prevent name clashes
312 "{\n"
313 " return float4(x, yzw[0], yzw[1], yzw[2]);\n"
314 "}\n"
315 "\n"
316 "float4 vec4(float4 xyzw)\n" // FIXME: Prevent name clashes
317 "{\n"
318 " return xyzw;\n"
319 "}\n"
320 "\n"
321 "bool xor(bool p, bool q)\n" // FIXME: Prevent name clashes
322 "{\n"
323 " return (p || q) && !(p && q);\n"
324 "}\n"
325 "\n"
326 "float mod(float x, float y)\n" // FIXME: Prevent name clashes
327 "{\n"
328 " return x - y * floor(x / y);\n"
329 "}\n"
daniel@transgaming.com680553b2010-03-08 21:30:52 +0000330 "\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000331 "float2 mod(float2 x, float y)\n" // FIXME: Prevent name clashes
332 "{\n"
333 " return x - y * floor(x / y);\n"
334 "}\n"
daniel@transgaming.com680553b2010-03-08 21:30:52 +0000335 "\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000336 "float3 mod(float3 x, float y)\n" // FIXME: Prevent name clashes
337 "{\n"
338 " return x - y * floor(x / y);\n"
339 "}\n"
daniel@transgaming.com680553b2010-03-08 21:30:52 +0000340 "\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000341 "float4 mod(float4 x, float y)\n" // FIXME: Prevent name clashes
342 "{\n"
343 " return x - y * floor(x / y);\n"
344 "}\n"
daniel@transgaming.com680553b2010-03-08 21:30:52 +0000345 "\n"
346 "float faceforward(float N, float I, float Nref)\n" // FIXME: Prevent name clashes
347 "{\n"
348 " if(dot(Nref, I) < 0)\n"
349 " {\n"
350 " return N;\n"
351 " }\n"
352 " else\n"
353 " {\n"
354 " return -N;\n"
355 " }\n"
356 "}\n"
357 "\n"
358 "float2 faceforward(float2 N, float2 I, float2 Nref)\n" // FIXME: Prevent name clashes
359 "{\n"
360 " if(dot(Nref, I) < 0)\n"
361 " {\n"
362 " return N;\n"
363 " }\n"
364 " else\n"
365 " {\n"
366 " return -N;\n"
367 " }\n"
368 "}\n"
369 "\n"
370 "float3 faceforward(float3 N, float3 I, float3 Nref)\n" // FIXME: Prevent name clashes
371 "{\n"
372 " if(dot(Nref, I) < 0)\n"
373 " {\n"
374 " return N;\n"
375 " }\n"
376 " else\n"
377 " {\n"
378 " return -N;\n"
379 " }\n"
380 "}\n"
daniel@transgaming.com49bce7e2010-03-17 03:58:51 +0000381 "\n"
daniel@transgaming.com680553b2010-03-08 21:30:52 +0000382 "float4 faceforward(float4 N, float4 I, float4 Nref)\n" // FIXME: Prevent name clashes
383 "{\n"
384 " if(dot(Nref, I) < 0)\n"
385 " {\n"
386 " return N;\n"
387 " }\n"
388 " else\n"
389 " {\n"
390 " return -N;\n"
391 " }\n"
392 "}\n"
daniel@transgaming.com49bce7e2010-03-17 03:58:51 +0000393 "\n"
394 "bool __equal(float2x2 m, float2x2 n)\n"
395 "{\n"
396 " return m[0][0] == n[0][0] && m[0][1] == n[0][1] &&\n"
397 " m[1][0] == n[1][0] && m[1][1] == n[1][1];\n"
398 "}\n"
399 "\n"
400 "bool __equal(float3x3 m, float3x3 n)\n"
401 "{\n"
402 " return m[0][0] == n[0][0] && m[0][1] == n[0][1] && m[0][2] == n[0][2] &&\n"
403 " m[1][0] == n[1][0] && m[1][1] == n[1][1] && m[1][2] == n[1][2] &&\n"
404 " m[2][0] == n[2][0] && m[2][1] == n[2][1] && m[2][2] == n[2][2];\n"
405 "}\n"
406 "\n"
407 "bool __equal(float4x4 m, float4x4 n)\n"
408 "{\n"
409 " return m[0][0] == n[0][0] && m[0][1] == n[0][1] && m[0][2] == n[0][2] && m[0][3] == n[0][3] &&\n"
410 " m[1][0] == n[1][0] && m[1][1] == n[1][1] && m[1][2] == n[1][2] && m[1][3] == n[1][3] &&\n"
411 " m[2][0] == n[2][0] && m[2][1] == n[2][1] && m[2][2] == n[2][2] && m[2][3] == n[2][3] &&\n"
412 " m[3][0] == n[3][0] && m[3][1] == n[3][1] && m[3][2] == n[3][2] && m[3][3] == n[3][3];\n"
413 "}\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000414 "\n";
415}
416
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000417void OutputHLSL::footer()
418{
419 EShLanguage language = context.language;
420 TInfoSinkBase &out = context.infoSink.obj;
421 TSymbolTableLevel *symbols = context.symbolTable.getGlobalLevel();
422
423 if (language == EShLangFragment)
424 {
425 out << "PS_OUTPUT main(PS_INPUT input)\n" // FIXME: Prevent name clashes
426 "{\n"
427 " float rhw = 1.0 / input.gl_FragCoord.w;\n"
428 " gl_FragCoord.x = (input.gl_FragCoord.x * rhw) * gl_Window.x + gl_Window.z;\n"
429 " gl_FragCoord.y = (input.gl_FragCoord.y * rhw) * gl_Window.y + gl_Window.w;\n"
430 " gl_FragCoord.z = (input.gl_FragCoord.z * rhw) * gl_Depth.x + gl_Depth.y;\n"
431 " gl_FragCoord.w = rhw;\n"
432 " gl_FrontFacing = __frontCCW ? (input.__vFace >= 0.0) : (input.__vFace <= 0.0);\n";
433
434 for (TSymbolTableLevel::const_iterator namedSymbol = symbols->begin(); namedSymbol != symbols->end(); namedSymbol++)
435 {
436 const TSymbol *symbol = (*namedSymbol).second;
437 const TString &name = symbol->getName();
438
439 if (symbol->isVariable())
440 {
441 const TVariable *variable = static_cast<const TVariable*>(symbol);
442 const TType &type = variable->getType();
443 TQualifier qualifier = type.getQualifier();
444
445 if (qualifier == EvqVaryingIn)
446 {
447 out << " " + name + " = input." + name + ";\n"; // FIXME: Prevent name clashes
448 }
449 }
450 }
451
452 out << "\n"
453 " gl_main();\n"
454 "\n"
455 " PS_OUTPUT output;\n" // FIXME: Prevent name clashes
456 " output.gl_Color[0] = gl_Color[0];\n"; // FIXME: Prevent name clashes
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000457 }
daniel@transgaming.comd25ab252010-03-30 03:36:26 +0000458 else // Vertex shader
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000459 {
460 out << "VS_OUTPUT main(VS_INPUT input)\n" // FIXME: Prevent name clashes
461 "{\n";
462
463 for (TSymbolTableLevel::const_iterator namedSymbol = symbols->begin(); namedSymbol != symbols->end(); namedSymbol++)
464 {
465 const TSymbol *symbol = (*namedSymbol).second;
466 const TString &name = symbol->getName();
467
468 if (symbol->isVariable())
469 {
470 const TVariable *variable = static_cast<const TVariable*>(symbol);
471 const TType &type = variable->getType();
472 TQualifier qualifier = type.getQualifier();
473
474 if (qualifier == EvqAttribute)
475 {
476 out << " " + name + " = input." + name + ";\n"; // FIXME: Prevent name clashes
477 }
478 }
479 }
480
481 out << "\n"
482 " gl_main();\n"
483 "\n"
484 " VS_OUTPUT output;\n" // FIXME: Prevent name clashes
485 " output.gl_Position.x = gl_Position.x - gl_HalfPixelSize.x * gl_Position.w;\n"
486 " output.gl_Position.y = -(gl_Position.y - gl_HalfPixelSize.y * gl_Position.w);\n"
487 " output.gl_Position.z = (gl_Position.z + gl_Position.w) * 0.5;\n"
488 " output.gl_Position.w = gl_Position.w;\n"
489 " output.gl_PointSize = gl_PointSize;\n"
490 " output.gl_FragCoord = gl_Position;\n";
491
492 TSymbolTableLevel *symbols = context.symbolTable.getGlobalLevel();
493
494 for (TSymbolTableLevel::const_iterator namedSymbol = symbols->begin(); namedSymbol != symbols->end(); namedSymbol++)
495 {
496 const TSymbol *symbol = (*namedSymbol).second;
497 const TString &name = symbol->getName();
498
499 if (symbol->isVariable())
500 {
501 const TVariable *variable = static_cast<const TVariable*>(symbol);
502 TQualifier qualifier = variable->getType().getQualifier();
503
504 if (qualifier == EvqVaryingOut || qualifier == EvqInvariantVaryingOut)
505 {
daniel@transgaming.com0e3358a2010-04-05 20:32:42 +0000506 // Program linking depends on this exact format
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000507 out << " output." + name + " = " + name + ";\n"; // FIXME: Prevent name clashes
508 }
509 }
510 }
511 }
512
513 out << " return output;\n" // FIXME: Prevent name clashes
514 "}\n";
515}
516
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000517void OutputHLSL::visitSymbol(TIntermSymbol *node)
518{
519 TInfoSinkBase &out = context.infoSink.obj;
520
521 TString name = node->getSymbol();
522
523 if (name == "gl_FragColor")
524 {
525 out << "gl_Color[0]";
526 }
527 else if (name == "gl_FragData")
528 {
529 out << "gl_Color";
530 }
531 else
532 {
533 out << name;
534 }
535}
536
537bool OutputHLSL::visitBinary(Visit visit, TIntermBinary *node)
538{
539 TInfoSinkBase &out = context.infoSink.obj;
540
541 switch (node->getOp())
542 {
daniel@transgaming.com93a96c32010-03-28 19:36:13 +0000543 case EOpAssign: outputTriplet(visit, "(", " = ", ")"); break;
544 case EOpInitialize: outputTriplet(visit, NULL, " = ", NULL); break;
545 case EOpAddAssign: outputTriplet(visit, "(", " += ", ")"); break;
546 case EOpSubAssign: outputTriplet(visit, "(", " -= ", ")"); break;
547 case EOpMulAssign: outputTriplet(visit, "(", " *= ", ")"); break;
548 case EOpVectorTimesScalarAssign: outputTriplet(visit, "(", " *= ", ")"); break;
549 case EOpMatrixTimesScalarAssign: outputTriplet(visit, "(", " *= ", ")"); break;
550 case EOpVectorTimesMatrixAssign:
551 case EOpMatrixTimesMatrixAssign:
552 if (visit == PreVisit)
553 {
554 out << "(";
555 }
556 else if (visit == InVisit)
557 {
558 out << " = mul(";
daniel@transgaming.com279e38a2010-04-03 20:56:13 +0000559
daniel@transgaming.com5441d662010-04-07 03:25:11 +0000560 if (node->getLeft()->getQualifier() == EvqUniform)
daniel@transgaming.com279e38a2010-04-03 20:56:13 +0000561 {
562 out << "transpose(";
563 }
564
daniel@transgaming.com93a96c32010-03-28 19:36:13 +0000565 node->getLeft()->traverse(this);
daniel@transgaming.com279e38a2010-04-03 20:56:13 +0000566
daniel@transgaming.com5441d662010-04-07 03:25:11 +0000567 if (node->getLeft()->getQualifier() == EvqUniform)
daniel@transgaming.com279e38a2010-04-03 20:56:13 +0000568 {
569 out << ")";
570 }
571
daniel@transgaming.com5441d662010-04-07 03:25:11 +0000572 out << ", ";
573
574 if (node->getRight()->getQualifier() == EvqUniform)
575 {
576 out << "transpose(";
577 }
daniel@transgaming.com93a96c32010-03-28 19:36:13 +0000578 }
579 else
580 {
daniel@transgaming.com5441d662010-04-07 03:25:11 +0000581 if (node->getRight()->getQualifier() == EvqUniform)
582 {
583 out << ")";
584 }
585
586 out << "))";
daniel@transgaming.com93a96c32010-03-28 19:36:13 +0000587 }
588 break;
589 case EOpDivAssign: outputTriplet(visit, "(", " /= ", ")"); break;
590 case EOpIndexDirect: outputTriplet(visit, NULL, "[", "]"); break;
591 case EOpIndexIndirect: outputTriplet(visit, NULL, "[", "]"); break;
592 case EOpIndexDirectStruct: outputTriplet(visit, NULL, ".", NULL); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000593 case EOpVectorSwizzle:
594 if (visit == InVisit)
595 {
596 out << ".";
597
598 TIntermAggregate *swizzle = node->getRight()->getAsAggregate();
599
600 if (swizzle)
601 {
602 TIntermSequence &sequence = swizzle->getSequence();
603
604 for (TIntermSequence::iterator sit = sequence.begin(); sit != sequence.end(); sit++)
605 {
606 TIntermConstantUnion *element = (*sit)->getAsConstantUnion();
607
608 if (element)
609 {
610 int i = element->getUnionArrayPointer()[0].getIConst();
611
612 switch (i)
613 {
614 case 0: out << "x"; break;
615 case 1: out << "y"; break;
616 case 2: out << "z"; break;
617 case 3: out << "w"; break;
618 default: UNREACHABLE();
619 }
620 }
621 else UNREACHABLE();
622 }
623 }
624 else UNREACHABLE();
625
626 return false; // Fully processed
627 }
628 break;
629 case EOpAdd: outputTriplet(visit, "(", " + ", ")"); break;
630 case EOpSub: outputTriplet(visit, "(", " - ", ")"); break;
631 case EOpMul: outputTriplet(visit, "(", " * ", ")"); break;
632 case EOpDiv: outputTriplet(visit, "(", " / ", ")"); break;
daniel@transgaming.com49bce7e2010-03-17 03:58:51 +0000633 case EOpEqual:
634 if (!node->getLeft()->isMatrix())
635 {
636 outputTriplet(visit, "(", " == ", ")");
637 }
638 else
639 {
640 outputTriplet(visit, "__equal(", ", ", ")");
641 }
642 break;
643 case EOpNotEqual:
644 if (!node->getLeft()->isMatrix())
645 {
646 outputTriplet(visit, "(", " != ", ")");
647 }
648 else
649 {
650 outputTriplet(visit, "!__equal(", ", ", ")");
651 }
652 break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000653 case EOpLessThan: outputTriplet(visit, "(", " < ", ")"); break;
654 case EOpGreaterThan: outputTriplet(visit, "(", " > ", ")"); break;
655 case EOpLessThanEqual: outputTriplet(visit, "(", " <= ", ")"); break;
656 case EOpGreaterThanEqual: outputTriplet(visit, "(", " >= ", ")"); break;
657 case EOpVectorTimesScalar: outputTriplet(visit, "(", " * ", ")"); break;
daniel@transgaming.com93a96c32010-03-28 19:36:13 +0000658 case EOpMatrixTimesScalar: outputTriplet(visit, "(", " * ", ")"); break;
daniel@transgaming.com5441d662010-04-07 03:25:11 +0000659 case EOpVectorTimesMatrix:
660 if (node->getRight()->getQualifier() == EvqUniform)
661 {
662 outputTriplet(visit, "mul(", ", transpose(", "))");
663 }
664 else
665 {
666 outputTriplet(visit, "mul(", ", ", ")");
667 }
668 break;
669 case EOpMatrixTimesVector:
670 if (node->getLeft()->getQualifier() == EvqUniform)
671 {
672 outputTriplet(visit, "mul(transpose(", "), ", ")");
673 }
674 else
675 {
676 outputTriplet(visit, "mul(", ", ", ")");
677 }
678 break;
679 case EOpMatrixTimesMatrix:
680 if (node->getLeft()->getQualifier() == EvqUniform)
681 {
682 if (node->getRight()->getQualifier() == EvqUniform)
683 {
684 outputTriplet(visit, "mul(transpose(", "), transpose(", "))");
685 }
686 else
687 {
688 outputTriplet(visit, "mul(transpose(", "), ", ")");
689 }
690 }
691 else
692 {
693 if (node->getRight()->getQualifier() == EvqUniform)
694 {
695 outputTriplet(visit, "mul(", ", transpose(", "))");
696 }
697 else
698 {
699 outputTriplet(visit, "mul(", ", ", ")");
700 }
701 }
702 break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000703 case EOpLogicalOr: outputTriplet(visit, "(", " || ", ")"); break;
704 case EOpLogicalXor: outputTriplet(visit, "xor(", ", ", ")"); break; // FIXME: Prevent name clashes
705 case EOpLogicalAnd: outputTriplet(visit, "(", " && ", ")"); break;
706 default: UNREACHABLE();
707 }
708
709 return true;
710}
711
712bool OutputHLSL::visitUnary(Visit visit, TIntermUnary *node)
713{
714 TInfoSinkBase &out = context.infoSink.obj;
715
716 switch (node->getOp())
717 {
718 case EOpNegative: outputTriplet(visit, "(-", NULL, ")"); break;
719 case EOpVectorLogicalNot: outputTriplet(visit, "(!", NULL, ")"); break;
720 case EOpLogicalNot: outputTriplet(visit, "(!", NULL, ")"); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000721 case EOpPostIncrement: outputTriplet(visit, "(", NULL, "++)"); break;
722 case EOpPostDecrement: outputTriplet(visit, "(", NULL, "--)"); break;
723 case EOpPreIncrement: outputTriplet(visit, "(++", NULL, ")"); break;
724 case EOpPreDecrement: outputTriplet(visit, "(--", NULL, ")"); break;
725 case EOpConvIntToBool:
726 case EOpConvFloatToBool:
727 switch (node->getOperand()->getType().getNominalSize())
728 {
729 case 1: outputTriplet(visit, "bool(", NULL, ")"); break;
730 case 2: outputTriplet(visit, "bool2(", NULL, ")"); break;
731 case 3: outputTriplet(visit, "bool3(", NULL, ")"); break;
732 case 4: outputTriplet(visit, "bool4(", NULL, ")"); break;
733 default: UNREACHABLE();
734 }
735 break;
736 case EOpConvBoolToFloat:
737 case EOpConvIntToFloat:
738 switch (node->getOperand()->getType().getNominalSize())
739 {
740 case 1: outputTriplet(visit, "float(", NULL, ")"); break;
741 case 2: outputTriplet(visit, "float2(", NULL, ")"); break;
742 case 3: outputTriplet(visit, "float3(", NULL, ")"); break;
743 case 4: outputTriplet(visit, "float4(", NULL, ")"); break;
744 default: UNREACHABLE();
745 }
746 break;
747 case EOpConvFloatToInt:
748 case EOpConvBoolToInt:
749 switch (node->getOperand()->getType().getNominalSize())
750 {
751 case 1: outputTriplet(visit, "int(", NULL, ")"); break;
752 case 2: outputTriplet(visit, "int2(", NULL, ")"); break;
753 case 3: outputTriplet(visit, "int3(", NULL, ")"); break;
754 case 4: outputTriplet(visit, "int4(", NULL, ")"); break;
755 default: UNREACHABLE();
756 }
757 break;
758 case EOpRadians: outputTriplet(visit, "radians(", NULL, ")"); break;
759 case EOpDegrees: outputTriplet(visit, "degrees(", NULL, ")"); break;
760 case EOpSin: outputTriplet(visit, "sin(", NULL, ")"); break;
761 case EOpCos: outputTriplet(visit, "cos(", NULL, ")"); break;
762 case EOpTan: outputTriplet(visit, "tan(", NULL, ")"); break;
763 case EOpAsin: outputTriplet(visit, "asin(", NULL, ")"); break;
764 case EOpAcos: outputTriplet(visit, "acos(", NULL, ")"); break;
765 case EOpAtan: outputTriplet(visit, "atan(", NULL, ")"); break;
766 case EOpExp: outputTriplet(visit, "exp(", NULL, ")"); break;
767 case EOpLog: outputTriplet(visit, "log(", NULL, ")"); break;
768 case EOpExp2: outputTriplet(visit, "exp2(", NULL, ")"); break;
769 case EOpLog2: outputTriplet(visit, "log2(", NULL, ")"); break;
770 case EOpSqrt: outputTriplet(visit, "sqrt(", NULL, ")"); break;
771 case EOpInverseSqrt: outputTriplet(visit, "rsqrt(", NULL, ")"); break;
772 case EOpAbs: outputTriplet(visit, "abs(", NULL, ")"); break;
773 case EOpSign: outputTriplet(visit, "sign(", NULL, ")"); break;
774 case EOpFloor: outputTriplet(visit, "floor(", NULL, ")"); break;
775 case EOpCeil: outputTriplet(visit, "ceil(", NULL, ")"); break;
776 case EOpFract: outputTriplet(visit, "frac(", NULL, ")"); break;
777 case EOpLength: outputTriplet(visit, "length(", NULL, ")"); break;
778 case EOpNormalize: outputTriplet(visit, "normalize(", NULL, ")"); break;
daniel@transgaming.com45d03582010-03-11 19:41:29 +0000779// case EOpDPdx: outputTriplet(visit, "ddx(", NULL, ")"); break;
780// case EOpDPdy: outputTriplet(visit, "ddy(", NULL, ")"); break;
781// case EOpFwidth: outputTriplet(visit, "fwidth(", NULL, ")"); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000782 case EOpAny: outputTriplet(visit, "any(", NULL, ")"); break;
783 case EOpAll: outputTriplet(visit, "all(", NULL, ")"); break;
784 default: UNREACHABLE();
785 }
786
787 return true;
788}
789
790bool OutputHLSL::visitAggregate(Visit visit, TIntermAggregate *node)
791{
792 EShLanguage language = context.language;
793 TInfoSinkBase &out = context.infoSink.obj;
794
795 if (node->getOp() == EOpNull)
796 {
797 out.message(EPrefixError, "node is still EOpNull!");
798 return true;
799 }
800
801 switch (node->getOp())
802 {
803 case EOpSequence: outputTriplet(visit, NULL, ";\n", ";\n"); break;
804 case EOpDeclaration:
805 if (visit == PreVisit)
806 {
807 TIntermSequence &sequence = node->getSequence();
808 TIntermTyped *variable = sequence[0]->getAsTyped();
809 bool visit = true;
810
daniel@transgaming.comd25ab252010-03-30 03:36:26 +0000811 if (variable && (variable->getQualifier() == EvqTemporary || variable->getQualifier() == EvqGlobal))
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000812 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000813 if (!variable->getAsSymbolNode() || variable->getAsSymbolNode()->getSymbol() != "") // Variable declaration
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000814 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000815 out << typeString(variable->getType()) + " ";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000816
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000817 for (TIntermSequence::iterator sit = sequence.begin(); sit != sequence.end(); sit++)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000818 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000819 TIntermSymbol *symbol = (*sit)->getAsSymbolNode();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000820
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000821 if (symbol)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000822 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000823 symbol->traverse(this);
824
825 out << arrayString(symbol->getType());
826 }
827 else
828 {
829 (*sit)->traverse(this);
830 }
831
832 if (visit && this->inVisit)
833 {
834 if (*sit != sequence.back())
835 {
836 visit = this->visitAggregate(InVisit, node);
837 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000838 }
839 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000840
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000841 if (visit && this->postVisit)
842 {
843 this->visitAggregate(PostVisit, node);
844 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000845 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000846 else if (variable->getAsSymbolNode() && variable->getAsSymbolNode()->getSymbol() == "") // Type (struct) declaration
847 {
848 const TType &type = variable->getType();
849 const TTypeList &fields = *type.getStruct();
850
851 out << "struct " + type.getTypeName() + "\n"
852 "{\n";
853
854 for (unsigned int i = 0; i < fields.size(); i++)
855 {
856 const TType &field = *fields[i].type;
857
858 out << " " + typeString(field) + " " + field.getFieldName() + ";\n";
859 }
860
861 out << "};\n";
862 }
863 else UNREACHABLE();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000864 }
865
866 return false;
867 }
868 else if (visit == InVisit)
869 {
870 out << ", ";
871 }
872 break;
873 case EOpComma: UNIMPLEMENTED(); /* FIXME */ out << "Comma\n"; return true;
874 case EOpFunction:
875 {
alokp@chromium.org43884872010-03-30 00:08:52 +0000876 TString name = TFunction::unmangleName(node->getName());
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000877
878 if (visit == PreVisit)
879 {
880 if (name == "main")
881 {
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000882 name = "gl_main";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000883 }
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000884
885 out << typeString(node->getType()) << " " << name << "(";
886
887 TIntermSequence &sequence = node->getSequence();
888 TIntermSequence &arguments = sequence[0]->getAsAggregate()->getSequence();
889
890 for (unsigned int i = 0; i < arguments.size(); i++)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000891 {
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000892 TIntermSymbol *symbol = arguments[i]->getAsSymbolNode();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000893
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000894 if (symbol)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000895 {
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000896 const TType &type = symbol->getType();
897 const TString &name = symbol->getSymbol();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000898
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000899 out << typeString(type) + " " + name;
900
901 if (i < arguments.size() - 1)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000902 {
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000903 out << ", ";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000904 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000905 }
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000906 else UNREACHABLE();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000907 }
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000908
909 sequence.erase(sequence.begin());
910
911 out << ")\n"
912 "{\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000913 }
914 else if (visit == PostVisit)
915 {
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000916 out << "}\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000917 }
918 }
919 break;
920 case EOpFunctionCall:
921 {
922 if (visit == PreVisit)
923 {
alokp@chromium.org43884872010-03-30 00:08:52 +0000924 TString name = TFunction::unmangleName(node->getName());
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000925
926 if (node->isUserDefined())
927 {
928 out << name << "(";
929 }
930 else
931 {
932 if (name == "texture2D")
933 {
934 if (node->getSequence().size() == 2)
935 {
936 out << "gl_texture2D(";
937 }
938 else if (node->getSequence().size() == 3)
939 {
940 out << "gl_texture2DBias(";
941 }
942 else UNREACHABLE();
943 }
944 else if (name == "texture2DProj")
945 {
946 out << "gl_texture2DProj(";
947 }
948 else if (name == "texture2DLod")
949 {
950 out << "gl_texture2DLod(";
951 UNIMPLEMENTED(); // FIXME: Move lod to last texture coordinate component
952 }
953 else if (name == "texture2DProjLod")
954 {
955 out << "gl_texture2DProjLod(";
956 UNIMPLEMENTED(); // FIXME: Move lod to last texture coordinate component
957 }
958 else if (name == "textureCube")
959 {
960 out << "gl_textureCube("; // FIXME: Incorrect sampling location
961 }
962 else
963 {
964 UNIMPLEMENTED(); // FIXME
965 }
966 }
967 }
968 else if (visit == InVisit)
969 {
970 out << ", ";
971 }
972 else
973 {
974 out << ")";
975 }
976 }
977 break;
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000978 case EOpParameters: outputTriplet(visit, "(", ", ", ")\n{\n"); break;
979 case EOpConstructFloat: outputTriplet(visit, "vec1(", NULL, ")"); break;
980 case EOpConstructVec2: outputTriplet(visit, "vec2(", ", ", ")"); break;
981 case EOpConstructVec3: outputTriplet(visit, "vec3(", ", ", ")"); break;
982 case EOpConstructVec4: outputTriplet(visit, "vec4(", ", ", ")"); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000983 case EOpConstructBool: UNIMPLEMENTED(); /* FIXME */ out << "Construct bool"; break;
984 case EOpConstructBVec2: UNIMPLEMENTED(); /* FIXME */ out << "Construct bvec2"; break;
985 case EOpConstructBVec3: UNIMPLEMENTED(); /* FIXME */ out << "Construct bvec3"; break;
986 case EOpConstructBVec4: UNIMPLEMENTED(); /* FIXME */ out << "Construct bvec4"; break;
987 case EOpConstructInt: UNIMPLEMENTED(); /* FIXME */ out << "Construct int"; break;
988 case EOpConstructIVec2: UNIMPLEMENTED(); /* FIXME */ out << "Construct ivec2"; break;
989 case EOpConstructIVec3: UNIMPLEMENTED(); /* FIXME */ out << "Construct ivec3"; break;
990 case EOpConstructIVec4: UNIMPLEMENTED(); /* FIXME */ out << "Construct ivec4"; break;
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000991 case EOpConstructMat2: outputTriplet(visit, "float2x2(", ", ", ")"); break;
992 case EOpConstructMat3: outputTriplet(visit, "float3x3(", ", ", ")"); break;
993 case EOpConstructMat4: outputTriplet(visit, "float4x4(", ", ", ")"); break;
994 case EOpConstructStruct: outputTriplet(visit, "{", ", ", "}"); break;
995 case EOpLessThan: outputTriplet(visit, "(", " < ", ")"); break;
996 case EOpGreaterThan: outputTriplet(visit, "(", " > ", ")"); break;
997 case EOpLessThanEqual: outputTriplet(visit, "(", " <= ", ")"); break;
998 case EOpGreaterThanEqual: outputTriplet(visit, "(", " >= ", ")"); break;
999 case EOpVectorEqual: outputTriplet(visit, "(", " == ", ")"); break;
1000 case EOpVectorNotEqual: outputTriplet(visit, "(", " != ", ")"); break;
1001 case EOpMod: outputTriplet(visit, "mod(", ", ", ")"); break; // FIXME: Prevent name clashes
1002 case EOpPow: outputTriplet(visit, "pow(", ", ", ")"); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001003 case EOpAtan:
1004 if (node->getSequence().size() == 1)
1005 {
1006 outputTriplet(visit, "atan(", ", ", ")");
1007 }
1008 else if (node->getSequence().size() == 2)
1009 {
1010 outputTriplet(visit, "atan2(", ", ", ")");
1011 }
1012 else UNREACHABLE();
1013 break;
1014 case EOpMin: outputTriplet(visit, "min(", ", ", ")"); break;
1015 case EOpMax: outputTriplet(visit, "max(", ", ", ")"); break;
1016 case EOpClamp: outputTriplet(visit, "clamp(", ", ", ")"); break;
1017 case EOpMix: outputTriplet(visit, "lerp(", ", ", ")"); break;
1018 case EOpStep: outputTriplet(visit, "step(", ", ", ")"); break;
1019 case EOpSmoothStep: outputTriplet(visit, "smoothstep(", ", ", ")"); break;
1020 case EOpDistance: outputTriplet(visit, "distance(", ", ", ")"); break;
1021 case EOpDot: outputTriplet(visit, "dot(", ", ", ")"); break;
1022 case EOpCross: outputTriplet(visit, "cross(", ", ", ")"); break;
daniel@transgaming.com680553b2010-03-08 21:30:52 +00001023 case EOpFaceForward: outputTriplet(visit, "faceforward(", ", ", ")"); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001024 case EOpReflect: outputTriplet(visit, "reflect(", ", ", ")"); break;
1025 case EOpRefract: outputTriplet(visit, "refract(", ", ", ")"); break;
1026 case EOpMul: outputTriplet(visit, "(", " * ", ")"); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001027 default: UNREACHABLE();
1028 }
1029
1030 return true;
1031}
1032
1033bool OutputHLSL::visitSelection(Visit visit, TIntermSelection *node)
1034{
1035 TInfoSinkBase &out = context.infoSink.obj;
1036
alokp@chromium.org60fe4072010-03-29 20:58:29 +00001037 if (node->usesTernaryOperator())
1038 {
1039 out << "(";
1040 node->getCondition()->traverse(this);
1041 out << ") ? (";
1042 node->getTrueBlock()->traverse(this);
1043 out << ") : (";
1044 node->getFalseBlock()->traverse(this);
1045 out << ")\n";
1046 }
1047 else // if/else statement
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001048 {
daniel@transgaming.com3d53fda2010-03-21 04:30:55 +00001049 out << "if(";
1050
1051 node->getCondition()->traverse(this);
1052
1053 out << ")\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001054 "{\n";
1055
daniel@transgaming.com3d53fda2010-03-21 04:30:55 +00001056 node->getTrueBlock()->traverse(this);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001057
1058 out << ";}\n";
daniel@transgaming.com3d53fda2010-03-21 04:30:55 +00001059
1060 if (node->getFalseBlock())
1061 {
1062 out << "else\n"
1063 "{\n";
1064
1065 node->getFalseBlock()->traverse(this);
1066
1067 out << ";}\n";
1068 }
1069 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001070
1071 return false;
1072}
1073
1074void OutputHLSL::visitConstantUnion(TIntermConstantUnion *node)
1075{
1076 TInfoSinkBase &out = context.infoSink.obj;
1077
alokp@chromium.orgdd037b22010-03-30 18:47:20 +00001078 const TType &type = node->getType();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001079
daniel@transgaming.com86487c22010-03-11 19:41:43 +00001080 if (type.isField())
1081 {
1082 out << type.getFieldName();
1083 }
1084 else
1085 {
1086 int size = type.getObjectSize();
daniel@transgaming.com86487c22010-03-11 19:41:43 +00001087
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001088 if (type.getBasicType() == EbtStruct)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001089 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001090 out << "{";
1091 }
1092 else
1093 {
1094 bool matrix = type.isMatrix();
1095 TBasicType elementType = node->getUnionArrayPointer()[0].getType();
1096
1097 switch (elementType)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001098 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001099 case EbtBool:
1100 if (!matrix)
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001101 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001102 switch (size)
1103 {
1104 case 1: out << "bool("; break;
1105 case 2: out << "bool2("; break;
1106 case 3: out << "bool3("; break;
1107 case 4: out << "bool4("; break;
1108 default: UNREACHABLE();
1109 }
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001110 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001111 else
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001112 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001113 UNIMPLEMENTED();
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001114 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001115 break;
1116 case EbtFloat:
1117 if (!matrix)
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001118 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001119 switch (size)
1120 {
1121 case 1: out << "float("; break;
1122 case 2: out << "float2("; break;
1123 case 3: out << "float3("; break;
1124 case 4: out << "float4("; break;
1125 default: UNREACHABLE();
1126 }
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001127 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001128 else
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001129 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001130 switch (size)
1131 {
1132 case 4: out << "float2x2("; break;
1133 case 9: out << "float3x3("; break;
1134 case 16: out << "float4x4("; break;
1135 default: UNREACHABLE();
1136 }
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001137 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001138 break;
1139 case EbtInt:
1140 if (!matrix)
1141 {
1142 switch (size)
1143 {
1144 case 1: out << "int("; break;
1145 case 2: out << "int2("; break;
1146 case 3: out << "int3("; break;
1147 case 4: out << "int4("; break;
1148 default: UNREACHABLE();
1149 }
1150 }
1151 else
1152 {
1153 UNIMPLEMENTED();
1154 }
1155 break;
1156 default:
1157 UNIMPLEMENTED(); // FIXME
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001158 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001159 }
1160
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001161 for (int i = 0; i < size; i++)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001162 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001163 switch (node->getUnionArrayPointer()[i].getType())
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001164 {
1165 case EbtBool:
1166 if (node->getUnionArrayPointer()[i].getBConst())
1167 {
1168 out << "true";
1169 }
1170 else
1171 {
1172 out << "false";
1173 }
1174 break;
1175 case EbtFloat:
1176 out << node->getUnionArrayPointer()[i].getFConst();
1177 break;
1178 case EbtInt:
1179 out << node->getUnionArrayPointer()[i].getIConst();
1180 break;
1181 default:
1182 UNIMPLEMENTED(); // FIXME
1183 }
1184
1185 if (i != size - 1)
1186 {
1187 out << ", ";
1188 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001189 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001190
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001191 if (type.getBasicType() == EbtStruct)
1192 {
1193 out << "}";
1194 }
1195 else
1196 {
1197 out << ")";
1198 }
daniel@transgaming.com86487c22010-03-11 19:41:43 +00001199 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001200}
1201
1202bool OutputHLSL::visitLoop(Visit visit, TIntermLoop *node)
1203{
daniel@transgaming.com4a35ef22010-04-08 03:51:06 +00001204 if (handleExcessiveLoop(node))
1205 {
1206 return false;
1207 }
1208
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001209 TInfoSinkBase &out = context.infoSink.obj;
1210
1211 if (!node->testFirst())
1212 {
1213 out << "do\n"
1214 "{\n";
1215 }
1216 else
1217 {
1218 out << "for(";
1219
1220 if (node->getInit())
1221 {
1222 node->getInit()->traverse(this);
1223 }
1224
1225 out << "; ";
1226
1227 if (node->getTest())
1228 {
1229 node->getTest()->traverse(this);
1230 }
1231
1232 out << "; ";
1233
1234 if (node->getTerminal())
1235 {
1236 node->getTerminal()->traverse(this);
1237 }
1238
1239 out << ")\n"
1240 "{\n";
1241 }
1242
1243 if (node->getBody())
1244 {
1245 node->getBody()->traverse(this);
1246 }
1247
1248 out << "}\n";
1249
1250 if (!node->testFirst())
1251 {
1252 out << "while(\n";
1253
1254 node->getTest()->traverse(this);
1255
1256 out << ")";
1257 }
1258
1259 out << ";\n";
1260
1261 return false;
1262}
1263
1264bool OutputHLSL::visitBranch(Visit visit, TIntermBranch *node)
1265{
1266 TInfoSinkBase &out = context.infoSink.obj;
1267
1268 switch (node->getFlowOp())
1269 {
daniel@transgaming.comf67f82e2010-03-17 03:58:54 +00001270 case EOpKill: outputTriplet(visit, "discard", NULL, NULL); break;
1271 case EOpBreak: outputTriplet(visit, "break", NULL, NULL); break;
1272 case EOpContinue: outputTriplet(visit, "continue", NULL, NULL); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001273 case EOpReturn:
1274 if (visit == PreVisit)
1275 {
1276 if (node->getExpression())
1277 {
1278 out << "return ";
1279 }
1280 else
1281 {
1282 out << "return;\n";
1283 }
1284 }
1285 else if (visit == PostVisit)
1286 {
1287 out << ";\n";
1288 }
1289 break;
1290 default: UNREACHABLE();
1291 }
1292
1293 return true;
1294}
1295
daniel@transgaming.com4a35ef22010-04-08 03:51:06 +00001296// Handle loops with more than 255 iterations (unsupported by D3D9) by splitting them
1297bool OutputHLSL::handleExcessiveLoop(TIntermLoop *node)
1298{
1299 TInfoSinkBase &out = context.infoSink.obj;
1300
1301 // Parse loops of the form:
1302 // for(int index = initial; index [comparator] limit; index += increment)
1303 TIntermSymbol *index = NULL;
1304 TOperator comparator = EOpNull;
1305 int initial = 0;
1306 int limit = 0;
1307 int increment = 0;
1308
1309 // Parse index name and intial value
1310 if (node->getInit())
1311 {
1312 TIntermAggregate *init = node->getInit()->getAsAggregate();
1313
1314 if (init)
1315 {
1316 TIntermSequence &sequence = init->getSequence();
1317 TIntermTyped *variable = sequence[0]->getAsTyped();
1318
1319 if (variable && variable->getQualifier() == EvqTemporary)
1320 {
1321 TIntermBinary *assign = variable->getAsBinaryNode();
1322
1323 if (assign->getOp() == EOpInitialize)
1324 {
1325 TIntermSymbol *symbol = assign->getLeft()->getAsSymbolNode();
1326 TIntermConstantUnion *constant = assign->getRight()->getAsConstantUnion();
1327
1328 if (symbol && constant)
1329 {
1330 if (constant->getBasicType() == EbtInt && constant->getSize() == 1)
1331 {
1332 index = symbol;
1333 initial = constant->getUnionArrayPointer()[0].getIConst();
1334 }
1335 }
1336 }
1337 }
1338 }
1339 }
1340
1341 // Parse comparator and limit value
1342 if (index != NULL && node->getTest())
1343 {
1344 TIntermBinary *test = node->getTest()->getAsBinaryNode();
1345
1346 if (test && test->getLeft()->getAsSymbolNode()->getId() == index->getId())
1347 {
1348 TIntermConstantUnion *constant = test->getRight()->getAsConstantUnion();
1349
1350 if (constant)
1351 {
1352 if (constant->getBasicType() == EbtInt && constant->getSize() == 1)
1353 {
1354 comparator = test->getOp();
1355 limit = constant->getUnionArrayPointer()[0].getIConst();
1356 }
1357 }
1358 }
1359 }
1360
1361 // Parse increment
1362 if (index != NULL && comparator != EOpNull && node->getTerminal())
1363 {
1364 TIntermBinary *binaryTerminal = node->getTerminal()->getAsBinaryNode();
1365 TIntermUnary *unaryTerminal = node->getTerminal()->getAsUnaryNode();
1366
1367 if (binaryTerminal)
1368 {
1369 TOperator op = binaryTerminal->getOp();
1370 TIntermConstantUnion *constant = binaryTerminal->getRight()->getAsConstantUnion();
1371
1372 if (constant)
1373 {
1374 if (constant->getBasicType() == EbtInt && constant->getSize() == 1)
1375 {
1376 int value = constant->getUnionArrayPointer()[0].getIConst();
1377
1378 switch (op)
1379 {
1380 case EOpAddAssign: increment = value; break;
1381 case EOpSubAssign: increment = -value; break;
1382 default: UNIMPLEMENTED();
1383 }
1384 }
1385 }
1386 }
1387 else if (unaryTerminal)
1388 {
1389 TOperator op = unaryTerminal->getOp();
1390
1391 switch (op)
1392 {
1393 case EOpPostIncrement: increment = 1; break;
1394 case EOpPostDecrement: increment = -1; break;
1395 case EOpPreIncrement: increment = 1; break;
1396 case EOpPreDecrement: increment = -1; break;
1397 default: UNIMPLEMENTED();
1398 }
1399 }
1400 }
1401
1402 if (index != NULL && comparator != EOpNull && increment != 0)
1403 {
1404 if (comparator == EOpLessThanEqual)
1405 {
1406 comparator = EOpLessThan;
1407 limit += 1;
1408 }
1409
1410 if (comparator == EOpLessThan)
1411 {
1412 int iterations = (limit - initial + 1) / increment;
1413
1414 if (iterations <= 255)
1415 {
1416 return false; // Not an excessive loop
1417 }
1418
1419 while (iterations > 0)
1420 {
1421 int remainder = (limit - initial + 1) % increment;
1422 int clampedLimit = initial + increment * min(255, iterations) - 1 - remainder;
1423
1424 // for(int index = initial; index < clampedLimit; index += increment)
1425
1426 out << "for(int ";
1427 index->traverse(this);
1428 out << " = ";
1429 out << initial;
1430
1431 out << "; ";
1432 index->traverse(this);
1433 out << " < ";
1434 out << clampedLimit;
1435
1436 out << "; ";
1437 index->traverse(this);
1438 out << " += ";
1439 out << increment;
1440 out << ")\n"
1441 "{\n";
1442
1443 if (node->getBody())
1444 {
1445 node->getBody()->traverse(this);
1446 }
1447
1448 out << "}\n";
1449
1450 initial += 255 * increment;
1451 iterations -= 255;
1452 }
1453
1454 return true;
1455 }
1456 else UNIMPLEMENTED();
1457 }
1458
1459 return false; // Not handled as an excessive loop
1460}
1461
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001462void OutputHLSL::outputTriplet(Visit visit, const char *preString, const char *inString, const char *postString)
1463{
1464 TInfoSinkBase &out = context.infoSink.obj;
1465
1466 if (visit == PreVisit && preString)
1467 {
1468 out << preString;
1469 }
1470 else if (visit == InVisit && inString)
1471 {
1472 out << inString;
1473 }
1474 else if (visit == PostVisit && postString)
1475 {
1476 out << postString;
1477 }
1478}
1479
1480TString OutputHLSL::typeString(const TType &type)
1481{
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001482 if (type.getBasicType() == EbtStruct)
1483 {
1484 return type.getTypeName();
1485 }
1486 else if (type.isMatrix())
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001487 {
1488 switch (type.getNominalSize())
1489 {
1490 case 2: return "float2x2";
1491 case 3: return "float3x3";
1492 case 4: return "float4x4";
1493 }
1494 }
1495 else
1496 {
1497 switch (type.getBasicType())
1498 {
1499 case EbtFloat:
1500 switch (type.getNominalSize())
1501 {
1502 case 1: return "float";
1503 case 2: return "float2";
1504 case 3: return "float3";
1505 case 4: return "float4";
1506 }
1507 case EbtInt:
1508 switch (type.getNominalSize())
1509 {
1510 case 1: return "int";
1511 case 2: return "int2";
1512 case 3: return "int3";
1513 case 4: return "int4";
1514 }
1515 case EbtBool:
1516 switch (type.getNominalSize())
1517 {
1518 case 1: return "bool";
1519 case 2: return "bool2";
1520 case 3: return "bool3";
1521 case 4: return "bool4";
1522 }
1523 case EbtVoid:
1524 return "void";
1525 case EbtSampler2D:
1526 return "sampler2D";
1527 case EbtSamplerCube:
1528 return "samplerCUBE";
1529 }
1530 }
1531
1532 UNIMPLEMENTED(); // FIXME
1533 return "<unknown type>";
1534}
1535
1536TString OutputHLSL::arrayString(const TType &type)
1537{
1538 if (!type.isArray())
1539 {
1540 return "";
1541 }
1542
1543 char buffer[100];
1544 sprintf(buffer, "[%d]", type.getArraySize());
1545
1546 return buffer;
1547}
1548
1549TString OutputHLSL::initializer(const TType &type)
1550{
1551 TString string;
1552
1553 int arraySize = type.isArray() ? type.getArraySize() : 1;
1554
1555 if (type.isArray())
1556 {
1557 string += "{";
1558 }
1559
1560 for (int element = 0; element < arraySize; element++)
1561 {
1562 string += typeString(type) + "(";
1563
1564 for (int component = 0; component < type.getNominalSize(); component++)
1565 {
1566 string += "0";
1567
1568 if (component < type.getNominalSize() - 1)
1569 {
1570 string += ", ";
1571 }
1572 }
1573
1574 string += ")";
1575
1576 if (element < arraySize - 1)
1577 {
1578 string += ", ";
1579 }
1580 }
1581
1582 if (type.isArray())
1583 {
1584 string += "}";
1585 }
1586
1587 return string;
1588}
1589}