namespace GL {
namespace SL {
-class DeclarationCombiner: public BlockModifier
+/** Assigns IDs to specialization constants with an automatic ID. */
+class ConstantIdAssigner: private TraversingVisitor
{
private:
- bool toplevel;
- std::map<std::string, std::vector<FunctionDeclaration *> > functions;
- std::map<std::string, VariableDeclaration *> variables;
+ std::set<unsigned> used_ids;
+ std::map<std::string, unsigned> existing_constants;
+ std::vector<VariableDeclaration *> auto_constants;
public:
- DeclarationCombiner();
+ void apply(Module &, const Features &);
- using StageVisitor::visit;
- virtual void visit(Block &);
- virtual void visit(FunctionDeclaration &);
+private:
virtual void visit(VariableDeclaration &);
};
-class VariableResolver: public StageVisitor
+/** 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:
- std::vector<Block *> blocks;
- StructDeclaration *type;
- bool anonymous;
- std::string block_interface;
- bool record_target;
- VariableDeclaration *assignment_target;
- bool self_referencing;
+ Stage *stage = 0;
+ std::string in_prefix;
+ std::string out_prefix;
+ bool function_scope = false;
+ bool copy_block = false;
+ std::vector<VariableDeclaration *> declared_inputs;
+ Block *iface_target_block = 0;
+ NodeList<Statement>::iterator iface_insert_point;
+ NodeList<Statement>::iterator assignment_insert_point;
+ std::set<Node *> nodes_to_remove;
public:
- VariableResolver();
+ void apply(Stage &);
- virtual void apply(Stage &);
- using StageVisitor::visit;
+private:
+ static std::string get_out_prefix(Stage::Type);
+ std::string change_prefix(const std::string &, const std::string &) const;
virtual void visit(Block &);
+ VariableDeclaration *generate_interface(VariableDeclaration &, const std::string &, const std::string &);
+ ExpressionStatement &insert_assignment(const std::string &, Expression *);
virtual void visit(VariableReference &);
- virtual void visit(MemberAccess &);
- virtual void visit(BinaryExpression &);
- virtual void visit(Assignment &);
- virtual void visit(StructDeclaration &);
virtual void visit(VariableDeclaration &);
- virtual void visit(InterfaceBlock &);
-};
-
-class FunctionResolver: public StageVisitor
-{
-private:
- std::map<std::string, std::vector<FunctionDeclaration *> > functions;
-
-public:
- using StageVisitor::visit;
- virtual void visit(FunctionCall &);
virtual void visit(FunctionDeclaration &);
+ virtual void visit(Passthrough &);
};
-class InterfaceGenerator: public BlockModifier
+class LayoutDefaulter: private TraversingVisitor
{
private:
- std::string in_prefix;
- std::string out_prefix;
- unsigned scope_level;
- std::map<std::string, RefPtr<VariableDeclaration> > iface_declarations;
+ InterfaceLayout *in_iface = 0;
+ bool need_winding = true;
+ bool need_spacing = true;
public:
- InterfaceGenerator();
+ void apply(Stage &);
- static std::string get_out_prefix(StageType);
- virtual void apply(Stage &);
- using StageVisitor::visit;
- virtual void visit(Block &);
- std::string change_prefix(const std::string &, const std::string &) const;
- bool generate_interface(VariableDeclaration &, const std::string &, const std::string &);
- ExpressionStatement &insert_assignment(const std::string &, Expression *);
- virtual void visit(VariableReference &);
- virtual void visit(VariableDeclaration &);
- virtual void visit(Passthrough &);
+private:
+ virtual void visit(InterfaceLayout &);
};
-class DeclarationReorderer: public StageVisitor
+/**
+Assigns sizes to arrays which don't have a size. Geometry shader inputs are
+sized by topology. Other arrays are sized by their use with literal indices.
+*/
+class ArraySizer: private TraversingVisitor
{
private:
- enum DeclarationKind
- {
- NO_DECLARATION,
- LAYOUT,
- STRUCT,
- VARIABLE,
- FUNCTION
- };
-
- unsigned scope_level;
- DeclarationKind kind;
- std::set<Node *> ordered_funcs;
- std::set<Node *> needed_funcs;
+ std::map<VariableDeclaration *, int> max_indices;
+ unsigned input_size = 0;
+ VariableDeclaration *r_declaration;
public:
- DeclarationReorderer();
+ void apply(Stage &);
- using StageVisitor::visit;
- virtual void visit(Block &);
+private:
+ virtual void visit(VariableReference &);
+ virtual void visit(MemberAccess &);
+ virtual void visit(Swizzle &);
+ virtual void visit(UnaryExpression&);
+ virtual void visit(BinaryExpression &);
+ virtual void visit(TernaryExpression &);
virtual void visit(FunctionCall &);
- virtual void visit(InterfaceLayout &) { kind = LAYOUT; }
- virtual void visit(StructDeclaration &) { kind = STRUCT; }
+ virtual void visit(InterfaceLayout &);
virtual void visit(VariableDeclaration &);
- virtual void visit(InterfaceBlock &) { kind = VARIABLE; }
- virtual void visit(FunctionDeclaration &);
};
} // namespace SL