[Mesa-dev] [PATCH 36/74] glsl: Add parser/compiler support for unsized array's length()

Iago Toral Quiroga itoral at igalia.com
Thu May 14 07:06:39 PDT 2015


From: Samuel Iglesias Gonsalvez <siglesias at igalia.com>

It also creates unop and triop expressions to tell the driver to
calculate the unsized array length.

It is needed two expressions to do the calculation:

* The unop expression saves the ir_rvalue* whose length should be
  calculated.
* Afterwards, this unop is going to be processed by a lowering pass
that will convert it to a triop that includes the block index,
offset of the variable inside the shader storage block and the array
stride. All of them are needed for length() calculation following
GL_ARB_shader_storage_buffer spec:

   array.length() =
      max((buffer_object_size - offset_of_array) / stride_of_array, 0)

Signed-off-by: Samuel Iglesias Gonsalvez <siglesias at igalia.com>
---
 src/glsl/hir_field_selection.cpp           | 15 +++++++++++----
 src/glsl/ir.cpp                            |  9 +++++++++
 src/glsl/ir.h                              | 19 ++++++++++++++++++-
 src/glsl/ir_validate.cpp                   | 13 +++++++++++++
 src/glsl/link_uniforms.cpp                 |  8 +++++++-
 src/mesa/program/ir_to_mesa.cpp            |  2 ++
 src/mesa/state_tracker/st_glsl_to_tgsi.cpp |  5 +++++
 7 files changed, 65 insertions(+), 6 deletions(-)

