← Back to branch summary

~mmach/netext73/mesa-haswell

« back to all changes in this revision

Viewing changes to src/mesa/state_tracker/st_cb_drawpixels_shader.c

  • Committer: mmach
  • Date: 2022-09-22 19:56:13 UTC
  • Revision ID: netbit73@gmail.com-20220922195613-wtik9mmy20tmor0i
2022-09-22 21:17:09

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/mesa/state_tracker/st_cb_drawpixels_shader.c
1
 
/**************************************************************************
2
 
 *
3
 
 * Copyright (C) 2015 Advanced Micro Devices, Inc.
4
 
 * Copyright 2007 VMware, Inc.
5
 
 * 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
 
#include "main/macros.h"
30
 
#include "st_cb_drawpixels.h"
31
 
#include "tgsi/tgsi_transform.h"
32
 
#include "tgsi/tgsi_scan.h"
33
 
 
34
 
struct tgsi_drawpix_transform {
35
 
   struct tgsi_transform_context base;
36
 
   struct tgsi_shader_info info;
37
 
   bool use_texcoord;
38
 
   bool scale_and_bias;
39
 
   bool pixel_maps;
40
 
   bool first_instruction_emitted;
41
 
   unsigned scale_const;
42
 
   unsigned bias_const;
43
 
   unsigned color_temp;
44
 
   unsigned drawpix_sampler;
45
 
   unsigned pixelmap_sampler;
46
 
   unsigned texcoord_const;
47
 
   unsigned tex_target;
48
 
};
49
 
 
50
 
static inline struct tgsi_drawpix_transform *
51
 
tgsi_drawpix_transform(struct tgsi_transform_context *tctx)
52
 
{
53
 
   return (struct tgsi_drawpix_transform *)tctx;
54
 
}
55
 
 
56
 
static void
57
 
set_src(struct tgsi_full_instruction *inst, unsigned i, unsigned file, unsigned index,
58
 
        unsigned x, unsigned y, unsigned z, unsigned w)
59
 
{
60
 
   inst->Src[i].Register.File  = file;
61
 
   inst->Src[i].Register.Index = index;
62
 
   inst->Src[i].Register.SwizzleX = x;
63
 
   inst->Src[i].Register.SwizzleY = y;
64
 
   inst->Src[i].Register.SwizzleZ = z;
65
 
   inst->Src[i].Register.SwizzleW = w;
66
 
}
67
 
 
68
 
#define SET_SRC(inst, i, file, index, x, y, z, w) \
69
 
   set_src(inst, i, file, index, TGSI_SWIZZLE_##x, TGSI_SWIZZLE_##y, \
70
 
           TGSI_SWIZZLE_##z, TGSI_SWIZZLE_##w)
71
 
 
72
 
static void
73
 
transform_instr(struct tgsi_transform_context *tctx,
74
 
                struct tgsi_full_instruction *current_inst)
75
 
{
76
 
   struct tgsi_drawpix_transform *ctx = tgsi_drawpix_transform(tctx);
77
 
   const unsigned tgsi_tex_target = ctx->tex_target == PIPE_TEXTURE_2D
78
 
      ? TGSI_TEXTURE_2D : TGSI_TEXTURE_RECT;
79
 
   unsigned i, sem_texcoord = ctx->use_texcoord ? TGSI_SEMANTIC_TEXCOORD :
80
 
                                                  TGSI_SEMANTIC_GENERIC;
81
 
   int texcoord_index = -1;
82
 
 
83
 
   if (ctx->first_instruction_emitted)
84
 
      goto transform_inst;
85
 
 
86
 
   ctx->first_instruction_emitted = true;
87
 
 
88
 
   /* Add scale and bias constants. */
89
 
   if (ctx->scale_and_bias) {
90
 
      if (ctx->info.const_file_max[0] < (int)ctx->scale_const) {
91
 
         tgsi_transform_const_decl(tctx, ctx->scale_const, ctx->scale_const);
92
 
      }
93
 
 
94
 
      if (ctx->info.const_file_max[0] < (int)ctx->bias_const) {
95
 
         tgsi_transform_const_decl(tctx, ctx->bias_const, ctx->bias_const);
96
 
      }
97
 
   }
