Re: [Mesa-dev] [PATCH 3/3] nir: Add helpers for creating variables and adding them to lists

2015-10-13 Thread Iago Toral
On Mon, 2015-10-12 at 07:47 -0700, Jason Ekstrand wrote:
> 
> On Oct 12, 2015 1:26 AM, "Iago Toral"  wrote:
> >
> > On Fri, 2015-10-09 at 07:09 -0700, Jason Ekstrand wrote:
> > > ---
> > >  src/glsl/nir/glsl_to_nir.cpp   | 40 -
> > >  src/glsl/nir/nir.c | 66
> ++
> > >  src/glsl/nir/nir.h | 20 +
> > >  src/mesa/program/prog_to_nir.c | 19 +---
> > >  4 files changed, 99 insertions(+), 46 deletions(-)
> > >
> > > diff --git a/src/glsl/nir/glsl_to_nir.cpp
> b/src/glsl/nir/glsl_to_nir.cpp
> > > index 874e361..5250080 100644
> > > --- a/src/glsl/nir/glsl_to_nir.cpp
> > > +++ b/src/glsl/nir/glsl_to_nir.cpp
> > > @@ -389,35 +389,10 @@ nir_visitor::visit(ir_variable *ir)
> > >
> > > var->interface_type = ir->get_interface_type();
> > >
> > > -   switch (var->data.mode) {
> > > -   case nir_var_local:
> > > -  exec_list_push_tail(>locals, >node);
> > > -  break;
> > > -
> > > -   case nir_var_global:
> > > -  exec_list_push_tail(>globals, >node);
> > > -  break;
> > > -
> > > -   case nir_var_shader_in:
> > > -  exec_list_push_tail(>inputs, >node);
> > > -  break;
> > > -
> > > -   case nir_var_shader_out:
> > > -  exec_list_push_tail(>outputs, >node);
> > > -  break;
> > > -
> > > -   case nir_var_uniform:
> > > -   case nir_var_shader_storage:
> > > -  exec_list_push_tail(>uniforms, >node);
> > > -  break;
> > > -
> > > -   case nir_var_system_value:
> > > -  exec_list_push_tail(>system_values, >node);
> > > -  break;
> > > -
> > > -   default:
> > > -  unreachable("not reached");
> > > -   }
> > > +   if (var->data.mode == nir_var_local)
> > > +  nir_function_impl_add_variable(impl, var);
> > > +   else
> > > +  nir_shader_add_variable(shader, var);
> > >
> > > _mesa_hash_table_insert(var_table, ir, var);
> > > this->var = var;
> > > @@ -2023,13 +1998,10 @@ nir_visitor::visit(ir_constant *ir)
> > >  * constant initializer and return a dereference.
> > >  */
> > >
> > > -   nir_variable *var = ralloc(this->shader, nir_variable);
> > > -   var->name = ralloc_strdup(var, "const_temp");
> > > -   var->type = ir->type;
> > > -   var->data.mode = nir_var_local;
> > > +   nir_variable *var =
> > > +  nir_local_variable_create(this->impl, ir->type,
> "const_temp");
> > > var->data.read_only = true;
> > > var->constant_initializer = constant_copy(ir, var);
> > > -   exec_list_push_tail(>impl->locals, >node);
> > >
> > > this->deref_head = nir_deref_var_create(this->shader, var);
> > > this->deref_tail = >deref_head->deref;
> > > diff --git a/src/glsl/nir/nir.c b/src/glsl/nir/nir.c
> > > index e12da80..3645726 100644
> > > --- a/src/glsl/nir/nir.c
> > > +++ b/src/glsl/nir/nir.c
> > > @@ -103,6 +103,72 @@ nir_reg_remove(nir_register *reg)
> > > exec_node_remove(>node);
> > >  }
> > >
> > > +void
> > > +nir_shader_add_variable(nir_shader *shader, nir_variable *var)
> > > +{
> > > +   switch (var->data.mode) {
> > > +   case nir_var_local:
> > > +  assert(!"nir_shader_add_variable cannot be used for local
> variables");
> > > +  break;
> > > +
> > > +   case nir_var_global:
> > > +  exec_list_push_tail(>globals, >node);
> > > +  break;
> > > +
> > > +   case nir_var_shader_in:
> > > +  exec_list_push_tail(>inputs, >node);
> > > +  break;
> > > +
> > > +   case nir_var_shader_out:
> > > +  exec_list_push_tail(>outputs, >node);
> > > +  break;
> > > +
> > > +   case nir_var_uniform:
> > > +   case nir_var_shader_storage:
> > > +  exec_list_push_tail(>uniforms, >node);
> > > +  break;
> > > +
> > > +   case nir_var_system_value:
> > > +  exec_list_push_tail(>system_values, >node);
> > > +  break;
> > > +   }
> > > +}
> > > +
> > > +nir_variable *
> > > +nir_variable_create(nir_shader *shader, nir_variable_mode mode,
> > > +const struct glsl_type *type, const char
> *name)
> > > +{
> > > +   nir_variable *var = rzalloc(shader, nir_variable);
> > > +   var->name = ralloc_strdup(var, name);
> > > +   var->type = type;
> > > +   var->data.mode = mode;
> > > +
> > > +   if ((mode == nir_var_shader_in && shader->stage !=
> MESA_SHADER_VERTEX) ||
> > > +   (mode == nir_var_shader_out && shader->stage !=
> MESA_SHADER_FRAGMENT))
> > > +  var->data.interpolation = INTERP_QUALIFIER_SMOOTH;
> > > +
> > > +   if (mode == nir_var_shader_in || mode == nir_var_uniform)
> > > +  var->data.read_only = true;
> > > +
> > > +   nir_shader_add_variable(shader, var);
> > > +
> > > +   return var;
> > > +}
> > > +
> > > +nir_variable *
> > > +nir_local_variable_create(nir_function_impl *impl,
> > > +  const struct glsl_type *type, const
> char *name)
> > > +{
> > > +   nir_variable *var = rzalloc(impl->overload->function->shader,
> nir_variable);
> > > +   var->name = ralloc_strdup(var, name);
> > > +   var->type = type;

Re: [Mesa-dev] [PATCH 3/3] nir: Add helpers for creating variables and adding them to lists

2015-10-12 Thread Iago Toral
On Fri, 2015-10-09 at 07:09 -0700, Jason Ekstrand wrote:
> ---
>  src/glsl/nir/glsl_to_nir.cpp   | 40 -
>  src/glsl/nir/nir.c | 66 
> ++
>  src/glsl/nir/nir.h | 20 +
>  src/mesa/program/prog_to_nir.c | 19 +---
>  4 files changed, 99 insertions(+), 46 deletions(-)
> 
> diff --git a/src/glsl/nir/glsl_to_nir.cpp b/src/glsl/nir/glsl_to_nir.cpp
> index 874e361..5250080 100644
> --- a/src/glsl/nir/glsl_to_nir.cpp
> +++ b/src/glsl/nir/glsl_to_nir.cpp
> @@ -389,35 +389,10 @@ nir_visitor::visit(ir_variable *ir)
>  
> var->interface_type = ir->get_interface_type();
>  
> -   switch (var->data.mode) {
> -   case nir_var_local:
> -  exec_list_push_tail(>locals, >node);
> -  break;
> -
> -   case nir_var_global:
> -  exec_list_push_tail(>globals, >node);
> -  break;
> -
> -   case nir_var_shader_in:
> -  exec_list_push_tail(>inputs, >node);
> -  break;
> -
> -   case nir_var_shader_out:
> -  exec_list_push_tail(>outputs, >node);
> -  break;
> -
> -   case nir_var_uniform:
> -   case nir_var_shader_storage:
> -  exec_list_push_tail(>uniforms, >node);
> -  break;
> -
> -   case nir_var_system_value:
> -  exec_list_push_tail(>system_values, >node);
> -  break;
> -
> -   default:
> -  unreachable("not reached");
> -   }
> +   if (var->data.mode == nir_var_local)
> +  nir_function_impl_add_variable(impl, var);
> +   else
> +  nir_shader_add_variable(shader, var);
>  
> _mesa_hash_table_insert(var_table, ir, var);
> this->var = var;
> @@ -2023,13 +1998,10 @@ nir_visitor::visit(ir_constant *ir)
>  * constant initializer and return a dereference.
>  */
>  
> -   nir_variable *var = ralloc(this->shader, nir_variable);
> -   var->name = ralloc_strdup(var, "const_temp");
> -   var->type = ir->type;
> -   var->data.mode = nir_var_local;
> +   nir_variable *var =
> +  nir_local_variable_create(this->impl, ir->type, "const_temp");
> var->data.read_only = true;
> var->constant_initializer = constant_copy(ir, var);
> -   exec_list_push_tail(>impl->locals, >node);
>  
> this->deref_head = nir_deref_var_create(this->shader, var);
> this->deref_tail = >deref_head->deref;
> diff --git a/src/glsl/nir/nir.c b/src/glsl/nir/nir.c
> index e12da80..3645726 100644
> --- a/src/glsl/nir/nir.c
> +++ b/src/glsl/nir/nir.c
> @@ -103,6 +103,72 @@ nir_reg_remove(nir_register *reg)
> exec_node_remove(>node);
>  }
>  
> +void
> +nir_shader_add_variable(nir_shader *shader, nir_variable *var)
> +{
> +   switch (var->data.mode) {
> +   case nir_var_local:
> +  assert(!"nir_shader_add_variable cannot be used for local variables");
> +  break;
> +
> +   case nir_var_global:
> +  exec_list_push_tail(>globals, >node);
> +  break;
> +
> +   case nir_var_shader_in:
> +  exec_list_push_tail(>inputs, >node);
> +  break;
> +
> +   case nir_var_shader_out:
> +  exec_list_push_tail(>outputs, >node);
> +  break;
> +
> +   case nir_var_uniform:
> +   case nir_var_shader_storage:
> +  exec_list_push_tail(>uniforms, >node);
> +  break;
> +
> +   case nir_var_system_value:
> +  exec_list_push_tail(>system_values, >node);
> +  break;
> +   }
> +}
> +
> +nir_variable *
> +nir_variable_create(nir_shader *shader, nir_variable_mode mode,
> +const struct glsl_type *type, const char *name)
> +{
> +   nir_variable *var = rzalloc(shader, nir_variable);
> +   var->name = ralloc_strdup(var, name);
> +   var->type = type;
> +   var->data.mode = mode;
> +
> +   if ((mode == nir_var_shader_in && shader->stage != MESA_SHADER_VERTEX) ||
> +   (mode == nir_var_shader_out && shader->stage != MESA_SHADER_FRAGMENT))
> +  var->data.interpolation = INTERP_QUALIFIER_SMOOTH;
> +
> +   if (mode == nir_var_shader_in || mode == nir_var_uniform)
> +  var->data.read_only = true;
> +
> +   nir_shader_add_variable(shader, var);
> +
> +   return var;
> +}
> +
> +nir_variable *
> +nir_local_variable_create(nir_function_impl *impl,
> +  const struct glsl_type *type, const char *name)
> +{
> +   nir_variable *var = rzalloc(impl->overload->function->shader, 
> nir_variable);
> +   var->name = ralloc_strdup(var, name);
> +   var->type = type;
> +   var->data.mode = nir_var_local;
> +
> +   nir_function_impl_add_variable(impl, var);
> +
> +   return var;
> +}
> +
>  nir_function *
>  nir_function_create(nir_shader *shader, const char *name)
>  {
> diff --git a/src/glsl/nir/nir.h b/src/glsl/nir/nir.h
> index bde9f49..997e2bd 100644
> --- a/src/glsl/nir/nir.h
> +++ b/src/glsl/nir/nir.h
> @@ -1559,6 +1559,26 @@ nir_register *nir_local_reg_create(nir_function_impl 
> *impl);
>  
>  void nir_reg_remove(nir_register *reg);
>  
> +/** Adds a variable to the appropreate list in nir_shader */
> +void nir_shader_add_variable(nir_shader *shader, nir_variable *var);
> +
> +static inline void
> 

Re: [Mesa-dev] [PATCH 3/3] nir: Add helpers for creating variables and adding them to lists

2015-10-12 Thread Jason Ekstrand
On Oct 12, 2015 1:26 AM, "Iago Toral"  wrote:
>
> On Fri, 2015-10-09 at 07:09 -0700, Jason Ekstrand wrote:
> > ---
> >  src/glsl/nir/glsl_to_nir.cpp   | 40 -
> >  src/glsl/nir/nir.c | 66
++
> >  src/glsl/nir/nir.h | 20 +
> >  src/mesa/program/prog_to_nir.c | 19 +---
> >  4 files changed, 99 insertions(+), 46 deletions(-)
> >
> > diff --git a/src/glsl/nir/glsl_to_nir.cpp b/src/glsl/nir/glsl_to_nir.cpp
> > index 874e361..5250080 100644
> > --- a/src/glsl/nir/glsl_to_nir.cpp
> > +++ b/src/glsl/nir/glsl_to_nir.cpp
> > @@ -389,35 +389,10 @@ nir_visitor::visit(ir_variable *ir)
> >
> > var->interface_type = ir->get_interface_type();
> >
> > -   switch (var->data.mode) {
> > -   case nir_var_local:
> > -  exec_list_push_tail(>locals, >node);
> > -  break;
> > -
> > -   case nir_var_global:
> > -  exec_list_push_tail(>globals, >node);
> > -  break;
> > -
> > -   case nir_var_shader_in:
> > -  exec_list_push_tail(>inputs, >node);
> > -  break;
> > -
> > -   case nir_var_shader_out:
> > -  exec_list_push_tail(>outputs, >node);
> > -  break;
> > -
> > -   case nir_var_uniform:
> > -   case nir_var_shader_storage:
> > -  exec_list_push_tail(>uniforms, >node);
> > -  break;
> > -
> > -   case nir_var_system_value:
> > -  exec_list_push_tail(>system_values, >node);
> > -  break;
> > -
> > -   default:
> > -  unreachable("not reached");
> > -   }
> > +   if (var->data.mode == nir_var_local)
> > +  nir_function_impl_add_variable(impl, var);
> > +   else
> > +  nir_shader_add_variable(shader, var);
> >
> > _mesa_hash_table_insert(var_table, ir, var);
> > this->var = var;
> > @@ -2023,13 +1998,10 @@ nir_visitor::visit(ir_constant *ir)
> >  * constant initializer and return a dereference.
> >  */
> >
> > -   nir_variable *var = ralloc(this->shader, nir_variable);
> > -   var->name = ralloc_strdup(var, "const_temp");
> > -   var->type = ir->type;
> > -   var->data.mode = nir_var_local;
> > +   nir_variable *var =
> > +  nir_local_variable_create(this->impl, ir->type, "const_temp");
> > var->data.read_only = true;
> > var->constant_initializer = constant_copy(ir, var);
> > -   exec_list_push_tail(>impl->locals, >node);
> >
> > this->deref_head = nir_deref_var_create(this->shader, var);
> > this->deref_tail = >deref_head->deref;
> > diff --git a/src/glsl/nir/nir.c b/src/glsl/nir/nir.c
> > index e12da80..3645726 100644
> > --- a/src/glsl/nir/nir.c
> > +++ b/src/glsl/nir/nir.c
> > @@ -103,6 +103,72 @@ nir_reg_remove(nir_register *reg)
> > exec_node_remove(>node);
> >  }
> >
> > +void
> > +nir_shader_add_variable(nir_shader *shader, nir_variable *var)
> > +{
> > +   switch (var->data.mode) {
> > +   case nir_var_local:
> > +  assert(!"nir_shader_add_variable cannot be used for local
variables");
> > +  break;
> > +
> > +   case nir_var_global:
> > +  exec_list_push_tail(>globals, >node);
> > +  break;
> > +
> > +   case nir_var_shader_in:
> > +  exec_list_push_tail(>inputs, >node);
> > +  break;
> > +
> > +   case nir_var_shader_out:
> > +  exec_list_push_tail(>outputs, >node);
> > +  break;
> > +
> > +   case nir_var_uniform:
> > +   case nir_var_shader_storage:
> > +  exec_list_push_tail(>uniforms, >node);
> > +  break;
> > +
> > +   case nir_var_system_value:
> > +  exec_list_push_tail(>system_values, >node);
> > +  break;
> > +   }
> > +}
> > +
> > +nir_variable *
> > +nir_variable_create(nir_shader *shader, nir_variable_mode mode,
> > +const struct glsl_type *type, const char *name)
> > +{
> > +   nir_variable *var = rzalloc(shader, nir_variable);
> > +   var->name = ralloc_strdup(var, name);
> > +   var->type = type;
> > +   var->data.mode = mode;
> > +
> > +   if ((mode == nir_var_shader_in && shader->stage !=
MESA_SHADER_VERTEX) ||
> > +   (mode == nir_var_shader_out && shader->stage !=
MESA_SHADER_FRAGMENT))
> > +  var->data.interpolation = INTERP_QUALIFIER_SMOOTH;
> > +
> > +   if (mode == nir_var_shader_in || mode == nir_var_uniform)
> > +  var->data.read_only = true;
> > +
> > +   nir_shader_add_variable(shader, var);
> > +
> > +   return var;
> > +}
> > +
> > +nir_variable *
> > +nir_local_variable_create(nir_function_impl *impl,
> > +  const struct glsl_type *type, const char
*name)
> > +{
> > +   nir_variable *var = rzalloc(impl->overload->function->shader,
nir_variable);
> > +   var->name = ralloc_strdup(var, name);
> > +   var->type = type;
> > +   var->data.mode = nir_var_local;
> > +
> > +   nir_function_impl_add_variable(impl, var);
> > +
> > +   return var;
> > +}
> > +
> >  nir_function *
> >  nir_function_create(nir_shader *shader, const char *name)
> >  {
> > diff --git a/src/glsl/nir/nir.h b/src/glsl/nir/nir.h
> > index bde9f49..997e2bd 100644
> > --- a/src/glsl/nir/nir.h
> > 

[Mesa-dev] [PATCH 3/3] nir: Add helpers for creating variables and adding them to lists

2015-10-09 Thread Jason Ekstrand
---
 src/glsl/nir/glsl_to_nir.cpp   | 40 -
 src/glsl/nir/nir.c | 66 ++
 src/glsl/nir/nir.h | 20 +
 src/mesa/program/prog_to_nir.c | 19 +---
 4 files changed, 99 insertions(+), 46 deletions(-)

diff --git a/src/glsl/nir/glsl_to_nir.cpp b/src/glsl/nir/glsl_to_nir.cpp
index 874e361..5250080 100644
--- a/src/glsl/nir/glsl_to_nir.cpp
+++ b/src/glsl/nir/glsl_to_nir.cpp
@@ -389,35 +389,10 @@ nir_visitor::visit(ir_variable *ir)
 
var->interface_type = ir->get_interface_type();
 
-   switch (var->data.mode) {
-   case nir_var_local:
-  exec_list_push_tail(>locals, >node);
-  break;
-
-   case nir_var_global:
-  exec_list_push_tail(>globals, >node);
-  break;
-
-   case nir_var_shader_in:
-  exec_list_push_tail(>inputs, >node);
-  break;
-
-   case nir_var_shader_out:
-  exec_list_push_tail(>outputs, >node);
-  break;
-
-   case nir_var_uniform:
-   case nir_var_shader_storage:
-  exec_list_push_tail(>uniforms, >node);
-  break;
-
-   case nir_var_system_value:
-  exec_list_push_tail(>system_values, >node);
-  break;
-
-   default:
-  unreachable("not reached");
-   }
+   if (var->data.mode == nir_var_local)
+  nir_function_impl_add_variable(impl, var);
+   else
+  nir_shader_add_variable(shader, var);
 
_mesa_hash_table_insert(var_table, ir, var);
this->var = var;
@@ -2023,13 +1998,10 @@ nir_visitor::visit(ir_constant *ir)
 * constant initializer and return a dereference.
 */
 
-   nir_variable *var = ralloc(this->shader, nir_variable);
-   var->name = ralloc_strdup(var, "const_temp");
-   var->type = ir->type;
-   var->data.mode = nir_var_local;
+   nir_variable *var =
+  nir_local_variable_create(this->impl, ir->type, "const_temp");
var->data.read_only = true;
var->constant_initializer = constant_copy(ir, var);
-   exec_list_push_tail(>impl->locals, >node);
 
this->deref_head = nir_deref_var_create(this->shader, var);
this->deref_tail = >deref_head->deref;
diff --git a/src/glsl/nir/nir.c b/src/glsl/nir/nir.c
index e12da80..3645726 100644
--- a/src/glsl/nir/nir.c
+++ b/src/glsl/nir/nir.c
@@ -103,6 +103,72 @@ nir_reg_remove(nir_register *reg)
exec_node_remove(>node);
 }
 
+void
+nir_shader_add_variable(nir_shader *shader, nir_variable *var)
+{
+   switch (var->data.mode) {
+   case nir_var_local:
+  assert(!"nir_shader_add_variable cannot be used for local variables");
+  break;
+
+   case nir_var_global:
+  exec_list_push_tail(>globals, >node);
+  break;
+
+   case nir_var_shader_in:
+  exec_list_push_tail(>inputs, >node);
+  break;
+
+   case nir_var_shader_out:
+  exec_list_push_tail(>outputs, >node);
+  break;
+
+   case nir_var_uniform:
+   case nir_var_shader_storage:
+  exec_list_push_tail(>uniforms, >node);
+  break;
+
+   case nir_var_system_value:
+  exec_list_push_tail(>system_values, >node);
+  break;
+   }
+}
+
+nir_variable *
+nir_variable_create(nir_shader *shader, nir_variable_mode mode,
+const struct glsl_type *type, const char *name)
+{
+   nir_variable *var = rzalloc(shader, nir_variable);
+   var->name = ralloc_strdup(var, name);
+   var->type = type;
+   var->data.mode = mode;
+
+   if ((mode == nir_var_shader_in && shader->stage != MESA_SHADER_VERTEX) ||
+   (mode == nir_var_shader_out && shader->stage != MESA_SHADER_FRAGMENT))
+  var->data.interpolation = INTERP_QUALIFIER_SMOOTH;
+
+   if (mode == nir_var_shader_in || mode == nir_var_uniform)
+  var->data.read_only = true;
+
+   nir_shader_add_variable(shader, var);
+
+   return var;
+}
+
+nir_variable *
+nir_local_variable_create(nir_function_impl *impl,
+  const struct glsl_type *type, const char *name)
+{
+   nir_variable *var = rzalloc(impl->overload->function->shader, nir_variable);
+   var->name = ralloc_strdup(var, name);
+   var->type = type;
+   var->data.mode = nir_var_local;
+
+   nir_function_impl_add_variable(impl, var);
+
+   return var;
+}
+
 nir_function *
 nir_function_create(nir_shader *shader, const char *name)
 {
diff --git a/src/glsl/nir/nir.h b/src/glsl/nir/nir.h
index bde9f49..997e2bd 100644
--- a/src/glsl/nir/nir.h
+++ b/src/glsl/nir/nir.h
@@ -1559,6 +1559,26 @@ nir_register *nir_local_reg_create(nir_function_impl 
*impl);
 
 void nir_reg_remove(nir_register *reg);
 
+/** Adds a variable to the appropreate list in nir_shader */
+void nir_shader_add_variable(nir_shader *shader, nir_variable *var);
+
+static inline void
+nir_function_impl_add_variable(nir_function_impl *impl, nir_variable *var)
+{
+   assert(var->data.mode == nir_var_local);
+   exec_list_push_tail(>locals, >node);
+}
+
+/** creates a variable, sets a few defaults, and adds it to the list */
+nir_variable *nir_variable_create(nir_shader *shader,
+  nir_variable_mode mode,
+  const