glsl_types.cpp revision bfd7c9ac228c7ed8aec04c3b3aa33f40ee00b035
1/* 2 * Copyright © 2009 Intel Corporation 3 * 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: 10 * 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 13 * Software. 14 * 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. 22 */ 23 24#include <cstdio> 25#include <stdlib.h> 26#include "main/core.h" /* for Elements */ 27#include "glsl_symbol_table.h" 28#include "glsl_parser_extras.h" 29#include "glsl_types.h" 30#include "builtin_types.h" 31extern "C" { 32#include "program/hash_table.h" 33} 34 35hash_table *glsl_type::array_types = NULL; 36hash_table *glsl_type::record_types = NULL; 37void *glsl_type::mem_ctx = NULL; 38 39void 40glsl_type::init_talloc_type_ctx(void) 41{ 42 if (glsl_type::mem_ctx == NULL) { 43 glsl_type::mem_ctx = talloc_autofree_context(); 44 assert(glsl_type::mem_ctx != NULL); 45 } 46} 47 48glsl_type::glsl_type(GLenum gl_type, 49 unsigned base_type, unsigned vector_elements, 50 unsigned matrix_columns, const char *name) : 51 gl_type(gl_type), 52 base_type(base_type), 53 sampler_dimensionality(0), sampler_shadow(0), sampler_array(0), 54 sampler_type(0), 55 vector_elements(vector_elements), matrix_columns(matrix_columns), 56 length(0) 57{ 58 init_talloc_type_ctx(); 59 this->name = talloc_strdup(this->mem_ctx, name); 60 /* Neither dimension is zero or both dimensions are zero. 61 */ 62 assert((vector_elements == 0) == (matrix_columns == 0)); 63 memset(& fields, 0, sizeof(fields)); 64} 65 66glsl_type::glsl_type(GLenum gl_type, 67 enum glsl_sampler_dim dim, bool shadow, bool array, 68 unsigned type, const char *name) : 69 gl_type(gl_type), 70 base_type(GLSL_TYPE_SAMPLER), 71 sampler_dimensionality(dim), sampler_shadow(shadow), 72 sampler_array(array), sampler_type(type), 73 vector_elements(0), matrix_columns(0), 74 length(0) 75{ 76 init_talloc_type_ctx(); 77 this->name = talloc_strdup(this->mem_ctx, name); 78 memset(& fields, 0, sizeof(fields)); 79} 80 81glsl_type::glsl_type(const glsl_struct_field *fields, unsigned num_fields, 82 const char *name) : 83 base_type(GLSL_TYPE_STRUCT), 84 sampler_dimensionality(0), sampler_shadow(0), sampler_array(0), 85 sampler_type(0), 86 vector_elements(0), matrix_columns(0), 87 length(num_fields) 88{ 89 unsigned int i; 90 91 init_talloc_type_ctx(); 92 this->name = talloc_strdup(this->mem_ctx, name); 93 this->fields.structure = talloc_array(this->mem_ctx, 94 glsl_struct_field, length); 95 for (i = 0; i < length; i++) { 96 this->fields.structure[i].type = fields[i].type; 97 this->fields.structure[i].name = talloc_strdup(this->fields.structure, 98 fields[i].name); 99 } 100} 101 102static void 103add_types_to_symbol_table(glsl_symbol_table *symtab, 104 const struct glsl_type *types, 105 unsigned num_types, bool warn) 106{ 107 (void) warn; 108 109 for (unsigned i = 0; i < num_types; i++) { 110 symtab->add_type(types[i].name, & types[i]); 111 } 112} 113 114 115void 116glsl_type::generate_110_types(glsl_symbol_table *symtab) 117{ 118 add_types_to_symbol_table(symtab, builtin_core_types, 119 Elements(builtin_core_types), 120 false); 121 add_types_to_symbol_table(symtab, builtin_structure_types, 122 Elements(builtin_structure_types), 123 false); 124 add_types_to_symbol_table(symtab, builtin_110_deprecated_structure_types, 125 Elements(builtin_110_deprecated_structure_types), 126 false); 127 add_types_to_symbol_table(symtab, & void_type, 1, false); 128} 129 130 131void 132glsl_type::generate_120_types(glsl_symbol_table *symtab) 133{ 134 generate_110_types(symtab); 135 136 add_types_to_symbol_table(symtab, builtin_120_types, 137 Elements(builtin_120_types), false); 138} 139 140 141void 142glsl_type::generate_130_types(glsl_symbol_table *symtab) 143{ 144 generate_120_types(symtab); 145 146 add_types_to_symbol_table(symtab, builtin_130_types, 147 Elements(builtin_130_types), false); 148 generate_EXT_texture_array_types(symtab, false); 149} 150 151 152void 153glsl_type::generate_ARB_texture_rectangle_types(glsl_symbol_table *symtab, 154 bool warn) 155{ 156 add_types_to_symbol_table(symtab, builtin_ARB_texture_rectangle_types, 157 Elements(builtin_ARB_texture_rectangle_types), 158 warn); 159} 160 161 162void 163glsl_type::generate_EXT_texture_array_types(glsl_symbol_table *symtab, 164 bool warn) 165{ 166 add_types_to_symbol_table(symtab, builtin_EXT_texture_array_types, 167 Elements(builtin_EXT_texture_array_types), 168 warn); 169} 170 171 172void 173_mesa_glsl_initialize_types(struct _mesa_glsl_parse_state *state) 174{ 175 switch (state->language_version) { 176 case 110: 177 glsl_type::generate_110_types(state->symbols); 178 break; 179 case 120: 180 glsl_type::generate_120_types(state->symbols); 181 break; 182 case 130: 183 glsl_type::generate_130_types(state->symbols); 184 break; 185 default: 186 /* error */ 187 break; 188 } 189 190 if (state->ARB_texture_rectangle_enable) { 191 glsl_type::generate_ARB_texture_rectangle_types(state->symbols, 192 state->ARB_texture_rectangle_warn); 193 } 194 195 if (state->EXT_texture_array_enable && state->language_version < 130) { 196 // These are already included in 130; don't create twice. 197 glsl_type::generate_EXT_texture_array_types(state->symbols, 198 state->EXT_texture_array_warn); 199 } 200} 201 202 203const glsl_type *glsl_type::get_base_type() const 204{ 205 switch (base_type) { 206 case GLSL_TYPE_UINT: 207 return uint_type; 208 case GLSL_TYPE_INT: 209 return int_type; 210 case GLSL_TYPE_FLOAT: 211 return float_type; 212 case GLSL_TYPE_BOOL: 213 return bool_type; 214 default: 215 return error_type; 216 } 217} 218 219 220void 221_mesa_glsl_release_types(void) 222{ 223 if (glsl_type::array_types != NULL) { 224 hash_table_dtor(glsl_type::array_types); 225 glsl_type::array_types = NULL; 226 } 227 228 if (glsl_type::record_types != NULL) { 229 hash_table_dtor(glsl_type::record_types); 230 glsl_type::record_types = NULL; 231 } 232} 233 234 235ir_function * 236glsl_type::generate_constructor(glsl_symbol_table *symtab) const 237{ 238 void *ctx = symtab; 239 240 /* Generate the function name and add it to the symbol table. 241 */ 242 ir_function *const f = new(ctx) ir_function(name); 243 244 bool added = symtab->add_function(name, f); 245 assert(added); 246 247 ir_function_signature *const sig = new(ctx) ir_function_signature(this); 248 f->add_signature(sig); 249 250 ir_variable **declarations = 251 (ir_variable **) malloc(sizeof(ir_variable *) * this->length); 252 for (unsigned i = 0; i < length; i++) { 253 char *const param_name = (char *) malloc(10); 254 255 snprintf(param_name, 10, "p%08X", i); 256 257 ir_variable *var = (this->base_type == GLSL_TYPE_ARRAY) 258 ? new(ctx) ir_variable(fields.array, param_name, ir_var_in) 259 : new(ctx) ir_variable(fields.structure[i].type, param_name, ir_var_in); 260 261 declarations[i] = var; 262 sig->parameters.push_tail(var); 263 } 264 265 /* Generate the body of the constructor. The body assigns each of the 266 * parameters to a portion of a local variable called _ret_val that has 267 * the same type as the constructor. After initializing _ret_val, 268 * _ret_val is returned. 269 */ 270 ir_variable *retval = new(ctx) ir_variable(this, "_ret_val", ir_var_auto); 271 sig->body.push_tail(retval); 272 273 for (unsigned i = 0; i < length; i++) { 274 ir_dereference *const lhs = (this->base_type == GLSL_TYPE_ARRAY) 275 ? (ir_dereference *) new(ctx) ir_dereference_array(retval, 276 new(ctx) ir_constant(i)) 277 : (ir_dereference *) new(ctx) ir_dereference_record(retval, 278 fields.structure[i].name); 279 280 ir_dereference *const rhs = new(ctx) ir_dereference_variable(declarations[i]); 281 ir_instruction *const assign = new(ctx) ir_assignment(lhs, rhs, NULL); 282 283 sig->body.push_tail(assign); 284 } 285 286 free(declarations); 287 288 ir_dereference *const retref = new(ctx) ir_dereference_variable(retval); 289 ir_instruction *const inst = new(ctx) ir_return(retref); 290 sig->body.push_tail(inst); 291 292 return f; 293} 294 295 296glsl_type::glsl_type(const glsl_type *array, unsigned length) : 297 base_type(GLSL_TYPE_ARRAY), 298 sampler_dimensionality(0), sampler_shadow(0), sampler_array(0), 299 sampler_type(0), 300 vector_elements(0), matrix_columns(0), 301 name(NULL), length(length) 302{ 303 this->fields.array = array; 304 /* Inherit the gl type of the base. The GL type is used for 305 * uniform/statevar handling in Mesa and the arrayness of the type 306 * is represented by the size rather than the type. 307 */ 308 this->gl_type = array->gl_type; 309 310 /* Allow a maximum of 10 characters for the array size. This is enough 311 * for 32-bits of ~0. The extra 3 are for the '[', ']', and terminating 312 * NUL. 313 */ 314 const unsigned name_length = strlen(array->name) + 10 + 3; 315 char *const n = (char *) talloc_size(this->mem_ctx, name_length); 316 317 if (length == 0) 318 snprintf(n, name_length, "%s[]", array->name); 319 else 320 snprintf(n, name_length, "%s[%u]", array->name, length); 321 322 this->name = n; 323} 324 325 326const glsl_type * 327glsl_type::get_instance(unsigned base_type, unsigned rows, unsigned columns) 328{ 329 if (base_type == GLSL_TYPE_VOID) 330 return &void_type; 331 332 if ((rows < 1) || (rows > 4) || (columns < 1) || (columns > 4)) 333 return error_type; 334 335 /* Treat GLSL vectors as Nx1 matrices. 336 */ 337 if (columns == 1) { 338 switch (base_type) { 339 case GLSL_TYPE_UINT: 340 return uint_type + (rows - 1); 341 case GLSL_TYPE_INT: 342 return int_type + (rows - 1); 343 case GLSL_TYPE_FLOAT: 344 return float_type + (rows - 1); 345 case GLSL_TYPE_BOOL: 346 return bool_type + (rows - 1); 347 default: 348 return error_type; 349 } 350 } else { 351 if ((base_type != GLSL_TYPE_FLOAT) || (rows == 1)) 352 return error_type; 353 354 /* GLSL matrix types are named mat{COLUMNS}x{ROWS}. Only the following 355 * combinations are valid: 356 * 357 * 1 2 3 4 358 * 1 359 * 2 x x x 360 * 3 x x x 361 * 4 x x x 362 */ 363#define IDX(c,r) (((c-1)*3) + (r-1)) 364 365 switch (IDX(columns, rows)) { 366 case IDX(2,2): return mat2_type; 367 case IDX(2,3): return mat2x3_type; 368 case IDX(2,4): return mat2x4_type; 369 case IDX(3,2): return mat3x2_type; 370 case IDX(3,3): return mat3_type; 371 case IDX(3,4): return mat3x4_type; 372 case IDX(4,2): return mat4x2_type; 373 case IDX(4,3): return mat4x3_type; 374 case IDX(4,4): return mat4_type; 375 default: return error_type; 376 } 377 } 378 379 assert(!"Should not get here."); 380 return error_type; 381} 382 383 384const glsl_type * 385glsl_type::get_array_instance(const glsl_type *base, unsigned array_size) 386{ 387 388 if (array_types == NULL) { 389 array_types = hash_table_ctor(64, hash_table_string_hash, 390 hash_table_string_compare); 391 } 392 393 /* Generate a name using the base type pointer in the key. This is 394 * done because the name of the base type may not be unique across 395 * shaders. For example, two shaders may have different record types 396 * named 'foo'. 397 */ 398 char key[128]; 399 snprintf(key, sizeof(key), "%p[%u]", (void *) base, array_size); 400 401 const glsl_type *t = (glsl_type *) hash_table_find(array_types, key); 402 if (t == NULL) { 403 t = new glsl_type(base, array_size); 404 405 hash_table_insert(array_types, (void *) t, talloc_strdup(mem_ctx, key)); 406 } 407 408 assert(t->base_type == GLSL_TYPE_ARRAY); 409 assert(t->length == array_size); 410 assert(t->fields.array == base); 411 412 return t; 413} 414 415 416int 417glsl_type::record_key_compare(const void *a, const void *b) 418{ 419 const glsl_type *const key1 = (glsl_type *) a; 420 const glsl_type *const key2 = (glsl_type *) b; 421 422 /* Return zero is the types match (there is zero difference) or non-zero 423 * otherwise. 424 */ 425 if (strcmp(key1->name, key2->name) != 0) 426 return 1; 427 428 if (key1->length != key2->length) 429 return 1; 430 431 for (unsigned i = 0; i < key1->length; i++) { 432 if (key1->fields.structure[i].type != key2->fields.structure[i].type) 433 return 1; 434 if (strcmp(key1->fields.structure[i].name, 435 key2->fields.structure[i].name) != 0) 436 return 1; 437 } 438 439 return 0; 440} 441 442 443unsigned 444glsl_type::record_key_hash(const void *a) 445{ 446 const glsl_type *const key = (glsl_type *) a; 447 char hash_key[128]; 448 unsigned size = 0; 449 450 size = snprintf(hash_key, sizeof(hash_key), "%08x", key->length); 451 452 for (unsigned i = 0; i < key->length; i++) { 453 if (size >= sizeof(hash_key)) 454 break; 455 456 size += snprintf(& hash_key[size], sizeof(hash_key) - size, 457 "%p", (void *) key->fields.structure[i].type); 458 } 459 460 return hash_table_string_hash(& hash_key); 461} 462 463 464const glsl_type * 465glsl_type::get_record_instance(const glsl_struct_field *fields, 466 unsigned num_fields, 467 const char *name) 468{ 469 const glsl_type key(fields, num_fields, name); 470 471 if (record_types == NULL) { 472 record_types = hash_table_ctor(64, record_key_hash, record_key_compare); 473 } 474 475 const glsl_type *t = (glsl_type *) hash_table_find(record_types, & key); 476 if (t == NULL) { 477 t = new glsl_type(fields, num_fields, name); 478 479 hash_table_insert(record_types, (void *) t, t); 480 } 481 482 assert(t->base_type == GLSL_TYPE_STRUCT); 483 assert(t->length == num_fields); 484 assert(strcmp(t->name, name) == 0); 485 486 return t; 487} 488 489 490const glsl_type * 491glsl_type::field_type(const char *name) const 492{ 493 if (this->base_type != GLSL_TYPE_STRUCT) 494 return error_type; 495 496 for (unsigned i = 0; i < this->length; i++) { 497 if (strcmp(name, this->fields.structure[i].name) == 0) 498 return this->fields.structure[i].type; 499 } 500 501 return error_type; 502} 503 504 505int 506glsl_type::field_index(const char *name) const 507{ 508 if (this->base_type != GLSL_TYPE_STRUCT) 509 return -1; 510 511 for (unsigned i = 0; i < this->length; i++) { 512 if (strcmp(name, this->fields.structure[i].name) == 0) 513 return i; 514 } 515 516 return -1; 517} 518 519 520unsigned 521glsl_type::component_slots() const 522{ 523 switch (this->base_type) { 524 case GLSL_TYPE_UINT: 525 case GLSL_TYPE_INT: 526 case GLSL_TYPE_FLOAT: 527 case GLSL_TYPE_BOOL: 528 return this->components(); 529 530 case GLSL_TYPE_STRUCT: { 531 unsigned size = 0; 532 533 for (unsigned i = 0; i < this->length; i++) 534 size += this->fields.structure[i].type->component_slots(); 535 536 return size; 537 } 538 539 case GLSL_TYPE_ARRAY: 540 return this->length * this->fields.array->component_slots(); 541 542 default: 543 return 0; 544 } 545} 546