src/compiler/glsl/ir.cpp

   1 /*
   2  * Copyright © 2010 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 #include <string.h>
  24 #include "ir.h"
  25 #include "compiler/glsl_types.h"
  26 #include "glsl_parser_extras.h"
  27
  28
  29 ir_rvalue::ir_rvalue(enum ir_node_type t)
  30    : ir_instruction(t)
  31 {
  32    this->type = glsl_type::error_type;
  33 }
  34
  35 bool ir_rvalue::is_zero() const
  36 {
  37    return false;
  38 }
  39
  40 bool ir_rvalue::is_one() const
  41 {
  42    return false;
  43 }
  44
  45 bool ir_rvalue::is_negative_one() const
  46 {
  47    return false;
  48 }
  49
  50 /**
  51  * Modify the swizzle make to move one component to another
  52  *
  53  * \param m    IR swizzle to be modified
  54  * \param from Component in the RHS that is to be swizzled
  55  * \param to   Desired swizzle location of \c from
  56  */
  57 static void
  58 update_rhs_swizzle(ir_swizzle_mask &m, unsigned from, unsigned to)
  59 {
  60    switch (to) {
  61    case 0: m.x = from; break;
  62    case 1: m.y = from; break;
  63    case 2: m.z = from; break;
  64    case 3: m.w = from; break;
  65    default: assert(!"Should not get here.");
  66    }
  67 }
  68
  69 void
  70 ir_assignment::set_lhs(ir_rvalue *lhs)
  71 {
  72    void *mem_ctx = this;
  73    bool swizzled = false;
  74
  75    while (lhs != NULL) {
  76       ir_swizzle *swiz = lhs->as_swizzle();
  77
  78       if (swiz == NULL)
  79          break;
  80
  81       unsigned write_mask = 0;
  82       ir_swizzle_mask rhs_swiz = { 0, 0, 0, 0, 0, 0 };
  83
  84       for (unsigned i = 0; i < swiz->mask.num_components; i++) {
  85          unsigned c = 0;
  86
  87          switch (i) {
  88          case 0: c = swiz->mask.x; break;
  89          case 1: c = swiz->mask.y; break;
  90          case 2: c = swiz->mask.z; break;
  91          case 3: c = swiz->mask.w; break;
  92          default: assert(!"Should not get here.");
  93          }
  94
  95          write_mask |= (((this->write_mask >> i) & 1) << c);
  96          update_rhs_swizzle(rhs_swiz, i, c);
  97          rhs_swiz.num_components = swiz->val->type->vector_elements;
  98       }
  99
 100       this->write_mask = write_mask;
 101       lhs = swiz->val;
 102
 103       this->rhs = new(mem_ctx) ir_swizzle(this->rhs, rhs_swiz);
 104       swizzled = true;
 105    }
 106
 107    if (swizzled) {
 108       /* Now, RHS channels line up with the LHS writemask.  Collapse it
 109        * to just the channels that will be written.
 110        */
 111       ir_swizzle_mask rhs_swiz = { 0, 0, 0, 0, 0, 0 };
 112       int rhs_chan = 0;
 113       for (int i = 0; i < 4; i++) {
 114          if (write_mask & (1 << i))
 115             update_rhs_swizzle(rhs_swiz, i, rhs_chan++);
 116       }
 117       rhs_swiz.num_components = rhs_chan;
 118       this->rhs = new(mem_ctx) ir_swizzle(this->rhs, rhs_swiz);
 119    }
 120
 121    assert((lhs == NULL) || lhs->as_dereference());
 122
 123    this->lhs = (ir_dereference *) lhs;
 124 }
 125
 126 ir_variable *
 127 ir_assignment::whole_variable_written()
 128 {
 129    ir_variable *v = this->lhs->whole_variable_referenced();
 130
 131    if (v == NULL)
 132       return NULL;
 133
 134    if (v->type->is_scalar())
 135       return v;
 136
 137    if (v->type->is_vector()) {
 138       const unsigned mask = (1U << v->type->vector_elements) - 1;
 139
 140       if (mask != this->write_mask)
 141          return NULL;
 142    }
 143
 144    /* Either all the vector components are assigned or the variable is some
 145     * composite type (and the whole thing is assigned.
 146     */
 147    return v;
 148 }
 149
 150 ir_assignment::ir_assignment(ir_dereference *lhs, ir_rvalue *rhs,
 151                              ir_rvalue *condition, unsigned write_mask)
 152    : ir_instruction(ir_type_assignment)
 153 {
 154    this->condition = condition;
 155    this->rhs = rhs;
 156    this->lhs = lhs;
 157    this->write_mask = write_mask;
 158
 159    if (lhs->type->is_scalar() || lhs->type->is_vector()) {
 160       int lhs_components = 0;
 161       for (int i = 0; i < 4; i++) {
 162          if (write_mask & (1 << i))
 163             lhs_components++;
 164       }
 165
 166       assert(lhs_components == this->rhs->type->vector_elements);
 167    }
 168 }
 169
 170 ir_assignment::ir_assignment(ir_rvalue *lhs, ir_rvalue *rhs,
 171                              ir_rvalue *condition)
 172    : ir_instruction(ir_type_assignment)
 173 {
 174    this->condition = condition;
 175    this->rhs = rhs;
 176
 177    /* If the RHS is a vector type, assume that all components of the vector
 178     * type are being written to the LHS.  The write mask comes from the RHS
 179     * because we can have a case where the LHS is a vec4 and the RHS is a
 180     * vec3.  In that case, the assignment is:
 181     *
 182     *     (assign (...) (xyz) (var_ref lhs) (var_ref rhs))
 183     */
 184    if (rhs->type->is_vector())
 185       this->write_mask = (1U << rhs->type->vector_elements) - 1;
 186    else if (rhs->type->is_scalar())
 187       this->write_mask = 1;
 188    else
 189       this->write_mask = 0;
 190
 191    this->set_lhs(lhs);
 192 }
 193
 194 ir_expression::ir_expression(int op, const struct glsl_type *type,
 195                              ir_rvalue *op0, ir_rvalue *op1,
 196                              ir_rvalue *op2, ir_rvalue *op3)
 197    : ir_rvalue(ir_type_expression)
 198 {
 199    this->type = type;
 200    this->operation = ir_expression_operation(op);
 201    this->operands[0] = op0;
 202    this->operands[1] = op1;
 203    this->operands[2] = op2;
 204    this->operands[3] = op3;
 205    init_num_operands();
 206
 207 #ifndef NDEBUG
 208    for (unsigned i = num_operands; i < 4; i++) {
 209       assert(this->operands[i] == NULL);
 210    }
 211
 212    for (unsigned i = 0; i < num_operands; i++) {
 213       assert(this->operands[i] != NULL);
 214    }
 215 #endif
 216 }
 217
 218 ir_expression::ir_expression(int op, ir_rvalue *op0)
 219    : ir_rvalue(ir_type_expression)
 220 {
 221    this->operation = ir_expression_operation(op);
 222    this->operands[0] = op0;
 223    this->operands[1] = NULL;
 224    this->operands[2] = NULL;
 225    this->operands[3] = NULL;
 226
 227    assert(op <= ir_last_unop);
 228    init_num_operands();
 229    assert(num_operands == 1);
 230    assert(this->operands[0]);
 231
 232    switch (this->operation) {
 233    case ir_unop_bit_not:
 234    case ir_unop_logic_not:
 235    case ir_unop_neg:
 236    case ir_unop_abs:
 237    case ir_unop_sign:
 238    case ir_unop_rcp:
 239    case ir_unop_rsq:
 240    case ir_unop_sqrt:
 241    case ir_unop_exp:
 242    case ir_unop_log:
 243    case ir_unop_exp2:
 244    case ir_unop_log2:
 245    case ir_unop_trunc:
 246    case ir_unop_ceil:
 247    case ir_unop_floor:
 248    case ir_unop_fract:
 249    case ir_unop_round_even:
 250    case ir_unop_sin:
 251    case ir_unop_cos:
 252    case ir_unop_dFdx:
 253    case ir_unop_dFdx_coarse:
 254    case ir_unop_dFdx_fine:
 255    case ir_unop_dFdy:
 256    case ir_unop_dFdy_coarse:
 257    case ir_unop_dFdy_fine:
 258    case ir_unop_bitfield_reverse:
 259    case ir_unop_interpolate_at_centroid:
 260    case ir_unop_saturate:
 261       this->type = op0->type;
 262       break;
 263
 264    case ir_unop_f2i:
 265    case ir_unop_b2i:
 266    case ir_unop_u2i:
 267    case ir_unop_d2i:
 268    case ir_unop_bitcast_f2i:
 269    case ir_unop_bit_count:
 270    case ir_unop_find_msb:
 271    case ir_unop_find_lsb:
 272    case ir_unop_subroutine_to_int:
 273    case ir_unop_i642i:
 274    case ir_unop_u642i:
 275       this->type = glsl_type::get_instance(GLSL_TYPE_INT,
 276                                            op0->type->vector_elements, 1);
 277       break;
 278
 279    case ir_unop_b2f:
 280    case ir_unop_i2f:
 281    case ir_unop_u2f:
 282    case ir_unop_d2f:
 283    case ir_unop_bitcast_i2f:
 284    case ir_unop_bitcast_u2f:
 285    case ir_unop_i642f:
 286    case ir_unop_u642f:
 287       this->type = glsl_type::get_instance(GLSL_TYPE_FLOAT,
 288                                            op0->type->vector_elements, 1);
 289       break;
 290
 291    case ir_unop_f2b:
 292    case ir_unop_i2b:
 293    case ir_unop_d2b:
 294    case ir_unop_i642b:
 295       this->type = glsl_type::get_instance(GLSL_TYPE_BOOL,
 296                                            op0->type->vector_elements, 1);
 297       break;
 298
 299    case ir_unop_f2d:
 300    case ir_unop_i2d:
 301    case ir_unop_u2d:
 302    case ir_unop_i642d:
 303    case ir_unop_u642d:
 304       this->type = glsl_type::get_instance(GLSL_TYPE_DOUBLE,
 305                                            op0->type->vector_elements, 1);
 306       break;
 307
 308    case ir_unop_i2u:
 309    case ir_unop_f2u:
 310    case ir_unop_d2u:
 311    case ir_unop_bitcast_f2u:
 312    case ir_unop_i642u:
 313    case ir_unop_u642u:
 314       this->type = glsl_type::get_instance(GLSL_TYPE_UINT,
 315                                            op0->type->vector_elements, 1);
 316       break;
 317
 318    case ir_unop_i2i64:
 319    case ir_unop_u2i64:
 320    case ir_unop_b2i64:
 321    case ir_unop_f2i64:
 322    case ir_unop_d2i64:
 323    case ir_unop_u642i64:
 324       this->type = glsl_type::get_instance(GLSL_TYPE_INT64,
 325                                            op0->type->vector_elements, 1);
 326       break;
 327
 328    case ir_unop_i2u64:
 329    case ir_unop_u2u64:
 330    case ir_unop_f2u64:
 331    case ir_unop_d2u64:
 332    case ir_unop_i642u64:
 333       this->type = glsl_type::get_instance(GLSL_TYPE_UINT64,
 334                                            op0->type->vector_elements, 1);
 335       break;
 336    case ir_unop_noise:
 337       this->type = glsl_type::float_type;
 338       break;
 339
 340    case ir_unop_unpack_double_2x32:
 341    case ir_unop_unpack_uint_2x32:
 342       this->type = glsl_type::uvec2_type;
 343       break;
 344
 345    case ir_unop_unpack_int_2x32:
 346       this->type = glsl_type::ivec2_type;
 347       break;
 348
 349    case ir_unop_pack_snorm_2x16:
 350    case ir_unop_pack_snorm_4x8:
 351    case ir_unop_pack_unorm_2x16:
 352    case ir_unop_pack_unorm_4x8:
 353    case ir_unop_pack_half_2x16:
 354       this->type = glsl_type::uint_type;
 355       break;
 356
 357    case ir_unop_pack_double_2x32:
 358       this->type = glsl_type::double_type;
 359       break;
 360
 361    case ir_unop_pack_int_2x32:
 362       this->type = glsl_type::int64_t_type;
 363       break;
 364
 365    case ir_unop_pack_uint_2x32:
 366       this->type = glsl_type::uint64_t_type;
 367       break;
 368
 369    case ir_unop_unpack_snorm_2x16:
 370    case ir_unop_unpack_unorm_2x16:
 371    case ir_unop_unpack_half_2x16:
 372       this->type = glsl_type::vec2_type;
 373       break;
 374
 375    case ir_unop_unpack_snorm_4x8:
 376    case ir_unop_unpack_unorm_4x8:
 377       this->type = glsl_type::vec4_type;
 378       break;
 379
 380    case ir_unop_unpack_sampler_2x32:
 381    case ir_unop_unpack_image_2x32:
 382       this->type = glsl_type::uvec2_type;
 383       break;
 384
 385    case ir_unop_pack_sampler_2x32:
 386    case ir_unop_pack_image_2x32:
 387       this->type = op0->type;
 388       break;
 389
 390    case ir_unop_frexp_sig:
 391       this->type = op0->type;
 392       break;
 393    case ir_unop_frexp_exp:
 394       this->type = glsl_type::get_instance(GLSL_TYPE_INT,
 395                                            op0->type->vector_elements, 1);
 396       break;
 397
 398    case ir_unop_get_buffer_size:
 399    case ir_unop_ssbo_unsized_array_length:
 400       this->type = glsl_type::int_type;
 401       break;
 402
 403    case ir_unop_bitcast_i642d:
 404    case ir_unop_bitcast_u642d:
 405       this->type = glsl_type::get_instance(GLSL_TYPE_DOUBLE,
 406                                            op0->type->vector_elements, 1);
 407       break;
 408
 409    case ir_unop_bitcast_d2i64:
 410       this->type = glsl_type::get_instance(GLSL_TYPE_INT64,
 411                                            op0->type->vector_elements, 1);
 412       break;
 413    case ir_unop_bitcast_d2u64:
 414       this->type = glsl_type::get_instance(GLSL_TYPE_UINT64,
 415                                            op0->type->vector_elements, 1);
 416       break;
 417
 418    default:
 419       assert(!"not reached: missing automatic type setup for ir_expression");
 420       this->type = op0->type;
 421       break;
 422    }
 423 }
 424
 425 ir_expression::ir_expression(int op, ir_rvalue *op0, ir_rvalue *op1)
 426    : ir_rvalue(ir_type_expression)
 427 {
 428    this->operation = ir_expression_operation(op);
 429    this->operands[0] = op0;
 430    this->operands[1] = op1;
 431    this->operands[2] = NULL;
 432    this->operands[3] = NULL;
 433
 434    assert(op > ir_last_unop);
 435    init_num_operands();
 436    assert(num_operands == 2);
 437    for (unsigned i = 0; i < num_operands; i++) {
 438       assert(this->operands[i] != NULL);
 439    }
 440
 441    switch (this->operation) {
 442    case ir_binop_all_equal:
 443    case ir_binop_any_nequal:
 444       this->type = glsl_type::bool_type;
 445       break;
 446
 447    case ir_binop_add:
 448    case ir_binop_sub:
 449    case ir_binop_min:
 450    case ir_binop_max:
 451    case ir_binop_pow:
 452    case ir_binop_mul:
 453    case ir_binop_div:
 454    case ir_binop_mod:
 455       if (op0->type->is_scalar()) {
 456          this->type = op1->type;
 457       } else if (op1->type->is_scalar()) {
 458          this->type = op0->type;
 459       } else {
 460          if (this->operation == ir_binop_mul) {
 461             this->type = glsl_type::get_mul_type(op0->type, op1->type);
 462          } else {
 463             assert(op0->type == op1->type);
 464             this->type = op0->type;
 465          }
 466       }
 467       break;
 468
 469    case ir_binop_logic_and:
 470    case ir_binop_logic_xor:
 471    case ir_binop_logic_or:
 472    case ir_binop_bit_and:
 473    case ir_binop_bit_xor:
 474    case ir_binop_bit_or:
 475        assert(!op0->type->is_matrix());
 476        assert(!op1->type->is_matrix());
 477       if (op0->type->is_scalar()) {
 478          this->type = op1->type;
 479       } else if (op1->type->is_scalar()) {
 480          this->type = op0->type;
 481       } else {
 482           assert(op0->type->vector_elements == op1->type->vector_elements);
 483           this->type = op0->type;
 484       }
 485       break;
 486
 487    case ir_binop_equal:
 488    case ir_binop_nequal:
 489    case ir_binop_gequal:
 490    case ir_binop_less:
 491       assert(op0->type == op1->type);
 492       this->type = glsl_type::get_instance(GLSL_TYPE_BOOL,
 493                                            op0->type->vector_elements, 1);
 494       break;
 495
 496    case ir_binop_dot:
 497       this->type = op0->type->get_base_type();
 498       break;
 499
 500    case ir_binop_imul_high:
 501    case ir_binop_carry:
 502    case ir_binop_borrow:
 503    case ir_binop_lshift:
 504    case ir_binop_rshift:
 505    case ir_binop_ldexp:
 506    case ir_binop_interpolate_at_offset:
 507    case ir_binop_interpolate_at_sample:
 508       this->type = op0->type;
 509       break;
 510
 511    case ir_binop_vector_extract:
 512       this->type = op0->type->get_scalar_type();
 513       break;
 514
 515    default:
 516       assert(!"not reached: missing automatic type setup for ir_expression");
 517       this->type = glsl_type::float_type;
 518    }
 519 }
 520
 521 ir_expression::ir_expression(int op, ir_rvalue *op0, ir_rvalue *op1,
 522                              ir_rvalue *op2)
 523    : ir_rvalue(ir_type_expression)
 524 {
 525    this->operation = ir_expression_operation(op);
 526    this->operands[0] = op0;
 527    this->operands[1] = op1;
 528    this->operands[2] = op2;
 529    this->operands[3] = NULL;
 530
 531    assert(op > ir_last_binop && op <= ir_last_triop);
 532    init_num_operands();
 533    assert(num_operands == 3);
 534    for (unsigned i = 0; i < num_operands; i++) {
 535       assert(this->operands[i] != NULL);
 536    }
 537
 538    switch (this->operation) {
 539    case ir_triop_fma:
 540    case ir_triop_lrp:
 541    case ir_triop_bitfield_extract:
 542    case ir_triop_vector_insert:
 543       this->type = op0->type;
 544       break;
 545
 546    case ir_triop_csel:
 547       this->type = op1->type;
 548       break;
 549
 550    default:
 551       assert(!"not reached: missing automatic type setup for ir_expression");
 552       this->type = glsl_type::float_type;
 553    }
 554 }
 555
 556 /**
 557  * This is only here for ir_reader to used for testing purposes. Please use
 558  * the precomputed num_operands field if you need the number of operands.
 559  */
 560 unsigned
 561 ir_expression::get_num_operands(ir_expression_operation op)
 562 {
 563    assert(op <= ir_last_opcode);
 564
 565    if (op <= ir_last_unop)
 566       return 1;
 567
 568    if (op <= ir_last_binop)
 569       return 2;
 570
 571    if (op <= ir_last_triop)
 572       return 3;
 573
 574    if (op <= ir_last_quadop)
 575       return 4;
 576
 577    unreachable("Could not calculate number of operands");
 578 }
 579
 580 #include "ir_expression_operation_strings.h"
 581
 582 const char*
 583 depth_layout_string(ir_depth_layout layout)
 584 {
 585    switch(layout) {
 586    case ir_depth_layout_none:      return "";
 587    case ir_depth_layout_any:       return "depth_any";
 588    case ir_depth_layout_greater:   return "depth_greater";
 589    case ir_depth_layout_less:      return "depth_less";
 590    case ir_depth_layout_unchanged: return "depth_unchanged";
 591
 592    default:
 593       assert(0);
 594       return "";
 595    }
 596 }
 597
 598 ir_expression_operation
 599 ir_expression::get_operator(const char *str)
 600 {
 601    for (int op = 0; op <= int(ir_last_opcode); op++) {
 602       if (strcmp(str, ir_expression_operation_strings[op]) == 0)
 603          return (ir_expression_operation) op;
 604    }
 605    return (ir_expression_operation) -1;
 606 }
 607
 608 ir_variable *
 609 ir_expression::variable_referenced() const
 610 {
 611    switch (operation) {
 612       case ir_binop_vector_extract:
 613       case ir_triop_vector_insert:
 614          /* We get these for things like a[0] where a is a vector type. In these
 615           * cases we want variable_referenced() to return the actual vector
 616           * variable this is wrapping.
 617           */
 618          return operands[0]->variable_referenced();
 619       default:
 620          return ir_rvalue::variable_referenced();
 621    }
 622 }
 623
 624 ir_constant::ir_constant()
 625    : ir_rvalue(ir_type_constant)
 626 {
 627    this->const_elements = NULL;
 628 }
 629
 630 ir_constant::ir_constant(const struct glsl_type *type,
 631                          const ir_constant_data *data)
 632    : ir_rvalue(ir_type_constant)
 633 {
 634    this->const_elements = NULL;
 635
 636    assert((type->base_type >= GLSL_TYPE_UINT)
 637           && (type->base_type <= GLSL_TYPE_IMAGE));
 638
 639    this->type = type;
 640    memcpy(& this->value, data, sizeof(this->value));
 641 }
 642
 643 ir_constant::ir_constant(float f, unsigned vector_elements)
 644    : ir_rvalue(ir_type_constant)
 645 {
 646    assert(vector_elements <= 4);
 647    this->type = glsl_type::get_instance(GLSL_TYPE_FLOAT, vector_elements, 1);
 648    for (unsigned i = 0; i < vector_elements; i++) {
 649       this->value.f[i] = f;
 650    }
 651    for (unsigned i = vector_elements; i < 16; i++)  {
 652       this->value.f[i] = 0;
 653    }
 654 }
 655
 656 ir_constant::ir_constant(double d, unsigned vector_elements)
 657    : ir_rvalue(ir_type_constant)
 658 {
 659    assert(vector_elements <= 4);
 660    this->type = glsl_type::get_instance(GLSL_TYPE_DOUBLE, vector_elements, 1);
 661    for (unsigned i = 0; i < vector_elements; i++) {
 662       this->value.d[i] = d;
 663    }
 664    for (unsigned i = vector_elements; i < 16; i++)  {
 665       this->value.d[i] = 0.0;
 666    }
 667 }
 668
 669 ir_constant::ir_constant(unsigned int u, unsigned vector_elements)
 670    : ir_rvalue(ir_type_constant)
 671 {
 672    assert(vector_elements <= 4);
 673    this->type = glsl_type::get_instance(GLSL_TYPE_UINT, vector_elements, 1);
 674    for (unsigned i = 0; i < vector_elements; i++) {
 675       this->value.u[i] = u;
 676    }
 677    for (unsigned i = vector_elements; i < 16; i++) {
 678       this->value.u[i] = 0;
 679    }
 680 }
 681
 682 ir_constant::ir_constant(int integer, unsigned vector_elements)
 683    : ir_rvalue(ir_type_constant)
 684 {
 685    assert(vector_elements <= 4);
 686    this->type = glsl_type::get_instance(GLSL_TYPE_INT, vector_elements, 1);
 687    for (unsigned i = 0; i < vector_elements; i++) {
 688       this->value.i[i] = integer;
 689    }
 690    for (unsigned i = vector_elements; i < 16; i++) {
 691       this->value.i[i] = 0;
 692    }
 693 }
 694
 695 ir_constant::ir_constant(uint64_t u64, unsigned vector_elements)
 696    : ir_rvalue(ir_type_constant)
 697 {
 698    assert(vector_elements <= 4);
 699    this->type = glsl_type::get_instance(GLSL_TYPE_UINT64, vector_elements, 1);
 700    for (unsigned i = 0; i < vector_elements; i++) {
 701       this->value.u64[i] = u64;
 702    }
 703    for (unsigned i = vector_elements; i < 16; i++) {
 704       this->value.u64[i] = 0;
 705    }
 706 }
 707
 708 ir_constant::ir_constant(int64_t int64, unsigned vector_elements)
 709    : ir_rvalue(ir_type_constant)
 710 {
 711    assert(vector_elements <= 4);
 712    this->type = glsl_type::get_instance(GLSL_TYPE_INT64, vector_elements, 1);
 713    for (unsigned i = 0; i < vector_elements; i++) {
 714       this->value.i64[i] = int64;
 715    }
 716    for (unsigned i = vector_elements; i < 16; i++) {
 717       this->value.i64[i] = 0;
 718    }
 719 }
 720
 721 ir_constant::ir_constant(bool b, unsigned vector_elements)
 722    : ir_rvalue(ir_type_constant)
 723 {
 724    assert(vector_elements <= 4);
 725    this->type = glsl_type::get_instance(GLSL_TYPE_BOOL, vector_elements, 1);
 726    for (unsigned i = 0; i < vector_elements; i++) {
 727       this->value.b[i] = b;
 728    }
 729    for (unsigned i = vector_elements; i < 16; i++) {
 730       this->value.b[i] = false;
 731    }
 732 }
 733
 734 ir_constant::ir_constant(const ir_constant *c, unsigned i)
 735    : ir_rvalue(ir_type_constant)
 736 {
 737    this->const_elements = NULL;
 738    this->type = c->type->get_base_type();
 739
 740    switch (this->type->base_type) {
 741    case GLSL_TYPE_UINT:  this->value.u[0] = c->value.u[i]; break;
 742    case GLSL_TYPE_INT:   this->value.i[0] = c->value.i[i]; break;
 743    case GLSL_TYPE_FLOAT: this->value.f[0] = c->value.f[i]; break;
 744    case GLSL_TYPE_BOOL:  this->value.b[0] = c->value.b[i]; break;
 745    case GLSL_TYPE_DOUBLE: this->value.d[0] = c->value.d[i]; break;
 746    default:              assert(!"Should not get here."); break;
 747    }
 748 }
 749
 750 ir_constant::ir_constant(const struct glsl_type *type, exec_list *value_list)
 751    : ir_rvalue(ir_type_constant)
 752 {
 753    this->const_elements = NULL;
 754    this->type = type;
 755
 756    assert(type->is_scalar() || type->is_vector() || type->is_matrix()
 757           || type->is_record() || type->is_array());
 758
 759    /* If the constant is a record, the types of each of the entries in
 760     * value_list must be a 1-for-1 match with the structure components.  Each
 761     * entry must also be a constant.  Just move the nodes from the value_list
 762     * to the list in the ir_constant.
 763     */
 764    if (type->is_array() || type->is_record()) {
 765       this->const_elements = ralloc_array(this, ir_constant *, type->length);
 766       unsigned i = 0;
 767       foreach_in_list(ir_constant, value, value_list) {
 768          assert(value->as_constant() != NULL);
 769
 770          this->const_elements[i++] = value;
 771       }
 772       return;
 773    }
 774
 775    for (unsigned i = 0; i < 16; i++) {
 776       this->value.u[i] = 0;
 777    }
 778
 779    ir_constant *value = (ir_constant *) (value_list->get_head_raw());
 780
 781    /* Constructors with exactly one scalar argument are special for vectors
 782     * and matrices.  For vectors, the scalar value is replicated to fill all
 783     * the components.  For matrices, the scalar fills the components of the
 784     * diagonal while the rest is filled with 0.
 785     */
 786    if (value->type->is_scalar() && value->next->is_tail_sentinel()) {
 787       if (type->is_matrix()) {
 788          /* Matrix - fill diagonal (rest is already set to 0) */
 789          assert(type->is_float() || type->is_double());
 790          for (unsigned i = 0; i < type->matrix_columns; i++) {
 791             if (type->is_float())
 792                this->value.f[i * type->vector_elements + i] =
 793                   value->value.f[0];
 794             else
 795                this->value.d[i * type->vector_elements + i] =
 796                   value->value.d[0];
 797          }
 798       } else {
 799          /* Vector or scalar - fill all components */
 800          switch (type->base_type) {
 801          case GLSL_TYPE_UINT:
 802          case GLSL_TYPE_INT:
 803             for (unsigned i = 0; i < type->components(); i++)
 804                this->value.u[i] = value->value.u[0];
 805             break;
 806          case GLSL_TYPE_FLOAT:
 807             for (unsigned i = 0; i < type->components(); i++)
 808                this->value.f[i] = value->value.f[0];
 809             break;
 810          case GLSL_TYPE_DOUBLE:
 811             for (unsigned i = 0; i < type->components(); i++)
 812                this->value.d[i] = value->value.d[0];
 813             break;
 814          case GLSL_TYPE_UINT64:
 815          case GLSL_TYPE_INT64:
 816             for (unsigned i = 0; i < type->components(); i++)
 817                this->value.u64[i] = value->value.u64[0];
 818             break;
 819          case GLSL_TYPE_BOOL:
 820             for (unsigned i = 0; i < type->components(); i++)
 821                this->value.b[i] = value->value.b[0];
 822             break;
 823          default:
 824             assert(!"Should not get here.");
 825             break;
 826          }
 827       }
 828       return;
 829    }
 830
 831    if (type->is_matrix() && value->type->is_matrix()) {
 832       assert(value->next->is_tail_sentinel());
 833
 834       /* From section 5.4.2 of the GLSL 1.20 spec:
 835        * "If a matrix is constructed from a matrix, then each component
 836        *  (column i, row j) in the result that has a corresponding component
 837        *  (column i, row j) in the argument will be initialized from there."
 838        */
 839       unsigned cols = MIN2(type->matrix_columns, value->type->matrix_columns);
 840       unsigned rows = MIN2(type->vector_elements, value->type->vector_elements);
 841       for (unsigned i = 0; i < cols; i++) {
 842          for (unsigned j = 0; j < rows; j++) {
 843             const unsigned src = i * value->type->vector_elements + j;
 844             const unsigned dst = i * type->vector_elements + j;
 845             this->value.f[dst] = value->value.f[src];
 846          }
 847       }
 848
 849       /* "All other components will be initialized to the identity matrix." */
 850       for (unsigned i = cols; i < type->matrix_columns; i++)
 851          this->value.f[i * type->vector_elements + i] = 1.0;
 852
 853       return;
 854    }
 855
 856    /* Use each component from each entry in the value_list to initialize one
 857     * component of the constant being constructed.
 858     */
 859    unsigned i = 0;
 860    for (;;) {
 861       assert(value->as_constant() != NULL);
 862       assert(!value->is_tail_sentinel());
 863
 864       for (unsigned j = 0; j < value->type->components(); j++) {
 865          switch (type->base_type) {
 866          case GLSL_TYPE_UINT:
 867             this->value.u[i] = value->get_uint_component(j);
 868             break;
 869          case GLSL_TYPE_INT:
 870             this->value.i[i] = value->get_int_component(j);
 871             break;
 872          case GLSL_TYPE_FLOAT:
 873             this->value.f[i] = value->get_float_component(j);
 874             break;
 875          case GLSL_TYPE_BOOL:
 876             this->value.b[i] = value->get_bool_component(j);
 877             break;
 878          case GLSL_TYPE_DOUBLE:
 879             this->value.d[i] = value->get_double_component(j);
 880             break;
 881          case GLSL_TYPE_UINT64:
 882             this->value.u64[i] = value->get_uint64_component(j);
 883             break;
 884          case GLSL_TYPE_INT64:
 885             this->value.i64[i] = value->get_int64_component(j);
 886             break;
 887          default:
 888             /* FINISHME: What to do?  Exceptions are not the answer.
 889              */
 890             break;
 891          }
 892
 893          i++;
 894          if (i >= type->components())
 895             break;
 896       }
 897
 898       if (i >= type->components())
 899          break; /* avoid downcasting a list sentinel */
 900       value = (ir_constant *) value->next;
 901    }
 902 }
 903
 904 ir_constant *
 905 ir_constant::zero(void *mem_ctx, const glsl_type *type)
 906 {
 907    assert(type->is_scalar() || type->is_vector() || type->is_matrix()
 908           || type->is_record() || type->is_array());
 909
 910    ir_constant *c = new(mem_ctx) ir_constant;
 911    c->type = type;
 912    memset(&c->value, 0, sizeof(c->value));
 913
 914    if (type->is_array()) {
 915       c->const_elements = ralloc_array(c, ir_constant *, type->length);
 916
 917       for (unsigned i = 0; i < type->length; i++)
 918          c->const_elements[i] = ir_constant::zero(c, type->fields.array);
 919    }
 920
 921    if (type->is_record()) {
 922       c->const_elements = ralloc_array(c, ir_constant *, type->length);
 923
 924       for (unsigned i = 0; i < type->length; i++) {
 925          c->const_elements[i] =
 926             ir_constant::zero(mem_ctx, type->fields.structure[i].type);
 927       }
 928    }
 929
 930    return c;
 931 }
 932
 933 bool
 934 ir_constant::get_bool_component(unsigned i) const
 935 {
 936    switch (this->type->base_type) {
 937    case GLSL_TYPE_UINT:  return this->value.u[i] != 0;
 938    case GLSL_TYPE_INT:   return this->value.i[i] != 0;
 939    case GLSL_TYPE_FLOAT: return ((int)this->value.f[i]) != 0;
 940    case GLSL_TYPE_BOOL:  return this->value.b[i];
 941    case GLSL_TYPE_DOUBLE: return this->value.d[i] != 0.0;
 942    case GLSL_TYPE_UINT64: return this->value.u64[i] != 0;
 943    case GLSL_TYPE_INT64:  return this->value.i64[i] != 0;
 944    default:              assert(!"Should not get here."); break;
 945    }
 946
 947    /* Must return something to make the compiler happy.  This is clearly an
 948     * error case.
 949     */
 950    return false;
 951 }
 952
 953 float
 954 ir_constant::get_float_component(unsigned i) const
 955 {
 956    switch (this->type->base_type) {
 957    case GLSL_TYPE_UINT:  return (float) this->value.u[i];
 958    case GLSL_TYPE_INT:   return (float) this->value.i[i];
 959    case GLSL_TYPE_FLOAT: return this->value.f[i];
 960    case GLSL_TYPE_BOOL:  return this->value.b[i] ? 1.0f : 0.0f;
 961    case GLSL_TYPE_DOUBLE: return (float) this->value.d[i];
 962    case GLSL_TYPE_UINT64: return (float) this->value.u64[i];
 963    case GLSL_TYPE_INT64:  return (float) this->value.i64[i];
 964    default:              assert(!"Should not get here."); break;
 965    }
 966
 967    /* Must return something to make the compiler happy.  This is clearly an
 968     * error case.
 969     */
 970    return 0.0;
 971 }
 972
 973 double
 974 ir_constant::get_double_component(unsigned i) const
 975 {
 976    switch (this->type->base_type) {
 977    case GLSL_TYPE_UINT:  return (double) this->value.u[i];
 978    case GLSL_TYPE_INT:   return (double) this->value.i[i];
 979    case GLSL_TYPE_FLOAT: return (double) this->value.f[i];
 980    case GLSL_TYPE_BOOL:  return this->value.b[i] ? 1.0 : 0.0;
 981    case GLSL_TYPE_DOUBLE: return this->value.d[i];
 982    case GLSL_TYPE_UINT64: return (double) this->value.u64[i];
 983    case GLSL_TYPE_INT64:  return (double) this->value.i64[i];
 984    default:              assert(!"Should not get here."); break;
 985    }
 986
 987    /* Must return something to make the compiler happy.  This is clearly an
 988     * error case.
 989     */
 990    return 0.0;
 991 }
 992
 993 int
 994 ir_constant::get_int_component(unsigned i) const
 995 {
 996    switch (this->type->base_type) {
 997    case GLSL_TYPE_UINT:  return this->value.u[i];
 998    case GLSL_TYPE_INT:   return this->value.i[i];
 999    case GLSL_TYPE_FLOAT: return (int) this->value.f[i];
1000    case GLSL_TYPE_BOOL:  return this->value.b[i] ? 1 : 0;
1001    case GLSL_TYPE_DOUBLE: return (int) this->value.d[i];
1002    case GLSL_TYPE_UINT64: return (int) this->value.u64[i];
1003    case GLSL_TYPE_INT64:  return (int) this->value.i64[i];
1004    default:              assert(!"Should not get here."); break;
1005    }
1006
1007    /* Must return something to make the compiler happy.  This is clearly an
1008     * error case.
1009     */
1010    return 0;
1011 }
1012
1013 unsigned
1014 ir_constant::get_uint_component(unsigned i) const
1015 {
1016    switch (this->type->base_type) {
1017    case GLSL_TYPE_UINT:  return this->value.u[i];
1018    case GLSL_TYPE_INT:   return this->value.i[i];
1019    case GLSL_TYPE_FLOAT: return (unsigned) this->value.f[i];
1020    case GLSL_TYPE_BOOL:  return this->value.b[i] ? 1 : 0;
1021    case GLSL_TYPE_DOUBLE: return (unsigned) this->value.d[i];
1022    case GLSL_TYPE_UINT64: return (unsigned) this->value.u64[i];
1023    case GLSL_TYPE_INT64:  return (unsigned) this->value.i64[i];
1024    default:              assert(!"Should not get here."); break;
1025    }
1026
1027    /* Must return something to make the compiler happy.  This is clearly an
1028     * error case.
1029     */
1030    return 0;
1031 }
1032
1033 int64_t
1034 ir_constant::get_int64_component(unsigned i) const
1035 {
1036    switch (this->type->base_type) {
1037    case GLSL_TYPE_UINT:  return this->value.u[i];
1038    case GLSL_TYPE_INT:   return this->value.i[i];
1039    case GLSL_TYPE_FLOAT: return (int64_t) this->value.f[i];
1040    case GLSL_TYPE_BOOL:  return this->value.b[i] ? 1 : 0;
1041    case GLSL_TYPE_DOUBLE: return (int64_t) this->value.d[i];
1042    case GLSL_TYPE_UINT64: return (int64_t) this->value.u64[i];
1043    case GLSL_TYPE_INT64:  return this->value.i64[i];
1044    default:              assert(!"Should not get here."); break;
1045    }
1046
1047    /* Must return something to make the compiler happy.  This is clearly an
1048     * error case.
1049     */
1050    return 0;
1051 }
1052
1053 uint64_t
1054 ir_constant::get_uint64_component(unsigned i) const
1055 {
1056    switch (this->type->base_type) {
1057    case GLSL_TYPE_UINT:  return this->value.u[i];
1058    case GLSL_TYPE_INT:   return this->value.i[i];
1059    case GLSL_TYPE_FLOAT: return (uint64_t) this->value.f[i];
1060    case GLSL_TYPE_BOOL:  return this->value.b[i] ? 1 : 0;
1061    case GLSL_TYPE_DOUBLE: return (uint64_t) this->value.d[i];
1062    case GLSL_TYPE_UINT64: return this->value.u64[i];
1063    case GLSL_TYPE_INT64:  return (uint64_t) this->value.i64[i];
1064    default:              assert(!"Should not get here."); break;
1065    }
1066
1067    /* Must return something to make the compiler happy.  This is clearly an
1068     * error case.
1069     */
1070    return 0;
1071 }
1072
1073 ir_constant *
1074 ir_constant::get_array_element(unsigned i) const
1075 {
1076    assert(this->type->is_array());
1077
1078    /* From page 35 (page 41 of the PDF) of the GLSL 1.20 spec:
1079     *
1080     *     "Behavior is undefined if a shader subscripts an array with an index
1081     *     less than 0 or greater than or equal to the size the array was
1082     *     declared with."
1083     *
1084     * Most out-of-bounds accesses are removed before things could get this far.
1085     * There are cases where non-constant array index values can get constant
1086     * folded.
1087     */
1088    if (int(i) < 0)
1089       i = 0;
1090    else if (i >= this->type->length)
1091       i = this->type->length - 1;
1092
1093    return const_elements[i];
1094 }
1095
1096 ir_constant *
1097 ir_constant::get_record_field(int idx)
1098 {
1099    assert(this->type->is_record());
1100    assert(idx >= 0 && (unsigned) idx < this->type->length);
1101
1102    return const_elements[idx];
1103 }
1104
1105 void
1106 ir_constant::copy_offset(ir_constant *src, int offset)
1107 {
1108    switch (this->type->base_type) {
1109    case GLSL_TYPE_UINT:
1110    case GLSL_TYPE_INT:
1111    case GLSL_TYPE_FLOAT:
1112    case GLSL_TYPE_DOUBLE:
1113    case GLSL_TYPE_UINT64:
1114    case GLSL_TYPE_INT64:
1115    case GLSL_TYPE_BOOL: {
1116       unsigned int size = src->type->components();
1117       assert (size <= this->type->components() - offset);
1118       for (unsigned int i=0; i<size; i++) {
1119          switch (this->type->base_type) {
1120          case GLSL_TYPE_UINT:
1121             value.u[i+offset] = src->get_uint_component(i);
1122             break;
1123          case GLSL_TYPE_INT:
1124             value.i[i+offset] = src->get_int_component(i);
1125             break;
1126          case GLSL_TYPE_FLOAT:
1127             value.f[i+offset] = src->get_float_component(i);
1128             break;
1129          case GLSL_TYPE_BOOL:
1130             value.b[i+offset] = src->get_bool_component(i);
1131             break;
1132          case GLSL_TYPE_DOUBLE:
1133             value.d[i+offset] = src->get_double_component(i);
1134             break;
1135          case GLSL_TYPE_UINT64:
1136             value.u64[i+offset] = src->get_uint64_component(i);
1137             break;
1138          case GLSL_TYPE_INT64:
1139             value.i64[i+offset] = src->get_int64_component(i);
1140             break;
1141          default: // Shut up the compiler
1142             break;
1143          }
1144       }
1145       break;
1146    }
1147
1148    case GLSL_TYPE_STRUCT:
1149    case GLSL_TYPE_ARRAY: {
1150       assert (src->type == this->type);
1151       for (unsigned i = 0; i < this->type->length; i++) {
1152          this->const_elements[i] = src->const_elements[i]->clone(this, NULL);
1153       }
1154       break;
1155    }
1156
1157    default:
1158       assert(!"Should not get here.");
1159       break;
1160    }
1161 }
1162
1163 void
1164 ir_constant::copy_masked_offset(ir_constant *src, int offset, unsigned int mask)
1165 {
1166    assert (!type->is_array() && !type->is_record());
1167
1168    if (!type->is_vector() && !type->is_matrix()) {
1169       offset = 0;
1170       mask = 1;
1171    }
1172
1173    int id = 0;
1174    for (int i=0; i<4; i++) {
1175       if (mask & (1 << i)) {
1176          switch (this->type->base_type) {
1177          case GLSL_TYPE_UINT:
1178             value.u[i+offset] = src->get_uint_component(id++);
1179             break;
1180          case GLSL_TYPE_INT:
1181             value.i[i+offset] = src->get_int_component(id++);
1182             break;
1183          case GLSL_TYPE_FLOAT:
1184             value.f[i+offset] = src->get_float_component(id++);
1185             break;
1186          case GLSL_TYPE_BOOL:
1187             value.b[i+offset] = src->get_bool_component(id++);
1188             break;
1189          case GLSL_TYPE_DOUBLE:
1190             value.d[i+offset] = src->get_double_component(id++);
1191             break;
1192          case GLSL_TYPE_UINT64:
1193             value.u64[i+offset] = src->get_uint64_component(id++);
1194             break;
1195          case GLSL_TYPE_INT64:
1196             value.i64[i+offset] = src->get_int64_component(id++);
1197             break;
1198          default:
1199             assert(!"Should not get here.");
1200             return;
1201          }
1202       }
1203    }
1204 }
1205
1206 bool
1207 ir_constant::has_value(const ir_constant *c) const
1208 {
1209    if (this->type != c->type)
1210       return false;
1211
1212    if (this->type->is_array() || this->type->is_record()) {
1213       for (unsigned i = 0; i < this->type->length; i++) {
1214          if (!this->const_elements[i]->has_value(c->const_elements[i]))
1215             return false;
1216       }
1217       return true;
1218    }
1219
1220    for (unsigned i = 0; i < this->type->components(); i++) {
1221       switch (this->type->base_type) {
1222       case GLSL_TYPE_UINT:
1223          if (this->value.u[i] != c->value.u[i])
1224             return false;
1225          break;
1226       case GLSL_TYPE_INT:
1227          if (this->value.i[i] != c->value.i[i])
1228             return false;
1229          break;
1230       case GLSL_TYPE_FLOAT:
1231          if (this->value.f[i] != c->value.f[i])
1232             return false;
1233          break;
1234       case GLSL_TYPE_BOOL:
1235          if (this->value.b[i] != c->value.b[i])
1236             return false;
1237          break;
1238       case GLSL_TYPE_DOUBLE:
1239          if (this->value.d[i] != c->value.d[i])
1240             return false;
1241          break;
1242       case GLSL_TYPE_UINT64:
1243          if (this->value.u64[i] != c->value.u64[i])
1244             return false;
1245          break;
1246       case GLSL_TYPE_INT64:
1247          if (this->value.i64[i] != c->value.i64[i])
1248             return false;
1249          break;
1250       default:
1251          assert(!"Should not get here.");
1252          return false;
1253       }
1254    }
1255
1256    return true;
1257 }
1258
1259 bool
1260 ir_constant::is_value(float f, int i) const
1261 {
1262    if (!this->type->is_scalar() && !this->type->is_vector())
1263       return false;
1264
1265    /* Only accept boolean values for 0/1. */
1266    if (int(bool(i)) != i && this->type->is_boolean())
1267       return false;
1268
1269    for (unsigned c = 0; c < this->type->vector_elements; c++) {
1270       switch (this->type->base_type) {
1271       case GLSL_TYPE_FLOAT:
1272          if (this->value.f[c] != f)
1273             return false;
1274          break;
1275       case GLSL_TYPE_INT:
1276          if (this->value.i[c] != i)
1277             return false;
1278          break;
1279       case GLSL_TYPE_UINT:
1280          if (this->value.u[c] != unsigned(i))
1281             return false;
1282          break;
1283       case GLSL_TYPE_BOOL:
1284          if (this->value.b[c] != bool(i))
1285             return false;
1286          break;
1287       case GLSL_TYPE_DOUBLE:
1288          if (this->value.d[c] != double(f))
1289             return false;
1290          break;
1291       case GLSL_TYPE_UINT64:
1292          if (this->value.u64[c] != uint64_t(i))
1293             return false;
1294          break;
1295       case GLSL_TYPE_INT64:
1296          if (this->value.i64[c] != i)
1297             return false;
1298          break;
1299       default:
1300          /* The only other base types are structures, arrays, and samplers.
1301           * Samplers cannot be constants, and the others should have been
1302           * filtered out above.
1303           */
1304          assert(!"Should not get here.");
1305          return false;
1306       }
1307    }
1308
1309    return true;
1310 }
1311
1312 bool
1313 ir_constant::is_zero() const
1314 {
1315    return is_value(0.0, 0);
1316 }
1317
1318 bool
1319 ir_constant::is_one() const
1320 {
1321    return is_value(1.0, 1);
1322 }
1323
1324 bool
1325 ir_constant::is_negative_one() const
1326 {
1327    return is_value(-1.0, -1);
1328 }
1329
1330 bool
1331 ir_constant::is_uint16_constant() const
1332 {
1333    if (!type->is_integer())
1334       return false;
1335
1336    return value.u[0] < (1 << 16);
1337 }
1338
1339 ir_loop::ir_loop()
1340    : ir_instruction(ir_type_loop)
1341 {
1342 }
1343
1344
1345 ir_dereference_variable::ir_dereference_variable(ir_variable *var)
1346    : ir_dereference(ir_type_dereference_variable)
1347 {
1348    assert(var != NULL);
1349
1350    this->var = var;
1351    this->type = var->type;
1352 }
1353
1354
1355 ir_dereference_array::ir_dereference_array(ir_rvalue *value,
1356                                            ir_rvalue *array_index)
1357    : ir_dereference(ir_type_dereference_array)
1358 {
1359    this->array_index = array_index;
1360    this->set_array(value);
1361 }
1362
1363
1364 ir_dereference_array::ir_dereference_array(ir_variable *var,
1365                                            ir_rvalue *array_index)
1366    : ir_dereference(ir_type_dereference_array)
1367 {
1368    void *ctx = ralloc_parent(var);
1369
1370    this->array_index = array_index;
1371    this->set_array(new(ctx) ir_dereference_variable(var));
1372 }
1373
1374
1375 void
1376 ir_dereference_array::set_array(ir_rvalue *value)
1377 {
1378    assert(value != NULL);
1379
1380    this->array = value;
1381
1382    const glsl_type *const vt = this->array->type;
1383
1384    if (vt->is_array()) {
1385       type = vt->fields.array;
1386    } else if (vt->is_matrix()) {
1387       type = vt->column_type();
1388    } else if (vt->is_vector()) {
1389       type = vt->get_base_type();
1390    }
1391 }
1392
1393
1394 ir_dereference_record::ir_dereference_record(ir_rvalue *value,
1395                                              const char *field)
1396    : ir_dereference(ir_type_dereference_record)
1397 {
1398    assert(value != NULL);
1399
1400    this->record = value;
1401    this->type = this->record->type->field_type(field);
1402    this->field_idx = this->record->type->field_index(field);
1403 }
1404
1405
1406 ir_dereference_record::ir_dereference_record(ir_variable *var,
1407                                              const char *field)
1408    : ir_dereference(ir_type_dereference_record)
1409 {
1410    void *ctx = ralloc_parent(var);
1411
1412    this->record = new(ctx) ir_dereference_variable(var);
1413    this->type = this->record->type->field_type(field);
1414    this->field_idx = this->record->type->field_index(field);
1415 }
1416
1417 bool
1418 ir_dereference::is_lvalue(const struct _mesa_glsl_parse_state *state) const
1419 {
1420    ir_variable *var = this->variable_referenced();
1421
1422    /* Every l-value derference chain eventually ends in a variable.
1423     */
1424    if ((var == NULL) || var->data.read_only)
1425       return false;
1426
1427    /* From section 4.1.7 of the ARB_bindless_texture spec:
1428     *
1429     * "Samplers can be used as l-values, so can be assigned into and used as
1430     *  "out" and "inout" function parameters."
1431     *
1432     * From section 4.1.X of the ARB_bindless_texture spec:
1433     *
1434     * "Images can be used as l-values, so can be assigned into and used as
1435     *  "out" and "inout" function parameters."
1436     */
1437    if ((!state || state->has_bindless()) &&
1438        (this->type->contains_sampler() || this->type->contains_image()))
1439       return true;
1440
1441    /* From section 4.1.7 of the GLSL 4.40 spec:
1442     *
1443     *   "Opaque variables cannot be treated as l-values; hence cannot
1444     *    be used as out or inout function parameters, nor can they be
1445     *    assigned into."
1446     */
1447    if (this->type->contains_opaque())
1448       return false;
1449
1450    return true;
1451 }
1452
1453
1454 static const char * const tex_opcode_strs[] = { "tex", "txb", "txl", "txd", "txf", "txf_ms", "txs", "lod", "tg4", "query_levels", "texture_samples", "samples_identical" };
1455
1456 const char *ir_texture::opcode_string()
1457 {
1458    assert((unsigned int) op < ARRAY_SIZE(tex_opcode_strs));
1459    return tex_opcode_strs[op];
1460 }
1461
1462 ir_texture_opcode
1463 ir_texture::get_opcode(const char *str)
1464 {
1465    const int count = sizeof(tex_opcode_strs) / sizeof(tex_opcode_strs[0]);
1466    for (int op = 0; op < count; op++) {
1467       if (strcmp(str, tex_opcode_strs[op]) == 0)
1468          return (ir_texture_opcode) op;
1469    }
1470    return (ir_texture_opcode) -1;
1471 }
1472
1473
1474 void
1475 ir_texture::set_sampler(ir_dereference *sampler, const glsl_type *type)
1476 {
1477    assert(sampler != NULL);
1478    assert(type != NULL);
1479    this->sampler = sampler;
1480    this->type = type;
1481
1482    if (this->op == ir_txs || this->op == ir_query_levels ||
1483        this->op == ir_texture_samples) {
1484       assert(type->base_type == GLSL_TYPE_INT);
1485    } else if (this->op == ir_lod) {
1486       assert(type->vector_elements == 2);
1487       assert(type->is_float());
1488    } else if (this->op == ir_samples_identical) {
1489       assert(type == glsl_type::bool_type);
1490       assert(sampler->type->is_sampler());
1491       assert(sampler->type->sampler_dimensionality == GLSL_SAMPLER_DIM_MS);
1492    } else {
1493       assert(sampler->type->sampled_type == (int) type->base_type);
1494       if (sampler->type->sampler_shadow)
1495          assert(type->vector_elements == 4 || type->vector_elements == 1);
1496       else
1497          assert(type->vector_elements == 4);
1498    }
1499 }
1500
1501
1502 void
1503 ir_swizzle::init_mask(const unsigned *comp, unsigned count)
1504 {
1505    assert((count >= 1) && (count <= 4));
1506
1507    memset(&this->mask, 0, sizeof(this->mask));
1508    this->mask.num_components = count;
1509
1510    unsigned dup_mask = 0;
1511    switch (count) {
1512    case 4:
1513       assert(comp[3] <= 3);
1514       dup_mask |= (1U << comp[3])
1515          & ((1U << comp[0]) | (1U << comp[1]) | (1U << comp[2]));
1516       this->mask.w = comp[3];
1517
1518    case 3:
1519       assert(comp[2] <= 3);
1520       dup_mask |= (1U << comp[2])
1521          & ((1U << comp[0]) | (1U << comp[1]));
1522       this->mask.z = comp[2];
1523
1524    case 2:
1525       assert(comp[1] <= 3);
1526       dup_mask |= (1U << comp[1])
1527          & ((1U << comp[0]));
1528       this->mask.y = comp[1];
1529
1530    case 1:
1531       assert(comp[0] <= 3);
1532       this->mask.x = comp[0];
1533    }
1534
1535    this->mask.has_duplicates = dup_mask != 0;
1536
1537    /* Based on the number of elements in the swizzle and the base type
1538     * (i.e., float, int, unsigned, or bool) of the vector being swizzled,
1539     * generate the type of the resulting value.
1540     */
1541    type = glsl_type::get_instance(val->type->base_type, mask.num_components, 1);
1542 }
1543
1544 ir_swizzle::ir_swizzle(ir_rvalue *val, unsigned x, unsigned y, unsigned z,
1545                        unsigned w, unsigned count)
1546    : ir_rvalue(ir_type_swizzle), val(val)
1547 {
1548    const unsigned components[4] = { x, y, z, w };
1549    this->init_mask(components, count);
1550 }
1551
1552 ir_swizzle::ir_swizzle(ir_rvalue *val, const unsigned *comp,
1553                        unsigned count)
1554    : ir_rvalue(ir_type_swizzle), val(val)
1555 {
1556    this->init_mask(comp, count);
1557 }
1558
1559 ir_swizzle::ir_swizzle(ir_rvalue *val, ir_swizzle_mask mask)
1560    : ir_rvalue(ir_type_swizzle), val(val), mask(mask)
1561 {
1562    this->type = glsl_type::get_instance(val->type->base_type,
1563                                         mask.num_components, 1);
1564 }
1565
1566 #define X 1
1567 #define R 5
1568 #define S 9
1569 #define I 13
1570
1571 ir_swizzle *
1572 ir_swizzle::create(ir_rvalue *val, const char *str, unsigned vector_length)
1573 {
1574    void *ctx = ralloc_parent(val);
1575
1576    /* For each possible swizzle character, this table encodes the value in
1577     * \c idx_map that represents the 0th element of the vector.  For invalid
1578     * swizzle characters (e.g., 'k'), a special value is used that will allow
1579     * detection of errors.
1580     */
1581    static const unsigned char base_idx[26] = {
1582    /* a  b  c  d  e  f  g  h  i  j  k  l  m */
1583       R, R, I, I, I, I, R, I, I, I, I, I, I,
1584    /* n  o  p  q  r  s  t  u  v  w  x  y  z */
1585       I, I, S, S, R, S, S, I, I, X, X, X, X
1586    };
1587
1588    /* Each valid swizzle character has an entry in the previous table.  This
1589     * table encodes the base index encoded in the previous table plus the actual
1590     * index of the swizzle character.  When processing swizzles, the first
1591     * character in the string is indexed in the previous table.  Each character
1592     * in the string is indexed in this table, and the value found there has the
1593     * value form the first table subtracted.  The result must be on the range
1594     * [0,3].
1595     *
1596     * For example, the string "wzyx" will get X from the first table.  Each of
1597     * the charcaters will get X+3, X+2, X+1, and X+0 from this table.  After
1598     * subtraction, the swizzle values are { 3, 2, 1, 0 }.
1599     *
1600     * The string "wzrg" will get X from the first table.  Each of the characters
1601     * will get X+3, X+2, R+0, and R+1 from this table.  After subtraction, the
1602     * swizzle values are { 3, 2, 4, 5 }.  Since 4 and 5 are outside the range
1603     * [0,3], the error is detected.
1604     */
1605    static const unsigned char idx_map[26] = {
1606    /* a    b    c    d    e    f    g    h    i    j    k    l    m */
1607       R+3, R+2, 0,   0,   0,   0,   R+1, 0,   0,   0,   0,   0,   0,
1608    /* n    o    p    q    r    s    t    u    v    w    x    y    z */
1609       0,   0,   S+2, S+3, R+0, S+0, S+1, 0,   0,   X+3, X+0, X+1, X+2
1610    };
1611
1612    int swiz_idx[4] = { 0, 0, 0, 0 };
1613    unsigned i;
1614
1615
1616    /* Validate the first character in the swizzle string and look up the base
1617     * index value as described above.
1618     */
1619    if ((str[0] < 'a') || (str[0] > 'z'))
1620       return NULL;
1621
1622    const unsigned base = base_idx[str[0] - 'a'];
1623
1624
1625    for (i = 0; (i < 4) && (str[i] != '\0'); i++) {
1626       /* Validate the next character, and, as described above, convert it to a
1627        * swizzle index.
1628        */
1629       if ((str[i] < 'a') || (str[i] > 'z'))
1630          return NULL;
1631
1632       swiz_idx[i] = idx_map[str[i] - 'a'] - base;
1633       if ((swiz_idx[i] < 0) || (swiz_idx[i] >= (int) vector_length))
1634          return NULL;
1635    }
1636
1637    if (str[i] != '\0')
1638          return NULL;
1639
1640    return new(ctx) ir_swizzle(val, swiz_idx[0], swiz_idx[1], swiz_idx[2],
1641                               swiz_idx[3], i);
1642 }
1643
1644 #undef X
1645 #undef R
1646 #undef S
1647 #undef I
1648
1649 ir_variable *
1650 ir_swizzle::variable_referenced() const
1651 {
1652    return this->val->variable_referenced();
1653 }
1654
1655
1656 bool ir_variable::temporaries_allocate_names = false;
1657
1658 const char ir_variable::tmp_name[] = "compiler_temp";
1659
1660 ir_variable::ir_variable(const struct glsl_type *type, const char *name,
1661                          ir_variable_mode mode)
1662    : ir_instruction(ir_type_variable)
1663 {
1664    this->type = type;
1665
1666    if (mode == ir_var_temporary && !ir_variable::temporaries_allocate_names)
1667       name = NULL;
1668
1669    /* The ir_variable clone method may call this constructor with name set to
1670     * tmp_name.
1671     */
1672    assert(name != NULL
1673           || mode == ir_var_temporary
1674           || mode == ir_var_function_in
1675           || mode == ir_var_function_out
1676           || mode == ir_var_function_inout);
1677    assert(name != ir_variable::tmp_name
1678           || mode == ir_var_temporary);
1679    if (mode == ir_var_temporary
1680        && (name == NULL || name == ir_variable::tmp_name)) {
1681       this->name = ir_variable::tmp_name;
1682    } else if (name == NULL ||
1683               strlen(name) < ARRAY_SIZE(this->name_storage)) {
1684       strcpy(this->name_storage, name ? name : "");
1685       this->name = this->name_storage;
1686    } else {
1687       this->name = ralloc_strdup(this, name);
1688    }
1689
1690    this->u.max_ifc_array_access = NULL;
1691
1692    this->data.explicit_location = false;
1693    this->data.has_initializer = false;
1694    this->data.location = -1;
1695    this->data.location_frac = 0;
1696    this->data.binding = 0;
1697    this->data.warn_extension_index = 0;
1698    this->constant_value = NULL;
1699    this->constant_initializer = NULL;
1700    this->data.origin_upper_left = false;
1701    this->data.pixel_center_integer = false;
1702    this->data.depth_layout = ir_depth_layout_none;
1703    this->data.used = false;
1704    this->data.always_active_io = false;
1705    this->data.read_only = false;
1706    this->data.centroid = false;
1707    this->data.sample = false;
1708    this->data.patch = false;
1709    this->data.invariant = false;
1710    this->data.how_declared = ir_var_declared_normally;
1711    this->data.mode = mode;
1712    this->data.interpolation = INTERP_MODE_NONE;
1713    this->data.max_array_access = -1;
1714    this->data.offset = 0;
1715    this->data.precision = GLSL_PRECISION_NONE;
1716    this->data.memory_read_only = false;
1717    this->data.memory_write_only = false;
1718    this->data.memory_coherent = false;
1719    this->data.memory_volatile = false;
1720    this->data.memory_restrict = false;
1721    this->data.from_ssbo_unsized_array = false;
1722    this->data.fb_fetch_output = false;
1723    this->data.bindless = false;
1724    this->data.bound = false;
1725
1726    if (type != NULL) {
1727       if (type->is_interface())
1728          this->init_interface_type(type);
1729       else if (type->without_array()->is_interface())
1730          this->init_interface_type(type->without_array());
1731    }
1732 }
1733
1734
1735 const char *
1736 interpolation_string(unsigned interpolation)
1737 {
1738    switch (interpolation) {
1739    case INTERP_MODE_NONE:          return "no";
1740    case INTERP_MODE_SMOOTH:        return "smooth";
1741    case INTERP_MODE_FLAT:          return "flat";
1742    case INTERP_MODE_NOPERSPECTIVE: return "noperspective";
1743    }
1744
1745    assert(!"Should not get here.");
1746    return "";
1747 }
1748
1749 const char *const ir_variable::warn_extension_table[] = {
1750    "",
1751    "GL_ARB_shader_stencil_export",
1752    "GL_AMD_shader_stencil_export",
1753 };
1754
1755 void
1756 ir_variable::enable_extension_warning(const char *extension)
1757 {
1758    for (unsigned i = 0; i < ARRAY_SIZE(warn_extension_table); i++) {
1759       if (strcmp(warn_extension_table[i], extension) == 0) {
1760          this->data.warn_extension_index = i;
1761          return;
1762       }
1763    }
1764
1765    assert(!"Should not get here.");
1766    this->data.warn_extension_index = 0;
1767 }
1768
1769 const char *
1770 ir_variable::get_extension_warning() const
1771 {
1772    return this->data.warn_extension_index == 0
1773       ? NULL : warn_extension_table[this->data.warn_extension_index];
1774 }
1775
1776 ir_function_signature::ir_function_signature(const glsl_type *return_type,
1777                                              builtin_available_predicate b)
1778    : ir_instruction(ir_type_function_signature),
1779      return_type(return_type), is_defined(false),
1780      intrinsic_id(ir_intrinsic_invalid), builtin_avail(b), _function(NULL)
1781 {
1782    this->origin = NULL;
1783 }
1784
1785
1786 bool
1787 ir_function_signature::is_builtin() const
1788 {
1789    return builtin_avail != NULL;
1790 }
1791
1792
1793 bool
1794 ir_function_signature::is_builtin_available(const _mesa_glsl_parse_state *state) const
1795 {
1796    /* We can't call the predicate without a state pointer, so just say that
1797     * the signature is available.  At compile time, we need the filtering,
1798     * but also receive a valid state pointer.  At link time, we're resolving
1799     * imported built-in prototypes to their definitions, which will always
1800     * be an exact match.  So we can skip the filtering.
1801     */
1802    if (state == NULL)
1803       return true;
1804
1805    assert(builtin_avail != NULL);
1806    return builtin_avail(state);
1807 }
1808
1809
1810 static bool
1811 modes_match(unsigned a, unsigned b)
1812 {
1813    if (a == b)
1814       return true;
1815
1816    /* Accept "in" vs. "const in" */
1817    if ((a == ir_var_const_in && b == ir_var_function_in) ||
1818        (b == ir_var_const_in && a == ir_var_function_in))
1819       return true;
1820
1821    return false;
1822 }
1823
1824
1825 const char *
1826 ir_function_signature::qualifiers_match(exec_list *params)
1827 {
1828    /* check that the qualifiers match. */
1829    foreach_two_lists(a_node, &this->parameters, b_node, params) {
1830       ir_variable *a = (ir_variable *) a_node;
1831       ir_variable *b = (ir_variable *) b_node;
1832
1833       if (a->data.read_only != b->data.read_only ||
1834           !modes_match(a->data.mode, b->data.mode) ||
1835           a->data.interpolation != b->data.interpolation ||
1836           a->data.centroid != b->data.centroid ||
1837           a->data.sample != b->data.sample ||
1838           a->data.patch != b->data.patch ||
1839           a->data.memory_read_only != b->data.memory_read_only ||
1840           a->data.memory_write_only != b->data.memory_write_only ||
1841           a->data.memory_coherent != b->data.memory_coherent ||
1842           a->data.memory_volatile != b->data.memory_volatile ||
1843           a->data.memory_restrict != b->data.memory_restrict) {
1844
1845          /* parameter a's qualifiers don't match */
1846          return a->name;
1847       }
1848    }
1849    return NULL;
1850 }
1851
1852
1853 void
1854 ir_function_signature::replace_parameters(exec_list *new_params)
1855 {
1856    /* Destroy all of the previous parameter information.  If the previous
1857     * parameter information comes from the function prototype, it may either
1858     * specify incorrect parameter names or not have names at all.
1859     */
1860    new_params->move_nodes_to(&parameters);
1861 }
1862
1863
1864 ir_function::ir_function(const char *name)
1865    : ir_instruction(ir_type_function)
1866 {
1867    this->subroutine_index = -1;
1868    this->name = ralloc_strdup(this, name);
1869 }
1870
1871
1872 bool
1873 ir_function::has_user_signature()
1874 {
1875    foreach_in_list(ir_function_signature, sig, &this->signatures) {
1876       if (!sig->is_builtin())
1877          return true;
1878    }
1879    return false;
1880 }
1881
1882
1883 ir_rvalue *
1884 ir_rvalue::error_value(void *mem_ctx)
1885 {
1886    ir_rvalue *v = new(mem_ctx) ir_rvalue(ir_type_unset);
1887
1888    v->type = glsl_type::error_type;
1889    return v;
1890 }
1891
1892
1893 void
1894 visit_exec_list(exec_list *list, ir_visitor *visitor)
1895 {
1896    foreach_in_list_safe(ir_instruction, node, list) {
1897       node->accept(visitor);
1898    }
1899 }
1900
1901
1902 static void
1903 steal_memory(ir_instruction *ir, void *new_ctx)
1904 {
1905    ir_variable *var = ir->as_variable();
1906    ir_function *fn = ir->as_function();
1907    ir_constant *constant = ir->as_constant();
1908    if (var != NULL && var->constant_value != NULL)
1909       steal_memory(var->constant_value, ir);
1910
1911    if (var != NULL && var->constant_initializer != NULL)
1912       steal_memory(var->constant_initializer, ir);
1913
1914    if (fn != NULL && fn->subroutine_types)
1915       ralloc_steal(new_ctx, fn->subroutine_types);
1916
1917    /* The components of aggregate constants are not visited by the normal
1918     * visitor, so steal their values by hand.
1919     */
1920    if (constant != NULL &&
1921        (constant->type->is_array() || constant->type->is_record())) {
1922       for (unsigned int i = 0; i < constant->type->length; i++) {
1923          steal_memory(constant->const_elements[i], ir);
1924       }
1925    }
1926
1927    ralloc_steal(new_ctx, ir);
1928 }
1929
1930
1931 void
1932 reparent_ir(exec_list *list, void *mem_ctx)
1933 {
1934    foreach_in_list(ir_instruction, node, list) {
1935       visit_tree(node, steal_memory, mem_ctx);
1936    }
1937 }
1938
1939
1940 static ir_rvalue *
1941 try_min_one(ir_rvalue *ir)
1942 {
1943    ir_expression *expr = ir->as_expression();
1944
1945    if (!expr || expr->operation != ir_binop_min)
1946       return NULL;
1947
1948    if (expr->operands[0]->is_one())
1949       return expr->operands[1];
1950
1951    if (expr->operands[1]->is_one())
1952       return expr->operands[0];
1953
1954    return NULL;
1955 }
1956
1957 static ir_rvalue *
1958 try_max_zero(ir_rvalue *ir)
1959 {
1960    ir_expression *expr = ir->as_expression();
1961
1962    if (!expr || expr->operation != ir_binop_max)
1963       return NULL;
1964
1965    if (expr->operands[0]->is_zero())
1966       return expr->operands[1];
1967
1968    if (expr->operands[1]->is_zero())
1969       return expr->operands[0];
1970
1971    return NULL;
1972 }
1973
1974 ir_rvalue *
1975 ir_rvalue::as_rvalue_to_saturate()
1976 {
1977    ir_expression *expr = this->as_expression();
1978
1979    if (!expr)
1980       return NULL;
1981
1982    ir_rvalue *max_zero = try_max_zero(expr);
1983    if (max_zero) {
1984       return try_min_one(max_zero);
1985    } else {
1986       ir_rvalue *min_one = try_min_one(expr);
1987       if (min_one) {
1988          return try_max_zero(min_one);
1989       }
1990    }
1991
1992    return NULL;
1993 }
1994
1995
1996 unsigned
1997 vertices_per_prim(GLenum prim)
1998 {
1999    switch (prim) {
2000    case GL_POINTS:
2001       return 1;
2002    case GL_LINES:
2003       return 2;
2004    case GL_TRIANGLES:
2005       return 3;
2006    case GL_LINES_ADJACENCY:
2007       return 4;
2008    case GL_TRIANGLES_ADJACENCY:
2009       return 6;
2010    default:
2011       assert(!"Bad primitive");
2012       return 3;
2013    }
2014 }
2015
2016 /**
2017  * Generate a string describing the mode of a variable
2018  */
2019 const char *
2020 mode_string(const ir_variable *var)
2021 {
2022    switch (var->data.mode) {
2023    case ir_var_auto:
2024       return (var->data.read_only) ? "global constant" : "global variable";
2025
2026    case ir_var_uniform:
2027       return "uniform";
2028
2029    case ir_var_shader_storage:
2030       return "buffer";
2031
2032    case ir_var_shader_in:
2033       return "shader input";
2034
2035    case ir_var_shader_out:
2036       return "shader output";
2037
2038    case ir_var_function_in:
2039    case ir_var_const_in:
2040       return "function input";
2041
2042    case ir_var_function_out:
2043       return "function output";
2044
2045    case ir_var_function_inout:
2046       return "function inout";
2047
2048    case ir_var_system_value:
2049       return "shader input";
2050
2051    case ir_var_temporary:
2052       return "compiler temporary";
2053
2054    case ir_var_mode_count:
2055       break;
2056    }
2057
2058    assert(!"Should not get here.");
2059    return "invalid variable";
2060 }