blob: d102e20cf6f43d740661238bbb112e268c8f22c5 [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{
daniel@transgaming.com950f9932010-04-13 03:26:14 +000014OutputHLSL::OutputHLSL(TParseContext &context) : TIntermTraverser(true, true, true), mContext(context)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000015{
16}
17
daniel@transgaming.com950f9932010-04-13 03:26:14 +000018void OutputHLSL::output()
19{
20 mContext.treeRoot->traverse(this); // Output the body first to determine what has to go in the header and footer
21 header();
22 footer();
23
24 mContext.infoSink.obj << mHeader.c_str();
25 mContext.infoSink.obj << mBody.c_str();
26 mContext.infoSink.obj << mFooter.c_str();
27}
28
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000029void OutputHLSL::header()
30{
daniel@transgaming.com950f9932010-04-13 03:26:14 +000031 EShLanguage language = mContext.language;
32 TInfoSinkBase &out = mHeader;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000033
34 if (language == EShLangFragment)
35 {
36 TString uniforms;
37 TString varyingInput;
38 TString varyingGlobals;
39
daniel@transgaming.com950f9932010-04-13 03:26:14 +000040 TSymbolTableLevel *symbols = mContext.symbolTable.getGlobalLevel();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000041 int semanticIndex = 0;
42
43 for (TSymbolTableLevel::const_iterator namedSymbol = symbols->begin(); namedSymbol != symbols->end(); namedSymbol++)
44 {
45 const TSymbol *symbol = (*namedSymbol).second;
46 const TString &name = symbol->getName();
47
48 if (symbol->isVariable())
49 {
50 const TVariable *variable = static_cast<const TVariable*>(symbol);
51 const TType &type = variable->getType();
52 TQualifier qualifier = type.getQualifier();
53
54 if (qualifier == EvqUniform)
55 {
56 uniforms += "uniform " + typeString(type) + " " + name + arrayString(type) + ";\n";
57 }
58 else if (qualifier == EvqVaryingIn || qualifier == EvqInvariantVaryingIn)
59 {
60 char semantic[100];
61 sprintf(semantic, " : TEXCOORD%d", semanticIndex);
62 semanticIndex += type.isArray() ? type.getArraySize() : 1;
63
daniel@transgaming.com0e3358a2010-04-05 20:32:42 +000064 // Program linking depends on this exact format
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000065 varyingInput += " " + typeString(type) + " " + name + arrayString(type) + semantic + ";\n";
66 varyingGlobals += "static " + typeString(type) + " " + name + arrayString(type) + " = " + initializer(type) + ";\n";
67 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +000068 else if (qualifier == EvqGlobal || qualifier == EvqTemporary)
daniel@transgaming.comd25ab252010-03-30 03:36:26 +000069 {
70 // Globals are declared and intialized as an aggregate node
71 }
daniel@transgaming.com49bce7e2010-03-17 03:58:51 +000072 else if (qualifier == EvqConst)
73 {
74 // Constants are repeated as literals where used
75 }
76 else UNREACHABLE();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000077 }
78 }
79
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +000080 out << "uniform float4 gl_Window;\n"
81 "uniform float2 gl_Depth;\n"
daniel@transgaming.com79b820b2010-03-16 05:48:57 +000082 "uniform bool __frontCCW;\n"
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +000083 "\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000084 out << uniforms;
daniel@transgaming.com86487c22010-03-11 19:41:43 +000085 out << "\n"
86 "struct PS_INPUT\n" // FIXME: Prevent name clashes
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000087 "{\n";
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +000088 out << varyingInput;
daniel@transgaming.com0e3358a2010-04-05 20:32:42 +000089 out << " float4 gl_FragCoord : TEXCOORD" << semanticIndex << ";\n";
daniel@transgaming.com79b820b2010-03-16 05:48:57 +000090 out << " float __vFace : VFACE;\n"
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +000091 "};\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000092 "\n";
93 out << varyingGlobals;
94 out << "\n"
95 "struct PS_OUTPUT\n" // FIXME: Prevent name clashes
96 "{\n"
97 " float4 gl_Color[1] : COLOR;\n"
98 "};\n"
99 "\n"
100 "static float4 gl_Color[1] = {float4(0, 0, 0, 0)};\n"
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +0000101 "static float4 gl_FragCoord = float4(0, 0, 0, 0);\n"
daniel@transgaming.comccad59f2010-03-26 04:08:39 +0000102 "static float2 gl_PointCoord = float2(0.5, 0.5);\n"
daniel@transgaming.com79b820b2010-03-16 05:48:57 +0000103 "static bool gl_FrontFacing = false;\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000104 "\n"
105 "float4 gl_texture2D(sampler2D s, float2 t)\n"
106 "{\n"
107 " return tex2D(s, t);\n"
108 "}\n"
109 "\n"
110 "float4 gl_texture2DProj(sampler2D s, float3 t)\n"
111 "{\n"
112 " return tex2Dproj(s, float4(t.x, t.y, 0, t.z));\n"
113 "}\n"
114 "float4 gl_texture2DBias(sampler2D s, float2 t, float bias)\n"
115 "{\n"
116 " return tex2Dbias(s, float4(t.x, t.y, 0, bias));\n"
117 "}\n"
118 "\n"
119 "float4 gl_textureCube(samplerCUBE s, float3 t)\n"
120 "{\n"
121 " return texCUBE(s, t);\n"
122 "}\n"
123 "\n";
124 }
daniel@transgaming.comd25ab252010-03-30 03:36:26 +0000125 else // Vertex shader
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000126 {
127 TString uniforms;
128 TString attributeInput;
129 TString attributeGlobals;
130 TString varyingOutput;
131 TString varyingGlobals;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000132
daniel@transgaming.com950f9932010-04-13 03:26:14 +0000133 TSymbolTableLevel *symbols = mContext.symbolTable.getGlobalLevel();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000134 int semanticIndex = 0;
135
136 for (TSymbolTableLevel::const_iterator namedSymbol = symbols->begin(); namedSymbol != symbols->end(); namedSymbol++)
137 {
138 const TSymbol *symbol = (*namedSymbol).second;
139 const TString &name = symbol->getName();
140
141 if (symbol->isVariable())
142 {
143 const TVariable *variable = static_cast<const TVariable*>(symbol);
144 const TType &type = variable->getType();
145 TQualifier qualifier = type.getQualifier();
146
147 if (qualifier == EvqUniform)
148 {
149 uniforms += "uniform " + typeString(type) + " " + name + arrayString(type) + ";\n";
150 }
151 else if (qualifier == EvqAttribute)
152 {
153 char semantic[100];
154 sprintf(semantic, " : TEXCOORD%d", semanticIndex);
155 semanticIndex += type.isArray() ? type.getArraySize() : 1;
156
157 attributeInput += " " + typeString(type) + " " + name + arrayString(type) + semantic + ";\n";
158 attributeGlobals += "static " + typeString(type) + " " + name + arrayString(type) + " = " + initializer(type) + ";\n";
159 }
160 else if (qualifier == EvqVaryingOut || qualifier == EvqInvariantVaryingOut)
161 {
daniel@transgaming.com0e3358a2010-04-05 20:32:42 +0000162 // Program linking depends on this exact format
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000163 varyingOutput += " " + typeString(type) + " " + name + arrayString(type) + " : TEXCOORD0;\n"; // Actual semantic index assigned during link
164 varyingGlobals += "static " + typeString(type) + " " + name + arrayString(type) + " = " + initializer(type) + ";\n";
165 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000166 else if (qualifier == EvqGlobal || qualifier == EvqTemporary)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000167 {
daniel@transgaming.comd25ab252010-03-30 03:36:26 +0000168 // Globals are declared and intialized as an aggregate node
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000169 }
daniel@transgaming.com49bce7e2010-03-17 03:58:51 +0000170 else if (qualifier == EvqConst)
171 {
172 // Constants are repeated as literals where used
173 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000174 else UNREACHABLE();
175 }
176 }
177
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +0000178 out << "uniform float2 gl_HalfPixelSize;\n"
179 "\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000180 out << uniforms;
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +0000181 out << "\n"
182 "struct VS_INPUT\n" // FIXME: Prevent name clashes
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000183 "{\n";
184 out << attributeInput;
185 out << "};\n"
186 "\n";
187 out << attributeGlobals;
188 out << "\n"
189 "struct VS_OUTPUT\n" // FIXME: Prevent name clashes
190 "{\n"
191 " float4 gl_Position : POSITION;\n"
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +0000192 " float gl_PointSize : PSIZE;\n"
daniel@transgaming.com0e3358a2010-04-05 20:32:42 +0000193 " float4 gl_FragCoord : TEXCOORD0;\n"; // Actual semantic index assigned during link
daniel@transgaming.com9b5f5442010-03-16 05:43:55 +0000194 out << varyingOutput;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000195 out << "};\n"
196 "\n"
197 "static float4 gl_Position = float4(0, 0, 0, 0);\n"
daniel@transgaming.comccad59f2010-03-26 04:08:39 +0000198 "static float gl_PointSize = float(1);\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000199 out << varyingGlobals;
200 out << "\n";
201 }
202
daniel@transgaming.com86487c22010-03-11 19:41:43 +0000203 out << "struct gl_DepthRangeParameters\n"
204 "{\n"
205 " float near;\n"
206 " float far;\n"
207 " float diff;\n"
208 "};\n"
209 "\n"
210 "uniform gl_DepthRangeParameters gl_DepthRange;\n"
211 "\n"
212 "float vec1(float x)\n" // FIXME: Prevent name clashes
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000213 "{\n"
214 " return x;\n"
215 "}\n"
216 "\n"
217 "float vec1(float2 xy)\n" // FIXME: Prevent name clashes
218 "{\n"
219 " return xy[0];\n"
220 "}\n"
221 "\n"
222 "float vec1(float3 xyz)\n" // FIXME: Prevent name clashes
223 "{\n"
224 " return xyz[0];\n"
225 "}\n"
226 "\n"
227 "float vec1(float4 xyzw)\n" // FIXME: Prevent name clashes
228 "{\n"
229 " return xyzw[0];\n"
230 "}\n"
231 "\n"
232 "float2 vec2(float x)\n" // FIXME: Prevent name clashes
233 "{\n"
234 " return float2(x, x);\n"
235 "}\n"
236 "\n"
237 "float2 vec2(float x, float y)\n" // FIXME: Prevent name clashes
238 "{\n"
239 " return float2(x, y);\n"
240 "}\n"
241 "\n"
242 "float2 vec2(float2 xy)\n" // FIXME: Prevent name clashes
243 "{\n"
244 " return xy;\n"
245 "}\n"
246 "\n"
247 "float2 vec2(float3 xyz)\n" // FIXME: Prevent name clashes
248 "{\n"
249 " return float2(xyz[0], xyz[1]);\n"
250 "}\n"
251 "\n"
252 "float2 vec2(float4 xyzw)\n" // FIXME: Prevent name clashes
253 "{\n"
254 " return float2(xyzw[0], xyzw[1]);\n"
255 "}\n"
256 "\n"
257 "float3 vec3(float x)\n" // FIXME: Prevent name clashes
258 "{\n"
259 " return float3(x, x, x);\n"
260 "}\n"
261 "\n"
262 "float3 vec3(float x, float y, float z)\n" // FIXME: Prevent name clashes
263 "{\n"
264 " return float3(x, y, z);\n"
265 "}\n"
266 "\n"
267 "float3 vec3(float2 xy, float z)\n" // FIXME: Prevent name clashes
268 "{\n"
269 " return float3(xy[0], xy[1], z);\n"
270 "}\n"
271 "\n"
272 "float3 vec3(float x, float2 yz)\n" // FIXME: Prevent name clashes
273 "{\n"
274 " return float3(x, yz[0], yz[1]);\n"
275 "}\n"
276 "\n"
277 "float3 vec3(float3 xyz)\n" // FIXME: Prevent name clashes
278 "{\n"
279 " return xyz;\n"
280 "}\n"
281 "\n"
282 "float3 vec3(float4 xyzw)\n" // FIXME: Prevent name clashes
283 "{\n"
284 " return float3(xyzw[0], xyzw[1], xyzw[2]);\n"
285 "}\n"
286 "\n"
287 "float4 vec4(float x)\n" // FIXME: Prevent name clashes
288 "{\n"
289 " return float4(x, x, x, x);\n"
290 "}\n"
291 "\n"
292 "float4 vec4(float x, float y, float z, float w)\n" // FIXME: Prevent name clashes
293 "{\n"
294 " return float4(x, y, z, w);\n"
295 "}\n"
296 "\n"
297 "float4 vec4(float2 xy, float z, float w)\n" // FIXME: Prevent name clashes
298 "{\n"
299 " return float4(xy[0], xy[1], z, w);\n"
300 "}\n"
301 "\n"
302 "float4 vec4(float x, float2 yz, float w)\n" // FIXME: Prevent name clashes
303 "{\n"
304 " return float4(x, yz[0], yz[1], w);\n"
305 "}\n"
306 "\n"
307 "float4 vec4(float x, float y, float2 zw)\n" // FIXME: Prevent name clashes
308 "{\n"
309 " return float4(x, y, zw[0], zw[1]);\n"
310 "}\n"
311 "\n"
312 "float4 vec4(float2 xy, float2 zw)\n" // FIXME: Prevent name clashes
313 "{\n"
314 " return float4(xy[0], xy[1], zw[0], zw[1]);\n"
315 "}\n"
316 "\n"
317 "float4 vec4(float3 xyz, float w)\n" // FIXME: Prevent name clashes
318 "{\n"
319 " return float4(xyz[0], xyz[1], xyz[2], w);\n"
320 "}\n"
321 "\n"
322 "float4 vec4(float x, float3 yzw)\n" // FIXME: Prevent name clashes
323 "{\n"
324 " return float4(x, yzw[0], yzw[1], yzw[2]);\n"
325 "}\n"
326 "\n"
327 "float4 vec4(float4 xyzw)\n" // FIXME: Prevent name clashes
328 "{\n"
329 " return xyzw;\n"
330 "}\n"
331 "\n"
332 "bool xor(bool p, bool q)\n" // FIXME: Prevent name clashes
333 "{\n"
334 " return (p || q) && !(p && q);\n"
335 "}\n"
336 "\n"
337 "float mod(float x, float y)\n" // FIXME: Prevent name clashes
338 "{\n"
339 " return x - y * floor(x / y);\n"
340 "}\n"
daniel@transgaming.com680553b2010-03-08 21:30:52 +0000341 "\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000342 "float2 mod(float2 x, float y)\n" // FIXME: Prevent name clashes
343 "{\n"
344 " return x - y * floor(x / y);\n"
345 "}\n"
daniel@transgaming.com680553b2010-03-08 21:30:52 +0000346 "\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000347 "float3 mod(float3 x, float y)\n" // FIXME: Prevent name clashes
348 "{\n"
349 " return x - y * floor(x / y);\n"
350 "}\n"
daniel@transgaming.com680553b2010-03-08 21:30:52 +0000351 "\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000352 "float4 mod(float4 x, float y)\n" // FIXME: Prevent name clashes
353 "{\n"
354 " return x - y * floor(x / y);\n"
355 "}\n"
daniel@transgaming.com680553b2010-03-08 21:30:52 +0000356 "\n"
357 "float faceforward(float N, float I, float Nref)\n" // FIXME: Prevent name clashes
358 "{\n"
359 " if(dot(Nref, I) < 0)\n"
360 " {\n"
361 " return N;\n"
362 " }\n"
363 " else\n"
364 " {\n"
365 " return -N;\n"
366 " }\n"
367 "}\n"
368 "\n"
369 "float2 faceforward(float2 N, float2 I, float2 Nref)\n" // FIXME: Prevent name clashes
370 "{\n"
371 " if(dot(Nref, I) < 0)\n"
372 " {\n"
373 " return N;\n"
374 " }\n"
375 " else\n"
376 " {\n"
377 " return -N;\n"
378 " }\n"
379 "}\n"
380 "\n"
381 "float3 faceforward(float3 N, float3 I, float3 Nref)\n" // FIXME: Prevent name clashes
382 "{\n"
383 " if(dot(Nref, I) < 0)\n"
384 " {\n"
385 " return N;\n"
386 " }\n"
387 " else\n"
388 " {\n"
389 " return -N;\n"
390 " }\n"
391 "}\n"
daniel@transgaming.com49bce7e2010-03-17 03:58:51 +0000392 "\n"
daniel@transgaming.com680553b2010-03-08 21:30:52 +0000393 "float4 faceforward(float4 N, float4 I, float4 Nref)\n" // FIXME: Prevent name clashes
394 "{\n"
395 " if(dot(Nref, I) < 0)\n"
396 " {\n"
397 " return N;\n"
398 " }\n"
399 " else\n"
400 " {\n"
401 " return -N;\n"
402 " }\n"
403 "}\n"
daniel@transgaming.com49bce7e2010-03-17 03:58:51 +0000404 "\n"
405 "bool __equal(float2x2 m, float2x2 n)\n"
406 "{\n"
407 " return m[0][0] == n[0][0] && m[0][1] == n[0][1] &&\n"
408 " m[1][0] == n[1][0] && m[1][1] == n[1][1];\n"
409 "}\n"
410 "\n"
411 "bool __equal(float3x3 m, float3x3 n)\n"
412 "{\n"
413 " return m[0][0] == n[0][0] && m[0][1] == n[0][1] && m[0][2] == n[0][2] &&\n"
414 " m[1][0] == n[1][0] && m[1][1] == n[1][1] && m[1][2] == n[1][2] &&\n"
415 " m[2][0] == n[2][0] && m[2][1] == n[2][1] && m[2][2] == n[2][2];\n"
416 "}\n"
417 "\n"
418 "bool __equal(float4x4 m, float4x4 n)\n"
419 "{\n"
420 " 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"
421 " 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"
422 " 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"
423 " 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"
424 "}\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000425 "\n";
426}
427
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000428void OutputHLSL::footer()
429{
daniel@transgaming.com950f9932010-04-13 03:26:14 +0000430 EShLanguage language = mContext.language;
431 TInfoSinkBase &out = mFooter;
432 TSymbolTableLevel *symbols = mContext.symbolTable.getGlobalLevel();
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000433
434 if (language == EShLangFragment)
435 {
436 out << "PS_OUTPUT main(PS_INPUT input)\n" // FIXME: Prevent name clashes
437 "{\n"
438 " float rhw = 1.0 / input.gl_FragCoord.w;\n"
439 " gl_FragCoord.x = (input.gl_FragCoord.x * rhw) * gl_Window.x + gl_Window.z;\n"
440 " gl_FragCoord.y = (input.gl_FragCoord.y * rhw) * gl_Window.y + gl_Window.w;\n"
441 " gl_FragCoord.z = (input.gl_FragCoord.z * rhw) * gl_Depth.x + gl_Depth.y;\n"
442 " gl_FragCoord.w = rhw;\n"
443 " gl_FrontFacing = __frontCCW ? (input.__vFace >= 0.0) : (input.__vFace <= 0.0);\n";
444
445 for (TSymbolTableLevel::const_iterator namedSymbol = symbols->begin(); namedSymbol != symbols->end(); namedSymbol++)
446 {
447 const TSymbol *symbol = (*namedSymbol).second;
448 const TString &name = symbol->getName();
449
450 if (symbol->isVariable())
451 {
452 const TVariable *variable = static_cast<const TVariable*>(symbol);
453 const TType &type = variable->getType();
454 TQualifier qualifier = type.getQualifier();
455
456 if (qualifier == EvqVaryingIn)
457 {
458 out << " " + name + " = input." + name + ";\n"; // FIXME: Prevent name clashes
459 }
460 }
461 }
462
463 out << "\n"
464 " gl_main();\n"
465 "\n"
466 " PS_OUTPUT output;\n" // FIXME: Prevent name clashes
467 " output.gl_Color[0] = gl_Color[0];\n"; // FIXME: Prevent name clashes
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000468 }
daniel@transgaming.comd25ab252010-03-30 03:36:26 +0000469 else // Vertex shader
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000470 {
471 out << "VS_OUTPUT main(VS_INPUT input)\n" // FIXME: Prevent name clashes
472 "{\n";
473
474 for (TSymbolTableLevel::const_iterator namedSymbol = symbols->begin(); namedSymbol != symbols->end(); namedSymbol++)
475 {
476 const TSymbol *symbol = (*namedSymbol).second;
477 const TString &name = symbol->getName();
478
479 if (symbol->isVariable())
480 {
481 const TVariable *variable = static_cast<const TVariable*>(symbol);
482 const TType &type = variable->getType();
483 TQualifier qualifier = type.getQualifier();
484
485 if (qualifier == EvqAttribute)
486 {
487 out << " " + name + " = input." + name + ";\n"; // FIXME: Prevent name clashes
488 }
489 }
490 }
491
492 out << "\n"
493 " gl_main();\n"
494 "\n"
495 " VS_OUTPUT output;\n" // FIXME: Prevent name clashes
496 " output.gl_Position.x = gl_Position.x - gl_HalfPixelSize.x * gl_Position.w;\n"
497 " output.gl_Position.y = -(gl_Position.y - gl_HalfPixelSize.y * gl_Position.w);\n"
498 " output.gl_Position.z = (gl_Position.z + gl_Position.w) * 0.5;\n"
499 " output.gl_Position.w = gl_Position.w;\n"
500 " output.gl_PointSize = gl_PointSize;\n"
501 " output.gl_FragCoord = gl_Position;\n";
502
daniel@transgaming.com950f9932010-04-13 03:26:14 +0000503 TSymbolTableLevel *symbols = mContext.symbolTable.getGlobalLevel();
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000504
505 for (TSymbolTableLevel::const_iterator namedSymbol = symbols->begin(); namedSymbol != symbols->end(); namedSymbol++)
506 {
507 const TSymbol *symbol = (*namedSymbol).second;
508 const TString &name = symbol->getName();
509
510 if (symbol->isVariable())
511 {
512 const TVariable *variable = static_cast<const TVariable*>(symbol);
513 TQualifier qualifier = variable->getType().getQualifier();
514
515 if (qualifier == EvqVaryingOut || qualifier == EvqInvariantVaryingOut)
516 {
daniel@transgaming.com0e3358a2010-04-05 20:32:42 +0000517 // Program linking depends on this exact format
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000518 out << " output." + name + " = " + name + ";\n"; // FIXME: Prevent name clashes
519 }
520 }
521 }
522 }
523
524 out << " return output;\n" // FIXME: Prevent name clashes
525 "}\n";
526}
527
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000528void OutputHLSL::visitSymbol(TIntermSymbol *node)
529{
daniel@transgaming.com950f9932010-04-13 03:26:14 +0000530 TInfoSinkBase &out = mBody;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000531
532 TString name = node->getSymbol();
533
534 if (name == "gl_FragColor")
535 {
536 out << "gl_Color[0]";
537 }
538 else if (name == "gl_FragData")
539 {
540 out << "gl_Color";
541 }
542 else
543 {
544 out << name;
545 }
546}
547
548bool OutputHLSL::visitBinary(Visit visit, TIntermBinary *node)
549{
daniel@transgaming.com950f9932010-04-13 03:26:14 +0000550 TInfoSinkBase &out = mBody;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000551
552 switch (node->getOp())
553 {
daniel@transgaming.com93a96c32010-03-28 19:36:13 +0000554 case EOpAssign: outputTriplet(visit, "(", " = ", ")"); break;
555 case EOpInitialize: outputTriplet(visit, NULL, " = ", NULL); break;
556 case EOpAddAssign: outputTriplet(visit, "(", " += ", ")"); break;
557 case EOpSubAssign: outputTriplet(visit, "(", " -= ", ")"); break;
558 case EOpMulAssign: outputTriplet(visit, "(", " *= ", ")"); break;
559 case EOpVectorTimesScalarAssign: outputTriplet(visit, "(", " *= ", ")"); break;
560 case EOpMatrixTimesScalarAssign: outputTriplet(visit, "(", " *= ", ")"); break;
561 case EOpVectorTimesMatrixAssign:
562 case EOpMatrixTimesMatrixAssign:
563 if (visit == PreVisit)
564 {
565 out << "(";
566 }
567 else if (visit == InVisit)
568 {
569 out << " = mul(";
daniel@transgaming.com279e38a2010-04-03 20:56:13 +0000570
daniel@transgaming.com5441d662010-04-07 03:25:11 +0000571 if (node->getLeft()->getQualifier() == EvqUniform)
daniel@transgaming.com279e38a2010-04-03 20:56:13 +0000572 {
573 out << "transpose(";
574 }
575
daniel@transgaming.com93a96c32010-03-28 19:36:13 +0000576 node->getLeft()->traverse(this);
daniel@transgaming.com279e38a2010-04-03 20:56:13 +0000577
daniel@transgaming.com5441d662010-04-07 03:25:11 +0000578 if (node->getLeft()->getQualifier() == EvqUniform)
daniel@transgaming.com279e38a2010-04-03 20:56:13 +0000579 {
580 out << ")";
581 }
582
daniel@transgaming.com5441d662010-04-07 03:25:11 +0000583 out << ", ";
584
585 if (node->getRight()->getQualifier() == EvqUniform)
586 {
587 out << "transpose(";
588 }
daniel@transgaming.com93a96c32010-03-28 19:36:13 +0000589 }
590 else
591 {
daniel@transgaming.com5441d662010-04-07 03:25:11 +0000592 if (node->getRight()->getQualifier() == EvqUniform)
593 {
594 out << ")";
595 }
596
597 out << "))";
daniel@transgaming.com93a96c32010-03-28 19:36:13 +0000598 }
599 break;
600 case EOpDivAssign: outputTriplet(visit, "(", " /= ", ")"); break;
601 case EOpIndexDirect: outputTriplet(visit, NULL, "[", "]"); break;
602 case EOpIndexIndirect: outputTriplet(visit, NULL, "[", "]"); break;
603 case EOpIndexDirectStruct: outputTriplet(visit, NULL, ".", NULL); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000604 case EOpVectorSwizzle:
605 if (visit == InVisit)
606 {
607 out << ".";
608
609 TIntermAggregate *swizzle = node->getRight()->getAsAggregate();
610
611 if (swizzle)
612 {
613 TIntermSequence &sequence = swizzle->getSequence();
614
615 for (TIntermSequence::iterator sit = sequence.begin(); sit != sequence.end(); sit++)
616 {
617 TIntermConstantUnion *element = (*sit)->getAsConstantUnion();
618
619 if (element)
620 {
621 int i = element->getUnionArrayPointer()[0].getIConst();
622
623 switch (i)
624 {
625 case 0: out << "x"; break;
626 case 1: out << "y"; break;
627 case 2: out << "z"; break;
628 case 3: out << "w"; break;
629 default: UNREACHABLE();
630 }
631 }
632 else UNREACHABLE();
633 }
634 }
635 else UNREACHABLE();
636
637 return false; // Fully processed
638 }
639 break;
640 case EOpAdd: outputTriplet(visit, "(", " + ", ")"); break;
641 case EOpSub: outputTriplet(visit, "(", " - ", ")"); break;
642 case EOpMul: outputTriplet(visit, "(", " * ", ")"); break;
643 case EOpDiv: outputTriplet(visit, "(", " / ", ")"); break;
daniel@transgaming.com49bce7e2010-03-17 03:58:51 +0000644 case EOpEqual:
645 if (!node->getLeft()->isMatrix())
646 {
647 outputTriplet(visit, "(", " == ", ")");
648 }
649 else
650 {
651 outputTriplet(visit, "__equal(", ", ", ")");
652 }
653 break;
654 case EOpNotEqual:
655 if (!node->getLeft()->isMatrix())
656 {
657 outputTriplet(visit, "(", " != ", ")");
658 }
659 else
660 {
661 outputTriplet(visit, "!__equal(", ", ", ")");
662 }
663 break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000664 case EOpLessThan: outputTriplet(visit, "(", " < ", ")"); break;
665 case EOpGreaterThan: outputTriplet(visit, "(", " > ", ")"); break;
666 case EOpLessThanEqual: outputTriplet(visit, "(", " <= ", ")"); break;
667 case EOpGreaterThanEqual: outputTriplet(visit, "(", " >= ", ")"); break;
668 case EOpVectorTimesScalar: outputTriplet(visit, "(", " * ", ")"); break;
daniel@transgaming.com93a96c32010-03-28 19:36:13 +0000669 case EOpMatrixTimesScalar: outputTriplet(visit, "(", " * ", ")"); break;
daniel@transgaming.com5441d662010-04-07 03:25:11 +0000670 case EOpVectorTimesMatrix:
671 if (node->getRight()->getQualifier() == EvqUniform)
672 {
673 outputTriplet(visit, "mul(", ", transpose(", "))");
674 }
675 else
676 {
677 outputTriplet(visit, "mul(", ", ", ")");
678 }
679 break;
680 case EOpMatrixTimesVector:
681 if (node->getLeft()->getQualifier() == EvqUniform)
682 {
683 outputTriplet(visit, "mul(transpose(", "), ", ")");
684 }
685 else
686 {
687 outputTriplet(visit, "mul(", ", ", ")");
688 }
689 break;
690 case EOpMatrixTimesMatrix:
691 if (node->getLeft()->getQualifier() == EvqUniform)
692 {
693 if (node->getRight()->getQualifier() == EvqUniform)
694 {
695 outputTriplet(visit, "mul(transpose(", "), transpose(", "))");
696 }
697 else
698 {
699 outputTriplet(visit, "mul(transpose(", "), ", ")");
700 }
701 }
702 else
703 {
704 if (node->getRight()->getQualifier() == EvqUniform)
705 {
706 outputTriplet(visit, "mul(", ", transpose(", "))");
707 }
708 else
709 {
710 outputTriplet(visit, "mul(", ", ", ")");
711 }
712 }
713 break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000714 case EOpLogicalOr: outputTriplet(visit, "(", " || ", ")"); break;
715 case EOpLogicalXor: outputTriplet(visit, "xor(", ", ", ")"); break; // FIXME: Prevent name clashes
716 case EOpLogicalAnd: outputTriplet(visit, "(", " && ", ")"); break;
717 default: UNREACHABLE();
718 }
719
720 return true;
721}
722
723bool OutputHLSL::visitUnary(Visit visit, TIntermUnary *node)
724{
daniel@transgaming.com950f9932010-04-13 03:26:14 +0000725 TInfoSinkBase &out = mBody;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000726
727 switch (node->getOp())
728 {
729 case EOpNegative: outputTriplet(visit, "(-", NULL, ")"); break;
730 case EOpVectorLogicalNot: outputTriplet(visit, "(!", NULL, ")"); break;
731 case EOpLogicalNot: outputTriplet(visit, "(!", NULL, ")"); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000732 case EOpPostIncrement: outputTriplet(visit, "(", NULL, "++)"); break;
733 case EOpPostDecrement: outputTriplet(visit, "(", NULL, "--)"); break;
734 case EOpPreIncrement: outputTriplet(visit, "(++", NULL, ")"); break;
735 case EOpPreDecrement: outputTriplet(visit, "(--", NULL, ")"); break;
736 case EOpConvIntToBool:
737 case EOpConvFloatToBool:
738 switch (node->getOperand()->getType().getNominalSize())
739 {
740 case 1: outputTriplet(visit, "bool(", NULL, ")"); break;
741 case 2: outputTriplet(visit, "bool2(", NULL, ")"); break;
742 case 3: outputTriplet(visit, "bool3(", NULL, ")"); break;
743 case 4: outputTriplet(visit, "bool4(", NULL, ")"); break;
744 default: UNREACHABLE();
745 }
746 break;
747 case EOpConvBoolToFloat:
748 case EOpConvIntToFloat:
749 switch (node->getOperand()->getType().getNominalSize())
750 {
751 case 1: outputTriplet(visit, "float(", NULL, ")"); break;
752 case 2: outputTriplet(visit, "float2(", NULL, ")"); break;
753 case 3: outputTriplet(visit, "float3(", NULL, ")"); break;
754 case 4: outputTriplet(visit, "float4(", NULL, ")"); break;
755 default: UNREACHABLE();
756 }
757 break;
758 case EOpConvFloatToInt:
759 case EOpConvBoolToInt:
760 switch (node->getOperand()->getType().getNominalSize())
761 {
762 case 1: outputTriplet(visit, "int(", NULL, ")"); break;
763 case 2: outputTriplet(visit, "int2(", NULL, ")"); break;
764 case 3: outputTriplet(visit, "int3(", NULL, ")"); break;
765 case 4: outputTriplet(visit, "int4(", NULL, ")"); break;
766 default: UNREACHABLE();
767 }
768 break;
769 case EOpRadians: outputTriplet(visit, "radians(", NULL, ")"); break;
770 case EOpDegrees: outputTriplet(visit, "degrees(", NULL, ")"); break;
771 case EOpSin: outputTriplet(visit, "sin(", NULL, ")"); break;
772 case EOpCos: outputTriplet(visit, "cos(", NULL, ")"); break;
773 case EOpTan: outputTriplet(visit, "tan(", NULL, ")"); break;
774 case EOpAsin: outputTriplet(visit, "asin(", NULL, ")"); break;
775 case EOpAcos: outputTriplet(visit, "acos(", NULL, ")"); break;
776 case EOpAtan: outputTriplet(visit, "atan(", NULL, ")"); break;
777 case EOpExp: outputTriplet(visit, "exp(", NULL, ")"); break;
778 case EOpLog: outputTriplet(visit, "log(", NULL, ")"); break;
779 case EOpExp2: outputTriplet(visit, "exp2(", NULL, ")"); break;
780 case EOpLog2: outputTriplet(visit, "log2(", NULL, ")"); break;
781 case EOpSqrt: outputTriplet(visit, "sqrt(", NULL, ")"); break;
782 case EOpInverseSqrt: outputTriplet(visit, "rsqrt(", NULL, ")"); break;
783 case EOpAbs: outputTriplet(visit, "abs(", NULL, ")"); break;
784 case EOpSign: outputTriplet(visit, "sign(", NULL, ")"); break;
785 case EOpFloor: outputTriplet(visit, "floor(", NULL, ")"); break;
786 case EOpCeil: outputTriplet(visit, "ceil(", NULL, ")"); break;
787 case EOpFract: outputTriplet(visit, "frac(", NULL, ")"); break;
788 case EOpLength: outputTriplet(visit, "length(", NULL, ")"); break;
789 case EOpNormalize: outputTriplet(visit, "normalize(", NULL, ")"); break;
daniel@transgaming.com45d03582010-03-11 19:41:29 +0000790// case EOpDPdx: outputTriplet(visit, "ddx(", NULL, ")"); break;
791// case EOpDPdy: outputTriplet(visit, "ddy(", NULL, ")"); break;
792// case EOpFwidth: outputTriplet(visit, "fwidth(", NULL, ")"); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000793 case EOpAny: outputTriplet(visit, "any(", NULL, ")"); break;
794 case EOpAll: outputTriplet(visit, "all(", NULL, ")"); break;
795 default: UNREACHABLE();
796 }
797
798 return true;
799}
800
801bool OutputHLSL::visitAggregate(Visit visit, TIntermAggregate *node)
802{
daniel@transgaming.com950f9932010-04-13 03:26:14 +0000803 EShLanguage language = mContext.language;
804 TInfoSinkBase &out = mBody;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000805
806 if (node->getOp() == EOpNull)
807 {
808 out.message(EPrefixError, "node is still EOpNull!");
809 return true;
810 }
811
812 switch (node->getOp())
813 {
814 case EOpSequence: outputTriplet(visit, NULL, ";\n", ";\n"); break;
815 case EOpDeclaration:
816 if (visit == PreVisit)
817 {
818 TIntermSequence &sequence = node->getSequence();
819 TIntermTyped *variable = sequence[0]->getAsTyped();
820 bool visit = true;
821
daniel@transgaming.comd25ab252010-03-30 03:36:26 +0000822 if (variable && (variable->getQualifier() == EvqTemporary || variable->getQualifier() == EvqGlobal))
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000823 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000824 if (!variable->getAsSymbolNode() || variable->getAsSymbolNode()->getSymbol() != "") // Variable declaration
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000825 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000826 out << typeString(variable->getType()) + " ";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000827
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000828 for (TIntermSequence::iterator sit = sequence.begin(); sit != sequence.end(); sit++)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000829 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000830 TIntermSymbol *symbol = (*sit)->getAsSymbolNode();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000831
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000832 if (symbol)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000833 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000834 symbol->traverse(this);
835
836 out << arrayString(symbol->getType());
837 }
838 else
839 {
840 (*sit)->traverse(this);
841 }
842
843 if (visit && this->inVisit)
844 {
845 if (*sit != sequence.back())
846 {
847 visit = this->visitAggregate(InVisit, node);
848 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000849 }
850 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000851
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000852 if (visit && this->postVisit)
853 {
854 this->visitAggregate(PostVisit, node);
855 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000856 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +0000857 else if (variable->getAsSymbolNode() && variable->getAsSymbolNode()->getSymbol() == "") // Type (struct) declaration
858 {
859 const TType &type = variable->getType();
860 const TTypeList &fields = *type.getStruct();
861
862 out << "struct " + type.getTypeName() + "\n"
863 "{\n";
864
865 for (unsigned int i = 0; i < fields.size(); i++)
866 {
867 const TType &field = *fields[i].type;
868
869 out << " " + typeString(field) + " " + field.getFieldName() + ";\n";
870 }
871
872 out << "};\n";
873 }
874 else UNREACHABLE();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000875 }
876
877 return false;
878 }
879 else if (visit == InVisit)
880 {
881 out << ", ";
882 }
883 break;
daniel@transgaming.comd1acd1e2010-04-13 03:25:57 +0000884 case EOpPrototype:
885 if (visit == PreVisit)
886 {
887 out << typeString(node->getType()) << " " << node->getName() << "(";
888
889 TIntermSequence &arguments = node->getSequence();
890
891 for (unsigned int i = 0; i < arguments.size(); i++)
892 {
893 TIntermSymbol *symbol = arguments[i]->getAsSymbolNode();
894
895 if (symbol)
896 {
897 out << argumentString(symbol);
898
899 if (i < arguments.size() - 1)
900 {
901 out << ", ";
902 }
903 }
904 else UNREACHABLE();
905 }
906
907 out << ");\n";
908
909 return false;
910 }
911 break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000912 case EOpComma: UNIMPLEMENTED(); /* FIXME */ out << "Comma\n"; return true;
913 case EOpFunction:
914 {
alokp@chromium.org43884872010-03-30 00:08:52 +0000915 TString name = TFunction::unmangleName(node->getName());
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000916
917 if (visit == PreVisit)
918 {
919 if (name == "main")
920 {
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000921 name = "gl_main";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000922 }
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000923
924 out << typeString(node->getType()) << " " << name << "(";
925
926 TIntermSequence &sequence = node->getSequence();
927 TIntermSequence &arguments = sequence[0]->getAsAggregate()->getSequence();
928
929 for (unsigned int i = 0; i < arguments.size(); i++)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000930 {
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000931 TIntermSymbol *symbol = arguments[i]->getAsSymbolNode();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000932
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000933 if (symbol)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000934 {
daniel@transgaming.comd1acd1e2010-04-13 03:25:57 +0000935 out << argumentString(symbol);
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000936
937 if (i < arguments.size() - 1)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000938 {
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000939 out << ", ";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000940 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000941 }
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000942 else UNREACHABLE();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000943 }
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000944
945 sequence.erase(sequence.begin());
946
947 out << ")\n"
948 "{\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000949 }
950 else if (visit == PostVisit)
951 {
daniel@transgaming.come78c0c92010-03-28 19:36:06 +0000952 out << "}\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000953 }
954 }
955 break;
956 case EOpFunctionCall:
957 {
958 if (visit == PreVisit)
959 {
alokp@chromium.org43884872010-03-30 00:08:52 +0000960 TString name = TFunction::unmangleName(node->getName());
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000961
962 if (node->isUserDefined())
963 {
964 out << name << "(";
965 }
966 else
967 {
968 if (name == "texture2D")
969 {
970 if (node->getSequence().size() == 2)
971 {
972 out << "gl_texture2D(";
973 }
974 else if (node->getSequence().size() == 3)
975 {
976 out << "gl_texture2DBias(";
977 }
978 else UNREACHABLE();
979 }
980 else if (name == "texture2DProj")
981 {
982 out << "gl_texture2DProj(";
983 }
984 else if (name == "texture2DLod")
985 {
986 out << "gl_texture2DLod(";
987 UNIMPLEMENTED(); // FIXME: Move lod to last texture coordinate component
988 }
989 else if (name == "texture2DProjLod")
990 {
991 out << "gl_texture2DProjLod(";
992 UNIMPLEMENTED(); // FIXME: Move lod to last texture coordinate component
993 }
994 else if (name == "textureCube")
995 {
996 out << "gl_textureCube("; // FIXME: Incorrect sampling location
997 }
998 else
999 {
1000 UNIMPLEMENTED(); // FIXME
1001 }
1002 }
1003 }
1004 else if (visit == InVisit)
1005 {
1006 out << ", ";
1007 }
1008 else
1009 {
1010 out << ")";
1011 }
1012 }
1013 break;
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001014 case EOpParameters: outputTriplet(visit, "(", ", ", ")\n{\n"); break;
1015 case EOpConstructFloat: outputTriplet(visit, "vec1(", NULL, ")"); break;
1016 case EOpConstructVec2: outputTriplet(visit, "vec2(", ", ", ")"); break;
1017 case EOpConstructVec3: outputTriplet(visit, "vec3(", ", ", ")"); break;
1018 case EOpConstructVec4: outputTriplet(visit, "vec4(", ", ", ")"); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001019 case EOpConstructBool: UNIMPLEMENTED(); /* FIXME */ out << "Construct bool"; break;
1020 case EOpConstructBVec2: UNIMPLEMENTED(); /* FIXME */ out << "Construct bvec2"; break;
1021 case EOpConstructBVec3: UNIMPLEMENTED(); /* FIXME */ out << "Construct bvec3"; break;
1022 case EOpConstructBVec4: UNIMPLEMENTED(); /* FIXME */ out << "Construct bvec4"; break;
1023 case EOpConstructInt: UNIMPLEMENTED(); /* FIXME */ out << "Construct int"; break;
1024 case EOpConstructIVec2: UNIMPLEMENTED(); /* FIXME */ out << "Construct ivec2"; break;
1025 case EOpConstructIVec3: UNIMPLEMENTED(); /* FIXME */ out << "Construct ivec3"; break;
1026 case EOpConstructIVec4: UNIMPLEMENTED(); /* FIXME */ out << "Construct ivec4"; break;
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001027 case EOpConstructMat2: outputTriplet(visit, "float2x2(", ", ", ")"); break;
1028 case EOpConstructMat3: outputTriplet(visit, "float3x3(", ", ", ")"); break;
1029 case EOpConstructMat4: outputTriplet(visit, "float4x4(", ", ", ")"); break;
1030 case EOpConstructStruct: outputTriplet(visit, "{", ", ", "}"); break;
1031 case EOpLessThan: outputTriplet(visit, "(", " < ", ")"); break;
1032 case EOpGreaterThan: outputTriplet(visit, "(", " > ", ")"); break;
1033 case EOpLessThanEqual: outputTriplet(visit, "(", " <= ", ")"); break;
1034 case EOpGreaterThanEqual: outputTriplet(visit, "(", " >= ", ")"); break;
1035 case EOpVectorEqual: outputTriplet(visit, "(", " == ", ")"); break;
1036 case EOpVectorNotEqual: outputTriplet(visit, "(", " != ", ")"); break;
1037 case EOpMod: outputTriplet(visit, "mod(", ", ", ")"); break; // FIXME: Prevent name clashes
1038 case EOpPow: outputTriplet(visit, "pow(", ", ", ")"); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001039 case EOpAtan:
1040 if (node->getSequence().size() == 1)
1041 {
1042 outputTriplet(visit, "atan(", ", ", ")");
1043 }
1044 else if (node->getSequence().size() == 2)
1045 {
1046 outputTriplet(visit, "atan2(", ", ", ")");
1047 }
1048 else UNREACHABLE();
1049 break;
1050 case EOpMin: outputTriplet(visit, "min(", ", ", ")"); break;
1051 case EOpMax: outputTriplet(visit, "max(", ", ", ")"); break;
1052 case EOpClamp: outputTriplet(visit, "clamp(", ", ", ")"); break;
1053 case EOpMix: outputTriplet(visit, "lerp(", ", ", ")"); break;
1054 case EOpStep: outputTriplet(visit, "step(", ", ", ")"); break;
1055 case EOpSmoothStep: outputTriplet(visit, "smoothstep(", ", ", ")"); break;
1056 case EOpDistance: outputTriplet(visit, "distance(", ", ", ")"); break;
1057 case EOpDot: outputTriplet(visit, "dot(", ", ", ")"); break;
1058 case EOpCross: outputTriplet(visit, "cross(", ", ", ")"); break;
daniel@transgaming.com680553b2010-03-08 21:30:52 +00001059 case EOpFaceForward: outputTriplet(visit, "faceforward(", ", ", ")"); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001060 case EOpReflect: outputTriplet(visit, "reflect(", ", ", ")"); break;
1061 case EOpRefract: outputTriplet(visit, "refract(", ", ", ")"); break;
1062 case EOpMul: outputTriplet(visit, "(", " * ", ")"); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001063 default: UNREACHABLE();
1064 }
1065
1066 return true;
1067}
1068
1069bool OutputHLSL::visitSelection(Visit visit, TIntermSelection *node)
1070{
daniel@transgaming.com950f9932010-04-13 03:26:14 +00001071 TInfoSinkBase &out = mBody;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001072
alokp@chromium.org60fe4072010-03-29 20:58:29 +00001073 if (node->usesTernaryOperator())
1074 {
1075 out << "(";
1076 node->getCondition()->traverse(this);
1077 out << ") ? (";
1078 node->getTrueBlock()->traverse(this);
1079 out << ") : (";
1080 node->getFalseBlock()->traverse(this);
1081 out << ")\n";
1082 }
1083 else // if/else statement
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001084 {
daniel@transgaming.com3d53fda2010-03-21 04:30:55 +00001085 out << "if(";
1086
1087 node->getCondition()->traverse(this);
1088
1089 out << ")\n"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001090 "{\n";
1091
daniel@transgaming.com3d53fda2010-03-21 04:30:55 +00001092 node->getTrueBlock()->traverse(this);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001093
1094 out << ";}\n";
daniel@transgaming.com3d53fda2010-03-21 04:30:55 +00001095
1096 if (node->getFalseBlock())
1097 {
1098 out << "else\n"
1099 "{\n";
1100
1101 node->getFalseBlock()->traverse(this);
1102
1103 out << ";}\n";
1104 }
1105 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001106
1107 return false;
1108}
1109
1110void OutputHLSL::visitConstantUnion(TIntermConstantUnion *node)
1111{
daniel@transgaming.com950f9932010-04-13 03:26:14 +00001112 TInfoSinkBase &out = mBody;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001113
alokp@chromium.orgdd037b22010-03-30 18:47:20 +00001114 const TType &type = node->getType();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001115
daniel@transgaming.com86487c22010-03-11 19:41:43 +00001116 if (type.isField())
1117 {
1118 out << type.getFieldName();
1119 }
1120 else
1121 {
1122 int size = type.getObjectSize();
daniel@transgaming.com86487c22010-03-11 19:41:43 +00001123
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001124 if (type.getBasicType() == EbtStruct)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001125 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001126 out << "{";
1127 }
1128 else
1129 {
1130 bool matrix = type.isMatrix();
1131 TBasicType elementType = node->getUnionArrayPointer()[0].getType();
1132
1133 switch (elementType)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001134 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001135 case EbtBool:
1136 if (!matrix)
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001137 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001138 switch (size)
1139 {
1140 case 1: out << "bool("; break;
1141 case 2: out << "bool2("; break;
1142 case 3: out << "bool3("; break;
1143 case 4: out << "bool4("; break;
1144 default: UNREACHABLE();
1145 }
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001146 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001147 else
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001148 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001149 UNIMPLEMENTED();
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001150 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001151 break;
1152 case EbtFloat:
1153 if (!matrix)
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001154 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001155 switch (size)
1156 {
1157 case 1: out << "float("; break;
1158 case 2: out << "float2("; break;
1159 case 3: out << "float3("; break;
1160 case 4: out << "float4("; break;
1161 default: UNREACHABLE();
1162 }
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001163 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001164 else
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001165 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001166 switch (size)
1167 {
1168 case 4: out << "float2x2("; break;
1169 case 9: out << "float3x3("; break;
1170 case 16: out << "float4x4("; break;
1171 default: UNREACHABLE();
1172 }
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001173 }
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001174 break;
1175 case EbtInt:
1176 if (!matrix)
1177 {
1178 switch (size)
1179 {
1180 case 1: out << "int("; break;
1181 case 2: out << "int2("; break;
1182 case 3: out << "int3("; break;
1183 case 4: out << "int4("; break;
1184 default: UNREACHABLE();
1185 }
1186 }
1187 else
1188 {
1189 UNIMPLEMENTED();
1190 }
1191 break;
1192 default:
1193 UNIMPLEMENTED(); // FIXME
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001194 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001195 }
1196
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001197 for (int i = 0; i < size; i++)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001198 {
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001199 switch (node->getUnionArrayPointer()[i].getType())
daniel@transgaming.com45d03582010-03-11 19:41:29 +00001200 {
1201 case EbtBool:
1202 if (node->getUnionArrayPointer()[i].getBConst())
1203 {
1204 out << "true";
1205 }
1206 else
1207 {
1208 out << "false";
1209 }
1210 break;
1211 case EbtFloat:
1212 out << node->getUnionArrayPointer()[i].getFConst();
1213 break;
1214 case EbtInt:
1215 out << node->getUnionArrayPointer()[i].getIConst();
1216 break;
1217 default:
1218 UNIMPLEMENTED(); // FIXME
1219 }
1220
1221 if (i != size - 1)
1222 {
1223 out << ", ";
1224 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001225 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001226
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001227 if (type.getBasicType() == EbtStruct)
1228 {
1229 out << "}";
1230 }
1231 else
1232 {
1233 out << ")";
1234 }
daniel@transgaming.com86487c22010-03-11 19:41:43 +00001235 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001236}
1237
1238bool OutputHLSL::visitLoop(Visit visit, TIntermLoop *node)
1239{
daniel@transgaming.com4a35ef22010-04-08 03:51:06 +00001240 if (handleExcessiveLoop(node))
1241 {
1242 return false;
1243 }
1244
daniel@transgaming.com950f9932010-04-13 03:26:14 +00001245 TInfoSinkBase &out = mBody;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001246
1247 if (!node->testFirst())
1248 {
1249 out << "do\n"
1250 "{\n";
1251 }
1252 else
1253 {
1254 out << "for(";
1255
1256 if (node->getInit())
1257 {
1258 node->getInit()->traverse(this);
1259 }
1260
1261 out << "; ";
1262
1263 if (node->getTest())
1264 {
1265 node->getTest()->traverse(this);
1266 }
1267
1268 out << "; ";
1269
1270 if (node->getTerminal())
1271 {
1272 node->getTerminal()->traverse(this);
1273 }
1274
1275 out << ")\n"
1276 "{\n";
1277 }
1278
1279 if (node->getBody())
1280 {
1281 node->getBody()->traverse(this);
1282 }
1283
1284 out << "}\n";
1285
1286 if (!node->testFirst())
1287 {
1288 out << "while(\n";
1289
1290 node->getTest()->traverse(this);
1291
1292 out << ")";
1293 }
1294
1295 out << ";\n";
1296
1297 return false;
1298}
1299
1300bool OutputHLSL::visitBranch(Visit visit, TIntermBranch *node)
1301{
daniel@transgaming.com950f9932010-04-13 03:26:14 +00001302 TInfoSinkBase &out = mBody;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001303
1304 switch (node->getFlowOp())
1305 {
daniel@transgaming.comf67f82e2010-03-17 03:58:54 +00001306 case EOpKill: outputTriplet(visit, "discard", NULL, NULL); break;
1307 case EOpBreak: outputTriplet(visit, "break", NULL, NULL); break;
1308 case EOpContinue: outputTriplet(visit, "continue", NULL, NULL); break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001309 case EOpReturn:
1310 if (visit == PreVisit)
1311 {
1312 if (node->getExpression())
1313 {
1314 out << "return ";
1315 }
1316 else
1317 {
1318 out << "return;\n";
1319 }
1320 }
1321 else if (visit == PostVisit)
1322 {
1323 out << ";\n";
1324 }
1325 break;
1326 default: UNREACHABLE();
1327 }
1328
1329 return true;
1330}
1331
daniel@transgaming.com4a35ef22010-04-08 03:51:06 +00001332// Handle loops with more than 255 iterations (unsupported by D3D9) by splitting them
1333bool OutputHLSL::handleExcessiveLoop(TIntermLoop *node)
1334{
daniel@transgaming.com950f9932010-04-13 03:26:14 +00001335 TInfoSinkBase &out = mBody;
daniel@transgaming.com4a35ef22010-04-08 03:51:06 +00001336
1337 // Parse loops of the form:
1338 // for(int index = initial; index [comparator] limit; index += increment)
1339 TIntermSymbol *index = NULL;
1340 TOperator comparator = EOpNull;
1341 int initial = 0;
1342 int limit = 0;
1343 int increment = 0;
1344
1345 // Parse index name and intial value
1346 if (node->getInit())
1347 {
1348 TIntermAggregate *init = node->getInit()->getAsAggregate();
1349
1350 if (init)
1351 {
1352 TIntermSequence &sequence = init->getSequence();
1353 TIntermTyped *variable = sequence[0]->getAsTyped();
1354
1355 if (variable && variable->getQualifier() == EvqTemporary)
1356 {
1357 TIntermBinary *assign = variable->getAsBinaryNode();
1358
1359 if (assign->getOp() == EOpInitialize)
1360 {
1361 TIntermSymbol *symbol = assign->getLeft()->getAsSymbolNode();
1362 TIntermConstantUnion *constant = assign->getRight()->getAsConstantUnion();
1363
1364 if (symbol && constant)
1365 {
1366 if (constant->getBasicType() == EbtInt && constant->getSize() == 1)
1367 {
1368 index = symbol;
1369 initial = constant->getUnionArrayPointer()[0].getIConst();
1370 }
1371 }
1372 }
1373 }
1374 }
1375 }
1376
1377 // Parse comparator and limit value
1378 if (index != NULL && node->getTest())
1379 {
1380 TIntermBinary *test = node->getTest()->getAsBinaryNode();
1381
1382 if (test && test->getLeft()->getAsSymbolNode()->getId() == index->getId())
1383 {
1384 TIntermConstantUnion *constant = test->getRight()->getAsConstantUnion();
1385
1386 if (constant)
1387 {
1388 if (constant->getBasicType() == EbtInt && constant->getSize() == 1)
1389 {
1390 comparator = test->getOp();
1391 limit = constant->getUnionArrayPointer()[0].getIConst();
1392 }
1393 }
1394 }
1395 }
1396
1397 // Parse increment
1398 if (index != NULL && comparator != EOpNull && node->getTerminal())
1399 {
1400 TIntermBinary *binaryTerminal = node->getTerminal()->getAsBinaryNode();
1401 TIntermUnary *unaryTerminal = node->getTerminal()->getAsUnaryNode();
1402
1403 if (binaryTerminal)
1404 {
1405 TOperator op = binaryTerminal->getOp();
1406 TIntermConstantUnion *constant = binaryTerminal->getRight()->getAsConstantUnion();
1407
1408 if (constant)
1409 {
1410 if (constant->getBasicType() == EbtInt && constant->getSize() == 1)
1411 {
1412 int value = constant->getUnionArrayPointer()[0].getIConst();
1413
1414 switch (op)
1415 {
1416 case EOpAddAssign: increment = value; break;
1417 case EOpSubAssign: increment = -value; break;
1418 default: UNIMPLEMENTED();
1419 }
1420 }
1421 }
1422 }
1423 else if (unaryTerminal)
1424 {
1425 TOperator op = unaryTerminal->getOp();
1426
1427 switch (op)
1428 {
1429 case EOpPostIncrement: increment = 1; break;
1430 case EOpPostDecrement: increment = -1; break;
1431 case EOpPreIncrement: increment = 1; break;
1432 case EOpPreDecrement: increment = -1; break;
1433 default: UNIMPLEMENTED();
1434 }
1435 }
1436 }
1437
1438 if (index != NULL && comparator != EOpNull && increment != 0)
1439 {
1440 if (comparator == EOpLessThanEqual)
1441 {
1442 comparator = EOpLessThan;
1443 limit += 1;
1444 }
1445
1446 if (comparator == EOpLessThan)
1447 {
1448 int iterations = (limit - initial + 1) / increment;
1449
1450 if (iterations <= 255)
1451 {
1452 return false; // Not an excessive loop
1453 }
1454
1455 while (iterations > 0)
1456 {
1457 int remainder = (limit - initial + 1) % increment;
1458 int clampedLimit = initial + increment * min(255, iterations) - 1 - remainder;
1459
1460 // for(int index = initial; index < clampedLimit; index += increment)
1461
1462 out << "for(int ";
1463 index->traverse(this);
1464 out << " = ";
1465 out << initial;
1466
1467 out << "; ";
1468 index->traverse(this);
1469 out << " < ";
1470 out << clampedLimit;
1471
1472 out << "; ";
1473 index->traverse(this);
1474 out << " += ";
1475 out << increment;
1476 out << ")\n"
1477 "{\n";
1478
1479 if (node->getBody())
1480 {
1481 node->getBody()->traverse(this);
1482 }
1483
1484 out << "}\n";
1485
1486 initial += 255 * increment;
1487 iterations -= 255;
1488 }
1489
1490 return true;
1491 }
1492 else UNIMPLEMENTED();
1493 }
1494
1495 return false; // Not handled as an excessive loop
1496}
1497
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001498void OutputHLSL::outputTriplet(Visit visit, const char *preString, const char *inString, const char *postString)
1499{
daniel@transgaming.com950f9932010-04-13 03:26:14 +00001500 TInfoSinkBase &out = mBody;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001501
1502 if (visit == PreVisit && preString)
1503 {
1504 out << preString;
1505 }
1506 else if (visit == InVisit && inString)
1507 {
1508 out << inString;
1509 }
1510 else if (visit == PostVisit && postString)
1511 {
1512 out << postString;
1513 }
1514}
1515
daniel@transgaming.comd1acd1e2010-04-13 03:25:57 +00001516TString OutputHLSL::argumentString(const TIntermSymbol *symbol)
1517{
1518 TQualifier qualifier = symbol->getQualifier();
1519 const TType &type = symbol->getType();
1520 const TString &name = symbol->getSymbol();
1521
1522 return qualifierString(qualifier) + " " + typeString(type) + " " + name + arrayString(type);
1523}
1524
1525TString OutputHLSL::qualifierString(TQualifier qualifier)
1526{
1527 switch(qualifier)
1528 {
1529 case EvqIn: return "in";
1530 case EvqOut: return "out";
1531 case EvqInOut: return "inout";
1532 case EvqConstReadOnly: return "const";
1533 default: UNREACHABLE();
1534 }
1535
1536 return "";
1537}
1538
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001539TString OutputHLSL::typeString(const TType &type)
1540{
daniel@transgaming.comfe565152010-04-10 05:29:07 +00001541 if (type.getBasicType() == EbtStruct)
1542 {
1543 return type.getTypeName();
1544 }
1545 else if (type.isMatrix())
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001546 {
1547 switch (type.getNominalSize())
1548 {
1549 case 2: return "float2x2";
1550 case 3: return "float3x3";
1551 case 4: return "float4x4";
1552 }
1553 }
1554 else
1555 {
1556 switch (type.getBasicType())
1557 {
1558 case EbtFloat:
1559 switch (type.getNominalSize())
1560 {
1561 case 1: return "float";
1562 case 2: return "float2";
1563 case 3: return "float3";
1564 case 4: return "float4";
1565 }
1566 case EbtInt:
1567 switch (type.getNominalSize())
1568 {
1569 case 1: return "int";
1570 case 2: return "int2";
1571 case 3: return "int3";
1572 case 4: return "int4";
1573 }
1574 case EbtBool:
1575 switch (type.getNominalSize())
1576 {
1577 case 1: return "bool";
1578 case 2: return "bool2";
1579 case 3: return "bool3";
1580 case 4: return "bool4";
1581 }
1582 case EbtVoid:
1583 return "void";
1584 case EbtSampler2D:
1585 return "sampler2D";
1586 case EbtSamplerCube:
1587 return "samplerCUBE";
1588 }
1589 }
1590
1591 UNIMPLEMENTED(); // FIXME
1592 return "<unknown type>";
1593}
1594
1595TString OutputHLSL::arrayString(const TType &type)
1596{
1597 if (!type.isArray())
1598 {
1599 return "";
1600 }
1601
1602 char buffer[100];
1603 sprintf(buffer, "[%d]", type.getArraySize());
1604
1605 return buffer;
1606}
1607
1608TString OutputHLSL::initializer(const TType &type)
1609{
1610 TString string;
1611
1612 int arraySize = type.isArray() ? type.getArraySize() : 1;
1613
1614 if (type.isArray())
1615 {
1616 string += "{";
1617 }
1618
1619 for (int element = 0; element < arraySize; element++)
1620 {
1621 string += typeString(type) + "(";
1622
1623 for (int component = 0; component < type.getNominalSize(); component++)
1624 {
1625 string += "0";
1626
1627 if (component < type.getNominalSize() - 1)
1628 {
1629 string += ", ";
1630 }
1631 }
1632
1633 string += ")";
1634
1635 if (element < arraySize - 1)
1636 {
1637 string += ", ";
1638 }
1639 }
1640
1641 if (type.isArray())
1642 {
1643 string += "}";
1644 }
1645
1646 return string;
1647}
1648}