diff --git a/src/glsl/hir_field_selection.cpp b/src/glsl/hir_field_selection.cpp
index 0fa9768..fedbef0 100644
--- a/src/glsl/hir_field_selection.cpp
+++ b/src/glsl/hir_field_selection.cpp
@@ -71,10 +71,17 @@ _mesa_ast_field_selection_to_hir(const ast_expression *expr,
             _mesa_glsl_error(&loc, state, "length method takes no arguments");
 
          if (op->type->is_array()) {
-            if (op->type->is_unsized_array())
-               _mesa_glsl_error(&loc, state, "length called on unsized array");
-
-            result = new(ctx) ir_constant(op->type->array_size());
+            if (op->type->is_unsized_array()) {
+               if (!state->ARB_shader_storage_buffer_object_enable) {
+                  _mesa_glsl_error(&loc, state, "length called on unsized array"
+                                                " only available with "
+                                                "ARB_shader_storage_buffer_object");
+               }
+               /* Calculate length of an unsized array in run-time */
+               result = new(ctx) ir_expression(ir_unop_ssbo_unsized_array_length, op);
+            } else {
+               result = new(ctx) ir_constant(op->type->array_size());
+            }
          } else if (op->type->is_vector()) {
             if (state->ARB_shading_language_420pack_enable) {
                /* .length() returns int. */
diff --git a/src/glsl/ir.cpp b/src/glsl/ir.cpp
index cbf2381..ffd853f 100644
--- a/src/glsl/ir.cpp
+++ b/src/glsl/ir.cpp
@@ -340,6 +340,9 @@ ir_expression::ir_expression(int op, ir_rvalue *op0)
       this->type = glsl_type::get_instance(GLSL_TYPE_INT,
 					   op0->type->vector_elements, 1);
       break;
+   case ir_unop_ssbo_unsized_array_length:
+      this->type = glsl_type::int_type;
+      break;
 
    default:
       assert(!"not reached: missing automatic type setup for ir_expression");
@@ -471,6 +474,10 @@ ir_expression::ir_expression(int op, ir_rvalue *op0, ir_rvalue *op1,
       this->type = op1->type;
       break;
 
+   case ir_triop_ssbo_unsized_array_length:
+      this->type = glsl_type::int_type;
+      break;
+
    default:
       assert(!"not reached: missing automatic type setup for ir_expression");
       this->type = glsl_type::float_type;
@@ -569,6 +576,7 @@ static const char *const operator_strs[] = {
    "frexp_exp",
    "noise",
    "interpolate_at_centroid",
+   "ssbo_unsized_array_length_parser",
    "+",
    "-",
    "*",
@@ -610,6 +618,7 @@ static const char *const operator_strs[] = {
    "csel",
    "bfi",
    "bitfield_extract",
+   "ssbo_unsized_array_length",
    "vector_insert",
    "bitfield_insert",
    "vector",
diff --git a/src/glsl/ir.h b/src/glsl/ir.h
index 4b113f5..63f40fc 100644
--- a/src/glsl/ir.h
+++ b/src/glsl/ir.h
@@ -1403,9 +1403,17 @@ enum ir_expression_operation {
    ir_unop_interpolate_at_centroid,
 
    /**
+    * Calculate length of an unsized array inside a buffer block.
+    * This opcode is going to be replaced in a lowering pass inside the linker.
+    *
+    * operand0 is the unsized array's ir_value for the calculation of its lenght.
+    */
+   ir_unop_ssbo_unsized_array_length,
+
+   /**
     * A sentinel marking the last of the unary operations.
     */
-   ir_last_unop = ir_unop_interpolate_at_centroid,
+   ir_last_unop = ir_unop_ssbo_unsized_array_length,
 
    ir_binop_add,
    ir_binop_sub,
@@ -1585,6 +1593,15 @@ enum ir_expression_operation {
    ir_triop_bitfield_extract,
 
    /**
+    * Calculate length of an unsized array inside a buffer block.
+    *
+    * operand0 is the ir_constant buffer block index in the linked shader.
+    * operand1 is a byte offset of the unsized array inside the buffer block
+    * operand2 is the array stride if the unsized array were sized.
+    */
+   ir_triop_ssbo_unsized_array_length,
+
+   /**
     * Generate a value with one field of a vector changed
     *
     * operand0 is the vector
diff --git a/src/glsl/ir_validate.cpp b/src/glsl/ir_validate.cpp
index b292ce4..f6ed1ff 100644
--- a/src/glsl/ir_validate.cpp
+++ b/src/glsl/ir_validate.cpp
@@ -408,6 +408,12 @@ ir_validate::visit_leave(ir_expression *ir)
       assert(ir->operands[0]->type->is_float());
       break;
 
+   case ir_unop_ssbo_unsized_array_length:
+      assert(ir->type == glsl_type::int_type);
+      assert(ir->operands[0]->type->is_array());
+      assert(ir->operands[0]->type->is_unsized_array());
+      break;
+
    case ir_unop_d2f:
       assert(ir->operands[0]->type->base_type == GLSL_TYPE_DOUBLE);
       assert(ir->type->base_type == GLSL_TYPE_FLOAT);
@@ -658,6 +664,13 @@ ir_validate::visit_leave(ir_expression *ir)
       assert(ir->operands[3]->type == glsl_type::int_type);
       break;
 
+   case ir_triop_ssbo_unsized_array_length:
+      assert(ir->type == glsl_type::int_type);
+      assert(ir->operands[0]->type == glsl_type::uint_type);
+      assert(ir->operands[1]->type == glsl_type::uint_type);
+      assert(ir->operands[2]->type == glsl_type::uint_type);
+      break;
+
    case ir_quadop_vector:
       /* The vector operator collects some number of scalars and generates a
        * vector from them.
diff --git a/src/glsl/link_uniforms.cpp b/src/glsl/link_uniforms.cpp
index 2137364..6debe78 100644
--- a/src/glsl/link_uniforms.cpp
+++ b/src/glsl/link_uniforms.cpp
@@ -221,7 +221,13 @@ program_resource_visitor::recursion(const glsl_type *t, char **name,
       if (record_type == NULL && t->fields.array->is_record())
          record_type = t->fields.array;
 
-      for (unsigned i = 0; i < t->length; i++) {
+      unsigned length = t->length;
+      /* Shader storage block unsized arrays: add subscript [0] to variable
+       * names */
+      if (t->is_unsized_array())
+         length = 1;
+
+      for (unsigned i = 0; i < length; i++) {
 	 size_t new_length = name_length;
 
 	 /* Append the subscript to the current variable name */
diff --git a/src/mesa/program/ir_to_mesa.cpp b/src/mesa/program/ir_to_mesa.cpp
index 8cdb704..e9f228a 100644
--- a/src/mesa/program/ir_to_mesa.cpp
+++ b/src/mesa/program/ir_to_mesa.cpp
@@ -1343,9 +1343,11 @@ ir_to_mesa_visitor::visit(ir_expression *ir)
    case ir_unop_dFdy_coarse:
    case ir_unop_dFdy_fine:
    case ir_binop_ssbo_load:
+   case ir_triop_ssbo_unsized_array_length:
       assert(!"not supported");
       break;
 
+   case ir_unop_ssbo_unsized_array_length:
    case ir_quadop_vector:
       /* This operation should have already been handled.
        */
diff --git a/src/mesa/state_tracker/st_glsl_to_tgsi.cpp b/src/mesa/state_tracker/st_glsl_to_tgsi.cpp
index 8265a8b..2a1127e 100644
--- a/src/mesa/state_tracker/st_glsl_to_tgsi.cpp
+++ b/src/mesa/state_tracker/st_glsl_to_tgsi.cpp
@@ -2192,10 +2192,15 @@ glsl_to_tgsi_visitor::visit(ir_expression *ir)
    case ir_triop_vector_insert:
    case ir_binop_carry:
    case ir_binop_borrow:
+   case ir_unop_ssbo_unsized_array_length:
       /* This operation is not supported, or should have already been handled.
        */
       assert(!"Invalid ir opcode in glsl_to_tgsi_visitor::visit()");
       break;
+
+   case ir_triop_ssbo_unsized_array_length:
+      assert(!"Not implemented yet");
+      break;
    }
 
    this->result = result_src;
-- 
1.9.1



More information about the mesa-dev mailing list