- std::map<std::string, std::vector<FunctionDeclaration *> > functions;
- std::map<std::string, VariableDeclaration *> variables;
- std::set<Node *> nodes_to_remove;
-
-public:
- void apply(Stage &);
-
-private:
- virtual void visit(Block &);
- 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 &);
-};
-
-/** Forms links between nested blocks in the syntax tree. */
-class BlockHierarchyResolver: private TraversingVisitor
-{
-public:
- void apply(Stage &s) { s.content.visit(*this); }
-
-private:
- virtual void enter(Block &);
-};
-
-/** Resolves types of variables and base types of other types. */
-class TypeResolver: private TraversingVisitor
-{
-private:
- Stage *stage;
- std::map<TypeDeclaration *, TypeDeclaration *> alias_map;
- std::map<TypeDeclaration *, TypeDeclaration *> array_types;
- NodeList<Statement>::iterator type_insert_point;
-
-public:
- TypeResolver();
-
- void apply(Stage &);
-
-private:
- TypeDeclaration *resolve_type(const std::string &);
- virtual void visit(Block &);
- virtual void visit(BasicTypeDeclaration &);
- virtual void visit(ImageTypeDeclaration &);
- virtual void visit(StructDeclaration &);
- virtual void visit(VariableDeclaration &);
- virtual void visit(FunctionDeclaration &);
-};
-
-/** 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;
- RefPtr<Expression> r_replacement_expr;
- std::string block_interface;
- bool r_any_resolved;
- bool record_target;
- bool r_self_referencing;
- VariableDeclaration *r_assignment_target;
-
-public:
- VariableResolver();
-
- bool apply(Stage &);
-
-private:
- virtual void enter(Block &);
- void visit_and_replace(RefPtr<Expression> &);
- virtual void visit(VariableReference &);
- virtual void visit(InterfaceBlockReference &);
- virtual void visit(MemberAccess &);
- virtual void visit(UnaryExpression &);
- virtual void visit(BinaryExpression &);
- virtual void visit(Assignment &);
- virtual void visit(FunctionCall &);
- virtual void visit(VariableDeclaration &);
- virtual void visit(InterfaceBlock &);
-};
-
-/** Resolves types and lvalueness of expressions. */
-class ExpressionResolver: private TraversingVisitor
-{
-private:
- enum Compatibility
- {
- NOT_COMPATIBLE,
- LEFT_CONVERTIBLE,
- RIGHT_CONVERTIBLE,
- SAME_TYPE
- };
-
- Stage *stage;
- std::vector<BasicTypeDeclaration *> basic_types;
- bool r_any_resolved;