glsl: Modify ir_end_primitive to have a stream.

[mesa.git] / src / glsl / ir_rvalue_visitor.cpp

diff --git a/src/glsl/ir_rvalue_visitor.cpp b/src/glsl/ir_rvalue_visitor.cpp
index eb6691ea1d978e37ff7f1484465552f690af1136..0370170b3bfcfaa8bba06ac2e8bd6c2e33281dc3 100644 (file)
--- a/src/glsl/ir_rvalue_visitor.cpp
+++ b/src/glsl/ir_rvalue_visitor.cpp
@@ -156,6 +156,13 @@ ir_rvalue_base_visitor::rvalue_visit(ir_emit_vertex *ir)
    return visit_continue;
 }
 
+ir_visitor_status
+ir_rvalue_base_visitor::rvalue_visit(ir_end_primitive *ir)
+{
+   handle_rvalue(&ir->stream);
+   return visit_continue;
+}
+
 ir_visitor_status
 ir_rvalue_visitor::visit_leave(ir_expression *ir)
 {
@@ -216,6 +223,12 @@ ir_rvalue_visitor::visit_leave(ir_emit_vertex *ir)
    return rvalue_visit(ir);
 }
 
+ir_visitor_status
+ir_rvalue_visitor::visit_leave(ir_end_primitive *ir)
+{
+   return rvalue_visit(ir);
+}
+
 ir_visitor_status
 ir_rvalue_enter_visitor::visit_enter(ir_expression *ir)
 {
@@ -275,3 +288,9 @@ ir_rvalue_enter_visitor::visit_enter(ir_emit_vertex *ir)
 {
    return rvalue_visit(ir);
 }
+
+ir_visitor_status
+ir_rvalue_enter_visitor::visit_enter(ir_end_primitive *ir)
+{
+   return rvalue_visit(ir);
+}