X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;ds=sidebyside;f=source%2Fglsl%2Fgenerate.h;h=ce95754eb4d68f9d28addbbceaa06839988be507;hb=ab5f2e6f1ddd35f8f117460530d76c0ba0c9bc87;hp=a09c552f1fd9f05c5329edabae112ece237017b0;hpb=7a16308e72aef363727b21348779673edf8e5c07;p=libs%2Fgl.git

diff --git a/source/glsl/generate.h b/source/glsl/generate.h
index a09c552f..ce95754e 100644
--- a/source/glsl/generate.h
+++ b/source/glsl/generate.h
@@ -11,23 +11,27 @@ namespace Msp {
 namespace GL {
 namespace SL {
 
-class DeclarationCombiner: public BlockModifier
+class DeclarationCombiner: private TraversingVisitor
 {
 private:
 	bool toplevel;
 	std::map<std::string, std::vector<FunctionDeclaration *> > functions;
 	std::map<std::string, VariableDeclaration *> variables;
+	std::set<Node *> nodes_to_remove;
 
 public:
 	DeclarationCombiner();
 
-	using StageVisitor::visit;
+	void apply(Stage &);
+
+private:
 	virtual void visit(Block &);
 	virtual void visit(FunctionDeclaration &);
 	virtual void visit(VariableDeclaration &);
+	using TraversingVisitor::visit;
 };
 
-class VariableResolver: public StageVisitor
+class VariableResolver: private TraversingVisitor
 {
 private:
 	std::vector<Block *> blocks;
@@ -41,8 +45,9 @@ private:
 public:
 	VariableResolver();
 
-	virtual void apply(Stage &);
-	using StageVisitor::visit;
+	void apply(Stage &);
+
+private:
 	virtual void visit(Block &);
 	virtual void visit(VariableReference &);
 	virtual void visit(MemberAccess &);
@@ -51,43 +56,53 @@ public:
 	virtual void visit(StructDeclaration &);
 	virtual void visit(VariableDeclaration &);
 	virtual void visit(InterfaceBlock &);
+	using TraversingVisitor::visit;
 };
 
-class FunctionResolver: public StageVisitor
+class FunctionResolver: private TraversingVisitor
 {
 private:
 	std::map<std::string, std::vector<FunctionDeclaration *> > functions;
 
 public:
-	using StageVisitor::visit;
+	void apply(Stage &s) { visit(s.content); }
+
+private:
 	virtual void visit(FunctionCall &);
 	virtual void visit(FunctionDeclaration &);
+	using TraversingVisitor::visit;
 };
 
-class InterfaceGenerator: public BlockModifier
+class InterfaceGenerator: private TraversingVisitor
 {
 private:
+	Stage *stage;
 	std::string in_prefix;
 	std::string out_prefix;
 	unsigned scope_level;
-	std::map<std::string, RefPtr<VariableDeclaration> > iface_declarations;
+	Block *current_block;
+	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);
-	virtual void apply(Stage &);
-	using StageVisitor::visit;
-	virtual void visit(Block &);
 	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 &);
 	ExpressionStatement &insert_assignment(const std::string &, Expression *);
 	virtual void visit(VariableReference &);
 	virtual void visit(VariableDeclaration &);
 	virtual void visit(Passthrough &);
+	using TraversingVisitor::visit;
 };
 
-class DeclarationReorderer: public StageVisitor
+class DeclarationReorderer: private TraversingVisitor
 {
 private:
 	enum DeclarationKind
@@ -107,7 +122,9 @@ private:
 public:
 	DeclarationReorderer();
 
-	using StageVisitor::visit;
+	void apply(Stage &s) { visit(s.content); }
+
+private:
 	virtual void visit(Block &);
 	virtual void visit(FunctionCall &);
 	virtual void visit(InterfaceLayout &) { kind = LAYOUT; }
@@ -115,6 +132,7 @@ public:
 	virtual void visit(VariableDeclaration &);
 	virtual void visit(InterfaceBlock &) { kind = VARIABLE; }
 	virtual void visit(FunctionDeclaration &);
+	using TraversingVisitor::visit;
 };
 
 } // namespace SL