r500: add cmp support in theory
[mesa.git] / src / mesa / drivers / dri / r300 / r500_fragprog.c
1 /*
2 * Copyright (C) 2005 Ben Skeggs.
3 *
4 * All Rights Reserved.
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining
7 * a 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, sublicense, 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
16 * portions of the Software.
17 *
18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
19 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
21 * IN NO EVENT SHALL THE COPYRIGHT OWNER(S) AND/OR ITS SUPPLIERS BE
22 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
23 * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
24 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25 *
26 */
27
28 /**
29 * \file
30 *
31 * \author Ben Skeggs <darktama@iinet.net.au>
32 *
33 * \author Jerome Glisse <j.glisse@gmail.com>
34 *
35 * \author Corbin Simpson <MostAwesomeDude@gmail.com>
36 *
37 * \todo Depth write, WPOS/FOGC inputs
38 *
39 * \todo FogOption
40 *
41 * \todo Verify results of opcodes for accuracy, I've only checked them in
42 * specific cases.
43 */
44
45 #include "glheader.h"
46 #include "macros.h"
47 #include "enums.h"
48 #include "shader/prog_instruction.h"
49 #include "shader/prog_parameter.h"
50 #include "shader/prog_print.h"
51
52 #include "r300_context.h"
53 #include "r500_fragprog.h"
54 #include "r300_reg.h"
55 #include "r300_state.h"
56
57 /*
58 * Useful macros and values
59 */
60 #define ERROR(fmt, args...) do { \
61 fprintf(stderr, "%s::%s(): " fmt "\n", \
62 __FILE__, __FUNCTION__, ##args); \
63 fp->error = GL_TRUE; \
64 } while(0)
65
66 #define COMPILE_STATE struct r300_pfs_compile_state *cs = fp->cs
67
68 #define R500_US_NUM_TEMP_REGS 128
69 #define R500_US_NUM_CONST_REGS 256
70
71 /* "Register" flags */
72 #define REG_CONSTANT (1 << 8)
73 #define REG_SRC_REL (1 << 9)
74 #define REG_DEST_REL (1 << 7)
75
76 /* Swizzle tools */
77 #define R500_SWIZZLE_ZERO 4
78 #define R500_SWIZZLE_HALF 5
79 #define R500_SWIZZLE_ONE 6
80 #define R500_SWIZ_RGB_ZERO ((4 << 0) | (4 << 3) | (4 << 6))
81 #define R500_SWIZ_RGB_ONE ((6 << 0) | (6 << 3) | (6 << 6))
82 #define R500_SWIZ_RGB_RGB ((0 << 0) | (1 << 3) | (2 << 6))
83 /* Swizzles for inst2 */
84 #define MAKE_SWIZ_TEX_STRQ(x) (x << 8)
85 #define MAKE_SWIZ_TEX_RGBA(x) (x << 24)
86 /* Swizzles for inst3 */
87 #define MAKE_SWIZ_RGB_A(x) (x << 2)
88 #define MAKE_SWIZ_RGB_B(x) (x << 15)
89 /* Swizzles for inst4 */
90 #define MAKE_SWIZ_ALPHA_A(x) (x << 14)
91 #define MAKE_SWIZ_ALPHA_B(x) (x << 21)
92 /* Swizzle for inst5 */
93 #define MAKE_SWIZ_RGBA_C(x) (x << 14)
94 #define MAKE_SWIZ_ALPHA_C(x) (x << 27)
95
96 static inline GLuint make_rgb_swizzle(struct prog_src_register src) {
97 GLuint swiz = 0x0;
98 GLuint temp;
99 /* This could be optimized, but it should be plenty fast already. */
100 int i;
101 for (i = 0; i < 3; i++) {
102 temp = (src.Swizzle >> i*3) & 0x7;
103 /* Fix SWIZZLE_ONE */
104 if (temp == 5) temp++;
105 swiz += temp << i*3;
106 }
107 return swiz;
108 }
109
110 static inline GLuint make_alpha_swizzle(struct prog_src_register src) {
111 GLuint swiz = (src.Swizzle >> 12) & 0x7;
112 if (swiz == 5) swiz++;
113 return swiz;
114 }
115
116 static inline GLuint make_strq_swizzle(struct prog_src_register src) {
117 GLuint swiz = 0x0;
118 GLuint temp = src.Swizzle;
119 int i;
120 for (i = 0; i < 4; i++) {
121 swiz += (temp & 0x3) << i*2;
122 temp >>= 3;
123 }
124 return swiz;
125 }
126
127 static int get_temp(struct r500_fragment_program *fp, int slot) {
128
129 COMPILE_STATE;
130
131 int r = slot;
132
133 while (cs->inputs[r].refcount != 0) {
134 /* Crap, taken. */
135 r++;
136 }
137
138 fp->temp_reg_offset = r - slot;
139
140 if (r >= R500_US_NUM_TEMP_REGS) {
141 ERROR("Out of hardware temps!\n");
142 return 0;
143 }
144
145 if (r > fp->max_temp_idx)
146 fp->max_temp_idx = r;
147
148 return r;
149 }
150
151 /* Borrowed verbatim from r300_fragprog since it hasn't changed. */
152 static GLuint emit_const4fv(struct r500_fragment_program *fp,
153 const GLfloat * cp)
154 {
155 GLuint reg = 0x0;
156 int index;
157
158 for (index = 0; index < fp->const_nr; ++index) {
159 if (fp->constant[index] == cp)
160 break;
161 }
162
163 if (index >= fp->const_nr) {
164 if (index >= R500_US_NUM_CONST_REGS) {
165 ERROR("Out of hw constants!\n");
166 return reg;
167 }
168
169 fp->const_nr++;
170 fp->constant[index] = cp;
171 }
172
173 reg = index | REG_CONSTANT;
174 return reg;
175 }
176
177 static GLuint make_src(struct r500_fragment_program *fp, struct prog_src_register src) {
178 COMPILE_STATE;
179 GLuint reg;
180 switch (src.File) {
181 case PROGRAM_TEMPORARY:
182 reg = src.Index + fp->temp_reg_offset;
183 break;
184 case PROGRAM_INPUT:
185 reg = cs->inputs[src.Index].reg;
186 break;
187 case PROGRAM_STATE_VAR:
188 case PROGRAM_NAMED_PARAM:
189 case PROGRAM_CONSTANT:
190 reg = emit_const4fv(fp, fp->mesa_program.Base.Parameters->
191 ParameterValues[src.Index]);
192 break;
193 default:
194 ERROR("Can't handle src.File %x\n", src.File);
195 reg = 0x0;
196 break;
197 }
198 return reg;
199 }
200
201 static GLuint make_dest(struct r500_fragment_program *fp, struct prog_dst_register dest) {
202 GLuint reg;
203 switch (dest.File) {
204 case PROGRAM_TEMPORARY:
205 reg = dest.Index + fp->temp_reg_offset;
206 break;
207 case PROGRAM_OUTPUT:
208 /* Eventually we may need to handle multiple
209 * rendering targets... */
210 reg = dest.Index;
211 break;
212 default:
213 ERROR("Can't handle dest.File %x\n", dest.File);
214 reg = 0x0;
215 break;
216 }
217 return reg;
218 }
219
220 static void emit_tex(struct r500_fragment_program *fp,
221 struct prog_instruction *fpi, int opcode, int dest, int counter)
222 {
223 int hwsrc, hwdest;
224 GLuint mask;
225
226 mask = fpi->DstReg.WriteMask << 11;
227 hwsrc = make_src(fp, fpi->SrcReg[0]);
228
229 fp->inst[counter].inst0 = R500_INST_TYPE_TEX | mask
230 | R500_INST_TEX_SEM_WAIT;
231
232 fp->inst[counter].inst1 = fpi->TexSrcUnit
233 | R500_TEX_SEM_ACQUIRE | R500_TEX_IGNORE_UNCOVERED;
234
235 if (fpi->TexSrcTarget == TEXTURE_RECT_INDEX)
236 fp->inst[counter].inst1 |= R500_TEX_UNSCALED;
237
238 switch (opcode) {
239 case OPCODE_TEX:
240 fp->inst[counter].inst1 |= R500_TEX_INST_LD;
241 break;
242 case OPCODE_TXB:
243 fp->inst[counter].inst1 |= R500_TEX_INST_LODBIAS;
244 break;
245 case OPCODE_TXP:
246 fp->inst[counter].inst1 |= R500_TEX_INST_PROJ;
247 break;
248 default:
249 ERROR("emit_tex can't handle opcode %x\n", opcode);
250 }
251
252 fp->inst[counter].inst2 = R500_TEX_SRC_ADDR(hwsrc)
253 /* | MAKE_SWIZ_TEX_STRQ(make_strq_swizzle(fpi->SrcReg[0])) */
254 | R500_TEX_SRC_S_SWIZ_R | R500_TEX_SRC_T_SWIZ_G
255 | R500_TEX_SRC_R_SWIZ_B | R500_TEX_SRC_Q_SWIZ_A
256 | R500_TEX_DST_ADDR(dest)
257 | R500_TEX_DST_R_SWIZ_R | R500_TEX_DST_G_SWIZ_G
258 | R500_TEX_DST_B_SWIZ_B | R500_TEX_DST_A_SWIZ_A;
259
260
261
262 fp->inst[counter].inst3 = 0x0;
263 fp->inst[counter].inst4 = 0x0;
264 fp->inst[counter].inst5 = 0x0;
265 }
266
267 static void dumb_shader(struct r500_fragment_program *fp)
268 {
269 fp->inst[0].inst0 = R500_INST_TYPE_TEX
270 | R500_INST_TEX_SEM_WAIT
271 | R500_INST_RGB_WMASK_R
272 | R500_INST_RGB_WMASK_G
273 | R500_INST_RGB_WMASK_B
274 | R500_INST_ALPHA_WMASK
275 | R500_INST_RGB_CLAMP
276 | R500_INST_ALPHA_CLAMP;
277 fp->inst[0].inst1 = R500_TEX_ID(0)
278 | R500_TEX_INST_LD
279 | R500_TEX_SEM_ACQUIRE
280 | R500_TEX_IGNORE_UNCOVERED;
281 fp->inst[0].inst2 = R500_TEX_SRC_ADDR(0)
282 | R500_TEX_SRC_S_SWIZ_R
283 | R500_TEX_SRC_T_SWIZ_G
284 | R500_TEX_DST_ADDR(0)
285 | R500_TEX_DST_R_SWIZ_R
286 | R500_TEX_DST_G_SWIZ_G
287 | R500_TEX_DST_B_SWIZ_B
288 | R500_TEX_DST_A_SWIZ_A;
289 fp->inst[0].inst3 = R500_DX_ADDR(0)
290 | R500_DX_S_SWIZ_R
291 | R500_DX_T_SWIZ_R
292 | R500_DX_R_SWIZ_R
293 | R500_DX_Q_SWIZ_R
294 | R500_DY_ADDR(0)
295 | R500_DY_S_SWIZ_R
296 | R500_DY_T_SWIZ_R
297 | R500_DY_R_SWIZ_R
298 | R500_DY_Q_SWIZ_R;
299 fp->inst[0].inst4 = 0x0;
300 fp->inst[0].inst5 = 0x0;
301
302 fp->inst[1].inst0 = R500_INST_TYPE_OUT |
303 R500_INST_TEX_SEM_WAIT |
304 R500_INST_LAST |
305 R500_INST_RGB_OMASK_R |
306 R500_INST_RGB_OMASK_G |
307 R500_INST_RGB_OMASK_B |
308 R500_INST_ALPHA_OMASK;
309 fp->inst[1].inst1 = R500_RGB_ADDR0(0) |
310 R500_RGB_ADDR1(0) |
311 R500_RGB_ADDR1_CONST |
312 R500_RGB_ADDR2(0) |
313 R500_RGB_ADDR2_CONST |
314 R500_RGB_SRCP_OP_1_MINUS_2RGB0;
315 fp->inst[1].inst2 = R500_ALPHA_ADDR0(0) |
316 R500_ALPHA_ADDR1(0) |
317 R500_ALPHA_ADDR1_CONST |
318 R500_ALPHA_ADDR2(0) |
319 R500_ALPHA_ADDR2_CONST |
320 R500_ALPHA_SRCP_OP_1_MINUS_2A0;
321 fp->inst[1].inst3 = R500_ALU_RGB_SEL_A_SRC0 |
322 R500_ALU_RGB_R_SWIZ_A_R |
323 R500_ALU_RGB_G_SWIZ_A_G |
324 R500_ALU_RGB_B_SWIZ_A_B |
325 R500_ALU_RGB_SEL_B_SRC0 |
326 R500_ALU_RGB_R_SWIZ_B_1 |
327 R500_ALU_RGB_B_SWIZ_B_1 |
328 R500_ALU_RGB_G_SWIZ_B_1;
329 fp->inst[1].inst4 = R500_ALPHA_OP_MAD |
330 R500_ALPHA_SWIZ_A_A |
331 R500_ALPHA_SWIZ_B_1;
332 fp->inst[1].inst5 = R500_ALU_RGBA_OP_MAD |
333 R500_ALU_RGBA_R_SWIZ_0 |
334 R500_ALU_RGBA_G_SWIZ_0 |
335 R500_ALU_RGBA_B_SWIZ_0 |
336 R500_ALU_RGBA_A_SWIZ_0;
337
338 fp->cs->nrslots = 2;
339 fp->translated = GL_TRUE;
340 }
341
342 /* static void emit_alu(struct r500_fragment_program *fp) {
343 * } */
344
345 static void emit_mov(struct r500_fragment_program *fp, int counter, struct prog_src_register src, GLuint dest) {
346 /* The r3xx shader uses MAD to implement MOV. We are using CMP, since
347 * it is technically more accurate and recommended by ATI/AMD. */
348 GLuint src_reg = make_src(fp, src);
349 fp->inst[counter].inst0 = R500_INST_TYPE_ALU | R500_INST_TEX_SEM_WAIT;
350 fp->inst[counter].inst1 = R500_RGB_ADDR0(src_reg);
351 fp->inst[counter].inst2 = R500_ALPHA_ADDR0(src_reg);
352 fp->inst[counter].inst3 = R500_ALU_RGB_SEL_A_SRC0
353 | MAKE_SWIZ_RGB_A(make_rgb_swizzle(src))
354 | R500_ALU_RGB_SEL_B_SRC0
355 | MAKE_SWIZ_RGB_B(make_rgb_swizzle(src))
356 | R500_ALU_RGB_OMOD_DISABLE;
357 fp->inst[counter].inst4 = R500_ALPHA_OP_CMP
358 | R500_ALPHA_ADDRD(dest)
359 | R500_ALPHA_SEL_A_SRC0 | MAKE_SWIZ_ALPHA_A(make_alpha_swizzle(src))
360 | R500_ALPHA_SEL_B_SRC0 | MAKE_SWIZ_ALPHA_B(make_alpha_swizzle(src))
361 | R500_ALPHA_OMOD_DISABLE;
362 fp->inst[counter].inst5 = R500_ALU_RGBA_OP_CMP
363 | R500_ALU_RGBA_ADDRD(dest)
364 | MAKE_SWIZ_RGBA_C(R500_SWIZ_RGB_ZERO)
365 | MAKE_SWIZ_ALPHA_C(R500_SWIZZLE_ZERO);
366 }
367
368 static GLboolean parse_program(struct r500_fragment_program *fp)
369 {
370 struct gl_fragment_program *mp = &fp->mesa_program;
371 const struct prog_instruction *inst = mp->Base.Instructions;
372 struct prog_instruction *fpi;
373 GLuint src[3], dest, temp[2];
374 int flags, mask, counter = 0;
375
376 if (!inst || inst[0].Opcode == OPCODE_END) {
377 ERROR("The program is empty!\n");
378 return GL_FALSE;
379 }
380
381 for (fpi = mp->Base.Instructions; fpi->Opcode != OPCODE_END; fpi++) {
382
383 if (fpi->Opcode != OPCODE_KIL) {
384 dest = make_dest(fp, fpi->DstReg);
385 mask = fpi->DstReg.WriteMask << 11;
386 }
387
388 switch (fpi->Opcode) {
389 case OPCODE_ABS:
390 emit_mov(fp, counter, fpi->SrcReg[0], dest);
391 fp->inst[counter].inst0 |= mask;
392 fp->inst[counter].inst3 |= R500_ALU_RGB_MOD_A_ABS
393 | R500_ALU_RGB_MOD_B_ABS;
394 fp->inst[counter].inst4 |= R500_ALPHA_MOD_A_ABS
395 | R500_ALPHA_MOD_B_ABS;
396 break;
397 case OPCODE_ADD:
398 src[0] = make_src(fp, fpi->SrcReg[0]);
399 src[1] = make_src(fp, fpi->SrcReg[1]);
400 /* Variation on MAD: 1*src0+src1 */
401 fp->inst[counter].inst0 = R500_INST_TYPE_ALU
402 | mask;
403 fp->inst[counter].inst1 = R500_RGB_ADDR0(src[0])
404 | R500_RGB_ADDR1(src[1]) | R500_RGB_ADDR2(0);
405 fp->inst[counter].inst2 = R500_ALPHA_ADDR0(src[0])
406 | R500_ALPHA_ADDR1(src[1]) | R500_ALPHA_ADDR2(0);
407 fp->inst[counter].inst3 = /* 1 */
408 MAKE_SWIZ_RGB_A(R500_SWIZ_RGB_ONE)
409 | R500_ALU_RGB_SEL_B_SRC0 | MAKE_SWIZ_RGB_B(make_rgb_swizzle(fpi->SrcReg[0]));
410 fp->inst[counter].inst4 = R500_ALPHA_OP_MAD
411 | R500_ALPHA_ADDRD(dest)
412 | MAKE_SWIZ_ALPHA_A(R500_SWIZZLE_ONE)
413 | R500_ALPHA_SEL_B_SRC0 | MAKE_SWIZ_ALPHA_B(make_alpha_swizzle(fpi->SrcReg[0]));
414 fp->inst[counter].inst5 = R500_ALU_RGBA_OP_MAD
415 | R500_ALU_RGBA_ADDRD(dest)
416 | R500_ALU_RGBA_SEL_C_SRC1
417 | MAKE_SWIZ_RGBA_C(make_rgb_swizzle(fpi->SrcReg[1]))
418 | R500_ALU_RGBA_ALPHA_SEL_C_SRC1
419 | MAKE_SWIZ_ALPHA_C(make_alpha_swizzle(fpi->SrcReg[1]));
420 break;
421 case OPCODE_CMP:
422 src[0] = make_src(fp, fpi->SrcReg[0]);
423 src[1] = make_src(fp, fpi->SrcReg[1]);
424 src[2] = make_src(fp, fpi->SrcReg[2]);
425 fp->inst[counter].inst0 = R500_INST_TYPE_ALU
426 | mask;
427 fp->inst[counter].inst1 = R500_RGB_ADDR0(src[0])
428 | R500_RGB_ADDR1(src[1]) | R500_RGB_ADDR2(src[2]);
429 fp->inst[counter].inst2 = R500_ALPHA_ADDR0(src[0])
430 | R500_ALPHA_ADDR1(src[1]) | R500_ALPHA_ADDR2(src[2]);
431 fp->inst[counter].inst3 = R500_ALU_RGB_SEL_A_SRC0
432 | MAKE_SWIZ_RGB_A(make_rgb_swizzle(fpi->SrcReg[0]))
433 | R500_ALU_RGB_SEL_B_SRC1 | MAKE_SWIZ_RGB_B(make_rgb_swizzle(fpi->SrcReg[1]));
434 fp->inst[counter].inst4 = R500_ALPHA_OP_CMP
435 | R500_ALPHA_ADDRD(dest)
436 | R500_ALPHA_SEL_A_SRC0 | MAKE_SWIZ_ALPHA_A(make_alpha_swizzle(fpi->SrcReg[0]))
437 | R500_ALPHA_SEL_B_SRC1 | MAKE_SWIZ_ALPHA_B(make_alpha_swizzle(fpi->SrcReg[1]));
438 fp->inst[counter].inst5 = R500_ALU_RGBA_OP_CMP
439 | R500_ALU_RGBA_ADDRD(dest)
440 | R500_ALU_RGBA_SEL_C_SRC2
441 | MAKE_SWIZ_RGBA_C(make_rgb_swizzle(fpi->SrcReg[2]))
442 | R500_ALU_RGBA_ALPHA_SEL_C_SRC2
443 | MAKE_SWIZ_ALPHA_C(make_alpha_swizzle(fpi->SrcReg[2]));
444 break;
445 case OPCODE_DP3:
446 src[0] = make_src(fp, fpi->SrcReg[0]);
447 src[1] = make_src(fp, fpi->SrcReg[1]);
448 fp->inst[counter].inst0 = R500_INST_TYPE_ALU
449 | R500_INST_TEX_SEM_WAIT | mask;
450 fp->inst[counter].inst1 = R500_RGB_ADDR0(src[0])
451 | R500_RGB_ADDR1(src[1]);
452 fp->inst[counter].inst2 = R500_ALPHA_ADDR0(src[0])
453 | R500_ALPHA_ADDR1(src[1]);
454 fp->inst[counter].inst3 = R500_ALU_RGB_SEL_A_SRC0
455 | MAKE_SWIZ_RGB_A(make_rgb_swizzle(fpi->SrcReg[0]))
456 | R500_ALU_RGB_SEL_B_SRC1 | MAKE_SWIZ_RGB_B(make_rgb_swizzle(fpi->SrcReg[1]));
457 fp->inst[counter].inst4 = R500_ALPHA_OP_DP
458 | R500_ALPHA_ADDRD(dest)
459 | R500_ALPHA_SEL_A_SRC0 | MAKE_SWIZ_ALPHA_A(make_alpha_swizzle(fpi->SrcReg[0]))
460 | R500_ALPHA_SEL_B_SRC1 | MAKE_SWIZ_ALPHA_B(make_alpha_swizzle(fpi->SrcReg[1]));
461 fp->inst[counter].inst5 = R500_ALU_RGBA_OP_DP3
462 | R500_ALU_RGBA_ADDRD(dest);
463 break;
464 case OPCODE_DP4:
465 src[0] = make_src(fp, fpi->SrcReg[0]);
466 src[1] = make_src(fp, fpi->SrcReg[1]);
467 /* Based on DP3 */
468 fp->inst[counter].inst0 = R500_INST_TYPE_ALU
469 | R500_INST_TEX_SEM_WAIT | mask;
470 fp->inst[counter].inst1 = R500_RGB_ADDR0(src[0])
471 | R500_RGB_ADDR1(src[1]);
472 fp->inst[counter].inst2 = R500_ALPHA_ADDR0(src[0])
473 | R500_ALPHA_ADDR1(src[1]);
474 fp->inst[counter].inst3 = R500_ALU_RGB_SEL_A_SRC0
475 | MAKE_SWIZ_RGB_A(make_rgb_swizzle(fpi->SrcReg[0]))
476 | R500_ALU_RGB_SEL_B_SRC1 | MAKE_SWIZ_RGB_B(make_rgb_swizzle(fpi->SrcReg[1]));
477 fp->inst[counter].inst4 = R500_ALPHA_OP_DP
478 | R500_ALPHA_ADDRD(dest)
479 | R500_ALPHA_SEL_A_SRC0 | MAKE_SWIZ_ALPHA_A(make_alpha_swizzle(fpi->SrcReg[0]))
480 | R500_ALPHA_SEL_B_SRC1 | MAKE_SWIZ_ALPHA_B(make_alpha_swizzle(fpi->SrcReg[1]));
481 fp->inst[counter].inst5 = R500_ALU_RGBA_OP_DP4
482 | R500_ALU_RGBA_ADDRD(dest);
483 break;
484 case OPCODE_MAD:
485 src[0] = make_src(fp, fpi->SrcReg[0]);
486 src[1] = make_src(fp, fpi->SrcReg[1]);
487 src[2] = make_src(fp, fpi->SrcReg[2]);
488 fp->inst[counter].inst0 = R500_INST_TYPE_ALU
489 | mask;
490 fp->inst[counter].inst1 = R500_RGB_ADDR0(src[0])
491 | R500_RGB_ADDR1(src[1]) | R500_RGB_ADDR2(src[2]);
492 fp->inst[counter].inst2 = R500_ALPHA_ADDR0(src[0])
493 | R500_ALPHA_ADDR1(src[1]) | R500_ALPHA_ADDR2(src[2]);
494 fp->inst[counter].inst3 = R500_ALU_RGB_SEL_A_SRC0
495 | MAKE_SWIZ_RGB_A(make_rgb_swizzle(fpi->SrcReg[0]))
496 | R500_ALU_RGB_SEL_B_SRC1 | MAKE_SWIZ_RGB_B(make_rgb_swizzle(fpi->SrcReg[1]));
497 fp->inst[counter].inst4 = R500_ALPHA_OP_MAD
498 | R500_ALPHA_ADDRD(dest)
499 | R500_ALPHA_SEL_A_SRC0 | MAKE_SWIZ_ALPHA_A(make_alpha_swizzle(fpi->SrcReg[0]))
500 | R500_ALPHA_SEL_B_SRC1 | MAKE_SWIZ_ALPHA_B(make_alpha_swizzle(fpi->SrcReg[1]));
501 fp->inst[counter].inst5 = R500_ALU_RGBA_OP_MAD
502 | R500_ALU_RGBA_ADDRD(dest)
503 | R500_ALU_RGBA_SEL_C_SRC2
504 | MAKE_SWIZ_RGBA_C(make_rgb_swizzle(fpi->SrcReg[2]))
505 | R500_ALU_RGBA_ALPHA_SEL_C_SRC2
506 | MAKE_SWIZ_ALPHA_C(make_alpha_swizzle(fpi->SrcReg[2]));
507 break;
508 case OPCODE_MAX:
509 src[0] = make_src(fp, fpi->SrcReg[0]);
510 src[1] = make_src(fp, fpi->SrcReg[0]);
511 fp->inst[counter].inst0 = R500_INST_TYPE_ALU | mask;
512 fp->inst[counter].inst1 = R500_RGB_ADDR0(src[0]) | R500_RGB_ADDR1(src[1]);
513 fp->inst[counter].inst2 = R500_ALPHA_ADDR0(src[0]) | R500_ALPHA_ADDR1(src[1]);
514 fp->inst[counter].inst3 = R500_ALU_RGB_SEL_A_SRC0
515 | MAKE_SWIZ_RGB_A(make_rgb_swizzle(fpi->SrcReg[0]))
516 | R500_ALU_RGB_SEL_B_SRC1
517 | MAKE_SWIZ_RGB_B(make_rgb_swizzle(fpi->SrcReg[1]));
518 fp->inst[counter].inst4 = R500_ALPHA_OP_MAX
519 | R500_ALPHA_ADDRD(dest)
520 | R500_ALPHA_SEL_A_SRC0 | MAKE_SWIZ_ALPHA_A(make_alpha_swizzle(fpi->SrcReg[0]))
521 | R500_ALPHA_SEL_B_SRC1 | MAKE_SWIZ_ALPHA_B(make_alpha_swizzle(fpi->SrcReg[1]));
522 fp->inst[counter].inst5 = R500_ALU_RGBA_OP_MAX
523 | R500_ALU_RGBA_ADDRD(dest);
524 break;
525 case OPCODE_MIN:
526 src[0] = make_src(fp, fpi->SrcReg[0]);
527 src[1] = make_src(fp, fpi->SrcReg[0]);
528 fp->inst[counter].inst0 = R500_INST_TYPE_ALU | mask;
529 fp->inst[counter].inst1 = R500_RGB_ADDR0(src[0]) | R500_RGB_ADDR1(src[1]);
530 fp->inst[counter].inst2 = R500_ALPHA_ADDR0(src[0]) | R500_ALPHA_ADDR1(src[1]);
531 fp->inst[counter].inst3 = R500_ALU_RGB_SEL_A_SRC0
532 | MAKE_SWIZ_RGB_A(make_rgb_swizzle(fpi->SrcReg[0]))
533 | R500_ALU_RGB_SEL_B_SRC1
534 | MAKE_SWIZ_RGB_B(make_rgb_swizzle(fpi->SrcReg[1]));
535 fp->inst[counter].inst4 = R500_ALPHA_OP_MIN
536 | R500_ALPHA_ADDRD(dest)
537 | R500_ALPHA_SEL_A_SRC0 | MAKE_SWIZ_ALPHA_A(make_alpha_swizzle(fpi->SrcReg[0]))
538 | R500_ALPHA_SEL_B_SRC1 | MAKE_SWIZ_ALPHA_B(make_alpha_swizzle(fpi->SrcReg[1]));
539 fp->inst[counter].inst5 = R500_ALU_RGBA_OP_MIN
540 | R500_ALU_RGBA_ADDRD(dest);
541 break;
542 case OPCODE_MOV:
543 emit_mov(fp, counter, fpi->SrcReg[0], dest);
544 fp->inst[counter].inst0 |= mask;
545 break;
546 case OPCODE_MUL:
547 src[0] = make_src(fp, fpi->SrcReg[0]);
548 src[1] = make_src(fp, fpi->SrcReg[1]);
549 /* Variation on MAD: src0*src1+0 */
550 fp->inst[counter].inst0 = R500_INST_TYPE_ALU
551 | R500_INST_TEX_SEM_WAIT | mask;
552 fp->inst[counter].inst1 = R500_RGB_ADDR0(src[0])
553 | R500_RGB_ADDR1(src[1]);
554 fp->inst[counter].inst2 = R500_ALPHA_ADDR0(src[0])
555 | R500_ALPHA_ADDR1(src[1]);
556 fp->inst[counter].inst3 = R500_ALU_RGB_SEL_A_SRC0
557 | MAKE_SWIZ_RGB_A(make_rgb_swizzle(fpi->SrcReg[0]))
558 | R500_ALU_RGB_SEL_B_SRC1 | MAKE_SWIZ_RGB_B(make_rgb_swizzle(fpi->SrcReg[1]));
559 fp->inst[counter].inst4 = R500_ALPHA_OP_MAD
560 | R500_ALPHA_ADDRD(dest)
561 | R500_ALPHA_SEL_A_SRC0 | MAKE_SWIZ_ALPHA_A(make_alpha_swizzle(fpi->SrcReg[0]))
562 | R500_ALPHA_SEL_B_SRC1 | MAKE_SWIZ_ALPHA_B(make_alpha_swizzle(fpi->SrcReg[1]));
563 fp->inst[counter].inst5 = R500_ALU_RGBA_OP_MAD
564 | R500_ALU_RGBA_ADDRD(dest)
565 // | R500_ALU_RGBA_SEL_C_SRC2
566 | MAKE_SWIZ_RGBA_C(R500_SWIZ_RGB_ZERO)
567 // | R500_ALU_RGBA_ALPHA_SEL_C_SRC2
568 | MAKE_SWIZ_ALPHA_C(R500_SWIZZLE_ZERO);
569 break;
570 case OPCODE_SUB:
571 src[0] = make_src(fp, fpi->SrcReg[0]);
572 src[1] = make_src(fp, fpi->SrcReg[1]);
573 /* Variation on MAD: 1*src0-src1 */
574 fp->inst[counter].inst0 = R500_INST_TYPE_ALU
575 | mask;
576 fp->inst[counter].inst1 = R500_RGB_ADDR1(src[0])
577 | R500_RGB_ADDR2(src[1]);
578 fp->inst[counter].inst2 = R500_ALPHA_ADDR1(src[0])
579 | R500_ALPHA_ADDR2(src[1]);
580 fp->inst[counter].inst3 = /* 1 */
581 MAKE_SWIZ_RGB_A(R500_SWIZ_RGB_ONE)
582 | R500_ALU_RGB_SEL_B_SRC1 | MAKE_SWIZ_RGB_B(make_rgb_swizzle(fpi->SrcReg[0]));
583 fp->inst[counter].inst4 = R500_ALPHA_OP_MAD
584 | R500_ALPHA_ADDRD(dest)
585 | R500_ALPHA_SEL_A_SRC0 | MAKE_SWIZ_ALPHA_A(R500_SWIZZLE_ONE)
586 | R500_ALPHA_SEL_B_SRC1 | MAKE_SWIZ_ALPHA_B(make_alpha_swizzle(fpi->SrcReg[0]));
587 fp->inst[counter].inst5 = R500_ALU_RGBA_OP_MAD
588 | R500_ALU_RGBA_ADDRD(dest)
589 | R500_ALU_RGBA_SEL_C_SRC2
590 | MAKE_SWIZ_RGBA_C(make_rgb_swizzle(fpi->SrcReg[1]))
591 | R500_ALU_RGBA_MOD_C_NEG
592 | R500_ALU_RGBA_ALPHA_SEL_C_SRC2
593 | MAKE_SWIZ_ALPHA_C(make_alpha_swizzle(fpi->SrcReg[1]))
594 | R500_ALU_RGBA_ALPHA_MOD_C_NEG;
595 break;
596 case OPCODE_TEX:
597 emit_tex(fp, fpi, OPCODE_TEX, dest, counter);
598 break;
599 case OPCODE_TXB:
600 emit_tex(fp, fpi, OPCODE_TXB, dest, counter);
601 break;
602 case OPCODE_TXP:
603 emit_tex(fp, fpi, OPCODE_TXP, dest, counter);
604 break;
605 default:
606 ERROR("unknown fpi->Opcode %d\n", fpi->Opcode);
607 break;
608 }
609
610 /* Finishing touches */
611 if (fpi->SaturateMode == SATURATE_ZERO_ONE) {
612 fp->inst[counter].inst0 |= R500_INST_RGB_CLAMP | R500_INST_ALPHA_CLAMP;
613 }
614 if (fpi->DstReg.File == PROGRAM_OUTPUT) {
615 fp->inst[counter].inst0 |= R500_INST_TYPE_OUT
616 | R500_INST_RGB_OMASK_R | R500_INST_RGB_OMASK_G
617 | R500_INST_RGB_OMASK_B | R500_INST_ALPHA_OMASK;
618 }
619
620 counter++;
621
622 if (fp->error)
623 return GL_FALSE;
624
625 }
626
627 /* Finish him! (If it's an ALU/OUT instruction...) */
628 if ((fp->inst[counter-1].inst0 & 0x3) <= 1) {
629 fp->inst[counter-1].inst0 |= R500_INST_TYPE_OUT
630 | R500_INST_TEX_SEM_WAIT | R500_INST_LAST;
631 } else {
632 /* We still need to put an output inst, right? */
633 fp->inst[counter].inst0 = R500_INST_TYPE_OUT
634 | R500_INST_TEX_SEM_WAIT | R500_INST_LAST
635 | R500_INST_RGB_OMASK_R | R500_INST_RGB_OMASK_G
636 | R500_INST_RGB_OMASK_B | R500_INST_ALPHA_OMASK;
637 fp->inst[counter].inst1 = R500_RGB_ADDR0(dest);
638 fp->inst[counter].inst2 = R500_ALPHA_ADDR0(dest);
639 fp->inst[counter].inst3 = R500_ALU_RGB_SEL_A_SRC0
640 | MAKE_SWIZ_RGB_A(R500_SWIZ_RGB_RGB)
641 | R500_ALU_RGB_SEL_B_SRC0
642 | MAKE_SWIZ_RGB_B(R500_SWIZ_RGB_ONE);
643 fp->inst[counter].inst4 = R500_ALPHA_OP_MAD
644 | R500_ALPHA_ADDRD(0)
645 | R500_ALPHA_SEL_A_SRC0 | R500_ALPHA_SEL_B_SRC0
646 | R500_ALPHA_SWIZ_A_A | R500_ALPHA_SWIZ_B_1;
647 fp->inst[counter].inst5 = R500_ALU_RGBA_OP_MAD
648 | R500_ALU_RGBA_ADDRD(0)
649 | MAKE_SWIZ_RGBA_C(R500_SWIZ_RGB_ZERO)
650 | MAKE_SWIZ_ALPHA_C(R500_SWIZZLE_ZERO);
651 counter++;
652 }
653
654 fp->cs->nrslots = counter;
655
656 fp->max_temp_idx++;
657
658 return GL_TRUE;
659 }
660
661 static void init_program(r300ContextPtr r300, struct r500_fragment_program *fp)
662 {
663 struct r300_pfs_compile_state *cs = NULL;
664 struct gl_fragment_program *mp = &fp->mesa_program;
665 struct prog_instruction *fpi;
666 GLuint InputsRead = mp->Base.InputsRead;
667 GLuint temps_used = 0; /* for fp->temps[] */
668 int i, j;
669
670 /* New compile, reset tracking data */
671 fp->optimization =
672 driQueryOptioni(&r300->radeon.optionCache, "fp_optimization");
673 fp->translated = GL_FALSE;
674 fp->error = GL_FALSE;
675 fp->cs = cs = &(R300_CONTEXT(fp->ctx)->state.pfs_compile);
676 fp->cur_node = 0;
677 fp->first_node_has_tex = 0;
678 fp->const_nr = 0;
679 /* Size of pixel stack, plus 1. */
680 fp->max_temp_idx = 1;
681 /* Temp register offset. */
682 fp->temp_reg_offset = 0;
683 fp->node[0].alu_end = -1;
684 fp->node[0].tex_end = -1;
685
686 _mesa_memset(cs, 0, sizeof(*fp->cs));
687 for (i = 0; i < PFS_MAX_ALU_INST; i++) {
688 for (j = 0; j < 3; j++) {
689 cs->slot[i].vsrc[j] = SRC_CONST;
690 cs->slot[i].ssrc[j] = SRC_CONST;
691 }
692 }
693
694 /* Work out what temps the Mesa inputs correspond to, this must match
695 * what setup_rs_unit does, which shouldn't be a problem as rs_unit
696 * configures itself based on the fragprog's InputsRead
697 *
698 * NOTE: this depends on get_hw_temp() allocating registers in order,
699 * starting from register 0, so we're just going to do that instead.
700 */
701
702 /* Texcoords come first */
703 for (i = 0; i < fp->ctx->Const.MaxTextureUnits; i++) {
704 if (InputsRead & (FRAG_BIT_TEX0 << i)) {
705 cs->inputs[FRAG_ATTRIB_TEX0 + i].refcount = 0;
706 cs->inputs[FRAG_ATTRIB_TEX0 + i].reg =
707 fp->temp_reg_offset;
708 fp->temp_reg_offset++;
709 }
710 }
711 InputsRead &= ~FRAG_BITS_TEX_ANY;
712
713 /* fragment position treated as a texcoord */
714 if (InputsRead & FRAG_BIT_WPOS) {
715 cs->inputs[FRAG_ATTRIB_WPOS].refcount = 0;
716 cs->inputs[FRAG_ATTRIB_WPOS].reg =
717 fp->temp_reg_offset;
718 fp->temp_reg_offset++;
719 }
720 InputsRead &= ~FRAG_BIT_WPOS;
721
722 /* Then primary colour */
723 if (InputsRead & FRAG_BIT_COL0) {
724 cs->inputs[FRAG_ATTRIB_COL0].refcount = 0;
725 cs->inputs[FRAG_ATTRIB_COL0].reg =
726 fp->temp_reg_offset;
727 fp->temp_reg_offset++;
728 }
729 InputsRead &= ~FRAG_BIT_COL0;
730
731 /* Secondary color */
732 if (InputsRead & FRAG_BIT_COL1) {
733 cs->inputs[FRAG_ATTRIB_COL1].refcount = 0;
734 cs->inputs[FRAG_ATTRIB_COL1].reg =
735 fp->temp_reg_offset;
736 fp->temp_reg_offset++;
737 }
738 InputsRead &= ~FRAG_BIT_COL1;
739
740 /* Anything else */
741 if (InputsRead) {
742 WARN_ONCE("Don't know how to handle inputs 0x%x\n", InputsRead);
743 /* force read from hwreg 0 for now */
744 for (i = 0; i < 32; i++)
745 if (InputsRead & (1 << i))
746 cs->inputs[i].reg = 0;
747 }
748
749 /* Pre-parse the mesa program, grabbing refcounts on input/temp regs.
750 * That way, we can free up the reg when it's no longer needed
751 */
752 if (!mp->Base.Instructions) {
753 ERROR("No instructions found in program, going to go die now.\n");
754 return;
755 }
756
757 #if 0
758 for (fpi = mp->Base.Instructions; fpi->Opcode != OPCODE_END; fpi++) {
759 int idx;
760 for (i = 0; i < 3; i++) {
761 idx = fpi->SrcReg[i].Index;
762 if (fpi->SrcReg[i].File == PROGRAM_INPUT) {
763 cs->inputs[idx].refcount++;
764 if (fp->max_temp_idx < idx)
765 fp->max_temp_idx = idx;
766 }
767 }
768 }
769 #endif
770
771 fp->max_temp_idx = fp->temp_reg_offset + 1;
772
773 cs->temp_in_use = temps_used;
774 }
775
776 static void update_params(struct r500_fragment_program *fp)
777 {
778 struct gl_fragment_program *mp = &fp->mesa_program;
779
780 /* Ask Mesa nicely to fill in ParameterValues for us */
781 if (mp->Base.Parameters)
782 _mesa_load_state_parameters(fp->ctx, mp->Base.Parameters);
783 }
784
785 void r500TranslateFragmentShader(r300ContextPtr r300,
786 struct r500_fragment_program *fp)
787 {
788
789 struct r300_pfs_compile_state *cs = NULL;
790
791 if (!fp->translated) {
792
793 /* I need to see what I'm working with! */
794 fprintf(stderr, "Mesa program:\n");
795 fprintf(stderr, "-------------\n");
796 _mesa_print_program(&fp->mesa_program.Base);
797 fflush(stdout);
798
799 init_program(r300, fp);
800 cs = fp->cs;
801
802 if (parse_program(fp) == GL_FALSE) {
803 ERROR("Huh. Couldn't parse program. There should be additional errors explaining why.\nUsing dumb shader...\n");
804 dumb_shader(fp);
805 return;
806 }
807
808 fp->translated = GL_TRUE;
809 r300UpdateStateParameters(fp->ctx, _NEW_PROGRAM);
810 }
811
812 update_params(fp);
813 }