98
 
 
99
 
   if (ctx->info.const_file_max[0] < (int)ctx->texcoord_const) {
100
 
      tgsi_transform_const_decl(tctx, ctx->texcoord_const, ctx->texcoord_const);
101
 
   }
102
 
 
103
 
   /* Add a new temp. */
104
 
   ctx->color_temp = ctx->info.file_max[TGSI_FILE_TEMPORARY] + 1;
105
 
   tgsi_transform_temp_decl(tctx, ctx->color_temp);
106
 
 
107
 
   /* Add TEXCOORD[texcoord_slot] if it's missing. */
108
 
   for (i = 0; i < ctx->info.num_inputs; i++) {
109
 
      if (ctx->info.input_semantic_name[i] == sem_texcoord &&
110
 
          ctx->info.input_semantic_index[i] == 0) {
111
 
         texcoord_index = i;
112
 
         break;
113
 
      }
114
 
   }
115
 
 
116
 
   if (texcoord_index == -1) {
117
 
      texcoord_index = ctx->info.num_inputs;
118
 
      tgsi_transform_input_decl(tctx, texcoord_index, sem_texcoord, 0,
119
 
                                TGSI_INTERPOLATE_PERSPECTIVE);
120
 
   }
121
 
 
122
 
   /* Declare the drawpix sampler if it's missing. */
123
 
   if (!(ctx->info.samplers_declared & (1 << ctx->drawpix_sampler))) {
124
 
      tgsi_transform_sampler_decl(tctx, ctx->drawpix_sampler);
125
 
 
126
 
      /* emit sampler view declaration */
127
 
      tgsi_transform_sampler_view_decl(tctx, ctx->drawpix_sampler,
128
 
                                       tgsi_tex_target, TGSI_RETURN_TYPE_FLOAT);
129
 
   }
130
 
 
131
 
   /* Declare the pixel map sampler if it's missing. */
132
 
   if (ctx->pixel_maps &&
133
 
       !(ctx->info.samplers_declared & (1 << ctx->pixelmap_sampler))) {
134
 
      tgsi_transform_sampler_decl(tctx, ctx->pixelmap_sampler);
135
 
 
136
 
      /* emit sampler view declaration */
137
 
      tgsi_transform_sampler_view_decl(tctx, ctx->pixelmap_sampler,
138
 
                                       TGSI_TEXTURE_2D, TGSI_RETURN_TYPE_FLOAT);
139
 
   }
140
 
 
141
 
   /* Get initial pixel color from the texture.
142
 
    * TEX temp, fragment.texcoord[0], texture[0], 2D;
143
 
    */
144
 
   tgsi_transform_tex_inst(tctx, TGSI_FILE_TEMPORARY, ctx->color_temp,
145
 
                           TGSI_FILE_INPUT, texcoord_index,
146
 
                           tgsi_tex_target, ctx->drawpix_sampler);
147
 
 
148
 
   /* Apply the scale and bias. */
149
 
   if (ctx->scale_and_bias) {
150
 
      /* MAD temp, temp, scale, bias; */
151
 
      tgsi_transform_op3_inst(tctx, TGSI_OPCODE_MAD,
152
 
                              TGSI_FILE_TEMPORARY, ctx->color_temp,
153
 
                              TGSI_WRITEMASK_XYZW,
154
 
                              TGSI_FILE_TEMPORARY, ctx->color_temp,
155
 
                              TGSI_FILE_CONSTANT, ctx->scale_const,
156
 
                              TGSI_FILE_CONSTANT, ctx->bias_const);
157
 
   }
