glsl: Refactor AST-to-HIR code handling variable initializers

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

diff --git a/src/glsl/ir_expression_flattening.cpp b/src/glsl/ir_expression_flattening.cpp
index 0f10b671936312970f83b70df2912e8c522bf809..0b7c537bd8aafcb555317f0df91b0f46b5bff66c 100644 (file)
--- a/src/glsl/ir_expression_flattening.cpp
+++ b/src/glsl/ir_expression_flattening.cpp
@@ -35,15 +35,14 @@
 
 #include "ir.h"
 #include "ir_visitor.h"
+#include "ir_rvalue_visitor.h"
 #include "ir_expression_flattening.h"
 #include "glsl_types.h"
 
-class ir_expression_flattening_visitor : public ir_hierarchical_visitor {
+class ir_expression_flattening_visitor : public ir_rvalue_visitor {
 public:
-   ir_expression_flattening_visitor(ir_instruction *base_ir,
-                                   bool (*predicate)(ir_instruction *ir))
+   ir_expression_flattening_visitor(bool (*predicate)(ir_instruction *ir))
    {
-      this->base_ir = base_ir;
       this->predicate = predicate;
    }
 
@@ -52,44 +51,36 @@ public:
       /* empty */
    }
 
-   virtual ir_visitor_status visit_enter(ir_call *);
-   virtual ir_visitor_status visit_enter(ir_return *);
-   virtual ir_visitor_status visit_enter(ir_function_signature *);
-   virtual ir_visitor_status visit_enter(ir_if *);
-   virtual ir_visitor_status visit_enter(ir_loop *);
-   virtual ir_visitor_status visit_leave(ir_assignment *);
-   virtual ir_visitor_status visit_leave(ir_expression *);
-   virtual ir_visitor_status visit_leave(ir_swizzle *);
-
-   ir_rvalue *operand_to_temp(ir_rvalue *val);
+   void handle_rvalue(ir_rvalue **rvalue);
    bool (*predicate)(ir_instruction *ir);
-   ir_instruction *base_ir;
 };
 
 void
 do_expression_flattening(exec_list *instructions,
                         bool (*predicate)(ir_instruction *ir))
 {
+   ir_expression_flattening_visitor v(predicate);
+
    foreach_iter(exec_list_iterator, iter, *instructions) {
       ir_instruction *ir = (ir_instruction *)iter.get();
 
-      ir_expression_flattening_visitor v(ir, predicate);
       ir->accept(&v);
    }
 }
 
-
-ir_rvalue *
-ir_expression_flattening_visitor::operand_to_temp(ir_rvalue *ir)
+void
+ir_expression_flattening_visitor::handle_rvalue(ir_rvalue **rvalue)
 {
-   void *ctx = talloc_parent(base_ir);
    ir_variable *var;
    ir_assignment *assign;
+   ir_rvalue *ir = *rvalue;
+
+   if (!ir || !this->predicate(ir))
+      return;
 
-   if (!this->predicate(ir))
-      return ir;
+   void *ctx = ralloc_parent(ir);
 
-   var = new(ctx) ir_variable(ir->type, "flattening_tmp");
+   var = new(ctx) ir_variable(ir->type, "flattening_tmp", ir_var_temporary);
    base_ir->insert_before(var);
 
    assign = new(ctx) ir_assignment(new(ctx) ir_dereference_variable(var),
@@ -97,88 +88,5 @@ ir_expression_flattening_visitor::operand_to_temp(ir_rvalue *ir)
                                   NULL);
    base_ir->insert_before(assign);
 
-   return new(ctx) ir_dereference_variable(var);
-}
-
-ir_visitor_status
-ir_expression_flattening_visitor::visit_enter(ir_function_signature *ir)
-{
-   do_expression_flattening(&ir->body, this->predicate);
-
-   return visit_continue_with_parent;
-}
-
-ir_visitor_status
-ir_expression_flattening_visitor::visit_enter(ir_loop *ir)
-{
-   do_expression_flattening(&ir->body_instructions, this->predicate);
-
-   return visit_continue_with_parent;
-}
-
-ir_visitor_status
-ir_expression_flattening_visitor::visit_enter(ir_if *ir)
-{
-   ir->condition->accept(this);
-
-   do_expression_flattening(&ir->then_instructions, this->predicate);
-   do_expression_flattening(&ir->else_instructions, this->predicate);
-
-   return visit_continue_with_parent;
-}
-
-ir_visitor_status
-ir_expression_flattening_visitor::visit_leave(ir_expression *ir)
-{
-   unsigned int operand;
-
-   for (operand = 0; operand < ir->get_num_operands(); operand++) {
-      /* If the operand matches the predicate, then we'll assign its
-       * value to a temporary and deref the temporary as the operand.
-       */
-      ir->operands[operand] = operand_to_temp(ir->operands[operand]);
-   }
-
-   return visit_continue;
-}
-
-ir_visitor_status
-ir_expression_flattening_visitor::visit_leave(ir_assignment *ir)
-{
-   ir->rhs = operand_to_temp(ir->rhs);
-   if (ir->condition)
-      ir->condition = operand_to_temp(ir->condition);
-
-   return visit_continue;
-}
-
-ir_visitor_status
-ir_expression_flattening_visitor::visit_leave(ir_swizzle *ir)
-{
-   if (this->predicate(ir->val)) {
-      ir->val = operand_to_temp(ir->val);
-   }
-
-   return visit_continue;
-}
-
-ir_visitor_status
-ir_expression_flattening_visitor::visit_enter(ir_call *ir)
-{
-   /* FINISHME: Why not process the call parameters? (Same behavior as original
-    * FINISHME: code.)
-    */
-   (void) ir;
-   return visit_continue_with_parent;
-}
-
-
-ir_visitor_status
-ir_expression_flattening_visitor::visit_enter(ir_return *ir)
-{
-   /* FINISHME: Why not process the return value? (Same behavior as original
-    * FINISHME: code.)
-    */
-   (void) ir;
-   return visit_continue_with_parent;
+   *rvalue = new(ctx) ir_dereference_variable(var);
 }