]> git.tdb.fi Git - libs/gl.git/blobdiff - source/glsl/generate.h
Check the flat qualifier from the correct member
[libs/gl.git] / source / glsl / generate.h
index 90cf272219b9f2ba6bde8b5c4e6c349f5676218a..aba2c353f827239f7f4aaf3b2d2d73560d680db2 100644 (file)
@@ -11,119 +11,96 @@ namespace Msp {
 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 &);
 
-       void apply(Stage &s) { visit(s.content); }
-
-       using BlockModifier::visit;
-       virtual void visit(Block &);
-       virtual void visit(FunctionDeclaration &);
-       virtual void visit(VariableDeclaration &);
-};
-
-class VariableResolver: public TraversingVisitor
-{
 private:
-       std::vector<Block *> blocks;
-       StructDeclaration *type;
-       bool anonymous;
-       std::string block_interface;
-       bool record_target;
-       VariableDeclaration *assignment_target;
-       bool self_referencing;
-
-public:
-       VariableResolver();
-
-       void apply(Stage &);
-
-       using TraversingVisitor::visit;
-       virtual void visit(Block &);
-       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 TraversingVisitor
-{
-private:
-       std::map<std::string, std::vector<FunctionDeclaration *> > functions;
+/** Materializes implicitly declared interfaces.
 
-public:
-       void apply(Stage &s) { visit(s.content); }
+Out variable declarations inside functions are moved to the global scope.
 
-       using TraversingVisitor::visit;
-       virtual void visit(FunctionCall &);
-       virtual void visit(FunctionDeclaration &);
-};
+Passthrough statements are processed, generating out variables to match in
+variables and copying values.
 
-class InterfaceGenerator: public BlockModifier
+Unresolved variables are looked up in the previous stage's out variables. */
+class InterfaceGenerator: private TraversingVisitor
 {
 private:
-       Stage *stage;
+       Stage *stage = 0;
        std::string in_prefix;
        std::string out_prefix;
-       unsigned scope_level;
-       std::map<std::string, RefPtr<VariableDeclaration> > iface_declarations;
+       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:
-       InterfaceGenerator();
-
        void apply(Stage &);
 
+private:
        static std::string get_out_prefix(Stage::Type);
-       using BlockModifier::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 &);
+       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(VariableDeclaration &);
+       virtual void visit(FunctionDeclaration &);
        virtual void visit(Passthrough &);
 };
 
-class DeclarationReorderer: public TraversingVisitor
+class LayoutDefaulter: 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;
+       InterfaceLayout *in_iface = 0;
+       bool need_winding = true;
+       bool need_spacing = true;
 
 public:
-       DeclarationReorderer();
+       void apply(Stage &);
 
-       void apply(Stage &s) { visit(s.content); }
+private:
+       virtual void visit(InterfaceLayout &);
+};
 
-       using TraversingVisitor::visit;
-       virtual void visit(Block &);
+/**
+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:
+       std::map<VariableDeclaration *, int> max_indices;
+       unsigned input_size = 0;
+       VariableDeclaration *r_declaration;
+
+public:
+       void apply(Stage &);
+
+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