X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fglsl%2Foptimize.h;h=ea772b3d841b1ef78171d99ebb9afef064a76dfd;hb=ab5f2e6f1ddd35f8f117460530d76c0ba0c9bc87;hp=2d1b20c99e72d7b1361508cc9bdda14c2424952b;hpb=ff8a0248f1a3c0c1f48f670867bc9106c898c55b;p=libs%2Fgl.git

diff --git a/source/glsl/optimize.h b/source/glsl/optimize.h
index 2d1b20c9..ea772b3d 100644
--- a/source/glsl/optimize.h
+++ b/source/glsl/optimize.h
@@ -53,17 +53,20 @@ private:
 	using TraversingVisitor::visit;
 };
 
-class ConstantConditionEliminator: private BlockModifier
+class ConstantConditionEliminator: private TraversingVisitor
 {
 private:
 	unsigned scope_level;
+	Block *current_block;
 	bool record_only;
 	ExpressionEvaluator::ValueMap variable_values;
+	NodeList<Statement>::iterator insert_point;
+	std::set<Node *> nodes_to_remove;
 
 public:
 	ConstantConditionEliminator();
 
-	void apply(Stage &s) { visit(s.content); }
+	void apply(Stage &);
 
 private:
 	virtual void visit(Block &);
@@ -72,7 +75,7 @@ private:
 	virtual void visit(VariableDeclaration &);
 	virtual void visit(Conditional &);
 	virtual void visit(Iteration &);
-	using BlockModifier::visit;
+	using TraversingVisitor::visit;
 };
 
 class UnusedVariableRemover: private TraversingVisitor