OSDN Git Service

Fixed infinite loop crash
[android-x86/external-swiftshader.git] / src / OpenGL / compiler / OutputASM.cpp
1 // SwiftShader Software Renderer\r
2 //\r
3 // Copyright(c) 2005-2013 TransGaming Inc.\r
4 //\r
5 // All rights reserved. No part of this software may be copied, distributed, transmitted,\r
6 // transcribed, stored in a retrieval system, translated into any human or computer\r
7 // language by any means, or disclosed to third parties without the explicit written\r
8 // agreement of TransGaming Inc. Without such an agreement, no rights or licenses, express\r
9 // or implied, including but not limited to any patent rights, are granted to you.\r
10 //\r
11 \r
12 #include "OutputASM.h"\r
13 \r
14 #include "common/debug.h"\r
15 #include "InfoSink.h"\r
16 \r
17 #include "libGLESv2/Shader.h"\r
18 \r
19 #include <GLES2/gl2.h>\r
20 #include <GLES2/gl2ext.h>\r
21 #include <GLES3/gl3.h>\r
22 \r
23 namespace glsl\r
24 {\r
25         // Integer to TString conversion\r
26         TString str(int i)\r
27         {\r
28                 char buffer[20];\r
29                 sprintf(buffer, "%d", i);\r
30                 return buffer;\r
31         }\r
32 \r
33         class Temporary : public TIntermSymbol\r
34         {\r
35         public:\r
36                 Temporary(OutputASM *assembler) : TIntermSymbol(0, "tmp", TType(EbtFloat, EbpHigh, EvqTemporary, 4, 1, false)), assembler(assembler)\r
37                 {\r
38                 }\r
39 \r
40                 ~Temporary()\r
41                 {\r
42                         assembler->freeTemporary(this);\r
43                 }\r
44 \r
45         private:\r
46                 OutputASM *const assembler;\r
47         };\r
48 \r
49         class Constant : public TIntermConstantUnion\r
50         {\r
51         public:\r
52                 Constant(float x, float y, float z, float w) : TIntermConstantUnion(constants, TType(EbtFloat, EbpHigh, EvqConstExpr, 4, 1, false))\r
53                 {\r
54                         constants[0].setFConst(x);\r
55                         constants[1].setFConst(y);\r
56                         constants[2].setFConst(z);\r
57                         constants[3].setFConst(w);\r
58                 }\r
59 \r
60                 Constant(bool b) : TIntermConstantUnion(constants, TType(EbtBool, EbpHigh, EvqConstExpr, 1, 1, false))\r
61                 {\r
62                         constants[0].setBConst(b);\r
63                 }\r
64 \r
65                 Constant(int i) : TIntermConstantUnion(constants, TType(EbtInt, EbpHigh, EvqConstExpr, 1, 1, false))\r
66                 {\r
67                         constants[0].setIConst(i);\r
68                 }\r
69 \r
70                 ~Constant()\r
71                 {\r
72                 }\r
73 \r
74         private:\r
75                 ConstantUnion constants[4];\r
76         };\r
77 \r
78         Uniform::Uniform(GLenum type, GLenum precision, const std::string &name, int arraySize, int registerIndex)\r
79         {\r
80                 this->type = type;\r
81                 this->precision = precision;\r
82                 this->name = name;\r
83                 this->arraySize = arraySize;\r
84                 this->registerIndex = registerIndex;\r
85         }\r
86 \r
87         Attribute::Attribute()\r
88         {\r
89                 type = GL_NONE;\r
90                 arraySize = 0;\r
91                 registerIndex = 0;\r
92         }\r
93 \r
94         Attribute::Attribute(GLenum type, const std::string &name, int arraySize, int registerIndex)\r
95         {\r
96                 this->type = type;\r
97                 this->name = name;\r
98                 this->arraySize = arraySize;\r
99                 this->registerIndex = registerIndex;\r
100         }\r
101 \r
102         sw::PixelShader *Shader::getPixelShader() const\r
103         {\r
104                 return 0;\r
105         }\r
106 \r
107         sw::VertexShader *Shader::getVertexShader() const\r
108         {\r
109                 return 0;\r
110         }\r
111 \r
112         OutputASM::OutputASM(TParseContext &context, Shader *shaderObject) : TIntermTraverser(true, true, true), mContext(context), shaderObject(shaderObject)\r
113         {\r
114                 shader = 0;\r
115                 pixelShader = 0;\r
116                 vertexShader = 0;\r
117 \r
118                 if(shaderObject)\r
119                 {\r
120                         shader = shaderObject->getShader();\r
121                         pixelShader = shaderObject->getPixelShader();\r
122                         vertexShader = shaderObject->getVertexShader();\r
123                 }\r
124 \r
125                 functionArray.push_back(Function(0, "main(", 0, 0));\r
126                 currentFunction = 0;\r
127                 outputQualifier = EvqOutput; // Set outputQualifier to any value other than EvqFragColor or EvqFragData\r
128         }\r
129 \r
130         OutputASM::~OutputASM()\r
131         {\r
132         }\r
133 \r
134         void OutputASM::output()\r
135         {\r
136                 if(shader)\r
137                 {\r
138                         emitShader(GLOBAL);\r
139 \r
140                         if(functionArray.size() > 1)   // Only call main() when there are other functions\r
141                         {\r
142                                 Instruction *callMain = emit(sw::Shader::OPCODE_CALL);\r
143                                 callMain->dst.type = sw::Shader::PARAMETER_LABEL;\r
144                                 callMain->dst.index = 0;   // main()\r
145 \r
146                                 emit(sw::Shader::OPCODE_RET);\r
147                         }\r
148 \r
149                         emitShader(FUNCTION);\r
150                 }\r
151         }\r
152 \r
153         void OutputASM::emitShader(Scope scope)\r
154         {\r
155                 emitScope = scope;\r
156                 currentScope = GLOBAL;\r
157                 mContext.treeRoot->traverse(this);\r
158         }\r
159 \r
160         void OutputASM::freeTemporary(Temporary *temporary)\r
161         {\r
162                 free(temporaries, temporary);\r
163         }\r
164 \r
165         void OutputASM::visitSymbol(TIntermSymbol *symbol)\r
166         {\r
167                 // Vertex varyings don't have to be actively used to successfully link\r
168                 // against pixel shaders that use them. So make sure they're declared.\r
169                 if(symbol->getQualifier() == EvqVaryingOut || symbol->getQualifier() == EvqInvariantVaryingOut || symbol->getQualifier() == EvqVertexOut)\r
170                 {\r
171                         if(symbol->getBasicType() != EbtInvariant)   // Typeless declarations are not new varyings\r
172                         {\r
173                                 declareVarying(symbol, -1);\r
174                         }\r
175                 }\r
176         }\r
177 \r
178         bool OutputASM::visitBinary(Visit visit, TIntermBinary *node)\r
179         {\r
180                 if(currentScope != emitScope)\r
181                 {\r
182                         return false;\r
183                 }\r
184 \r
185                 TIntermTyped *result = node;\r
186                 TIntermTyped *left = node->getLeft();\r
187                 TIntermTyped *right = node->getRight();\r
188                 const TType &leftType = left->getType();\r
189                 const TType &rightType = right->getType();\r
190                 const TType &resultType = node->getType();\r
191                 \r
192                 switch(node->getOp())\r
193                 {\r
194                 case EOpAssign:\r
195                         if(visit == PostVisit)\r
196                         {\r
197                                 assignLvalue(left, right);\r
198                                 copy(result, right);\r
199                         }\r
200                         break;\r
201                 case EOpInitialize:\r
202                         if(visit == PostVisit)\r
203                         {\r
204                                 copy(left, right);\r
205                         }\r
206                         break;\r
207                 case EOpMatrixTimesScalarAssign:\r
208                         if(visit == PostVisit)\r
209                         {\r
210                                 for(int i = 0; i < leftType.getNominalSize(); i++)\r
211                                 {\r
212                                         Instruction *mul = emit(sw::Shader::OPCODE_MUL, result, left, right);\r
213                                         mul->dst.index += i;\r
214                                         argument(mul->src[0], left, i);\r
215                                 }\r
216 \r
217                                 assignLvalue(left, result);\r
218                         }\r
219                         break;\r
220                 case EOpVectorTimesMatrixAssign:\r
221                         if(visit == PostVisit)\r
222                         {\r
223                                 int size = leftType.getNominalSize();\r
224 \r
225                                 for(int i = 0; i < size; i++)\r
226                                 {\r
227                                         Instruction *dot = emit(sw::Shader::OPCODE_DP(size), result, left, right);\r
228                                         dot->dst.mask = 1 << i;\r
229                                         argument(dot->src[1], right, i);\r
230                                 }\r
231 \r
232                                 assignLvalue(left, result);\r
233                         }\r
234                         break;\r
235                 case EOpMatrixTimesMatrixAssign:\r
236                         if(visit == PostVisit)\r
237                         {\r
238                                 int dim = leftType.getNominalSize();\r
239 \r
240                                 for(int i = 0; i < dim; i++)\r
241                                 {\r
242                                         Instruction *mul = emit(sw::Shader::OPCODE_MUL, result, left, right);\r
243                                         mul->dst.index += i;\r
244                                         argument(mul->src[1], right, i);\r
245                                         mul->src[1].swizzle = 0x00;\r
246 \r
247                                         for(int j = 1; j < dim; j++)\r
248                                         {\r
249                                                 Instruction *mad = emit(sw::Shader::OPCODE_MAD, result, left, right, result);\r
250                                                 mad->dst.index += i;\r
251                                                 argument(mad->src[0], left, j);\r
252                                                 argument(mad->src[1], right, i);\r
253                                                 mad->src[1].swizzle = j * 0x55;\r
254                                                 argument(mad->src[2], result, i);\r
255                                         }\r
256                                 }\r
257 \r
258                                 assignLvalue(left, result);\r
259                         }\r
260                         break;\r
261                 case EOpIndexDirect:\r
262                         if(visit == PostVisit)\r
263                         {\r
264                                 int index = right->getAsConstantUnion()->getIConst(0);\r
265 \r
266                                 if(result->isMatrix() || result->isStruct())\r
267                                 {\r
268                                         ASSERT(left->isArray());\r
269                                         copy(result, left, index * left->elementRegisterCount());\r
270                                 }\r
271                                 else if(result->isRegister())\r
272                                 {\r
273                                         Instruction *mov = emit(sw::Shader::OPCODE_MOV, result, left);\r
274 \r
275                                         if(left->isRegister())\r
276                                         {\r
277                                                 mov->src[0].swizzle = index;\r
278                                         }\r
279                                         else if(left->isArray())\r
280                                         {\r
281                                                 argument(mov->src[0], left, index * left->elementRegisterCount());\r
282                                         }\r
283                                         else if(left->isMatrix())\r
284                                         {\r
285                                                 ASSERT(index < left->getNominalSize());   // FIXME: Report semantic error\r
286                                                 argument(mov->src[0], left, index);\r
287                                         }\r
288                                         else UNREACHABLE(0);\r
289                                 }\r
290                                 else UNREACHABLE(0);\r
291                         }\r
292                         break;\r
293                 case EOpIndexIndirect:\r
294                         if(visit == PostVisit)\r
295                         {\r
296                                 if(left->isArray() || left->isMatrix())\r
297                                 {\r
298                                         for(int index = 0; index < result->totalRegisterCount(); index++)\r
299                                         {\r
300                                                 Instruction *mov = emit(sw::Shader::OPCODE_MOV, result, left);\r
301                                                 mov->dst.index += index;\r
302                                                 mov->dst.mask = writeMask(result, index);\r
303                                                 argument(mov->src[0], left, index);\r
304 \r
305                                                 if(left->totalRegisterCount() > 1)\r
306                                                 {\r
307                                                         sw::Shader::SourceParameter relativeRegister;\r
308                                                         argument(relativeRegister, right);\r
309 \r
310                                                         mov->src[0].rel.type = relativeRegister.type;\r
311                                                         mov->src[0].rel.index = relativeRegister.index;\r
312                                                         mov->src[0].rel.scale = result->totalRegisterCount();\r
313                                                         mov->src[0].rel.deterministic = !(vertexShader && left->getQualifier() == EvqUniform);\r
314                                                 }\r
315                                         }\r
316                                 }\r
317                                 else if(left->isRegister())\r
318                                 {\r
319                                         emit(sw::Shader::OPCODE_EXTRACT, result, left, right);\r
320                                 }\r
321                                 else UNREACHABLE(0);\r
322                         }\r
323                         break;\r
324                 case EOpIndexDirectStruct:\r
325                         if(visit == PostVisit)\r
326                         {\r
327                                 ASSERT(leftType.isStruct());\r
328 \r
329                                 const TFieldList& fields = leftType.getStruct()->fields();\r
330                                 int index = right->getAsConstantUnion()->getIConst(0);\r
331                                 int fieldOffset = 0;\r
332 \r
333                                 for(int i = 0; i < index; i++)\r
334                                 {\r
335                                         fieldOffset += fields[i]->type()->totalRegisterCount();\r
336                                 }\r
337 \r
338                                 copy(result, left, fieldOffset);\r
339                         }\r
340                         break;\r
341                 case EOpVectorSwizzle:\r
342                         if(visit == PostVisit)\r
343                         {\r
344                                 int swizzle = 0;\r
345                                 TIntermAggregate *components = right->getAsAggregate();\r
346 \r
347                                 if(components)\r
348                                 {\r
349                                         TIntermSequence &sequence = components->getSequence();\r
350                                         int component = 0;\r
351 \r
352                                         for(TIntermSequence::iterator sit = sequence.begin(); sit != sequence.end(); sit++)\r
353                                         {\r
354                                                 TIntermConstantUnion *element = (*sit)->getAsConstantUnion();\r
355 \r
356                                                 if(element)\r
357                                                 {\r
358                                                         int i = element->getUnionArrayPointer()[0].getIConst();\r
359                                                         swizzle |= i << (component * 2);\r
360                                                         component++;\r
361                                                 }\r
362                                                 else UNREACHABLE(0);\r
363                                         }\r
364                                 }\r
365                                 else UNREACHABLE(0);\r
366 \r
367                                 Instruction *mov = emit(sw::Shader::OPCODE_MOV, result, left);\r
368                                 mov->src[0].swizzle = swizzle;\r
369                         }\r
370                         break;\r
371                 case EOpAddAssign: if(visit == PostVisit) emitAssign(sw::Shader::OPCODE_ADD, result, left, left, right); break;\r
372                 case EOpAdd:       if(visit == PostVisit) emitBinary(sw::Shader::OPCODE_ADD, result, left, right);       break;\r
373                 case EOpSubAssign: if(visit == PostVisit) emitAssign(sw::Shader::OPCODE_SUB, result, left, left, right); break;\r
374                 case EOpSub:       if(visit == PostVisit) emitBinary(sw::Shader::OPCODE_SUB, result, left, right);       break;\r
375                 case EOpMulAssign: if(visit == PostVisit) emitAssign(sw::Shader::OPCODE_MUL, result, left, left, right); break;\r
376                 case EOpMul:       if(visit == PostVisit) emitBinary(sw::Shader::OPCODE_MUL, result, left, right);       break;\r
377                 case EOpDivAssign: if(visit == PostVisit) emitAssign(sw::Shader::OPCODE_DIV, result, left, left, right); break;\r
378                 case EOpDiv:       if(visit == PostVisit) emitBinary(sw::Shader::OPCODE_DIV, result, left, right);       break;\r
379                 case EOpEqual:\r
380                         if(visit == PostVisit)\r
381                         {\r
382                                 emitCmp(sw::Shader::CONTROL_EQ, result, left, right);\r
383 \r
384                                 for(int index = 1; index < left->totalRegisterCount(); index++)\r
385                                 {\r
386                                         Temporary equal(this);\r
387                                         emitCmp(sw::Shader::CONTROL_EQ, &equal, left, right, index);\r
388                                         emit(sw::Shader::OPCODE_AND, result, result, &equal);\r
389                                 }\r
390                         }\r
391                         break;\r
392                 case EOpNotEqual:\r
393                         if(visit == PostVisit)\r
394                         {\r
395                                 emitCmp(sw::Shader::CONTROL_NE, result, left, right);\r
396 \r
397                                 for(int index = 1; index < left->totalRegisterCount(); index++)\r
398                                 {\r
399                                         Temporary notEqual(this);\r
400                                         emitCmp(sw::Shader::CONTROL_NE, &notEqual, left, right, index);\r
401                                         emit(sw::Shader::OPCODE_OR, result, result, &notEqual);\r
402                                 }\r
403                         }\r
404                         break;\r
405                 case EOpLessThan:                if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_LT, result, left, right); break;\r
406                 case EOpGreaterThan:             if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_GT, result, left, right); break;\r
407                 case EOpLessThanEqual:           if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_LE, result, left, right); break;\r
408                 case EOpGreaterThanEqual:        if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_GE, result, left, right); break;\r
409                 case EOpVectorTimesScalarAssign: if(visit == PostVisit) emitAssign(sw::Shader::OPCODE_MUL, result, left, left, right); break;\r
410                 case EOpVectorTimesScalar:       if(visit == PostVisit) emit(sw::Shader::OPCODE_MUL, result, left, right); break;\r
411                 case EOpMatrixTimesScalar:\r
412                         if(visit == PostVisit)\r
413                         {\r
414                                 for(int i = 0; i < leftType.getNominalSize(); i++)\r
415                                 {\r
416                                         Instruction *mul = emit(sw::Shader::OPCODE_MUL, result, left, right);\r
417                                         mul->dst.index += i;\r
418                                         argument(mul->src[0], left, i);\r
419                                 }\r
420                         }\r
421                         break;\r
422                 case EOpVectorTimesMatrix:\r
423                         if(visit == PostVisit)\r
424                         {\r
425                                 int size = leftType.getNominalSize();\r
426 \r
427                                 for(int i = 0; i < size; i++)\r
428                                 {\r
429                                         Instruction *dot = emit(sw::Shader::OPCODE_DP(size), result, left, right);\r
430                                         dot->dst.mask = 1 << i;\r
431                                         argument(dot->src[1], right, i);\r
432                                 }\r
433                         }\r
434                         break;\r
435                 case EOpMatrixTimesVector:\r
436                         if(visit == PostVisit)\r
437                         {\r
438                                 Instruction *mul = emit(sw::Shader::OPCODE_MUL, result, left, right);\r
439                                 mul->src[1].swizzle = 0x00;\r
440 \r
441                                 for(int i = 1; i < leftType.getNominalSize(); i++)\r
442                                 {\r
443                                         Instruction *mad = emit(sw::Shader::OPCODE_MAD, result, left, right, result);\r
444                                         argument(mad->src[0], left, i);\r
445                                         mad->src[1].swizzle = i * 0x55;\r
446                                 }\r
447                         }\r
448                         break;\r
449                 case EOpMatrixTimesMatrix:\r
450                         if(visit == PostVisit)\r
451                         {\r
452                                 int dim = leftType.getNominalSize();\r
453 \r
454                                 for(int i = 0; i < dim; i++)\r
455                                 {\r
456                                         Instruction *mul = emit(sw::Shader::OPCODE_MUL, result, left, right);\r
457                                         mul->dst.index += i;\r
458                                         argument(mul->src[1], right, i);\r
459                                         mul->src[1].swizzle = 0x00;\r
460 \r
461                                         for(int j = 1; j < dim; j++)\r
462                                         {\r
463                                                 Instruction *mad = emit(sw::Shader::OPCODE_MAD, result, left, right, result);\r
464                                                 mad->dst.index += i;\r
465                                                 argument(mad->src[0], left, j);\r
466                                                 argument(mad->src[1], right, i);\r
467                                                 mad->src[1].swizzle = j * 0x55;\r
468                                                 argument(mad->src[2], result, i);\r
469                                         }\r
470                                 }\r
471                         }\r
472                         break;\r
473                 case EOpLogicalOr:\r
474                         if(trivial(right, 6))\r
475                         {\r
476                                 if(visit == PostVisit)\r
477                                 {\r
478                                         emit(sw::Shader::OPCODE_OR, result, left, right);\r
479                                 }\r
480                         }\r
481                         else   // Short-circuit evaluation\r
482                         {\r
483                                 if(visit == InVisit)\r
484                                 {\r
485                                         emit(sw::Shader::OPCODE_MOV, result, left);\r
486                                         Instruction *ifnot = emit(sw::Shader::OPCODE_IF, 0, result);\r
487                                         ifnot->src[0].modifier = sw::Shader::MODIFIER_NOT;\r
488                                 }\r
489                                 else if(visit == PostVisit)\r
490                                 {\r
491                                         emit(sw::Shader::OPCODE_MOV, result, right);\r
492                                         emit(sw::Shader::OPCODE_ENDIF);\r
493                                 }\r
494                         }\r
495                         break;\r
496                 case EOpLogicalXor:        if(visit == PostVisit) emit(sw::Shader::OPCODE_XOR, result, left, right); break;\r
497                 case EOpLogicalAnd:\r
498                         if(trivial(right, 6))\r
499                         {\r
500                                 if(visit == PostVisit)\r
501                                 {\r
502                                         emit(sw::Shader::OPCODE_AND, result, left, right);\r
503                                 }\r
504                         }\r
505                         else   // Short-circuit evaluation\r
506                         {\r
507                                 if(visit == InVisit)\r
508                                 {\r
509                                         emit(sw::Shader::OPCODE_MOV, result, left);\r
510                                         emit(sw::Shader::OPCODE_IF, 0, result);\r
511                                 }\r
512                                 else if(visit == PostVisit)\r
513                                 {\r
514                                         emit(sw::Shader::OPCODE_MOV, result, right);\r
515                                         emit(sw::Shader::OPCODE_ENDIF);\r
516                                 }\r
517                         }\r
518                         break;\r
519                 default: UNREACHABLE(node->getOp());\r
520                 }\r
521 \r
522                 return true;\r
523         }\r
524 \r
525         bool OutputASM::visitUnary(Visit visit, TIntermUnary *node)\r
526         {\r
527                 if(currentScope != emitScope)\r
528                 {\r
529                         return false;\r
530                 }\r
531 \r
532                 Constant one(1.0f, 1.0f, 1.0f, 1.0f);\r
533                 Constant rad(1.74532925e-2f, 1.74532925e-2f, 1.74532925e-2f, 1.74532925e-2f);\r
534                 Constant deg(5.72957795e+1f, 5.72957795e+1f, 5.72957795e+1f, 5.72957795e+1f);\r
535 \r
536                 TIntermTyped *result = node;\r
537                 TIntermTyped *arg = node->getOperand();\r
538 \r
539                 switch(node->getOp())\r
540                 {\r
541                 case EOpNegative:\r
542                         if(visit == PostVisit)\r
543                         {\r
544                                 for(int index = 0; index < arg->totalRegisterCount(); index++)\r
545                                 {\r
546                                         Instruction *neg = emit(sw::Shader::OPCODE_MOV, result, arg);\r
547                                         neg->dst.index += index;\r
548                                         argument(neg->src[0], arg, index);\r
549                                         neg->src[0].modifier = sw::Shader::MODIFIER_NEGATE;\r
550                                 }\r
551                         }\r
552                         break;\r
553                 case EOpVectorLogicalNot: if(visit == PostVisit) emit(sw::Shader::OPCODE_NOT, result, arg); break;\r
554                 case EOpLogicalNot:       if(visit == PostVisit) emit(sw::Shader::OPCODE_NOT, result, arg); break;\r
555                 case EOpPostIncrement:\r
556                         if(visit == PostVisit)\r
557                         {\r
558                                 copy(result, arg);\r
559 \r
560                                 for(int index = 0; index < arg->totalRegisterCount(); index++)\r
561                                 {\r
562                                         Instruction *add = emit(sw::Shader::OPCODE_ADD, arg, arg, &one);\r
563                                         add->dst.index += index;\r
564                                         argument(add->src[0], arg, index);\r
565                                 }\r
566 \r
567                                 assignLvalue(arg, arg);\r
568                         }\r
569                         break;\r
570                 case EOpPostDecrement:\r
571                         if(visit == PostVisit)\r
572                         {\r
573                                 copy(result, arg);\r
574 \r
575                                 for(int index = 0; index < arg->totalRegisterCount(); index++)\r
576                                 {\r
577                                         Instruction *sub = emit(sw::Shader::OPCODE_SUB, arg, arg, &one);\r
578                                         sub->dst.index += index;\r
579                                         argument(sub->src[0], arg, index);\r
580                                 }\r
581 \r
582                                 assignLvalue(arg, arg);\r
583                         }\r
584                         break;\r
585                 case EOpPreIncrement:\r
586                         if(visit == PostVisit)\r
587                         {\r
588                                 for(int index = 0; index < arg->totalRegisterCount(); index++)\r
589                                 {\r
590                                         Instruction *add = emit(sw::Shader::OPCODE_ADD, result, arg, &one);\r
591                                         add->dst.index += index;\r
592                                         argument(add->src[0], arg, index);\r
593                                 }\r
594 \r
595                                 assignLvalue(arg, result);\r
596                         }\r
597                         break;\r
598                 case EOpPreDecrement:\r
599                         if(visit == PostVisit)\r
600                         {\r
601                                 for(int index = 0; index < arg->totalRegisterCount(); index++)\r
602                                 {\r
603                                         Instruction *sub = emit(sw::Shader::OPCODE_SUB, result, arg, &one);\r
604                                         sub->dst.index += index;\r
605                                         argument(sub->src[0], arg, index);\r
606                                 }\r
607 \r
608                                 assignLvalue(arg, result);\r
609                         }\r
610                         break;\r
611                 case EOpRadians:          if(visit == PostVisit) emit(sw::Shader::OPCODE_MUL, result, arg, &rad); break;\r
612                 case EOpDegrees:          if(visit == PostVisit) emit(sw::Shader::OPCODE_MUL, result, arg, &deg); break;\r
613                 case EOpSin:              if(visit == PostVisit) emit(sw::Shader::OPCODE_SIN, result, arg); break;\r
614                 case EOpCos:              if(visit == PostVisit) emit(sw::Shader::OPCODE_COS, result, arg); break;\r
615                 case EOpTan:              if(visit == PostVisit) emit(sw::Shader::OPCODE_TAN, result, arg); break;\r
616                 case EOpAsin:             if(visit == PostVisit) emit(sw::Shader::OPCODE_ASIN, result, arg); break;\r
617                 case EOpAcos:             if(visit == PostVisit) emit(sw::Shader::OPCODE_ACOS, result, arg); break;\r
618                 case EOpAtan:             if(visit == PostVisit) emit(sw::Shader::OPCODE_ATAN, result, arg); break;\r
619                 case EOpSinh:             if(visit == PostVisit) emit(sw::Shader::OPCODE_SINH, result, arg); break;\r
620                 case EOpCosh:             if(visit == PostVisit) emit(sw::Shader::OPCODE_COSH, result, arg); break;\r
621                 case EOpTanh:             if(visit == PostVisit) emit(sw::Shader::OPCODE_TANH, result, arg); break;\r
622                 case EOpAsinh:            if(visit == PostVisit) emit(sw::Shader::OPCODE_ASINH, result, arg); break;\r
623                 case EOpAcosh:            if(visit == PostVisit) emit(sw::Shader::OPCODE_ACOSH, result, arg); break;\r
624                 case EOpAtanh:            if(visit == PostVisit) emit(sw::Shader::OPCODE_ATANH, result, arg); break;\r
625                 case EOpExp:              if(visit == PostVisit) emit(sw::Shader::OPCODE_EXP, result, arg); break;\r
626                 case EOpLog:              if(visit == PostVisit) emit(sw::Shader::OPCODE_LOG, result, arg); break;\r
627                 case EOpExp2:             if(visit == PostVisit) emit(sw::Shader::OPCODE_EXP2, result, arg); break;\r
628                 case EOpLog2:             if(visit == PostVisit) emit(sw::Shader::OPCODE_LOG2, result, arg); break;\r
629                 case EOpSqrt:             if(visit == PostVisit) emit(sw::Shader::OPCODE_SQRT, result, arg); break;\r
630                 case EOpInverseSqrt:      if(visit == PostVisit) emit(sw::Shader::OPCODE_RSQ, result, arg); break;\r
631                 case EOpAbs:              if(visit == PostVisit) emit(sw::Shader::OPCODE_ABS, result, arg); break;\r
632                 case EOpSign:             if(visit == PostVisit) emit(sw::Shader::OPCODE_SGN, result, arg); break;\r
633                 case EOpFloor:            if(visit == PostVisit) emit(sw::Shader::OPCODE_FLOOR, result, arg); break;\r
634                 case EOpTrunc:            if(visit == PostVisit) emit(sw::Shader::OPCODE_TRUNC, result, arg); break;\r
635                 case EOpRound:            if(visit == PostVisit) emit(sw::Shader::OPCODE_ROUND, result, arg); break;\r
636                 case EOpRoundEven:        if(visit == PostVisit) emit(sw::Shader::OPCODE_ROUNDEVEN, result, arg); break;\r
637                 case EOpCeil:             if(visit == PostVisit) emit(sw::Shader::OPCODE_CEIL, result, arg, result); break;\r
638                 case EOpFract:            if(visit == PostVisit) emit(sw::Shader::OPCODE_FRC, result, arg); break;\r
639                 case EOpIsNan:            if(visit == PostVisit) emit(sw::Shader::OPCODE_ISNAN, result, arg); break;\r
640                 case EOpIsInf:            if(visit == PostVisit) emit(sw::Shader::OPCODE_ISINF, result, arg); break;\r
641                 case EOpLength:           if(visit == PostVisit) emit(sw::Shader::OPCODE_LEN(dim(arg)), result, arg); break;\r
642                 case EOpNormalize:        if(visit == PostVisit) emit(sw::Shader::OPCODE_NRM(dim(arg)), result, arg); break;\r
643                 case EOpDFdx:             if(visit == PostVisit) emit(sw::Shader::OPCODE_DFDX, result, arg); break;\r
644                 case EOpDFdy:             if(visit == PostVisit) emit(sw::Shader::OPCODE_DFDY, result, arg); break;\r
645                 case EOpFwidth:           if(visit == PostVisit) emit(sw::Shader::OPCODE_FWIDTH, result, arg); break;\r
646                 case EOpAny:              if(visit == PostVisit) emit(sw::Shader::OPCODE_ANY, result, arg); break;\r
647                 case EOpAll:              if(visit == PostVisit) emit(sw::Shader::OPCODE_ALL, result, arg); break;\r
648                 case EOpTranspose:\r
649                         if(visit == PostVisit)\r
650                         {\r
651                                 int numCols = arg->getNominalSize();\r
652                                 int numRows = arg->getSecondarySize();\r
653                                 for(int i = 0; i < numCols; ++i)\r
654                                 {\r
655                                         for(int j = 0; j < numRows; ++j)\r
656                                         {\r
657                                                 Instruction *mov = emit(sw::Shader::OPCODE_MOV, result, arg);\r
658                                                 mov->src[0].index += i;\r
659                                                 mov->src[0].swizzle = 0x55 * j;\r
660                                                 mov->dst.index += j;\r
661                                                 mov->dst.mask = 1 << i;\r
662                                         }\r
663                                 }\r
664                         }\r
665                         break;\r
666                 default: UNREACHABLE(node->getOp());\r
667                 }\r
668 \r
669                 return true;\r
670         }\r
671 \r
672         bool OutputASM::visitAggregate(Visit visit, TIntermAggregate *node)\r
673         {\r
674                 if(currentScope != emitScope && node->getOp() != EOpFunction && node->getOp() != EOpSequence)\r
675                 {\r
676                         return false;\r
677                 }\r
678 \r
679                 Constant zero(0.0f, 0.0f, 0.0f, 0.0f);\r
680 \r
681                 TIntermTyped *result = node;\r
682                 const TType &resultType = node->getType();\r
683                 TIntermSequence &arg = node->getSequence();\r
684                 int argumentCount = arg.size();\r
685 \r
686                 switch(node->getOp())\r
687                 {\r
688                 case EOpSequence:           break;\r
689                 case EOpDeclaration:        break;\r
690                 case EOpPrototype:          break;\r
691                 case EOpComma:\r
692                         if(visit == PostVisit)\r
693                         {\r
694                                 copy(result, arg[1]);\r
695                         }\r
696                         break;\r
697                 case EOpFunction:\r
698                         if(visit == PreVisit)\r
699                         {\r
700                                 const TString &name = node->getName();\r
701 \r
702                                 if(emitScope == FUNCTION)\r
703                                 {\r
704                                         if(functionArray.size() > 1)   // No need for a label when there's only main()\r
705                                         {\r
706                                                 Instruction *label = emit(sw::Shader::OPCODE_LABEL);\r
707                                                 label->dst.type = sw::Shader::PARAMETER_LABEL;\r
708 \r
709                                                 const Function *function = findFunction(name);\r
710                                                 ASSERT(function);   // Should have been added during global pass\r
711                                                 label->dst.index = function->label;\r
712                                                 currentFunction = function->label;\r
713                                         }\r
714                                 }\r
715                                 else if(emitScope == GLOBAL)\r
716                                 {\r
717                                         if(name != "main(")\r
718                                         {\r
719                                                 TIntermSequence &arguments = node->getSequence()[0]->getAsAggregate()->getSequence();\r
720                                                 functionArray.push_back(Function(functionArray.size(), name, &arguments, node));\r
721                                         }\r
722                                 }\r
723                                 else UNREACHABLE(emitScope);\r
724 \r
725                                 currentScope = FUNCTION;\r
726                         }\r
727                         else if(visit == PostVisit)\r
728                         {\r
729                                 if(emitScope == FUNCTION)\r
730                                 {\r
731                                         if(functionArray.size() > 1)   // No need to return when there's only main()\r
732                                         {\r
733                                                 emit(sw::Shader::OPCODE_RET);\r
734                                         }\r
735                                 }\r
736 \r
737                                 currentScope = GLOBAL;\r
738                         }\r
739                         break;\r
740                 case EOpFunctionCall:\r
741                         if(visit == PostVisit)\r
742                         {\r
743                                 if(node->isUserDefined())\r
744                                 {\r
745                                         const TString &name = node->getName();\r
746                                         const Function *function = findFunction(name);\r
747 \r
748                                         if(!function)\r
749                                         {\r
750                                                 mContext.error(node->getLine(), "function definition not found", name.c_str());\r
751                                                 return false;\r
752                                         }\r
753 \r
754                                         TIntermSequence &arguments = *function->arg;\r
755 \r
756                                         for(int i = 0; i < argumentCount; i++)\r
757                                         {\r
758                                                 TIntermTyped *in = arguments[i]->getAsTyped();\r
759 \r
760                                                 if(in->getQualifier() == EvqIn ||\r
761                                                    in->getQualifier() == EvqInOut ||\r
762                                                    in->getQualifier() == EvqConstReadOnly)\r
763                                                 {\r
764                                                         copy(in, arg[i]);\r
765                                                 }\r
766                                         }\r
767 \r
768                                         Instruction *call = emit(sw::Shader::OPCODE_CALL);\r
769                                         call->dst.type = sw::Shader::PARAMETER_LABEL;\r
770                                         call->dst.index = function->label;\r
771 \r
772                                         if(function->ret && function->ret->getType().getBasicType() != EbtVoid)\r
773                                         {\r
774                                                 copy(result, function->ret);\r
775                                         }\r
776 \r
777                                         for(int i = 0; i < argumentCount; i++)\r
778                                         {\r
779                                                 TIntermTyped *argument = arguments[i]->getAsTyped();\r
780                                                 TIntermTyped *out = arg[i]->getAsTyped();\r
781                                                                 \r
782                                                 if(argument->getQualifier() == EvqOut ||\r
783                                                    argument->getQualifier() == EvqInOut)\r
784                                                 {\r
785                                                         copy(out, argument);\r
786                                                 }\r
787                                         }\r
788                                 }\r
789                                 else\r
790                                 {\r
791                                         TString name = TFunction::unmangleName(node->getName());\r
792 \r
793                                         if(name == "texture" || name == "texture2D" || name == "textureCube" || name == "texture3D")\r
794                                         {\r
795                                                 if(argumentCount == 2)\r
796                                                 {\r
797                                                         emit(sw::Shader::OPCODE_TEX, result, arg[1], arg[0]);\r
798                                                 }\r
799                                                 else if(argumentCount == 3)   // bias\r
800                                                 {\r
801                                                         Temporary uvwb(this);\r
802                                                         emit(sw::Shader::OPCODE_MOV, &uvwb, arg[1]);\r
803                                                         Instruction *bias = emit(sw::Shader::OPCODE_MOV, &uvwb, arg[2]);\r
804                                                         bias->dst.mask = 0x8;\r
805 \r
806                                                         Instruction *tex = emit(sw::Shader::OPCODE_TEX, result, &uvwb, arg[0]);   // FIXME: Implement an efficient TEXLDB instruction\r
807                                                         tex->bias = true;\r
808                                                 }\r
809                                                 else UNREACHABLE(argumentCount);\r
810                                         }\r
811                                         else if(name == "texture2DProj")\r
812                                         {\r
813                                                 TIntermTyped *t = arg[1]->getAsTyped();\r
814 \r
815                                                 if(argumentCount == 2)\r
816                                                 {\r
817                                                         Instruction *tex = emit(sw::Shader::OPCODE_TEX, result, arg[1], arg[0]);\r
818                                                         tex->project = true;\r
819 \r
820                                                         if(t->getNominalSize() == 3)\r
821                                                         {\r
822                                                                 tex->src[0].swizzle = 0xA4;\r
823                                                         }\r
824                                                         else ASSERT(t->getNominalSize() == 4);\r
825                                                 }\r
826                                                 else if(argumentCount == 3)   // bias\r
827                                                 {\r
828                                                         Temporary proj(this);\r
829 \r
830                                                         if(t->getNominalSize() == 3)\r
831                                                         {\r
832                                                                 Instruction *div = emit(sw::Shader::OPCODE_DIV, &proj, arg[1], arg[1]);\r
833                                                                 div->src[1].swizzle = 0xAA;\r
834                                                                 div->dst.mask = 0x3;\r
835                                                         }\r
836                                                         else if(t->getNominalSize() == 4)\r
837                                                         {\r
838                                                                 Instruction *div = emit(sw::Shader::OPCODE_DIV, &proj, arg[1], arg[1]);\r
839                                                                 div->src[1].swizzle = 0xFF;\r
840                                                                 div->dst.mask = 0x3;\r
841                                                         }\r
842                                                         else UNREACHABLE(t->getNominalSize());\r
843 \r
844                                                         Instruction *bias = emit(sw::Shader::OPCODE_MOV, &proj, arg[2]);\r
845                                                         bias->dst.mask = 0x8;\r
846 \r
847                                                         Instruction *tex = emit(sw::Shader::OPCODE_TEX, result, &proj, arg[0]);\r
848                                                         tex->bias = true;\r
849                                                 }\r
850                                                 else UNREACHABLE(argumentCount);\r
851                                         }\r
852                                         else if(name == "texture2DLod" || name == "textureCubeLod")\r
853                                         {\r
854                                                 Temporary uvwb(this);\r
855                                                 emit(sw::Shader::OPCODE_MOV, &uvwb, arg[1]);\r
856                                                 Instruction *lod = emit(sw::Shader::OPCODE_MOV, &uvwb, arg[2]);\r
857                                                 lod->dst.mask = 0x8;\r
858 \r
859                                                 emit(sw::Shader::OPCODE_TEXLDL, result, &uvwb, arg[0]);\r
860                                         }\r
861                                         else if(name == "texture2DProjLod")\r
862                                         {\r
863                                                 TIntermTyped *t = arg[1]->getAsTyped();\r
864                                                 Temporary proj(this);\r
865 \r
866                                                 if(t->getNominalSize() == 3)\r
867                                                 {\r
868                                                         Instruction *div = emit(sw::Shader::OPCODE_DIV, &proj, arg[1], arg[1]);\r
869                                                         div->src[1].swizzle = 0xAA;\r
870                                                         div->dst.mask = 0x3;\r
871                                                 }\r
872                                                 else if(t->getNominalSize() == 4)\r
873                                                 {\r
874                                                         Instruction *div = emit(sw::Shader::OPCODE_DIV, &proj, arg[1], arg[1]);\r
875                                                         div->src[1].swizzle = 0xFF;\r
876                                                         div->dst.mask = 0x3;\r
877                                                 }\r
878                                                 else UNREACHABLE(t->getNominalSize());\r
879 \r
880                                                 Instruction *lod = emit(sw::Shader::OPCODE_MOV, &proj, arg[2]);\r
881                                                 lod->dst.mask = 0x8;\r
882 \r
883                                                 emit(sw::Shader::OPCODE_TEXLDL, result, &proj, arg[0]);\r
884                                         }\r
885                                         else UNREACHABLE(0);\r
886                                 }\r
887                         }\r
888                         break;\r
889                 case EOpParameters:\r
890                         break;\r
891                 case EOpConstructFloat:\r
892                 case EOpConstructVec2:\r
893                 case EOpConstructVec3:\r
894                 case EOpConstructVec4:\r
895                 case EOpConstructBool:\r
896                 case EOpConstructBVec2:\r
897                 case EOpConstructBVec3:\r
898                 case EOpConstructBVec4:\r
899                 case EOpConstructInt:\r
900                 case EOpConstructIVec2:\r
901                 case EOpConstructIVec3:\r
902                 case EOpConstructIVec4:\r
903                 case EOpConstructUInt:\r
904                 case EOpConstructUVec2:\r
905                 case EOpConstructUVec3:\r
906                 case EOpConstructUVec4:\r
907                         if(visit == PostVisit)\r
908                         {\r
909                                 int component = 0;\r
910 \r
911                                 for(int i = 0; i < argumentCount; i++)\r
912                                 {\r
913                                         TIntermTyped *argi = arg[i]->getAsTyped();\r
914                                         int size = argi->getNominalSize();\r
915 \r
916                                         if(!argi->isMatrix())\r
917                                         {\r
918                                                 Instruction *mov = emitCast(result, argi);\r
919                                                 mov->dst.mask = (0xF << component) & 0xF;\r
920                                                 mov->src[0].swizzle = readSwizzle(argi, size) << (component * 2);\r
921 \r
922                                                 component += size;\r
923                                         }\r
924                                         else   // Matrix\r
925                                         {\r
926                                                 int column = 0;\r
927 \r
928                                                 while(component < resultType.getNominalSize())\r
929                                                 {\r
930                                                         Instruction *mov = emitCast(result, argi);\r
931                                                         mov->dst.mask = (0xF << component) & 0xF;\r
932                                                         mov->src[0].index += column;\r
933                                                         mov->src[0].swizzle = readSwizzle(argi, size) << (component * 2);\r
934 \r
935                                                         column++;\r
936                                                         component += size;\r
937                                                 }\r
938                                         }\r
939                                 }\r
940                         }\r
941                         break;\r
942                 case EOpConstructMat2:\r
943                 case EOpConstructMat2x3:\r
944                 case EOpConstructMat2x4:\r
945                 case EOpConstructMat3x2:\r
946                 case EOpConstructMat3:\r
947                 case EOpConstructMat3x4:\r
948                 case EOpConstructMat4x2:\r
949                 case EOpConstructMat4x3:\r
950                 case EOpConstructMat4:\r
951                         if(visit == PostVisit)\r
952                         {\r
953                                 TIntermTyped *arg0 = arg[0]->getAsTyped();\r
954                                 const int dim = result->getNominalSize();\r
955 \r
956                                 if(arg0->isScalar() && arg.size() == 1)   // Construct scale matrix\r
957                                 {\r
958                                         for(int i = 0; i < dim; i++)\r
959                                         {\r
960                                                 Instruction *init = emit(sw::Shader::OPCODE_MOV, result, &zero);\r
961                                                 init->dst.index += i;\r
962                                                 Instruction *mov = emitCast(result, arg0);\r
963                                                 mov->dst.index += i;\r
964                                                 mov->dst.mask = 1 << i;\r
965                                                 ASSERT(mov->src[0].swizzle == 0x00);\r
966                                         }\r
967                                 }\r
968                                 else if(arg0->isMatrix())\r
969                                 {\r
970                                         for(int i = 0; i < dim; i++)\r
971                                         {\r
972                                                 if(dim > dim2(arg0))\r
973                                                 {\r
974                                                         // Initialize to identity matrix\r
975                                                         Constant col((i == 0 ? 1.0f : 0.0f), (i == 1 ? 1.0f : 0.0f), (i == 2 ? 1.0f : 0.0f), (i == 3 ? 1.0f : 0.0f));\r
976                                                         Instruction *mov = emitCast(result, &col);\r
977                                                         mov->dst.index += i;\r
978                                                 }\r
979 \r
980                                                 if(i < dim2(arg0))\r
981                                                 {\r
982                                                         Instruction *mov = emitCast(result, arg0);\r
983                                                         mov->dst.index += i;\r
984                                                         mov->dst.mask = 0xF >> (4 - dim2(arg0));\r
985                                                         argument(mov->src[0], arg0, i);\r
986                                                 }\r
987                                         }\r
988                                 }\r
989                                 else\r
990                                 {\r
991                                         int column = 0;\r
992                                         int row = 0;\r
993 \r
994                                         for(int i = 0; i < argumentCount; i++)\r
995                                         {\r
996                                                 TIntermTyped *argi = arg[i]->getAsTyped();\r
997                                                 int size = argi->getNominalSize();\r
998                                                 int element = 0;\r
999 \r
1000                                                 while(element < size)\r
1001                                                 {\r
1002                                                         Instruction *mov = emitCast(result, argi);\r
1003                                                         mov->dst.index += column;\r
1004                                                         mov->dst.mask = (0xF << row) & 0xF;\r
1005                                                         mov->src[0].swizzle = (readSwizzle(argi, size) << (row * 2)) + 0x55 * element;\r
1006 \r
1007                                                         int end = row + size - element;\r
1008                                                         column = end >= dim ? column + 1 : column;\r
1009                                                         element = element + dim - row;\r
1010                                                         row = end >= dim ? 0 : end;\r
1011                                                 }\r
1012                                         }\r
1013                                 }\r
1014                         }\r
1015                         break;\r
1016                 case EOpConstructStruct:\r
1017                         if(visit == PostVisit)\r
1018                         {\r
1019                                 int offset = 0;\r
1020                                 for(int i = 0; i < argumentCount; i++)\r
1021                                 {\r
1022                                         TIntermTyped *argi = arg[i]->getAsTyped();\r
1023                                         int size = argi->totalRegisterCount();\r
1024 \r
1025                                         for(int index = 0; index < size; index++)\r
1026                                         {\r
1027                                                 Instruction *mov = emit(sw::Shader::OPCODE_MOV, result, argi);\r
1028                                                 mov->dst.index += index + offset;\r
1029                                                 mov->dst.mask = writeMask(result, offset + index);\r
1030                                                 argument(mov->src[0], argi, index);\r
1031                                         }\r
1032 \r
1033                                         offset += size;\r
1034                                 }\r
1035                         }\r
1036                         break;\r
1037                 case EOpLessThan:         if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_LT, result, arg[0], arg[1]); break;\r
1038                 case EOpGreaterThan:      if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_GT, result, arg[0], arg[1]); break;\r
1039                 case EOpLessThanEqual:    if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_LE, result, arg[0], arg[1]); break;\r
1040                 case EOpGreaterThanEqual: if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_GE, result, arg[0], arg[1]); break;\r
1041                 case EOpVectorEqual:      if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_EQ, result, arg[0], arg[1]); break;\r
1042                 case EOpVectorNotEqual:   if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_NE, result, arg[0], arg[1]); break;\r
1043                 case EOpMod:              if(visit == PostVisit) emit(sw::Shader::OPCODE_MOD, result, arg[0], arg[1]); break;\r
1044                 case EOpPow:              if(visit == PostVisit) emit(sw::Shader::OPCODE_POW, result, arg[0], arg[1]); break;\r
1045                 case EOpAtan:             if(visit == PostVisit) emit(sw::Shader::OPCODE_ATAN2, result, arg[0], arg[1]); break;\r
1046                 case EOpMin:              if(visit == PostVisit) emit(sw::Shader::OPCODE_MIN, result, arg[0], arg[1]); break;\r
1047                 case EOpMax:              if(visit == PostVisit) emit(sw::Shader::OPCODE_MAX, result, arg[0], arg[1]); break;\r
1048                 case EOpClamp:\r
1049                         if(visit == PostVisit)\r
1050                         {\r
1051                                 emit(sw::Shader::OPCODE_MAX, result, arg[0], arg[1]);\r
1052                                 emit(sw::Shader::OPCODE_MIN, result, result, arg[2]);\r
1053                         }\r
1054                         break;\r
1055                 case EOpMix:         if(visit == PostVisit) emit(sw::Shader::OPCODE_LRP, result, arg[2], arg[1], arg[0]); break;\r
1056                 case EOpStep:        if(visit == PostVisit) emit(sw::Shader::OPCODE_STEP, result, arg[0], arg[1]); break;\r
1057                 case EOpSmoothStep:  if(visit == PostVisit) emit(sw::Shader::OPCODE_SMOOTH, result, arg[0], arg[1], arg[2]); break;\r
1058                 case EOpFloatBitsToInt:  if(visit == PostVisit) emit(sw::Shader::OPCODE_FLOATBITSTOINT, result, arg[0]); break;\r
1059                 case EOpFloatBitsToUint: if(visit == PostVisit) emit(sw::Shader::OPCODE_FLOATBITSTOUINT, result, arg[0]); break;\r
1060                 case EOpIntBitsToFloat:  if(visit == PostVisit) emit(sw::Shader::OPCODE_INTBITSTOFLOAT, result, arg[0]); break;\r
1061                 case EOpUintBitsToFloat: if(visit == PostVisit) emit(sw::Shader::OPCODE_UINTBITSTOFLOAT, result, arg[0]); break;\r
1062                 case EOpPackSnorm2x16:   if(visit == PostVisit) emit(sw::Shader::OPCODE_PACKSNORM2x16, result, arg[0]); break;\r
1063                 case EOpPackUnorm2x16:   if(visit == PostVisit) emit(sw::Shader::OPCODE_PACKUNORM2x16, result, arg[0]); break;\r
1064                 case EOpPackHalf2x16:    if(visit == PostVisit) emit(sw::Shader::OPCODE_PACKHALF2x16, result, arg[0]); break;\r
1065                 case EOpUnpackSnorm2x16: if(visit == PostVisit) emit(sw::Shader::OPCODE_UNPACKSNORM2x16, result, arg[0]); break;\r
1066                 case EOpUnpackUnorm2x16: if(visit == PostVisit) emit(sw::Shader::OPCODE_UNPACKUNORM2x16, result, arg[0]); break;\r
1067                 case EOpUnpackHalf2x16:  if(visit == PostVisit) emit(sw::Shader::OPCODE_UNPACKHALF2x16, result, arg[0]); break;\r
1068                 case EOpDistance:    if(visit == PostVisit) emit(sw::Shader::OPCODE_DIST(dim(arg[0])), result, arg[0], arg[1]); break;\r
1069                 case EOpDot:         if(visit == PostVisit) emit(sw::Shader::OPCODE_DP(dim(arg[0])), result, arg[0], arg[1]); break;\r
1070                 case EOpCross:       if(visit == PostVisit) emit(sw::Shader::OPCODE_CRS, result, arg[0], arg[1]); break;\r
1071                 case EOpFaceForward: if(visit == PostVisit) emit(sw::Shader::OPCODE_FORWARD(dim(arg[0])), result, arg[0], arg[1], arg[2]); break;\r
1072                 case EOpReflect:     if(visit == PostVisit) emit(sw::Shader::OPCODE_REFLECT(dim(arg[0])), result, arg[0], arg[1]); break;\r
1073                 case EOpRefract:     if(visit == PostVisit) emit(sw::Shader::OPCODE_REFRACT(dim(arg[0])), result, arg[0], arg[1], arg[2]); break;\r
1074                 case EOpMul:\r
1075                         if(visit == PostVisit)\r
1076                         {\r
1077                                 ASSERT(dim2(arg[0]) == dim2(arg[1]));\r
1078 \r
1079                                 for(int i = 0; i < dim2(arg[0]); i++)\r
1080                                 {\r
1081                                         Instruction *mul = emit(sw::Shader::OPCODE_MUL, result, arg[0], arg[1]);\r
1082                                         mul->dst.index += i;\r
1083                                         argument(mul->src[0], arg[0], i);\r
1084                                         argument(mul->src[1], arg[1], i);\r
1085                                 }\r
1086                         }\r
1087                         break;\r
1088                 case EOpOuterProduct:\r
1089                         if(visit == PostVisit)\r
1090                         {\r
1091                                 for(int i = 0; i < dim(arg[1]); i++)\r
1092                                 {\r
1093                                         Instruction *mul = emit(sw::Shader::OPCODE_MUL, result, arg[0], arg[1]);\r
1094                                         mul->dst.index += i;\r
1095                                         mul->src[1].swizzle = 0x55 * i;\r
1096                                 }\r
1097                         }\r
1098                         break;\r
1099                 default: UNREACHABLE(node->getOp());\r
1100                 }\r
1101 \r
1102                 return true;\r
1103         }\r
1104 \r
1105         bool OutputASM::visitSelection(Visit visit, TIntermSelection *node)\r
1106         {\r
1107                 if(currentScope != emitScope)\r
1108                 {\r
1109                         return false;\r
1110                 }\r
1111 \r
1112                 TIntermTyped *condition = node->getCondition();\r
1113                 TIntermNode *trueBlock = node->getTrueBlock();\r
1114                 TIntermNode *falseBlock = node->getFalseBlock();\r
1115                 TIntermConstantUnion *constantCondition = condition->getAsConstantUnion();\r
1116 \r
1117                 condition->traverse(this);\r
1118 \r
1119                 if(node->usesTernaryOperator())\r
1120                 {\r
1121                         if(constantCondition)\r
1122                         {\r
1123                                 bool trueCondition = constantCondition->getUnionArrayPointer()->getBConst();\r
1124 \r
1125                                 if(trueCondition)\r
1126                                 {\r
1127                                         trueBlock->traverse(this);\r
1128                                         copy(node, trueBlock);\r
1129                                 }\r
1130                                 else\r
1131                                 {\r
1132                                         falseBlock->traverse(this);\r
1133                                         copy(node, falseBlock);\r
1134                                 }\r
1135                         }\r
1136                         else if(trivial(node, 6))   // Fast to compute both potential results and no side effects\r
1137                         {\r
1138                                 trueBlock->traverse(this);\r
1139                                 falseBlock->traverse(this);\r
1140                                 emit(sw::Shader::OPCODE_SELECT, node, condition, trueBlock, falseBlock);\r
1141                         }\r
1142                         else\r
1143                         {\r
1144                                 emit(sw::Shader::OPCODE_IF, 0, condition);\r
1145 \r
1146                                 if(trueBlock)\r
1147                                 {\r
1148                                         trueBlock->traverse(this);\r
1149                                         copy(node, trueBlock);\r
1150                                 }\r
1151 \r
1152                                 if(falseBlock)\r
1153                                 {\r
1154                                         emit(sw::Shader::OPCODE_ELSE);\r
1155                                         falseBlock->traverse(this);\r
1156                                         copy(node, falseBlock);\r
1157                                 }\r
1158 \r
1159                                 emit(sw::Shader::OPCODE_ENDIF);\r
1160                         }\r
1161                 }\r
1162                 else  // if/else statement\r
1163                 {\r
1164                         if(constantCondition)\r
1165                         {\r
1166                                 bool trueCondition = constantCondition->getUnionArrayPointer()->getBConst();\r
1167 \r
1168                                 if(trueCondition)\r
1169                                 {\r
1170                                         if(trueBlock)\r
1171                                         {\r
1172                                                 trueBlock->traverse(this);\r
1173                                         }\r
1174                                 }\r
1175                                 else\r
1176                                 {\r
1177                                         if(falseBlock)\r
1178                                         {\r
1179                                                 falseBlock->traverse(this);\r
1180                                         }\r
1181                                 }\r
1182                         }\r
1183                         else\r
1184                         {\r
1185                                 emit(sw::Shader::OPCODE_IF, 0, condition);\r
1186 \r
1187                                 if(trueBlock)\r
1188                                 {\r
1189                                         trueBlock->traverse(this);\r
1190                                 }\r
1191 \r
1192                                 if(falseBlock)\r
1193                                 {\r
1194                                         emit(sw::Shader::OPCODE_ELSE);\r
1195                                         falseBlock->traverse(this);\r
1196                                 }\r
1197 \r
1198                                 emit(sw::Shader::OPCODE_ENDIF);\r
1199                         }\r
1200                 }\r
1201 \r
1202                 return false;\r
1203         }\r
1204 \r
1205         bool OutputASM::visitLoop(Visit visit, TIntermLoop *node)\r
1206         {\r
1207                 if(currentScope != emitScope)\r
1208                 {\r
1209                         return false;\r
1210                 }\r
1211 \r
1212                 unsigned int iterations = loopCount(node);\r
1213 \r
1214                 if(iterations == 0)\r
1215                 {\r
1216                         return false;\r
1217                 }\r
1218 \r
1219                 bool unroll = (iterations <= 4);\r
1220 \r
1221                 if(unroll)\r
1222                 {\r
1223                         DetectLoopDiscontinuity detectLoopDiscontinuity;\r
1224                         unroll = !detectLoopDiscontinuity.traverse(node);\r
1225                 }\r
1226 \r
1227                 TIntermNode *init = node->getInit();\r
1228                 TIntermTyped *condition = node->getCondition();\r
1229                 TIntermTyped *expression = node->getExpression();\r
1230                 TIntermNode *body = node->getBody();\r
1231 \r
1232                 if(node->getType() == ELoopDoWhile)\r
1233                 {\r
1234                         Temporary iterate(this);\r
1235                         Constant True(true);\r
1236                         emit(sw::Shader::OPCODE_MOV, &iterate, &True);\r
1237 \r
1238                         emit(sw::Shader::OPCODE_WHILE, 0, &iterate);   // FIXME: Implement real do-while\r
1239 \r
1240                         if(body)\r
1241                         {\r
1242                                 body->traverse(this);\r
1243                         }\r
1244 \r
1245                         emit(sw::Shader::OPCODE_TEST);\r
1246 \r
1247                         condition->traverse(this);\r
1248                         emit(sw::Shader::OPCODE_MOV, &iterate, condition);\r
1249 \r
1250                         emit(sw::Shader::OPCODE_ENDWHILE);\r
1251                 }\r
1252                 else\r
1253                 {\r
1254                         if(init)\r
1255                         {\r
1256                                 init->traverse(this);\r
1257                         }\r
1258 \r
1259                         if(unroll)\r
1260                         {\r
1261                                 for(unsigned int i = 0; i < iterations; i++)\r
1262                                 {\r
1263                                 //      condition->traverse(this);   // Condition could contain statements, but not in an unrollable loop\r
1264 \r
1265                                         if(body)\r
1266                                         {\r
1267                                                 body->traverse(this);\r
1268                                         }\r
1269 \r
1270                                         if(expression)\r
1271                                         {\r
1272                                                 expression->traverse(this);\r
1273                                         }\r
1274                                 }\r
1275                         }\r
1276                         else\r
1277                         {\r
1278                                 if(condition)\r
1279                                 {\r
1280                                         condition->traverse(this);\r
1281                                 }\r
1282 \r
1283                                 emit(sw::Shader::OPCODE_WHILE, 0, condition);\r
1284 \r
1285                                 if(body)\r
1286                                 {\r
1287                                         body->traverse(this);\r
1288                                 }\r
1289 \r
1290                                 emit(sw::Shader::OPCODE_TEST);\r
1291 \r
1292                                 if(expression)\r
1293                                 {\r
1294                                         expression->traverse(this);\r
1295                                 }\r
1296 \r
1297                                 if(condition)\r
1298                                 {\r
1299                                         condition->traverse(this);\r
1300                                 }\r
1301 \r
1302                                 emit(sw::Shader::OPCODE_ENDWHILE);\r
1303                         }\r
1304                 }\r
1305 \r
1306                 return false;\r
1307         }\r
1308 \r
1309         bool OutputASM::visitBranch(Visit visit, TIntermBranch *node)\r
1310         {\r
1311                 if(currentScope != emitScope)\r
1312                 {\r
1313                         return false;\r
1314                 }\r
1315 \r
1316                 switch(node->getFlowOp())\r
1317                 {\r
1318                 case EOpKill:      if(visit == PostVisit) emit(sw::Shader::OPCODE_DISCARD);  break;\r
1319                 case EOpBreak:     if(visit == PostVisit) emit(sw::Shader::OPCODE_BREAK);    break;\r
1320                 case EOpContinue:  if(visit == PostVisit) emit(sw::Shader::OPCODE_CONTINUE); break;\r
1321                 case EOpReturn:\r
1322                         if(visit == PostVisit)\r
1323                         {\r
1324                                 TIntermTyped *value = node->getExpression();\r
1325 \r
1326                                 if(value)\r
1327                                 {\r
1328                                         copy(functionArray[currentFunction].ret, value);\r
1329                                 }\r
1330 \r
1331                                 emit(sw::Shader::OPCODE_LEAVE);\r
1332                         }\r
1333                         break;\r
1334                 default: UNREACHABLE(node->getFlowOp());\r
1335                 }\r
1336 \r
1337                 return true;\r
1338         }\r
1339 \r
1340         bool OutputASM::isSamplerRegister(TIntermTyped *operand)\r
1341         {\r
1342                 return operand && isSamplerRegister(operand->getType());\r
1343         }\r
1344 \r
1345         bool OutputASM::isSamplerRegister(const TType &type)\r
1346         {\r
1347                 // A sampler register's qualifiers can be:\r
1348                 // - EvqUniform: The sampler uniform is used as is in the code (default case).\r
1349                 // - EvqTemporary: The sampler is indexed. It's still a sampler register.\r
1350                 // - EvqIn (and other similar types): The sampler has been passed as a function argument. At this point,\r
1351                 //                                    the sampler has been copied and is no longer a sampler register.\r
1352                 return IsSampler(type.getBasicType()) && (type.getQualifier() == EvqUniform || type.getQualifier() == EvqTemporary);\r
1353         }\r
1354 \r
1355         Instruction *OutputASM::emit(sw::Shader::Opcode op, TIntermTyped *dst, TIntermNode *src0, TIntermNode *src1, TIntermNode *src2, int index)\r
1356         {\r
1357                 if(isSamplerRegister(dst))\r
1358                 {\r
1359                         op = sw::Shader::OPCODE_NULL;   // Can't assign to a sampler, but this is hit when indexing sampler arrays\r
1360                 }\r
1361 \r
1362                 Instruction *instruction = new Instruction(op);\r
1363 \r
1364                 if(dst)\r
1365                 {\r
1366                         instruction->dst.type = registerType(dst);\r
1367                         instruction->dst.index = registerIndex(dst) + index;\r
1368                         instruction->dst.mask = writeMask(dst);\r
1369                         instruction->dst.integer = (dst->getBasicType() == EbtInt);\r
1370                 }\r
1371 \r
1372                 argument(instruction->src[0], src0, index);\r
1373                 argument(instruction->src[1], src1, index);\r
1374                 argument(instruction->src[2], src2, index);\r
1375 \r
1376                 shader->append(instruction);\r
1377 \r
1378                 return instruction;\r
1379         }\r
1380 \r
1381         Instruction *OutputASM::emitCast(TIntermTyped *dst, TIntermTyped *src)\r
1382         {\r
1383                 // Integers are implemented as float\r
1384                 if((dst->getBasicType() == EbtFloat || dst->getBasicType() == EbtInt) && src->getBasicType() == EbtBool)\r
1385                 {\r
1386                         return emit(sw::Shader::OPCODE_B2F, dst, src);\r
1387                 }\r
1388                 if(dst->getBasicType() == EbtBool && (src->getBasicType() == EbtFloat || src->getBasicType() == EbtInt))\r
1389                 {\r
1390                         return emit(sw::Shader::OPCODE_F2B, dst, src);\r
1391                 }\r
1392                 if(dst->getBasicType() == EbtInt && src->getBasicType() == EbtFloat)\r
1393                 {\r
1394                         return emit(sw::Shader::OPCODE_TRUNC, dst, src);\r
1395                 }\r
1396 \r
1397                 return emit(sw::Shader::OPCODE_MOV, dst, src);\r
1398         }\r
1399 \r
1400         void OutputASM::emitBinary(sw::Shader::Opcode op, TIntermTyped *dst, TIntermNode *src0, TIntermNode *src1, TIntermNode *src2)\r
1401         {\r
1402                 for(int index = 0; index < dst->elementRegisterCount(); index++)\r
1403                 {\r
1404                         emit(op, dst, src0, src1, src2, index);\r
1405                 }\r
1406         }\r
1407 \r
1408         void OutputASM::emitAssign(sw::Shader::Opcode op, TIntermTyped *result, TIntermTyped *lhs, TIntermTyped *src0, TIntermTyped *src1)\r
1409         {\r
1410                 emitBinary(op, result, src0, src1);\r
1411                 assignLvalue(lhs, result);\r
1412         }\r
1413 \r
1414         void OutputASM::emitCmp(sw::Shader::Control cmpOp, TIntermTyped *dst, TIntermNode *left, TIntermNode *right, int index)\r
1415         {\r
1416                 bool boolean = (left->getAsTyped()->getBasicType() == EbtBool);\r
1417                 sw::Shader::Opcode opcode = boolean ? sw::Shader::OPCODE_ICMP : sw::Shader::OPCODE_CMP;\r
1418 \r
1419                 Instruction *cmp = emit(opcode, dst, left, right);\r
1420                 cmp->control = cmpOp;\r
1421                 argument(cmp->src[0], left, index);\r
1422                 argument(cmp->src[1], right, index);\r
1423         }\r
1424 \r
1425         int componentCount(const TType &type, int registers)\r
1426         {\r
1427                 if(registers == 0)\r
1428                 {\r
1429                         return 0;\r
1430                 }\r
1431 \r
1432                 if(type.isArray() && registers >= type.elementRegisterCount())\r
1433                 {\r
1434                         int index = registers / type.elementRegisterCount();\r
1435                         registers -= index * type.elementRegisterCount();\r
1436                         return index * type.getElementSize() + componentCount(type, registers);\r
1437                 }\r
1438 \r
1439                 if(type.isStruct())\r
1440                 {\r
1441                         const TFieldList& fields = type.getStruct()->fields();\r
1442                         int elements = 0;\r
1443 \r
1444                         for(TFieldList::const_iterator field = fields.begin(); field != fields.end(); field++)\r
1445                         {\r
1446                                 const TType &fieldType = *((*field)->type());\r
1447 \r
1448                                 if(fieldType.totalRegisterCount() <= registers)\r
1449                                 {\r
1450                                         registers -= fieldType.totalRegisterCount();\r
1451                                         elements += fieldType.getObjectSize();\r
1452                                 }\r
1453                                 else   // Register within this field\r
1454                                 {\r
1455                                         return elements + componentCount(fieldType, registers);\r
1456                                 }\r
1457                         }\r
1458                 }\r
1459                 else if(type.isMatrix())\r
1460                 {\r
1461                         return registers * type.getSecondarySize();\r
1462                 }\r
1463                 \r
1464                 UNREACHABLE(0);\r
1465                 return 0;\r
1466         }\r
1467 \r
1468         int registerSize(const TType &type, int registers)\r
1469         {\r
1470                 if(registers == 0)\r
1471                 {\r
1472                         if(type.isStruct())\r
1473                         {\r
1474                                 return registerSize(*((*(type.getStruct()->fields().begin()))->type()), 0);\r
1475                         }\r
1476 \r
1477                         return type.isMatrix() ? type.getSecondarySize() : type.getNominalSize();\r
1478                 }\r
1479 \r
1480                 if(type.isArray() && registers >= type.elementRegisterCount())\r
1481                 {\r
1482                         int index = registers / type.elementRegisterCount();\r
1483                         registers -= index * type.elementRegisterCount();\r
1484                         return registerSize(type, registers);\r
1485                 }\r
1486 \r
1487                 if(type.isStruct())\r
1488                 {\r
1489                         const TFieldList& fields = type.getStruct()->fields();\r
1490                         int elements = 0;\r
1491 \r
1492                         for(TFieldList::const_iterator field = fields.begin(); field != fields.end(); field++)\r
1493                         {\r
1494                                 const TType &fieldType = *((*field)->type());\r
1495                                 \r
1496                                 if(fieldType.totalRegisterCount() <= registers)\r
1497                                 {\r
1498                                         registers -= fieldType.totalRegisterCount();\r
1499                                         elements += fieldType.getObjectSize();\r
1500                                 }\r
1501                                 else   // Register within this field\r
1502                                 {\r
1503                                         return registerSize(fieldType, registers);\r
1504                                 }\r
1505                         }\r
1506                 }\r
1507                 else if(type.isMatrix())\r
1508                 {\r
1509                         return registerSize(type, 0);\r
1510                 }\r
1511                 \r
1512                 UNREACHABLE(0);\r
1513                 return 0;\r
1514         }\r
1515 \r
1516         void OutputASM::argument(sw::Shader::SourceParameter &parameter, TIntermNode *argument, int index)\r
1517         {\r
1518                 if(argument)\r
1519                 {\r
1520                         TIntermTyped *arg = argument->getAsTyped();\r
1521                         const TType &type = arg->getType();\r
1522                         index = (index >= arg->totalRegisterCount()) ? arg->totalRegisterCount() - 1 : index;\r
1523 \r
1524                         int size = registerSize(type, index);\r
1525 \r
1526                         parameter.type = registerType(arg);\r
1527 \r
1528                         if(arg->getQualifier() == EvqConstExpr)\r
1529                         {\r
1530                                 int component = componentCount(type, index);\r
1531                                 ConstantUnion *constants = arg->getAsConstantUnion()->getUnionArrayPointer();\r
1532 \r
1533                                 for(int i = 0; i < 4; i++)\r
1534                                 {\r
1535                                         if(size == 1)   // Replicate\r
1536                                         {\r
1537                                                 parameter.value[i] = constants[component + 0].getAsFloat();\r
1538                                         }\r
1539                                         else if(i < size)\r
1540                                         {\r
1541                                                 parameter.value[i] = constants[component + i].getAsFloat();\r
1542                                         }\r
1543                                         else\r
1544                                         {\r
1545                                                 parameter.value[i] = 0.0f;\r
1546                                         }\r
1547                                 }\r
1548                         }\r
1549                         else\r
1550                         {\r
1551                                 parameter.index = registerIndex(arg) + index;\r
1552 \r
1553                                 if(isSamplerRegister(arg))\r
1554                                 {\r
1555                                         TIntermBinary *binary = argument->getAsBinaryNode();\r
1556 \r
1557                                         if(binary)\r
1558                                         {\r
1559                                                 TIntermTyped *left = binary->getLeft();\r
1560                                                 TIntermTyped *right = binary->getRight();\r
1561 \r
1562                                                 if(binary->getOp() == EOpIndexDirect)\r
1563                                                 {\r
1564                                                         parameter.index += right->getAsConstantUnion()->getIConst(0);\r
1565                                                 }\r
1566                                                 else if(binary->getOp() == EOpIndexIndirect)\r
1567                                                 {\r
1568                                                         if(left->getArraySize() > 1)\r
1569                                                         {\r
1570                                                                 parameter.rel.type = registerType(binary->getRight());\r
1571                                                                 parameter.rel.index = registerIndex(binary->getRight());\r
1572                                                                 parameter.rel.scale = 1;\r
1573                                                                 parameter.rel.deterministic = true;\r
1574                                                         }\r
1575                                                 }\r
1576                                                 else if(binary->getOp() == EOpIndexDirectStruct)\r
1577                                                 {\r
1578                                                         parameter.index += right->getAsConstantUnion()->getIConst(0);\r
1579                                                 }\r
1580                                                 else UNREACHABLE(binary->getOp());\r
1581                                         }\r
1582                                 }\r
1583                         }\r
1584 \r
1585                         if(!IsSampler(arg->getBasicType()))\r
1586                         {\r
1587                                 parameter.swizzle = readSwizzle(arg, size);\r
1588                         }\r
1589                 }\r
1590         }\r
1591 \r
1592         void OutputASM::copy(TIntermTyped *dst, TIntermNode *src, int offset)\r
1593         {\r
1594                 for(int index = 0; index < dst->totalRegisterCount(); index++)\r
1595                 {\r
1596                         Instruction *mov = emit(sw::Shader::OPCODE_MOV, dst, src);\r
1597                         mov->dst.index += index;\r
1598                         mov->dst.mask = writeMask(dst, index);\r
1599                         argument(mov->src[0], src, offset + index);\r
1600                 }\r
1601         }\r
1602 \r
1603         int swizzleElement(int swizzle, int index)\r
1604         {\r
1605                 return (swizzle >> (index * 2)) & 0x03;\r
1606         }\r
1607 \r
1608         int swizzleSwizzle(int leftSwizzle, int rightSwizzle)\r
1609         {\r
1610                 return (swizzleElement(leftSwizzle, swizzleElement(rightSwizzle, 0)) << 0) |\r
1611                        (swizzleElement(leftSwizzle, swizzleElement(rightSwizzle, 1)) << 2) |\r
1612                        (swizzleElement(leftSwizzle, swizzleElement(rightSwizzle, 2)) << 4) |\r
1613                        (swizzleElement(leftSwizzle, swizzleElement(rightSwizzle, 3)) << 6);\r
1614         }\r
1615 \r
1616         void OutputASM::assignLvalue(TIntermTyped *dst, TIntermTyped *src)\r
1617         {\r
1618                 if(src &&\r
1619                         ((src->isVector() && (!dst->isVector() || (dst->getNominalSize() != dst->getNominalSize()))) ||\r
1620                          (src->isMatrix() && (!dst->isMatrix() || (src->getNominalSize() != dst->getNominalSize()) || (src->getSecondarySize() != dst->getSecondarySize())))))\r
1621                 {\r
1622                         return mContext.error(src->getLine(), "Result type should match the l-value type in compound assignment", src->isVector() ? "vector" : "matrix");\r
1623                 }\r
1624 \r
1625                 TIntermBinary *binary = dst->getAsBinaryNode();\r
1626 \r
1627                 if(binary && binary->getOp() == EOpIndexIndirect && dst->isScalar())\r
1628                 {\r
1629                         Instruction *insert = new Instruction(sw::Shader::OPCODE_INSERT);\r
1630                         \r
1631                         Temporary address(this);\r
1632                         lvalue(insert->dst, address, dst);\r
1633 \r
1634                         insert->src[0].type = insert->dst.type;\r
1635                         insert->src[0].index = insert->dst.index;\r
1636                         insert->src[0].rel = insert->dst.rel;\r
1637                         argument(insert->src[1], src);\r
1638                         argument(insert->src[2], binary->getRight());\r
1639 \r
1640                         shader->append(insert);\r
1641                 }\r
1642                 else\r
1643                 {\r
1644                         for(int offset = 0; offset < dst->totalRegisterCount(); offset++)\r
1645                         {\r
1646                                 Instruction *mov = new Instruction(sw::Shader::OPCODE_MOV);\r
1647                         \r
1648                                 Temporary address(this);\r
1649                                 int swizzle = lvalue(mov->dst, address, dst);\r
1650                                 mov->dst.index += offset;\r
1651 \r
1652                                 if(offset > 0)\r
1653                                 {\r
1654                                         mov->dst.mask = writeMask(dst, offset);\r
1655                                 }\r
1656 \r
1657                                 argument(mov->src[0], src, offset);\r
1658                                 mov->src[0].swizzle = swizzleSwizzle(mov->src[0].swizzle, swizzle);\r
1659 \r
1660                                 shader->append(mov);\r
1661                         }\r
1662                 }\r
1663         }\r
1664 \r
1665         int OutputASM::lvalue(sw::Shader::DestinationParameter &dst, Temporary &address, TIntermTyped *node)\r
1666         {\r
1667                 TIntermTyped *result = node;\r
1668                 TIntermBinary *binary = node->getAsBinaryNode();\r
1669                 TIntermSymbol *symbol = node->getAsSymbolNode();\r
1670 \r
1671                 if(binary)\r
1672                 {\r
1673                         TIntermTyped *left = binary->getLeft();\r
1674                         TIntermTyped *right = binary->getRight();\r
1675 \r
1676                         int leftSwizzle = lvalue(dst, address, left);   // Resolve the l-value of the left side\r
1677 \r
1678                         switch(binary->getOp())\r
1679                         {\r
1680                         case EOpIndexDirect:\r
1681                                 {\r
1682                                         int rightIndex = right->getAsConstantUnion()->getIConst(0);\r
1683 \r
1684                                         if(left->isRegister())\r
1685                                         {\r
1686                                                 int leftMask = dst.mask;\r
1687                                                 \r
1688                                                 dst.mask = 1;\r
1689                                                 while((leftMask & dst.mask) == 0)\r
1690                                                 {\r
1691                                                         dst.mask = dst.mask << 1;\r
1692                                                 }\r
1693 \r
1694                                                 int element = swizzleElement(leftSwizzle, rightIndex);\r
1695                                                 dst.mask = 1 << element;\r
1696                                                 \r
1697                                                 return element;\r
1698                                         }\r
1699                                         else if(left->isArray() || left->isMatrix())\r
1700                                         {\r
1701                                                 dst.index += rightIndex * result->totalRegisterCount();\r
1702                                                 return 0xE4;\r
1703                                         }\r
1704                                         else UNREACHABLE(0);\r
1705                                 }\r
1706                                 break;\r
1707                         case EOpIndexIndirect:\r
1708                                 {\r
1709                                         if(left->isRegister())\r
1710                                         {\r
1711                                                 // Requires INSERT instruction (handled by calling function)\r
1712                                         }\r
1713                                         else if(left->isArray() || left->isMatrix())\r
1714                                         {\r
1715                                                 int scale = result->totalRegisterCount();\r
1716 \r
1717                                                 if(dst.rel.type == sw::Shader::PARAMETER_VOID)   // Use the index register as the relative address directly\r
1718                                                 {\r
1719                                                         if(left->totalRegisterCount() > 1)\r
1720                                                         {\r
1721                                                                 sw::Shader::SourceParameter relativeRegister;\r
1722                                                                 argument(relativeRegister, right);\r
1723 \r
1724                                                                 dst.rel.index = relativeRegister.index;\r
1725                                                                 dst.rel.type = relativeRegister.type;\r
1726                                                                 dst.rel.scale = scale;\r
1727                                                                 dst.rel.deterministic = !(vertexShader && left->getQualifier() == EvqUniform);\r
1728                                                         }\r
1729                                                 }\r
1730                                                 else if(dst.rel.index != registerIndex(&address))   // Move the previous index register to the address register\r
1731                                                 {\r
1732                                                         if(scale == 1)\r
1733                                                         {\r
1734                                                                 Constant oldScale((int)dst.rel.scale);\r
1735                                                                 Instruction *mad = emit(sw::Shader::OPCODE_MAD, &address, &address, &oldScale, right);\r
1736                                                                 mad->src[0].index = dst.rel.index;\r
1737                                                                 mad->src[0].type = dst.rel.type;\r
1738                                                         }\r
1739                                                         else\r
1740                                                         {\r
1741                                                                 Constant oldScale((int)dst.rel.scale);\r
1742                                                                 Instruction *mul = emit(sw::Shader::OPCODE_MUL, &address, &address, &oldScale);\r
1743                                                                 mul->src[0].index = dst.rel.index;\r
1744                                                                 mul->src[0].type = dst.rel.type;\r
1745 \r
1746                                                                 Constant newScale(scale);\r
1747                                                                 emit(sw::Shader::OPCODE_MAD, &address, right, &newScale, &address);\r
1748                                                         }\r
1749 \r
1750                                                         dst.rel.type = sw::Shader::PARAMETER_TEMP;\r
1751                                                         dst.rel.index = registerIndex(&address);\r
1752                                                         dst.rel.scale = 1;\r
1753                                                 }\r
1754                                                 else   // Just add the new index to the address register\r
1755                                                 {\r
1756                                                         if(scale == 1)\r
1757                                                         {\r
1758                                                                 emit(sw::Shader::OPCODE_ADD, &address, &address, right);\r
1759                                                         }\r
1760                                                         else\r
1761                                                         {\r
1762                                                                 Constant newScale(scale);\r
1763                                                                 emit(sw::Shader::OPCODE_MAD, &address, right, &newScale, &address);\r
1764                                                         }\r
1765                                                 }\r
1766                                         }\r
1767                                         else UNREACHABLE(0);\r
1768                                 }\r
1769                                 break;\r
1770                         case EOpIndexDirectStruct:\r
1771                                 {\r
1772                                         const TFieldList& fields = left->getType().getStruct()->fields();\r
1773                                         int index = right->getAsConstantUnion()->getIConst(0);\r
1774                                         int fieldOffset = 0;\r
1775 \r
1776                                         for(int i = 0; i < index; i++)\r
1777                                         {\r
1778                                                 fieldOffset += fields[i]->type()->totalRegisterCount();\r
1779                                         }\r
1780 \r
1781                                         dst.type = registerType(left);\r
1782                                         dst.index += fieldOffset;\r
1783                                         dst.mask = writeMask(right);\r
1784 \r
1785                                         return 0xE4;\r
1786                                 }\r
1787                                 break;\r
1788                         case EOpVectorSwizzle:\r
1789                                 {\r
1790                                         ASSERT(left->isRegister());\r
1791 \r
1792                                         int leftMask = dst.mask;\r
1793 \r
1794                                         int swizzle = 0;\r
1795                                         int rightMask = 0;\r
1796 \r
1797                                         TIntermSequence &sequence = right->getAsAggregate()->getSequence();\r
1798 \r
1799                                         for(unsigned int i = 0; i < sequence.size(); i++)\r
1800                                         {\r
1801                                                 int index = sequence[i]->getAsConstantUnion()->getIConst(0);\r
1802 \r
1803                                                 int element = swizzleElement(leftSwizzle, index);\r
1804                                                 rightMask = rightMask | (1 << element);\r
1805                                                 swizzle = swizzle | swizzleElement(leftSwizzle, i) << (element * 2);\r
1806                                         }\r
1807                                         \r
1808                                         dst.mask = leftMask & rightMask;\r
1809 \r
1810                                         return swizzle;\r
1811                                 }\r
1812                                 break;\r
1813                         default:\r
1814                                 UNREACHABLE(binary->getOp());   // Not an l-value operator\r
1815                                 break;\r
1816                         }\r
1817                 }\r
1818                 else if(symbol)\r
1819                 {\r
1820                         dst.type = registerType(symbol);\r
1821                         dst.index = registerIndex(symbol);\r
1822                         dst.mask = writeMask(symbol);\r
1823                         return 0xE4;\r
1824                 }\r
1825 \r
1826                 return 0xE4;\r
1827         }\r
1828 \r
1829         sw::Shader::ParameterType OutputASM::registerType(TIntermTyped *operand)\r
1830         {\r
1831                 if(isSamplerRegister(operand))\r
1832                 {\r
1833                         return sw::Shader::PARAMETER_SAMPLER;\r
1834                 }\r
1835 \r
1836                 const TQualifier qualifier = operand->getQualifier();\r
1837                 if((EvqFragColor == qualifier) || (EvqFragData == qualifier))\r
1838                 {\r
1839                         if(((EvqFragData == qualifier) && (EvqFragColor == outputQualifier)) ||\r
1840                            ((EvqFragColor == qualifier) && (EvqFragData == outputQualifier)))\r
1841                         {\r
1842                                 mContext.error(operand->getLine(), "static assignment to both gl_FragData and gl_FragColor", "");\r
1843                         }\r
1844                         outputQualifier = qualifier;\r
1845                 }\r
1846 \r
1847                 switch(qualifier)\r
1848                 {\r
1849                 case EvqTemporary:           return sw::Shader::PARAMETER_TEMP;\r
1850                 case EvqGlobal:              return sw::Shader::PARAMETER_TEMP;\r
1851                 case EvqConstExpr:           return sw::Shader::PARAMETER_FLOAT4LITERAL;   // All converted to float\r
1852                 case EvqAttribute:           return sw::Shader::PARAMETER_INPUT;\r
1853                 case EvqVaryingIn:           return sw::Shader::PARAMETER_INPUT;\r
1854                 case EvqVaryingOut:          return sw::Shader::PARAMETER_OUTPUT;\r
1855                 case EvqVertexIn:            return sw::Shader::PARAMETER_INPUT;\r
1856                 case EvqFragmentOut:         return sw::Shader::PARAMETER_COLOROUT;\r
1857                 case EvqVertexOut:           return sw::Shader::PARAMETER_OUTPUT;\r
1858                 case EvqFragmentIn:          return sw::Shader::PARAMETER_INPUT;\r
1859                 case EvqInvariantVaryingIn:  return sw::Shader::PARAMETER_INPUT;    // FIXME: Guarantee invariance at the backend\r
1860                 case EvqInvariantVaryingOut: return sw::Shader::PARAMETER_OUTPUT;   // FIXME: Guarantee invariance at the backend \r
1861                 case EvqUniform:             return sw::Shader::PARAMETER_CONST;\r
1862                 case EvqIn:                  return sw::Shader::PARAMETER_TEMP;\r
1863                 case EvqOut:                 return sw::Shader::PARAMETER_TEMP;\r
1864                 case EvqInOut:               return sw::Shader::PARAMETER_TEMP;\r
1865                 case EvqConstReadOnly:       return sw::Shader::PARAMETER_TEMP;\r
1866                 case EvqPosition:            return sw::Shader::PARAMETER_OUTPUT;\r
1867                 case EvqPointSize:           return sw::Shader::PARAMETER_OUTPUT;\r
1868                 case EvqInstanceID:          return sw::Shader::PARAMETER_MISCTYPE;\r
1869                 case EvqFragCoord:           return sw::Shader::PARAMETER_MISCTYPE;\r
1870                 case EvqFrontFacing:         return sw::Shader::PARAMETER_MISCTYPE;\r
1871                 case EvqPointCoord:          return sw::Shader::PARAMETER_INPUT;\r
1872                 case EvqFragColor:           return sw::Shader::PARAMETER_COLOROUT;\r
1873                 case EvqFragData:            return sw::Shader::PARAMETER_COLOROUT;\r
1874                 default: UNREACHABLE(qualifier);\r
1875                 }\r
1876 \r
1877                 return sw::Shader::PARAMETER_VOID;\r
1878         }\r
1879 \r
1880         int OutputASM::registerIndex(TIntermTyped *operand)\r
1881         {\r
1882                 if(isSamplerRegister(operand))\r
1883                 {\r
1884                         return samplerRegister(operand);\r
1885                 }\r
1886 \r
1887                 switch(operand->getQualifier())\r
1888                 {\r
1889                 case EvqTemporary:           return temporaryRegister(operand);\r
1890                 case EvqGlobal:              return temporaryRegister(operand);\r
1891                 case EvqConstExpr:           UNREACHABLE(EvqConstExpr);\r
1892                 case EvqAttribute:           return attributeRegister(operand);\r
1893                 case EvqVaryingIn:           return varyingRegister(operand);\r
1894                 case EvqVaryingOut:          return varyingRegister(operand);\r
1895                 case EvqVertexIn:            return attributeRegister(operand);\r
1896                 case EvqFragmentOut:         return 0;\r
1897                 case EvqVertexOut:           return varyingRegister(operand);\r
1898                 case EvqFragmentIn:          return varyingRegister(operand);\r
1899                 case EvqInvariantVaryingIn:  return varyingRegister(operand);\r
1900                 case EvqInvariantVaryingOut: return varyingRegister(operand);\r
1901                 case EvqUniform:             return uniformRegister(operand);\r
1902                 case EvqIn:                  return temporaryRegister(operand);\r
1903                 case EvqOut:                 return temporaryRegister(operand);\r
1904                 case EvqInOut:               return temporaryRegister(operand);\r
1905                 case EvqConstReadOnly:       return temporaryRegister(operand);\r
1906                 case EvqPosition:            return varyingRegister(operand);\r
1907                 case EvqPointSize:           return varyingRegister(operand);\r
1908                 case EvqInstanceID:          vertexShader->instanceIdDeclared = true; return 0;\r
1909                 case EvqFragCoord:           pixelShader->vPosDeclared = true;  return 0;\r
1910                 case EvqFrontFacing:         pixelShader->vFaceDeclared = true; return 1;\r
1911                 case EvqPointCoord:          return varyingRegister(operand);\r
1912                 case EvqFragColor:           return 0;\r
1913                 case EvqFragData:            return 0;\r
1914                 default: UNREACHABLE(operand->getQualifier());\r
1915                 }\r
1916 \r
1917                 return 0;\r
1918         }\r
1919 \r
1920         int OutputASM::writeMask(TIntermTyped *destination, int index)\r
1921         {\r
1922                 if(destination->getQualifier() == EvqPointSize)\r
1923                 {\r
1924                         return 0x2;   // Point size stored in the y component\r
1925                 }\r
1926 \r
1927                 return 0xF >> (4 - registerSize(destination->getType(), index));\r
1928         }\r
1929 \r
1930         int OutputASM::readSwizzle(TIntermTyped *argument, int size)\r
1931         {\r
1932                 if(argument->getQualifier() == EvqPointSize)\r
1933                 {\r
1934                         return 0x55;   // Point size stored in the y component\r
1935                 }\r
1936 \r
1937                 static const unsigned char swizzleSize[5] = {0x00, 0x00, 0x54, 0xA4, 0xE4};   // (void), xxxx, xyyy, xyzz, xyzw\r
1938 \r
1939                 return swizzleSize[size];\r
1940         }\r
1941 \r
1942         // Conservatively checks whether an expression is fast to compute and has no side effects\r
1943         bool OutputASM::trivial(TIntermTyped *expression, int budget)\r
1944         {\r
1945                 if(!expression->isRegister())\r
1946                 {\r
1947                         return false;\r
1948                 }\r
1949 \r
1950                 return cost(expression, budget) >= 0;\r
1951         }\r
1952 \r
1953         // Returns the remaining computing budget (if < 0 the expression is too expensive or has side effects)\r
1954         int OutputASM::cost(TIntermNode *expression, int budget)\r
1955         {\r
1956                 if(budget < 0)\r
1957                 {\r
1958                         return budget;\r
1959                 }\r
1960 \r
1961                 if(expression->getAsSymbolNode())\r
1962                 {\r
1963                         return budget;\r
1964                 }\r
1965                 else if(expression->getAsConstantUnion())\r
1966                 {\r
1967                         return budget;\r
1968                 }\r
1969                 else if(expression->getAsBinaryNode())\r
1970                 {\r
1971                         TIntermBinary *binary = expression->getAsBinaryNode();\r
1972 \r
1973                         switch(binary->getOp())\r
1974                         {\r
1975                         case EOpVectorSwizzle:\r
1976                         case EOpIndexDirect:\r
1977                         case EOpIndexDirectStruct:\r
1978                                 return cost(binary->getLeft(), budget - 0);\r
1979                         case EOpAdd:\r
1980                         case EOpSub:\r
1981                         case EOpMul:\r
1982                                 return cost(binary->getLeft(), cost(binary->getRight(), budget - 1));\r
1983                         default:\r
1984                                 return -1;\r
1985                         }\r
1986                 }\r
1987                 else if(expression->getAsUnaryNode())\r
1988                 {\r
1989                         TIntermUnary *unary = expression->getAsUnaryNode();\r
1990 \r
1991                         switch(unary->getOp())\r
1992                         {\r
1993                         case EOpAbs:\r
1994                         case EOpNegative:\r
1995                                 return cost(unary->getOperand(), budget - 1);\r
1996                         default:\r
1997                                 return -1;\r
1998                         }\r
1999                 }\r
2000                 else if(expression->getAsSelectionNode())\r
2001                 {\r
2002                         TIntermSelection *selection = expression->getAsSelectionNode();\r
2003 \r
2004                         if(selection->usesTernaryOperator())\r
2005                         {\r
2006                                 TIntermTyped *condition = selection->getCondition();\r
2007                                 TIntermNode *trueBlock = selection->getTrueBlock();\r
2008                                 TIntermNode *falseBlock = selection->getFalseBlock();\r
2009                                 TIntermConstantUnion *constantCondition = condition->getAsConstantUnion();\r
2010 \r
2011                                 if(constantCondition)\r
2012                                 {\r
2013                                         bool trueCondition = constantCondition->getUnionArrayPointer()->getBConst();\r
2014 \r
2015                                         if(trueCondition)\r
2016                                         {\r
2017                                                 return cost(trueBlock, budget - 0);\r
2018                                         }\r
2019                                         else\r
2020                                         {\r
2021                                                 return cost(falseBlock, budget - 0);\r
2022                                         }\r
2023                                 }\r
2024                                 else\r
2025                                 {\r
2026                                         return cost(trueBlock, cost(falseBlock, budget - 2));\r
2027                                 }\r
2028                         }\r
2029                 }\r
2030 \r
2031                 return -1;\r
2032         }\r
2033 \r
2034         const Function *OutputASM::findFunction(const TString &name)\r
2035         {\r
2036                 for(unsigned int f = 0; f < functionArray.size(); f++)\r
2037                 {\r
2038                         if(functionArray[f].name == name)\r
2039                         {\r
2040                                 return &functionArray[f];\r
2041                         }\r
2042                 }\r
2043 \r
2044                 return 0;\r
2045         }\r
2046         \r
2047         int OutputASM::temporaryRegister(TIntermTyped *temporary)\r
2048         {\r
2049                 return allocate(temporaries, temporary);\r
2050         }\r
2051 \r
2052         int OutputASM::varyingRegister(TIntermTyped *varying)\r
2053         {\r
2054                 int var = lookup(varyings, varying);\r
2055 \r
2056                 if(var == -1)\r
2057                 {\r
2058                         var = allocate(varyings, varying);\r
2059                         int componentCount = varying->getNominalSize();\r
2060                         int registerCount = varying->totalRegisterCount();\r
2061 \r
2062                         if(pixelShader)\r
2063                         {\r
2064                                 if((var + registerCount) > sw::PixelShader::MAX_INPUT_VARYINGS)\r
2065                                 {\r
2066                                         mContext.error(varying->getLine(), "Varyings packing failed: Too many varyings", "fragment shader");\r
2067                                         return 0;\r
2068                                 }\r
2069 \r
2070                                 if(varying->getQualifier() == EvqPointCoord)\r
2071                                 {\r
2072                                         ASSERT(varying->isRegister());\r
2073                                         if(componentCount >= 1) pixelShader->semantic[var][0] = sw::Shader::Semantic(sw::Shader::USAGE_TEXCOORD, var);\r
2074                                         if(componentCount >= 2) pixelShader->semantic[var][1] = sw::Shader::Semantic(sw::Shader::USAGE_TEXCOORD, var);\r
2075                                         if(componentCount >= 3) pixelShader->semantic[var][2] = sw::Shader::Semantic(sw::Shader::USAGE_TEXCOORD, var);\r
2076                                         if(componentCount >= 4) pixelShader->semantic[var][3] = sw::Shader::Semantic(sw::Shader::USAGE_TEXCOORD, var);\r
2077                                 }\r
2078                                 else\r
2079                                 {\r
2080                                         for(int i = 0; i < varying->totalRegisterCount(); i++)\r
2081                                         {\r
2082                                                 if(componentCount >= 1) pixelShader->semantic[var + i][0] = sw::Shader::Semantic(sw::Shader::USAGE_COLOR, var + i);\r
2083                                                 if(componentCount >= 2) pixelShader->semantic[var + i][1] = sw::Shader::Semantic(sw::Shader::USAGE_COLOR, var + i);\r
2084                                                 if(componentCount >= 3) pixelShader->semantic[var + i][2] = sw::Shader::Semantic(sw::Shader::USAGE_COLOR, var + i);\r
2085                                                 if(componentCount >= 4) pixelShader->semantic[var + i][3] = sw::Shader::Semantic(sw::Shader::USAGE_COLOR, var + i);\r
2086                                         }\r
2087                                 }\r
2088                         }\r
2089                         else if(vertexShader)\r
2090                         {\r
2091                                 if((var + registerCount) > sw::VertexShader::MAX_OUTPUT_VARYINGS)\r
2092                                 {\r
2093                                         mContext.error(varying->getLine(), "Varyings packing failed: Too many varyings", "vertex shader");\r
2094                                         return 0;\r
2095                                 }\r
2096 \r
2097                                 if(varying->getQualifier() == EvqPosition)\r
2098                                 {\r
2099                                         ASSERT(varying->isRegister());\r
2100                                         vertexShader->output[var][0] = sw::Shader::Semantic(sw::Shader::USAGE_POSITION, 0);\r
2101                                         vertexShader->output[var][1] = sw::Shader::Semantic(sw::Shader::USAGE_POSITION, 0);\r
2102                                         vertexShader->output[var][2] = sw::Shader::Semantic(sw::Shader::USAGE_POSITION, 0);\r
2103                                         vertexShader->output[var][3] = sw::Shader::Semantic(sw::Shader::USAGE_POSITION, 0);\r
2104                                         vertexShader->positionRegister = var;\r
2105                                 }\r
2106                                 else if(varying->getQualifier() == EvqPointSize)\r
2107                                 {\r
2108                                         ASSERT(varying->isRegister());\r
2109                                         vertexShader->output[var][0] = sw::Shader::Semantic(sw::Shader::USAGE_PSIZE, 0);\r
2110                                         vertexShader->output[var][1] = sw::Shader::Semantic(sw::Shader::USAGE_PSIZE, 0);\r
2111                                         vertexShader->output[var][2] = sw::Shader::Semantic(sw::Shader::USAGE_PSIZE, 0);\r
2112                                         vertexShader->output[var][3] = sw::Shader::Semantic(sw::Shader::USAGE_PSIZE, 0);\r
2113                                         vertexShader->pointSizeRegister = var;\r
2114                                 }\r
2115                                 else\r
2116                                 {\r
2117                                         // Semantic indexes for user varyings will be assigned during program link to match the pixel shader\r
2118                                 }\r
2119                         }\r
2120                         else UNREACHABLE(0);\r
2121 \r
2122                         declareVarying(varying, var);\r
2123                 }\r
2124 \r
2125                 return var;\r
2126         }\r
2127 \r
2128         void OutputASM::declareVarying(TIntermTyped *varying, int reg)\r
2129         {\r
2130                 if(varying->getQualifier() != EvqPointCoord)   // gl_PointCoord does not need linking\r
2131                 {\r
2132                         const TType &type = varying->getType();\r
2133                         const char *name = varying->getAsSymbolNode()->getSymbol().c_str();\r
2134                         VaryingList &activeVaryings = shaderObject->varyings;\r
2135                         \r
2136                         // Check if this varying has been declared before without having a register assigned\r
2137                         for(VaryingList::iterator v = activeVaryings.begin(); v != activeVaryings.end(); v++)\r
2138                         {\r
2139                                 if(v->name == name)\r
2140                                 {\r
2141                                         if(reg >= 0)\r
2142                                         {\r
2143                                                 ASSERT(v->reg < 0 || v->reg == reg);\r
2144                                                 v->reg = reg;\r
2145                                         }\r
2146 \r
2147                                         return;\r
2148                                 }\r
2149                         }\r
2150                         \r
2151                         activeVaryings.push_back(glsl::Varying(glVariableType(type), name, varying->getArraySize(), reg, 0));\r
2152                 }\r
2153         }\r
2154 \r
2155         int OutputASM::uniformRegister(TIntermTyped *uniform)\r
2156         {\r
2157                 const TType &type = uniform->getType();\r
2158                 ASSERT(!IsSampler(type.getBasicType()));\r
2159                 TIntermSymbol *symbol = uniform->getAsSymbolNode();\r
2160                 ASSERT(symbol);\r
2161 \r
2162                 if(symbol)\r
2163                 {\r
2164                         int index = lookup(uniforms, uniform);\r
2165 \r
2166                         if(index == -1)\r
2167                         {\r
2168                                 index = allocate(uniforms, uniform);\r
2169                                 const TString &name = symbol->getSymbol().c_str();\r
2170 \r
2171                                 declareUniform(type, name, index);\r
2172                         }\r
2173 \r
2174                         return index;\r
2175                 }\r
2176 \r
2177                 return 0;\r
2178         }\r
2179 \r
2180         int OutputASM::attributeRegister(TIntermTyped *attribute)\r
2181         {\r
2182                 ASSERT(!attribute->isArray());\r
2183                 ASSERT(attribute->getBasicType() == EbtFloat);\r
2184 \r
2185                 int index = lookup(attributes, attribute);\r
2186 \r
2187                 if(index == -1)\r
2188                 {\r
2189                         TIntermSymbol *symbol = attribute->getAsSymbolNode();\r
2190                         ASSERT(symbol);\r
2191 \r
2192                         if(symbol)\r
2193                         {\r
2194                                 index = allocate(attributes, attribute);\r
2195                                 const TType &type = attribute->getType();\r
2196                                 int registerCount = attribute->totalRegisterCount();\r
2197 \r
2198                                 if(vertexShader && (index + registerCount) <= sw::VertexShader::MAX_INPUT_ATTRIBUTES)\r
2199                                 {\r
2200                                         for(int i = 0; i < registerCount; i++)\r
2201                                         {\r
2202                                                 vertexShader->input[index + i] = sw::Shader::Semantic(sw::Shader::USAGE_TEXCOORD, index + i);\r
2203                                         }\r
2204                                 }\r
2205 \r
2206                                 ActiveAttributes &activeAttributes = shaderObject->activeAttributes;\r
2207 \r
2208                                 const char *name = symbol->getSymbol().c_str();\r
2209                                 activeAttributes.push_back(Attribute(glVariableType(type), name, 0, index));\r
2210                         }\r
2211                 }\r
2212 \r
2213                 return index;\r
2214         }\r
2215 \r
2216         int OutputASM::samplerRegister(TIntermTyped *sampler)\r
2217         {\r
2218                 ASSERT(IsSampler(sampler->getType().getBasicType()));\r
2219                 TIntermSymbol *symbol = sampler->getAsSymbolNode();\r
2220                 TIntermBinary *binary = sampler->getAsBinaryNode();\r
2221 \r
2222                 if(symbol)\r
2223                 {\r
2224                         return samplerRegister(symbol);\r
2225                 }\r
2226                 else if(binary)\r
2227                 {\r
2228                         ASSERT(binary->getOp() == EOpIndexDirect || binary->getOp() == EOpIndexIndirect || binary->getOp() == EOpIndexDirectStruct);\r
2229 \r
2230                         return samplerRegister(binary->getLeft());   // Index added later\r
2231                 }\r
2232                 else UNREACHABLE(0);\r
2233 \r
2234                 return 0;\r
2235         }\r
2236 \r
2237         int OutputASM::samplerRegister(TIntermSymbol *sampler)\r
2238         {\r
2239                 const TType &type = sampler->getType();\r
2240                 ASSERT(IsSampler(type.getBasicType()) || type.getStruct());   // Structures can contain samplers\r
2241 \r
2242                 int index = lookup(samplers, sampler);\r
2243 \r
2244                 if(index == -1)\r
2245                 {\r
2246                         index = allocate(samplers, sampler);\r
2247 \r
2248                         if(sampler->getQualifier() == EvqUniform)\r
2249                         {\r
2250                                 const char *name = sampler->getSymbol().c_str();\r
2251                                 declareUniform(type, name, index);\r
2252                         }\r
2253                 }\r
2254 \r
2255                 return index;\r
2256         }\r
2257 \r
2258         int OutputASM::lookup(VariableArray &list, TIntermTyped *variable)\r
2259         {\r
2260                 for(unsigned int i = 0; i < list.size(); i++)\r
2261                 {\r
2262                         if(list[i] == variable)\r
2263                         {\r
2264                                 return i;   // Pointer match\r
2265                         }\r
2266                 }\r
2267 \r
2268                 TIntermSymbol *varSymbol = variable->getAsSymbolNode();\r
2269 \r
2270                 if(varSymbol)\r
2271                 {\r
2272                         for(unsigned int i = 0; i < list.size(); i++)\r
2273                         {\r
2274                                 if(list[i])\r
2275                                 {\r
2276                                         TIntermSymbol *listSymbol = list[i]->getAsSymbolNode();\r
2277 \r
2278                                         if(listSymbol)\r
2279                                         {\r
2280                                                 if(listSymbol->getId() == varSymbol->getId())\r
2281                                                 {\r
2282                                                         ASSERT(listSymbol->getSymbol() == varSymbol->getSymbol());\r
2283                                                         ASSERT(listSymbol->getType() == varSymbol->getType());\r
2284                                                         ASSERT(listSymbol->getQualifier() == varSymbol->getQualifier());\r
2285 \r
2286                                                         return i;\r
2287                                                 }\r
2288                                         }\r
2289                                 }\r
2290                         }\r
2291                 }\r
2292 \r
2293                 return -1;\r
2294         }\r
2295 \r
2296         int OutputASM::allocate(VariableArray &list, TIntermTyped *variable)\r
2297         {\r
2298                 int index = lookup(list, variable);\r
2299 \r
2300                 if(index == -1)\r
2301                 {\r
2302                         unsigned int registerCount = variable->totalRegisterCount();\r
2303 \r
2304                         for(unsigned int i = 0; i < list.size(); i++)\r
2305                         {\r
2306                                 if(list[i] == 0)\r
2307                                 {\r
2308                                         unsigned int j = 1;\r
2309                                         for( ; j < registerCount && (i + j) < list.size(); j++)\r
2310                                         {\r
2311                                                 if(list[i + j] != 0)\r
2312                                                 {\r
2313                                                         break;\r
2314                                                 }\r
2315                                         }\r
2316 \r
2317                                         if(j == registerCount)   // Found free slots\r
2318                                         {\r
2319                                                 for(unsigned int j = 0; j < registerCount; j++)\r
2320                                                 {\r
2321                                                         list[i + j] = variable;\r
2322                                                 }\r
2323 \r
2324                                                 return i;\r
2325                                         }\r
2326                                 }\r
2327                         }\r
2328 \r
2329                         index = list.size();\r
2330 \r
2331                         for(unsigned int i = 0; i < registerCount; i++)\r
2332                         {\r
2333                                 list.push_back(variable);\r
2334                         }\r
2335                 }\r
2336 \r
2337                 return index;\r
2338         }\r
2339 \r
2340         void OutputASM::free(VariableArray &list, TIntermTyped *variable)\r
2341         {\r
2342                 int index = lookup(list, variable);\r
2343 \r
2344                 if(index >= 0)\r
2345                 {\r
2346                         list[index] = 0;\r
2347                 }\r
2348         }\r
2349 \r
2350         void OutputASM::declareUniform(const TType &type, const TString &name, int index)\r
2351         {\r
2352                 const TStructure *structure = type.getStruct();\r
2353                 ActiveUniforms &activeUniforms = shaderObject->activeUniforms;\r
2354 \r
2355                 if(!structure)\r
2356                 {\r
2357                         activeUniforms.push_back(Uniform(glVariableType(type), glVariablePrecision(type), name.c_str(), type.getArraySize(), index));\r
2358 \r
2359                         if(isSamplerRegister(type))\r
2360                         {\r
2361                                 for(int i = 0; i < type.totalRegisterCount(); i++)\r
2362                                 {\r
2363                                         shader->declareSampler(index + i);\r
2364                                 }\r
2365                         }\r
2366                 }\r
2367                 else\r
2368                 {\r
2369                         const TFieldList& fields = structure->fields();\r
2370                         if(type.isArray())\r
2371                         {\r
2372                                 int elementIndex = index;\r
2373 \r
2374                                 for(int i = 0; i < type.getArraySize(); i++)\r
2375                                 {\r
2376                                         for(size_t j = 0; j < fields.size(); j++)\r
2377                                         {\r
2378                                                 const TType &fieldType = *(fields[j]->type());\r
2379                                                 const TString &fieldName = fields[j]->name();\r
2380 \r
2381                                                 const TString uniformName = name + "[" + str(i) + "]." + fieldName;\r
2382                                                 declareUniform(fieldType, uniformName, elementIndex);\r
2383                                                 elementIndex += fieldType.totalRegisterCount();\r
2384                                         }\r
2385                                 }\r
2386                         }\r
2387                         else\r
2388                         {\r
2389                                 int fieldIndex = index;\r
2390 \r
2391                                 for(size_t i = 0; i < fields.size(); i++)\r
2392                                 {\r
2393                                         const TType &fieldType = *(fields[i]->type());\r
2394                                         const TString &fieldName = fields[i]->name();\r
2395 \r
2396                                         const TString uniformName = name + "." + fieldName;\r
2397                                         declareUniform(fieldType, uniformName, fieldIndex);\r
2398                                         fieldIndex += fieldType.totalRegisterCount();\r
2399                                 }\r
2400                         }\r
2401                 }\r
2402         }\r
2403 \r
2404         GLenum OutputASM::glVariableType(const TType &type)\r
2405         {\r
2406                 switch(type.getBasicType())\r
2407                 {\r
2408                 case EbtFloat:\r
2409                         if(type.isScalar())\r
2410                         {\r
2411                                 return GL_FLOAT;\r
2412                         }\r
2413                         else if(type.isVector())\r
2414                         {\r
2415                                 switch(type.getNominalSize())\r
2416                                 {\r
2417                                 case 2: return GL_FLOAT_VEC2;\r
2418                                 case 3: return GL_FLOAT_VEC3;\r
2419                                 case 4: return GL_FLOAT_VEC4;\r
2420                                 default: UNREACHABLE(type.getNominalSize());\r
2421                                 }\r
2422                         }\r
2423                         else if(type.isMatrix())\r
2424                         {\r
2425                                 switch(type.getNominalSize())\r
2426                                 {\r
2427                                 case 2:\r
2428                                         switch(type.getSecondarySize())\r
2429                                         {\r
2430                                         case 2: return GL_FLOAT_MAT2;\r
2431                                         case 3: return GL_FLOAT_MAT2x3;\r
2432                                         case 4: return GL_FLOAT_MAT2x4;\r
2433                                         default: UNREACHABLE(type.getSecondarySize());\r
2434                                         }\r
2435                                 case 3:\r
2436                                         switch(type.getSecondarySize())\r
2437                                         {\r
2438                                         case 2: return GL_FLOAT_MAT3x2;\r
2439                                         case 3: return GL_FLOAT_MAT3;\r
2440                                         case 4: return GL_FLOAT_MAT3x4;\r
2441                                         default: UNREACHABLE(type.getSecondarySize());\r
2442                                         }\r
2443                                 case 4:\r
2444                                         switch(type.getSecondarySize())\r
2445                                         {\r
2446                                         case 2: return GL_FLOAT_MAT4x2;\r
2447                                         case 3: return GL_FLOAT_MAT4x3;\r
2448                                         case 4: return GL_FLOAT_MAT4;\r
2449                                         default: UNREACHABLE(type.getSecondarySize());\r
2450                                         }\r
2451                                 default: UNREACHABLE(type.getNominalSize());\r
2452                                 }\r
2453                         }\r
2454                         else UNREACHABLE(0);\r
2455                         break;\r
2456                 case EbtInt:\r
2457                         if(type.isScalar())\r
2458                         {\r
2459                                 return GL_INT;\r
2460                         }\r
2461                         else if(type.isVector())\r
2462                         {\r
2463                                 switch(type.getNominalSize())\r
2464                                 {\r
2465                                 case 2: return GL_INT_VEC2;\r
2466                                 case 3: return GL_INT_VEC3;\r
2467                                 case 4: return GL_INT_VEC4;\r
2468                                 default: UNREACHABLE(type.getNominalSize());\r
2469                                 }\r
2470                         }\r
2471                         else UNREACHABLE(0);\r
2472                         break;\r
2473                 case EbtUInt:\r
2474                         if(type.isScalar())\r
2475                         {\r
2476                                 return GL_UNSIGNED_INT;\r
2477                         }\r
2478                         else if(type.isVector())\r
2479                         {\r
2480                                 switch(type.getNominalSize())\r
2481                                 {\r
2482                                 case 2: return GL_UNSIGNED_INT_VEC2;\r
2483                                 case 3: return GL_UNSIGNED_INT_VEC3;\r
2484                                 case 4: return GL_UNSIGNED_INT_VEC4;\r
2485                                 default: UNREACHABLE(type.getNominalSize());\r
2486                                 }\r
2487                         }\r
2488                         else UNREACHABLE(0);\r
2489                         break;\r
2490                 case EbtBool:\r
2491                         if(type.isScalar())\r
2492                         {\r
2493                                 return GL_BOOL;\r
2494                         }\r
2495                         else if(type.isVector())\r
2496                         {\r
2497                                 switch(type.getNominalSize())\r
2498                                 {\r
2499                                 case 2: return GL_BOOL_VEC2;\r
2500                                 case 3: return GL_BOOL_VEC3;\r
2501                                 case 4: return GL_BOOL_VEC4;\r
2502                                 default: UNREACHABLE(type.getNominalSize());\r
2503                                 }\r
2504                         }\r
2505                         else UNREACHABLE(0);\r
2506                         break;\r
2507                 case EbtSampler2D:\r
2508                 case EbtISampler2D:\r
2509                 case EbtUSampler2D:\r
2510                         return GL_SAMPLER_2D;\r
2511                 case EbtSamplerCube:\r
2512                 case EbtISamplerCube:\r
2513                 case EbtUSamplerCube:\r
2514                         return GL_SAMPLER_CUBE;\r
2515                 case EbtSamplerExternalOES:\r
2516                         return GL_SAMPLER_EXTERNAL_OES;\r
2517                 case EbtSampler3D:\r
2518                 case EbtISampler3D:\r
2519                 case EbtUSampler3D:\r
2520                         return GL_SAMPLER_3D_OES;\r
2521                 case EbtSampler2DArray:\r
2522                 case EbtISampler2DArray:\r
2523                 case EbtUSampler2DArray:\r
2524                         return GL_SAMPLER_2D_ARRAY;\r
2525                 default:\r
2526                         UNREACHABLE(type.getBasicType());\r
2527                         break;\r
2528                 }\r
2529 \r
2530                 return GL_NONE;\r
2531         }\r
2532 \r
2533         GLenum OutputASM::glVariablePrecision(const TType &type)\r
2534         {\r
2535                 if(type.getBasicType() == EbtFloat)\r
2536                 {\r
2537                         switch(type.getPrecision())\r
2538                         {\r
2539                         case EbpHigh:   return GL_HIGH_FLOAT;\r
2540                         case EbpMedium: return GL_MEDIUM_FLOAT;\r
2541                         case EbpLow:    return GL_LOW_FLOAT;\r
2542                         case EbpUndefined:\r
2543                                 // Should be defined as the default precision by the parser\r
2544                         default: UNREACHABLE(type.getPrecision());\r
2545                         }\r
2546                 }\r
2547                 else if(type.getBasicType() == EbtInt)\r
2548                 {\r
2549                         switch(type.getPrecision())\r
2550                         {\r
2551                         case EbpHigh:   return GL_HIGH_INT;\r
2552                         case EbpMedium: return GL_MEDIUM_INT;\r
2553                         case EbpLow:    return GL_LOW_INT;\r
2554                         case EbpUndefined:\r
2555                                 // Should be defined as the default precision by the parser\r
2556                         default: UNREACHABLE(type.getPrecision());\r
2557                         }\r
2558                 }\r
2559 \r
2560                 // Other types (boolean, sampler) don't have a precision\r
2561                 return GL_NONE;\r
2562         }\r
2563 \r
2564         int OutputASM::dim(TIntermNode *v)\r
2565         {\r
2566                 TIntermTyped *vector = v->getAsTyped();\r
2567                 ASSERT(vector && vector->isRegister());\r
2568                 return vector->getNominalSize();\r
2569         }\r
2570 \r
2571         int OutputASM::dim2(TIntermNode *m)\r
2572         {\r
2573                 TIntermTyped *matrix = m->getAsTyped();\r
2574                 ASSERT(matrix && matrix->isMatrix() && !matrix->isArray());\r
2575                 return matrix->getSecondarySize();\r
2576         }\r
2577 \r
2578         // Returns ~0 if no loop count could be determined\r
2579         unsigned int OutputASM::loopCount(TIntermLoop *node)\r
2580         {\r
2581                 // Parse loops of the form:\r
2582                 // for(int index = initial; index [comparator] limit; index += increment)\r
2583                 TIntermSymbol *index = 0;\r
2584                 TOperator comparator = EOpNull;\r
2585                 int initial = 0;\r
2586                 int limit = 0;\r
2587                 int increment = 0;\r
2588 \r
2589                 // Parse index name and intial value\r
2590                 if(node->getInit())\r
2591                 {\r
2592                         TIntermAggregate *init = node->getInit()->getAsAggregate();\r
2593 \r
2594                         if(init)\r
2595                         {\r
2596                                 TIntermSequence &sequence = init->getSequence();\r
2597                                 TIntermTyped *variable = sequence[0]->getAsTyped();\r
2598 \r
2599                                 if(variable && variable->getQualifier() == EvqTemporary)\r
2600                                 {\r
2601                                         TIntermBinary *assign = variable->getAsBinaryNode();\r
2602 \r
2603                                         if(assign->getOp() == EOpInitialize)\r
2604                                         {\r
2605                                                 TIntermSymbol *symbol = assign->getLeft()->getAsSymbolNode();\r
2606                                                 TIntermConstantUnion *constant = assign->getRight()->getAsConstantUnion();\r
2607 \r
2608                                                 if(symbol && constant)\r
2609                                                 {\r
2610                                                         if(constant->getBasicType() == EbtInt && constant->getNominalSize() == 1)\r
2611                                                         {\r
2612                                                                 index = symbol;\r
2613                                                                 initial = constant->getUnionArrayPointer()[0].getIConst();\r
2614                                                         }\r
2615                                                 }\r
2616                                         }\r
2617                                 }\r
2618                         }\r
2619                 }\r
2620 \r
2621                 // Parse comparator and limit value\r
2622                 if(index && node->getCondition())\r
2623                 {\r
2624                         TIntermBinary *test = node->getCondition()->getAsBinaryNode();\r
2625 \r
2626                         if(test && test->getLeft()->getAsSymbolNode()->getId() == index->getId())\r
2627                         {\r
2628                                 TIntermConstantUnion *constant = test->getRight()->getAsConstantUnion();\r
2629 \r
2630                                 if(constant)\r
2631                                 {\r
2632                                         if(constant->getBasicType() == EbtInt && constant->getNominalSize() == 1)\r
2633                                         {\r
2634                                                 comparator = test->getOp();\r
2635                                                 limit = constant->getUnionArrayPointer()[0].getIConst();\r
2636                                         }\r
2637                                 }\r
2638                         }\r
2639                 }\r
2640 \r
2641                 // Parse increment\r
2642                 if(index && comparator != EOpNull && node->getExpression())\r
2643                 {\r
2644                         TIntermBinary *binaryTerminal = node->getExpression()->getAsBinaryNode();\r
2645                         TIntermUnary *unaryTerminal = node->getExpression()->getAsUnaryNode();\r
2646 \r
2647                         if(binaryTerminal)\r
2648                         {\r
2649                                 TOperator op = binaryTerminal->getOp();\r
2650                                 TIntermConstantUnion *constant = binaryTerminal->getRight()->getAsConstantUnion();\r
2651 \r
2652                                 if(constant)\r
2653                                 {\r
2654                                         if(constant->getBasicType() == EbtInt && constant->getNominalSize() == 1)\r
2655                                         {\r
2656                                                 int value = constant->getUnionArrayPointer()[0].getIConst();\r
2657 \r
2658                                                 switch(op)\r
2659                                                 {\r
2660                                                 case EOpAddAssign: increment = value;  break;\r
2661                                                 case EOpSubAssign: increment = -value; break;\r
2662                                                 default: UNIMPLEMENTED();\r
2663                                                 }\r
2664                                         }\r
2665                                 }\r
2666                         }\r
2667                         else if(unaryTerminal)\r
2668                         {\r
2669                                 TOperator op = unaryTerminal->getOp();\r
2670 \r
2671                                 switch(op)\r
2672                                 {\r
2673                                 case EOpPostIncrement: increment = 1;  break;\r
2674                                 case EOpPostDecrement: increment = -1; break;\r
2675                                 case EOpPreIncrement:  increment = 1;  break;\r
2676                                 case EOpPreDecrement:  increment = -1; break;\r
2677                                 default: UNIMPLEMENTED();\r
2678                                 }\r
2679                         }\r
2680                 }\r
2681 \r
2682                 if(index && comparator != EOpNull && increment != 0)\r
2683                 {\r
2684                         if(comparator == EOpLessThanEqual)\r
2685                         {\r
2686                                 comparator = EOpLessThan;\r
2687                                 limit += 1;\r
2688                         }\r
2689 \r
2690                         if(comparator == EOpLessThan)\r
2691                         {\r
2692                                 int iterations = (limit - initial) / increment;\r
2693 \r
2694                                 if(iterations <= 0)\r
2695                                 {\r
2696                                         iterations = 0;\r
2697                                 }\r
2698 \r
2699                                 return iterations;\r
2700                         }\r
2701                         else UNIMPLEMENTED();   // Falls through\r
2702                 }\r
2703 \r
2704                 return ~0;\r
2705         }\r
2706 \r
2707         bool DetectLoopDiscontinuity::traverse(TIntermNode *node)\r
2708         {\r
2709                 loopDepth = 0;\r
2710                 loopDiscontinuity = false;\r
2711                 \r
2712                 node->traverse(this);\r
2713                 \r
2714                 return loopDiscontinuity;\r
2715         }\r
2716 \r
2717         bool DetectLoopDiscontinuity::visitLoop(Visit visit, TIntermLoop *loop)\r
2718         {\r
2719                 if(visit == PreVisit)\r
2720                 {\r
2721                         loopDepth++;\r
2722                 }\r
2723                 else if(visit == PostVisit)\r
2724                 {\r
2725                         loopDepth++;\r
2726                 }\r
2727 \r
2728                 return true;\r
2729         }\r
2730 \r
2731         bool DetectLoopDiscontinuity::visitBranch(Visit visit, TIntermBranch *node)\r
2732         {\r
2733                 if(loopDiscontinuity)\r
2734                 {\r
2735                         return false;\r
2736                 }\r
2737 \r
2738                 if(!loopDepth)\r
2739                 {\r
2740                         return true;\r
2741                 }\r
2742         \r
2743                 switch(node->getFlowOp())\r
2744                 {\r
2745                 case EOpKill:\r
2746                         break;\r
2747                 case EOpBreak:\r
2748                 case EOpContinue:\r
2749                 case EOpReturn:\r
2750                         loopDiscontinuity = true;\r
2751                         break;\r
2752                 default: UNREACHABLE(node->getFlowOp());\r
2753                 }\r
2754 \r
2755                 return !loopDiscontinuity;\r
2756         }\r
2757 \r
2758         bool DetectLoopDiscontinuity::visitAggregate(Visit visit, TIntermAggregate *node)\r
2759         {\r
2760                 return !loopDiscontinuity;\r
2761         }\r
2762 }\r