X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fglsl%2Fvisitor.cpp;fp=source%2Fglsl%2Fvisitor.cpp;h=d8b1813959f17e4535ea1ea8c60001c22a36d76f;hb=39141d961e55f56d540b3678f9f8fe94868f2652;hp=fd25bf342a7ea588525be88079166453474c4e8e;hpb=f6dfd3e78fc8c4c4a14619f3fdece53600ab0db7;p=libs%2Fgl.git

diff --git a/source/glsl/visitor.cpp b/source/glsl/visitor.cpp
index fd25bf34..d8b18139 100644
--- a/source/glsl/visitor.cpp
+++ b/source/glsl/visitor.cpp
@@ -186,6 +186,13 @@ void NodeRemover::visit(InterfaceBlock &iface)
 	TraversingVisitor::visit(iface);
 }
 
+void NodeRemover::visit(FunctionDeclaration &func)
+{
+	if(to_remove->count(&func))
+		remove_from_map(stage->functions, func.name+func.signature, func);
+	TraversingVisitor::visit(func);
+}
+
 void NodeRemover::visit(Iteration &iter)
 {
 	if(to_remove->count(iter.init_statement.get()))