OSDN Git Service

tgsi/scan: collect information about output vertex streams
[android-x86/external-mesa.git] / 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          if (fulldecl->Declaration.UsageMask & TGSI_WRITEMASK_X) {
551             info->output_streams[reg] |= (ubyte)fulldecl->Semantic.StreamX;
552             info->num_stream_output_components[fulldecl->Semantic.StreamX]++;
553          }
554          if (fulldecl->Declaration.UsageMask & TGSI_WRITEMASK_Y) {
555             info->output_streams[reg] |= (ubyte)fulldecl->Semantic.StreamY << 2;
556             info->num_stream_output_components[fulldecl->Semantic.StreamY]++;
557          }
558          if (fulldecl->Declaration.UsageMask & TGSI_WRITEMASK_Z) {
559             info->output_streams[reg] |= (ubyte)fulldecl->Semantic.StreamZ << 4;
560             info->num_stream_output_components[fulldecl->Semantic.StreamZ]++;
561          }
562          if (fulldecl->Declaration.UsageMask & TGSI_WRITEMASK_W) {
563             info->output_streams[reg] |= (ubyte)fulldecl->Semantic.StreamW << 6;
564             info->num_stream_output_components[fulldecl->Semantic.StreamW]++;
565          }
566
567          switch (semName) {
568          case TGSI_SEMANTIC_PRIMID:
569             info->writes_primid = true;
570             break;
571          case TGSI_SEMANTIC_VIEWPORT_INDEX:
572             info->writes_viewport_index = true;
573             break;
574          case TGSI_SEMANTIC_LAYER:
575             info->writes_layer = true;
576             break;
577          case TGSI_SEMANTIC_PSIZE:
578             info->writes_psize = true;
579             break;
580          case TGSI_SEMANTIC_CLIPVERTEX:
581             info->writes_clipvertex = true;
582             break;
583          case TGSI_SEMANTIC_COLOR:
584             info->colors_written |= 1 << semIndex;
585             break;
586          case TGSI_SEMANTIC_STENCIL:
587             info->writes_stencil = true;
588             break;
589          case TGSI_SEMANTIC_SAMPLEMASK:
590             info->writes_samplemask = true;
591             break;
592          case TGSI_SEMANTIC_EDGEFLAG:
593             info->writes_edgeflag = true;
594             break;
595          case TGSI_SEMANTIC_POSITION:
596             if (procType == PIPE_SHADER_FRAGMENT)
597                info->writes_z = true;
598             else
599                info->writes_position = true;
600             break;
601          }
602          break;
603
604       case TGSI_FILE_SAMPLER:
605          STATIC_ASSERT(sizeof(info->samplers_declared) * 8 >= PIPE_MAX_SAMPLERS);
606          info->samplers_declared |= 1u << reg;
607          break;
608
609       case TGSI_FILE_SAMPLER_VIEW:
610          target = fulldecl->SamplerView.Resource;
611          type = fulldecl->SamplerView.ReturnTypeX;
612
613          assert(target < TGSI_TEXTURE_UNKNOWN);
614          if (info->sampler_targets[reg] == TGSI_TEXTURE_UNKNOWN) {
615             /* Save sampler target for this sampler index */
616             info->sampler_targets[reg] = target;
617             info->sampler_type[reg] = type;
618          } else {
619             /* if previously declared, make sure targets agree */
620             assert(info->sampler_targets[reg] == target);
621             assert(info->sampler_type[reg] == type);
622          }
623          break;
624       }
625    }
626 }
627
628
629 static void
630 scan_immediate(struct tgsi_shader_info *info)
631 {
632    uint reg = info->immediate_count++;
633    uint file = TGSI_FILE_IMMEDIATE;
634
635    info->file_mask[file] |= (1 << reg);
636    info->file_count[file]++;
637    info->file_max[file] = MAX2(info->file_max[file], (int)reg);
638 }
639
640
641 static void
642 scan_property(struct tgsi_shader_info *info,
643               const struct tgsi_full_property *fullprop)
644 {
645    unsigned name = fullprop->Property.PropertyName;
646    unsigned value = fullprop->u[0].Data;
647
648    assert(name < ARRAY_SIZE(info->properties));
649    info->properties[name] = value;
650
651    switch (name) {
652    case TGSI_PROPERTY_NUM_CLIPDIST_ENABLED:
653       info->num_written_clipdistance = value;
654       info->clipdist_writemask |= (1 << value) - 1;
655       break;
656    case TGSI_PROPERTY_NUM_CULLDIST_ENABLED:
657       info->num_written_culldistance = value;
658       info->culldist_writemask |= (1 << value) - 1;
659       break;
660    }
661 }
662
663
664 /**
665  * Scan the given TGSI shader to collect information such as number of
666  * registers used, special instructions used, etc.
667  * \return info  the result of the scan
668  */
669 void
670 tgsi_scan_shader(const struct tgsi_token *tokens,
671                  struct tgsi_shader_info *info)
672 {
673    uint procType, i;
674    struct tgsi_parse_context parse;
675    unsigned current_depth = 0;
676
677    memset(info, 0, sizeof(*info));
678    for (i = 0; i < TGSI_FILE_COUNT; i++)
679       info->file_max[i] = -1;
680    for (i = 0; i < ARRAY_SIZE(info->const_file_max); i++)
681       info->const_file_max[i] = -1;
682    info->properties[TGSI_PROPERTY_GS_INVOCATIONS] = 1;
683    for (i = 0; i < ARRAY_SIZE(info->sampler_targets); i++)
684       info->sampler_targets[i] = TGSI_TEXTURE_UNKNOWN;
685
686    /**
687     ** Setup to begin parsing input shader
688     **/
689    if (tgsi_parse_init( &parse, tokens ) != TGSI_PARSE_OK) {
690       debug_printf("tgsi_parse_init() failed in tgsi_scan_shader()!\n");
691       return;
692    }
693    procType = parse.FullHeader.Processor.Processor;
694    assert(procType == PIPE_SHADER_FRAGMENT ||
695           procType == PIPE_SHADER_VERTEX ||
696           procType == PIPE_SHADER_GEOMETRY ||
697           procType == PIPE_SHADER_TESS_CTRL ||
698           procType == PIPE_SHADER_TESS_EVAL ||
699           procType == PIPE_SHADER_COMPUTE);
700    info->processor = procType;
701
702    /**
703     ** Loop over incoming program tokens/instructions
704     */
705    while (!tgsi_parse_end_of_tokens(&parse)) {
706       info->num_tokens++;
707
708       tgsi_parse_token( &parse );
709
710       switch( parse.FullToken.Token.Type ) {
711       case TGSI_TOKEN_TYPE_INSTRUCTION:
712          scan_instruction(info, &parse.FullToken.FullInstruction,
713                           &current_depth);
714          break;
715       case TGSI_TOKEN_TYPE_DECLARATION:
716          scan_declaration(info, &parse.FullToken.FullDeclaration);
717          break;
718       case TGSI_TOKEN_TYPE_IMMEDIATE:
719          scan_immediate(info);
720          break;
721       case TGSI_TOKEN_TYPE_PROPERTY:
722          scan_property(info, &parse.FullToken.FullProperty);
723          break;
724       default:
725          assert(!"Unexpected TGSI token type");
726       }
727    }
728
729    info->uses_kill = (info->opcode_count[TGSI_OPCODE_KILL_IF] ||
730                       info->opcode_count[TGSI_OPCODE_KILL]);
731
732    /* The dimensions of the IN decleration in geometry shader have
733     * to be deduced from the type of the input primitive.
734     */
735    if (procType == PIPE_SHADER_GEOMETRY) {
736       unsigned input_primitive =
737             info->properties[TGSI_PROPERTY_GS_INPUT_PRIM];
738       int num_verts = u_vertices_per_prim(input_primitive);
739       int j;
740       info->file_count[TGSI_FILE_INPUT] = num_verts;
741       info->file_max[TGSI_FILE_INPUT] =
742             MAX2(info->file_max[TGSI_FILE_INPUT], num_verts - 1);
743       for (j = 0; j < num_verts; ++j) {
744          info->file_mask[TGSI_FILE_INPUT] |= (1 << j);
745       }
746    }
747
748    tgsi_parse_free(&parse);
749 }
750
751 /**
752  * Collect information about the arrays of a given register file.
753  *
754  * @param tokens TGSI shader
755  * @param file the register file to scan through
756  * @param max_array_id number of entries in @p arrays; should be equal to the
757  *                     highest array id, i.e. tgsi_shader_info::array_max[file].
758  * @param arrays info for array of each ID will be written to arrays[ID - 1].
759  */
760 void
761 tgsi_scan_arrays(const struct tgsi_token *tokens,
762                  unsigned file,
763                  unsigned max_array_id,
764                  struct tgsi_array_info *arrays)
765 {
766    struct tgsi_parse_context parse;
767
768    if (tgsi_parse_init(&parse, tokens) != TGSI_PARSE_OK) {
769       debug_printf("tgsi_parse_init() failed in tgsi_scan_arrays()!\n");
770       return;
771    }
772
773    memset(arrays, 0, sizeof(arrays[0]) * max_array_id);
774
775    while (!tgsi_parse_end_of_tokens(&parse)) {
776       struct tgsi_full_instruction *inst;
777
778       tgsi_parse_token(&parse);
779
780       if (parse.FullToken.Token.Type == TGSI_TOKEN_TYPE_DECLARATION) {
781          struct tgsi_full_declaration *decl = &parse.FullToken.FullDeclaration;
782
783          if (decl->Declaration.Array && decl->Declaration.File == file &&
784              decl->Array.ArrayID > 0 && decl->Array.ArrayID <= max_array_id) {
785             struct tgsi_array_info *array = &arrays[decl->Array.ArrayID - 1];
786             assert(!array->declared);
787             array->declared = true;
788             array->range = decl->Range;
789          }
790       }
791
792       if (parse.FullToken.Token.Type != TGSI_TOKEN_TYPE_INSTRUCTION)
793          continue;
794
795       inst = &parse.FullToken.FullInstruction;
796       for (unsigned i = 0; i < inst->Instruction.NumDstRegs; i++) {
797          const struct tgsi_full_dst_register *dst = &inst->Dst[i];
798          if (dst->Register.File != file)
799             continue;
800
801          if (dst->Register.Indirect) {
802             if (dst->Indirect.ArrayID > 0 &&
803                 dst->Indirect.ArrayID <= max_array_id) {
804                arrays[dst->Indirect.ArrayID - 1].writemask |= dst->Register.WriteMask;
805             } else {
806                /* Indirect writes without an ArrayID can write anywhere. */
807                for (unsigned j = 0; j < max_array_id; ++j)
808                   arrays[j].writemask |= dst->Register.WriteMask;
809             }
810          } else {
811             /* Check whether the write falls into any of the arrays anyway. */
812             for (unsigned j = 0; j < max_array_id; ++j) {
813                struct tgsi_array_info *array = &arrays[j];
814                if (array->declared &&
815                    dst->Register.Index >= array->range.First &&
816                    dst->Register.Index <= array->range.Last)
817                   array->writemask |= dst->Register.WriteMask;
818             }
819          }
820       }
821    }
822
823    tgsi_parse_free(&parse);
824
825    return;
826 }
827
828
829 /**
830  * Check if the given shader is a "passthrough" shader consisting of only
831  * MOV instructions of the form:  MOV OUT[n], IN[n]
832  *  
833  */
834 boolean
835 tgsi_is_passthrough_shader(const struct tgsi_token *tokens)
836 {
837    struct tgsi_parse_context parse;
838
839    /**
840     ** Setup to begin parsing input shader
841     **/
842    if (tgsi_parse_init(&parse, tokens) != TGSI_PARSE_OK) {
843       debug_printf("tgsi_parse_init() failed in tgsi_is_passthrough_shader()!\n");
844       return FALSE;
845    }
846
847    /**
848     ** Loop over incoming program tokens/instructions
849     */
850    while (!tgsi_parse_end_of_tokens(&parse)) {
851
852       tgsi_parse_token(&parse);
853
854       switch (parse.FullToken.Token.Type) {
855       case TGSI_TOKEN_TYPE_INSTRUCTION:
856          {
857             struct tgsi_full_instruction *fullinst =
858                &parse.FullToken.FullInstruction;
859             const struct tgsi_full_src_register *src =
860                &fullinst->Src[0];
861             const struct tgsi_full_dst_register *dst =
862                &fullinst->Dst[0];
863
864             /* Do a whole bunch of checks for a simple move */
865             if (fullinst->Instruction.Opcode != TGSI_OPCODE_MOV ||
866                 (src->Register.File != TGSI_FILE_INPUT &&
867                  src->Register.File != TGSI_FILE_SYSTEM_VALUE) ||
868                 dst->Register.File != TGSI_FILE_OUTPUT ||
869                 src->Register.Index != dst->Register.Index ||
870
871                 src->Register.Negate ||
872                 src->Register.Absolute ||
873
874                 src->Register.SwizzleX != TGSI_SWIZZLE_X ||
875                 src->Register.SwizzleY != TGSI_SWIZZLE_Y ||
876                 src->Register.SwizzleZ != TGSI_SWIZZLE_Z ||
877                 src->Register.SwizzleW != TGSI_SWIZZLE_W ||
878
879                 dst->Register.WriteMask != TGSI_WRITEMASK_XYZW)
880             {
881                tgsi_parse_free(&parse);
882                return FALSE;
883             }
884          }
885          break;
886
887       case TGSI_TOKEN_TYPE_DECLARATION:
888          /* fall-through */
889       case TGSI_TOKEN_TYPE_IMMEDIATE:
890          /* fall-through */
891       case TGSI_TOKEN_TYPE_PROPERTY:
892          /* fall-through */
893       default:
894          ; /* no-op */
895       }
896    }
897
898    tgsi_parse_free(&parse);
899
900    /* if we get here, it's a pass-through shader */
901    return TRUE;
902 }