src/gallium/auxiliary/tgsi/tgsi_scan.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2008 VMware, Inc.
   4  * All Rights Reserved.
   5  * Copyright 2008 VMware, Inc.  All rights Reserved.
   6  *
   7  * Permission is hereby granted, free of charge, to any person obtaining a
   8  * copy of this software and associated documentation files (the
   9  * "Software"), to deal in the Software without restriction, including
  10  * without limitation the rights to use, copy, modify, merge, publish,
  11  * distribute, sub license, and/or sell copies of the Software, and to
  12  * permit persons to whom the Software is furnished to do so, subject to
  13  * the following conditions:
  14  *
  15  * The above copyright notice and this permission notice (including the
  16  * next paragraph) shall be included in all copies or substantial portions
  17  * of the Software.
  18  *
  19  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  20  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  21  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  22  * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
  23  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  24  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  25  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  26  *
  27  **************************************************************************/
  28
  29 /**
  30  * TGSI program scan utility.
  31  * Used to determine which registers and instructions are used by a shader.
  32  *
  33  * Authors:  Brian Paul
  34  */
  35
  36
  37 #include "util/u_debug.h"
  38 #include "util/u_math.h"
  39 #include "util/u_memory.h"
  40 #include "util/u_prim.h"
  41 #include "tgsi/tgsi_info.h"
  42 #include "tgsi/tgsi_parse.h"
  43 #include "tgsi/tgsi_util.h"
  44 #include "tgsi/tgsi_scan.h"
  45
  46
  47 static bool
  48 is_memory_file(unsigned file)
  49 {
  50    return file == TGSI_FILE_SAMPLER ||
  51           file == TGSI_FILE_SAMPLER_VIEW ||
  52           file == TGSI_FILE_IMAGE ||
  53           file == TGSI_FILE_BUFFER;
  54 }
  55
  56
  57 static bool
  58 is_mem_query_inst(unsigned opcode)
  59 {
  60    return opcode == TGSI_OPCODE_RESQ ||
  61           opcode == TGSI_OPCODE_TXQ ||
  62           opcode == TGSI_OPCODE_TXQS ||
  63           opcode == TGSI_OPCODE_TXQ_LZ ||
  64           opcode == TGSI_OPCODE_LODQ;
  65 }
  66
  67 /**
  68  * Is the opcode a "true" texture instruction which samples from a
  69  * texture map?
  70  */
  71 static bool
  72 is_texture_inst(unsigned opcode)
  73 {
  74    return (!is_mem_query_inst(opcode) &&
  75            tgsi_get_opcode_info(opcode)->is_tex);
  76 }
  77
  78
  79 /**
  80  * Is the opcode an instruction which computes a derivative explicitly or
  81  * implicitly?
  82  */
  83 static bool
  84 computes_derivative(unsigned opcode)
  85 {
  86    if (tgsi_get_opcode_info(opcode)->is_tex) {
  87       return opcode != TGSI_OPCODE_TG4 &&
  88              opcode != TGSI_OPCODE_TXD &&
  89              opcode != TGSI_OPCODE_TXF &&
  90              opcode != TGSI_OPCODE_TXL &&
  91              opcode != TGSI_OPCODE_TXL2 &&
  92              opcode != TGSI_OPCODE_TXQ &&
  93              opcode != TGSI_OPCODE_TXQ_LZ &&
  94              opcode != TGSI_OPCODE_TXQS;
  95    }
  96
  97    return opcode == TGSI_OPCODE_DDX || opcode == TGSI_OPCODE_DDX_FINE ||
  98           opcode == TGSI_OPCODE_DDY || opcode == TGSI_OPCODE_DDY_FINE ||
  99           opcode == TGSI_OPCODE_SAMPLE ||
 100           opcode == TGSI_OPCODE_SAMPLE_B ||
 101           opcode == TGSI_OPCODE_SAMPLE_C;
 102 }
 103
 104
 105 static void
 106 scan_src_operand(struct tgsi_shader_info *info,
 107                  const struct tgsi_full_instruction *fullinst,
 108                  const struct tgsi_full_src_register *src,
 109                  unsigned src_index,
 110                  unsigned usage_mask,
 111                  bool is_interp_instruction,
 112                  bool *is_mem_inst)
 113 {
 114    int ind = src->Register.Index;
 115
 116    /* Mark which inputs are effectively used */
 117    if (src->Register.File == TGSI_FILE_INPUT) {
 118       if (src->Register.Indirect) {
 119          for (ind = 0; ind < info->num_inputs; ++ind) {
 120             info->input_usage_mask[ind] |= usage_mask;
 121          }
 122       } else {
 123          assert(ind >= 0);
 124          assert(ind < PIPE_MAX_SHADER_INPUTS);
 125          info->input_usage_mask[ind] |= usage_mask;
 126       }
 127
 128       if (info->processor == PIPE_SHADER_FRAGMENT) {
 129          unsigned name, index, input;
 130
 131          if (src->Register.Indirect && src->Indirect.ArrayID)
 132             input = info->input_array_first[src->Indirect.ArrayID];
 133          else
 134             input = src->Register.Index;
 135
 136          name = info->input_semantic_name[input];
 137          index = info->input_semantic_index[input];
 138
 139          if (name == TGSI_SEMANTIC_POSITION &&
 140              (src->Register.SwizzleX == TGSI_SWIZZLE_Z ||
 141               src->Register.SwizzleY == TGSI_SWIZZLE_Z ||
 142               src->Register.SwizzleZ == TGSI_SWIZZLE_Z ||
 143               src->Register.SwizzleW == TGSI_SWIZZLE_Z))
 144             info->reads_z = TRUE;
 145
 146          if (name == TGSI_SEMANTIC_COLOR) {
 147             unsigned mask =
 148                (1 << src->Register.SwizzleX) |
 149                (1 << src->Register.SwizzleY) |
 150                (1 << src->Register.SwizzleZ) |
 151                (1 << src->Register.SwizzleW);
 152
 153             info->colors_read |= mask << (index * 4);
 154          }
 155
 156          /* Process only interpolated varyings. Don't include POSITION.
 157           * Don't include integer varyings, because they are not
 158           * interpolated. Don't process inputs interpolated by INTERP
 159           * opcodes. Those are tracked separately.
 160           */
 161          if ((!is_interp_instruction || src_index != 0) &&
 162              (name == TGSI_SEMANTIC_GENERIC ||
 163               name == TGSI_SEMANTIC_TEXCOORD ||
 164               name == TGSI_SEMANTIC_COLOR ||
 165               name == TGSI_SEMANTIC_BCOLOR ||
 166               name == TGSI_SEMANTIC_FOG ||
 167               name == TGSI_SEMANTIC_CLIPDIST)) {
 168             switch (info->input_interpolate[input]) {
 169             case TGSI_INTERPOLATE_COLOR:
 170             case TGSI_INTERPOLATE_PERSPECTIVE:
 171                switch (info->input_interpolate_loc[input]) {
 172                case TGSI_INTERPOLATE_LOC_CENTER:
 173                   info->uses_persp_center = TRUE;
 174                   break;
 175                case TGSI_INTERPOLATE_LOC_CENTROID:
 176                   info->uses_persp_centroid = TRUE;
 177                   break;
 178                case TGSI_INTERPOLATE_LOC_SAMPLE:
 179                   info->uses_persp_sample = TRUE;
 180                   break;
 181                }
 182                break;
 183             case TGSI_INTERPOLATE_LINEAR:
 184                switch (info->input_interpolate_loc[input]) {
 185                case TGSI_INTERPOLATE_LOC_CENTER:
 186                   info->uses_linear_center = TRUE;
 187                   break;
 188                case TGSI_INTERPOLATE_LOC_CENTROID:
 189                   info->uses_linear_centroid = TRUE;
 190                   break;
 191                case TGSI_INTERPOLATE_LOC_SAMPLE:
 192                   info->uses_linear_sample = TRUE;
 193                   break;
 194                }
 195                break;
 196                /* TGSI_INTERPOLATE_CONSTANT doesn't do any interpolation. */
 197             }
 198          }
 199       }
 200    }
 201
 202    /* check for indirect register reads */
 203    if (src->Register.Indirect) {
 204       info->indirect_files |= (1 << src->Register.File);
 205       info->indirect_files_read |= (1 << src->Register.File);
 206
 207       /* record indirect constant buffer indexing */
 208       if (src->Register.File == TGSI_FILE_CONSTANT) {
 209          if (src->Register.Dimension) {
 210             if (src->Dimension.Indirect)
 211                info->const_buffers_indirect = info->const_buffers_declared;
 212             else
 213                info->const_buffers_indirect |= 1u << src->Dimension.Index;
 214          } else {
 215             info->const_buffers_indirect |= 1;
 216          }
 217       }
 218    }
 219
 220    if (src->Register.Dimension && src->Dimension.Indirect)
 221       info->dim_indirect_files |= 1u << src->Register.File;
 222
 223    /* Texture samplers */
 224    if (src->Register.File == TGSI_FILE_SAMPLER) {
 225       const unsigned index = src->Register.Index;
 226
 227       assert(fullinst->Instruction.Texture);
 228       assert(index < ARRAY_SIZE(info->is_msaa_sampler));
 229       assert(index < PIPE_MAX_SAMPLERS);
 230
 231       if (is_texture_inst(fullinst->Instruction.Opcode)) {
 232          const unsigned target = fullinst->Texture.Texture;
 233          assert(target < TGSI_TEXTURE_UNKNOWN);
 234          /* for texture instructions, check that the texture instruction
 235           * target matches the previous sampler view declaration (if there
 236           * was one.)
 237           */
 238          if (info->sampler_targets[index] == TGSI_TEXTURE_UNKNOWN) {
 239             /* probably no sampler view declaration */
 240             info->sampler_targets[index] = target;
 241          } else {
 242             /* Make sure the texture instruction's sampler/target info
 243              * agrees with the sampler view declaration.
 244              */
 245             assert(info->sampler_targets[index] == target);
 246          }
 247          /* MSAA samplers */
 248          if (target == TGSI_TEXTURE_2D_MSAA ||
 249              target == TGSI_TEXTURE_2D_ARRAY_MSAA) {
 250             info->is_msaa_sampler[src->Register.Index] = TRUE;
 251          }
 252       }
 253    }
 254
 255    if (is_memory_file(src->Register.File) &&
 256        !is_mem_query_inst(fullinst->Instruction.Opcode)) {
 257       *is_mem_inst = true;
 258
 259       if (tgsi_get_opcode_info(fullinst->Instruction.Opcode)->is_store) {
 260          info->writes_memory = TRUE;
 261
 262          if (src->Register.File == TGSI_FILE_IMAGE) {
 263             if (src->Register.Indirect)
 264                info->images_writemask = info->images_declared;
 265             else
 266                info->images_writemask |= 1 << src->Register.Index;
 267          } else if (src->Register.File == TGSI_FILE_BUFFER) {
 268             if (src->Register.Indirect)
 269                info->shader_buffers_atomic = info->shader_buffers_declared;
 270             else
 271                info->shader_buffers_atomic |= 1 << src->Register.Index;
 272          }
 273       } else {
 274          if (src->Register.File == TGSI_FILE_BUFFER) {
 275             if (src->Register.Indirect)
 276                info->shader_buffers_load = info->shader_buffers_declared;
 277             else
 278                info->shader_buffers_load |= 1 << src->Register.Index;
 279          }
 280       }
 281    }
 282 }
 283
 284
 285 static void
 286 scan_instruction(struct tgsi_shader_info *info,
 287                  const struct tgsi_full_instruction *fullinst,
 288                  unsigned *current_depth)
 289 {
 290    unsigned i;
 291    bool is_mem_inst = false;
 292    bool is_interp_instruction = false;
 293
 294    assert(fullinst->Instruction.Opcode < TGSI_OPCODE_LAST);
 295    info->opcode_count[fullinst->Instruction.Opcode]++;
 296
 297    switch (fullinst->Instruction.Opcode) {
 298    case TGSI_OPCODE_IF:
 299    case TGSI_OPCODE_UIF:
 300    case TGSI_OPCODE_BGNLOOP:
 301       (*current_depth)++;
 302       info->max_depth = MAX2(info->max_depth, *current_depth);
 303       break;
 304    case TGSI_OPCODE_ENDIF:
 305    case TGSI_OPCODE_ENDLOOP:
 306       (*current_depth)--;
 307       break;
 308    default:
 309       break;
 310    }
 311
 312    if (fullinst->Instruction.Opcode == TGSI_OPCODE_INTERP_CENTROID ||
 313        fullinst->Instruction.Opcode == TGSI_OPCODE_INTERP_OFFSET ||
 314        fullinst->Instruction.Opcode == TGSI_OPCODE_INTERP_SAMPLE) {
 315       const struct tgsi_full_src_register *src0 = &fullinst->Src[0];
 316       unsigned input;
 317
 318       is_interp_instruction = true;
 319
 320       if (src0->Register.Indirect && src0->Indirect.ArrayID)
 321          input = info->input_array_first[src0->Indirect.ArrayID];
 322       else
 323          input = src0->Register.Index;
 324
 325       /* For the INTERP opcodes, the interpolation is always
 326        * PERSPECTIVE unless LINEAR is specified.
 327        */
 328       switch (info->input_interpolate[input]) {
 329       case TGSI_INTERPOLATE_COLOR:
 330       case TGSI_INTERPOLATE_CONSTANT:
 331       case TGSI_INTERPOLATE_PERSPECTIVE:
 332          switch (fullinst->Instruction.Opcode) {
 333          case TGSI_OPCODE_INTERP_CENTROID:
 334             info->uses_persp_opcode_interp_centroid = TRUE;
 335             break;
 336          case TGSI_OPCODE_INTERP_OFFSET:
 337             info->uses_persp_opcode_interp_offset = TRUE;
 338             break;
 339          case TGSI_OPCODE_INTERP_SAMPLE:
 340             info->uses_persp_opcode_interp_sample = TRUE;
 341             break;
 342          }
 343          break;
 344
 345       case TGSI_INTERPOLATE_LINEAR:
 346          switch (fullinst->Instruction.Opcode) {
 347          case TGSI_OPCODE_INTERP_CENTROID:
 348             info->uses_linear_opcode_interp_centroid = TRUE;
 349             break;
 350          case TGSI_OPCODE_INTERP_OFFSET:
 351             info->uses_linear_opcode_interp_offset = TRUE;
 352             break;
 353          case TGSI_OPCODE_INTERP_SAMPLE:
 354             info->uses_linear_opcode_interp_sample = TRUE;
 355             break;
 356          }
 357          break;
 358       }
 359    }
 360
 361    if (fullinst->Instruction.Opcode >= TGSI_OPCODE_F2D &&
 362        fullinst->Instruction.Opcode <= TGSI_OPCODE_DSSG)
 363       info->uses_doubles = TRUE;
 364
 365    for (i = 0; i < fullinst->Instruction.NumSrcRegs; i++) {
 366       scan_src_operand(info, fullinst, &fullinst->Src[i], i,
 367                        tgsi_util_get_inst_usage_mask(fullinst, i),
 368                        is_interp_instruction, &is_mem_inst);
 369    }
 370
 371    if (fullinst->Instruction.Texture) {
 372       for (i = 0; i < fullinst->Texture.NumOffsets; i++) {
 373          struct tgsi_full_src_register src = {{0}};
 374
 375          src.Register.File = fullinst->TexOffsets[i].File;
 376          src.Register.Index = fullinst->TexOffsets[i].Index;
 377          src.Register.SwizzleX = fullinst->TexOffsets[i].SwizzleX;
 378          src.Register.SwizzleY = fullinst->TexOffsets[i].SwizzleY;
 379          src.Register.SwizzleZ = fullinst->TexOffsets[i].SwizzleZ;
 380
 381          /* The usage mask is suboptimal but should be safe. */
 382          scan_src_operand(info, fullinst, &src, 0, TGSI_WRITEMASK_XYZ,
 383                           false, &is_mem_inst);
 384       }
 385    }
 386
 387    /* check for indirect register writes */
 388    for (i = 0; i < fullinst->Instruction.NumDstRegs; i++) {
 389       const struct tgsi_full_dst_register *dst = &fullinst->Dst[i];
 390       if (dst->Register.Indirect) {
 391          info->indirect_files |= (1 << dst->Register.File);
 392          info->indirect_files_written |= (1 << dst->Register.File);
 393       }
 394
 395       if (dst->Register.Dimension && dst->Dimension.Indirect)
 396          info->dim_indirect_files |= 1u << dst->Register.File;
 397
 398       if (is_memory_file(dst->Register.File)) {
 399          assert(fullinst->Instruction.Opcode == TGSI_OPCODE_STORE);
 400
 401          is_mem_inst = true;
 402          info->writes_memory = TRUE;
 403
 404          if (dst->Register.File == TGSI_FILE_IMAGE) {
 405             if (dst->Register.Indirect)
 406                info->images_writemask = info->images_declared;
 407             else
 408                info->images_writemask |= 1 << dst->Register.Index;
 409          } else if (dst->Register.File == TGSI_FILE_BUFFER) {
 410             if (dst->Register.Indirect)
 411                info->shader_buffers_store = info->shader_buffers_declared;
 412             else
 413                info->shader_buffers_store |= 1 << dst->Register.Index;
 414          }
 415       }
 416    }
 417
 418    if (is_mem_inst)
 419       info->num_memory_instructions++;
 420
 421    if (computes_derivative(fullinst->Instruction.Opcode))
 422       info->uses_derivatives = true;
 423
 424    info->num_instructions++;
 425 }
 426
 427
 428 static void
 429 scan_declaration(struct tgsi_shader_info *info,
 430                  const struct tgsi_full_declaration *fulldecl)
 431 {
 432    const uint file = fulldecl->Declaration.File;
 433    const unsigned procType = info->processor;
 434    uint reg;
 435
 436    if (fulldecl->Declaration.Array) {
 437       unsigned array_id = fulldecl->Array.ArrayID;
 438
 439       switch (file) {
 440       case TGSI_FILE_INPUT:
 441          assert(array_id < ARRAY_SIZE(info->input_array_first));
 442          info->input_array_first[array_id] = fulldecl->Range.First;
 443          info->input_array_last[array_id] = fulldecl->Range.Last;
 444          break;
 445       case TGSI_FILE_OUTPUT:
 446          assert(array_id < ARRAY_SIZE(info->output_array_first));
 447          info->output_array_first[array_id] = fulldecl->Range.First;
 448          info->output_array_last[array_id] = fulldecl->Range.Last;
 449          break;
 450       }
 451       info->array_max[file] = MAX2(info->array_max[file], array_id);
 452    }
 453
 454    for (reg = fulldecl->Range.First; reg <= fulldecl->Range.Last; reg++) {
 455       unsigned semName = fulldecl->Semantic.Name;
 456       unsigned semIndex = fulldecl->Semantic.Index +
 457          (reg - fulldecl->Range.First);
 458       int buffer;
 459       unsigned index, target, type;
 460
 461       /* only first 32 regs will appear in this bitfield */
 462       info->file_mask[file] |= (1 << reg);
 463       info->file_count[file]++;
 464       info->file_max[file] = MAX2(info->file_max[file], (int)reg);
 465
 466       switch (file) {
 467       case TGSI_FILE_CONSTANT:
 468          buffer = 0;
 469
 470          if (fulldecl->Declaration.Dimension)
 471             buffer = fulldecl->Dim.Index2D;
 472
 473          info->const_file_max[buffer] =
 474             MAX2(info->const_file_max[buffer], (int)reg);
 475          info->const_buffers_declared |= 1u << buffer;
 476          break;
 477
 478       case TGSI_FILE_IMAGE:
 479          info->images_declared |= 1u << reg;
 480          if (fulldecl->Image.Resource == TGSI_TEXTURE_BUFFER)
 481             info->images_buffers |= 1 << reg;
 482          break;
 483
 484       case TGSI_FILE_BUFFER:
 485          info->shader_buffers_declared |= 1u << reg;
 486          break;
 487
 488       case TGSI_FILE_INPUT:
 489          info->input_semantic_name[reg] = (ubyte) semName;
 490          info->input_semantic_index[reg] = (ubyte) semIndex;
 491          info->input_interpolate[reg] = (ubyte)fulldecl->Interp.Interpolate;
 492          info->input_interpolate_loc[reg] = (ubyte)fulldecl->Interp.Location;
 493          info->input_cylindrical_wrap[reg] = (ubyte)fulldecl->Interp.CylindricalWrap;
 494
 495          /* Vertex shaders can have inputs with holes between them. */
 496          info->num_inputs = MAX2(info->num_inputs, reg + 1);
 497
 498          if (semName == TGSI_SEMANTIC_PRIMID)
 499             info->uses_primid = TRUE;
 500          else if (procType == PIPE_SHADER_FRAGMENT) {
 501             if (semName == TGSI_SEMANTIC_POSITION)
 502                info->reads_position = TRUE;
 503             else if (semName == TGSI_SEMANTIC_FACE)
 504                info->uses_frontface = TRUE;
 505          }
 506          break;
 507
 508       case TGSI_FILE_SYSTEM_VALUE:
 509          index = fulldecl->Range.First;
 510
 511          info->system_value_semantic_name[index] = semName;
 512          info->num_system_values = MAX2(info->num_system_values, index + 1);
 513
 514          switch (semName) {
 515          case TGSI_SEMANTIC_INSTANCEID:
 516             info->uses_instanceid = TRUE;
 517             break;
 518          case TGSI_SEMANTIC_VERTEXID:
 519             info->uses_vertexid = TRUE;
 520             break;
 521          case TGSI_SEMANTIC_VERTEXID_NOBASE:
 522             info->uses_vertexid_nobase = TRUE;
 523             break;
 524          case TGSI_SEMANTIC_BASEVERTEX:
 525             info->uses_basevertex = TRUE;
 526             break;
 527          case TGSI_SEMANTIC_PRIMID:
 528             info->uses_primid = TRUE;
 529             break;
 530          case TGSI_SEMANTIC_INVOCATIONID:
 531             info->uses_invocationid = TRUE;
 532             break;
 533          case TGSI_SEMANTIC_POSITION:
 534             info->reads_position = TRUE;
 535             break;
 536          case TGSI_SEMANTIC_FACE:
 537             info->uses_frontface = TRUE;
 538             break;
 539          case TGSI_SEMANTIC_SAMPLEMASK:
 540             info->reads_samplemask = TRUE;
 541             break;
 542          }
 543          break;
 544
 545       case TGSI_FILE_OUTPUT:
 546          info->output_semantic_name[reg] = (ubyte) semName;
 547          info->output_semantic_index[reg] = (ubyte) semIndex;
 548          info->num_outputs = MAX2(info->num_outputs, reg + 1);
 549
 550          switch (semName) {
 551          case TGSI_SEMANTIC_PRIMID:
 552             info->writes_primid = true;
 553             break;
 554          case TGSI_SEMANTIC_VIEWPORT_INDEX:
 555             info->writes_viewport_index = true;
 556             break;
 557          case TGSI_SEMANTIC_LAYER:
 558             info->writes_layer = true;
 559             break;
 560          case TGSI_SEMANTIC_PSIZE:
 561             info->writes_psize = true;
 562             break;
 563          case TGSI_SEMANTIC_CLIPVERTEX:
 564             info->writes_clipvertex = true;
 565             break;
 566          case TGSI_SEMANTIC_COLOR:
 567             info->colors_written |= 1 << semIndex;
 568             break;
 569          case TGSI_SEMANTIC_STENCIL:
 570             info->writes_stencil = true;
 571             break;
 572          case TGSI_SEMANTIC_SAMPLEMASK:
 573             info->writes_samplemask = true;
 574             break;
 575          case TGSI_SEMANTIC_EDGEFLAG:
 576             info->writes_edgeflag = true;
 577             break;
 578          case TGSI_SEMANTIC_POSITION:
 579             if (procType == PIPE_SHADER_FRAGMENT)
 580                info->writes_z = true;
 581             else
 582                info->writes_position = true;
 583             break;
 584          }
 585          break;
 586
 587       case TGSI_FILE_SAMPLER:
 588          STATIC_ASSERT(sizeof(info->samplers_declared) * 8 >= PIPE_MAX_SAMPLERS);
 589          info->samplers_declared |= 1u << reg;
 590          break;
 591
 592       case TGSI_FILE_SAMPLER_VIEW:
 593          target = fulldecl->SamplerView.Resource;
 594          type = fulldecl->SamplerView.ReturnTypeX;
 595
 596          assert(target < TGSI_TEXTURE_UNKNOWN);
 597          if (info->sampler_targets[reg] == TGSI_TEXTURE_UNKNOWN) {
 598             /* Save sampler target for this sampler index */
 599             info->sampler_targets[reg] = target;
 600             info->sampler_type[reg] = type;
 601          } else {
 602             /* if previously declared, make sure targets agree */
 603             assert(info->sampler_targets[reg] == target);
 604             assert(info->sampler_type[reg] == type);
 605          }
 606          break;
 607       }
 608    }
 609 }
 610
 611
 612 static void
 613 scan_immediate(struct tgsi_shader_info *info)
 614 {
 615    uint reg = info->immediate_count++;
 616    uint file = TGSI_FILE_IMMEDIATE;
 617
 618    info->file_mask[file] |= (1 << reg);
 619    info->file_count[file]++;
 620    info->file_max[file] = MAX2(info->file_max[file], (int)reg);
 621 }
 622
 623
 624 static void
 625 scan_property(struct tgsi_shader_info *info,
 626               const struct tgsi_full_property *fullprop)
 627 {
 628    unsigned name = fullprop->Property.PropertyName;
 629    unsigned value = fullprop->u[0].Data;
 630
 631    assert(name < ARRAY_SIZE(info->properties));
 632    info->properties[name] = value;
 633
 634    switch (name) {
 635    case TGSI_PROPERTY_NUM_CLIPDIST_ENABLED:
 636       info->num_written_clipdistance = value;
 637       info->clipdist_writemask |= (1 << value) - 1;
 638       break;
 639    case TGSI_PROPERTY_NUM_CULLDIST_ENABLED:
 640       info->num_written_culldistance = value;
 641       info->culldist_writemask |= (1 << value) - 1;
 642       break;
 643    }
 644 }
 645
 646
 647 /**
 648  * Scan the given TGSI shader to collect information such as number of
 649  * registers used, special instructions used, etc.
 650  * \return info  the result of the scan
 651  */
 652 void
 653 tgsi_scan_shader(const struct tgsi_token *tokens,
 654                  struct tgsi_shader_info *info)
 655 {
 656    uint procType, i;
 657    struct tgsi_parse_context parse;
 658    unsigned current_depth = 0;
 659
 660    memset(info, 0, sizeof(*info));
 661    for (i = 0; i < TGSI_FILE_COUNT; i++)
 662       info->file_max[i] = -1;
 663    for (i = 0; i < ARRAY_SIZE(info->const_file_max); i++)
 664       info->const_file_max[i] = -1;
 665    info->properties[TGSI_PROPERTY_GS_INVOCATIONS] = 1;
 666    for (i = 0; i < ARRAY_SIZE(info->sampler_targets); i++)
 667       info->sampler_targets[i] = TGSI_TEXTURE_UNKNOWN;
 668
 669    /**
 670     ** Setup to begin parsing input shader
 671     **/
 672    if (tgsi_parse_init( &parse, tokens ) != TGSI_PARSE_OK) {
 673       debug_printf("tgsi_parse_init() failed in tgsi_scan_shader()!\n");
 674       return;
 675    }
 676    procType = parse.FullHeader.Processor.Processor;
 677    assert(procType == PIPE_SHADER_FRAGMENT ||
 678           procType == PIPE_SHADER_VERTEX ||
 679           procType == PIPE_SHADER_GEOMETRY ||
 680           procType == PIPE_SHADER_TESS_CTRL ||
 681           procType == PIPE_SHADER_TESS_EVAL ||
 682           procType == PIPE_SHADER_COMPUTE);
 683    info->processor = procType;
 684
 685    /**
 686     ** Loop over incoming program tokens/instructions
 687     */
 688    while (!tgsi_parse_end_of_tokens(&parse)) {
 689       info->num_tokens++;
 690
 691       tgsi_parse_token( &parse );
 692
 693       switch( parse.FullToken.Token.Type ) {
 694       case TGSI_TOKEN_TYPE_INSTRUCTION:
 695          scan_instruction(info, &parse.FullToken.FullInstruction,
 696                           &current_depth);
 697          break;
 698       case TGSI_TOKEN_TYPE_DECLARATION:
 699          scan_declaration(info, &parse.FullToken.FullDeclaration);
 700          break;
 701       case TGSI_TOKEN_TYPE_IMMEDIATE:
 702          scan_immediate(info);
 703          break;
 704       case TGSI_TOKEN_TYPE_PROPERTY:
 705          scan_property(info, &parse.FullToken.FullProperty);
 706          break;
 707       default:
 708          assert(!"Unexpected TGSI token type");
 709       }
 710    }
 711
 712    info->uses_kill = (info->opcode_count[TGSI_OPCODE_KILL_IF] ||
 713                       info->opcode_count[TGSI_OPCODE_KILL]);
 714
 715    /* The dimensions of the IN decleration in geometry shader have
 716     * to be deduced from the type of the input primitive.
 717     */
 718    if (procType == PIPE_SHADER_GEOMETRY) {
 719       unsigned input_primitive =
 720             info->properties[TGSI_PROPERTY_GS_INPUT_PRIM];
 721       int num_verts = u_vertices_per_prim(input_primitive);
 722       int j;
 723       info->file_count[TGSI_FILE_INPUT] = num_verts;
 724       info->file_max[TGSI_FILE_INPUT] =
 725             MAX2(info->file_max[TGSI_FILE_INPUT], num_verts - 1);
 726       for (j = 0; j < num_verts; ++j) {
 727          info->file_mask[TGSI_FILE_INPUT] |= (1 << j);
 728       }
 729    }
 730
 731    tgsi_parse_free(&parse);
 732 }
 733
 734 /**
 735  * Collect information about the arrays of a given register file.
 736  *
 737  * @param tokens TGSI shader
 738  * @param file the register file to scan through
 739  * @param max_array_id number of entries in @p arrays; should be equal to the
 740  *                     highest array id, i.e. tgsi_shader_info::array_max[file].
 741  * @param arrays info for array of each ID will be written to arrays[ID - 1].
 742  */
 743 void
 744 tgsi_scan_arrays(const struct tgsi_token *tokens,
 745                  unsigned file,
 746                  unsigned max_array_id,
 747                  struct tgsi_array_info *arrays)
 748 {
 749    struct tgsi_parse_context parse;
 750
 751    if (tgsi_parse_init(&parse, tokens) != TGSI_PARSE_OK) {
 752       debug_printf("tgsi_parse_init() failed in tgsi_scan_arrays()!\n");
 753       return;
 754    }
 755
 756    memset(arrays, 0, sizeof(arrays[0]) * max_array_id);
 757
 758    while (!tgsi_parse_end_of_tokens(&parse)) {
 759       struct tgsi_full_instruction *inst;
 760
 761       tgsi_parse_token(&parse);
 762
 763       if (parse.FullToken.Token.Type == TGSI_TOKEN_TYPE_DECLARATION) {
 764          struct tgsi_full_declaration *decl = &parse.FullToken.FullDeclaration;
 765
 766          if (decl->Declaration.Array && decl->Declaration.File == file &&
 767              decl->Array.ArrayID > 0 && decl->Array.ArrayID <= max_array_id) {
 768             struct tgsi_array_info *array = &arrays[decl->Array.ArrayID - 1];
 769             assert(!array->declared);
 770             array->declared = true;
 771             array->range = decl->Range;
 772          }
 773       }
 774
 775       if (parse.FullToken.Token.Type != TGSI_TOKEN_TYPE_INSTRUCTION)
 776          continue;
 777
 778       inst = &parse.FullToken.FullInstruction;
 779       for (unsigned i = 0; i < inst->Instruction.NumDstRegs; i++) {
 780          const struct tgsi_full_dst_register *dst = &inst->Dst[i];
 781          if (dst->Register.File != file)
 782             continue;
 783
 784          if (dst->Register.Indirect) {
 785             if (dst->Indirect.ArrayID > 0 &&
 786                 dst->Indirect.ArrayID <= max_array_id) {
 787                arrays[dst->Indirect.ArrayID - 1].writemask |= dst->Register.WriteMask;
 788             } else {
 789                /* Indirect writes without an ArrayID can write anywhere. */
 790                for (unsigned j = 0; j < max_array_id; ++j)
 791                   arrays[j].writemask |= dst->Register.WriteMask;
 792             }
 793          } else {
 794             /* Check whether the write falls into any of the arrays anyway. */
 795             for (unsigned j = 0; j < max_array_id; ++j) {
 796                struct tgsi_array_info *array = &arrays[j];
 797                if (array->declared &&
 798                    dst->Register.Index >= array->range.First &&
 799                    dst->Register.Index <= array->range.Last)
 800                   array->writemask |= dst->Register.WriteMask;
 801             }
 802          }
 803       }
 804    }
 805
 806    tgsi_parse_free(&parse);
 807
 808    return;
 809 }
 810
 811
 812 /**
 813  * Check if the given shader is a "passthrough" shader consisting of only
 814  * MOV instructions of the form:  MOV OUT[n], IN[n]
 815  *
 816  */
 817 boolean
 818 tgsi_is_passthrough_shader(const struct tgsi_token *tokens)
 819 {
 820    struct tgsi_parse_context parse;
 821
 822    /**
 823     ** Setup to begin parsing input shader
 824     **/
 825    if (tgsi_parse_init(&parse, tokens) != TGSI_PARSE_OK) {
 826       debug_printf("tgsi_parse_init() failed in tgsi_is_passthrough_shader()!\n");
 827       return FALSE;
 828    }
 829
 830    /**
 831     ** Loop over incoming program tokens/instructions
 832     */
 833    while (!tgsi_parse_end_of_tokens(&parse)) {
 834
 835       tgsi_parse_token(&parse);
 836
 837       switch (parse.FullToken.Token.Type) {
 838       case TGSI_TOKEN_TYPE_INSTRUCTION:
 839          {
 840             struct tgsi_full_instruction *fullinst =
 841                &parse.FullToken.FullInstruction;
 842             const struct tgsi_full_src_register *src =
 843                &fullinst->Src[0];
 844             const struct tgsi_full_dst_register *dst =
 845                &fullinst->Dst[0];
 846
 847             /* Do a whole bunch of checks for a simple move */
 848             if (fullinst->Instruction.Opcode != TGSI_OPCODE_MOV ||
 849                 (src->Register.File != TGSI_FILE_INPUT &&
 850                  src->Register.File != TGSI_FILE_SYSTEM_VALUE) ||
 851                 dst->Register.File != TGSI_FILE_OUTPUT ||
 852                 src->Register.Index != dst->Register.Index ||
 853
 854                 src->Register.Negate ||
 855                 src->Register.Absolute ||
 856
 857                 src->Register.SwizzleX != TGSI_SWIZZLE_X ||
 858                 src->Register.SwizzleY != TGSI_SWIZZLE_Y ||
 859                 src->Register.SwizzleZ != TGSI_SWIZZLE_Z ||
 860                 src->Register.SwizzleW != TGSI_SWIZZLE_W ||
 861
 862                 dst->Register.WriteMask != TGSI_WRITEMASK_XYZW)
 863             {
 864                tgsi_parse_free(&parse);
 865                return FALSE;
 866             }
 867          }
 868          break;
 869
 870       case TGSI_TOKEN_TYPE_DECLARATION:
 871          /* fall-through */
 872       case TGSI_TOKEN_TYPE_IMMEDIATE:
 873          /* fall-through */
 874       case TGSI_TOKEN_TYPE_PROPERTY:
 875          /* fall-through */
 876       default:
 877          ; /* no-op */
 878       }
 879    }
 880
 881    tgsi_parse_free(&parse);
 882
 883    /* if we get here, it's a pass-through shader */
 884    return TRUE;
 885 }