X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fglsl%2Fgenerate.h;h=68816b8b50acec9411ec331e8b866749412bb3d2;hb=6288c42adde9ee7d39a47de51fa2856cf965dccc;hp=6740f38ea5c49e531b9fbbabf9b285e08ea7f24b;hpb=5c33b56c3b97ca0381ac216a603c7553f4bea499;p=libs%2Fgl.git

diff --git a/source/glsl/generate.h b/source/glsl/generate.h
index 6740f38e..68816b8b 100644
--- a/source/glsl/generate.h
+++ b/source/glsl/generate.h
@@ -27,21 +27,36 @@ private:
 	virtual void visit(VariableDeclaration &);
 };
 
+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 BlockResolver: private TraversingVisitor
 {
 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 +70,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 +100,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 +118,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 &);
 };