OSDN Git Service

Fixed some unary operators
[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 #include "Common/Math.hpp"\r
14 \r
15 #include "common/debug.h"\r
16 #include "InfoSink.h"\r
17 \r
18 #include "libGLESv2/Shader.h"\r
19 \r
20 #include <GLES2/gl2.h>\r
21 #include <GLES2/gl2ext.h>\r
22 #include <GLES3/gl3.h>\r
23 \r
24 namespace glsl\r
25 {\r
26         // Integer to TString conversion\r
27         TString str(int i)\r
28         {\r
29                 char buffer[20];\r
30                 sprintf(buffer, "%d", i);\r
31                 return buffer;\r
32         }\r
33 \r
34         class Temporary : public TIntermSymbol\r
35         {\r
36         public:\r
37                 Temporary(OutputASM *assembler) : TIntermSymbol(TSymbolTableLevel::nextUniqueId(), "tmp", TType(EbtFloat, EbpHigh, EvqTemporary, 4, 1, false)), assembler(assembler)\r
38                 {\r
39                 }\r
40 \r
41                 ~Temporary()\r
42                 {\r
43                         assembler->freeTemporary(this);\r
44                 }\r
45 \r
46         private:\r
47                 OutputASM *const assembler;\r
48         };\r
49 \r
50         class Constant : public TIntermConstantUnion\r
51         {\r
52         public:\r
53                 Constant(float x, float y, float z, float w) : TIntermConstantUnion(constants, TType(EbtFloat, EbpHigh, EvqConstExpr, 4, 1, false))\r
54                 {\r
55                         constants[0].setFConst(x);\r
56                         constants[1].setFConst(y);\r
57                         constants[2].setFConst(z);\r
58                         constants[3].setFConst(w);\r
59                 }\r
60 \r
61                 Constant(bool b) : TIntermConstantUnion(constants, TType(EbtBool, EbpHigh, EvqConstExpr, 1, 1, false))\r
62                 {\r
63                         constants[0].setBConst(b);\r
64                 }\r
65 \r
66                 Constant(int i) : TIntermConstantUnion(constants, TType(EbtInt, EbpHigh, EvqConstExpr, 1, 1, false))\r
67                 {\r
68                         constants[0].setIConst(i);\r
69                 }\r
70 \r
71                 ~Constant()\r
72                 {\r
73                 }\r
74 \r
75         private:\r
76                 ConstantUnion constants[4];\r
77         };\r
78 \r
79         Uniform::Uniform(GLenum type, GLenum precision, const std::string &name, int arraySize, int registerIndex, int blockId, const BlockMemberInfo& blockMemberInfo) :\r
80                 type(type), precision(precision), name(name), arraySize(arraySize), registerIndex(registerIndex), blockId(blockId), blockInfo(blockMemberInfo)\r
81         {\r
82         }\r
83 \r
84         UniformBlock::UniformBlock(const std::string& name, unsigned int dataSize, unsigned int arraySize,\r
85                                    TLayoutBlockStorage layout, bool isRowMajorLayout, int registerIndex, int blockId) :\r
86                 name(name), dataSize(dataSize), arraySize(arraySize), layout(layout),\r
87                 isRowMajorLayout(isRowMajorLayout), registerIndex(registerIndex), blockId(blockId)\r
88         {\r
89         }\r
90 \r
91         BlockLayoutEncoder::BlockLayoutEncoder(bool rowMajor)\r
92                 : mCurrentOffset(0), isRowMajor(rowMajor)\r
93         {\r
94         }\r
95 \r
96         BlockMemberInfo BlockLayoutEncoder::encodeType(const TType &type)\r
97         {\r
98                 int arrayStride;\r
99                 int matrixStride;\r
100 \r
101                 getBlockLayoutInfo(type, type.getArraySize(), isRowMajor, &arrayStride, &matrixStride);\r
102 \r
103                 const BlockMemberInfo memberInfo(static_cast<int>(mCurrentOffset * BytesPerComponent),\r
104                                                  static_cast<int>(arrayStride * BytesPerComponent),\r
105                                                  static_cast<int>(matrixStride * BytesPerComponent),\r
106                                                  (matrixStride > 0) && isRowMajor);\r
107 \r
108                 advanceOffset(type, type.getArraySize(), isRowMajor, arrayStride, matrixStride);\r
109 \r
110                 return memberInfo;\r
111         }\r
112 \r
113         // static\r
114         size_t BlockLayoutEncoder::getBlockRegister(const BlockMemberInfo &info)\r
115         {\r
116                 return (info.offset / BytesPerComponent) / ComponentsPerRegister;\r
117         }\r
118 \r
119         // static\r
120         size_t BlockLayoutEncoder::getBlockRegisterElement(const BlockMemberInfo &info)\r
121         {\r
122                 return (info.offset / BytesPerComponent) % ComponentsPerRegister;\r
123         }\r
124 \r
125         void BlockLayoutEncoder::nextRegister()\r
126         {\r
127                 mCurrentOffset = sw::align(mCurrentOffset, ComponentsPerRegister);\r
128         }\r
129 \r
130         Std140BlockEncoder::Std140BlockEncoder(bool rowMajor) : BlockLayoutEncoder(rowMajor)\r
131         {\r
132         }\r
133 \r
134         void Std140BlockEncoder::enterAggregateType()\r
135         {\r
136                 nextRegister();\r
137         }\r
138 \r
139         void Std140BlockEncoder::exitAggregateType()\r
140         {\r
141                 nextRegister();\r
142         }\r
143 \r
144         void Std140BlockEncoder::getBlockLayoutInfo(const TType &type, unsigned int arraySize, bool isRowMajorMatrix, int *arrayStrideOut, int *matrixStrideOut)\r
145         {\r
146                 size_t baseAlignment = 0;\r
147                 int matrixStride = 0;\r
148                 int arrayStride = 0;\r
149 \r
150                 if(type.isMatrix())\r
151                 {\r
152                         baseAlignment = ComponentsPerRegister;\r
153                         matrixStride = ComponentsPerRegister;\r
154 \r
155                         if(arraySize > 0)\r
156                         {\r
157                                 const int numRegisters = isRowMajorMatrix ? type.getSecondarySize() : type.getNominalSize();\r
158                                 arrayStride = ComponentsPerRegister * numRegisters;\r
159                         }\r
160                 }\r
161                 else if(arraySize > 0)\r
162                 {\r
163                         baseAlignment = ComponentsPerRegister;\r
164                         arrayStride = ComponentsPerRegister;\r
165                 }\r
166                 else\r
167                 {\r
168                         const int numComponents = type.getElementSize();\r
169                         baseAlignment = (numComponents == 3 ? 4u : static_cast<size_t>(numComponents));\r
170                 }\r
171 \r
172                 mCurrentOffset = sw::align(mCurrentOffset, baseAlignment);\r
173 \r
174                 *matrixStrideOut = matrixStride;\r
175                 *arrayStrideOut = arrayStride;\r
176         }\r
177 \r
178         void Std140BlockEncoder::advanceOffset(const TType &type, unsigned int arraySize, bool isRowMajorMatrix, int arrayStride, int matrixStride)\r
179         {\r
180                 if(arraySize > 0)\r
181                 {\r
182                         mCurrentOffset += arrayStride * arraySize;\r
183                 }\r
184                 else if(type.isMatrix())\r
185                 {\r
186                         ASSERT(matrixStride == ComponentsPerRegister);\r
187                         const int numRegisters = isRowMajorMatrix ? type.getSecondarySize() : type.getNominalSize();\r
188                         mCurrentOffset += ComponentsPerRegister * numRegisters;\r
189                 }\r
190                 else\r
191                 {\r
192                         mCurrentOffset += type.getElementSize();\r
193                 }\r
194         }\r
195 \r
196         Attribute::Attribute()\r
197         {\r
198                 type = GL_NONE;\r
199                 arraySize = 0;\r
200                 registerIndex = 0;\r
201         }\r
202 \r
203         Attribute::Attribute(GLenum type, const std::string &name, int arraySize, int location, int registerIndex)\r
204         {\r
205                 this->type = type;\r
206                 this->name = name;\r
207                 this->arraySize = arraySize;\r
208                 this->location = location;\r
209                 this->registerIndex = registerIndex;\r
210         }\r
211 \r
212         sw::PixelShader *Shader::getPixelShader() const\r
213         {\r
214                 return 0;\r
215         }\r
216 \r
217         sw::VertexShader *Shader::getVertexShader() const\r
218         {\r
219                 return 0;\r
220         }\r
221 \r
222         OutputASM::TextureFunction::TextureFunction(const TString& nodeName) : method(IMPLICIT), proj(false), offset(false)\r
223         {\r
224                 TString name = TFunction::unmangleName(nodeName);\r
225 \r
226                 if(name == "texture2D" || name == "textureCube" || name == "texture" || name == "texture3D")\r
227                 {\r
228                         method = IMPLICIT;\r
229                 }\r
230                 else if(name == "texture2DProj" || name == "textureProj")\r
231                 {\r
232                         method = IMPLICIT;\r
233                         proj = true;\r
234                 }\r
235                 else if(name == "texture2DLod" || name == "textureCubeLod" || name == "textureLod")\r
236                 {\r
237                         method = LOD;\r
238                 }\r
239                 else if(name == "texture2DProjLod" || name == "textureProjLod")\r
240                 {\r
241                         method = LOD;\r
242                         proj = true;\r
243                 }\r
244                 else if(name == "textureSize")\r
245                 {\r
246                         method = SIZE;\r
247                 }\r
248                 else if(name == "textureOffset")\r
249                 {\r
250                         method = IMPLICIT;\r
251                         offset = true;\r
252                 }\r
253                 else if(name == "textureProjOffset")\r
254                 {\r
255                         method = IMPLICIT;\r
256                         offset = true;\r
257                         proj = true;\r
258                 }\r
259                 else if(name == "textureLodOffset")\r
260                 {\r
261                         method = LOD;\r
262                         offset = true;\r
263                 }\r
264                 else if(name == "textureProjLodOffset")\r
265                 {\r
266                         method = LOD;\r
267                         proj = true;\r
268                         offset = true;\r
269                 }\r
270                 else if(name == "texelFetch")\r
271                 {\r
272                         method = FETCH;\r
273                 }\r
274                 else if(name == "texelFetchOffset")\r
275                 {\r
276                         method = FETCH;\r
277                         offset = true;\r
278                 }\r
279                 else if(name == "textureGrad")\r
280                 {\r
281                         method = GRAD;\r
282                 }\r
283                 else if(name == "textureGradOffset")\r
284                 {\r
285                         method = GRAD;\r
286                         offset = true;\r
287                 }\r
288                 else if(name == "textureProjGrad")\r
289                 {\r
290                         method = GRAD;\r
291                         proj = true;\r
292                 }\r
293                 else if(name == "textureProjGradOffset")\r
294                 {\r
295                         method = GRAD;\r
296                         proj = true;\r
297                         offset = true;\r
298                 }\r
299                 else UNREACHABLE(0);\r
300         }\r
301 \r
302         OutputASM::OutputASM(TParseContext &context, Shader *shaderObject) : TIntermTraverser(true, true, true), shaderObject(shaderObject), mContext(context)\r
303         {\r
304                 shader = 0;\r
305                 pixelShader = 0;\r
306                 vertexShader = 0;\r
307 \r
308                 if(shaderObject)\r
309                 {\r
310                         shader = shaderObject->getShader();\r
311                         pixelShader = shaderObject->getPixelShader();\r
312                         vertexShader = shaderObject->getVertexShader();\r
313                 }\r
314 \r
315                 functionArray.push_back(Function(0, "main(", 0, 0));\r
316                 currentFunction = 0;\r
317                 outputQualifier = EvqOutput; // Set outputQualifier to any value other than EvqFragColor or EvqFragData\r
318         }\r
319 \r
320         OutputASM::~OutputASM()\r
321         {\r
322         }\r
323 \r
324         void OutputASM::output()\r
325         {\r
326                 if(shader)\r
327                 {\r
328                         emitShader(GLOBAL);\r
329 \r
330                         if(functionArray.size() > 1)   // Only call main() when there are other functions\r
331                         {\r
332                                 Instruction *callMain = emit(sw::Shader::OPCODE_CALL);\r
333                                 callMain->dst.type = sw::Shader::PARAMETER_LABEL;\r
334                                 callMain->dst.index = 0;   // main()\r
335 \r
336                                 emit(sw::Shader::OPCODE_RET);\r
337                         }\r
338 \r
339                         emitShader(FUNCTION);\r
340                 }\r
341         }\r
342 \r
343         void OutputASM::emitShader(Scope scope)\r
344         {\r
345                 emitScope = scope;\r
346                 currentScope = GLOBAL;\r
347                 mContext.getTreeRoot()->traverse(this);\r
348         }\r
349 \r
350         void OutputASM::freeTemporary(Temporary *temporary)\r
351         {\r
352                 free(temporaries, temporary);\r
353         }\r
354 \r
355         sw::Shader::Opcode OutputASM::getOpcode(sw::Shader::Opcode op, TIntermTyped *in) const\r
356         {\r
357                 TBasicType baseType = in->getType().getBasicType();\r
358 \r
359                 switch(op)\r
360                 {\r
361                 case sw::Shader::OPCODE_NEG:\r
362                         switch(baseType)\r
363                         {\r
364                         case EbtInt:\r
365                         case EbtUInt:\r
366                                 return sw::Shader::OPCODE_INEG;\r
367                         case EbtFloat:\r
368                         default:\r
369                                 return op;\r
370                         }\r
371                 case sw::Shader::OPCODE_ABS:\r
372                         switch(baseType)\r
373                         {\r
374                         case EbtInt:\r
375                                 return sw::Shader::OPCODE_IABS;\r
376                         case EbtFloat:\r
377                         default:\r
378                                 return op;\r
379                         }\r
380                 case sw::Shader::OPCODE_SGN:\r
381                         switch(baseType)\r
382                         {\r
383                         case EbtInt:\r
384                                 return sw::Shader::OPCODE_ISGN;\r
385                         case EbtFloat:\r
386                         default:\r
387                                 return op;\r
388                         }\r
389                 case sw::Shader::OPCODE_ADD:\r
390                         switch(baseType)\r
391                         {\r
392                         case EbtInt:\r
393                         case EbtUInt:\r
394                                 return sw::Shader::OPCODE_IADD;\r
395                         case EbtFloat:\r
396                         default:\r
397                                 return op;\r
398                         }\r
399                 case sw::Shader::OPCODE_SUB:\r
400                         switch(baseType)\r
401                         {\r
402                         case EbtInt:\r
403                         case EbtUInt:\r
404                                 return sw::Shader::OPCODE_ISUB;\r
405                         case EbtFloat:\r
406                         default:\r
407                                 return op;\r
408                         }\r
409                 case sw::Shader::OPCODE_MUL:\r
410                         switch(baseType)\r
411                         {\r
412                         case EbtInt:\r
413                         case EbtUInt:\r
414                                 return sw::Shader::OPCODE_IMUL;\r
415                         case EbtFloat:\r
416                         default:\r
417                                 return op;\r
418                         }\r
419                 case sw::Shader::OPCODE_DIV:\r
420                         switch(baseType)\r
421                         {\r
422                         case EbtInt:\r
423                                 return sw::Shader::OPCODE_IDIV;\r
424                         case EbtUInt:\r
425                                 return sw::Shader::OPCODE_UDIV;\r
426                         case EbtFloat:\r
427                         default:\r
428                                 return op;\r
429                         }\r
430                 case sw::Shader::OPCODE_IMOD:\r
431                         return baseType == EbtUInt ? sw::Shader::OPCODE_UMOD : op;\r
432                 case sw::Shader::OPCODE_ISHR:\r
433                         return baseType == EbtUInt ? sw::Shader::OPCODE_USHR : op;\r
434                 case sw::Shader::OPCODE_MIN:\r
435                         switch(baseType)\r
436                         {\r
437                         case EbtInt:\r
438                                 return sw::Shader::OPCODE_IMIN;\r
439                         case EbtUInt:\r
440                                 return sw::Shader::OPCODE_UMIN;\r
441                         case EbtFloat:\r
442                         default:\r
443                                 return op;\r
444                         }\r
445                 case sw::Shader::OPCODE_MAX:\r
446                         switch(baseType)\r
447                         {\r
448                         case EbtInt:\r
449                                 return sw::Shader::OPCODE_IMAX;\r
450                         case EbtUInt:\r
451                                 return sw::Shader::OPCODE_UMAX;\r
452                         case EbtFloat:\r
453                         default:\r
454                                 return op;\r
455                         }\r
456                 default:\r
457                         return op;\r
458                 }\r
459         }\r
460 \r
461         void OutputASM::visitSymbol(TIntermSymbol *symbol)\r
462         {\r
463                 // Vertex varyings don't have to be actively used to successfully link\r
464                 // against pixel shaders that use them. So make sure they're declared.\r
465                 if(symbol->getQualifier() == EvqVaryingOut || symbol->getQualifier() == EvqInvariantVaryingOut || symbol->getQualifier() == EvqVertexOut)\r
466                 {\r
467                         if(symbol->getBasicType() != EbtInvariant)   // Typeless declarations are not new varyings\r
468                         {\r
469                                 declareVarying(symbol, -1);\r
470                         }\r
471                 }\r
472 \r
473                 TInterfaceBlock* block = symbol->getType().getInterfaceBlock();\r
474                 // OpenGL ES 3.0.4 spec, section 2.12.6 Uniform Variables:\r
475                 // "All members of a named uniform block declared with a shared or std140 layout qualifier\r
476                 // are considered active, even if they are not referenced in any shader in the program.\r
477                 // The uniform block itself is also considered active, even if no member of the block is referenced."\r
478                 if(block && ((block->blockStorage() == EbsShared) || (block->blockStorage() == EbsStd140)))\r
479                 {\r
480                         uniformRegister(symbol);\r
481                 }\r
482         }\r
483 \r
484         bool OutputASM::visitBinary(Visit visit, TIntermBinary *node)\r
485         {\r
486                 if(currentScope != emitScope)\r
487                 {\r
488                         return false;\r
489                 }\r
490 \r
491                 TIntermTyped *result = node;\r
492                 TIntermTyped *left = node->getLeft();\r
493                 TIntermTyped *right = node->getRight();\r
494                 const TType &leftType = left->getType();\r
495                 const TType &rightType = right->getType();\r
496                 const TType &resultType = node->getType();\r
497 \r
498                 switch(node->getOp())\r
499                 {\r
500                 case EOpAssign:\r
501                         if(visit == PostVisit)\r
502                         {\r
503                                 assignLvalue(left, right);\r
504                                 copy(result, right);\r
505                         }\r
506                         break;\r
507                 case EOpInitialize:\r
508                         if(visit == PostVisit)\r
509                         {\r
510                                 copy(left, right);\r
511                         }\r
512                         break;\r
513                 case EOpMatrixTimesScalarAssign:\r
514                         if(visit == PostVisit)\r
515                         {\r
516                                 for(int i = 0; i < leftType.getNominalSize(); i++)\r
517                                 {\r
518                                         emit(sw::Shader::OPCODE_MUL, result, i, left, i, right);\r
519                                 }\r
520 \r
521                                 assignLvalue(left, result);\r
522                         }\r
523                         break;\r
524                 case EOpVectorTimesMatrixAssign:\r
525                         if(visit == PostVisit)\r
526                         {\r
527                                 int size = leftType.getNominalSize();\r
528 \r
529                                 for(int i = 0; i < size; i++)\r
530                                 {\r
531                                         Instruction *dot = emit(sw::Shader::OPCODE_DP(size), result, 0, left, 0, right, i);\r
532                                         dot->dst.mask = 1 << i;\r
533                                 }\r
534 \r
535                                 assignLvalue(left, result);\r
536                         }\r
537                         break;\r
538                 case EOpMatrixTimesMatrixAssign:\r
539                         if(visit == PostVisit)\r
540                         {\r
541                                 int dim = leftType.getNominalSize();\r
542 \r
543                                 for(int i = 0; i < dim; i++)\r
544                                 {\r
545                                         Instruction *mul = emit(sw::Shader::OPCODE_MUL, result, i, left, 0, right, i);\r
546                                         mul->src[1].swizzle = 0x00;\r
547 \r
548                                         for(int j = 1; j < dim; j++)\r
549                                         {\r
550                                                 Instruction *mad = emit(sw::Shader::OPCODE_MAD, result, i, left, j, right, i, result, i);\r
551                                                 mad->src[1].swizzle = j * 0x55;\r
552                                         }\r
553                                 }\r
554 \r
555                                 assignLvalue(left, result);\r
556                         }\r
557                         break;\r
558                 case EOpIndexDirect:\r
559                         if(visit == PostVisit)\r
560                         {\r
561                                 int index = right->getAsConstantUnion()->getIConst(0);\r
562 \r
563                                 if(result->isMatrix() || result->isStruct() || result->isInterfaceBlock())\r
564                                 {\r
565                                         ASSERT(left->isArray());\r
566                                         copy(result, left, index * left->elementRegisterCount());\r
567                                 }\r
568                                 else if(result->isRegister())\r
569                                 {\r
570                                         int srcIndex = 0;\r
571                                         if(left->isRegister())\r
572                                         {\r
573                                                 srcIndex = 0;\r
574                                         }\r
575                                         else if(left->isArray())\r
576                                         {\r
577                                                 srcIndex = index * left->elementRegisterCount();\r
578                                         }\r
579                                         else if(left->isMatrix())\r
580                                         {\r
581                                                 ASSERT(index < left->getNominalSize());   // FIXME: Report semantic error\r
582                                                 srcIndex = index;\r
583                                         }\r
584                                         else UNREACHABLE(0);\r
585 \r
586                                         Instruction *mov = emit(sw::Shader::OPCODE_MOV, result, 0, left, srcIndex);\r
587 \r
588                                         if(left->isRegister())\r
589                                         {\r
590                                                 mov->src[0].swizzle = index;\r
591                                         }\r
592                                 }\r
593                                 else UNREACHABLE(0);\r
594                         }\r
595                         break;\r
596                 case EOpIndexIndirect:\r
597                         if(visit == PostVisit)\r
598                         {\r
599                                 if(left->isArray() || left->isMatrix())\r
600                                 {\r
601                                         for(int index = 0; index < result->totalRegisterCount(); index++)\r
602                                         {\r
603                                                 Instruction *mov = emit(sw::Shader::OPCODE_MOV, result, index, left, index);\r
604                                                 mov->dst.mask = writeMask(result, index);\r
605 \r
606                                                 if(left->totalRegisterCount() > 1)\r
607                                                 {\r
608                                                         sw::Shader::SourceParameter relativeRegister;\r
609                                                         argument(relativeRegister, right);\r
610 \r
611                                                         mov->src[0].rel.type = relativeRegister.type;\r
612                                                         mov->src[0].rel.index = relativeRegister.index;\r
613                                                         mov->src[0].rel.scale = result->totalRegisterCount();\r
614                                                         mov->src[0].rel.deterministic = !(vertexShader && left->getQualifier() == EvqUniform);\r
615                                                 }\r
616                                         }\r
617                                 }\r
618                                 else if(left->isRegister())\r
619                                 {\r
620                                         emit(sw::Shader::OPCODE_EXTRACT, result, left, right);\r
621                                 }\r
622                                 else UNREACHABLE(0);\r
623                         }\r
624                         break;\r
625                 case EOpIndexDirectStruct:\r
626                 case EOpIndexDirectInterfaceBlock:\r
627                         if(visit == PostVisit)\r
628                         {\r
629                                 ASSERT(leftType.isStruct() || (leftType.isInterfaceBlock()));\r
630 \r
631                                 const TFieldList& fields = (node->getOp() == EOpIndexDirectStruct) ?\r
632                                                            leftType.getStruct()->fields() :\r
633                                                            leftType.getInterfaceBlock()->fields();\r
634                                 int index = right->getAsConstantUnion()->getIConst(0);\r
635                                 int fieldOffset = 0;\r
636 \r
637                                 for(int i = 0; i < index; i++)\r
638                                 {\r
639                                         fieldOffset += fields[i]->type()->totalRegisterCount();\r
640                                 }\r
641 \r
642                                 copy(result, left, fieldOffset);\r
643                         }\r
644                         break;\r
645                 case EOpVectorSwizzle:\r
646                         if(visit == PostVisit)\r
647                         {\r
648                                 int swizzle = 0;\r
649                                 TIntermAggregate *components = right->getAsAggregate();\r
650 \r
651                                 if(components)\r
652                                 {\r
653                                         TIntermSequence &sequence = components->getSequence();\r
654                                         int component = 0;\r
655 \r
656                                         for(TIntermSequence::iterator sit = sequence.begin(); sit != sequence.end(); sit++)\r
657                                         {\r
658                                                 TIntermConstantUnion *element = (*sit)->getAsConstantUnion();\r
659 \r
660                                                 if(element)\r
661                                                 {\r
662                                                         int i = element->getUnionArrayPointer()[0].getIConst();\r
663                                                         swizzle |= i << (component * 2);\r
664                                                         component++;\r
665                                                 }\r
666                                                 else UNREACHABLE(0);\r
667                                         }\r
668                                 }\r
669                                 else UNREACHABLE(0);\r
670 \r
671                                 Instruction *mov = emit(sw::Shader::OPCODE_MOV, result, left);\r
672                                 mov->src[0].swizzle = swizzle;\r
673                         }\r
674                         break;\r
675                 case EOpAddAssign: if(visit == PostVisit) emitAssign(getOpcode(sw::Shader::OPCODE_ADD, result), result, left, left, right); break;\r
676                 case EOpAdd:       if(visit == PostVisit) emitBinary(getOpcode(sw::Shader::OPCODE_ADD, result), result, left, right);       break;\r
677                 case EOpSubAssign: if(visit == PostVisit) emitAssign(getOpcode(sw::Shader::OPCODE_SUB, result), result, left, left, right); break;\r
678                 case EOpSub:       if(visit == PostVisit) emitBinary(getOpcode(sw::Shader::OPCODE_SUB, result), result, left, right);       break;\r
679                 case EOpMulAssign: if(visit == PostVisit) emitAssign(getOpcode(sw::Shader::OPCODE_MUL, result), result, left, left, right); break;\r
680                 case EOpMul:       if(visit == PostVisit) emitBinary(getOpcode(sw::Shader::OPCODE_MUL, result), result, left, right);       break;\r
681                 case EOpDivAssign: if(visit == PostVisit) emitAssign(getOpcode(sw::Shader::OPCODE_DIV, result), result, left, left, right); break;\r
682                 case EOpDiv:       if(visit == PostVisit) emitBinary(getOpcode(sw::Shader::OPCODE_DIV, result), result, left, right);       break;\r
683                 case EOpIModAssign:          if(visit == PostVisit) emitAssign(getOpcode(sw::Shader::OPCODE_IMOD, result), result, left, left, right); break;\r
684                 case EOpIMod:                if(visit == PostVisit) emitBinary(getOpcode(sw::Shader::OPCODE_IMOD, result), result, left, right);       break;\r
685                 case EOpBitShiftLeftAssign:  if(visit == PostVisit) emitAssign(sw::Shader::OPCODE_SHL, result, left, left, right); break;\r
686                 case EOpBitShiftLeft:        if(visit == PostVisit) emitBinary(sw::Shader::OPCODE_SHL, result, left, right);       break;\r
687                 case EOpBitShiftRightAssign: if(visit == PostVisit) emitAssign(getOpcode(sw::Shader::OPCODE_ISHR, result), result, left, left, right); break;\r
688                 case EOpBitShiftRight:       if(visit == PostVisit) emitBinary(getOpcode(sw::Shader::OPCODE_ISHR, result), result, left, right);       break;\r
689                 case EOpBitwiseAndAssign:    if(visit == PostVisit) emitAssign(sw::Shader::OPCODE_AND, result, left, left, right); break;\r
690                 case EOpBitwiseAnd:          if(visit == PostVisit) emitBinary(sw::Shader::OPCODE_AND, result, left, right);       break;\r
691                 case EOpBitwiseXorAssign:    if(visit == PostVisit) emitAssign(sw::Shader::OPCODE_XOR, result, left, left, right); break;\r
692                 case EOpBitwiseXor:          if(visit == PostVisit) emitBinary(sw::Shader::OPCODE_XOR, result, left, right);       break;\r
693                 case EOpBitwiseOrAssign:     if(visit == PostVisit) emitAssign(sw::Shader::OPCODE_OR, result, left, left, right);  break;\r
694                 case EOpBitwiseOr:           if(visit == PostVisit) emitBinary(sw::Shader::OPCODE_OR, result, left, right);        break;\r
695                 case EOpEqual:\r
696                         if(visit == PostVisit)\r
697                         {\r
698                                 emitBinary(sw::Shader::OPCODE_EQ, result, left, right);\r
699 \r
700                                 for(int index = 1; index < left->totalRegisterCount(); index++)\r
701                                 {\r
702                                         Temporary equal(this);\r
703                                         emit(sw::Shader::OPCODE_EQ, &equal, 0, left, index, right, index);\r
704                                         emit(sw::Shader::OPCODE_AND, result, result, &equal);\r
705                                 }\r
706                         }\r
707                         break;\r
708                 case EOpNotEqual:\r
709                         if(visit == PostVisit)\r
710                         {\r
711                                 emitBinary(sw::Shader::OPCODE_NE, result, left, right);\r
712 \r
713                                 for(int index = 1; index < left->totalRegisterCount(); index++)\r
714                                 {\r
715                                         Temporary notEqual(this);\r
716                                         emit(sw::Shader::OPCODE_NE, &notEqual, 0, left, index, right, index);\r
717                                         emit(sw::Shader::OPCODE_OR, result, result, &notEqual);\r
718                                 }\r
719                         }\r
720                         break;\r
721                 case EOpLessThan:                if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_LT, result, left, right); break;\r
722                 case EOpGreaterThan:             if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_GT, result, left, right); break;\r
723                 case EOpLessThanEqual:           if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_LE, result, left, right); break;\r
724                 case EOpGreaterThanEqual:        if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_GE, result, left, right); break;\r
725                 case EOpVectorTimesScalarAssign: if(visit == PostVisit) emitAssign(getOpcode(sw::Shader::OPCODE_MUL, left), result, left, left, right); break;\r
726                 case EOpVectorTimesScalar:       if(visit == PostVisit) emit(getOpcode(sw::Shader::OPCODE_MUL, left), result, left, right); break;\r
727                 case EOpMatrixTimesScalar:\r
728                         if(visit == PostVisit)\r
729                         {\r
730                                 for(int i = 0; i < leftType.getNominalSize(); i++)\r
731                                 {\r
732                                         emit(sw::Shader::OPCODE_MUL, result, i, left, i, right);\r
733                                 }\r
734                         }\r
735                         break;\r
736                 case EOpVectorTimesMatrix:\r
737                         if(visit == PostVisit)\r
738                         {\r
739                                 sw::Shader::Opcode dpOpcode = sw::Shader::OPCODE_DP(leftType.getNominalSize());\r
740 \r
741                                 int size = rightType.getNominalSize();\r
742                                 for(int i = 0; i < size; i++)\r
743                                 {\r
744                                         Instruction *dot = emit(dpOpcode, result, 0, left, 0, right, i);\r
745                                         dot->dst.mask = 1 << i;\r
746                                 }\r
747                         }\r
748                         break;\r
749                 case EOpMatrixTimesVector:\r
750                         if(visit == PostVisit)\r
751                         {\r
752                                 Instruction *mul = emit(sw::Shader::OPCODE_MUL, result, left, right);\r
753                                 mul->src[1].swizzle = 0x00;\r
754 \r
755                                 int size = rightType.getNominalSize();\r
756                                 for(int i = 1; i < size; i++)\r
757                                 {\r
758                                         Instruction *mad = emit(sw::Shader::OPCODE_MAD, result, 0, left, i, right, 0, result);\r
759                                         mad->src[1].swizzle = i * 0x55;\r
760                                 }\r
761                         }\r
762                         break;\r
763                 case EOpMatrixTimesMatrix:\r
764                         if(visit == PostVisit)\r
765                         {\r
766                                 int dim = leftType.getNominalSize();\r
767 \r
768                                 int size = rightType.getNominalSize();\r
769                                 for(int i = 0; i < size; i++)\r
770                                 {\r
771                                         Instruction *mul = emit(sw::Shader::OPCODE_MUL, result, i, left, 0, right, i);\r
772                                         mul->src[1].swizzle = 0x00;\r
773 \r
774                                         for(int j = 1; j < dim; j++)\r
775                                         {\r
776                                                 Instruction *mad = emit(sw::Shader::OPCODE_MAD, result, i, left, j, right, i, result, i);\r
777                                                 mad->src[1].swizzle = j * 0x55;\r
778                                         }\r
779                                 }\r
780                         }\r
781                         break;\r
782                 case EOpLogicalOr:\r
783                         if(trivial(right, 6))\r
784                         {\r
785                                 if(visit == PostVisit)\r
786                                 {\r
787                                         emit(sw::Shader::OPCODE_OR, result, left, right);\r
788                                 }\r
789                         }\r
790                         else   // Short-circuit evaluation\r
791                         {\r
792                                 if(visit == InVisit)\r
793                                 {\r
794                                         emit(sw::Shader::OPCODE_MOV, result, left);\r
795                                         Instruction *ifnot = emit(sw::Shader::OPCODE_IF, 0, result);\r
796                                         ifnot->src[0].modifier = sw::Shader::MODIFIER_NOT;\r
797                                 }\r
798                                 else if(visit == PostVisit)\r
799                                 {\r
800                                         emit(sw::Shader::OPCODE_MOV, result, right);\r
801                                         emit(sw::Shader::OPCODE_ENDIF);\r
802                                 }\r
803                         }\r
804                         break;\r
805                 case EOpLogicalXor:        if(visit == PostVisit) emit(sw::Shader::OPCODE_XOR, result, left, right); break;\r
806                 case EOpLogicalAnd:\r
807                         if(trivial(right, 6))\r
808                         {\r
809                                 if(visit == PostVisit)\r
810                                 {\r
811                                         emit(sw::Shader::OPCODE_AND, result, left, right);\r
812                                 }\r
813                         }\r
814                         else   // Short-circuit evaluation\r
815                         {\r
816                                 if(visit == InVisit)\r
817                                 {\r
818                                         emit(sw::Shader::OPCODE_MOV, result, left);\r
819                                         emit(sw::Shader::OPCODE_IF, 0, result);\r
820                                 }\r
821                                 else if(visit == PostVisit)\r
822                                 {\r
823                                         emit(sw::Shader::OPCODE_MOV, result, right);\r
824                                         emit(sw::Shader::OPCODE_ENDIF);\r
825                                 }\r
826                         }\r
827                         break;\r
828                 default: UNREACHABLE(node->getOp());\r
829                 }\r
830 \r
831                 return true;\r
832         }\r
833 \r
834         void OutputASM::emitDeterminant(TIntermTyped *result, TIntermTyped *arg, int size, int col, int row, int outCol, int outRow)\r
835         {\r
836                 switch(size)\r
837                 {\r
838                 case 1: // Used for cofactor computation only\r
839                         {\r
840                                 // For a 2x2 matrix, the cofactor is simply a transposed move or negate\r
841                                 bool isMov = (row == col);\r
842                                 sw::Shader::Opcode op = isMov ? sw::Shader::OPCODE_MOV : sw::Shader::OPCODE_NEG;\r
843                                 Instruction *mov = emit(op, result, outCol, arg, isMov ? 1 - row : row);\r
844                                 mov->src[0].swizzle = 0x55 * (isMov ? 1 - col : col);\r
845                                 mov->dst.mask = 1 << outRow;\r
846                         }\r
847                         break;\r
848                 case 2:\r
849                         {\r
850                                 static const unsigned int swizzle[3] = { 0x99, 0x88, 0x44 }; // xy?? : yzyz, xzxz, xyxy\r
851 \r
852                                 bool isCofactor = (col >= 0) && (row >= 0);\r
853                                 int col0 = (isCofactor && (col <= 0)) ? 1 : 0;\r
854                                 int col1 = (isCofactor && (col <= 1)) ? 2 : 1;\r
855                                 bool negate = isCofactor && ((col & 0x01) ^ (row & 0x01));\r
856 \r
857                                 Instruction *det = emit(sw::Shader::OPCODE_DET2, result, outCol, arg, negate ? col1 : col0, arg, negate ? col0 : col1);\r
858                                 det->src[0].swizzle = det->src[1].swizzle = swizzle[isCofactor ? row : 2];\r
859                                 det->dst.mask = 1 << outRow;\r
860                         }\r
861                         break;\r
862                 case 3:\r
863                         {\r
864                                 static const unsigned int swizzle[4] = { 0xF9, 0xF8, 0xF4, 0xE4 }; // xyz? : yzww, xzww, xyww, xyzw\r
865 \r
866                                 bool isCofactor = (col >= 0) && (row >= 0);\r
867                                 int col0 = (isCofactor && (col <= 0)) ? 1 : 0;\r
868                                 int col1 = (isCofactor && (col <= 1)) ? 2 : 1;\r
869                                 int col2 = (isCofactor && (col <= 2)) ? 3 : 2;\r
870                                 bool negate = isCofactor && ((col & 0x01) ^ (row & 0x01));\r
871 \r
872                                 Instruction *det = emit(sw::Shader::OPCODE_DET3, result, outCol, arg, col0, arg, negate ? col2 : col1, arg, negate ? col1 : col2);\r
873                                 det->src[0].swizzle = det->src[1].swizzle = det->src[2].swizzle = swizzle[isCofactor ? row : 3];\r
874                                 det->dst.mask = 1 << outRow;\r
875                         }\r
876                         break;\r
877                 case 4:\r
878                         {\r
879                                 Instruction *det = emit(sw::Shader::OPCODE_DET4, result, outCol, arg, 0, arg, 1, arg, 2, arg, 3);\r
880                                 det->dst.mask = 1 << outRow;\r
881                         }\r
882                         break;\r
883                 default:\r
884                         UNREACHABLE(size);\r
885                         break;\r
886                 }\r
887         }\r
888 \r
889         bool OutputASM::visitUnary(Visit visit, TIntermUnary *node)\r
890         {\r
891                 if(currentScope != emitScope)\r
892                 {\r
893                         return false;\r
894                 }\r
895 \r
896                 TIntermTyped *result = node;\r
897                 TIntermTyped *arg = node->getOperand();\r
898                 TBasicType basicType = arg->getType().getBasicType();\r
899 \r
900                 union\r
901                 {\r
902                         float f;\r
903                         int i;\r
904                 } one_value;\r
905 \r
906                 if(basicType == EbtInt || basicType == EbtUInt)\r
907                 {\r
908                         one_value.i = 1;\r
909                 }\r
910                 else\r
911                 {\r
912                         one_value.f = 1.0f;\r
913                 }\r
914 \r
915                 Constant one(one_value.f, one_value.f, one_value.f, one_value.f);\r
916                 Constant rad(1.74532925e-2f, 1.74532925e-2f, 1.74532925e-2f, 1.74532925e-2f);\r
917                 Constant deg(5.72957795e+1f, 5.72957795e+1f, 5.72957795e+1f, 5.72957795e+1f);\r
918 \r
919                 switch(node->getOp())\r
920                 {\r
921                 case EOpNegative:\r
922                         if(visit == PostVisit)\r
923                         {\r
924                                 sw::Shader::Opcode negOpcode = getOpcode(sw::Shader::OPCODE_NEG, arg);\r
925                                 for(int index = 0; index < arg->totalRegisterCount(); index++)\r
926                                 {\r
927                                         emit(negOpcode, result, index, arg, index);\r
928                                 }\r
929                         }\r
930                         break;\r
931                 case EOpVectorLogicalNot: if(visit == PostVisit) emit(sw::Shader::OPCODE_NOT, result, arg); break;\r
932                 case EOpLogicalNot:       if(visit == PostVisit) emit(sw::Shader::OPCODE_NOT, result, arg); break;\r
933                 case EOpPostIncrement:\r
934                         if(visit == PostVisit)\r
935                         {\r
936                                 copy(result, arg);\r
937 \r
938                                 sw::Shader::Opcode addOpcode = getOpcode(sw::Shader::OPCODE_ADD, arg);\r
939                                 for(int index = 0; index < arg->totalRegisterCount(); index++)\r
940                                 {\r
941                                         emit(addOpcode, arg, index, arg, index, &one);\r
942                                 }\r
943 \r
944                                 assignLvalue(arg, arg);\r
945                         }\r
946                         break;\r
947                 case EOpPostDecrement:\r
948                         if(visit == PostVisit)\r
949                         {\r
950                                 copy(result, arg);\r
951 \r
952                                 sw::Shader::Opcode subOpcode = getOpcode(sw::Shader::OPCODE_SUB, arg);\r
953                                 for(int index = 0; index < arg->totalRegisterCount(); index++)\r
954                                 {\r
955                                         emit(subOpcode, arg, index, arg, index, &one);\r
956                                 }\r
957 \r
958                                 assignLvalue(arg, arg);\r
959                         }\r
960                         break;\r
961                 case EOpPreIncrement:\r
962                         if(visit == PostVisit)\r
963                         {\r
964                                 sw::Shader::Opcode addOpcode = getOpcode(sw::Shader::OPCODE_ADD, arg);\r
965                                 for(int index = 0; index < arg->totalRegisterCount(); index++)\r
966                                 {\r
967                                         emit(addOpcode, result, index, arg, index, &one);\r
968                                 }\r
969 \r
970                                 assignLvalue(arg, result);\r
971                         }\r
972                         break;\r
973                 case EOpPreDecrement:\r
974                         if(visit == PostVisit)\r
975                         {\r
976                                 sw::Shader::Opcode subOpcode = getOpcode(sw::Shader::OPCODE_SUB, arg);\r
977                                 for(int index = 0; index < arg->totalRegisterCount(); index++)\r
978                                 {\r
979                                         emit(subOpcode, result, index, arg, index, &one);\r
980                                 }\r
981 \r
982                                 assignLvalue(arg, result);\r
983                         }\r
984                         break;\r
985                 case EOpRadians:          if(visit == PostVisit) emit(sw::Shader::OPCODE_MUL, result, arg, &rad); break;\r
986                 case EOpDegrees:          if(visit == PostVisit) emit(sw::Shader::OPCODE_MUL, result, arg, &deg); break;\r
987                 case EOpSin:              if(visit == PostVisit) emit(sw::Shader::OPCODE_SIN, result, arg); break;\r
988                 case EOpCos:              if(visit == PostVisit) emit(sw::Shader::OPCODE_COS, result, arg); break;\r
989                 case EOpTan:              if(visit == PostVisit) emit(sw::Shader::OPCODE_TAN, result, arg); break;\r
990                 case EOpAsin:             if(visit == PostVisit) emit(sw::Shader::OPCODE_ASIN, result, arg); break;\r
991                 case EOpAcos:             if(visit == PostVisit) emit(sw::Shader::OPCODE_ACOS, result, arg); break;\r
992                 case EOpAtan:             if(visit == PostVisit) emit(sw::Shader::OPCODE_ATAN, result, arg); break;\r
993                 case EOpSinh:             if(visit == PostVisit) emit(sw::Shader::OPCODE_SINH, result, arg); break;\r
994                 case EOpCosh:             if(visit == PostVisit) emit(sw::Shader::OPCODE_COSH, result, arg); break;\r
995                 case EOpTanh:             if(visit == PostVisit) emit(sw::Shader::OPCODE_TANH, result, arg); break;\r
996                 case EOpAsinh:            if(visit == PostVisit) emit(sw::Shader::OPCODE_ASINH, result, arg); break;\r
997                 case EOpAcosh:            if(visit == PostVisit) emit(sw::Shader::OPCODE_ACOSH, result, arg); break;\r
998                 case EOpAtanh:            if(visit == PostVisit) emit(sw::Shader::OPCODE_ATANH, result, arg); break;\r
999                 case EOpExp:              if(visit == PostVisit) emit(sw::Shader::OPCODE_EXP, result, arg); break;\r
1000                 case EOpLog:              if(visit == PostVisit) emit(sw::Shader::OPCODE_LOG, result, arg); break;\r
1001                 case EOpExp2:             if(visit == PostVisit) emit(sw::Shader::OPCODE_EXP2, result, arg); break;\r
1002                 case EOpLog2:             if(visit == PostVisit) emit(sw::Shader::OPCODE_LOG2, result, arg); break;\r
1003                 case EOpSqrt:             if(visit == PostVisit) emit(sw::Shader::OPCODE_SQRT, result, arg); break;\r
1004                 case EOpInverseSqrt:      if(visit == PostVisit) emit(sw::Shader::OPCODE_RSQ, result, arg); break;\r
1005                 case EOpAbs:              if(visit == PostVisit) emit(getOpcode(sw::Shader::OPCODE_ABS, result), result, arg); break;\r
1006                 case EOpSign:             if(visit == PostVisit) emit(getOpcode(sw::Shader::OPCODE_SGN, result), result, arg); break;\r
1007                 case EOpFloor:            if(visit == PostVisit) emit(sw::Shader::OPCODE_FLOOR, result, arg); break;\r
1008                 case EOpTrunc:            if(visit == PostVisit) emit(sw::Shader::OPCODE_TRUNC, result, arg); break;\r
1009                 case EOpRound:            if(visit == PostVisit) emit(sw::Shader::OPCODE_ROUND, result, arg); break;\r
1010                 case EOpRoundEven:        if(visit == PostVisit) emit(sw::Shader::OPCODE_ROUNDEVEN, result, arg); break;\r
1011                 case EOpCeil:             if(visit == PostVisit) emit(sw::Shader::OPCODE_CEIL, result, arg, result); break;\r
1012                 case EOpFract:            if(visit == PostVisit) emit(sw::Shader::OPCODE_FRC, result, arg); break;\r
1013                 case EOpIsNan:            if(visit == PostVisit) emit(sw::Shader::OPCODE_ISNAN, result, arg); break;\r
1014                 case EOpIsInf:            if(visit == PostVisit) emit(sw::Shader::OPCODE_ISINF, result, arg); break;\r
1015                 case EOpLength:           if(visit == PostVisit) emit(sw::Shader::OPCODE_LEN(dim(arg)), result, arg); break;\r
1016                 case EOpNormalize:        if(visit == PostVisit) emit(sw::Shader::OPCODE_NRM(dim(arg)), result, arg); break;\r
1017                 case EOpDFdx:             if(visit == PostVisit) emit(sw::Shader::OPCODE_DFDX, result, arg); break;\r
1018                 case EOpDFdy:             if(visit == PostVisit) emit(sw::Shader::OPCODE_DFDY, result, arg); break;\r
1019                 case EOpFwidth:           if(visit == PostVisit) emit(sw::Shader::OPCODE_FWIDTH, result, arg); break;\r
1020                 case EOpAny:              if(visit == PostVisit) emit(sw::Shader::OPCODE_ANY, result, arg); break;\r
1021                 case EOpAll:              if(visit == PostVisit) emit(sw::Shader::OPCODE_ALL, result, arg); break;\r
1022                 case EOpFloatBitsToInt:   if(visit == PostVisit) emit(sw::Shader::OPCODE_FLOATBITSTOINT, result, arg); break;\r
1023                 case EOpFloatBitsToUint:  if(visit == PostVisit) emit(sw::Shader::OPCODE_FLOATBITSTOUINT, result, arg); break;\r
1024                 case EOpIntBitsToFloat:   if(visit == PostVisit) emit(sw::Shader::OPCODE_INTBITSTOFLOAT, result, arg); break;\r
1025                 case EOpUintBitsToFloat:  if(visit == PostVisit) emit(sw::Shader::OPCODE_UINTBITSTOFLOAT, result, arg); break;\r
1026                 case EOpPackSnorm2x16:    if(visit == PostVisit) emit(sw::Shader::OPCODE_PACKSNORM2x16, result, arg); break;\r
1027                 case EOpPackUnorm2x16:    if(visit == PostVisit) emit(sw::Shader::OPCODE_PACKUNORM2x16, result, arg); break;\r
1028                 case EOpPackHalf2x16:     if(visit == PostVisit) emit(sw::Shader::OPCODE_PACKHALF2x16, result, arg); break;\r
1029                 case EOpUnpackSnorm2x16:  if(visit == PostVisit) emit(sw::Shader::OPCODE_UNPACKSNORM2x16, result, arg); break;\r
1030                 case EOpUnpackUnorm2x16:  if(visit == PostVisit) emit(sw::Shader::OPCODE_UNPACKUNORM2x16, result, arg); break;\r
1031                 case EOpUnpackHalf2x16:   if(visit == PostVisit) emit(sw::Shader::OPCODE_UNPACKHALF2x16, result, arg); break;\r
1032                 case EOpTranspose:\r
1033                         if(visit == PostVisit)\r
1034                         {\r
1035                                 int numCols = arg->getNominalSize();\r
1036                                 int numRows = arg->getSecondarySize();\r
1037                                 for(int i = 0; i < numCols; ++i)\r
1038                                 {\r
1039                                         for(int j = 0; j < numRows; ++j)\r
1040                                         {\r
1041                                                 Instruction *mov = emit(sw::Shader::OPCODE_MOV, result, j, arg, i);\r
1042                                                 mov->src[0].swizzle = 0x55 * j;\r
1043                                                 mov->dst.mask = 1 << i;\r
1044                                         }\r
1045                                 }\r
1046                         }\r
1047                         break;\r
1048                 case EOpDeterminant:\r
1049                         if(visit == PostVisit)\r
1050                         {\r
1051                                 int size = arg->getNominalSize();\r
1052                                 ASSERT(size == arg->getSecondarySize());\r
1053 \r
1054                                 emitDeterminant(result, arg, size);\r
1055                         }\r
1056                         break;\r
1057                 case EOpInverse:\r
1058                         if(visit == PostVisit)\r
1059                         {\r
1060                                 int size = arg->getNominalSize();\r
1061                                 ASSERT(size == arg->getSecondarySize());\r
1062 \r
1063                                 // Compute transposed matrix of cofactors\r
1064                                 for(int i = 0; i < size; ++i)\r
1065                                 {\r
1066                                         for(int j = 0; j < size; ++j)\r
1067                                         {\r
1068                                                 // For a 2x2 matrix, the cofactor is simply a transposed move or negate\r
1069                                                 // For a 3x3 or 4x4 matrix, the cofactor is a transposed determinant\r
1070                                                 emitDeterminant(result, arg, size - 1, j, i, i, j);\r
1071                                         }\r
1072                                 }\r
1073 \r
1074                                 // Compute 1 / determinant\r
1075                                 Temporary invDet(this);\r
1076                                 emitDeterminant(&invDet, arg, size);\r
1077                                 Constant one(1.0f, 1.0f, 1.0f, 1.0f);\r
1078                                 Instruction *div = emit(sw::Shader::OPCODE_DIV, &invDet, &one, &invDet);\r
1079                                 div->src[1].swizzle = 0x00; // xxxx\r
1080 \r
1081                                 // Divide transposed matrix of cofactors by determinant\r
1082                                 for(int i = 0; i < size; ++i)\r
1083                                 {\r
1084                                         emit(sw::Shader::OPCODE_MUL, result, i, result, i, &invDet);\r
1085                                 }\r
1086                         }\r
1087                         break;\r
1088                 default: UNREACHABLE(node->getOp());\r
1089                 }\r
1090 \r
1091                 return true;\r
1092         }\r
1093 \r
1094         bool OutputASM::visitAggregate(Visit visit, TIntermAggregate *node)\r
1095         {\r
1096                 if(currentScope != emitScope && node->getOp() != EOpFunction && node->getOp() != EOpSequence)\r
1097                 {\r
1098                         return false;\r
1099                 }\r
1100 \r
1101                 Constant zero(0.0f, 0.0f, 0.0f, 0.0f);\r
1102 \r
1103                 TIntermTyped *result = node;\r
1104                 const TType &resultType = node->getType();\r
1105                 TIntermSequence &arg = node->getSequence();\r
1106                 int argumentCount = arg.size();\r
1107 \r
1108                 switch(node->getOp())\r
1109                 {\r
1110                 case EOpSequence:             break;\r
1111                 case EOpDeclaration:          break;\r
1112                 case EOpInvariantDeclaration: break;\r
1113                 case EOpPrototype:            break;\r
1114                 case EOpComma:\r
1115                         if(visit == PostVisit)\r
1116                         {\r
1117                                 copy(result, arg[1]);\r
1118                         }\r
1119                         break;\r
1120                 case EOpFunction:\r
1121                         if(visit == PreVisit)\r
1122                         {\r
1123                                 const TString &name = node->getName();\r
1124 \r
1125                                 if(emitScope == FUNCTION)\r
1126                                 {\r
1127                                         if(functionArray.size() > 1)   // No need for a label when there's only main()\r
1128                                         {\r
1129                                                 Instruction *label = emit(sw::Shader::OPCODE_LABEL);\r
1130                                                 label->dst.type = sw::Shader::PARAMETER_LABEL;\r
1131 \r
1132                                                 const Function *function = findFunction(name);\r
1133                                                 ASSERT(function);   // Should have been added during global pass\r
1134                                                 label->dst.index = function->label;\r
1135                                                 currentFunction = function->label;\r
1136                                         }\r
1137                                 }\r
1138                                 else if(emitScope == GLOBAL)\r
1139                                 {\r
1140                                         if(name != "main(")\r
1141                                         {\r
1142                                                 TIntermSequence &arguments = node->getSequence()[0]->getAsAggregate()->getSequence();\r
1143                                                 functionArray.push_back(Function(functionArray.size(), name, &arguments, node));\r
1144                                         }\r
1145                                 }\r
1146                                 else UNREACHABLE(emitScope);\r
1147 \r
1148                                 currentScope = FUNCTION;\r
1149                         }\r
1150                         else if(visit == PostVisit)\r
1151                         {\r
1152                                 if(emitScope == FUNCTION)\r
1153                                 {\r
1154                                         if(functionArray.size() > 1)   // No need to return when there's only main()\r
1155                                         {\r
1156                                                 emit(sw::Shader::OPCODE_RET);\r
1157                                         }\r
1158                                 }\r
1159 \r
1160                                 currentScope = GLOBAL;\r
1161                         }\r
1162                         break;\r
1163                 case EOpFunctionCall:\r
1164                         if(visit == PostVisit)\r
1165                         {\r
1166                                 if(node->isUserDefined())\r
1167                                 {\r
1168                                         const TString &name = node->getName();\r
1169                                         const Function *function = findFunction(name);\r
1170 \r
1171                                         if(!function)\r
1172                                         {\r
1173                                                 mContext.error(node->getLine(), "function definition not found", name.c_str());\r
1174                                                 return false;\r
1175                                         }\r
1176 \r
1177                                         TIntermSequence &arguments = *function->arg;\r
1178 \r
1179                                         for(int i = 0; i < argumentCount; i++)\r
1180                                         {\r
1181                                                 TIntermTyped *in = arguments[i]->getAsTyped();\r
1182 \r
1183                                                 if(in->getQualifier() == EvqIn ||\r
1184                                                    in->getQualifier() == EvqInOut ||\r
1185                                                    in->getQualifier() == EvqConstReadOnly)\r
1186                                                 {\r
1187                                                         copy(in, arg[i]);\r
1188                                                 }\r
1189                                         }\r
1190 \r
1191                                         Instruction *call = emit(sw::Shader::OPCODE_CALL);\r
1192                                         call->dst.type = sw::Shader::PARAMETER_LABEL;\r
1193                                         call->dst.index = function->label;\r
1194 \r
1195                                         if(function->ret && function->ret->getType().getBasicType() != EbtVoid)\r
1196                                         {\r
1197                                                 copy(result, function->ret);\r
1198                                         }\r
1199 \r
1200                                         for(int i = 0; i < argumentCount; i++)\r
1201                                         {\r
1202                                                 TIntermTyped *argument = arguments[i]->getAsTyped();\r
1203                                                 TIntermTyped *out = arg[i]->getAsTyped();\r
1204 \r
1205                                                 if(argument->getQualifier() == EvqOut ||\r
1206                                                    argument->getQualifier() == EvqInOut)\r
1207                                                 {\r
1208                                                         copy(out, argument);\r
1209                                                 }\r
1210                                         }\r
1211                                 }\r
1212                                 else\r
1213                                 {\r
1214                                         const TextureFunction textureFunction(node->getName());\r
1215                                         switch(textureFunction.method)\r
1216                                         {\r
1217                                         case TextureFunction::IMPLICIT:\r
1218                                                 {\r
1219                                                         TIntermTyped *t = arg[1]->getAsTyped();\r
1220 \r
1221                                                         TIntermNode* offset = textureFunction.offset ? arg[2] : 0;\r
1222 \r
1223                                                         if(argumentCount == 2 || (textureFunction.offset && argumentCount == 3))\r
1224                                                         {\r
1225                                                                 Instruction *tex = emit(textureFunction.offset ? sw::Shader::OPCODE_TEXOFFSET : sw::Shader::OPCODE_TEX,\r
1226                                                                                         result, arg[1], arg[0], offset);\r
1227                                                                 if(textureFunction.proj)\r
1228                                                                 {\r
1229                                                                         tex->project = true;\r
1230 \r
1231                                                                         switch(t->getNominalSize())\r
1232                                                                         {\r
1233                                                                         case 2: tex->src[0].swizzle = 0x54; break; // xyyy\r
1234                                                                         case 3: tex->src[0].swizzle = 0xA4; break; // xyzz\r
1235                                                                         case 4: break; // xyzw\r
1236                                                                         default:\r
1237                                                                                 UNREACHABLE(t->getNominalSize());\r
1238                                                                                 break;\r
1239                                                                         }\r
1240                                                                 }\r
1241                                                         }\r
1242                                                         else if(argumentCount == 3 || (textureFunction.offset && argumentCount == 4))   // bias\r
1243                                                         {\r
1244                                                                 Temporary proj(this);\r
1245                                                                 if(textureFunction.proj)\r
1246                                                                 {\r
1247                                                                         Instruction *div = emit(sw::Shader::OPCODE_DIV, &proj, arg[1], arg[1]);\r
1248                                                                         div->dst.mask = 0x3;\r
1249 \r
1250                                                                         switch(t->getNominalSize())\r
1251                                                                         {\r
1252                                                                         case 2:\r
1253                                                                         case 3:\r
1254                                                                         case 4:\r
1255                                                                                 div->src[1].swizzle = 0x55 * (t->getNominalSize() - 1);\r
1256                                                                                 break;\r
1257                                                                         default:\r
1258                                                                                 UNREACHABLE(t->getNominalSize());\r
1259                                                                                 break;\r
1260                                                                         }\r
1261                                                                 }\r
1262                                                                 else\r
1263                                                                 {\r
1264                                                                         emit(sw::Shader::OPCODE_MOV, &proj, arg[1]);\r
1265                                                                 }\r
1266 \r
1267                                                                 Instruction *bias = emit(sw::Shader::OPCODE_MOV, &proj, arg[textureFunction.offset ? 3 : 2]);\r
1268                                                                 bias->dst.mask = 0x8;\r
1269 \r
1270                                                                 Instruction *tex = emit(textureFunction.offset ? sw::Shader::OPCODE_TEXOFFSET : sw::Shader::OPCODE_TEX,\r
1271                                                                                         result, &proj, arg[0], offset); // FIXME: Implement an efficient TEXLDB instruction\r
1272                                                                 tex->bias = true;\r
1273                                                         }\r
1274                                                         else UNREACHABLE(argumentCount);\r
1275                                                 }\r
1276                                                 break;\r
1277                                         case TextureFunction::LOD:\r
1278                                                 {\r
1279                                                         TIntermTyped *t = arg[1]->getAsTyped();\r
1280                                                         Temporary proj(this);\r
1281 \r
1282                                                         if(textureFunction.proj)\r
1283                                                         {\r
1284                                                                 Instruction *div = emit(sw::Shader::OPCODE_DIV, &proj, arg[1], arg[1]);\r
1285                                                                 div->dst.mask = 0x3;\r
1286 \r
1287                                                                 switch(t->getNominalSize())\r
1288                                                                 {\r
1289                                                                 case 2:\r
1290                                                                 case 3:\r
1291                                                                 case 4:\r
1292                                                                         div->src[1].swizzle = 0x55 * (t->getNominalSize() - 1);\r
1293                                                                         break;\r
1294                                                                 default:\r
1295                                                                         UNREACHABLE(t->getNominalSize());\r
1296                                                                         break;\r
1297                                                                 }\r
1298                                                         }\r
1299                                                         else\r
1300                                                         {\r
1301                                                                 emit(sw::Shader::OPCODE_MOV, &proj, arg[1]);\r
1302                                                         }\r
1303 \r
1304                                                         Instruction *lod = emit(sw::Shader::OPCODE_MOV, &proj, arg[2]);\r
1305                                                         lod->dst.mask = 0x8;\r
1306 \r
1307                                                         emit(textureFunction.offset ? sw::Shader::OPCODE_TEXLDLOFFSET : sw::Shader::OPCODE_TEXLDL,\r
1308                                                              result, &proj, arg[0], textureFunction.offset ? arg[3] : 0);\r
1309                                                 }\r
1310                                                 break;\r
1311                                         case TextureFunction::FETCH:\r
1312                                                 {\r
1313                                                         TIntermTyped *t = arg[1]->getAsTyped();\r
1314 \r
1315                                                         if(argumentCount == 3 || (textureFunction.offset && argumentCount == 4))\r
1316                                                         {\r
1317                                                                 TIntermNode* offset = textureFunction.offset ? arg[3] : 0;\r
1318 \r
1319                                                                 emit(textureFunction.offset ? sw::Shader::OPCODE_TEXELFETCHOFFSET : sw::Shader::OPCODE_TEXELFETCH,\r
1320                                                                      result, arg[1], arg[0], arg[2], offset);\r
1321                                                         }\r
1322                                                         else UNREACHABLE(argumentCount);\r
1323                                                 }\r
1324                                                 break;\r
1325                                         case TextureFunction::GRAD:\r
1326                                                 {\r
1327                                                         TIntermTyped *t = arg[1]->getAsTyped();\r
1328 \r
1329                                                         if(argumentCount == 4 || (textureFunction.offset && argumentCount == 5))\r
1330                                                         {\r
1331                                                                 Temporary uvwb(this);\r
1332 \r
1333                                                                 if(textureFunction.proj)\r
1334                                                                 {\r
1335                                                                         Instruction *div = emit(sw::Shader::OPCODE_DIV, &uvwb, arg[1], arg[1]);\r
1336                                                                         div->dst.mask = 0x3;\r
1337 \r
1338                                                                         switch(t->getNominalSize())\r
1339                                                                         {\r
1340                                                                         case 2:\r
1341                                                                         case 3:\r
1342                                                                         case 4:\r
1343                                                                                 div->src[1].swizzle = 0x55 * (t->getNominalSize() - 1);\r
1344                                                                                 break;\r
1345                                                                         default:\r
1346                                                                                 UNREACHABLE(t->getNominalSize());\r
1347                                                                                 break;\r
1348                                                                         }\r
1349                                                                 }\r
1350                                                                 else\r
1351                                                                 {\r
1352                                                                         emit(sw::Shader::OPCODE_MOV, &uvwb, arg[1]);\r
1353                                                                 }\r
1354 \r
1355                                                                 TIntermNode* offset = textureFunction.offset ? arg[4] : 0;\r
1356 \r
1357                                                                 emit(textureFunction.offset ? sw::Shader::OPCODE_TEXGRADOFFSET : sw::Shader::OPCODE_TEXGRAD,\r
1358                                                                      result, &uvwb, arg[0], arg[2], arg[3], offset);\r
1359                                                         }\r
1360                                                         else UNREACHABLE(argumentCount);\r
1361                                                 }\r
1362                                                 break;\r
1363                                         case TextureFunction::SIZE:\r
1364                                                 emit(sw::Shader::OPCODE_TEXSIZE, result, arg[1], arg[0]);\r
1365                                                 break;\r
1366                                         default:\r
1367                                                 UNREACHABLE(textureFunction.method);\r
1368                                         }\r
1369                                 }\r
1370                         }\r
1371                         break;\r
1372                 case EOpParameters:\r
1373                         break;\r
1374                 case EOpConstructFloat:\r
1375                 case EOpConstructVec2:\r
1376                 case EOpConstructVec3:\r
1377                 case EOpConstructVec4:\r
1378                 case EOpConstructBool:\r
1379                 case EOpConstructBVec2:\r
1380                 case EOpConstructBVec3:\r
1381                 case EOpConstructBVec4:\r
1382                 case EOpConstructInt:\r
1383                 case EOpConstructIVec2:\r
1384                 case EOpConstructIVec3:\r
1385                 case EOpConstructIVec4:\r
1386                 case EOpConstructUInt:\r
1387                 case EOpConstructUVec2:\r
1388                 case EOpConstructUVec3:\r
1389                 case EOpConstructUVec4:\r
1390                         if(visit == PostVisit)\r
1391                         {\r
1392                                 int component = 0;\r
1393 \r
1394                                 for(int i = 0; i < argumentCount; i++)\r
1395                                 {\r
1396                                         TIntermTyped *argi = arg[i]->getAsTyped();\r
1397                                         int size = argi->getNominalSize();\r
1398 \r
1399                                         if(!argi->isMatrix())\r
1400                                         {\r
1401                                                 Instruction *mov = emitCast(result, argi);\r
1402                                                 mov->dst.mask = (0xF << component) & 0xF;\r
1403                                                 mov->src[0].swizzle = readSwizzle(argi, size) << (component * 2);\r
1404 \r
1405                                                 component += size;\r
1406                                         }\r
1407                                         else   // Matrix\r
1408                                         {\r
1409                                                 int column = 0;\r
1410 \r
1411                                                 while(component < resultType.getNominalSize())\r
1412                                                 {\r
1413                                                         Instruction *mov = emitCast(result, 0, argi, column);\r
1414                                                         mov->dst.mask = (0xF << component) & 0xF;\r
1415                                                         mov->src[0].swizzle = readSwizzle(argi, size) << (component * 2);\r
1416 \r
1417                                                         column++;\r
1418                                                         component += size;\r
1419                                                 }\r
1420                                         }\r
1421                                 }\r
1422                         }\r
1423                         break;\r
1424                 case EOpConstructMat2:\r
1425                 case EOpConstructMat2x3:\r
1426                 case EOpConstructMat2x4:\r
1427                 case EOpConstructMat3x2:\r
1428                 case EOpConstructMat3:\r
1429                 case EOpConstructMat3x4:\r
1430                 case EOpConstructMat4x2:\r
1431                 case EOpConstructMat4x3:\r
1432                 case EOpConstructMat4:\r
1433                         if(visit == PostVisit)\r
1434                         {\r
1435                                 TIntermTyped *arg0 = arg[0]->getAsTyped();\r
1436                                 const int outCols = result->getNominalSize();\r
1437                                 const int outRows = result->getSecondarySize();\r
1438 \r
1439                                 if(arg0->isScalar() && arg.size() == 1)   // Construct scale matrix\r
1440                                 {\r
1441                                         for(int i = 0; i < outCols; i++)\r
1442                                         {\r
1443                                                 Instruction *init = emit(sw::Shader::OPCODE_MOV, result, i, &zero);\r
1444                                                 Instruction *mov = emitCast(result, i, arg0, 0);\r
1445                                                 mov->dst.mask = 1 << i;\r
1446                                                 ASSERT(mov->src[0].swizzle == 0x00);\r
1447                                         }\r
1448                                 }\r
1449                                 else if(arg0->isMatrix())\r
1450                                 {\r
1451                                         const int inCols = arg0->getNominalSize();\r
1452                                         const int inRows = arg0->getSecondarySize();\r
1453 \r
1454                                         for(int i = 0; i < outCols; i++)\r
1455                                         {\r
1456                                                 if(i >= inCols || outRows > inRows)\r
1457                                                 {\r
1458                                                         // Initialize to identity matrix\r
1459                                                         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
1460                                                         Instruction *mov = emitCast(result, i, &col, 0);\r
1461                                                 }\r
1462 \r
1463                                                 if(i < inCols)\r
1464                                                 {\r
1465                                                         Instruction *mov = emitCast(result, i, arg0, i);\r
1466                                                         mov->dst.mask = 0xF >> (4 - inRows);\r
1467                                                 }\r
1468                                         }\r
1469                                 }\r
1470                                 else\r
1471                                 {\r
1472                                         int column = 0;\r
1473                                         int row = 0;\r
1474 \r
1475                                         for(int i = 0; i < argumentCount; i++)\r
1476                                         {\r
1477                                                 TIntermTyped *argi = arg[i]->getAsTyped();\r
1478                                                 int size = argi->getNominalSize();\r
1479                                                 int element = 0;\r
1480 \r
1481                                                 while(element < size)\r
1482                                                 {\r
1483                                                         Instruction *mov = emitCast(result, column, argi, 0);\r
1484                                                         mov->dst.mask = (0xF << row) & 0xF;\r
1485                                                         mov->src[0].swizzle = (readSwizzle(argi, size) << (row * 2)) + 0x55 * element;\r
1486 \r
1487                                                         int end = row + size - element;\r
1488                                                         column = end >= outRows ? column + 1 : column;\r
1489                                                         element = element + outRows - row;\r
1490                                                         row = end >= outRows ? 0 : end;\r
1491                                                 }\r
1492                                         }\r
1493                                 }\r
1494                         }\r
1495                         break;\r
1496                 case EOpConstructStruct:\r
1497                         if(visit == PostVisit)\r
1498                         {\r
1499                                 int offset = 0;\r
1500                                 for(int i = 0; i < argumentCount; i++)\r
1501                                 {\r
1502                                         TIntermTyped *argi = arg[i]->getAsTyped();\r
1503                                         int size = argi->totalRegisterCount();\r
1504 \r
1505                                         for(int index = 0; index < size; index++)\r
1506                                         {\r
1507                                                 Instruction *mov = emit(sw::Shader::OPCODE_MOV, result, index + offset, argi, index);\r
1508                                                 mov->dst.mask = writeMask(result, offset + index);\r
1509                                         }\r
1510 \r
1511                                         offset += size;\r
1512                                 }\r
1513                         }\r
1514                         break;\r
1515                 case EOpLessThan:         if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_LT, result, arg[0], arg[1]); break;\r
1516                 case EOpGreaterThan:      if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_GT, result, arg[0], arg[1]); break;\r
1517                 case EOpLessThanEqual:    if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_LE, result, arg[0], arg[1]); break;\r
1518                 case EOpGreaterThanEqual: if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_GE, result, arg[0], arg[1]); break;\r
1519                 case EOpVectorEqual:      if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_EQ, result, arg[0], arg[1]); break;\r
1520                 case EOpVectorNotEqual:   if(visit == PostVisit) emitCmp(sw::Shader::CONTROL_NE, result, arg[0], arg[1]); break;\r
1521                 case EOpMod:              if(visit == PostVisit) emit(sw::Shader::OPCODE_MOD, result, arg[0], arg[1]); break;\r
1522                 case EOpPow:              if(visit == PostVisit) emit(sw::Shader::OPCODE_POW, result, arg[0], arg[1]); break;\r
1523                 case EOpAtan:             if(visit == PostVisit) emit(sw::Shader::OPCODE_ATAN2, result, arg[0], arg[1]); break;\r
1524                 case EOpMin:              if(visit == PostVisit) emit(getOpcode(sw::Shader::OPCODE_MIN, result), result, arg[0], arg[1]); break;\r
1525                 case EOpMax:              if(visit == PostVisit) emit(getOpcode(sw::Shader::OPCODE_MAX, result), result, arg[0], arg[1]); break;\r
1526                 case EOpClamp:\r
1527                         if(visit == PostVisit)\r
1528                         {\r
1529                                 emit(getOpcode(sw::Shader::OPCODE_MAX, result), result, arg[0], arg[1]);\r
1530                                 emit(getOpcode(sw::Shader::OPCODE_MIN, result), result, result, arg[2]);\r
1531                         }\r
1532                         break;\r
1533                 case EOpMix:         if(visit == PostVisit) emit(sw::Shader::OPCODE_LRP, result, arg[2], arg[1], arg[0]); break;\r
1534                 case EOpStep:        if(visit == PostVisit) emit(sw::Shader::OPCODE_STEP, result, arg[0], arg[1]); break;\r
1535                 case EOpSmoothStep:  if(visit == PostVisit) emit(sw::Shader::OPCODE_SMOOTH, result, arg[0], arg[1], arg[2]); break;\r
1536                 case EOpDistance:    if(visit == PostVisit) emit(sw::Shader::OPCODE_DIST(dim(arg[0])), result, arg[0], arg[1]); break;\r
1537                 case EOpDot:         if(visit == PostVisit) emit(sw::Shader::OPCODE_DP(dim(arg[0])), result, arg[0], arg[1]); break;\r
1538                 case EOpCross:       if(visit == PostVisit) emit(sw::Shader::OPCODE_CRS, result, arg[0], arg[1]); break;\r
1539                 case EOpFaceForward: if(visit == PostVisit) emit(sw::Shader::OPCODE_FORWARD(dim(arg[0])), result, arg[0], arg[1], arg[2]); break;\r
1540                 case EOpReflect:     if(visit == PostVisit) emit(sw::Shader::OPCODE_REFLECT(dim(arg[0])), result, arg[0], arg[1]); break;\r
1541                 case EOpRefract:     if(visit == PostVisit) emit(sw::Shader::OPCODE_REFRACT(dim(arg[0])), result, arg[0], arg[1], arg[2]); break;\r
1542                 case EOpMul:\r
1543                         if(visit == PostVisit)\r
1544                         {\r
1545                                 TIntermTyped *arg0 = arg[0]->getAsTyped();\r
1546                                 TIntermTyped *arg1 = arg[1]->getAsTyped();\r
1547                                 ASSERT((arg0->getNominalSize() == arg1->getNominalSize()) && (arg0->getSecondarySize() == arg1->getSecondarySize()));\r
1548 \r
1549                                 int size = arg0->getNominalSize();\r
1550                                 for(int i = 0; i < size; i++)\r
1551                                 {\r
1552                                         emit(sw::Shader::OPCODE_MUL, result, i, arg[0], i, arg[1], i);\r
1553                                 }\r
1554                         }\r
1555                         break;\r
1556                 case EOpOuterProduct:\r
1557                         if(visit == PostVisit)\r
1558                         {\r
1559                                 for(int i = 0; i < dim(arg[1]); i++)\r
1560                                 {\r
1561                                         Instruction *mul = emit(sw::Shader::OPCODE_MUL, result, i, arg[0], 0, arg[1]);\r
1562                                         mul->src[1].swizzle = 0x55 * i;\r
1563                                 }\r
1564                         }\r
1565                         break;\r
1566                 default: UNREACHABLE(node->getOp());\r
1567                 }\r
1568 \r
1569                 return true;\r
1570         }\r
1571 \r
1572         bool OutputASM::visitSelection(Visit visit, TIntermSelection *node)\r
1573         {\r
1574                 if(currentScope != emitScope)\r
1575                 {\r
1576                         return false;\r
1577                 }\r
1578 \r
1579                 TIntermTyped *condition = node->getCondition();\r
1580                 TIntermNode *trueBlock = node->getTrueBlock();\r
1581                 TIntermNode *falseBlock = node->getFalseBlock();\r
1582                 TIntermConstantUnion *constantCondition = condition->getAsConstantUnion();\r
1583 \r
1584                 condition->traverse(this);\r
1585 \r
1586                 if(node->usesTernaryOperator())\r
1587                 {\r
1588                         if(constantCondition)\r
1589                         {\r
1590                                 bool trueCondition = constantCondition->getUnionArrayPointer()->getBConst();\r
1591 \r
1592                                 if(trueCondition)\r
1593                                 {\r
1594                                         trueBlock->traverse(this);\r
1595                                         copy(node, trueBlock);\r
1596                                 }\r
1597                                 else\r
1598                                 {\r
1599                                         falseBlock->traverse(this);\r
1600                                         copy(node, falseBlock);\r
1601                                 }\r
1602                         }\r
1603                         else if(trivial(node, 6))   // Fast to compute both potential results and no side effects\r
1604                         {\r
1605                                 trueBlock->traverse(this);\r
1606                                 falseBlock->traverse(this);\r
1607                                 emit(sw::Shader::OPCODE_SELECT, node, condition, trueBlock, falseBlock);\r
1608                         }\r
1609                         else\r
1610                         {\r
1611                                 emit(sw::Shader::OPCODE_IF, 0, condition);\r
1612 \r
1613                                 if(trueBlock)\r
1614                                 {\r
1615                                         trueBlock->traverse(this);\r
1616                                         copy(node, trueBlock);\r
1617                                 }\r
1618 \r
1619                                 if(falseBlock)\r
1620                                 {\r
1621                                         emit(sw::Shader::OPCODE_ELSE);\r
1622                                         falseBlock->traverse(this);\r
1623                                         copy(node, falseBlock);\r
1624                                 }\r
1625 \r
1626                                 emit(sw::Shader::OPCODE_ENDIF);\r
1627                         }\r
1628                 }\r
1629                 else  // if/else statement\r
1630                 {\r
1631                         if(constantCondition)\r
1632                         {\r
1633                                 bool trueCondition = constantCondition->getUnionArrayPointer()->getBConst();\r
1634 \r
1635                                 if(trueCondition)\r
1636                                 {\r
1637                                         if(trueBlock)\r
1638                                         {\r
1639                                                 trueBlock->traverse(this);\r
1640                                         }\r
1641                                 }\r
1642                                 else\r
1643                                 {\r
1644                                         if(falseBlock)\r
1645                                         {\r
1646                                                 falseBlock->traverse(this);\r
1647                                         }\r
1648                                 }\r
1649                         }\r
1650                         else\r
1651                         {\r
1652                                 emit(sw::Shader::OPCODE_IF, 0, condition);\r
1653 \r
1654                                 if(trueBlock)\r
1655                                 {\r
1656                                         trueBlock->traverse(this);\r
1657                                 }\r
1658 \r
1659                                 if(falseBlock)\r
1660                                 {\r
1661                                         emit(sw::Shader::OPCODE_ELSE);\r
1662                                         falseBlock->traverse(this);\r
1663                                 }\r
1664 \r
1665                                 emit(sw::Shader::OPCODE_ENDIF);\r
1666                         }\r
1667                 }\r
1668 \r
1669                 return false;\r
1670         }\r
1671 \r
1672         bool OutputASM::visitLoop(Visit visit, TIntermLoop *node)\r
1673         {\r
1674                 if(currentScope != emitScope)\r
1675                 {\r
1676                         return false;\r
1677                 }\r
1678 \r
1679                 unsigned int iterations = loopCount(node);\r
1680 \r
1681                 if(iterations == 0)\r
1682                 {\r
1683                         return false;\r
1684                 }\r
1685 \r
1686                 bool unroll = (iterations <= 4);\r
1687 \r
1688                 if(unroll)\r
1689                 {\r
1690                         DetectLoopDiscontinuity detectLoopDiscontinuity;\r
1691                         unroll = !detectLoopDiscontinuity.traverse(node);\r
1692                 }\r
1693 \r
1694                 TIntermNode *init = node->getInit();\r
1695                 TIntermTyped *condition = node->getCondition();\r
1696                 TIntermTyped *expression = node->getExpression();\r
1697                 TIntermNode *body = node->getBody();\r
1698                 Constant True(true);\r
1699 \r
1700                 if(node->getType() == ELoopDoWhile)\r
1701                 {\r
1702                         Temporary iterate(this);\r
1703                         emit(sw::Shader::OPCODE_MOV, &iterate, &True);\r
1704 \r
1705                         emit(sw::Shader::OPCODE_WHILE, 0, &iterate);   // FIXME: Implement real do-while\r
1706 \r
1707                         if(body)\r
1708                         {\r
1709                                 body->traverse(this);\r
1710                         }\r
1711 \r
1712                         emit(sw::Shader::OPCODE_TEST);\r
1713 \r
1714                         condition->traverse(this);\r
1715                         emit(sw::Shader::OPCODE_MOV, &iterate, condition);\r
1716 \r
1717                         emit(sw::Shader::OPCODE_ENDWHILE);\r
1718                 }\r
1719                 else\r
1720                 {\r
1721                         if(init)\r
1722                         {\r
1723                                 init->traverse(this);\r
1724                         }\r
1725 \r
1726                         if(unroll)\r
1727                         {\r
1728                                 for(unsigned int i = 0; i < iterations; i++)\r
1729                                 {\r
1730                                 //      condition->traverse(this);   // Condition could contain statements, but not in an unrollable loop\r
1731 \r
1732                                         if(body)\r
1733                                         {\r
1734                                                 body->traverse(this);\r
1735                                         }\r
1736 \r
1737                                         if(expression)\r
1738                                         {\r
1739                                                 expression->traverse(this);\r
1740                                         }\r
1741                                 }\r
1742                         }\r
1743                         else\r
1744                         {\r
1745                                 if(condition)\r
1746                                 {\r
1747                                         condition->traverse(this);\r
1748                                 }\r
1749                                 else\r
1750                                 {\r
1751                                         condition = &True;\r
1752                                 }\r
1753 \r
1754                                 emit(sw::Shader::OPCODE_WHILE, 0, condition);\r
1755 \r
1756                                 if(body)\r
1757                                 {\r
1758                                         body->traverse(this);\r
1759                                 }\r
1760 \r
1761                                 emit(sw::Shader::OPCODE_TEST);\r
1762 \r
1763                                 if(expression)\r
1764                                 {\r
1765                                         expression->traverse(this);\r
1766                                 }\r
1767 \r
1768                                 if(condition)\r
1769                                 {\r
1770                                         condition->traverse(this);\r
1771                                 }\r
1772 \r
1773                                 emit(sw::Shader::OPCODE_ENDWHILE);\r
1774                         }\r
1775                 }\r
1776 \r
1777                 return false;\r
1778         }\r
1779 \r
1780         bool OutputASM::visitBranch(Visit visit, TIntermBranch *node)\r
1781         {\r
1782                 if(currentScope != emitScope)\r
1783                 {\r
1784                         return false;\r
1785                 }\r
1786 \r
1787                 switch(node->getFlowOp())\r
1788                 {\r
1789                 case EOpKill:      if(visit == PostVisit) emit(sw::Shader::OPCODE_DISCARD);  break;\r
1790                 case EOpBreak:     if(visit == PostVisit) emit(sw::Shader::OPCODE_BREAK);    break;\r
1791                 case EOpContinue:  if(visit == PostVisit) emit(sw::Shader::OPCODE_CONTINUE); break;\r
1792                 case EOpReturn:\r
1793                         if(visit == PostVisit)\r
1794                         {\r
1795                                 TIntermTyped *value = node->getExpression();\r
1796 \r
1797                                 if(value)\r
1798                                 {\r
1799                                         copy(functionArray[currentFunction].ret, value);\r
1800                                 }\r
1801 \r
1802                                 emit(sw::Shader::OPCODE_LEAVE);\r
1803                         }\r
1804                         break;\r
1805                 default: UNREACHABLE(node->getFlowOp());\r
1806                 }\r
1807 \r
1808                 return true;\r
1809         }\r
1810 \r
1811         bool OutputASM::isSamplerRegister(TIntermTyped *operand)\r
1812         {\r
1813                 return operand && isSamplerRegister(operand->getType());\r
1814         }\r
1815 \r
1816         bool OutputASM::isSamplerRegister(const TType &type)\r
1817         {\r
1818                 // A sampler register's qualifiers can be:\r
1819                 // - EvqUniform: The sampler uniform is used as is in the code (default case).\r
1820                 // - EvqTemporary: The sampler is indexed. It's still a sampler register.\r
1821                 // - EvqIn (and other similar types): The sampler has been passed as a function argument. At this point,\r
1822                 //                                    the sampler has been copied and is no longer a sampler register.\r
1823                 return IsSampler(type.getBasicType()) && (type.getQualifier() == EvqUniform || type.getQualifier() == EvqTemporary);\r
1824         }\r
1825 \r
1826         Instruction *OutputASM::emit(sw::Shader::Opcode op, TIntermTyped *dst, TIntermNode *src0, TIntermNode *src1, TIntermNode *src2, TIntermNode *src3, TIntermNode *src4)\r
1827         {\r
1828                 return emit(op, dst, 0, src0, 0, src1, 0, src2, 0, src3, 0, src4, 0);\r
1829         }\r
1830 \r
1831         Instruction *OutputASM::emit(sw::Shader::Opcode op, TIntermTyped *dst, int dstIndex, TIntermNode *src0, int index0, TIntermNode *src1, int index1,\r
1832                                      TIntermNode *src2, int index2, TIntermNode *src3, int index3, TIntermNode *src4, int index4)\r
1833         {\r
1834                 if(isSamplerRegister(dst))\r
1835                 {\r
1836                         op = sw::Shader::OPCODE_NULL;   // Can't assign to a sampler, but this is hit when indexing sampler arrays\r
1837                 }\r
1838 \r
1839                 Instruction *instruction = new Instruction(op);\r
1840 \r
1841                 if(dst)\r
1842                 {\r
1843                         instruction->dst.type = registerType(dst);\r
1844                         instruction->dst.index = registerIndex(dst) + dstIndex;\r
1845                         instruction->dst.mask = writeMask(dst);\r
1846                         instruction->dst.integer = (dst->getBasicType() == EbtInt);\r
1847                 }\r
1848 \r
1849                 argument(instruction->src[0], src0, index0);\r
1850                 argument(instruction->src[1], src1, index1);\r
1851                 argument(instruction->src[2], src2, index2);\r
1852                 argument(instruction->src[3], src3, index3);\r
1853                 argument(instruction->src[4], src4, index4);\r
1854 \r
1855                 shader->append(instruction);\r
1856 \r
1857                 return instruction;\r
1858         }\r
1859 \r
1860         Instruction *OutputASM::emitCast(TIntermTyped *dst, TIntermTyped *src)\r
1861         {\r
1862                 return emitCast(dst, 0, src, 0);\r
1863         }\r
1864 \r
1865         Instruction *OutputASM::emitCast(TIntermTyped *dst, int dstIndex, TIntermTyped *src, int srcIndex)\r
1866         {\r
1867                 switch(src->getBasicType())\r
1868                 {\r
1869                 case EbtBool:\r
1870                         switch(dst->getBasicType())\r
1871                         {\r
1872                         case EbtInt:   return emit(sw::Shader::OPCODE_B2I, dst, dstIndex, src, srcIndex);\r
1873                         case EbtUInt:  return emit(sw::Shader::OPCODE_B2I, dst, dstIndex, src, srcIndex);\r
1874                         case EbtFloat: return emit(sw::Shader::OPCODE_B2F, dst, dstIndex, src, srcIndex);\r
1875                         default:       break;\r
1876                         }\r
1877                         break;\r
1878                 case EbtInt:\r
1879                         switch(dst->getBasicType())\r
1880                         {\r
1881                         case EbtBool:  return emit(sw::Shader::OPCODE_I2B, dst, dstIndex, src, srcIndex);\r
1882                         case EbtFloat: return emit(sw::Shader::OPCODE_I2F, dst, dstIndex, src, srcIndex);\r
1883                         default:       break;\r
1884                         }\r
1885                         break;\r
1886                 case EbtUInt:\r
1887                         switch(dst->getBasicType())\r
1888                         {\r
1889                         case EbtBool:  return emit(sw::Shader::OPCODE_I2B, dst, dstIndex, src, srcIndex);\r
1890                         case EbtFloat: return emit(sw::Shader::OPCODE_U2F, dst, dstIndex, src, srcIndex);\r
1891                         default:       break;\r
1892                         }\r
1893                         break;\r
1894                 case EbtFloat:\r
1895                         switch(dst->getBasicType())\r
1896                         {\r
1897                         case EbtBool: return emit(sw::Shader::OPCODE_F2B, dst, dstIndex, src, srcIndex);\r
1898                         case EbtInt:  return emit(sw::Shader::OPCODE_F2I, dst, dstIndex, src, srcIndex);\r
1899                         case EbtUInt: return emit(sw::Shader::OPCODE_F2U, dst, dstIndex, src, srcIndex);\r
1900                         default:      break;\r
1901                         }\r
1902                         break;\r
1903                 default:\r
1904                         break;\r
1905                 }\r
1906 \r
1907                 ASSERT(src->getBasicType() == dst->getBasicType());\r
1908 \r
1909                 return emit(sw::Shader::OPCODE_MOV, dst, dstIndex, src, srcIndex);\r
1910         }\r
1911 \r
1912         void OutputASM::emitBinary(sw::Shader::Opcode op, TIntermTyped *dst, TIntermNode *src0, TIntermNode *src1, TIntermNode *src2)\r
1913         {\r
1914                 for(int index = 0; index < dst->elementRegisterCount(); index++)\r
1915                 {\r
1916                         emit(op, dst, index, src0, index, src1, index, src2, index);\r
1917                 }\r
1918         }\r
1919 \r
1920         void OutputASM::emitAssign(sw::Shader::Opcode op, TIntermTyped *result, TIntermTyped *lhs, TIntermTyped *src0, TIntermTyped *src1)\r
1921         {\r
1922                 emitBinary(op, result, src0, src1);\r
1923                 assignLvalue(lhs, result);\r
1924         }\r
1925 \r
1926         void OutputASM::emitCmp(sw::Shader::Control cmpOp, TIntermTyped *dst, TIntermNode *left, TIntermNode *right, int index)\r
1927         {\r
1928                 sw::Shader::Opcode opcode;\r
1929                 switch(left->getAsTyped()->getBasicType())\r
1930                 {\r
1931                 case EbtBool:\r
1932                 case EbtInt:\r
1933                         opcode = sw::Shader::OPCODE_ICMP;\r
1934                         break;\r
1935                 case EbtUInt:\r
1936                         opcode = sw::Shader::OPCODE_UCMP;\r
1937                         break;\r
1938                 default:\r
1939                         opcode = sw::Shader::OPCODE_CMP;\r
1940                         break;\r
1941                 }\r
1942 \r
1943                 Instruction *cmp = emit(opcode, dst, 0, left, index, right, index);\r
1944                 cmp->control = cmpOp;\r
1945         }\r
1946 \r
1947         int componentCount(const TType &type, int registers)\r
1948         {\r
1949                 if(registers == 0)\r
1950                 {\r
1951                         return 0;\r
1952                 }\r
1953 \r
1954                 if(type.isArray() && registers >= type.elementRegisterCount())\r
1955                 {\r
1956                         int index = registers / type.elementRegisterCount();\r
1957                         registers -= index * type.elementRegisterCount();\r
1958                         return index * type.getElementSize() + componentCount(type, registers);\r
1959                 }\r
1960 \r
1961                 if(type.isStruct() || type.isInterfaceBlock())\r
1962                 {\r
1963                         const TFieldList& fields = type.getStruct() ? type.getStruct()->fields() : type.getInterfaceBlock()->fields();\r
1964                         int elements = 0;\r
1965 \r
1966                         for(TFieldList::const_iterator field = fields.begin(); field != fields.end(); field++)\r
1967                         {\r
1968                                 const TType &fieldType = *((*field)->type());\r
1969 \r
1970                                 if(fieldType.totalRegisterCount() <= registers)\r
1971                                 {\r
1972                                         registers -= fieldType.totalRegisterCount();\r
1973                                         elements += fieldType.getObjectSize();\r
1974                                 }\r
1975                                 else   // Register within this field\r
1976                                 {\r
1977                                         return elements + componentCount(fieldType, registers);\r
1978                                 }\r
1979                         }\r
1980                 }\r
1981                 else if(type.isMatrix())\r
1982                 {\r
1983                         return registers * type.registerSize();\r
1984                 }\r
1985 \r
1986                 UNREACHABLE(0);\r
1987                 return 0;\r
1988         }\r
1989 \r
1990         int registerSize(const TType &type, int registers)\r
1991         {\r
1992                 if(registers == 0)\r
1993                 {\r
1994                         if(type.isStruct())\r
1995                         {\r
1996                                 return registerSize(*((*(type.getStruct()->fields().begin()))->type()), 0);\r
1997                         }\r
1998                         else if(type.isInterfaceBlock())\r
1999                         {\r
2000                                 return registerSize(*((*(type.getInterfaceBlock()->fields().begin()))->type()), 0);\r
2001                         }\r
2002 \r
2003                         return type.registerSize();\r
2004                 }\r
2005 \r
2006                 if(type.isArray() && registers >= type.elementRegisterCount())\r
2007                 {\r
2008                         int index = registers / type.elementRegisterCount();\r
2009                         registers -= index * type.elementRegisterCount();\r
2010                         return registerSize(type, registers);\r
2011                 }\r
2012 \r
2013                 if(type.isStruct() || type.isInterfaceBlock())\r
2014                 {\r
2015                         const TFieldList& fields = type.getStruct() ? type.getStruct()->fields() : type.getInterfaceBlock()->fields();\r
2016                         int elements = 0;\r
2017 \r
2018                         for(TFieldList::const_iterator field = fields.begin(); field != fields.end(); field++)\r
2019                         {\r
2020                                 const TType &fieldType = *((*field)->type());\r
2021 \r
2022                                 if(fieldType.totalRegisterCount() <= registers)\r
2023                                 {\r
2024                                         registers -= fieldType.totalRegisterCount();\r
2025                                         elements += fieldType.getObjectSize();\r
2026                                 }\r
2027                                 else   // Register within this field\r
2028                                 {\r
2029                                         return registerSize(fieldType, registers);\r
2030                                 }\r
2031                         }\r
2032                 }\r
2033                 else if(type.isMatrix())\r
2034                 {\r
2035                         return registerSize(type, 0);\r
2036                 }\r
2037 \r
2038                 UNREACHABLE(0);\r
2039                 return 0;\r
2040         }\r
2041 \r
2042         void OutputASM::argument(sw::Shader::SourceParameter &parameter, TIntermNode *argument, int index)\r
2043         {\r
2044                 if(argument)\r
2045                 {\r
2046                         TIntermTyped *arg = argument->getAsTyped();\r
2047                         const TType &type = arg->getType();\r
2048                         index = (index >= arg->totalRegisterCount()) ? arg->totalRegisterCount() - 1 : index;\r
2049 \r
2050                         int size = registerSize(type, index);\r
2051 \r
2052                         parameter.type = registerType(arg);\r
2053 \r
2054                         if(arg->getQualifier() == EvqConstExpr)\r
2055                         {\r
2056                                 int component = componentCount(type, index);\r
2057                                 ConstantUnion *constants = arg->getAsConstantUnion()->getUnionArrayPointer();\r
2058 \r
2059                                 for(int i = 0; i < 4; i++)\r
2060                                 {\r
2061                                         if(size == 1)   // Replicate\r
2062                                         {\r
2063                                                 parameter.value[i] = constants[component + 0].getAsFloat();\r
2064                                         }\r
2065                                         else if(i < size)\r
2066                                         {\r
2067                                                 parameter.value[i] = constants[component + i].getAsFloat();\r
2068                                         }\r
2069                                         else\r
2070                                         {\r
2071                                                 parameter.value[i] = 0.0f;\r
2072                                         }\r
2073                                 }\r
2074                         }\r
2075                         else\r
2076                         {\r
2077                                 parameter.index = registerIndex(arg) + index;\r
2078 \r
2079                                 if(isSamplerRegister(arg))\r
2080                                 {\r
2081                                         TIntermBinary *binary = argument->getAsBinaryNode();\r
2082 \r
2083                                         if(binary)\r
2084                                         {\r
2085                                                 TIntermTyped *left = binary->getLeft();\r
2086                                                 TIntermTyped *right = binary->getRight();\r
2087 \r
2088                                                 switch(binary->getOp())\r
2089                                                 {\r
2090                                                 case EOpIndexDirect:\r
2091                                                         parameter.index += right->getAsConstantUnion()->getIConst(0);\r
2092                                                         break;\r
2093                                                 case EOpIndexIndirect:\r
2094                                                         if(left->getArraySize() > 1)\r
2095                                                         {\r
2096                                                                 parameter.rel.type = registerType(binary->getRight());\r
2097                                                                 parameter.rel.index = registerIndex(binary->getRight());\r
2098                                                                 parameter.rel.scale = 1;\r
2099                                                                 parameter.rel.deterministic = true;\r
2100                                                         }\r
2101                                                         break;\r
2102                                                 case EOpIndexDirectStruct:\r
2103                                                 case EOpIndexDirectInterfaceBlock:\r
2104                                                         parameter.index += right->getAsConstantUnion()->getIConst(0);\r
2105                                                         break;\r
2106                                                 default:\r
2107                                                         UNREACHABLE(binary->getOp());\r
2108                                                 }\r
2109                                         }\r
2110                                 }\r
2111                         }\r
2112 \r
2113                         if(!IsSampler(arg->getBasicType()))\r
2114                         {\r
2115                                 parameter.swizzle = readSwizzle(arg, size);\r
2116                         }\r
2117                 }\r
2118         }\r
2119 \r
2120         void OutputASM::copy(TIntermTyped *dst, TIntermNode *src, int offset)\r
2121         {\r
2122                 for(int index = 0; index < dst->totalRegisterCount(); index++)\r
2123                 {\r
2124                         Instruction *mov = emit(sw::Shader::OPCODE_MOV, dst, index, src, offset + index);\r
2125                         mov->dst.mask = writeMask(dst, index);\r
2126                 }\r
2127         }\r
2128 \r
2129         int swizzleElement(int swizzle, int index)\r
2130         {\r
2131                 return (swizzle >> (index * 2)) & 0x03;\r
2132         }\r
2133 \r
2134         int swizzleSwizzle(int leftSwizzle, int rightSwizzle)\r
2135         {\r
2136                 return (swizzleElement(leftSwizzle, swizzleElement(rightSwizzle, 0)) << 0) |\r
2137                        (swizzleElement(leftSwizzle, swizzleElement(rightSwizzle, 1)) << 2) |\r
2138                        (swizzleElement(leftSwizzle, swizzleElement(rightSwizzle, 2)) << 4) |\r
2139                        (swizzleElement(leftSwizzle, swizzleElement(rightSwizzle, 3)) << 6);\r
2140         }\r
2141 \r
2142         void OutputASM::assignLvalue(TIntermTyped *dst, TIntermTyped *src)\r
2143         {\r
2144                 if(src &&\r
2145                         ((src->isVector() && (!dst->isVector() || (dst->getNominalSize() != dst->getNominalSize()))) ||\r
2146                          (src->isMatrix() && (!dst->isMatrix() || (src->getNominalSize() != dst->getNominalSize()) || (src->getSecondarySize() != dst->getSecondarySize())))))\r
2147                 {\r
2148                         return mContext.error(src->getLine(), "Result type should match the l-value type in compound assignment", src->isVector() ? "vector" : "matrix");\r
2149                 }\r
2150 \r
2151                 TIntermBinary *binary = dst->getAsBinaryNode();\r
2152 \r
2153                 if(binary && binary->getOp() == EOpIndexIndirect && dst->isScalar())\r
2154                 {\r
2155                         Instruction *insert = new Instruction(sw::Shader::OPCODE_INSERT);\r
2156 \r
2157                         Temporary address(this);\r
2158                         lvalue(insert->dst, address, dst);\r
2159 \r
2160                         insert->src[0].type = insert->dst.type;\r
2161                         insert->src[0].index = insert->dst.index;\r
2162                         insert->src[0].rel = insert->dst.rel;\r
2163                         argument(insert->src[1], src);\r
2164                         argument(insert->src[2], binary->getRight());\r
2165 \r
2166                         shader->append(insert);\r
2167                 }\r
2168                 else\r
2169                 {\r
2170                         for(int offset = 0; offset < dst->totalRegisterCount(); offset++)\r
2171                         {\r
2172                                 Instruction *mov = new Instruction(sw::Shader::OPCODE_MOV);\r
2173 \r
2174                                 Temporary address(this);\r
2175                                 int swizzle = lvalue(mov->dst, address, dst);\r
2176                                 mov->dst.index += offset;\r
2177 \r
2178                                 if(offset > 0)\r
2179                                 {\r
2180                                         mov->dst.mask = writeMask(dst, offset);\r
2181                                 }\r
2182 \r
2183                                 argument(mov->src[0], src, offset);\r
2184                                 mov->src[0].swizzle = swizzleSwizzle(mov->src[0].swizzle, swizzle);\r
2185 \r
2186                                 shader->append(mov);\r
2187                         }\r
2188                 }\r
2189         }\r
2190 \r
2191         int OutputASM::lvalue(sw::Shader::DestinationParameter &dst, Temporary &address, TIntermTyped *node)\r
2192         {\r
2193                 TIntermTyped *result = node;\r
2194                 TIntermBinary *binary = node->getAsBinaryNode();\r
2195                 TIntermSymbol *symbol = node->getAsSymbolNode();\r
2196 \r
2197                 if(binary)\r
2198                 {\r
2199                         TIntermTyped *left = binary->getLeft();\r
2200                         TIntermTyped *right = binary->getRight();\r
2201 \r
2202                         int leftSwizzle = lvalue(dst, address, left);   // Resolve the l-value of the left side\r
2203 \r
2204                         switch(binary->getOp())\r
2205                         {\r
2206                         case EOpIndexDirect:\r
2207                                 {\r
2208                                         int rightIndex = right->getAsConstantUnion()->getIConst(0);\r
2209 \r
2210                                         if(left->isRegister())\r
2211                                         {\r
2212                                                 int leftMask = dst.mask;\r
2213 \r
2214                                                 dst.mask = 1;\r
2215                                                 while((leftMask & dst.mask) == 0)\r
2216                                                 {\r
2217                                                         dst.mask = dst.mask << 1;\r
2218                                                 }\r
2219 \r
2220                                                 int element = swizzleElement(leftSwizzle, rightIndex);\r
2221                                                 dst.mask = 1 << element;\r
2222 \r
2223                                                 return element;\r
2224                                         }\r
2225                                         else if(left->isArray() || left->isMatrix())\r
2226                                         {\r
2227                                                 dst.index += rightIndex * result->totalRegisterCount();\r
2228                                                 return 0xE4;\r
2229                                         }\r
2230                                         else UNREACHABLE(0);\r
2231                                 }\r
2232                                 break;\r
2233                         case EOpIndexIndirect:\r
2234                                 {\r
2235                                         if(left->isRegister())\r
2236                                         {\r
2237                                                 // Requires INSERT instruction (handled by calling function)\r
2238                                         }\r
2239                                         else if(left->isArray() || left->isMatrix())\r
2240                                         {\r
2241                                                 int scale = result->totalRegisterCount();\r
2242 \r
2243                                                 if(dst.rel.type == sw::Shader::PARAMETER_VOID)   // Use the index register as the relative address directly\r
2244                                                 {\r
2245                                                         if(left->totalRegisterCount() > 1)\r
2246                                                         {\r
2247                                                                 sw::Shader::SourceParameter relativeRegister;\r
2248                                                                 argument(relativeRegister, right);\r
2249 \r
2250                                                                 dst.rel.index = relativeRegister.index;\r
2251                                                                 dst.rel.type = relativeRegister.type;\r
2252                                                                 dst.rel.scale = scale;\r
2253                                                                 dst.rel.deterministic = !(vertexShader && left->getQualifier() == EvqUniform);\r
2254                                                         }\r
2255                                                 }\r
2256                                                 else if(dst.rel.index != registerIndex(&address))   // Move the previous index register to the address register\r
2257                                                 {\r
2258                                                         if(scale == 1)\r
2259                                                         {\r
2260                                                                 Constant oldScale((int)dst.rel.scale);\r
2261                                                                 Instruction *mad = emit(sw::Shader::OPCODE_IMAD, &address, &address, &oldScale, right);\r
2262                                                                 mad->src[0].index = dst.rel.index;\r
2263                                                                 mad->src[0].type = dst.rel.type;\r
2264                                                         }\r
2265                                                         else\r
2266                                                         {\r
2267                                                                 Constant oldScale((int)dst.rel.scale);\r
2268                                                                 Instruction *mul = emit(sw::Shader::OPCODE_IMUL, &address, &address, &oldScale);\r
2269                                                                 mul->src[0].index = dst.rel.index;\r
2270                                                                 mul->src[0].type = dst.rel.type;\r
2271 \r
2272                                                                 Constant newScale(scale);\r
2273                                                                 emit(sw::Shader::OPCODE_IMAD, &address, right, &newScale, &address);\r
2274                                                         }\r
2275 \r
2276                                                         dst.rel.type = sw::Shader::PARAMETER_TEMP;\r
2277                                                         dst.rel.index = registerIndex(&address);\r
2278                                                         dst.rel.scale = 1;\r
2279                                                 }\r
2280                                                 else   // Just add the new index to the address register\r
2281                                                 {\r
2282                                                         if(scale == 1)\r
2283                                                         {\r
2284                                                                 emit(sw::Shader::OPCODE_IADD, &address, &address, right);\r
2285                                                         }\r
2286                                                         else\r
2287                                                         {\r
2288                                                                 Constant newScale(scale);\r
2289                                                                 emit(sw::Shader::OPCODE_IMAD, &address, right, &newScale, &address);\r
2290                                                         }\r
2291                                                 }\r
2292                                         }\r
2293                                         else UNREACHABLE(0);\r
2294                                 }\r
2295                                 break;\r
2296                         case EOpIndexDirectStruct:\r
2297                         case EOpIndexDirectInterfaceBlock:\r
2298                                 {\r
2299                                         const TFieldList& fields = (binary->getOp() == EOpIndexDirectStruct) ?\r
2300                                                                left->getType().getStruct()->fields() :\r
2301                                                                left->getType().getInterfaceBlock()->fields();\r
2302                                         int index = right->getAsConstantUnion()->getIConst(0);\r
2303                                         int fieldOffset = 0;\r
2304 \r
2305                                         for(int i = 0; i < index; i++)\r
2306                                         {\r
2307                                                 fieldOffset += fields[i]->type()->totalRegisterCount();\r
2308                                         }\r
2309 \r
2310                                         dst.type = registerType(left);\r
2311                                         dst.index += fieldOffset;\r
2312                                         dst.mask = writeMask(right);\r
2313 \r
2314                                         return 0xE4;\r
2315                                 }\r
2316                                 break;\r
2317                         case EOpVectorSwizzle:\r
2318                                 {\r
2319                                         ASSERT(left->isRegister());\r
2320 \r
2321                                         int leftMask = dst.mask;\r
2322 \r
2323                                         int swizzle = 0;\r
2324                                         int rightMask = 0;\r
2325 \r
2326                                         TIntermSequence &sequence = right->getAsAggregate()->getSequence();\r
2327 \r
2328                                         for(unsigned int i = 0; i < sequence.size(); i++)\r
2329                                         {\r
2330                                                 int index = sequence[i]->getAsConstantUnion()->getIConst(0);\r
2331 \r
2332                                                 int element = swizzleElement(leftSwizzle, index);\r
2333                                                 rightMask = rightMask | (1 << element);\r
2334                                                 swizzle = swizzle | swizzleElement(leftSwizzle, i) << (element * 2);\r
2335                                         }\r
2336 \r
2337                                         dst.mask = leftMask & rightMask;\r
2338 \r
2339                                         return swizzle;\r
2340                                 }\r
2341                                 break;\r
2342                         default:\r
2343                                 UNREACHABLE(binary->getOp());   // Not an l-value operator\r
2344                                 break;\r
2345                         }\r
2346                 }\r
2347                 else if(symbol)\r
2348                 {\r
2349                         dst.type = registerType(symbol);\r
2350                         dst.index = registerIndex(symbol);\r
2351                         dst.mask = writeMask(symbol);\r
2352                         return 0xE4;\r
2353                 }\r
2354 \r
2355                 return 0xE4;\r
2356         }\r
2357 \r
2358         sw::Shader::ParameterType OutputASM::registerType(TIntermTyped *operand)\r
2359         {\r
2360                 if(isSamplerRegister(operand))\r
2361                 {\r
2362                         return sw::Shader::PARAMETER_SAMPLER;\r
2363                 }\r
2364 \r
2365                 const TQualifier qualifier = operand->getQualifier();\r
2366                 if((EvqFragColor == qualifier) || (EvqFragData == qualifier))\r
2367                 {\r
2368                         if(((EvqFragData == qualifier) && (EvqFragColor == outputQualifier)) ||\r
2369                            ((EvqFragColor == qualifier) && (EvqFragData == outputQualifier)))\r
2370                         {\r
2371                                 mContext.error(operand->getLine(), "static assignment to both gl_FragData and gl_FragColor", "");\r
2372                         }\r
2373                         outputQualifier = qualifier;\r
2374                 }\r
2375 \r
2376                 switch(qualifier)\r
2377                 {\r
2378                 case EvqTemporary:           return sw::Shader::PARAMETER_TEMP;\r
2379                 case EvqGlobal:              return sw::Shader::PARAMETER_TEMP;\r
2380                 case EvqConstExpr:           return sw::Shader::PARAMETER_FLOAT4LITERAL;   // All converted to float\r
2381                 case EvqAttribute:           return sw::Shader::PARAMETER_INPUT;\r
2382                 case EvqVaryingIn:           return sw::Shader::PARAMETER_INPUT;\r
2383                 case EvqVaryingOut:          return sw::Shader::PARAMETER_OUTPUT;\r
2384                 case EvqVertexIn:            return sw::Shader::PARAMETER_INPUT;\r
2385                 case EvqFragmentOut:         return sw::Shader::PARAMETER_COLOROUT;\r
2386                 case EvqVertexOut:           return sw::Shader::PARAMETER_OUTPUT;\r
2387                 case EvqFragmentIn:          return sw::Shader::PARAMETER_INPUT;\r
2388                 case EvqInvariantVaryingIn:  return sw::Shader::PARAMETER_INPUT;    // FIXME: Guarantee invariance at the backend\r
2389                 case EvqInvariantVaryingOut: return sw::Shader::PARAMETER_OUTPUT;   // FIXME: Guarantee invariance at the backend\r
2390                 case EvqSmooth:              return sw::Shader::PARAMETER_OUTPUT;\r
2391                 case EvqFlat:                return sw::Shader::PARAMETER_OUTPUT;\r
2392                 case EvqCentroidOut:         return sw::Shader::PARAMETER_OUTPUT;\r
2393                 case EvqSmoothIn:            return sw::Shader::PARAMETER_INPUT;\r
2394                 case EvqFlatIn:              return sw::Shader::PARAMETER_INPUT;\r
2395                 case EvqCentroidIn:          return sw::Shader::PARAMETER_INPUT;\r
2396                 case EvqUniform:             return sw::Shader::PARAMETER_CONST;\r
2397                 case EvqIn:                  return sw::Shader::PARAMETER_TEMP;\r
2398                 case EvqOut:                 return sw::Shader::PARAMETER_TEMP;\r
2399                 case EvqInOut:               return sw::Shader::PARAMETER_TEMP;\r
2400                 case EvqConstReadOnly:       return sw::Shader::PARAMETER_TEMP;\r
2401                 case EvqPosition:            return sw::Shader::PARAMETER_OUTPUT;\r
2402                 case EvqPointSize:           return sw::Shader::PARAMETER_OUTPUT;\r
2403                 case EvqInstanceID:          return sw::Shader::PARAMETER_MISCTYPE;\r
2404                 case EvqFragCoord:           return sw::Shader::PARAMETER_MISCTYPE;\r
2405                 case EvqFrontFacing:         return sw::Shader::PARAMETER_MISCTYPE;\r
2406                 case EvqPointCoord:          return sw::Shader::PARAMETER_INPUT;\r
2407                 case EvqFragColor:           return sw::Shader::PARAMETER_COLOROUT;\r
2408                 case EvqFragData:            return sw::Shader::PARAMETER_COLOROUT;\r
2409                 case EvqFragDepth:           return sw::Shader::PARAMETER_DEPTHOUT;\r
2410                 default: UNREACHABLE(qualifier);\r
2411                 }\r
2412 \r
2413                 return sw::Shader::PARAMETER_VOID;\r
2414         }\r
2415 \r
2416         unsigned int OutputASM::registerIndex(TIntermTyped *operand)\r
2417         {\r
2418                 if(isSamplerRegister(operand))\r
2419                 {\r
2420                         return samplerRegister(operand);\r
2421                 }\r
2422 \r
2423                 switch(operand->getQualifier())\r
2424                 {\r
2425                 case EvqTemporary:           return temporaryRegister(operand);\r
2426                 case EvqGlobal:              return temporaryRegister(operand);\r
2427                 case EvqConstExpr:           UNREACHABLE(EvqConstExpr);\r
2428                 case EvqAttribute:           return attributeRegister(operand);\r
2429                 case EvqVaryingIn:           return varyingRegister(operand);\r
2430                 case EvqVaryingOut:          return varyingRegister(operand);\r
2431                 case EvqVertexIn:            return attributeRegister(operand);\r
2432                 case EvqFragmentOut:         return fragmentOutputRegister(operand);\r
2433                 case EvqVertexOut:           return varyingRegister(operand);\r
2434                 case EvqFragmentIn:          return varyingRegister(operand);\r
2435                 case EvqInvariantVaryingIn:  return varyingRegister(operand);\r
2436                 case EvqInvariantVaryingOut: return varyingRegister(operand);\r
2437                 case EvqSmooth:              return varyingRegister(operand);\r
2438                 case EvqFlat:                return varyingRegister(operand);\r
2439                 case EvqCentroidOut:         return varyingRegister(operand);\r
2440                 case EvqSmoothIn:            return varyingRegister(operand);\r
2441                 case EvqFlatIn:              return varyingRegister(operand);\r
2442                 case EvqCentroidIn:          return varyingRegister(operand);\r
2443                 case EvqUniform:             return uniformRegister(operand);\r
2444                 case EvqIn:                  return temporaryRegister(operand);\r
2445                 case EvqOut:                 return temporaryRegister(operand);\r
2446                 case EvqInOut:               return temporaryRegister(operand);\r
2447                 case EvqConstReadOnly:       return temporaryRegister(operand);\r
2448                 case EvqPosition:            return varyingRegister(operand);\r
2449                 case EvqPointSize:           return varyingRegister(operand);\r
2450                 case EvqInstanceID:          vertexShader->instanceIdDeclared = true; return 0;\r
2451                 case EvqFragCoord:           pixelShader->vPosDeclared = true;  return 0;\r
2452                 case EvqFrontFacing:         pixelShader->vFaceDeclared = true; return 1;\r
2453                 case EvqPointCoord:          return varyingRegister(operand);\r
2454                 case EvqFragColor:           return 0;\r
2455                 case EvqFragData:            return 0;\r
2456                 case EvqFragDepth:           return 0;\r
2457                 default: UNREACHABLE(operand->getQualifier());\r
2458                 }\r
2459 \r
2460                 return 0;\r
2461         }\r
2462 \r
2463         int OutputASM::writeMask(TIntermTyped *destination, int index)\r
2464         {\r
2465                 if(destination->getQualifier() == EvqPointSize)\r
2466                 {\r
2467                         return 0x2;   // Point size stored in the y component\r
2468                 }\r
2469 \r
2470                 return 0xF >> (4 - registerSize(destination->getType(), index));\r
2471         }\r
2472 \r
2473         int OutputASM::readSwizzle(TIntermTyped *argument, int size)\r
2474         {\r
2475                 if(argument->getQualifier() == EvqPointSize)\r
2476                 {\r
2477                         return 0x55;   // Point size stored in the y component\r
2478                 }\r
2479 \r
2480                 static const unsigned char swizzleSize[5] = {0x00, 0x00, 0x54, 0xA4, 0xE4};   // (void), xxxx, xyyy, xyzz, xyzw\r
2481 \r
2482                 return swizzleSize[size];\r
2483         }\r
2484 \r
2485         // Conservatively checks whether an expression is fast to compute and has no side effects\r
2486         bool OutputASM::trivial(TIntermTyped *expression, int budget)\r
2487         {\r
2488                 if(!expression->isRegister())\r
2489                 {\r
2490                         return false;\r
2491                 }\r
2492 \r
2493                 return cost(expression, budget) >= 0;\r
2494         }\r
2495 \r
2496         // Returns the remaining computing budget (if < 0 the expression is too expensive or has side effects)\r
2497         int OutputASM::cost(TIntermNode *expression, int budget)\r
2498         {\r
2499                 if(budget < 0)\r
2500                 {\r
2501                         return budget;\r
2502                 }\r
2503 \r
2504                 if(expression->getAsSymbolNode())\r
2505                 {\r
2506                         return budget;\r
2507                 }\r
2508                 else if(expression->getAsConstantUnion())\r
2509                 {\r
2510                         return budget;\r
2511                 }\r
2512                 else if(expression->getAsBinaryNode())\r
2513                 {\r
2514                         TIntermBinary *binary = expression->getAsBinaryNode();\r
2515 \r
2516                         switch(binary->getOp())\r
2517                         {\r
2518                         case EOpVectorSwizzle:\r
2519                         case EOpIndexDirect:\r
2520                         case EOpIndexDirectStruct:\r
2521                         case EOpIndexDirectInterfaceBlock:\r
2522                                 return cost(binary->getLeft(), budget - 0);\r
2523                         case EOpAdd:\r
2524                         case EOpSub:\r
2525                         case EOpMul:\r
2526                                 return cost(binary->getLeft(), cost(binary->getRight(), budget - 1));\r
2527                         default:\r
2528                                 return -1;\r
2529                         }\r
2530                 }\r
2531                 else if(expression->getAsUnaryNode())\r
2532                 {\r
2533                         TIntermUnary *unary = expression->getAsUnaryNode();\r
2534 \r
2535                         switch(unary->getOp())\r
2536                         {\r
2537                         case EOpAbs:\r
2538                         case EOpNegative:\r
2539                                 return cost(unary->getOperand(), budget - 1);\r
2540                         default:\r
2541                                 return -1;\r
2542                         }\r
2543                 }\r
2544                 else if(expression->getAsSelectionNode())\r
2545                 {\r
2546                         TIntermSelection *selection = expression->getAsSelectionNode();\r
2547 \r
2548                         if(selection->usesTernaryOperator())\r
2549                         {\r
2550                                 TIntermTyped *condition = selection->getCondition();\r
2551                                 TIntermNode *trueBlock = selection->getTrueBlock();\r
2552                                 TIntermNode *falseBlock = selection->getFalseBlock();\r
2553                                 TIntermConstantUnion *constantCondition = condition->getAsConstantUnion();\r
2554 \r
2555                                 if(constantCondition)\r
2556                                 {\r
2557                                         bool trueCondition = constantCondition->getUnionArrayPointer()->getBConst();\r
2558 \r
2559                                         if(trueCondition)\r
2560                                         {\r
2561                                                 return cost(trueBlock, budget - 0);\r
2562                                         }\r
2563                                         else\r
2564                                         {\r
2565                                                 return cost(falseBlock, budget - 0);\r
2566                                         }\r
2567                                 }\r
2568                                 else\r
2569                                 {\r
2570                                         return cost(trueBlock, cost(falseBlock, budget - 2));\r
2571                                 }\r
2572                         }\r
2573                 }\r
2574 \r
2575                 return -1;\r
2576         }\r
2577 \r
2578         const Function *OutputASM::findFunction(const TString &name)\r
2579         {\r
2580                 for(unsigned int f = 0; f < functionArray.size(); f++)\r
2581                 {\r
2582                         if(functionArray[f].name == name)\r
2583                         {\r
2584                                 return &functionArray[f];\r
2585                         }\r
2586                 }\r
2587 \r
2588                 return 0;\r
2589         }\r
2590 \r
2591         int OutputASM::temporaryRegister(TIntermTyped *temporary)\r
2592         {\r
2593                 return allocate(temporaries, temporary);\r
2594         }\r
2595 \r
2596         int OutputASM::varyingRegister(TIntermTyped *varying)\r
2597         {\r
2598                 int var = lookup(varyings, varying);\r
2599 \r
2600                 if(var == -1)\r
2601                 {\r
2602                         var = allocate(varyings, varying);\r
2603                         int componentCount = varying->registerSize();\r
2604                         int registerCount = varying->totalRegisterCount();\r
2605 \r
2606                         if(pixelShader)\r
2607                         {\r
2608                                 if((var + registerCount) > sw::PixelShader::MAX_INPUT_VARYINGS)\r
2609                                 {\r
2610                                         mContext.error(varying->getLine(), "Varyings packing failed: Too many varyings", "fragment shader");\r
2611                                         return 0;\r
2612                                 }\r
2613 \r
2614                                 if(varying->getQualifier() == EvqPointCoord)\r
2615                                 {\r
2616                                         ASSERT(varying->isRegister());\r
2617                                         if(componentCount >= 1) pixelShader->semantic[var][0] = sw::Shader::Semantic(sw::Shader::USAGE_TEXCOORD, var);\r
2618                                         if(componentCount >= 2) pixelShader->semantic[var][1] = sw::Shader::Semantic(sw::Shader::USAGE_TEXCOORD, var);\r
2619                                         if(componentCount >= 3) pixelShader->semantic[var][2] = sw::Shader::Semantic(sw::Shader::USAGE_TEXCOORD, var);\r
2620                                         if(componentCount >= 4) pixelShader->semantic[var][3] = sw::Shader::Semantic(sw::Shader::USAGE_TEXCOORD, var);\r
2621                                 }\r
2622                                 else\r
2623                                 {\r
2624                                         for(int i = 0; i < varying->totalRegisterCount(); i++)\r
2625                                         {\r
2626                                                 if(componentCount >= 1) pixelShader->semantic[var + i][0] = sw::Shader::Semantic(sw::Shader::USAGE_COLOR, var + i);\r
2627                                                 if(componentCount >= 2) pixelShader->semantic[var + i][1] = sw::Shader::Semantic(sw::Shader::USAGE_COLOR, var + i);\r
2628                                                 if(componentCount >= 3) pixelShader->semantic[var + i][2] = sw::Shader::Semantic(sw::Shader::USAGE_COLOR, var + i);\r
2629                                                 if(componentCount >= 4) pixelShader->semantic[var + i][3] = sw::Shader::Semantic(sw::Shader::USAGE_COLOR, var + i);\r
2630                                         }\r
2631                                 }\r
2632                         }\r
2633                         else if(vertexShader)\r
2634                         {\r
2635                                 if((var + registerCount) > sw::VertexShader::MAX_OUTPUT_VARYINGS)\r
2636                                 {\r
2637                                         mContext.error(varying->getLine(), "Varyings packing failed: Too many varyings", "vertex shader");\r
2638                                         return 0;\r
2639                                 }\r
2640 \r
2641                                 if(varying->getQualifier() == EvqPosition)\r
2642                                 {\r
2643                                         ASSERT(varying->isRegister());\r
2644                                         vertexShader->output[var][0] = sw::Shader::Semantic(sw::Shader::USAGE_POSITION, 0);\r
2645                                         vertexShader->output[var][1] = sw::Shader::Semantic(sw::Shader::USAGE_POSITION, 0);\r
2646                                         vertexShader->output[var][2] = sw::Shader::Semantic(sw::Shader::USAGE_POSITION, 0);\r
2647                                         vertexShader->output[var][3] = sw::Shader::Semantic(sw::Shader::USAGE_POSITION, 0);\r
2648                                         vertexShader->positionRegister = var;\r
2649                                 }\r
2650                                 else if(varying->getQualifier() == EvqPointSize)\r
2651                                 {\r
2652                                         ASSERT(varying->isRegister());\r
2653                                         vertexShader->output[var][0] = sw::Shader::Semantic(sw::Shader::USAGE_PSIZE, 0);\r
2654                                         vertexShader->output[var][1] = sw::Shader::Semantic(sw::Shader::USAGE_PSIZE, 0);\r
2655                                         vertexShader->output[var][2] = sw::Shader::Semantic(sw::Shader::USAGE_PSIZE, 0);\r
2656                                         vertexShader->output[var][3] = sw::Shader::Semantic(sw::Shader::USAGE_PSIZE, 0);\r
2657                                         vertexShader->pointSizeRegister = var;\r
2658                                 }\r
2659                                 else\r
2660                                 {\r
2661                                         // Semantic indexes for user varyings will be assigned during program link to match the pixel shader\r
2662                                 }\r
2663                         }\r
2664                         else UNREACHABLE(0);\r
2665 \r
2666                         declareVarying(varying, var);\r
2667                 }\r
2668 \r
2669                 return var;\r
2670         }\r
2671 \r
2672         void OutputASM::declareVarying(TIntermTyped *varying, int reg)\r
2673         {\r
2674                 if(varying->getQualifier() != EvqPointCoord)   // gl_PointCoord does not need linking\r
2675                 {\r
2676                         const TType &type = varying->getType();\r
2677                         const char *name = varying->getAsSymbolNode()->getSymbol().c_str();\r
2678                         VaryingList &activeVaryings = shaderObject->varyings;\r
2679 \r
2680                         // Check if this varying has been declared before without having a register assigned\r
2681                         for(VaryingList::iterator v = activeVaryings.begin(); v != activeVaryings.end(); v++)\r
2682                         {\r
2683                                 if(v->name == name)\r
2684                                 {\r
2685                                         if(reg >= 0)\r
2686                                         {\r
2687                                                 ASSERT(v->reg < 0 || v->reg == reg);\r
2688                                                 v->reg = reg;\r
2689                                         }\r
2690 \r
2691                                         return;\r
2692                                 }\r
2693                         }\r
2694 \r
2695                         activeVaryings.push_back(glsl::Varying(glVariableType(type), name, varying->getArraySize(), reg, 0));\r
2696                 }\r
2697         }\r
2698 \r
2699         int OutputASM::uniformRegister(TIntermTyped *uniform)\r
2700         {\r
2701                 const TType &type = uniform->getType();\r
2702                 ASSERT(!IsSampler(type.getBasicType()));\r
2703                 TInterfaceBlock *block = type.getAsInterfaceBlock();\r
2704                 TIntermSymbol *symbol = uniform->getAsSymbolNode();\r
2705                 ASSERT(symbol || block);\r
2706 \r
2707                 if(symbol || block)\r
2708                 {\r
2709                         int index = lookup(uniforms, uniform);\r
2710 \r
2711                         if(index == -1)\r
2712                         {\r
2713                                 index = allocate(uniforms, uniform);\r
2714                                 const TString &name = symbol ? symbol->getSymbol() : block->name();\r
2715 \r
2716                                 declareUniform(type, name, index);\r
2717                         }\r
2718 \r
2719                         return index;\r
2720                 }\r
2721 \r
2722                 return 0;\r
2723         }\r
2724 \r
2725         int OutputASM::attributeRegister(TIntermTyped *attribute)\r
2726         {\r
2727                 ASSERT(!attribute->isArray());\r
2728 \r
2729                 int index = lookup(attributes, attribute);\r
2730 \r
2731                 if(index == -1)\r
2732                 {\r
2733                         TIntermSymbol *symbol = attribute->getAsSymbolNode();\r
2734                         ASSERT(symbol);\r
2735 \r
2736                         if(symbol)\r
2737                         {\r
2738                                 index = allocate(attributes, attribute);\r
2739                                 const TType &type = attribute->getType();\r
2740                                 int registerCount = attribute->totalRegisterCount();\r
2741 \r
2742                                 if(vertexShader && (index + registerCount) <= sw::VertexShader::MAX_INPUT_ATTRIBUTES)\r
2743                                 {\r
2744                                         for(int i = 0; i < registerCount; i++)\r
2745                                         {\r
2746                                                 vertexShader->input[index + i] = sw::Shader::Semantic(sw::Shader::USAGE_TEXCOORD, index + i);\r
2747                                         }\r
2748                                 }\r
2749 \r
2750                                 ActiveAttributes &activeAttributes = shaderObject->activeAttributes;\r
2751 \r
2752                                 const char *name = symbol->getSymbol().c_str();\r
2753                                 activeAttributes.push_back(Attribute(glVariableType(type), name, type.getArraySize(), type.getLayoutQualifier().location, index));\r
2754                         }\r
2755                 }\r
2756 \r
2757                 return index;\r
2758         }\r
2759 \r
2760         int OutputASM::fragmentOutputRegister(TIntermTyped *fragmentOutput)\r
2761         {\r
2762                 return allocate(fragmentOutputs, fragmentOutput);\r
2763         }\r
2764 \r
2765         int OutputASM::samplerRegister(TIntermTyped *sampler)\r
2766         {\r
2767                 ASSERT(IsSampler(sampler->getType().getBasicType()));\r
2768                 TIntermSymbol *symbol = sampler->getAsSymbolNode();\r
2769                 TIntermBinary *binary = sampler->getAsBinaryNode();\r
2770 \r
2771                 if(symbol)\r
2772                 {\r
2773                         return samplerRegister(symbol);\r
2774                 }\r
2775                 else if(binary)\r
2776                 {\r
2777                         ASSERT(binary->getOp() == EOpIndexDirect || binary->getOp() == EOpIndexIndirect ||\r
2778                                    binary->getOp() == EOpIndexDirectStruct || binary->getOp() == EOpIndexDirectInterfaceBlock);\r
2779 \r
2780                         return samplerRegister(binary->getLeft());   // Index added later\r
2781                 }\r
2782                 else UNREACHABLE(0);\r
2783 \r
2784                 return 0;\r
2785         }\r
2786 \r
2787         int OutputASM::samplerRegister(TIntermSymbol *sampler)\r
2788         {\r
2789                 const TType &type = sampler->getType();\r
2790                 ASSERT(IsSampler(type.getBasicType()) || type.getStruct());   // Structures can contain samplers\r
2791 \r
2792                 int index = lookup(samplers, sampler);\r
2793 \r
2794                 if(index == -1)\r
2795                 {\r
2796                         index = allocate(samplers, sampler);\r
2797 \r
2798                         if(sampler->getQualifier() == EvqUniform)\r
2799                         {\r
2800                                 const char *name = sampler->getSymbol().c_str();\r
2801                                 declareUniform(type, name, index);\r
2802                         }\r
2803                 }\r
2804 \r
2805                 return index;\r
2806         }\r
2807 \r
2808         int OutputASM::lookup(VariableArray &list, TIntermTyped *variable)\r
2809         {\r
2810                 for(unsigned int i = 0; i < list.size(); i++)\r
2811                 {\r
2812                         if(list[i] == variable)\r
2813                         {\r
2814                                 return i;   // Pointer match\r
2815                         }\r
2816                 }\r
2817 \r
2818                 TIntermSymbol *varSymbol = variable->getAsSymbolNode();\r
2819                 TInterfaceBlock *varBlock = variable->getType().getAsInterfaceBlock();\r
2820 \r
2821                 if(varBlock)\r
2822                 {\r
2823                         for(unsigned int i = 0; i < list.size(); i++)\r
2824                         {\r
2825                                 if(list[i])\r
2826                                 {\r
2827                                         TInterfaceBlock *listBlock = list[i]->getType().getAsInterfaceBlock();\r
2828 \r
2829                                         if(listBlock)\r
2830                                         {\r
2831                                                 if(listBlock->name() == varBlock->name())\r
2832                                                 {\r
2833                                                         ASSERT(listBlock->arraySize() == varBlock->arraySize());\r
2834                                                         ASSERT(listBlock->fields() == varBlock->fields());\r
2835                                                         ASSERT(listBlock->blockStorage() == varBlock->blockStorage());\r
2836                                                         ASSERT(listBlock->matrixPacking() == varBlock->matrixPacking());\r
2837 \r
2838                                                         return i;\r
2839                                                 }\r
2840                                         }\r
2841                                 }\r
2842                         }\r
2843                 }\r
2844                 else if(varSymbol)\r
2845                 {\r
2846                         for(unsigned int i = 0; i < list.size(); i++)\r
2847                         {\r
2848                                 if(list[i])\r
2849                                 {\r
2850                                         TIntermSymbol *listSymbol = list[i]->getAsSymbolNode();\r
2851 \r
2852                                         if(listSymbol)\r
2853                                         {\r
2854                                                 if(listSymbol->getId() == varSymbol->getId())\r
2855                                                 {\r
2856                                                         ASSERT(listSymbol->getSymbol() == varSymbol->getSymbol());\r
2857                                                         ASSERT(listSymbol->getType() == varSymbol->getType());\r
2858                                                         ASSERT(listSymbol->getQualifier() == varSymbol->getQualifier());\r
2859 \r
2860                                                         return i;\r
2861                                                 }\r
2862                                         }\r
2863                                 }\r
2864                         }\r
2865                 }\r
2866 \r
2867                 return -1;\r
2868         }\r
2869 \r
2870         int OutputASM::allocate(VariableArray &list, TIntermTyped *variable)\r
2871         {\r
2872                 int index = lookup(list, variable);\r
2873 \r
2874                 if(index == -1)\r
2875                 {\r
2876                         unsigned int registerCount = variable->totalRegisterCount();\r
2877 \r
2878                         for(unsigned int i = 0; i < list.size(); i++)\r
2879                         {\r
2880                                 if(list[i] == 0)\r
2881                                 {\r
2882                                         unsigned int j = 1;\r
2883                                         for( ; j < registerCount && (i + j) < list.size(); j++)\r
2884                                         {\r
2885                                                 if(list[i + j] != 0)\r
2886                                                 {\r
2887                                                         break;\r
2888                                                 }\r
2889                                         }\r
2890 \r
2891                                         if(j == registerCount)   // Found free slots\r
2892                                         {\r
2893                                                 for(unsigned int j = 0; j < registerCount; j++)\r
2894                                                 {\r
2895                                                         list[i + j] = variable;\r
2896                                                 }\r
2897 \r
2898                                                 return i;\r
2899                                         }\r
2900                                 }\r
2901                         }\r
2902 \r
2903                         index = list.size();\r
2904 \r
2905                         for(unsigned int i = 0; i < registerCount; i++)\r
2906                         {\r
2907                                 list.push_back(variable);\r
2908                         }\r
2909                 }\r
2910 \r
2911                 return index;\r
2912         }\r
2913 \r
2914         void OutputASM::free(VariableArray &list, TIntermTyped *variable)\r
2915         {\r
2916                 int index = lookup(list, variable);\r
2917 \r
2918                 if(index >= 0)\r
2919                 {\r
2920                         list[index] = 0;\r
2921                 }\r
2922         }\r
2923 \r
2924         int OutputASM::declareUniform(const TType &type, const TString &name, int registerIndex, int blockId, BlockLayoutEncoder* encoder)\r
2925         {\r
2926                 const TStructure *structure = type.getStruct();\r
2927                 const TInterfaceBlock *block = (type.isInterfaceBlock() || (blockId == -1)) ? type.getInterfaceBlock() : nullptr;\r
2928                 ActiveUniforms &activeUniforms = shaderObject->activeUniforms;\r
2929 \r
2930                 if(!structure && !block)\r
2931                 {\r
2932                         const BlockMemberInfo blockInfo = encoder ? encoder->encodeType(type) : BlockMemberInfo::getDefaultBlockInfo();\r
2933                         if(blockId >= 0)\r
2934                         {\r
2935                                 blockDefinitions[blockId].indexMap[registerIndex] = TypedMemberInfo(blockInfo, type);\r
2936                                 shaderObject->activeUniformBlocks[blockId].fields.push_back(activeUniforms.size());\r
2937                         }\r
2938                         int fieldRegisterIndex = encoder ? shaderObject->activeUniformBlocks[blockId].registerIndex + BlockLayoutEncoder::getBlockRegister(blockInfo) : registerIndex;\r
2939                         activeUniforms.push_back(Uniform(glVariableType(type), glVariablePrecision(type), name.c_str(), type.getArraySize(),\r
2940                                                          fieldRegisterIndex, blockId, blockInfo));\r
2941                         if(isSamplerRegister(type))\r
2942                         {\r
2943                                 for(int i = 0; i < type.totalRegisterCount(); i++)\r
2944                                 {\r
2945                                         shader->declareSampler(fieldRegisterIndex + i);\r
2946                                 }\r
2947                         }\r
2948                 }\r
2949                 else if(block)\r
2950                 {\r
2951                         ActiveUniformBlocks &activeUniformBlocks = shaderObject->activeUniformBlocks;\r
2952                         const TFieldList& fields = block->fields();\r
2953                         const TString &blockName = block->name();\r
2954                         int fieldRegisterIndex = registerIndex;\r
2955                         bool isUniformBlockMember = !type.isInterfaceBlock() && (blockId == -1);\r
2956 \r
2957                         if(isUniformBlockMember)\r
2958                         {\r
2959                                 // This is a uniform that's part of a block, let's see if the block is already defined\r
2960                                 for(size_t i = 0; i < activeUniformBlocks.size(); ++i)\r
2961                                 {\r
2962                                         if(activeUniformBlocks[i].name == blockName.c_str())\r
2963                                         {\r
2964                                                 // The block is already defined, find the register for the current uniform and return it\r
2965                                                 for(size_t j = 0; j < fields.size(); j++)\r
2966                                                 {\r
2967                                                         const TString &fieldName = fields[j]->name();\r
2968                                                         if(fieldName == name)\r
2969                                                         {\r
2970                                                                 return fieldRegisterIndex;\r
2971                                                         }\r
2972 \r
2973                                                         fieldRegisterIndex += fields[j]->type()->totalRegisterCount();\r
2974                                                 }\r
2975 \r
2976                                                 ASSERT(false);\r
2977                                                 return fieldRegisterIndex;\r
2978                                         }\r
2979                                 }\r
2980                         }\r
2981 \r
2982                         blockId = activeUniformBlocks.size();\r
2983                         bool isRowMajor = block->matrixPacking() == EmpRowMajor;\r
2984                         activeUniformBlocks.push_back(UniformBlock(blockName.c_str(), 0, block->arraySize(),\r
2985                                                                    block->blockStorage(), isRowMajor, registerIndex, blockId));\r
2986                         blockDefinitions.push_back(BlockDefinition());\r
2987 \r
2988                         Std140BlockEncoder currentBlockEncoder(isRowMajor);\r
2989                         currentBlockEncoder.enterAggregateType();\r
2990                         for(size_t i = 0; i < fields.size(); i++)\r
2991                         {\r
2992                                 const TType &fieldType = *(fields[i]->type());\r
2993                                 const TString &fieldName = fields[i]->name();\r
2994                                 if(isUniformBlockMember && (fieldName == name))\r
2995                                 {\r
2996                                         registerIndex = fieldRegisterIndex;\r
2997                                 }\r
2998 \r
2999                                 const TString uniformName = block->hasInstanceName() ? blockName + "." + fieldName : fieldName;\r
3000 \r
3001                                 declareUniform(fieldType, uniformName, fieldRegisterIndex, blockId, &currentBlockEncoder);\r
3002                                 fieldRegisterIndex += fieldType.totalRegisterCount();\r
3003                         }\r
3004                         currentBlockEncoder.exitAggregateType();\r
3005                         activeUniformBlocks[blockId].dataSize = currentBlockEncoder.getBlockSize();\r
3006                 }\r
3007                 else\r
3008                 {\r
3009                         int fieldRegisterIndex = registerIndex;\r
3010 \r
3011                         const TFieldList& fields = structure->fields();\r
3012                         if(type.isArray() && (structure || type.isInterfaceBlock()))\r
3013                         {\r
3014                                 for(int i = 0; i < type.getArraySize(); i++)\r
3015                                 {\r
3016                                         if(encoder)\r
3017                                         {\r
3018                                                 encoder->enterAggregateType();\r
3019                                         }\r
3020                                         for(size_t j = 0; j < fields.size(); j++)\r
3021                                         {\r
3022                                                 const TType &fieldType = *(fields[j]->type());\r
3023                                                 const TString &fieldName = fields[j]->name();\r
3024                                                 const TString uniformName = name + "[" + str(i) + "]." + fieldName;\r
3025 \r
3026                                                 declareUniform(fieldType, uniformName, fieldRegisterIndex, blockId, encoder);\r
3027                                                 fieldRegisterIndex += fieldType.totalRegisterCount();\r
3028                                         }\r
3029                                         if(encoder)\r
3030                                         {\r
3031                                                 encoder->exitAggregateType();\r
3032                                         }\r
3033                                 }\r
3034                         }\r
3035                         else\r
3036                         {\r
3037                                 if(encoder)\r
3038                                 {\r
3039                                         encoder->enterAggregateType();\r
3040                                 }\r
3041                                 for(size_t i = 0; i < fields.size(); i++)\r
3042                                 {\r
3043                                         const TType &fieldType = *(fields[i]->type());\r
3044                                         const TString &fieldName = fields[i]->name();\r
3045                                         const TString uniformName = name + "." + fieldName;\r
3046 \r
3047                                         declareUniform(fieldType, uniformName, fieldRegisterIndex, blockId, encoder);\r
3048                                         fieldRegisterIndex += fieldType.totalRegisterCount();\r
3049                                 }\r
3050                                 if(encoder)\r
3051                                 {\r
3052                                         encoder->exitAggregateType();\r
3053                                 }\r
3054                         }\r
3055                 }\r
3056 \r
3057                 return registerIndex;\r
3058         }\r
3059 \r
3060         GLenum OutputASM::glVariableType(const TType &type)\r
3061         {\r
3062                 switch(type.getBasicType())\r
3063                 {\r
3064                 case EbtFloat:\r
3065                         if(type.isScalar())\r
3066                         {\r
3067                                 return GL_FLOAT;\r
3068                         }\r
3069                         else if(type.isVector())\r
3070                         {\r
3071                                 switch(type.getNominalSize())\r
3072                                 {\r
3073                                 case 2: return GL_FLOAT_VEC2;\r
3074                                 case 3: return GL_FLOAT_VEC3;\r
3075                                 case 4: return GL_FLOAT_VEC4;\r
3076                                 default: UNREACHABLE(type.getNominalSize());\r
3077                                 }\r
3078                         }\r
3079                         else if(type.isMatrix())\r
3080                         {\r
3081                                 switch(type.getNominalSize())\r
3082                                 {\r
3083                                 case 2:\r
3084                                         switch(type.getSecondarySize())\r
3085                                         {\r
3086                                         case 2: return GL_FLOAT_MAT2;\r
3087                                         case 3: return GL_FLOAT_MAT2x3;\r
3088                                         case 4: return GL_FLOAT_MAT2x4;\r
3089                                         default: UNREACHABLE(type.getSecondarySize());\r
3090                                         }\r
3091                                 case 3:\r
3092                                         switch(type.getSecondarySize())\r
3093                                         {\r
3094                                         case 2: return GL_FLOAT_MAT3x2;\r
3095                                         case 3: return GL_FLOAT_MAT3;\r
3096                                         case 4: return GL_FLOAT_MAT3x4;\r
3097                                         default: UNREACHABLE(type.getSecondarySize());\r
3098                                         }\r
3099                                 case 4:\r
3100                                         switch(type.getSecondarySize())\r
3101                                         {\r
3102                                         case 2: return GL_FLOAT_MAT4x2;\r
3103                                         case 3: return GL_FLOAT_MAT4x3;\r
3104                                         case 4: return GL_FLOAT_MAT4;\r
3105                                         default: UNREACHABLE(type.getSecondarySize());\r
3106                                         }\r
3107                                 default: UNREACHABLE(type.getNominalSize());\r
3108                                 }\r
3109                         }\r
3110                         else UNREACHABLE(0);\r
3111                         break;\r
3112                 case EbtInt:\r
3113                         if(type.isScalar())\r
3114                         {\r
3115                                 return GL_INT;\r
3116                         }\r
3117                         else if(type.isVector())\r
3118                         {\r
3119                                 switch(type.getNominalSize())\r
3120                                 {\r
3121                                 case 2: return GL_INT_VEC2;\r
3122                                 case 3: return GL_INT_VEC3;\r
3123                                 case 4: return GL_INT_VEC4;\r
3124                                 default: UNREACHABLE(type.getNominalSize());\r
3125                                 }\r
3126                         }\r
3127                         else UNREACHABLE(0);\r
3128                         break;\r
3129                 case EbtUInt:\r
3130                         if(type.isScalar())\r
3131                         {\r
3132                                 return GL_UNSIGNED_INT;\r
3133                         }\r
3134                         else if(type.isVector())\r
3135                         {\r
3136                                 switch(type.getNominalSize())\r
3137                                 {\r
3138                                 case 2: return GL_UNSIGNED_INT_VEC2;\r
3139                                 case 3: return GL_UNSIGNED_INT_VEC3;\r
3140                                 case 4: return GL_UNSIGNED_INT_VEC4;\r
3141                                 default: UNREACHABLE(type.getNominalSize());\r
3142                                 }\r
3143                         }\r
3144                         else UNREACHABLE(0);\r
3145                         break;\r
3146                 case EbtBool:\r
3147                         if(type.isScalar())\r
3148                         {\r
3149                                 return GL_BOOL;\r
3150                         }\r
3151                         else if(type.isVector())\r
3152                         {\r
3153                                 switch(type.getNominalSize())\r
3154                                 {\r
3155                                 case 2: return GL_BOOL_VEC2;\r
3156                                 case 3: return GL_BOOL_VEC3;\r
3157                                 case 4: return GL_BOOL_VEC4;\r
3158                                 default: UNREACHABLE(type.getNominalSize());\r
3159                                 }\r
3160                         }\r
3161                         else UNREACHABLE(0);\r
3162                         break;\r
3163                 case EbtSampler2D:\r
3164                         return GL_SAMPLER_2D;\r
3165                 case EbtISampler2D:\r
3166                         return GL_INT_SAMPLER_2D;\r
3167                 case EbtUSampler2D:\r
3168                         return GL_UNSIGNED_INT_SAMPLER_2D;\r
3169                 case EbtSamplerCube:\r
3170                         return GL_SAMPLER_CUBE;\r
3171                 case EbtISamplerCube:\r
3172                         return GL_INT_SAMPLER_CUBE;\r
3173                 case EbtUSamplerCube:\r
3174                         return GL_UNSIGNED_INT_SAMPLER_CUBE;\r
3175                 case EbtSamplerExternalOES:\r
3176                         return GL_SAMPLER_EXTERNAL_OES;\r
3177                 case EbtSampler3D:\r
3178                         return GL_SAMPLER_3D_OES;\r
3179                 case EbtISampler3D:\r
3180                         return GL_INT_SAMPLER_3D;\r
3181                 case EbtUSampler3D:\r
3182                         return GL_UNSIGNED_INT_SAMPLER_3D;\r
3183                 case EbtSampler2DArray:\r
3184                         return GL_SAMPLER_2D_ARRAY;\r
3185                 case EbtISampler2DArray:\r
3186                         return GL_INT_SAMPLER_2D_ARRAY;\r
3187                 case EbtUSampler2DArray:\r
3188                         return GL_UNSIGNED_INT_SAMPLER_2D_ARRAY;\r
3189                 case EbtSampler2DShadow:\r
3190                         return GL_SAMPLER_2D_SHADOW;\r
3191                 case EbtSamplerCubeShadow:\r
3192                         return GL_SAMPLER_CUBE_SHADOW;\r
3193                 case EbtSampler2DArrayShadow:\r
3194                         return GL_SAMPLER_2D_ARRAY_SHADOW;\r
3195                 default:\r
3196                         UNREACHABLE(type.getBasicType());\r
3197                         break;\r
3198                 }\r
3199 \r
3200                 return GL_NONE;\r
3201         }\r
3202 \r
3203         GLenum OutputASM::glVariablePrecision(const TType &type)\r
3204         {\r
3205                 if(type.getBasicType() == EbtFloat)\r
3206                 {\r
3207                         switch(type.getPrecision())\r
3208                         {\r
3209                         case EbpHigh:   return GL_HIGH_FLOAT;\r
3210                         case EbpMedium: return GL_MEDIUM_FLOAT;\r
3211                         case EbpLow:    return GL_LOW_FLOAT;\r
3212                         case EbpUndefined:\r
3213                                 // Should be defined as the default precision by the parser\r
3214                         default: UNREACHABLE(type.getPrecision());\r
3215                         }\r
3216                 }\r
3217                 else if(type.getBasicType() == EbtInt)\r
3218                 {\r
3219                         switch(type.getPrecision())\r
3220                         {\r
3221                         case EbpHigh:   return GL_HIGH_INT;\r
3222                         case EbpMedium: return GL_MEDIUM_INT;\r
3223                         case EbpLow:    return GL_LOW_INT;\r
3224                         case EbpUndefined:\r
3225                                 // Should be defined as the default precision by the parser\r
3226                         default: UNREACHABLE(type.getPrecision());\r
3227                         }\r
3228                 }\r
3229 \r
3230                 // Other types (boolean, sampler) don't have a precision\r
3231                 return GL_NONE;\r
3232         }\r
3233 \r
3234         int OutputASM::dim(TIntermNode *v)\r
3235         {\r
3236                 TIntermTyped *vector = v->getAsTyped();\r
3237                 ASSERT(vector && vector->isRegister());\r
3238                 return vector->getNominalSize();\r
3239         }\r
3240 \r
3241         int OutputASM::dim2(TIntermNode *m)\r
3242         {\r
3243                 TIntermTyped *matrix = m->getAsTyped();\r
3244                 ASSERT(matrix && matrix->isMatrix() && !matrix->isArray());\r
3245                 return matrix->getSecondarySize();\r
3246         }\r
3247 \r
3248         // Returns ~0u if no loop count could be determined\r
3249         unsigned int OutputASM::loopCount(TIntermLoop *node)\r
3250         {\r
3251                 // Parse loops of the form:\r
3252                 // for(int index = initial; index [comparator] limit; index += increment)\r
3253                 TIntermSymbol *index = 0;\r
3254                 TOperator comparator = EOpNull;\r
3255                 int initial = 0;\r
3256                 int limit = 0;\r
3257                 int increment = 0;\r
3258 \r
3259                 // Parse index name and intial value\r
3260                 if(node->getInit())\r
3261                 {\r
3262                         TIntermAggregate *init = node->getInit()->getAsAggregate();\r
3263 \r
3264                         if(init)\r
3265                         {\r
3266                                 TIntermSequence &sequence = init->getSequence();\r
3267                                 TIntermTyped *variable = sequence[0]->getAsTyped();\r
3268 \r
3269                                 if(variable && variable->getQualifier() == EvqTemporary)\r
3270                                 {\r
3271                                         TIntermBinary *assign = variable->getAsBinaryNode();\r
3272 \r
3273                                         if(assign->getOp() == EOpInitialize)\r
3274                                         {\r
3275                                                 TIntermSymbol *symbol = assign->getLeft()->getAsSymbolNode();\r
3276                                                 TIntermConstantUnion *constant = assign->getRight()->getAsConstantUnion();\r
3277 \r
3278                                                 if(symbol && constant)\r
3279                                                 {\r
3280                                                         if(constant->getBasicType() == EbtInt && constant->getNominalSize() == 1)\r
3281                                                         {\r
3282                                                                 index = symbol;\r
3283                                                                 initial = constant->getUnionArrayPointer()[0].getIConst();\r
3284                                                         }\r
3285                                                 }\r
3286                                         }\r
3287                                 }\r
3288                         }\r
3289                 }\r
3290 \r
3291                 // Parse comparator and limit value\r
3292                 if(index && node->getCondition())\r
3293                 {\r
3294                         TIntermBinary *test = node->getCondition()->getAsBinaryNode();\r
3295 \r
3296                         if(test && test->getLeft()->getAsSymbolNode()->getId() == index->getId())\r
3297                         {\r
3298                                 TIntermConstantUnion *constant = test->getRight()->getAsConstantUnion();\r
3299 \r
3300                                 if(constant)\r
3301                                 {\r
3302                                         if(constant->getBasicType() == EbtInt && constant->getNominalSize() == 1)\r
3303                                         {\r
3304                                                 comparator = test->getOp();\r
3305                                                 limit = constant->getUnionArrayPointer()[0].getIConst();\r
3306                                         }\r
3307                                 }\r
3308                         }\r
3309                 }\r
3310 \r
3311                 // Parse increment\r
3312                 if(index && comparator != EOpNull && node->getExpression())\r
3313                 {\r
3314                         TIntermBinary *binaryTerminal = node->getExpression()->getAsBinaryNode();\r
3315                         TIntermUnary *unaryTerminal = node->getExpression()->getAsUnaryNode();\r
3316 \r
3317                         if(binaryTerminal)\r
3318                         {\r
3319                                 TOperator op = binaryTerminal->getOp();\r
3320                                 TIntermConstantUnion *constant = binaryTerminal->getRight()->getAsConstantUnion();\r
3321 \r
3322                                 if(constant)\r
3323                                 {\r
3324                                         if(constant->getBasicType() == EbtInt && constant->getNominalSize() == 1)\r
3325                                         {\r
3326                                                 int value = constant->getUnionArrayPointer()[0].getIConst();\r
3327 \r
3328                                                 switch(op)\r
3329                                                 {\r
3330                                                 case EOpAddAssign: increment = value;  break;\r
3331                                                 case EOpSubAssign: increment = -value; break;\r
3332                                                 default: UNIMPLEMENTED();\r
3333                                                 }\r
3334                                         }\r
3335                                 }\r
3336                         }\r
3337                         else if(unaryTerminal)\r
3338                         {\r
3339                                 TOperator op = unaryTerminal->getOp();\r
3340 \r
3341                                 switch(op)\r
3342                                 {\r
3343                                 case EOpPostIncrement: increment = 1;  break;\r
3344                                 case EOpPostDecrement: increment = -1; break;\r
3345                                 case EOpPreIncrement:  increment = 1;  break;\r
3346                                 case EOpPreDecrement:  increment = -1; break;\r
3347                                 default: UNIMPLEMENTED();\r
3348                                 }\r
3349                         }\r
3350                 }\r
3351 \r
3352                 if(index && comparator != EOpNull && increment != 0)\r
3353                 {\r
3354                         if(comparator == EOpLessThanEqual)\r
3355                         {\r
3356                                 comparator = EOpLessThan;\r
3357                                 limit += 1;\r
3358                         }\r
3359 \r
3360                         if(comparator == EOpLessThan)\r
3361                         {\r
3362                                 int iterations = (limit - initial) / increment;\r
3363 \r
3364                                 if(iterations <= 0)\r
3365                                 {\r
3366                                         iterations = 0;\r
3367                                 }\r
3368 \r
3369                                 return iterations;\r
3370                         }\r
3371                         else UNIMPLEMENTED();   // Falls through\r
3372                 }\r
3373 \r
3374                 return ~0u;\r
3375         }\r
3376 \r
3377         bool DetectLoopDiscontinuity::traverse(TIntermNode *node)\r
3378         {\r
3379                 loopDepth = 0;\r
3380                 loopDiscontinuity = false;\r
3381 \r
3382                 node->traverse(this);\r
3383 \r
3384                 return loopDiscontinuity;\r
3385         }\r
3386 \r
3387         bool DetectLoopDiscontinuity::visitLoop(Visit visit, TIntermLoop *loop)\r
3388         {\r
3389                 if(visit == PreVisit)\r
3390                 {\r
3391                         loopDepth++;\r
3392                 }\r
3393                 else if(visit == PostVisit)\r
3394                 {\r
3395                         loopDepth++;\r
3396                 }\r
3397 \r
3398                 return true;\r
3399         }\r
3400 \r
3401         bool DetectLoopDiscontinuity::visitBranch(Visit visit, TIntermBranch *node)\r
3402         {\r
3403                 if(loopDiscontinuity)\r
3404                 {\r
3405                         return false;\r
3406                 }\r
3407 \r
3408                 if(!loopDepth)\r
3409                 {\r
3410                         return true;\r
3411                 }\r
3412 \r
3413                 switch(node->getFlowOp())\r
3414                 {\r
3415                 case EOpKill:\r
3416                         break;\r
3417                 case EOpBreak:\r
3418                 case EOpContinue:\r
3419                 case EOpReturn:\r
3420                         loopDiscontinuity = true;\r
3421                         break;\r
3422                 default: UNREACHABLE(node->getFlowOp());\r
3423                 }\r
3424 \r
3425                 return !loopDiscontinuity;\r
3426         }\r
3427 \r
3428         bool DetectLoopDiscontinuity::visitAggregate(Visit visit, TIntermAggregate *node)\r
3429         {\r
3430                 return !loopDiscontinuity;\r
3431         }\r
3432 }\r