158
 
 
159
 
   if (ctx->pixel_maps) {
160
 
      /* do four pixel map look-ups with two TEX instructions: */
161
 
      struct tgsi_full_instruction inst;
162
 
 
163
 
      /* TEX temp.xy, temp.xyyy, texture[1], 2D; */
164
 
      inst = tgsi_default_full_instruction();
165
 
      inst.Instruction.Opcode = TGSI_OPCODE_TEX;
166
 
      inst.Instruction.Texture = 1;
167
 
      inst.Texture.Texture = TGSI_TEXTURE_2D;
168
 
 
169
 
      inst.Instruction.NumDstRegs = 1;
170
 
      inst.Dst[0].Register.File  = TGSI_FILE_TEMPORARY;
171
 
      inst.Dst[0].Register.Index = ctx->color_temp;
172
 
      inst.Dst[0].Register.WriteMask = TGSI_WRITEMASK_XY;
173
 
 
174
 
      inst.Instruction.NumSrcRegs = 2;
175
 
      SET_SRC(&inst, 0, TGSI_FILE_TEMPORARY, ctx->color_temp, X, Y, Y, Y);
176
 
      inst.Src[1].Register.File  = TGSI_FILE_SAMPLER;
177
 
      inst.Src[1].Register.Index = ctx->pixelmap_sampler;
178
 
 
179
 
      tctx->emit_instruction(tctx, &inst);
180
 
 
181
 
      /* TEX temp.zw, temp.zwww, texture[1], 2D; */
182
 
      inst.Dst[0].Register.WriteMask = TGSI_WRITEMASK_ZW;
183
 
      SET_SRC(&inst, 0, TGSI_FILE_TEMPORARY, ctx->color_temp, Z, W, W, W);
184
 
      tctx->emit_instruction(tctx, &inst);
185
 
   }
186
 
 
187
 
   /* Now, "color_temp" should be used in place of IN:COLOR0,
188
 
    * and CONST[texcoord_slot] should be used in place of IN:TEXCOORD0.
189
 
    */
190
 
 
191
 
transform_inst:
192
 
 
193
 
   for (i = 0; i < current_inst->Instruction.NumSrcRegs; i++) {
194
 
      struct tgsi_full_src_register *src = &current_inst->Src[i];
195
 
      unsigned reg = src->Register.Index;
196
 
 
197
 
      if (src->Register.File != TGSI_FILE_INPUT || src->Register.Indirect)
198
 
         continue;
199
 
 
200
 
      if (ctx->info.input_semantic_name[reg] == TGSI_SEMANTIC_COLOR &&
201
 
          ctx->info.input_semantic_index[reg] == 0) {
202
 
         src->Register.File = TGSI_FILE_TEMPORARY;
203
 
         src->Register.Index = ctx->color_temp;
204
 
      } else if (ctx->info.input_semantic_name[reg] == sem_texcoord &&
205
 
                 ctx->info.input_semantic_index[reg] == 0) {
206
 
         src->Register.File = TGSI_FILE_CONSTANT;
207
 
         src->Register.Index = ctx->texcoord_const;
208
 
         src->Register.Dimension = 1;
209
 
         src->Dimension.Index = 0;
210
 
      }
211
 
   }
212
 
 
213
 
   tctx->emit_instruction(tctx, current_inst);
214
 
}
215
 
 
216
 
const struct tgsi_token *
217
 
st_get_drawpix_shader(const struct tgsi_token *tokens, bool use_texcoord,
218
 
                      bool scale_and_bias, unsigned scale_const,
219
 
                      unsigned bias_const, bool pixel_maps,
220
 
                      unsigned drawpix_sampler, unsigned pixelmap_sampler,
221
 
                      unsigned texcoord_const, unsigned tex_target)
222
 
{
223
 
   struct tgsi_drawpix_transform ctx;
224
 
   int newlen;
225
 
 
226
 
   assert(tex_target == PIPE_TEXTURE_2D ||
227
 
          tex_target == PIPE_TEXTURE_RECT);
228
 
 
229
 
   memset(&ctx, 0, sizeof(ctx));
230
 
   ctx.base.transform_instruction = transform_instr;
231
 
   ctx.use_texcoord = use_texcoord;
232
 
   ctx.scale_and_bias = scale_and_bias;
233
 
   ctx.scale_const = scale_const;
234
 
   ctx.bias_const = bias_const;
235
 
   ctx.pixel_maps = pixel_maps;
236
 
   ctx.drawpix_sampler = drawpix_sampler;
237
 
   ctx.pixelmap_sampler = pixelmap_sampler;
238
 
   ctx.texcoord_const = texcoord_const;
239
 
   ctx.tex_target = tex_target;
240
 
   tgsi_scan_shader(tokens, &ctx.info);
241
 
 
242
 
   newlen = tgsi_num_tokens(tokens) + 60;
243
 
 
244
 
   return tgsi_transform_shader(tokens, newlen, &ctx.base);
245
 
}