src/compiler/nir_types.cpp

   1 /*
   2  * Copyright © 2014 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 DEALINGS
  21  * IN THE SOFTWARE.
  22  *
  23  * Authors:
  24  *    Connor Abbott (cwabbott0@gmail.com)
  25  *
  26  */
  27
  28 #include "nir_types.h"
  29 #include "compiler/glsl/ir.h"
  30
  31 const char *
  32 glsl_get_type_name(const glsl_type *type)
  33 {
  34    return type->name;
  35 }
  36
  37 const glsl_type *
  38 glsl_get_array_element(const glsl_type* type)
  39 {
  40    if (type->is_matrix())
  41       return type->column_type();
  42    else if (type->is_vector())
  43       return type->get_scalar_type();
  44    return type->fields.array;
  45 }
  46
  47 const glsl_type *
  48 glsl_without_array(const glsl_type *type)
  49 {
  50    return type->without_array();
  51 }
  52
  53 const glsl_type *
  54 glsl_without_array_or_matrix(const glsl_type *type)
  55 {
  56    type = type->without_array();
  57    if (type->is_matrix())
  58       type = type->column_type();
  59    return type;
  60 }
  61
  62 const glsl_type *
  63 glsl_get_bare_type(const glsl_type *type)
  64 {
  65    return type->get_bare_type();
  66 }
  67
  68 const glsl_type *
  69 glsl_get_struct_field(const glsl_type *type, unsigned index)
  70 {
  71    return type->fields.structure[index].type;
  72 }
  73
  74 int
  75 glsl_get_struct_field_offset(const struct glsl_type *type,
  76                              unsigned index)
  77 {
  78    return type->fields.structure[index].offset;
  79 }
  80
  81 unsigned
  82 glsl_get_explicit_stride(const struct glsl_type *type)
  83 {
  84    return type->explicit_stride;
  85 }
  86
  87 const glsl_type *
  88 glsl_get_function_return_type(const glsl_type *type)
  89 {
  90    return type->fields.parameters[0].type;
  91 }
  92
  93 const glsl_function_param *
  94 glsl_get_function_param(const glsl_type *type, unsigned index)
  95 {
  96    return &type->fields.parameters[index + 1];
  97 }
  98
  99 const struct glsl_type *
 100 glsl_get_column_type(const struct glsl_type *type)
 101 {
 102    return type->column_type();
 103 }
 104
 105 GLenum
 106 glsl_get_gl_type(const struct glsl_type *type)
 107 {
 108    return type->gl_type;
 109 }
 110
 111 enum glsl_base_type
 112 glsl_get_base_type(const struct glsl_type *type)
 113 {
 114    return type->base_type;
 115 }
 116
 117 unsigned
 118 glsl_get_vector_elements(const struct glsl_type *type)
 119 {
 120    return type->vector_elements;
 121 }
 122
 123 unsigned
 124 glsl_get_components(const struct glsl_type *type)
 125 {
 126    return type->components();
 127 }
 128
 129 unsigned
 130 glsl_get_matrix_columns(const struct glsl_type *type)
 131 {
 132    return type->matrix_columns;
 133 }
 134
 135 unsigned
 136 glsl_get_length(const struct glsl_type *type)
 137 {
 138    return type->is_matrix() ? type->matrix_columns : type->length;
 139 }
 140
 141 unsigned
 142 glsl_get_aoa_size(const struct glsl_type *type)
 143 {
 144    return type->arrays_of_arrays_size();
 145 }
 146
 147 unsigned
 148 glsl_count_attribute_slots(const struct glsl_type *type,
 149                            bool is_gl_vertex_input)
 150 {
 151    return type->count_attribute_slots(is_gl_vertex_input);
 152 }
 153
 154 unsigned
 155 glsl_get_component_slots(const struct glsl_type *type)
 156 {
 157    return type->component_slots();
 158 }
 159
 160 unsigned
 161 glsl_varying_count(const struct glsl_type *type)
 162 {
 163    return type->varying_count();
 164 }
 165
 166 const char *
 167 glsl_get_struct_elem_name(const struct glsl_type *type, unsigned index)
 168 {
 169    return type->fields.structure[index].name;
 170 }
 171
 172 glsl_sampler_dim
 173 glsl_get_sampler_dim(const struct glsl_type *type)
 174 {
 175    assert(glsl_type_is_sampler(type) || glsl_type_is_image(type));
 176    return (glsl_sampler_dim)type->sampler_dimensionality;
 177 }
 178
 179 glsl_base_type
 180 glsl_get_sampler_result_type(const struct glsl_type *type)
 181 {
 182    assert(glsl_type_is_sampler(type) || glsl_type_is_image(type));
 183    return (glsl_base_type)type->sampled_type;
 184 }
 185
 186 unsigned
 187 glsl_get_sampler_target(const struct glsl_type *type)
 188 {
 189    assert(glsl_type_is_sampler(type));
 190    return type->sampler_index();
 191 }
 192
 193 int
 194 glsl_get_sampler_coordinate_components(const struct glsl_type *type)
 195 {
 196    assert(glsl_type_is_sampler(type) || glsl_type_is_image(type));
 197    return type->coordinate_components();
 198 }
 199
 200 unsigned
 201 glsl_get_struct_location_offset(const struct glsl_type *type,
 202                                 unsigned length)
 203 {
 204    return type->struct_location_offset(length);
 205 }
 206
 207 bool
 208 glsl_type_is_16bit(const glsl_type *type)
 209 {
 210    return type->is_16bit();
 211 }
 212
 213 bool
 214 glsl_type_is_32bit(const glsl_type *type)
 215 {
 216    return type->is_32bit();
 217 }
 218
 219 bool
 220 glsl_type_is_64bit(const glsl_type *type)
 221 {
 222    return type->is_64bit();
 223 }
 224
 225 bool
 226 glsl_type_is_void(const glsl_type *type)
 227 {
 228    return type->is_void();
 229 }
 230
 231 bool
 232 glsl_type_is_error(const glsl_type *type)
 233 {
 234    return type->is_error();
 235 }
 236
 237 bool
 238 glsl_type_is_vector(const struct glsl_type *type)
 239 {
 240    return type->is_vector();
 241 }
 242
 243 bool
 244 glsl_type_is_scalar(const struct glsl_type *type)
 245 {
 246    return type->is_scalar();
 247 }
 248
 249 bool
 250 glsl_type_is_vector_or_scalar(const struct glsl_type *type)
 251 {
 252    return type->is_vector() || type->is_scalar();
 253 }
 254
 255 bool
 256 glsl_type_is_matrix(const struct glsl_type *type)
 257 {
 258    return type->is_matrix();
 259 }
 260
 261 bool
 262 glsl_matrix_type_is_row_major(const struct glsl_type *type)
 263 {
 264    assert(type->is_matrix() && type->explicit_stride);
 265    return type->interface_row_major;
 266 }
 267
 268 bool
 269 glsl_type_is_array(const struct glsl_type *type)
 270 {
 271    return type->is_array();
 272 }
 273
 274 bool
 275 glsl_type_is_array_of_arrays(const struct glsl_type *type)
 276 {
 277    return type->is_array_of_arrays();
 278 }
 279
 280 bool
 281 glsl_type_is_array_or_matrix(const struct glsl_type *type)
 282 {
 283    return type->is_array() || type->is_matrix();
 284 }
 285
 286 bool
 287 glsl_type_is_struct(const struct glsl_type *type)
 288 {
 289    return type->is_struct();
 290 }
 291
 292 bool
 293 glsl_type_is_interface(const struct glsl_type *type)
 294 {
 295    return type->is_interface();
 296 }
 297
 298 bool
 299 glsl_type_is_struct_or_ifc(const struct glsl_type *type)
 300 {
 301    return type->is_struct() || type->is_interface();
 302 }
 303
 304 bool
 305 glsl_type_is_sampler(const struct glsl_type *type)
 306 {
 307    return type->is_sampler();
 308 }
 309
 310 bool
 311 glsl_type_is_image(const struct glsl_type *type)
 312 {
 313    return type->is_image();
 314 }
 315
 316 bool
 317 glsl_sampler_type_is_shadow(const struct glsl_type *type)
 318 {
 319    assert(glsl_type_is_sampler(type));
 320    return type->sampler_shadow;
 321 }
 322
 323 bool
 324 glsl_sampler_type_is_array(const struct glsl_type *type)
 325 {
 326    assert(glsl_type_is_sampler(type) || glsl_type_is_image(type));
 327    return type->sampler_array;
 328 }
 329
 330 bool
 331 glsl_type_is_dual_slot(const struct glsl_type *type)
 332 {
 333    return type->is_dual_slot();
 334 }
 335
 336 bool
 337 glsl_type_is_numeric(const struct glsl_type *type)
 338 {
 339    return type->is_numeric();
 340 }
 341
 342 bool
 343 glsl_type_is_boolean(const struct glsl_type *type)
 344 {
 345    return type->is_boolean();
 346 }
 347 bool
 348 glsl_type_is_integer(const struct glsl_type *type)
 349 {
 350    return type->is_integer();
 351 }
 352
 353 bool
 354 glsl_type_contains_64bit(const struct glsl_type *type)
 355 {
 356    return type->contains_64bit();
 357 }
 358
 359 const glsl_type *
 360 glsl_void_type(void)
 361 {
 362    return glsl_type::void_type;
 363 }
 364
 365 const glsl_type *
 366 glsl_float_type(void)
 367 {
 368    return glsl_type::float_type;
 369 }
 370
 371 const glsl_type *
 372 glsl_double_type(void)
 373 {
 374    return glsl_type::double_type;
 375 }
 376
 377 const glsl_type *
 378 glsl_float16_t_type(void)
 379 {
 380    return glsl_type::float16_t_type;
 381 }
 382
 383 const glsl_type *
 384 glsl_vec_type(unsigned n)
 385 {
 386    return glsl_type::vec(n);
 387 }
 388
 389 const glsl_type *
 390 glsl_dvec_type(unsigned n)
 391 {
 392    return glsl_type::dvec(n);
 393 }
 394
 395 const glsl_type *
 396 glsl_vec4_type(void)
 397 {
 398    return glsl_type::vec4_type;
 399 }
 400
 401 const glsl_type *
 402 glsl_uvec4_type(void)
 403 {
 404    return glsl_type::uvec4_type;
 405 }
 406
 407 const glsl_type *
 408 glsl_int_type(void)
 409 {
 410    return glsl_type::int_type;
 411 }
 412
 413 const glsl_type *
 414 glsl_uint_type(void)
 415 {
 416    return glsl_type::uint_type;
 417 }
 418
 419 const glsl_type *
 420 glsl_int64_t_type(void)
 421 {
 422    return glsl_type::int64_t_type;
 423 }
 424
 425 const glsl_type *
 426 glsl_uint64_t_type(void)
 427 {
 428    return glsl_type::uint64_t_type;
 429 }
 430
 431 const glsl_type *
 432 glsl_int16_t_type(void)
 433 {
 434    return glsl_type::int16_t_type;
 435 }
 436
 437 const glsl_type *
 438 glsl_uint16_t_type(void)
 439 {
 440    return glsl_type::uint16_t_type;
 441 }
 442
 443 const glsl_type *
 444 glsl_int8_t_type(void)
 445 {
 446    return glsl_type::int8_t_type;
 447 }
 448
 449 const glsl_type *
 450 glsl_uint8_t_type(void)
 451 {
 452    return glsl_type::uint8_t_type;
 453 }
 454
 455 const glsl_type *
 456 glsl_bool_type(void)
 457 {
 458    return glsl_type::bool_type;
 459 }
 460
 461 const glsl_type *
 462 glsl_scalar_type(enum glsl_base_type base_type)
 463 {
 464    return glsl_type::get_instance(base_type, 1, 1);
 465 }
 466
 467 const glsl_type *
 468 glsl_vector_type(enum glsl_base_type base_type, unsigned components)
 469 {
 470    const glsl_type *t = glsl_type::get_instance(base_type, components, 1);
 471    assert(t != glsl_type::error_type);
 472    return t;
 473 }
 474
 475 const glsl_type *
 476 glsl_matrix_type(enum glsl_base_type base_type, unsigned rows, unsigned columns)
 477 {
 478    const glsl_type *t = glsl_type::get_instance(base_type, rows, columns);
 479    assert(t != glsl_type::error_type);
 480    return t;
 481 }
 482
 483 const glsl_type *
 484 glsl_explicit_matrix_type(const glsl_type *mat,
 485                           unsigned stride, bool row_major)
 486 {
 487    assert(stride > 0);
 488    const glsl_type *t = glsl_type::get_instance(mat->base_type,
 489                                                 mat->vector_elements,
 490                                                 mat->matrix_columns,
 491                                                 stride, row_major);
 492    assert(t != glsl_type::error_type);
 493    return t;
 494 }
 495
 496 const glsl_type *
 497 glsl_array_type(const glsl_type *base, unsigned elements,
 498                 unsigned explicit_stride)
 499 {
 500    return glsl_type::get_array_instance(base, elements, explicit_stride);
 501 }
 502
 503 const glsl_type *
 504 glsl_struct_type(const glsl_struct_field *fields,
 505                  unsigned num_fields, const char *name)
 506 {
 507    return glsl_type::get_struct_instance(fields, num_fields, name);
 508 }
 509
 510 const glsl_type *
 511 glsl_interface_type(const glsl_struct_field *fields,
 512                     unsigned num_fields,
 513                     enum glsl_interface_packing packing,
 514                     bool row_major,
 515                     const char *block_name)
 516 {
 517    return glsl_type::get_interface_instance(fields, num_fields, packing,
 518                                             row_major, block_name);
 519 }
 520
 521 const struct glsl_type *
 522 glsl_sampler_type(enum glsl_sampler_dim dim, bool is_shadow, bool is_array,
 523                   enum glsl_base_type base_type)
 524 {
 525    return glsl_type::get_sampler_instance(dim, is_shadow, is_array, base_type);
 526 }
 527
 528 const struct glsl_type *
 529 glsl_bare_sampler_type()
 530 {
 531    return glsl_type::sampler_type;
 532 }
 533
 534 const struct glsl_type *
 535 glsl_image_type(enum glsl_sampler_dim dim, bool is_array,
 536                 enum glsl_base_type base_type)
 537 {
 538    return glsl_type::get_image_instance(dim, is_array, base_type);
 539 }
 540
 541 const glsl_type *
 542 glsl_function_type(const glsl_type *return_type,
 543                    const glsl_function_param *params, unsigned num_params)
 544 {
 545    return glsl_type::get_function_instance(return_type, params, num_params);
 546 }
 547
 548 const glsl_type *
 549 glsl_transposed_type(const struct glsl_type *type)
 550 {
 551    assert(glsl_type_is_matrix(type));
 552    return glsl_type::get_instance(type->base_type, type->matrix_columns,
 553                                   type->vector_elements);
 554 }
 555
 556 const glsl_type *
 557 glsl_channel_type(const glsl_type *t)
 558 {
 559    switch (t->base_type) {
 560    case GLSL_TYPE_ARRAY:
 561       return glsl_array_type(glsl_channel_type(t->fields.array), t->length,
 562                              t->explicit_stride);
 563    case GLSL_TYPE_UINT:
 564    case GLSL_TYPE_INT:
 565    case GLSL_TYPE_FLOAT:
 566    case GLSL_TYPE_FLOAT16:
 567    case GLSL_TYPE_DOUBLE:
 568    case GLSL_TYPE_UINT8:
 569    case GLSL_TYPE_INT8:
 570    case GLSL_TYPE_UINT16:
 571    case GLSL_TYPE_INT16:
 572    case GLSL_TYPE_UINT64:
 573    case GLSL_TYPE_INT64:
 574    case GLSL_TYPE_BOOL:
 575       return glsl_type::get_instance(t->base_type, 1, 1);
 576    default:
 577       unreachable("Unhandled base type glsl_channel_type()");
 578    }
 579 }
 580
 581 void
 582 glsl_get_natural_size_align_bytes(const struct glsl_type *type,
 583                                   unsigned *size, unsigned *align)
 584 {
 585    switch (type->base_type) {
 586    case GLSL_TYPE_BOOL:
 587       /* We special-case Booleans to 32 bits to not cause heartburn for
 588        * drivers that suddenly get an 8-bit load.
 589        */
 590       *size = 4 * type->components();
 591       *align = 4;
 592       break;
 593
 594    case GLSL_TYPE_UINT8:
 595    case GLSL_TYPE_INT8:
 596    case GLSL_TYPE_UINT16:
 597    case GLSL_TYPE_INT16:
 598    case GLSL_TYPE_FLOAT16:
 599    case GLSL_TYPE_UINT:
 600    case GLSL_TYPE_INT:
 601    case GLSL_TYPE_FLOAT:
 602    case GLSL_TYPE_DOUBLE:
 603    case GLSL_TYPE_UINT64:
 604    case GLSL_TYPE_INT64: {
 605       unsigned N = glsl_get_bit_size(type) / 8;
 606       *size = N * type->components();
 607       *align = N;
 608       break;
 609    }
 610
 611    case GLSL_TYPE_ARRAY: {
 612       unsigned elem_size, elem_align;
 613       glsl_get_natural_size_align_bytes(type->fields.array,
 614                                         &elem_size, &elem_align);
 615       *align = elem_align;
 616       *size = type->length * ALIGN_POT(elem_size, elem_align);
 617       break;
 618    }
 619
 620    case GLSL_TYPE_STRUCT:
 621       *size = 0;
 622       *align = 0;
 623       for (unsigned i = 0; i < type->length; i++) {
 624          unsigned elem_size, elem_align;
 625          glsl_get_natural_size_align_bytes(type->fields.structure[i].type,
 626                                            &elem_size, &elem_align);
 627          *align = MAX2(*align, elem_align);
 628          *size = ALIGN_POT(*size, elem_align) + elem_size;
 629       }
 630       break;
 631
 632    case GLSL_TYPE_SAMPLER:
 633    case GLSL_TYPE_ATOMIC_UINT:
 634    case GLSL_TYPE_SUBROUTINE:
 635    case GLSL_TYPE_IMAGE:
 636    case GLSL_TYPE_VOID:
 637    case GLSL_TYPE_ERROR:
 638    case GLSL_TYPE_INTERFACE:
 639    case GLSL_TYPE_FUNCTION:
 640       unreachable("type does not have a natural size");
 641    }
 642 }
 643
 644 const glsl_type *
 645 glsl_atomic_uint_type(void)
 646 {
 647    return glsl_type::atomic_uint_type;
 648 }
 649
 650 unsigned
 651 glsl_atomic_size(const struct glsl_type *type)
 652 {
 653    return type->atomic_size();
 654 }
 655
 656 bool
 657 glsl_contains_atomic(const struct glsl_type *type)
 658 {
 659    return type->contains_atomic();
 660 }