Store types and interface blocks in Stage

[libs/gl.git] / source / glsl / generate.h

diff --git a/source/glsl/generate.h b/source/glsl/generate.h
index 6740f38ea5c49e531b9fbbabf9b285e08ea7f24b..f1f10b897c281e9ec4a39ff032a167963b936b8a 100644 (file)
--- a/source/glsl/generate.h
+++ b/source/glsl/generate.h
@@ -33,15 +33,16 @@ public:
        void apply(Stage &s) { s.content.visit(*this); }
 
 private:
-       virtual void visit(Block &);
-       virtual void visit(InterfaceBlock &);
+       virtual void enter(Block &);
 };
 
 class VariableResolver: private TraversingVisitor
 {
 private:
+       Stage *stage;
        Block *builtins;
-       StructDeclaration *type;
+       std::map<std::string, VariableDeclaration *> *members;
+       RefPtr<InterfaceBlockReference> iface_ref;
        std::string block_interface;
        bool record_target;
        VariableDeclaration *assignment_target;
@@ -55,16 +56,15 @@ public:
 private:
        Block *next_block(Block &);
 
-       virtual void visit(Block &);
+       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(StructDeclaration &);
        virtual void visit(VariableDeclaration &);
        virtual void visit(InterfaceBlock &);
-       virtual void visit(FunctionDeclaration &);
-       virtual void visit(Iteration &);
 };
 
 class FunctionResolver: private TraversingVisitor
@@ -86,6 +86,10 @@ private:
        Stage *stage;
        std::string in_prefix;
        std::string out_prefix;
+       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;
@@ -100,9 +104,12 @@ private:
        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(InterfaceBlock &);
+       virtual void visit(FunctionDeclaration &);
        virtual void visit(Passthrough &);
 };