namespace GL {
namespace SL {
-class DeclarationCombiner: public BlockModifier
+/** Combines multiple declarations of the same identifier into one. */
+class DeclarationCombiner: private TraversingVisitor
{
private:
- bool toplevel;
std::map<std::string, std::vector<FunctionDeclaration *> > functions;
std::map<std::string, VariableDeclaration *> variables;
+ std::set<Node *> nodes_to_remove;
public:
- DeclarationCombiner();
+ void apply(Stage &);
- using StageVisitor::visit;
+private:
virtual void visit(Block &);
- virtual void visit(FunctionDeclaration &);
+ virtual void visit(VariableDeclaration &);
+ virtual void visit(FunctionDeclaration &) { }
+};
+
+/** Manipulates specialization constants. If values are specified, turns
+specialization constants into normal constants. Without values assigns
+automatic constant_ids to specialization constants. */
+class ConstantSpecializer: private TraversingVisitor
+{
+private:
+ const std::map<std::string, int> *values;
+
+public:
+ ConstantSpecializer();
+
+ void apply(Stage &, const std::map<std::string, int> *);
+
+private:
virtual void visit(VariableDeclaration &);
};
-class VariableResolver: public StageVisitor
+/** Forms links between nested blocks in the syntax tree. */
+class BlockHierarchyResolver: private TraversingVisitor
{
+public:
+ void apply(Stage &s) { s.content.visit(*this); }
+
private:
- std::vector<Block *> blocks;
- StructDeclaration *type;
- bool anonymous;
+ virtual void enter(Block &);
+};
+
+/** Resolves variable references. Variable references which match the name
+of an interface block are turned into interface block references. */
+class VariableResolver: private TraversingVisitor
+{
+private:
+ Stage *stage;
+ std::map<std::string, VariableDeclaration *> *r_members;
+ RefPtr<InterfaceBlockReference> r_iface_ref;
std::string block_interface;
bool record_target;
- VariableDeclaration *assignment_target;
- bool self_referencing;
+ VariableDeclaration *r_assignment_target;
+ bool r_self_referencing;
public:
VariableResolver();
- virtual void apply(Stage &);
- using StageVisitor::visit;
- virtual void visit(Block &);
+ void apply(Stage &);
+
+private:
+ virtual void enter(Block &);
virtual void visit(VariableReference &);
+ virtual void visit(InterfaceBlockReference &);
virtual void visit(MemberAccess &);
virtual void visit(BinaryExpression &);
virtual void visit(Assignment &);
virtual void visit(InterfaceBlock &);
};
-class FunctionResolver: public StageVisitor
+/** Resolves function declarations and calls. */
+class FunctionResolver: private TraversingVisitor
{
private:
- std::map<std::string, std::vector<FunctionDeclaration *> > functions;
+ Stage *stage;
+ std::map<std::string, std::vector<FunctionDeclaration *> > declarations;
public:
- using StageVisitor::visit;
+ void apply(Stage &);
+
+private:
virtual void visit(FunctionCall &);
virtual void visit(FunctionDeclaration &);
};
-class InterfaceGenerator: public BlockModifier
+/** Materializes implicitly declared interfaces.
+
+Out variable declarations inside functions are moved to the global scope.
+
+Passthrough statements are processed, generating out variables to match in
+variables and copying values.
+
+Unresolved variables are looked up in the previous stage's out variables. */
+class InterfaceGenerator: private TraversingVisitor
{
private:
+ Stage *stage;
std::string in_prefix;
std::string out_prefix;
- unsigned scope_level;
- std::map<std::string, RefPtr<VariableDeclaration> > iface_declarations;
+ bool function_scope;
+ InterfaceBlock *iface_block;
+ bool copy_block;
+ Block *iface_target_block;
+ NodeList<Statement>::iterator iface_insert_point;
+ NodeList<Statement>::iterator assignment_insert_point;
+ std::set<Node *> nodes_to_remove;
public:
InterfaceGenerator();
- static std::string get_out_prefix(StageType);
- virtual void apply(Stage &);
- using StageVisitor::visit;
- virtual void visit(Block &);
+ void apply(Stage &);
+
+private:
+ static std::string get_out_prefix(Stage::Type);
std::string change_prefix(const std::string &, const std::string &) const;
+ virtual void visit(Block &);
bool generate_interface(VariableDeclaration &, const std::string &, const std::string &);
+ bool generate_interface(InterfaceBlock &);
ExpressionStatement &insert_assignment(const std::string &, Expression *);
virtual void visit(VariableReference &);
virtual void visit(VariableDeclaration &);
- virtual void visit(Passthrough &);
-};
-
-class DeclarationReorderer: public StageVisitor
-{
-private:
- enum DeclarationKind
- {
- NO_DECLARATION,
- LAYOUT,
- STRUCT,
- VARIABLE,
- FUNCTION
- };
-
- unsigned scope_level;
- DeclarationKind kind;
- std::set<Node *> ordered_funcs;
- std::set<Node *> needed_funcs;
-
-public:
- DeclarationReorderer();
-
- using StageVisitor::visit;
- virtual void visit(Block &);
- virtual void visit(FunctionCall &);
- virtual void visit(InterfaceLayout &) { kind = LAYOUT; }
- virtual void visit(StructDeclaration &) { kind = STRUCT; }
- virtual void visit(VariableDeclaration &);
- virtual void visit(InterfaceBlock &) { kind = VARIABLE; }
+ virtual void visit(InterfaceBlock &);
virtual void visit(FunctionDeclaration &);
+ virtual void visit(Passthrough &);
};
} // namespace SL