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