2 * Copyright © 2012 Intel Corporation
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
11 * The above copyright notice and this permission notice (including the next
12 * paragraph) shall be included in all copies or substantial portions of the
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
18 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
21 * DEALINGS IN THE SOFTWARE.
24 #include "main/core.h"
27 #include "ir_uniform.h"
28 #include "link_uniform_block_active_visitor.h"
29 #include "main/hash_table.h"
34 class ubo_visitor
: public program_resource_visitor
{
36 ubo_visitor(void *mem_ctx
, gl_uniform_buffer_variable
*variables
,
37 unsigned num_variables
)
38 : index(0), offset(0), buffer_size(0), variables(variables
),
39 num_variables(num_variables
), mem_ctx(mem_ctx
), is_array_instance(false)
44 void process(const glsl_type
*type
, const char *name
)
47 this->buffer_size
= 0;
48 this->is_array_instance
= strchr(name
, ']') != NULL
;
49 this->program_resource_visitor::process(type
, name
);
55 gl_uniform_buffer_variable
*variables
;
56 unsigned num_variables
;
58 bool is_array_instance
;
61 virtual void visit_field(const glsl_type
*type
, const char *name
,
67 assert(!"Should not get here.");
70 virtual void visit_field(const glsl_type
*type
, const char *name
,
71 bool row_major
, const glsl_type
*record_type
)
73 assert(this->index
< this->num_variables
);
75 gl_uniform_buffer_variable
*v
= &this->variables
[this->index
++];
77 v
->Name
= ralloc_strdup(mem_ctx
, name
);
79 v
->RowMajor
= row_major
;
81 if (this->is_array_instance
) {
82 v
->IndexName
= ralloc_strdup(mem_ctx
, name
);
84 char *open_bracket
= strchr(v
->IndexName
, '[');
85 assert(open_bracket
!= NULL
);
87 char *close_bracket
= strchr(open_bracket
, ']');
88 assert(close_bracket
!= NULL
);
90 /* Length of the tail without the ']' but with the NUL.
92 unsigned len
= strlen(close_bracket
+ 1) + 1;
94 memmove(open_bracket
, close_bracket
+ 1, len
);
96 v
->IndexName
= v
->Name
;
99 const unsigned alignment
= record_type
100 ? record_type
->std140_base_alignment(v
->RowMajor
)
101 : type
->std140_base_alignment(v
->RowMajor
);
102 unsigned size
= type
->std140_size(v
->RowMajor
);
104 this->offset
= glsl_align(this->offset
, alignment
);
105 v
->Offset
= this->offset
;
106 this->offset
+= size
;
108 /* From the GL_ARB_uniform_buffer_object spec:
110 * "For uniform blocks laid out according to [std140] rules, the
111 * minimum buffer object size returned by the
112 * UNIFORM_BLOCK_DATA_SIZE query is derived by taking the offset of
113 * the last basic machine unit consumed by the last uniform of the
114 * uniform block (including any end-of-array or end-of-structure
115 * padding), adding one, and rounding up to the next multiple of
116 * the base alignment required for a vec4."
118 this->buffer_size
= glsl_align(this->offset
, 16);
121 virtual void visit_field(const glsl_struct_field
*field
)
123 /* FINISHME: When support for doubles (dvec4, etc.) is added to the
124 * FINISHME: compiler, this may be incorrect for a structure in a UBO
125 * FINISHME: like struct s { struct { float f } s1; dvec4 v; };.
127 this->offset
= glsl_align(this->offset
,
128 field
->type
->std140_base_alignment(false));
132 class count_block_size
: public program_resource_visitor
{
134 count_block_size() : num_active_uniforms(0)
139 unsigned num_active_uniforms
;
142 virtual void visit_field(const glsl_type
*type
, const char *name
,
148 this->num_active_uniforms
++;
152 } /* anonymous namespace */
155 const glsl_type
*type
;
156 bool has_instance_name
;
160 link_uniform_blocks(void *mem_ctx
,
161 struct gl_shader_program
*prog
,
162 struct gl_shader
**shader_list
,
163 unsigned num_shaders
,
164 struct gl_uniform_block
**blocks_ret
)
166 /* This hash table will track all of the uniform blocks that have been
167 * encountered. Since blocks with the same block-name must be the same,
168 * the hash is organized by block-name.
170 struct hash_table
*block_hash
=
171 _mesa_hash_table_create(mem_ctx
, _mesa_key_string_equal
);
173 if (block_hash
== NULL
) {
174 _mesa_error_no_memory(__func__
);
175 linker_error(prog
, "out of memory\n");
179 /* Determine which uniform blocks are active.
181 link_uniform_block_active_visitor
v(mem_ctx
, block_hash
, prog
);
182 for (unsigned i
= 0; i
< num_shaders
; i
++) {
183 visit_list_elements(&v
, shader_list
[i
]->ir
);
186 /* Count the number of active uniform blocks. Count the total number of
187 * active slots in those uniform blocks.
189 unsigned num_blocks
= 0;
190 unsigned num_variables
= 0;
191 count_block_size block_size
;
192 struct hash_entry
*entry
;
194 hash_table_foreach (block_hash
, entry
) {
195 const struct link_uniform_block_active
*const b
=
196 (const struct link_uniform_block_active
*) entry
->data
;
198 const glsl_type
*const block_type
=
199 b
->type
->is_array() ? b
->type
->fields
.array
: b
->type
;
201 assert((b
->num_array_elements
> 0) == b
->type
->is_array());
203 block_size
.num_active_uniforms
= 0;
204 block_size
.process(block_type
, "");
206 if (b
->num_array_elements
> 0) {
207 num_blocks
+= b
->num_array_elements
;
208 num_variables
+= b
->num_array_elements
209 * block_size
.num_active_uniforms
;
212 num_variables
+= block_size
.num_active_uniforms
;
217 if (num_blocks
== 0) {
218 assert(num_variables
== 0);
219 _mesa_hash_table_destroy(block_hash
, NULL
);
223 assert(num_variables
!= 0);
225 /* Allocate storage to hold all of the informatation related to uniform
226 * blocks that can be queried through the API.
228 gl_uniform_block
*blocks
=
229 ralloc_array(mem_ctx
, gl_uniform_block
, num_blocks
);
230 gl_uniform_buffer_variable
*variables
=
231 ralloc_array(blocks
, gl_uniform_buffer_variable
, num_variables
);
233 /* Add each variable from each uniform block to the API tracking
237 ubo_visitor
parcel(blocks
, variables
, num_variables
);
239 STATIC_ASSERT(unsigned(GLSL_INTERFACE_PACKING_STD140
)
240 == unsigned(ubo_packing_std140
));
241 STATIC_ASSERT(unsigned(GLSL_INTERFACE_PACKING_SHARED
)
242 == unsigned(ubo_packing_shared
));
243 STATIC_ASSERT(unsigned(GLSL_INTERFACE_PACKING_PACKED
)
244 == unsigned(ubo_packing_packed
));
247 hash_table_foreach (block_hash
, entry
) {
248 const struct link_uniform_block_active
*const b
=
249 (const struct link_uniform_block_active
*) entry
->data
;
250 const glsl_type
*block_type
= b
->type
;
252 if (b
->num_array_elements
> 0) {
253 const char *const name
= block_type
->fields
.array
->name
;
255 assert(b
->has_instance_name
);
256 for (unsigned j
= 0; j
< b
->num_array_elements
; j
++) {
257 blocks
[i
].Name
= ralloc_asprintf(blocks
, "%s[%u]", name
,
258 b
->array_elements
[j
]);
259 blocks
[i
].Uniforms
= &variables
[parcel
.index
];
261 /* The GL_ARB_shading_language_420pack spec says:
263 * "If the binding identifier is used with a uniform block
264 * instanced as an array then the first element of the array
265 * takes the specified block binding and each subsequent
266 * element takes the next consecutive uniform block binding
269 blocks
[i
].Binding
= (b
->has_binding
) ? b
->binding
+ j
: 0;
271 blocks
[i
].UniformBufferSize
= 0;
273 gl_uniform_block_packing(block_type
->interface_packing
);
275 parcel
.process(block_type
->fields
.array
,
278 blocks
[i
].UniformBufferSize
= parcel
.buffer_size
;
280 blocks
[i
].NumUniforms
=
281 (unsigned)(ptrdiff_t)(&variables
[parcel
.index
] - blocks
[i
].Uniforms
);
286 blocks
[i
].Name
= ralloc_strdup(blocks
, block_type
->name
);
287 blocks
[i
].Uniforms
= &variables
[parcel
.index
];
288 blocks
[i
].Binding
= (b
->has_binding
) ? b
->binding
: 0;
289 blocks
[i
].UniformBufferSize
= 0;
291 gl_uniform_block_packing(block_type
->interface_packing
);
293 parcel
.process(block_type
,
294 b
->has_instance_name
? block_type
->name
: "");
296 blocks
[i
].UniformBufferSize
= parcel
.buffer_size
;
298 blocks
[i
].NumUniforms
=
299 (unsigned)(ptrdiff_t)(&variables
[parcel
.index
] - blocks
[i
].Uniforms
);
305 assert(parcel
.index
== num_variables
);
307 _mesa_hash_table_destroy(block_hash
, NULL
);
309 *blocks_ret
= blocks
;
314 link_uniform_blocks_are_compatible(const gl_uniform_block
*a
,
315 const gl_uniform_block
*b
)
317 assert(strcmp(a
->Name
, b
->Name
) == 0);
319 /* Page 35 (page 42 of the PDF) in section 4.3.7 of the GLSL 1.50 spec says:
321 * "Matched block names within an interface (as defined above) must
322 * match in terms of having the same number of declarations with the
323 * same sequence of types and the same sequence of member names, as
324 * well as having the same member-wise layout qualification....if a
325 * matching block is declared as an array, then the array sizes must
326 * also match... Any mismatch will generate a link error."
328 * Arrays are not yet supported, so there is no check for that.
330 if (a
->NumUniforms
!= b
->NumUniforms
)
333 if (a
->_Packing
!= b
->_Packing
)
336 for (unsigned i
= 0; i
< a
->NumUniforms
; i
++) {
337 if (strcmp(a
->Uniforms
[i
].Name
, b
->Uniforms
[i
].Name
) != 0)
340 if (a
->Uniforms
[i
].Type
!= b
->Uniforms
[i
].Type
)
343 if (a
->Uniforms
[i
].RowMajor
!= b
->Uniforms
[i
].RowMajor
)