1a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick/* 2a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * Copyright © 2012 Intel Corporation 3a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * 4a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * Permission is hereby granted, free of charge, to any person obtaining a 5a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * copy of this software and associated documentation files (the "Software"), 6a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * to deal in the Software without restriction, including without limitation 7a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * the rights to use, copy, modify, merge, publish, distribute, sublicense, 8a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * and/or sell copies of the Software, and to permit persons to whom the 9a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * Software is furnished to do so, subject to the following conditions: 10a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * 11a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * The above copyright notice and this permission notice (including the next 12a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * paragraph) shall be included in all copies or substantial portions of the 13a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * Software. 14a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * 15a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 18a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 19a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 20a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER 21a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * DEALINGS IN THE SOFTWARE. 22a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick */ 23a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 24a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick#include "main/core.h" 25a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick#include "ir.h" 26a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick#include "linker.h" 27a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick#include "ir_uniform.h" 28a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick#include "glsl_symbol_table.h" 29a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick#include "program/hash_table.h" 30a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 31a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick/* These functions are put in a "private" namespace instead of being marked 32a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * static so that the unit tests can access them. See 33a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * http://code.google.com/p/googletest/wiki/AdvancedGuide#Testing_Private_Code 34a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick */ 35a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanicknamespace linker { 36a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 37a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanickgl_uniform_storage * 38a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanickget_storage(gl_uniform_storage *storage, unsigned num_storage, 39a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick const char *name) 40a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick{ 41a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick for (unsigned int i = 0; i < num_storage; i++) { 42a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick if (strcmp(name, storage[i].name) == 0) 43a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick return &storage[i]; 44a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } 45a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 46a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick return NULL; 47a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick} 48a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 49a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanickvoid 50a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanickcopy_constant_to_storage(union gl_constant_value *storage, 51a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick const ir_constant *val, 52a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick const enum glsl_base_type base_type, 53a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick const unsigned int elements) 54a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick{ 55a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick for (unsigned int i = 0; i < elements; i++) { 56a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick switch (base_type) { 57a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick case GLSL_TYPE_UINT: 58a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick storage[i].u = val->value.u[i]; 59a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick break; 60a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick case GLSL_TYPE_INT: 61a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick case GLSL_TYPE_SAMPLER: 62a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick storage[i].i = val->value.i[i]; 63a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick break; 64a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick case GLSL_TYPE_FLOAT: 65a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick storage[i].f = val->value.f[i]; 66a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick break; 67a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick case GLSL_TYPE_BOOL: 68a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick storage[i].b = int(val->value.b[i]); 69a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick break; 70a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick default: 71a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick /* All other types should have already been filtered by other 72a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick * paths in the caller. 73a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick */ 74a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick assert(!"Should not get here."); 75a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick break; 76a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } 77a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } 78a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick} 79a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 80a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanickvoid 81a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanickset_uniform_initializer(void *mem_ctx, gl_shader_program *prog, 82a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick const char *name, const glsl_type *type, 83a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick ir_constant *val) 84a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick{ 85a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick if (type->is_record()) { 86a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick ir_constant *field_constant; 87a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 88a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick field_constant = (ir_constant *)val->components.get_head(); 89a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 90a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick for (unsigned int i = 0; i < type->length; i++) { 91a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick const glsl_type *field_type = type->fields.structure[i].type; 92a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick const char *field_name = ralloc_asprintf(mem_ctx, "%s.%s", name, 93a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick type->fields.structure[i].name); 94a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick set_uniform_initializer(mem_ctx, prog, field_name, 95a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick field_type, field_constant); 96a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick field_constant = (ir_constant *)field_constant->next; 97a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } 98a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick return; 99a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } else if (type->is_array() && type->fields.array->is_record()) { 100a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick const glsl_type *const element_type = type->fields.array; 101a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 102a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick for (unsigned int i = 0; i < type->length; i++) { 103a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick const char *element_name = ralloc_asprintf(mem_ctx, "%s[%d]", name, i); 104a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 105a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick set_uniform_initializer(mem_ctx, prog, element_name, 106a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick element_type, val->array_elements[i]); 107a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } 108a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick return; 109a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } 110a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 111a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick struct gl_uniform_storage *const storage = 112a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick get_storage(prog->UniformStorage, 113a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick prog->NumUserUniformStorage, 114a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick name); 115a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick if (storage == NULL) { 116a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick assert(storage != NULL); 117a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick return; 118a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } 119a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 120a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick if (val->type->is_array()) { 121a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick const enum glsl_base_type base_type = 122a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick val->array_elements[0]->type->base_type; 123a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick const unsigned int elements = val->array_elements[0]->type->components(); 124a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick unsigned int idx = 0; 125a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 126a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick assert(val->type->length >= storage->array_elements); 127a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick for (unsigned int i = 0; i < storage->array_elements; i++) { 128a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick copy_constant_to_storage(& storage->storage[idx], 129a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick val->array_elements[i], 130a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick base_type, 131a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick elements); 132a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 133a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick idx += elements; 134a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } 13576027f5b5cd8017f8455e2df375fcea7cc888dceIan Romanick 13676027f5b5cd8017f8455e2df375fcea7cc888dceIan Romanick if (base_type == GLSL_TYPE_SAMPLER) { 13776027f5b5cd8017f8455e2df375fcea7cc888dceIan Romanick for (unsigned int i = 0; i < storage->array_elements; i++) { 13876027f5b5cd8017f8455e2df375fcea7cc888dceIan Romanick prog->SamplerUnits[storage->sampler + i] = storage->storage[i].i; 13976027f5b5cd8017f8455e2df375fcea7cc888dceIan Romanick } 14076027f5b5cd8017f8455e2df375fcea7cc888dceIan Romanick } 141a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } else { 142a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick copy_constant_to_storage(storage->storage, 143a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick val, 144a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick val->type->base_type, 145a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick val->type->components()); 14676027f5b5cd8017f8455e2df375fcea7cc888dceIan Romanick 14776027f5b5cd8017f8455e2df375fcea7cc888dceIan Romanick if (storage->type->is_sampler()) 14876027f5b5cd8017f8455e2df375fcea7cc888dceIan Romanick prog->SamplerUnits[storage->sampler] = storage->storage[0].i; 149a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } 150a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 151a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick storage->initialized = true; 152a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick} 153a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick} 154a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 155a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanickvoid 156a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanicklink_set_uniform_initializers(struct gl_shader_program *prog) 157a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick{ 158a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick void *mem_ctx = NULL; 159a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 160a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick for (unsigned int i = 0; i < MESA_SHADER_TYPES; i++) { 161a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick struct gl_shader *shader = prog->_LinkedShaders[i]; 162a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 163a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick if (shader == NULL) 164a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick continue; 165a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 166a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick foreach_list(node, shader->ir) { 167a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick ir_variable *const var = ((ir_instruction *) node)->as_variable(); 168a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 169a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick if (!var || var->mode != ir_var_uniform || !var->constant_value) 170a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick continue; 171a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 172a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick if (!mem_ctx) 173a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick mem_ctx = ralloc_context(NULL); 174a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 175a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick linker::set_uniform_initializer(mem_ctx, prog, var->name, 176a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick var->type, var->constant_value); 177a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } 178a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick } 179a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick 180a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick ralloc_free(mem_ctx); 181a2e623054b5d8eafebc7499efe93e4bceae16eadIan Romanick} 182