Refactor UnusedVariableRemover

[libs/gl.git] / source / glsl / visitor.cpp

diff --git a/source/glsl/visitor.cpp b/source/glsl/visitor.cpp
index dad8baddce0c5a5bd74c1b56c4abeb4a93d48fd9..dfe2d8d46523a00d83b024f1e13dc261db36703d 100644 (file)
--- a/source/glsl/visitor.cpp
+++ b/source/glsl/visitor.cpp
@@ -26,6 +26,11 @@ void TraversingVisitor::visit(MemberAccess &memacc)
        memacc.left->visit(*this);
 }
 
+void TraversingVisitor::visit(Swizzle &swizzle)
+{
+       swizzle.left->visit(*this);
+}
+
 void TraversingVisitor::visit(UnaryExpression &unary)
 {
        unary.expression->visit(*this);
@@ -39,7 +44,8 @@ void TraversingVisitor::visit(BinaryExpression &binary)
 
 void TraversingVisitor::visit(Assignment &assign)
 {
-       visit(static_cast<BinaryExpression &>(assign));
+       assign.left->visit(*this);
+       assign.right->visit(*this);
 }
 
 void TraversingVisitor::visit(FunctionCall &call)
@@ -75,7 +81,8 @@ void TraversingVisitor::visit(VariableDeclaration &var)
 
 void TraversingVisitor::visit(InterfaceBlock &iface)
 {
-       iface.members.visit(*this);
+       if(iface.members)
+               iface.members->visit(*this);
 }
 
 void TraversingVisitor::visit(FunctionDeclaration &func)