blob: 798a69af7eb647c5af2cd42f1908cebe7e918cbf [file] [log] [blame]
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00001//
2// Copyright (c) 2002-2010 The ANGLE Project Authors. All rights reserved.
3// Use of this source code is governed by a BSD-style license that can be
4// found in the LICENSE file.
5//
6
daniel@transgaming.combbf56f72010-04-20 18:52:13 +00007#include "compiler/localintermediate.h"
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +00008
9//
10// Two purposes:
11// 1. Show an example of how to iterate tree. Functions can
12// also directly call Traverse() on children themselves to
13// have finer grained control over the process than shown here.
14// See the last function for how to get started.
15// 2. Print out a text based description of the tree.
16//
17
18//
19// Use this class to carry along data from node to node in
20// the traversal
21//
22class TOutputTraverser : public TIntermTraverser {
23public:
alokp@chromium.org7beea402010-09-15 21:18:34 +000024 TOutputTraverser(TInfoSinkBase& i) : sink(i) { }
25 TInfoSinkBase& sink;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000026
27protected:
alokp@chromium.org76b82082010-03-24 17:59:39 +000028 void visitSymbol(TIntermSymbol*);
29 void visitConstantUnion(TIntermConstantUnion*);
30 bool visitBinary(Visit visit, TIntermBinary*);
31 bool visitUnary(Visit visit, TIntermUnary*);
32 bool visitSelection(Visit visit, TIntermSelection*);
33 bool visitAggregate(Visit visit, TIntermAggregate*);
34 bool visitLoop(Visit visit, TIntermLoop*);
35 bool visitBranch(Visit visit, TIntermBranch*);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000036};
37
38TString TType::getCompleteString() const
39{
alokp@chromium.org4e4facd2010-06-02 15:21:22 +000040 TStringStream stream;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000041
alokp@chromium.org76b82082010-03-24 17:59:39 +000042 if (qualifier != EvqTemporary && qualifier != EvqGlobal)
alokp@chromium.org4e4facd2010-06-02 15:21:22 +000043 stream << getQualifierString() << " " << getPrecisionString() << " ";
alokp@chromium.org76b82082010-03-24 17:59:39 +000044 if (array)
alokp@chromium.org4e4facd2010-06-02 15:21:22 +000045 stream << "array of ";
alokp@chromium.org76b82082010-03-24 17:59:39 +000046 if (matrix)
alokp@chromium.org4e4facd2010-06-02 15:21:22 +000047 stream << size << "X" << size << " matrix of ";
alokp@chromium.org76b82082010-03-24 17:59:39 +000048 else if (size > 1)
alokp@chromium.org4e4facd2010-06-02 15:21:22 +000049 stream << size << "-component vector of ";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000050
alokp@chromium.org4e4facd2010-06-02 15:21:22 +000051 stream << getBasicString();
52 return stream.str();
53}
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000054
55//
56// Helper functions for printing, not part of traversing.
57//
58
alokp@chromium.org7beea402010-09-15 21:18:34 +000059void OutputTreeText(TInfoSinkBase& sink, TIntermNode* node, const int depth)
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000060{
alokp@chromium.org76b82082010-03-24 17:59:39 +000061 int i;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000062
alokp@chromium.org7beea402010-09-15 21:18:34 +000063 sink.location(node->getLine());
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000064
alokp@chromium.org76b82082010-03-24 17:59:39 +000065 for (i = 0; i < depth; ++i)
alokp@chromium.org7beea402010-09-15 21:18:34 +000066 sink << " ";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000067}
68
69//
70// The rest of the file are the traversal functions. The last one
71// is the one that starts the traversal.
72//
73// Return true from interior nodes to have the external traversal
74// continue on to children. If you process children yourself,
75// return false.
76//
77
78void TOutputTraverser::visitSymbol(TIntermSymbol* node)
79{
alokp@chromium.org7beea402010-09-15 21:18:34 +000080 OutputTreeText(sink, node, depth);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000081
alokp@chromium.org7beea402010-09-15 21:18:34 +000082 sink << "'" << node->getSymbol() << "' ";
83 sink << "(" << node->getCompleteString() << ")\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000084}
85
86bool TOutputTraverser::visitBinary(Visit visit, TIntermBinary* node)
87{
alokp@chromium.org7beea402010-09-15 21:18:34 +000088 TInfoSinkBase& out = sink;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000089
alokp@chromium.org76b82082010-03-24 17:59:39 +000090 OutputTreeText(out, node, depth);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +000091
alokp@chromium.org76b82082010-03-24 17:59:39 +000092 switch (node->getOp()) {
alokp@chromium.org7beea402010-09-15 21:18:34 +000093 case EOpAssign: out << "move second child to first child"; break;
94 case EOpInitialize: out << "initialize first child with second child"; break;
95 case EOpAddAssign: out << "add second child into first child"; break;
96 case EOpSubAssign: out << "subtract second child into first child"; break;
97 case EOpMulAssign: out << "multiply second child into first child"; break;
98 case EOpVectorTimesMatrixAssign: out << "matrix mult second child into first child"; break;
99 case EOpVectorTimesScalarAssign: out << "vector scale second child into first child"; break;
100 case EOpMatrixTimesScalarAssign: out << "matrix scale second child into first child"; break;
101 case EOpMatrixTimesMatrixAssign: out << "matrix mult second child into first child"; break;
102 case EOpDivAssign: out << "divide second child into first child"; break;
103 case EOpIndexDirect: out << "direct index"; break;
104 case EOpIndexIndirect: out << "indirect index"; break;
105 case EOpIndexDirectStruct: out << "direct index for structure"; break;
106 case EOpVectorSwizzle: out << "vector swizzle"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000107
alokp@chromium.org7beea402010-09-15 21:18:34 +0000108 case EOpAdd: out << "add"; break;
109 case EOpSub: out << "subtract"; break;
110 case EOpMul: out << "component-wise multiply"; break;
111 case EOpDiv: out << "divide"; break;
112 case EOpEqual: out << "Compare Equal"; break;
113 case EOpNotEqual: out << "Compare Not Equal"; break;
114 case EOpLessThan: out << "Compare Less Than"; break;
115 case EOpGreaterThan: out << "Compare Greater Than"; break;
116 case EOpLessThanEqual: out << "Compare Less Than or Equal"; break;
117 case EOpGreaterThanEqual: out << "Compare Greater Than or Equal"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000118
alokp@chromium.org7beea402010-09-15 21:18:34 +0000119 case EOpVectorTimesScalar: out << "vector-scale"; break;
120 case EOpVectorTimesMatrix: out << "vector-times-matrix"; break;
121 case EOpMatrixTimesVector: out << "matrix-times-vector"; break;
122 case EOpMatrixTimesScalar: out << "matrix-scale"; break;
123 case EOpMatrixTimesMatrix: out << "matrix-multiply"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000124
alokp@chromium.org7beea402010-09-15 21:18:34 +0000125 case EOpLogicalOr: out << "logical-or"; break;
126 case EOpLogicalXor: out << "logical-xor"; break;
127 case EOpLogicalAnd: out << "logical-and"; break;
128 default: out << "<unknown op>";
alokp@chromium.org76b82082010-03-24 17:59:39 +0000129 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000130
alokp@chromium.org7beea402010-09-15 21:18:34 +0000131 out << " (" << node->getCompleteString() << ")";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000132
alokp@chromium.org7beea402010-09-15 21:18:34 +0000133 out << "\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000134
alokp@chromium.org76b82082010-03-24 17:59:39 +0000135 return true;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000136}
137
138bool TOutputTraverser::visitUnary(Visit visit, TIntermUnary* node)
139{
alokp@chromium.org7beea402010-09-15 21:18:34 +0000140 TInfoSinkBase& out = sink;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000141
alokp@chromium.org76b82082010-03-24 17:59:39 +0000142 OutputTreeText(out, node, depth);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000143
alokp@chromium.org76b82082010-03-24 17:59:39 +0000144 switch (node->getOp()) {
alokp@chromium.org7beea402010-09-15 21:18:34 +0000145 case EOpNegative: out << "Negate value"; break;
alokp@chromium.org76b82082010-03-24 17:59:39 +0000146 case EOpVectorLogicalNot:
alokp@chromium.org7beea402010-09-15 21:18:34 +0000147 case EOpLogicalNot: out << "Negate conditional"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000148
alokp@chromium.org7beea402010-09-15 21:18:34 +0000149 case EOpPostIncrement: out << "Post-Increment"; break;
150 case EOpPostDecrement: out << "Post-Decrement"; break;
151 case EOpPreIncrement: out << "Pre-Increment"; break;
152 case EOpPreDecrement: out << "Pre-Decrement"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000153
alokp@chromium.org7beea402010-09-15 21:18:34 +0000154 case EOpConvIntToBool: out << "Convert int to bool"; break;
155 case EOpConvFloatToBool:out << "Convert float to bool";break;
156 case EOpConvBoolToFloat:out << "Convert bool to float";break;
157 case EOpConvIntToFloat: out << "Convert int to float"; break;
158 case EOpConvFloatToInt: out << "Convert float to int"; break;
159 case EOpConvBoolToInt: out << "Convert bool to int"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000160
alokp@chromium.org7beea402010-09-15 21:18:34 +0000161 case EOpRadians: out << "radians"; break;
162 case EOpDegrees: out << "degrees"; break;
163 case EOpSin: out << "sine"; break;
164 case EOpCos: out << "cosine"; break;
165 case EOpTan: out << "tangent"; break;
166 case EOpAsin: out << "arc sine"; break;
167 case EOpAcos: out << "arc cosine"; break;
168 case EOpAtan: out << "arc tangent"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000169
alokp@chromium.org7beea402010-09-15 21:18:34 +0000170 case EOpExp: out << "exp"; break;
171 case EOpLog: out << "log"; break;
172 case EOpExp2: out << "exp2"; break;
173 case EOpLog2: out << "log2"; break;
174 case EOpSqrt: out << "sqrt"; break;
175 case EOpInverseSqrt: out << "inverse sqrt"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000176
alokp@chromium.org7beea402010-09-15 21:18:34 +0000177 case EOpAbs: out << "Absolute value"; break;
178 case EOpSign: out << "Sign"; break;
179 case EOpFloor: out << "Floor"; break;
180 case EOpCeil: out << "Ceiling"; break;
181 case EOpFract: out << "Fraction"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000182
alokp@chromium.org7beea402010-09-15 21:18:34 +0000183 case EOpLength: out << "length"; break;
184 case EOpNormalize: out << "normalize"; break;
185 // case EOpDPdx: out << "dPdx"; break;
186 // case EOpDPdy: out << "dPdy"; break;
187 // case EOpFwidth: out << "fwidth"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000188
alokp@chromium.org7beea402010-09-15 21:18:34 +0000189 case EOpAny: out << "any"; break;
190 case EOpAll: out << "all"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000191
alokp@chromium.org7beea402010-09-15 21:18:34 +0000192 default: out.message(EPrefixError, "Bad unary op");
alokp@chromium.org76b82082010-03-24 17:59:39 +0000193 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000194
alokp@chromium.org7beea402010-09-15 21:18:34 +0000195 out << " (" << node->getCompleteString() << ")";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000196
alokp@chromium.org7beea402010-09-15 21:18:34 +0000197 out << "\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000198
alokp@chromium.org76b82082010-03-24 17:59:39 +0000199 return true;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000200}
201
202bool TOutputTraverser::visitAggregate(Visit visit, TIntermAggregate* node)
203{
alokp@chromium.org7beea402010-09-15 21:18:34 +0000204 TInfoSinkBase& out = sink;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000205
alokp@chromium.org76b82082010-03-24 17:59:39 +0000206 if (node->getOp() == EOpNull) {
alokp@chromium.org7beea402010-09-15 21:18:34 +0000207 out.message(EPrefixError, "node is still EOpNull!");
alokp@chromium.org76b82082010-03-24 17:59:39 +0000208 return true;
209 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000210
alokp@chromium.org76b82082010-03-24 17:59:39 +0000211 OutputTreeText(out, node, depth);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000212
alokp@chromium.org76b82082010-03-24 17:59:39 +0000213 switch (node->getOp()) {
alokp@chromium.org7beea402010-09-15 21:18:34 +0000214 case EOpSequence: out << "Sequence\n"; return true;
215 case EOpComma: out << "Comma\n"; return true;
216 case EOpFunction: out << "Function Definition: " << node->getName(); break;
217 case EOpFunctionCall: out << "Function Call: " << node->getName(); break;
218 case EOpParameters: out << "Function Parameters: "; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000219
alokp@chromium.org7beea402010-09-15 21:18:34 +0000220 case EOpConstructFloat: out << "Construct float"; break;
221 case EOpConstructVec2: out << "Construct vec2"; break;
222 case EOpConstructVec3: out << "Construct vec3"; break;
223 case EOpConstructVec4: out << "Construct vec4"; break;
224 case EOpConstructBool: out << "Construct bool"; break;
225 case EOpConstructBVec2: out << "Construct bvec2"; break;
226 case EOpConstructBVec3: out << "Construct bvec3"; break;
227 case EOpConstructBVec4: out << "Construct bvec4"; break;
228 case EOpConstructInt: out << "Construct int"; break;
229 case EOpConstructIVec2: out << "Construct ivec2"; break;
230 case EOpConstructIVec3: out << "Construct ivec3"; break;
231 case EOpConstructIVec4: out << "Construct ivec4"; break;
232 case EOpConstructMat2: out << "Construct mat2"; break;
233 case EOpConstructMat3: out << "Construct mat3"; break;
234 case EOpConstructMat4: out << "Construct mat4"; break;
235 case EOpConstructStruct: out << "Construct structure"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000236
alokp@chromium.org7beea402010-09-15 21:18:34 +0000237 case EOpLessThan: out << "Compare Less Than"; break;
238 case EOpGreaterThan: out << "Compare Greater Than"; break;
239 case EOpLessThanEqual: out << "Compare Less Than or Equal"; break;
240 case EOpGreaterThanEqual: out << "Compare Greater Than or Equal"; break;
241 case EOpVectorEqual: out << "Equal"; break;
242 case EOpVectorNotEqual: out << "NotEqual"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000243
alokp@chromium.org7beea402010-09-15 21:18:34 +0000244 case EOpMod: out << "mod"; break;
245 case EOpPow: out << "pow"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000246
alokp@chromium.org7beea402010-09-15 21:18:34 +0000247 case EOpAtan: out << "arc tangent"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000248
alokp@chromium.org7beea402010-09-15 21:18:34 +0000249 case EOpMin: out << "min"; break;
250 case EOpMax: out << "max"; break;
251 case EOpClamp: out << "clamp"; break;
252 case EOpMix: out << "mix"; break;
253 case EOpStep: out << "step"; break;
254 case EOpSmoothStep: out << "smoothstep"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000255
alokp@chromium.org7beea402010-09-15 21:18:34 +0000256 case EOpDistance: out << "distance"; break;
257 case EOpDot: out << "dot-product"; break;
258 case EOpCross: out << "cross-product"; break;
259 case EOpFaceForward: out << "face-forward"; break;
260 case EOpReflect: out << "reflect"; break;
261 case EOpRefract: out << "refract"; break;
262 case EOpMul: out << "component-wise multiply"; break;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000263
alokp@chromium.org7beea402010-09-15 21:18:34 +0000264 default: out.message(EPrefixError, "Bad aggregation op");
alokp@chromium.org76b82082010-03-24 17:59:39 +0000265 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000266
alokp@chromium.org76b82082010-03-24 17:59:39 +0000267 if (node->getOp() != EOpSequence && node->getOp() != EOpParameters)
alokp@chromium.org7beea402010-09-15 21:18:34 +0000268 out << " (" << node->getCompleteString() << ")";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000269
alokp@chromium.org7beea402010-09-15 21:18:34 +0000270 out << "\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000271
alokp@chromium.org76b82082010-03-24 17:59:39 +0000272 return true;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000273}
274
275bool TOutputTraverser::visitSelection(Visit visit, TIntermSelection* node)
276{
alokp@chromium.org7beea402010-09-15 21:18:34 +0000277 TInfoSinkBase& out = sink;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000278
alokp@chromium.org76b82082010-03-24 17:59:39 +0000279 OutputTreeText(out, node, depth);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000280
alokp@chromium.org7beea402010-09-15 21:18:34 +0000281 out << "Test condition and select";
282 out << " (" << node->getCompleteString() << ")\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000283
alokp@chromium.org76b82082010-03-24 17:59:39 +0000284 ++depth;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000285
alokp@chromium.org7beea402010-09-15 21:18:34 +0000286 OutputTreeText(sink, node, depth);
287 out << "Condition\n";
alokp@chromium.org76b82082010-03-24 17:59:39 +0000288 node->getCondition()->traverse(this);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000289
alokp@chromium.org7beea402010-09-15 21:18:34 +0000290 OutputTreeText(sink, node, depth);
alokp@chromium.org76b82082010-03-24 17:59:39 +0000291 if (node->getTrueBlock()) {
alokp@chromium.org7beea402010-09-15 21:18:34 +0000292 out << "true case\n";
alokp@chromium.org76b82082010-03-24 17:59:39 +0000293 node->getTrueBlock()->traverse(this);
294 } else
alokp@chromium.org7beea402010-09-15 21:18:34 +0000295 out << "true case is null\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000296
alokp@chromium.org76b82082010-03-24 17:59:39 +0000297 if (node->getFalseBlock()) {
alokp@chromium.org7beea402010-09-15 21:18:34 +0000298 OutputTreeText(sink, node, depth);
299 out << "false case\n";
alokp@chromium.org76b82082010-03-24 17:59:39 +0000300 node->getFalseBlock()->traverse(this);
301 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000302
alokp@chromium.org76b82082010-03-24 17:59:39 +0000303 --depth;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000304
alokp@chromium.org76b82082010-03-24 17:59:39 +0000305 return false;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000306}
307
308void TOutputTraverser::visitConstantUnion(TIntermConstantUnion* node)
309{
alokp@chromium.org7beea402010-09-15 21:18:34 +0000310 TInfoSinkBase& out = sink;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000311
alokp@chromium.org76b82082010-03-24 17:59:39 +0000312 int size = node->getType().getObjectSize();
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000313
alokp@chromium.org76b82082010-03-24 17:59:39 +0000314 for (int i = 0; i < size; i++) {
315 OutputTreeText(out, node, depth);
316 switch (node->getUnionArrayPointer()[i].getType()) {
317 case EbtBool:
318 if (node->getUnionArrayPointer()[i].getBConst())
alokp@chromium.org7beea402010-09-15 21:18:34 +0000319 out << "true";
alokp@chromium.org76b82082010-03-24 17:59:39 +0000320 else
alokp@chromium.org7beea402010-09-15 21:18:34 +0000321 out << "false";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000322
alokp@chromium.org7beea402010-09-15 21:18:34 +0000323 out << " (" << "const bool" << ")";
324 out << "\n";
alokp@chromium.org76b82082010-03-24 17:59:39 +0000325 break;
326 case EbtFloat:
alokp@chromium.org7beea402010-09-15 21:18:34 +0000327 out << node->getUnionArrayPointer()[i].getFConst();
328 out << " (const float)\n";
alokp@chromium.org76b82082010-03-24 17:59:39 +0000329 break;
330 case EbtInt:
alokp@chromium.org7beea402010-09-15 21:18:34 +0000331 out << node->getUnionArrayPointer()[i].getIConst();
332 out << " (const int)\n";
alokp@chromium.org76b82082010-03-24 17:59:39 +0000333 break;
334 default:
alokp@chromium.org7beea402010-09-15 21:18:34 +0000335 out.message(EPrefixInternalError, "Unknown constant", node->getLine());
alokp@chromium.org76b82082010-03-24 17:59:39 +0000336 break;
337 }
338 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000339}
340
341bool TOutputTraverser::visitLoop(Visit visit, TIntermLoop* node)
342{
alokp@chromium.org7beea402010-09-15 21:18:34 +0000343 TInfoSinkBase& out = sink;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000344
alokp@chromium.org76b82082010-03-24 17:59:39 +0000345 OutputTreeText(out, node, depth);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000346
alokp@chromium.org7beea402010-09-15 21:18:34 +0000347 out << "Loop with condition ";
alokp@chromium.org52813552010-11-16 18:36:09 +0000348 if (node->getType() == ELoopDoWhile)
alokp@chromium.org7beea402010-09-15 21:18:34 +0000349 out << "not ";
350 out << "tested first\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000351
alokp@chromium.org76b82082010-03-24 17:59:39 +0000352 ++depth;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000353
alokp@chromium.org7beea402010-09-15 21:18:34 +0000354 OutputTreeText(sink, node, depth);
alokp@chromium.org52813552010-11-16 18:36:09 +0000355 if (node->getCondition()) {
alokp@chromium.org7beea402010-09-15 21:18:34 +0000356 out << "Loop Condition\n";
alokp@chromium.org52813552010-11-16 18:36:09 +0000357 node->getCondition()->traverse(this);
alokp@chromium.org76b82082010-03-24 17:59:39 +0000358 } else
alokp@chromium.org7beea402010-09-15 21:18:34 +0000359 out << "No loop condition\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000360
alokp@chromium.org7beea402010-09-15 21:18:34 +0000361 OutputTreeText(sink, node, depth);
alokp@chromium.org76b82082010-03-24 17:59:39 +0000362 if (node->getBody()) {
alokp@chromium.org7beea402010-09-15 21:18:34 +0000363 out << "Loop Body\n";
alokp@chromium.org76b82082010-03-24 17:59:39 +0000364 node->getBody()->traverse(this);
365 } else
alokp@chromium.org7beea402010-09-15 21:18:34 +0000366 out << "No loop body\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000367
alokp@chromium.org52813552010-11-16 18:36:09 +0000368 if (node->getExpression()) {
alokp@chromium.org7beea402010-09-15 21:18:34 +0000369 OutputTreeText(sink, node, depth);
370 out << "Loop Terminal Expression\n";
alokp@chromium.org52813552010-11-16 18:36:09 +0000371 node->getExpression()->traverse(this);
alokp@chromium.org76b82082010-03-24 17:59:39 +0000372 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000373
alokp@chromium.org76b82082010-03-24 17:59:39 +0000374 --depth;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000375
alokp@chromium.org76b82082010-03-24 17:59:39 +0000376 return false;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000377}
378
379bool TOutputTraverser::visitBranch(Visit visit, TIntermBranch* node)
380{
alokp@chromium.org7beea402010-09-15 21:18:34 +0000381 TInfoSinkBase& out = sink;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000382
alokp@chromium.org76b82082010-03-24 17:59:39 +0000383 OutputTreeText(out, node, depth);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000384
alokp@chromium.org76b82082010-03-24 17:59:39 +0000385 switch (node->getFlowOp()) {
alokp@chromium.org7beea402010-09-15 21:18:34 +0000386 case EOpKill: out << "Branch: Kill"; break;
387 case EOpBreak: out << "Branch: Break"; break;
388 case EOpContinue: out << "Branch: Continue"; break;
389 case EOpReturn: out << "Branch: Return"; break;
390 default: out << "Branch: Unknown Branch"; break;
alokp@chromium.org76b82082010-03-24 17:59:39 +0000391 }
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000392
alokp@chromium.org76b82082010-03-24 17:59:39 +0000393 if (node->getExpression()) {
alokp@chromium.org7beea402010-09-15 21:18:34 +0000394 out << " with expression\n";
alokp@chromium.org76b82082010-03-24 17:59:39 +0000395 ++depth;
396 node->getExpression()->traverse(this);
397 --depth;
398 } else
alokp@chromium.org7beea402010-09-15 21:18:34 +0000399 out << "\n";
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000400
alokp@chromium.org76b82082010-03-24 17:59:39 +0000401 return false;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000402}
403
404//
405// This function is the one to call externally to start the traversal.
406// Individual functions can be initialized to 0 to skip processing of that
407// type of node. It's children will still be processed.
408//
409void TIntermediate::outputTree(TIntermNode* root)
410{
alokp@chromium.org76b82082010-03-24 17:59:39 +0000411 if (root == 0)
412 return;
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000413
alokp@chromium.org7beea402010-09-15 21:18:34 +0000414 TOutputTraverser it(infoSink.info);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000415
alokp@chromium.org76b82082010-03-24 17:59:39 +0000416 root->traverse(&it);
daniel@transgaming.com4f39fd92010-03-08 20:26:45 +0000417}