X-Git-Url: http://git.tdb.fi/?p=libs%2Fgl.git;a=blobdiff_plain;f=source%2Fglsl%2Foptimize.h;h=551e2d56976155a0d5b8b1e5c56e848fe95281e1;hp=c932767e99807325f159372e626895aafa213c5b;hb=041ba4b1acd55337239c5ce24cc310118c621206;hpb=1cd0ea7f79b2b0dedd8a2a6622e5d2e8b2ea2512

diff --git a/source/glsl/optimize.h b/source/glsl/optimize.h
index c932767e..551e2d56 100644
--- a/source/glsl/optimize.h
+++ b/source/glsl/optimize.h
@@ -88,6 +88,7 @@ private:
 	virtual void visit(Block &);
 	virtual void visit(UnaryExpression &);
 	virtual void visit(BinaryExpression &);
+	virtual void visit(Assignment &a) { visit(static_cast<BinaryExpression &>(a)); }
 	virtual void visit(MemberAccess &);
 	virtual void visit(FunctionCall &);
 	virtual void visit(ExpressionStatement &);
@@ -165,6 +166,28 @@ private:
 	virtual void visit(Iteration &);
 };
 
+/** Removes types which are not used anywhere. */
+class UnusedTypeRemover: private TraversingVisitor
+{
+private:
+	std::set<Node *> unused_nodes;
+
+public:
+	bool apply(Stage &);
+
+private:
+	virtual void visit(Literal &);
+	virtual void visit(UnaryExpression &);
+	virtual void visit(BinaryExpression &);
+	virtual void visit(FunctionCall &);
+	virtual void visit(BasicTypeDeclaration &);
+	virtual void visit(ImageTypeDeclaration &);
+	virtual void visit(StructDeclaration &);
+	virtual void visit(VariableDeclaration &);
+	virtual void visit(InterfaceBlock &);
+	virtual void visit(FunctionDeclaration &);
+};
+
 /** Removes variable declarations with no references to them.  Assignment
 statements where the result is not used are also removed. */
 class UnusedVariableRemover: private TraversingVisitor