st_atom_pixeltransfer.c revision aa878f94ab77fe3be352d820aaa950b32836c814 
1/**************************************************************************
2 *
3 * Copyright 2007 Tungsten Graphics, Inc., Cedar Park, Texas.
4 * All Rights Reserved.
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the
8 * "Software"), to deal in the Software without restriction, including
9 * without limitation the rights to use, copy, modify, merge, publish,
10 * distribute, sub license, and/or sell copies of the Software, and to
11 * permit persons to whom the Software is furnished to do so, subject to
12 * the following conditions:
13 *
14 * The above copyright notice and this permission notice (including the
15 * next paragraph) shall be included in all copies or substantial portions
16 * of the Software.
17 *
18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21 * IN NO EVENT SHALL TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR
22 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25 *
26 **************************************************************************/
27
28/*
29 * Generate fragment programs to implement pixel transfer ops, such as
30 * scale/bias, colortable, convolution...
31 *
32 * Authors:
33 *   Brian Paul
34 */
35
36#include "main/imports.h"
37#include "main/image.h"
38#include "main/macros.h"
39#include "program/program.h"
40#include "program/prog_cache.h"
41#include "program/prog_instruction.h"
42#include "program/prog_parameter.h"
43#include "program/prog_print.h"
44
45#include "st_context.h"
46#include "st_format.h"
47#include "st_texture.h"
48
49#include "pipe/p_screen.h"
50#include "pipe/p_context.h"
51#include "util/u_inlines.h"
52#include "util/u_pack_color.h"
53
54
55struct state_key
56{
57   GLuint scaleAndBias:1;
58   GLuint pixelMaps:1;
59
60#if 0
61   GLfloat Maps[3][256][4];
62   int NumMaps;
63   GLint NumStages;
64   pipeline_stage Stages[STAGE_MAX];
65   GLboolean StagesUsed[STAGE_MAX];
66   GLfloat Scale1[4], Bias1[4];
67   GLfloat Scale2[4], Bias2[4];
68#endif
69};
70
71static void
72make_state_key(struct gl_context *ctx,  struct state_key *key)
73{
74   memset(key, 0, sizeof(*key));
75
76   if (ctx->Pixel.RedBias != 0.0 || ctx->Pixel.RedScale != 1.0 ||
77       ctx->Pixel.GreenBias != 0.0 || ctx->Pixel.GreenScale != 1.0 ||
78       ctx->Pixel.BlueBias != 0.0 || ctx->Pixel.BlueScale != 1.0 ||
79       ctx->Pixel.AlphaBias != 0.0 || ctx->Pixel.AlphaScale != 1.0) {
80      key->scaleAndBias = 1;
81   }
82
83   key->pixelMaps = ctx->Pixel.MapColorFlag;
84}
85
86
87static struct pipe_resource *
88create_color_map_texture(struct gl_context *ctx)
89{
90   struct st_context *st = st_context(ctx);
91   struct pipe_context *pipe = st->pipe;
92   struct pipe_resource *pt;
93   enum pipe_format format;
94   const uint texSize = 256; /* simple, and usually perfect */
95
96   /* find an RGBA texture format */
97   format = st_choose_format(pipe->screen, GL_RGBA,
98                             PIPE_TEXTURE_2D, 0, PIPE_BIND_SAMPLER_VIEW);
99
100   /* create texture for color map/table */
101   pt = st_texture_create(st, PIPE_TEXTURE_2D, format, 0,
102                          texSize, texSize, 1, 1, PIPE_BIND_SAMPLER_VIEW);
103   return pt;
104}
105
106
107/**
108 * Update the pixelmap texture with the contents of the R/G/B/A pixel maps.
109 */
110static void
111load_color_map_texture(struct gl_context *ctx, struct pipe_resource *pt)
112{
113   struct st_context *st = st_context(ctx);
114   struct pipe_context *pipe = st->pipe;
115   struct pipe_transfer *transfer;
116   const GLuint rSize = ctx->PixelMaps.RtoR.Size;
117   const GLuint gSize = ctx->PixelMaps.GtoG.Size;
118   const GLuint bSize = ctx->PixelMaps.BtoB.Size;
119   const GLuint aSize = ctx->PixelMaps.AtoA.Size;
120   const uint texSize = pt->width0;
121   uint *dest;
122   uint i, j;
123
124   transfer = pipe_get_transfer(st_context(ctx)->pipe,
125                                pt, 0, 0, PIPE_TRANSFER_WRITE,
126                                0, 0, texSize, texSize);
127   dest = (uint *) pipe_transfer_map(pipe, transfer);
128
129   /* Pack four 1D maps into a 2D texture:
130    * R map is placed horizontally, indexed by S, in channel 0
131    * G map is placed vertically, indexed by T, in channel 1
132    * B map is placed horizontally, indexed by S, in channel 2
133    * A map is placed vertically, indexed by T, in channel 3
134    */
135   for (i = 0; i < texSize; i++) {
136      for (j = 0; j < texSize; j++) {
137         union util_color uc;
138         int k = (i * texSize + j);
139         ubyte r = ctx->PixelMaps.RtoR.Map8[j * rSize / texSize];
140         ubyte g = ctx->PixelMaps.GtoG.Map8[i * gSize / texSize];
141         ubyte b = ctx->PixelMaps.BtoB.Map8[j * bSize / texSize];
142         ubyte a = ctx->PixelMaps.AtoA.Map8[i * aSize / texSize];
143         util_pack_color_ub(r, g, b, a, pt->format, &uc);
144         *(dest + k) = uc.ui;
145      }
146   }
147
148   pipe_transfer_unmap(pipe, transfer);
149   pipe->transfer_destroy(pipe, transfer);
150}
151
152
153
154#define MAX_INST 100
155
156/**
157 * Returns a fragment program which implements the current pixel transfer ops.
158 */
159static struct gl_fragment_program *
160get_pixel_transfer_program(struct gl_context *ctx, const struct state_key *key)
161{
162   struct st_context *st = st_context(ctx);
163   struct prog_instruction inst[MAX_INST];
164   struct gl_program_parameter_list *params;
165   struct gl_fragment_program *fp;
166   GLuint ic = 0;
167   const GLuint colorTemp = 0;
168
169   fp = (struct gl_fragment_program *)
170      ctx->Driver.NewProgram(ctx, GL_FRAGMENT_PROGRAM_ARB, 0);
171   if (!fp)
172      return NULL;
173
174   params = _mesa_new_parameter_list();
175
176   /*
177    * Get initial pixel color from the texture.
178    * TEX colorTemp, fragment.texcoord[0], texture[0], 2D;
179    */
180   _mesa_init_instructions(inst + ic, 1);
181   inst[ic].Opcode = OPCODE_TEX;
182   inst[ic].DstReg.File = PROGRAM_TEMPORARY;
183   inst[ic].DstReg.Index = colorTemp;
184   inst[ic].SrcReg[0].File = PROGRAM_INPUT;
185   inst[ic].SrcReg[0].Index = FRAG_ATTRIB_TEX0;
186   inst[ic].TexSrcUnit = 0;
187   inst[ic].TexSrcTarget = TEXTURE_2D_INDEX;
188   ic++;
189   fp->Base.InputsRead = (1 << FRAG_ATTRIB_TEX0);
190   fp->Base.OutputsWritten = BITFIELD64_BIT(FRAG_RESULT_COLOR);
191   fp->Base.SamplersUsed = 0x1;  /* sampler 0 (bit 0) is used */
192
193   if (key->scaleAndBias) {
194      static const gl_state_index scale_state[STATE_LENGTH] =
195         { STATE_INTERNAL, STATE_PT_SCALE, 0, 0, 0 };
196      static const gl_state_index bias_state[STATE_LENGTH] =
197         { STATE_INTERNAL, STATE_PT_BIAS, 0, 0, 0 };
198      GLfloat scale[4], bias[4];
199      GLint scale_p, bias_p;
200
201      scale[0] = ctx->Pixel.RedScale;
202      scale[1] = ctx->Pixel.GreenScale;
203      scale[2] = ctx->Pixel.BlueScale;
204      scale[3] = ctx->Pixel.AlphaScale;
205      bias[0] = ctx->Pixel.RedBias;
206      bias[1] = ctx->Pixel.GreenBias;
207      bias[2] = ctx->Pixel.BlueBias;
208      bias[3] = ctx->Pixel.AlphaBias;
209
210      scale_p = _mesa_add_state_reference(params, scale_state);
211      bias_p = _mesa_add_state_reference(params, bias_state);
212
213      /* MAD colorTemp, colorTemp, scale, bias; */
214      _mesa_init_instructions(inst + ic, 1);
215      inst[ic].Opcode = OPCODE_MAD;
216      inst[ic].DstReg.File = PROGRAM_TEMPORARY;
217      inst[ic].DstReg.Index = colorTemp;
218      inst[ic].SrcReg[0].File = PROGRAM_TEMPORARY;
219      inst[ic].SrcReg[0].Index = colorTemp;
220      inst[ic].SrcReg[1].File = PROGRAM_STATE_VAR;
221      inst[ic].SrcReg[1].Index = scale_p;
222      inst[ic].SrcReg[2].File = PROGRAM_STATE_VAR;
223      inst[ic].SrcReg[2].Index = bias_p;
224      ic++;
225   }
226
227   if (key->pixelMaps) {
228      const GLuint temp = 1;
229
230      /* create the colormap/texture now if not already done */
231      if (!st->pixel_xfer.pixelmap_texture) {
232         st->pixel_xfer.pixelmap_texture = create_color_map_texture(ctx);
233         st->pixel_xfer.pixelmap_sampler_view =
234            st_create_texture_sampler_view(st->pipe,
235                                           st->pixel_xfer.pixelmap_texture);
236      }
237
238      /* with a little effort, we can do four pixel map look-ups with
239       * two TEX instructions:
240       */
241
242      /* TEX temp.rg, colorTemp.rgba, texture[1], 2D; */
243      _mesa_init_instructions(inst + ic, 1);
244      inst[ic].Opcode = OPCODE_TEX;
245      inst[ic].DstReg.File = PROGRAM_TEMPORARY;
246      inst[ic].DstReg.Index = temp;
247      inst[ic].DstReg.WriteMask = WRITEMASK_XY; /* write R,G */
248      inst[ic].SrcReg[0].File = PROGRAM_TEMPORARY;
249      inst[ic].SrcReg[0].Index = colorTemp;
250      inst[ic].TexSrcUnit = 1;
251      inst[ic].TexSrcTarget = TEXTURE_2D_INDEX;
252      ic++;
253
254      /* TEX temp.ba, colorTemp.baba, texture[1], 2D; */
255      _mesa_init_instructions(inst + ic, 1);
256      inst[ic].Opcode = OPCODE_TEX;
257      inst[ic].DstReg.File = PROGRAM_TEMPORARY;
258      inst[ic].DstReg.Index = temp;
259      inst[ic].DstReg.WriteMask = WRITEMASK_ZW; /* write B,A */
260      inst[ic].SrcReg[0].File = PROGRAM_TEMPORARY;
261      inst[ic].SrcReg[0].Index = colorTemp;
262      inst[ic].SrcReg[0].Swizzle = MAKE_SWIZZLE4(SWIZZLE_Z, SWIZZLE_W,
263                                                 SWIZZLE_Z, SWIZZLE_W);
264      inst[ic].TexSrcUnit = 1;
265      inst[ic].TexSrcTarget = TEXTURE_2D_INDEX;
266      ic++;
267
268      /* MOV colorTemp, temp; */
269      _mesa_init_instructions(inst + ic, 1);
270      inst[ic].Opcode = OPCODE_MOV;
271      inst[ic].DstReg.File = PROGRAM_TEMPORARY;
272      inst[ic].DstReg.Index = colorTemp;
273      inst[ic].SrcReg[0].File = PROGRAM_TEMPORARY;
274      inst[ic].SrcReg[0].Index = temp;
275      ic++;
276
277      fp->Base.SamplersUsed |= (1 << 1);  /* sampler 1 is used */
278   }
279
280   /* Modify last instruction's dst reg to write to result.color */
281   {
282      struct prog_instruction *last = &inst[ic - 1];
283      last->DstReg.File = PROGRAM_OUTPUT;
284      last->DstReg.Index = FRAG_RESULT_COLOR;
285   }
286
287   /* END; */
288   _mesa_init_instructions(inst + ic, 1);
289   inst[ic].Opcode = OPCODE_END;
290   ic++;
291
292   assert(ic <= MAX_INST);
293
294
295   fp->Base.Instructions = _mesa_alloc_instructions(ic);
296   if (!fp->Base.Instructions) {
297      _mesa_error(ctx, GL_OUT_OF_MEMORY,
298                  "generating pixel transfer program");
299      return NULL;
300   }
301
302   _mesa_copy_instructions(fp->Base.Instructions, inst, ic);
303   fp->Base.NumInstructions = ic;
304   fp->Base.Parameters = params;
305
306#if 0
307   printf("========= pixel transfer prog\n");
308   _mesa_print_program(&fp->Base);
309   _mesa_print_parameter_list(fp->Base.Parameters);
310#endif
311
312   return fp;
313}
314
315
316
317/**
318 * Update st->pixel_xfer.program in response to new pixel-transfer state.
319 */
320static void
321update_pixel_transfer(struct st_context *st)
322{
323   struct gl_context *ctx = st->ctx;
324   struct state_key key;
325   struct gl_fragment_program *fp;
326
327   make_state_key(st->ctx, &key);
328
329   fp = (struct gl_fragment_program *)
330      _mesa_search_program_cache(st->pixel_xfer.cache, &key, sizeof(key));
331   if (!fp) {
332      fp = get_pixel_transfer_program(st->ctx, &key);
333      _mesa_program_cache_insert(st->ctx, st->pixel_xfer.cache,
334                                 &key, sizeof(key), &fp->Base);
335   }
336
337   if (ctx->Pixel.MapColorFlag) {
338      load_color_map_texture(ctx, st->pixel_xfer.pixelmap_texture);
339   }
340   st->pixel_xfer.pixelmap_enabled = ctx->Pixel.MapColorFlag;
341
342   st->pixel_xfer.program = (struct st_fragment_program *) fp;
343}
344
345
346
347const struct st_tracked_state st_update_pixel_transfer = {
348   "st_update_pixel_transfer",				/* name */
349   {							/* dirty */
350      _NEW_PIXEL,					/* mesa */
351      0,						/* st */
352   },
353   update_pixel_transfer				/* update */
354};
355