e1dc23102704a5acf4a3530579ecf8a5a66f5051
[mesa.git] / src / mesa / drivers / dri / r300 / r300_fragprog_common.c
1 /*
2 * Copyright (C) 2009 Maciej Cencora <m.cencora@gmail.com>
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 * Fragment program compiler. Perform transformations on the intermediate
32 * representation until the program is in a form where we can translate
33 * it more or less directly into machine-readable form.
34 *
35 * \author Ben Skeggs <darktama@iinet.net.au>
36 * \author Jerome Glisse <j.glisse@gmail.com>
37 */
38
39 #include "r300_fragprog_common.h"
40
41 #include "shader/program.h"
42 #include "shader/prog_parameter.h"
43 #include "shader/prog_print.h"
44
45 #include "compiler/radeon_compiler.h"
46
47 #include "r300_state.h"
48
49
50 static GLuint build_dtm(GLuint depthmode)
51 {
52 switch(depthmode) {
53 default:
54 case GL_LUMINANCE: return 0;
55 case GL_INTENSITY: return 1;
56 case GL_ALPHA: return 2;
57 }
58 }
59
60 static GLuint build_func(GLuint comparefunc)
61 {
62 return comparefunc - GL_NEVER;
63 }
64
65 /**
66 * Collect all external state that is relevant for compiling the given
67 * fragment program.
68 */
69 static void build_state(
70 r300ContextPtr r300,
71 struct gl_fragment_program *fp,
72 struct r300_fragment_program_external_state *state)
73 {
74 int unit;
75
76 _mesa_bzero(state, sizeof(*state));
77
78 for(unit = 0; unit < 16; ++unit) {
79 if (fp->Base.ShadowSamplers & (1 << unit)) {
80 struct gl_texture_object* tex = r300->radeon.glCtx->Texture.Unit[unit]._Current;
81
82 state->unit[unit].depth_texture_mode = build_dtm(tex->DepthMode);
83 state->unit[unit].texture_compare_func = build_func(tex->CompareFunc);
84 }
85 }
86 }
87
88
89 /**
90 * Transform the program to support fragment.position.
91 *
92 * Introduce a small fragment at the start of the program that will be
93 * the only code that directly reads the FRAG_ATTRIB_WPOS input.
94 * All other code pieces that reference that input will be rewritten
95 * to read from a newly allocated temporary.
96 *
97 */
98 static void insert_WPOS_trailer(struct r300_fragment_program_compiler *compiler, struct r300_fragment_program * fp)
99 {
100 int i;
101
102 if (!(compiler->Base.Program.InputsRead & FRAG_BIT_WPOS)) {
103 fp->wpos_attr = FRAG_ATTRIB_MAX;
104 return;
105 }
106
107 for (i = FRAG_ATTRIB_TEX0; i <= FRAG_ATTRIB_TEX7; ++i)
108 {
109 if (!(compiler->Base.Program.InputsRead & (1 << i))) {
110 fp->wpos_attr = i;
111 break;
112 }
113 }
114
115 rc_transform_fragment_wpos(&compiler->Base, FRAG_ATTRIB_WPOS, fp->wpos_attr);
116 }
117
118 /**
119 * Rewrite fragment.fogcoord to use a texture coordinate slot.
120 * Note that fogcoord is forced into an X001 pattern, and this enforcement
121 * is done here.
122 *
123 * See also the counterpart rewriting for vertex programs.
124 */
125 static void rewriteFog(struct r300_fragment_program_compiler *compiler, struct r300_fragment_program * fp)
126 {
127 struct prog_src_register src;
128 int i;
129
130 if (!(compiler->Base.Program.InputsRead & FRAG_BIT_FOGC)) {
131 fp->fog_attr = FRAG_ATTRIB_MAX;
132 return;
133 }
134
135 for (i = FRAG_ATTRIB_TEX0; i <= FRAG_ATTRIB_TEX7; ++i)
136 {
137 if (!(compiler->Base.Program.InputsRead & (1 << i))) {
138 fp->fog_attr = i;
139 break;
140 }
141 }
142
143 memset(&src, 0, sizeof(src));
144 src.File = PROGRAM_INPUT;
145 src.Index = fp->fog_attr;
146 src.Swizzle = MAKE_SWIZZLE4(SWIZZLE_X, SWIZZLE_ZERO, SWIZZLE_ZERO, SWIZZLE_ONE);
147 rc_move_input(&compiler->Base, FRAG_ATTRIB_FOGC, src);
148 }
149
150
151 /**
152 * Reserve hardware temporary registers for the program inputs.
153 *
154 * @note This allocation is performed explicitly, because the order of inputs
155 * is determined by the RS hardware.
156 */
157 static void allocate_hw_inputs(
158 struct r300_fragment_program_compiler * c,
159 void (*allocate)(void * data, unsigned input, unsigned hwreg),
160 void * mydata)
161 {
162 GLuint InputsRead = c->Base.Program.InputsRead;
163 int i;
164 GLuint hwindex = 0;
165
166 /* Primary colour */
167 if (InputsRead & FRAG_BIT_COL0)
168 allocate(mydata, FRAG_ATTRIB_COL0, hwindex++);
169 InputsRead &= ~FRAG_BIT_COL0;
170
171 /* Secondary color */
172 if (InputsRead & FRAG_BIT_COL1)
173 allocate(mydata, FRAG_ATTRIB_COL1, hwindex++);
174 InputsRead &= ~FRAG_BIT_COL1;
175
176 /* Texcoords */
177 for (i = 0; i < 8; i++) {
178 if (InputsRead & (FRAG_BIT_TEX0 << i))
179 allocate(mydata, FRAG_ATTRIB_TEX0+i, hwindex++);
180 }
181 InputsRead &= ~FRAG_BITS_TEX_ANY;
182
183 /* Fogcoords treated as a texcoord */
184 if (InputsRead & FRAG_BIT_FOGC)
185 allocate(mydata, FRAG_ATTRIB_FOGC, hwindex++);
186 InputsRead &= ~FRAG_BIT_FOGC;
187
188 /* fragment position treated as a texcoord */
189 if (InputsRead & FRAG_BIT_WPOS)
190 allocate(mydata, FRAG_ATTRIB_WPOS, hwindex++);
191 InputsRead &= ~FRAG_BIT_WPOS;
192
193 /* Anything else */
194 if (InputsRead)
195 rc_error(&c->Base, "Don't know how to handle inputs 0x%x\n", InputsRead);
196 }
197
198
199 static void translate_fragment_program(GLcontext *ctx, struct r300_fragment_program_cont *cont, struct r300_fragment_program *fp)
200 {
201 r300ContextPtr r300 = R300_CONTEXT(ctx);
202 struct r300_fragment_program_compiler compiler;
203
204 rc_init(&compiler.Base);
205 compiler.Base.Debug = (RADEON_DEBUG & DEBUG_PIXEL) ? GL_TRUE : GL_FALSE;
206
207 compiler.code = &fp->code;
208 compiler.state = fp->state;
209 compiler.is_r500 = (r300->radeon.radeonScreen->chip_family >= CHIP_FAMILY_RV515) ? GL_TRUE : GL_FALSE;
210 compiler.OutputDepth = FRAG_RESULT_DEPTH;
211 compiler.OutputColor = FRAG_RESULT_COLOR;
212 compiler.AllocateHwInputs = &allocate_hw_inputs;
213
214 if (compiler.Base.Debug) {
215 fflush(stderr);
216 _mesa_printf("Fragment Program: Initial program:\n");
217 _mesa_print_program(&cont->Base.Base);
218 fflush(stderr);
219 }
220
221 rc_mesa_to_rc_program(&compiler.Base, &cont->Base.Base);
222
223 insert_WPOS_trailer(&compiler, fp);
224
225 rewriteFog(&compiler, fp);
226
227 r3xx_compile_fragment_program(&compiler);
228 fp->error = compiler.Base.Error;
229
230 fp->InputsRead = compiler.Base.Program.InputsRead;
231
232 rc_destroy(&compiler.Base);
233 }
234
235 struct r300_fragment_program *r300SelectAndTranslateFragmentShader(GLcontext *ctx)
236 {
237 r300ContextPtr r300 = R300_CONTEXT(ctx);
238 struct r300_fragment_program_cont *fp_list;
239 struct r300_fragment_program *fp;
240 struct r300_fragment_program_external_state state;
241
242 fp_list = (struct r300_fragment_program_cont *)ctx->FragmentProgram._Current;
243 build_state(r300, ctx->FragmentProgram._Current, &state);
244
245 fp = fp_list->progs;
246 while (fp) {
247 if (_mesa_memcmp(&fp->state, &state, sizeof(state)) == 0) {
248 return r300->selected_fp = fp;
249 }
250 fp = fp->next;
251 }
252
253 fp = _mesa_calloc(sizeof(struct r300_fragment_program));
254
255 fp->state = state;
256
257 fp->next = fp_list->progs;
258 fp_list->progs = fp;
259
260 translate_fragment_program(ctx, fp_list, fp);
261
262 return r300->selected_fp = fp;
263 }