1 #include <msp/stringcodec/utf8.h>
2 #include <msp/strings/format.h>
11 const std::string &DumpTree::apply(Stage &stage)
13 formatted = format("Stage: %s\n", Stage::get_stage_name(stage.type));
14 tree.push_back(BRANCH);
15 append(format("Version: %d.%02d", stage.required_features.glsl_version.major, stage.required_features.glsl_version.minor));
17 for(std::map<string, TypeDeclaration *>::const_iterator i=stage.types.begin(); i!=stage.types.end(); ++i)
18 append(format("Type: %%%d %s", get_label(*i->second), i->first));
20 set<InterfaceBlock *> seen_interfaces;
21 for(std::map<string, InterfaceBlock *>::const_iterator i=stage.interface_blocks.begin(); i!=stage.interface_blocks.end(); ++i)
22 if(seen_interfaces.insert(i->second).second)
24 string text = format("Interface block: %%%d %s %s", get_label(*i->second), i->second->interface, i->second->name);
25 if(!i->second->instance_name.empty())
26 text += format(" %s", i->second->instance_name);
30 for(std::map<string, FunctionDeclaration *>::const_iterator i=stage.functions.begin(); i!=stage.functions.end(); ++i)
31 append(format("Function: %%%d %s", get_label(*i->second), i->first));
34 stage.content.visit(*this);
38 void DumpTree::append(const string &line)
40 StringCodec::Utf8::Encoder enc;
41 for(vector<TreeChars>::const_iterator i=tree.begin(); i!=tree.end(); )
43 enc.encode_char(*i++, formatted);
44 enc.encode_char((i==tree.end() ? REACH : EMPTY), formatted);
50 void DumpTree::append_subtree(const vector<Branch> &branches)
53 for(vector<Branch>::const_iterator i=branches.begin(); i!=branches.end(); )
55 vector<Branch>::const_iterator j = increment(i, branches);
59 annotated_branch(j->text, *j->node);
64 j->node->visit(*this);
69 void DumpTree::begin_sub()
71 tree.back() = (tree.back()==BRANCH_LAST ? EMPTY : STRAIGHT);
72 tree.push_back(BRANCH);
75 void DumpTree::last_branch()
77 tree.back() = BRANCH_LAST;
80 void DumpTree::end_sub()
83 if(tree.back()==STRAIGHT)
87 void DumpTree::annotated_branch(const string &annotation, Node &node)
96 unsigned DumpTree::get_label(const Node &node)
98 unsigned &label = node_labels[&node];
100 label = node_labels.size();
104 string DumpTree::format_type(TypeDeclaration *type)
106 return (type ? type->name : "?");
110 typename T::const_iterator DumpTree::increment(typename T::const_iterator &iter, const T &container)
112 typename T::const_iterator ret = iter++;
113 if(iter==container.end())
118 void DumpTree::visit(Block &block)
120 append(format("Block %s", (block.use_braces ? "{}" : "(inline)")));
123 for(std::map<string, VariableDeclaration *>::const_iterator i=block.variables.begin(); i!=block.variables.end(); ++i)
124 append(format("Variable: %%%d %s %s", get_label(*i->second), i->second->type, i->first));
126 bool labeled_body = !block.variables.empty();
133 for(NodeList<Statement>::const_iterator i=block.body.begin(); i!=block.body.end(); )
135 NodeList<Statement>::const_iterator j = increment(i, block.body);
144 void DumpTree::visit(Literal &literal)
146 append(format("Literal: %s -> %s", literal.token, format_type(literal.type)));
149 void DumpTree::visit(ParenthesizedExpression &parexpr)
151 annotated_branch(format("(expr) -> %s", format_type(parexpr.type)), *parexpr.expression);
154 void DumpTree::visit(VariableReference &var)
158 text += format("%%%d ", get_label(*var.declaration));
159 text += format("%s (var) -> %s", var.name, format_type(var.type));
163 void DumpTree::visit(InterfaceBlockReference &iface)
166 if(iface.declaration)
167 text += format("%%%d ", get_label(*iface.declaration));
168 text += format("%s (iface)", iface.name);
172 void DumpTree::visit(MemberAccess &memacc)
174 string text = "Member access:";
175 if(memacc.declaration)
176 text += format(" %%%d", get_label(*memacc.declaration));
177 text += format(" .%s -> %s", memacc.member, format_type(memacc.type));
178 annotated_branch(text, *memacc.left);
181 void DumpTree::visit(UnaryExpression &unary)
183 string text = format("Unary: %s, %sfix -> %s", unary.oper->token, (unary.oper->type==Operator::PREFIX ? "pre" : "post"), format_type(unary.type));
184 annotated_branch(text, *unary.expression);
187 void DumpTree::visit(BinaryExpression &binary)
189 append(format("Binary: %s -> %s", (binary.oper->token[0]=='[' ? "[]" : binary.oper->token), format_type(binary.type)));
191 binary.left->visit(*this);
193 binary.right->visit(*this);
197 void DumpTree::visit(Assignment &assign)
199 append(format("Assignment: %s%s -> %s", assign.oper->token, (assign.self_referencing ? " (self-referencing)" : ""), format_type(assign.type)));
201 if(assign.target_declaration)
202 append(format("Target: %%%d %s %s", get_label(*assign.target_declaration), assign.target_declaration->type, assign.target_declaration->name));
203 assign.left->visit(*this);
205 assign.right->visit(*this);
209 void DumpTree::visit(FunctionCall &call)
211 string head = "Function call: ";
213 head += format("%%%d ", get_label(*call.declaration));
216 head += " (constructor)";
217 head += format(" -> %s", format_type(call.type));
221 for(NodeArray<Expression>::const_iterator i=call.arguments.begin(); i!=call.arguments.end(); )
223 NodeArray<Expression>::const_iterator j = increment(i, call.arguments);
229 void DumpTree::visit(ExpressionStatement &expr)
231 annotated_branch("expr;", *expr.expression);
234 void DumpTree::visit(Import &import)
236 append(format("import %s", import.module));
239 void DumpTree::visit(Precision &prec)
241 append(format("precision %s %s", prec.precision, prec.type));
244 void DumpTree::visit(Layout &layout)
248 for(vector<Layout::Qualifier>::const_iterator i=layout.qualifiers.begin(); i!=layout.qualifiers.end(); )
250 vector<Layout::Qualifier>::const_iterator j = increment(i, layout.qualifiers);
251 string qualifier = j->name;
253 qualifier += format("=%d", j->value);
259 void DumpTree::visit(InterfaceLayout &layout)
261 annotated_branch(format("Layout: %s", layout.interface), layout.layout);
264 void DumpTree::visit(BasicTypeDeclaration &type)
266 append(format("%%%d typedef %s", get_label(type), type.name));
268 vector<Branch> branches;
270 branches.push_back(format("%s: %%%d %s", (type.kind==BasicTypeDeclaration::ALIAS ? "Alias of" : "Base"), get_label(*type.base_type), type.base_type->name));
271 if(type.kind==BasicTypeDeclaration::VECTOR)
272 branches.push_back(format("Vector: %d", type.size));
273 else if(type.kind==BasicTypeDeclaration::MATRIX)
274 branches.push_back(format("Matrix: %dx%d", type.size&0xFFFF, type.size>>16));
275 append_subtree(branches);
278 void DumpTree::visit(ImageTypeDeclaration &type)
280 static const char *dims[] = { "1D", "2D", "3D", "Cube" };
282 append(format("%%%d typedef %s", get_label(type), type.name));
284 vector<Branch> branches;
285 branches.push_back(format("Dimensions: %s%s", dims[type.dimensions-1], (type.array ? " array" : "")));
287 branches.push_back(format("Element type: %%%d %s", get_label(*type.base_type), type.base_type->name));
289 branches.push_back("Shadow");
290 append_subtree(branches);
293 void DumpTree::visit(StructDeclaration &strct)
295 annotated_branch(format("%%%d struct %s", get_label(strct), strct.name), strct.members);
298 void DumpTree::visit(VariableDeclaration &var)
300 string decl = format("%%%d ", get_label(var));
303 if(!var.interpolation.empty())
304 decl += format("%s ", var.interpolation);
305 if(!var.sampling.empty())
306 decl += format("%s ", var.sampling);
307 if(!var.interface.empty())
308 decl += format("%s ", var.interface);
309 if(!var.precision.empty())
310 decl += format("%s ", var.precision);
311 decl += format("%s %s", var.type, var.name);
312 if(var.source==BUILTIN_SOURCE)
313 decl += " (builtin)";
314 else if(var.linked_declaration)
318 vector<Branch> branches;
319 if(var.type_declaration)
320 branches.push_back(format("Type: %%%d %s", get_label(*var.type_declaration), var.type_declaration->name));
322 branches.push_back(var.layout.get());
326 branches.push_back(Branch("Array []", var.array_size.get()));
328 branches.push_back("Array []");
330 if(var.init_expression)
331 branches.push_back(var.init_expression.get());
332 append_subtree(branches);
335 void DumpTree::visit(InterfaceBlock &block)
338 if(!block.instance_name.empty())
339 head += format("%%%d ", get_label(block));
340 head += format("%s %s", block.interface, block.name);
341 if(!block.instance_name.empty())
342 head += format(" %s", block.instance_name);
345 if(block.source==BUILTIN_SOURCE)
346 head += " (builtin)";
347 else if(block.linked_block)
349 annotated_branch(head, block.members);
352 void DumpTree::visit(FunctionDeclaration &func)
354 string text = format("%%%d %s %s", get_label(func), func.return_type, func.name);
355 if(func.source==BUILTIN_SOURCE)
356 text += " (builtin)";
357 else if(!func.definition)
358 text += " (undefined)";
362 for(NodeArray<VariableDeclaration>::const_iterator i=func.parameters.begin(); i!=func.parameters.end(); ++i)
365 if(func.definition==&func)
366 func.body.visit(*this);
367 else if(func.definition)
368 append(format("Definition: %%%d", get_label(*func.definition)));
372 void DumpTree::visit(Conditional &cond)
376 vector<Branch> branches;
377 branches.push_back(cond.condition.get());
378 branches.push_back(&cond.body);
379 if(!cond.else_body.body.empty())
380 branches.push_back(&cond.else_body);
381 append_subtree(branches);
384 void DumpTree::visit(Iteration &iter)
389 if(iter.init_statement)
390 annotated_branch("Initialization", *iter.init_statement);
392 annotated_branch("Condition", *iter.condition);
393 if(iter.loop_expression)
394 annotated_branch("Loop", *iter.loop_expression);
396 annotated_branch("Body", iter.body);
400 void DumpTree::visit(Passthrough &pass)
403 annotated_branch("passthrough[]", *pass.subscript);
405 append("passthrough;");
408 void DumpTree::visit(Return &ret)
411 annotated_branch("return", *ret.expression);
416 void DumpTree::visit(Jump &jump)
418 append(format("%s;", jump.keyword));