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"
29 /* These functions are put in a "private" namespace instead of being marked
30 * static so that the unit tests can access them. See
31 * http://code.google.com/p/googletest/wiki/AdvancedGuide#Testing_Private_Code
36 get_storage(gl_uniform_storage
*storage
, unsigned num_storage
,
39 for (unsigned int i
= 0; i
< num_storage
; i
++) {
40 if (strcmp(name
, storage
[i
].name
) == 0)
48 get_uniform_block_index(const gl_shader_program
*shProg
,
49 const char *uniformBlockName
)
51 for (unsigned i
= 0; i
< shProg
->NumUniformBlocks
; i
++) {
52 if (!strcmp(shProg
->UniformBlocks
[i
].Name
, uniformBlockName
))
56 return GL_INVALID_INDEX
;
60 copy_constant_to_storage(union gl_constant_value
*storage
,
61 const ir_constant
*val
,
62 const enum glsl_base_type base_type
,
63 const unsigned int elements
,
64 unsigned int boolean_true
)
66 for (unsigned int i
= 0; i
< elements
; i
++) {
69 storage
[i
].u
= val
->value
.u
[i
];
72 case GLSL_TYPE_SAMPLER
:
73 storage
[i
].i
= val
->value
.i
[i
];
76 storage
[i
].f
= val
->value
.f
[i
];
78 case GLSL_TYPE_DOUBLE
:
79 /* XXX need to check on big-endian */
80 storage
[i
* 2].u
= *(uint32_t *)&val
->value
.d
[i
];
81 storage
[i
* 2 + 1].u
= *(((uint32_t *)&val
->value
.d
[i
]) + 1);
84 storage
[i
].b
= val
->value
.b
[i
] ? boolean_true
: 0;
87 case GLSL_TYPE_STRUCT
:
89 case GLSL_TYPE_ATOMIC_UINT
:
90 case GLSL_TYPE_INTERFACE
:
91 case GLSL_TYPE_FUNCTION
:
94 /* All other types should have already been filtered by other
95 * paths in the caller.
97 assert(!"Should not get here.");
104 set_sampler_binding(gl_shader_program
*prog
, const char *name
, int binding
)
106 struct gl_uniform_storage
*const storage
=
107 get_storage(prog
->UniformStorage
, prog
->NumUniformStorage
, name
);
109 if (storage
== NULL
) {
110 assert(storage
!= NULL
);
114 const unsigned elements
= MAX2(storage
->array_elements
, 1);
116 /* Section 4.4.4 (Opaque-Uniform Layout Qualifiers) of the GLSL 4.20 spec
119 * "If the binding identifier is used with an array, the first element
120 * of the array takes the specified unit and each subsequent element
121 * takes the next consecutive unit."
123 for (unsigned int i
= 0; i
< elements
; i
++) {
124 storage
->storage
[i
].i
= binding
+ i
;
127 for (int sh
= 0; sh
< MESA_SHADER_STAGES
; sh
++) {
128 gl_shader
*shader
= prog
->_LinkedShaders
[sh
];
130 if (shader
&& storage
->sampler
[sh
].active
) {
131 for (unsigned i
= 0; i
< elements
; i
++) {
132 unsigned index
= storage
->sampler
[sh
].index
+ i
;
134 shader
->SamplerUnits
[index
] = storage
->storage
[i
].i
;
139 storage
->initialized
= true;
143 set_block_binding(gl_shader_program
*prog
, const char *block_name
, int binding
)
145 const unsigned block_index
= get_uniform_block_index(prog
, block_name
);
147 if (block_index
== GL_INVALID_INDEX
) {
148 assert(block_index
!= GL_INVALID_INDEX
);
152 /* This is a field of a UBO. val is the binding index. */
153 for (int i
= 0; i
< MESA_SHADER_STAGES
; i
++) {
154 int stage_index
= prog
->UniformBlockStageIndex
[i
][block_index
];
156 if (stage_index
!= -1) {
157 struct gl_shader
*sh
= prog
->_LinkedShaders
[i
];
158 sh
->UniformBlocks
[stage_index
].Binding
= binding
;
164 set_uniform_initializer(void *mem_ctx
, gl_shader_program
*prog
,
165 const char *name
, const glsl_type
*type
,
166 ir_constant
*val
, unsigned int boolean_true
)
168 if (type
->is_record()) {
169 ir_constant
*field_constant
;
171 field_constant
= (ir_constant
*)val
->components
.get_head();
173 for (unsigned int i
= 0; i
< type
->length
; i
++) {
174 const glsl_type
*field_type
= type
->fields
.structure
[i
].type
;
175 const char *field_name
= ralloc_asprintf(mem_ctx
, "%s.%s", name
,
176 type
->fields
.structure
[i
].name
);
177 set_uniform_initializer(mem_ctx
, prog
, field_name
,
178 field_type
, field_constant
, boolean_true
);
179 field_constant
= (ir_constant
*)field_constant
->next
;
182 } else if (type
->is_array() && type
->fields
.array
->is_record()) {
183 const glsl_type
*const element_type
= type
->fields
.array
;
185 for (unsigned int i
= 0; i
< type
->length
; i
++) {
186 const char *element_name
= ralloc_asprintf(mem_ctx
, "%s[%d]", name
, i
);
188 set_uniform_initializer(mem_ctx
, prog
, element_name
,
189 element_type
, val
->array_elements
[i
],
195 struct gl_uniform_storage
*const storage
=
196 get_storage(prog
->UniformStorage
,
197 prog
->NumUniformStorage
,
199 if (storage
== NULL
) {
200 assert(storage
!= NULL
);
204 if (val
->type
->is_array()) {
205 const enum glsl_base_type base_type
=
206 val
->array_elements
[0]->type
->base_type
;
207 const unsigned int elements
= val
->array_elements
[0]->type
->components();
208 unsigned int idx
= 0;
209 unsigned dmul
= (base_type
== GLSL_TYPE_DOUBLE
) ? 2 : 1;
211 assert(val
->type
->length
>= storage
->array_elements
);
212 for (unsigned int i
= 0; i
< storage
->array_elements
; i
++) {
213 copy_constant_to_storage(& storage
->storage
[idx
],
214 val
->array_elements
[i
],
219 idx
+= elements
* dmul
;
222 copy_constant_to_storage(storage
->storage
,
224 val
->type
->base_type
,
225 val
->type
->components(),
228 if (storage
->type
->is_sampler()) {
229 for (int sh
= 0; sh
< MESA_SHADER_STAGES
; sh
++) {
230 gl_shader
*shader
= prog
->_LinkedShaders
[sh
];
232 if (shader
&& storage
->sampler
[sh
].active
) {
233 unsigned index
= storage
->sampler
[sh
].index
;
235 shader
->SamplerUnits
[index
] = storage
->storage
[0].i
;
241 storage
->initialized
= true;
246 link_set_uniform_initializers(struct gl_shader_program
*prog
,
247 unsigned int boolean_true
)
249 void *mem_ctx
= NULL
;
251 for (unsigned int i
= 0; i
< MESA_SHADER_STAGES
; i
++) {
252 struct gl_shader
*shader
= prog
->_LinkedShaders
[i
];
257 foreach_in_list(ir_instruction
, node
, shader
->ir
) {
258 ir_variable
*const var
= node
->as_variable();
260 if (!var
|| var
->data
.mode
!= ir_var_uniform
)
264 mem_ctx
= ralloc_context(NULL
);
266 if (var
->data
.explicit_binding
) {
267 const glsl_type
*const type
= var
->type
;
269 if (type
->without_array()->is_sampler()) {
270 linker::set_sampler_binding(prog
, var
->name
, var
->data
.binding
);
271 } else if (var
->is_in_uniform_block()) {
272 const glsl_type
*const iface_type
= var
->get_interface_type();
274 /* If the variable is an array and it is an interface instance,
275 * we need to set the binding for each array element. Just
276 * checking that the variable is an array is not sufficient.
277 * The variable could be an array element of a uniform block
278 * that lacks an instance name. For example:
284 * In this case "f" would pass is_in_uniform_block (above) and
285 * type->is_array(), but it will fail is_interface_instance().
287 if (var
->is_interface_instance() && var
->type
->is_array()) {
288 for (unsigned i
= 0; i
< var
->type
->length
; i
++) {
290 ralloc_asprintf(mem_ctx
, "%s[%u]", iface_type
->name
, i
);
292 /* Section 4.4.3 (Uniform Block Layout Qualifiers) of the
293 * GLSL 4.20 spec says:
295 * "If the binding identifier is used with a uniform
296 * block instanced as an array then the first element
297 * of the array takes the specified block binding and
298 * each subsequent element takes the next consecutive
299 * uniform block binding point."
301 linker::set_block_binding(prog
, name
,
302 var
->data
.binding
+ i
);
305 linker::set_block_binding(prog
, iface_type
->name
,
308 } else if (type
->contains_atomic()) {
309 /* we don't actually need to do anything. */
311 assert(!"Explicit binding not on a sampler, UBO or atomic.");
313 } else if (var
->constant_value
) {
314 linker::set_uniform_initializer(mem_ctx
, prog
, var
->name
,
315 var
->type
, var
->constant_value
,
321 ralloc_free(mem_ctx
);