From 0a0ab121f87a3b00e29c20a38fdd28e2fc31bae7 Mon Sep 17 00:00:00 2001
From: Eric Anholt <eric@anholt.net>
Date: Thu, 5 Aug 2010 13:25:39 -0700
Subject: [PATCH] ir_structure_splitting: Massive fixing to this.

I'd missed putting in the actual "find structures to split" part, so
most of the code didn't do anything.  I was running on too large of an
app and assuming the lack of progress was elsewhere.
---
 src/glsl/ir_structure_splitting.cpp | 43 +++++++++++++++++++++++------
 1 file changed, 34 insertions(+), 9 deletions(-)

diff --git a/src/glsl/ir_structure_splitting.cpp b/src/glsl/ir_structure_splitting.cpp
index f57ae44aae4..902eeadb2a4 100644
--- a/src/glsl/ir_structure_splitting.cpp
+++ b/src/glsl/ir_structure_splitting.cpp
@@ -34,8 +34,11 @@
 
 #include "ir.h"
 #include "ir_visitor.h"
+#include "ir_print_visitor.h"
 #include "glsl_types.h"
 
+static bool debug = false;
+
 class variable_entry : public exec_node
 {
 public:
@@ -76,7 +79,7 @@ public:
 
    virtual ir_visitor_status visit(ir_variable *);
    virtual ir_visitor_status visit(ir_dereference_variable *);
-   virtual ir_visitor_status visit(ir_dereference_record *);
+   virtual ir_visitor_status visit_enter(ir_dereference_record *);
 
    virtual ir_visitor_status visit_enter(ir_function_signature *);
 
@@ -93,7 +96,7 @@ ir_structure_reference_visitor::get_variable_entry(ir_variable *var)
 {
    assert(var);
 
-   if (!var->type->is_record())
+   if (!var->type->is_record() || var->mode == ir_var_uniform)
       return NULL;
 
    foreach_iter(exec_list_iterator, iter, this->variable_list) {
@@ -132,10 +135,10 @@ ir_structure_reference_visitor::visit(ir_dereference_variable *ir)
 }
 
 ir_visitor_status
-ir_structure_reference_visitor::visit(ir_dereference_record *ir)
+ir_structure_reference_visitor::visit_enter(ir_dereference_record *ir)
 {
    /* Don't descend into the ir_dereference_variable below. */
-   return visit_continue;
+   return visit_continue_with_parent;
 }
 
 ir_visitor_status
@@ -162,6 +165,7 @@ public:
    virtual ir_visitor_status visit_leave(ir_assignment *);
    virtual ir_visitor_status visit_leave(ir_call *);
    virtual ir_visitor_status visit_leave(ir_dereference_array *);
+   virtual ir_visitor_status visit_leave(ir_dereference_record *);
    virtual ir_visitor_status visit_leave(ir_expression *);
    virtual ir_visitor_status visit_leave(ir_if *);
    virtual ir_visitor_status visit_leave(ir_return *);
@@ -200,13 +204,13 @@ ir_structure_splitting_visitor::split_deref(ir_dereference **deref)
    if ((*deref)->ir_type != ir_type_dereference_record)
       return;
 
-   ir_dereference_record *deref_record = (ir_dereference_record *)deref;
-   ir_dereference_variable *deref_var = deref_record->as_dereference_variable();
+   ir_dereference_record *deref_record = (ir_dereference_record *)*deref;
+   ir_dereference_variable *deref_var = deref_record->record->as_dereference_variable();
    if (!deref_var)
       return;
 
    variable_entry *entry = get_splitting_entry(deref_var->var);
-   if (entry)
+   if (!entry)
       return;
 
    unsigned int i;
@@ -223,6 +227,9 @@ ir_structure_splitting_visitor::split_deref(ir_dereference **deref)
 void
 ir_structure_splitting_visitor::split_rvalue(ir_rvalue **rvalue)
 {
+   if (!*rvalue)
+      return;
+
    ir_dereference *deref = (*rvalue)->as_dereference();
 
    if (!deref)
@@ -281,6 +288,14 @@ ir_visitor_status
 ir_structure_splitting_visitor::visit_leave(ir_dereference_array *ir)
 {
    split_rvalue(&ir->array_index);
+   split_rvalue(&ir->array);
+   return visit_continue;
+}
+
+ir_visitor_status
+ir_structure_splitting_visitor::visit_leave(ir_dereference_record *ir)
+{
+   split_rvalue(&ir->record);
    return visit_continue;
 }
 
@@ -328,11 +343,19 @@ bool
 do_structure_splitting(exec_list *instructions)
 {
    ir_structure_reference_visitor refs;
-   void *mem_ctx = talloc_new(NULL);
+
+   visit_list_elements(&refs, instructions);
 
    /* Trim out variables we can't split. */
    foreach_iter(exec_list_iterator, iter, refs.variable_list) {
       variable_entry *entry = (variable_entry *)iter.get();
+
+      if (debug) {
+	 printf("structure %s@%p: decl %d, whole_access %d\n",
+		entry->var->name, entry->var, entry->declaration,
+		entry->whole_structure_access);
+      }
+
       if (!entry->declaration || entry->whole_structure_access) {
 	 entry->remove();
       }
@@ -341,6 +364,8 @@ do_structure_splitting(exec_list *instructions)
    if (refs.variable_list.is_empty())
       return false;
 
+   void *mem_ctx = talloc_new(NULL);
+
    /* Replace the decls of the structures to be split with their split
     * components.
     */
@@ -356,7 +381,7 @@ do_structure_splitting(exec_list *instructions)
 
       for (unsigned int i = 0; i < entry->var->type->length; i++) {
 	 const char *name = talloc_asprintf(mem_ctx, "%s_%s",
-					    type->name,
+					    entry->var->name,
 					    type->fields.structure[i].name);
 
 	 entry->components[i] =
-- 
2.30.2