projects / mesa.git / blob
? search:


35eb9cd8d08ed55a75c781771302e65112e82102
[mesa.git] / src / gallium / auxiliary / gallivm / lp_bld_sample_soa.c
1 /**************************************************************************
2 *
3 * Copyright 2009 VMware, Inc.
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 VMWARE 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 * @file
30 * Texture sampling -- SoA.
31 *
32 * @author Jose Fonseca <jfonseca@vmware.com>
33 * @author Brian Paul <brianp@vmware.com>
34 */
35
36 #include "pipe/p_defines.h"
37 #include "pipe/p_state.h"
38 #include "pipe/p_shader_tokens.h"
39 #include "util/u_debug.h"
40 #include "util/u_dump.h"
41 #include "util/u_memory.h"
42 #include "util/u_math.h"
43 #include "util/u_format.h"
44 #include "util/u_cpu_detect.h"
45 #include "lp_bld_debug.h"
46 #include "lp_bld_type.h"
47 #include "lp_bld_const.h"
48 #include "lp_bld_conv.h"
49 #include "lp_bld_arit.h"
50 #include "lp_bld_bitarit.h"
51 #include "lp_bld_logic.h"
52 #include "lp_bld_printf.h"
53 #include "lp_bld_swizzle.h"
54 #include "lp_bld_flow.h"
55 #include "lp_bld_gather.h"
56 #include "lp_bld_format.h"
57 #include "lp_bld_sample.h"
58 #include "lp_bld_sample_aos.h"
59 #include "lp_bld_struct.h"
60 #include "lp_bld_quad.h"
61 #include "lp_bld_pack.h"
62
63
64 /**
65 * Generate code to fetch a texel from a texture at int coords (x, y, z).
66 * The computation depends on whether the texture is 1D, 2D or 3D.
67 * The result, texel, will be float vectors:
68 * texel[0] = red values
69 * texel[1] = green values
70 * texel[2] = blue values
71 * texel[3] = alpha values
72 */
73 static void
74 lp_build_sample_texel_soa(struct lp_build_sample_context *bld,
75 unsigned unit,
76 LLVMValueRef width,
77 LLVMValueRef height,
78 LLVMValueRef depth,
79 LLVMValueRef x,
80 LLVMValueRef y,
81 LLVMValueRef z,
82 LLVMValueRef y_stride,
83 LLVMValueRef z_stride,
84 LLVMValueRef data_ptr,
85 LLVMValueRef mipoffsets,
86 LLVMValueRef texel_out[4])
87 {
88 const struct lp_sampler_static_state *static_state = bld->static_state;
89 const unsigned dims = bld->dims;
90 struct lp_build_context *int_coord_bld = &bld->int_coord_bld;
91 LLVMBuilderRef builder = bld->gallivm->builder;
92 LLVMValueRef offset;
93 LLVMValueRef i, j;
94 LLVMValueRef use_border = NULL;
95
96 /* use_border = x < 0 || x >= width || y < 0 || y >= height */
97 if (lp_sampler_wrap_mode_uses_border_color(static_state->wrap_s,
98 static_state->min_img_filter,
99 static_state->mag_img_filter)) {
100 LLVMValueRef b1, b2;
101 b1 = lp_build_cmp(int_coord_bld, PIPE_FUNC_LESS, x, int_coord_bld->zero);
102 b2 = lp_build_cmp(int_coord_bld, PIPE_FUNC_GEQUAL, x, width);
103 use_border = LLVMBuildOr(builder, b1, b2, "b1_or_b2");
104 }
105
106 if (dims >= 2 &&
107 lp_sampler_wrap_mode_uses_border_color(static_state->wrap_t,
108 static_state->min_img_filter,
109 static_state->mag_img_filter)) {
110 LLVMValueRef b1, b2;
111 b1 = lp_build_cmp(int_coord_bld, PIPE_FUNC_LESS, y, int_coord_bld->zero);
112 b2 = lp_build_cmp(int_coord_bld, PIPE_FUNC_GEQUAL, y, height);
113 if (use_border) {
114 use_border = LLVMBuildOr(builder, use_border, b1, "ub_or_b1");
115 use_border = LLVMBuildOr(builder, use_border, b2, "ub_or_b2");
116 }
117 else {
118 use_border = LLVMBuildOr(builder, b1, b2, "b1_or_b2");
119 }
120 }
121
122 if (dims == 3 &&
123 lp_sampler_wrap_mode_uses_border_color(static_state->wrap_r,
124 static_state->min_img_filter,
125 static_state->mag_img_filter)) {
126 LLVMValueRef b1, b2;
127 b1 = lp_build_cmp(int_coord_bld, PIPE_FUNC_LESS, z, int_coord_bld->zero);
128 b2 = lp_build_cmp(int_coord_bld, PIPE_FUNC_GEQUAL, z, depth);
129 if (use_border) {
130 use_border = LLVMBuildOr(builder, use_border, b1, "ub_or_b1");
131 use_border = LLVMBuildOr(builder, use_border, b2, "ub_or_b2");
132 }
133 else {
134 use_border = LLVMBuildOr(builder, b1, b2, "b1_or_b2");
135 }
136 }
137
138 /* convert x,y,z coords to linear offset from start of texture, in bytes */
139 lp_build_sample_offset(&bld->int_coord_bld,
140 bld->format_desc,
141 x, y, z, y_stride, z_stride,
142 &offset, &i, &j);
143 if (mipoffsets) {
144 offset = lp_build_add(&bld->int_coord_bld, offset, mipoffsets);
145 }
146
147 if (use_border) {
148 /* If we can sample the border color, it means that texcoords may
149 * lie outside the bounds of the texture image. We need to do
150 * something to prevent reading out of bounds and causing a segfault.
151 *
152 * Simply AND the texture coords with !use_border. This will cause
153 * coords which are out of bounds to become zero. Zero's guaranteed
154 * to be inside the texture image.
155 */
156 offset = lp_build_andnot(&bld->int_coord_bld, offset, use_border);
157 }
158
159 lp_build_fetch_rgba_soa(bld->gallivm,
160 bld->format_desc,
161 bld->texel_type,
162 data_ptr, offset,
163 i, j,
164 texel_out);
165
166 /*
167 * Note: if we find an app which frequently samples the texture border
168 * we might want to implement a true conditional here to avoid sampling
169 * the texture whenever possible (since that's quite a bit of code).
170 * Ex:
171 * if (use_border) {
172 * texel = border_color;
173 * }
174 * else {
175 * texel = sample_texture(coord);
176 * }
177 * As it is now, we always sample the texture, then selectively replace
178 * the texel color results with the border color.
179 */
180
181 if (use_border) {
182 /* select texel color or border color depending on use_border */
183 LLVMValueRef border_color_ptr =
184 bld->dynamic_state->border_color(bld->dynamic_state,
185 bld->gallivm, unit);
186 int chan;
187 for (chan = 0; chan < 4; chan++) {
188 LLVMValueRef border_chan =
189 lp_build_array_get(bld->gallivm, border_color_ptr,
190 lp_build_const_int32(bld->gallivm, chan));
191 LLVMValueRef border_chan_vec =
192 lp_build_broadcast_scalar(&bld->float_vec_bld, border_chan);
193 texel_out[chan] = lp_build_select(&bld->texel_bld, use_border,
194 border_chan_vec, texel_out[chan]);
195 }
196 }
197 }
198
199
200 /**
201 * Helper to compute the mirror function for the PIPE_WRAP_MIRROR modes.
202 */
203 static LLVMValueRef
204 lp_build_coord_mirror(struct lp_build_sample_context *bld,
205 LLVMValueRef coord)
206 {
207 struct lp_build_context *coord_bld = &bld->coord_bld;
208 struct lp_build_context *int_coord_bld = &bld->int_coord_bld;
209 LLVMValueRef fract, flr, isOdd;
210
211 lp_build_ifloor_fract(coord_bld, coord, &flr, &fract);
212
213 /* isOdd = flr & 1 */
214 isOdd = LLVMBuildAnd(bld->gallivm->builder, flr, int_coord_bld->one, "");
215
216 /* make coord positive or negative depending on isOdd */
217 coord = lp_build_set_sign(coord_bld, fract, isOdd);
218
219 /* convert isOdd to float */
220 isOdd = lp_build_int_to_float(coord_bld, isOdd);
221
222 /* add isOdd to coord */
223 coord = lp_build_add(coord_bld, coord, isOdd);
224
225 return coord;
226 }
227
228
229 /**
230 * Helper to compute the first coord and the weight for
231 * linear wrap repeat npot textures
232 */
233 void
234 lp_build_coord_repeat_npot_linear(struct lp_build_sample_context *bld,
235 LLVMValueRef coord_f,
236 LLVMValueRef length_i,
237 LLVMValueRef length_f,
238 LLVMValueRef *coord0_i,
239 LLVMValueRef *weight_f)
240 {
241 struct lp_build_context *coord_bld = &bld->coord_bld;
242 struct lp_build_context *int_coord_bld = &bld->int_coord_bld;
243 LLVMValueRef half = lp_build_const_vec(bld->gallivm, coord_bld->type, 0.5);
244 LLVMValueRef length_minus_one = lp_build_sub(int_coord_bld, length_i,
245 int_coord_bld->one);
246 LLVMValueRef mask;
247 /* wrap with normalized floats is just fract */
248 coord_f = lp_build_fract(coord_bld, coord_f);
249 /* mul by size and subtract 0.5 */
250 coord_f = lp_build_mul(coord_bld, coord_f, length_f);
251 coord_f = lp_build_sub(coord_bld, coord_f, half);
252 /*
253 * we avoided the 0.5/length division before the repeat wrap,
254 * now need to fix up edge cases with selects
255 */
256 /* convert to int, compute lerp weight */
257 lp_build_ifloor_fract(coord_bld, coord_f, coord0_i, weight_f);
258 mask = lp_build_compare(int_coord_bld->gallivm, int_coord_bld->type,
259 PIPE_FUNC_LESS, *coord0_i, int_coord_bld->zero);
260 *coord0_i = lp_build_select(int_coord_bld, mask, length_minus_one, *coord0_i);
261 }
262
263
264 /**
265 * Build LLVM code for texture wrap mode for linear filtering.
266 * \param x0_out returns first integer texcoord
267 * \param x1_out returns second integer texcoord
268 * \param weight_out returns linear interpolation weight
269 */
270 static void
271 lp_build_sample_wrap_linear(struct lp_build_sample_context *bld,
272 LLVMValueRef coord,
273 LLVMValueRef length,
274 LLVMValueRef length_f,
275 boolean is_pot,
276 unsigned wrap_mode,
277 LLVMValueRef *x0_out,
278 LLVMValueRef *x1_out,
279 LLVMValueRef *weight_out)
280 {
281 struct lp_build_context *coord_bld = &bld->coord_bld;
282 struct lp_build_context *int_coord_bld = &bld->int_coord_bld;
283 LLVMBuilderRef builder = bld->gallivm->builder;
284 LLVMValueRef half = lp_build_const_vec(bld->gallivm, coord_bld->type, 0.5);
285 LLVMValueRef length_minus_one = lp_build_sub(int_coord_bld, length, int_coord_bld->one);
286 LLVMValueRef coord0, coord1, weight;
287
288 switch(wrap_mode) {
289 case PIPE_TEX_WRAP_REPEAT:
290 if (is_pot) {
291 /* mul by size and subtract 0.5 */
292 coord = lp_build_mul(coord_bld, coord, length_f);
293 coord = lp_build_sub(coord_bld, coord, half);
294 /* convert to int, compute lerp weight */
295 lp_build_ifloor_fract(coord_bld, coord, &coord0, &weight);
296 coord1 = lp_build_add(int_coord_bld, coord0, int_coord_bld->one);
297 /* repeat wrap */
298 coord0 = LLVMBuildAnd(builder, coord0, length_minus_one, "");
299 coord1 = LLVMBuildAnd(builder, coord1, length_minus_one, "");
300 }
301 else {
302 LLVMValueRef mask;
303 lp_build_coord_repeat_npot_linear(bld, coord,
304 length, length_f,
305 &coord0, &weight);
306 mask = lp_build_compare(int_coord_bld->gallivm, int_coord_bld->type,
307 PIPE_FUNC_NOTEQUAL, coord0, length_minus_one);
308 coord1 = LLVMBuildAnd(builder,
309 lp_build_add(int_coord_bld, coord0, int_coord_bld->one),
310 mask, "");
311 }
312 break;
313
314 case PIPE_TEX_WRAP_CLAMP:
315 if (bld->static_state->normalized_coords) {
316 /* scale coord to length */
317 coord = lp_build_mul(coord_bld, coord, length_f);
318 }
319
320 /* clamp to [0, length] */
321 coord = lp_build_clamp(coord_bld, coord, coord_bld->zero, length_f);
322
323 coord = lp_build_sub(coord_bld, coord, half);
324
325 /* convert to int, compute lerp weight */
326 lp_build_ifloor_fract(coord_bld, coord, &coord0, &weight);
327 coord1 = lp_build_add(int_coord_bld, coord0, int_coord_bld->one);
328 break;
329
330 case PIPE_TEX_WRAP_CLAMP_TO_EDGE:
331 {
332 struct lp_build_context abs_coord_bld = bld->coord_bld;
333 abs_coord_bld.type.sign = FALSE;
334
335 if (bld->static_state->normalized_coords) {
336 /* mul by tex size */
337 coord = lp_build_mul(coord_bld, coord, length_f);
338 }
339 /* clamp to length max */
340 coord = lp_build_min(coord_bld, coord, length_f);
341 /* subtract 0.5 */
342 coord = lp_build_sub(coord_bld, coord, half);
343 /* clamp to [0, length - 0.5] */
344 coord = lp_build_max(coord_bld, coord, coord_bld->zero);
345 /* convert to int, compute lerp weight */
346 lp_build_ifloor_fract(&abs_coord_bld, coord, &coord0, &weight);
347 coord1 = lp_build_add(int_coord_bld, coord0, int_coord_bld->one);
348 /* coord1 = min(coord1, length-1) */
349 coord1 = lp_build_min(int_coord_bld, coord1, length_minus_one);
350 break;
351 }
352
353 case PIPE_TEX_WRAP_CLAMP_TO_BORDER:
354 {
355 LLVMValueRef min;
356 if (bld->static_state->normalized_coords) {
357 /* scale coord to length */
358 coord = lp_build_mul(coord_bld, coord, length_f);
359 }
360 /* was: clamp to [-0.5, length + 0.5], then sub 0.5 */
361 coord = lp_build_sub(coord_bld, coord, half);
362 min = lp_build_const_vec(bld->gallivm, coord_bld->type, -1.0F);
363 coord = lp_build_clamp(coord_bld, coord, min, length_f);
364 /* convert to int, compute lerp weight */
365 lp_build_ifloor_fract(coord_bld, coord, &coord0, &weight);
366 coord1 = lp_build_add(int_coord_bld, coord0, int_coord_bld->one);
367 }
368 break;
369
370 case PIPE_TEX_WRAP_MIRROR_REPEAT:
371 /* compute mirror function */
372 coord = lp_build_coord_mirror(bld, coord);
373
374 /* scale coord to length */
375 coord = lp_build_mul(coord_bld, coord, length_f);
376 coord = lp_build_sub(coord_bld, coord, half);
377
378 /* convert to int, compute lerp weight */
379 lp_build_ifloor_fract(coord_bld, coord, &coord0, &weight);
380 coord1 = lp_build_add(int_coord_bld, coord0, int_coord_bld->one);
381
382 /* coord0 = max(coord0, 0) */
383 coord0 = lp_build_max(int_coord_bld, coord0, int_coord_bld->zero);
384 /* coord1 = min(coord1, length-1) */
385 coord1 = lp_build_min(int_coord_bld, coord1, length_minus_one);
386 break;
387
388 case PIPE_TEX_WRAP_MIRROR_CLAMP:
389 coord = lp_build_abs(coord_bld, coord);
390
391 if (bld->static_state->normalized_coords) {
392 /* scale coord to length */
393 coord = lp_build_mul(coord_bld, coord, length_f);
394 }
395
396 /* clamp to [0, length] */
397 coord = lp_build_min(coord_bld, coord, length_f);
398
399 coord = lp_build_sub(coord_bld, coord, half);
400
401 /* convert to int, compute lerp weight */
402 lp_build_ifloor_fract(coord_bld, coord, &coord0, &weight);
403 coord1 = lp_build_add(int_coord_bld, coord0, int_coord_bld->one);
404 break;
405
406 case PIPE_TEX_WRAP_MIRROR_CLAMP_TO_EDGE:
407 {
408 LLVMValueRef min, max;
409 struct lp_build_context abs_coord_bld = bld->coord_bld;
410 abs_coord_bld.type.sign = FALSE;
411 coord = lp_build_abs(coord_bld, coord);
412
413 if (bld->static_state->normalized_coords) {
414 /* scale coord to length */
415 coord = lp_build_mul(coord_bld, coord, length_f);
416 }
417
418 /* clamp to [0.5, length - 0.5] */
419 min = half;
420 max = lp_build_sub(coord_bld, length_f, min);
421 coord = lp_build_clamp(coord_bld, coord, min, max);
422
423 coord = lp_build_sub(coord_bld, coord, half);
424
425 /* convert to int, compute lerp weight */
426 lp_build_ifloor_fract(&abs_coord_bld, coord, &coord0, &weight);
427 coord1 = lp_build_add(int_coord_bld, coord0, int_coord_bld->one);
428 }
429 break;
430
431 case PIPE_TEX_WRAP_MIRROR_CLAMP_TO_BORDER:
432 {
433 coord = lp_build_abs(coord_bld, coord);
434
435 if (bld->static_state->normalized_coords) {
436 /* scale coord to length */
437 coord = lp_build_mul(coord_bld, coord, length_f);
438 }
439
440 /* was: clamp to [-0.5, length + 0.5] then sub 0.5 */
441 /* skip -0.5 clamp (always positive), do sub first */
442 coord = lp_build_sub(coord_bld, coord, half);
443 coord = lp_build_min(coord_bld, coord, length_f);
444
445 /* convert to int, compute lerp weight */
446 lp_build_ifloor_fract(coord_bld, coord, &coord0, &weight);
447 coord1 = lp_build_add(int_coord_bld, coord0, int_coord_bld->one);
448 }
449 break;
450
451 default:
452 assert(0);
453 coord0 = NULL;
454 coord1 = NULL;
455 weight = NULL;
456 }
457
458 *x0_out = coord0;
459 *x1_out = coord1;
460 *weight_out = weight;
461 }
462
463
464 /**
465 * Build LLVM code for texture wrap mode for nearest filtering.
466 * \param coord the incoming texcoord (nominally in [0,1])
467 * \param length the texture size along one dimension, as int vector
468 * \param is_pot if TRUE, length is a power of two
469 * \param wrap_mode one of PIPE_TEX_WRAP_x
470 */
471 static LLVMValueRef
472 lp_build_sample_wrap_nearest(struct lp_build_sample_context *bld,
473 LLVMValueRef coord,
474 LLVMValueRef length,
475 LLVMValueRef length_f,
476 boolean is_pot,
477 unsigned wrap_mode)
478 {
479 struct lp_build_context *coord_bld = &bld->coord_bld;
480 struct lp_build_context *int_coord_bld = &bld->int_coord_bld;
481 LLVMBuilderRef builder = bld->gallivm->builder;
482 LLVMValueRef length_minus_one = lp_build_sub(int_coord_bld, length, int_coord_bld->one);
483 LLVMValueRef icoord;
484
485 switch(wrap_mode) {
486 case PIPE_TEX_WRAP_REPEAT:
487 if (is_pot) {
488 coord = lp_build_mul(coord_bld, coord, length_f);
489 icoord = lp_build_ifloor(coord_bld, coord);
490 icoord = LLVMBuildAnd(builder, icoord, length_minus_one, "");
491 }
492 else {
493 /* take fraction, unnormalize */
494 coord = lp_build_fract_safe(coord_bld, coord);
495 coord = lp_build_mul(coord_bld, coord, length_f);
496 icoord = lp_build_itrunc(coord_bld, coord);
497 }
498 break;
499
500 case PIPE_TEX_WRAP_CLAMP:
501 case PIPE_TEX_WRAP_CLAMP_TO_EDGE:
502 if (bld->static_state->normalized_coords) {
503 /* scale coord to length */
504 coord = lp_build_mul(coord_bld, coord, length_f);
505 }
506
507 /* floor */
508 /* use itrunc instead since we clamp to 0 anyway */
509 icoord = lp_build_itrunc(coord_bld, coord);
510
511 /* clamp to [0, length - 1]. */
512 icoord = lp_build_clamp(int_coord_bld, icoord, int_coord_bld->zero,
513 length_minus_one);
514 break;
515
516 case PIPE_TEX_WRAP_CLAMP_TO_BORDER:
517 /* Note: this is the same as CLAMP_TO_EDGE, except min = -1 */
518 {
519 LLVMValueRef min, max;
520
521 if (bld->static_state->normalized_coords) {
522 /* scale coord to length */
523 coord = lp_build_mul(coord_bld, coord, length_f);
524 }
525
526 icoord = lp_build_ifloor(coord_bld, coord);
527
528 /* clamp to [-1, length] */
529 min = lp_build_negate(int_coord_bld, int_coord_bld->one);
530 max = length;
531 icoord = lp_build_clamp(int_coord_bld, icoord, min, max);
532 }
533 break;
534
535 case PIPE_TEX_WRAP_MIRROR_REPEAT:
536 /* compute mirror function */
537 coord = lp_build_coord_mirror(bld, coord);
538
539 /* scale coord to length */
540 assert(bld->static_state->normalized_coords);
541 coord = lp_build_mul(coord_bld, coord, length_f);
542
543 /* itrunc == ifloor here */
544 icoord = lp_build_itrunc(coord_bld, coord);
545
546 /* clamp to [0, length - 1] */
547 icoord = lp_build_min(int_coord_bld, icoord, length_minus_one);
548 break;
549
550 case PIPE_TEX_WRAP_MIRROR_CLAMP:
551 case PIPE_TEX_WRAP_MIRROR_CLAMP_TO_EDGE:
552 coord = lp_build_abs(coord_bld, coord);
553
554 if (bld->static_state->normalized_coords) {
555 /* scale coord to length */
556 coord = lp_build_mul(coord_bld, coord, length_f);
557 }
558
559 /* itrunc == ifloor here */
560 icoord = lp_build_itrunc(coord_bld, coord);
561
562 /* clamp to [0, length - 1] */
563 icoord = lp_build_min(int_coord_bld, icoord, length_minus_one);
564 break;
565
566 case PIPE_TEX_WRAP_MIRROR_CLAMP_TO_BORDER:
567 coord = lp_build_abs(coord_bld, coord);
568
569 if (bld->static_state->normalized_coords) {
570 /* scale coord to length */
571 coord = lp_build_mul(coord_bld, coord, length_f);
572 }
573
574 /* itrunc == ifloor here */
575 icoord = lp_build_itrunc(coord_bld, coord);
576
577 /* clamp to [0, length] */
578 icoord = lp_build_min(int_coord_bld, icoord, length);
579 break;
580
581 default:
582 assert(0);
583 icoord = NULL;
584 }
585
586 return icoord;
587 }
588
589
590 /**
591 * Generate code to sample a mipmap level with nearest filtering.
592 * If sampling a cube texture, r = cube face in [0,5].
593 */
594 static void
595 lp_build_sample_image_nearest(struct lp_build_sample_context *bld,
596 unsigned unit,
597 LLVMValueRef size,
598 LLVMValueRef row_stride_vec,
599 LLVMValueRef img_stride_vec,
600 LLVMValueRef data_ptr,
601 LLVMValueRef mipoffsets,
602 LLVMValueRef s,
603 LLVMValueRef t,
604 LLVMValueRef r,
605 LLVMValueRef colors_out[4])
606 {
607 const unsigned dims = bld->dims;
608 LLVMValueRef width_vec;
609 LLVMValueRef height_vec;
610 LLVMValueRef depth_vec;
611 LLVMValueRef flt_size;
612 LLVMValueRef flt_width_vec;
613 LLVMValueRef flt_height_vec;
614 LLVMValueRef flt_depth_vec;
615 LLVMValueRef x, y, z;
616
617 lp_build_extract_image_sizes(bld,
618 &bld->int_size_bld,
619 bld->int_coord_type,
620 size,
621 &width_vec, &height_vec, &depth_vec);
622
623 flt_size = lp_build_int_to_float(&bld->float_size_bld, size);
624
625 lp_build_extract_image_sizes(bld,
626 &bld->float_size_bld,
627 bld->coord_type,
628 flt_size,
629 &flt_width_vec, &flt_height_vec, &flt_depth_vec);
630
631 /*
632 * Compute integer texcoords.
633 */
634 x = lp_build_sample_wrap_nearest(bld, s, width_vec, flt_width_vec,
635 bld->static_state->pot_width,
636 bld->static_state->wrap_s);
637 lp_build_name(x, "tex.x.wrapped");
638
639 if (dims >= 2) {
640 y = lp_build_sample_wrap_nearest(bld, t, height_vec, flt_height_vec,
641 bld->static_state->pot_height,
642 bld->static_state->wrap_t);
643 lp_build_name(y, "tex.y.wrapped");
644
645 if (dims == 3) {
646 z = lp_build_sample_wrap_nearest(bld, r, depth_vec, flt_depth_vec,
647 bld->static_state->pot_depth,
648 bld->static_state->wrap_r);
649 lp_build_name(z, "tex.z.wrapped");
650 }
651 else if (bld->static_state->target == PIPE_TEXTURE_CUBE) {
652 z = r;
653 }
654 else {
655 z = NULL;
656 }
657 }
658 else {
659 y = z = NULL;
660 }
661
662 /*
663 * Get texture colors.
664 */
665 lp_build_sample_texel_soa(bld, unit,
666 width_vec, height_vec, depth_vec,
667 x, y, z,
668 row_stride_vec, img_stride_vec,
669 data_ptr, mipoffsets, colors_out);
670 }
671
672
673 /**
674 * Generate code to sample a mipmap level with linear filtering.
675 * If sampling a cube texture, r = cube face in [0,5].
676 */
677 static void
678 lp_build_sample_image_linear(struct lp_build_sample_context *bld,
679 unsigned unit,
680 LLVMValueRef size,
681 LLVMValueRef row_stride_vec,
682 LLVMValueRef img_stride_vec,
683 LLVMValueRef data_ptr,
684 LLVMValueRef mipoffsets,
685 LLVMValueRef s,
686 LLVMValueRef t,
687 LLVMValueRef r,
688 LLVMValueRef colors_out[4])
689 {
690 const unsigned dims = bld->dims;
691 LLVMValueRef width_vec;
692 LLVMValueRef height_vec;
693 LLVMValueRef depth_vec;
694 LLVMValueRef flt_size;
695 LLVMValueRef flt_width_vec;
696 LLVMValueRef flt_height_vec;
697 LLVMValueRef flt_depth_vec;
698 LLVMValueRef x0, y0, z0, x1, y1, z1;
699 LLVMValueRef s_fpart, t_fpart, r_fpart;
700 LLVMValueRef neighbors[2][2][4];
701 int chan;
702
703 lp_build_extract_image_sizes(bld,
704 &bld->int_size_bld,
705 bld->int_coord_type,
706 size,
707 &width_vec, &height_vec, &depth_vec);
708
709 flt_size = lp_build_int_to_float(&bld->float_size_bld, size);
710
711 lp_build_extract_image_sizes(bld,
712 &bld->float_size_bld,
713 bld->coord_type,
714 flt_size,
715 &flt_width_vec, &flt_height_vec, &flt_depth_vec);
716
717 /*
718 * Compute integer texcoords.
719 */
720 lp_build_sample_wrap_linear(bld, s, width_vec, flt_width_vec,
721 bld->static_state->pot_width,
722 bld->static_state->wrap_s,
723 &x0, &x1, &s_fpart);
724 lp_build_name(x0, "tex.x0.wrapped");
725 lp_build_name(x1, "tex.x1.wrapped");
726
727 if (dims >= 2) {
728 lp_build_sample_wrap_linear(bld, t, height_vec, flt_height_vec,
729 bld->static_state->pot_height,
730 bld->static_state->wrap_t,
731 &y0, &y1, &t_fpart);
732 lp_build_name(y0, "tex.y0.wrapped");
733 lp_build_name(y1, "tex.y1.wrapped");
734
735 if (dims == 3) {
736 lp_build_sample_wrap_linear(bld, r, depth_vec, flt_depth_vec,
737 bld->static_state->pot_depth,
738 bld->static_state->wrap_r,
739 &z0, &z1, &r_fpart);
740 lp_build_name(z0, "tex.z0.wrapped");
741 lp_build_name(z1, "tex.z1.wrapped");
742 }
743 else if (bld->static_state->target == PIPE_TEXTURE_CUBE) {
744 z0 = z1 = r; /* cube face */
745 r_fpart = NULL;
746 }
747 else {
748 z0 = z1 = NULL;
749 r_fpart = NULL;
750 }
751 }
752 else {
753 y0 = y1 = t_fpart = NULL;
754 z0 = z1 = r_fpart = NULL;
755 }
756
757 /*
758 * Get texture colors.
759 */
760 /* get x0/x1 texels */
761 lp_build_sample_texel_soa(bld, unit,
762 width_vec, height_vec, depth_vec,
763 x0, y0, z0,
764 row_stride_vec, img_stride_vec,
765 data_ptr, mipoffsets, neighbors[0][0]);
766 lp_build_sample_texel_soa(bld, unit,
767 width_vec, height_vec, depth_vec,
768 x1, y0, z0,
769 row_stride_vec, img_stride_vec,
770 data_ptr, mipoffsets, neighbors[0][1]);
771
772 if (dims == 1) {
773 /* Interpolate two samples from 1D image to produce one color */
774 for (chan = 0; chan < 4; chan++) {
775 colors_out[chan] = lp_build_lerp(&bld->texel_bld, s_fpart,
776 neighbors[0][0][chan],
777 neighbors[0][1][chan]);
778 }
779 }
780 else {
781 /* 2D/3D texture */
782 LLVMValueRef colors0[4];
783
784 /* get x0/x1 texels at y1 */
785 lp_build_sample_texel_soa(bld, unit,
786 width_vec, height_vec, depth_vec,
787 x0, y1, z0,
788 row_stride_vec, img_stride_vec,
789 data_ptr, mipoffsets, neighbors[1][0]);
790 lp_build_sample_texel_soa(bld, unit,
791 width_vec, height_vec, depth_vec,
792 x1, y1, z0,
793 row_stride_vec, img_stride_vec,
794 data_ptr, mipoffsets, neighbors[1][1]);
795
796 /* Bilinear interpolate the four samples from the 2D image / 3D slice */
797 for (chan = 0; chan < 4; chan++) {
798 colors0[chan] = lp_build_lerp_2d(&bld->texel_bld,
799 s_fpart, t_fpart,
800 neighbors[0][0][chan],
801 neighbors[0][1][chan],
802 neighbors[1][0][chan],
803 neighbors[1][1][chan]);
804 }
805
806 if (dims == 3) {
807 LLVMValueRef neighbors1[2][2][4];
808 LLVMValueRef colors1[4];
809
810 /* get x0/x1/y0/y1 texels at z1 */
811 lp_build_sample_texel_soa(bld, unit,
812 width_vec, height_vec, depth_vec,
813 x0, y0, z1,
814 row_stride_vec, img_stride_vec,
815 data_ptr, mipoffsets, neighbors1[0][0]);
816 lp_build_sample_texel_soa(bld, unit,
817 width_vec, height_vec, depth_vec,
818 x1, y0, z1,
819 row_stride_vec, img_stride_vec,
820 data_ptr, mipoffsets, neighbors1[0][1]);
821 lp_build_sample_texel_soa(bld, unit,
822 width_vec, height_vec, depth_vec,
823 x0, y1, z1,
824 row_stride_vec, img_stride_vec,
825 data_ptr, mipoffsets, neighbors1[1][0]);
826 lp_build_sample_texel_soa(bld, unit,
827 width_vec, height_vec, depth_vec,
828 x1, y1, z1,
829 row_stride_vec, img_stride_vec,
830 data_ptr, mipoffsets, neighbors1[1][1]);
831
832 /* Bilinear interpolate the four samples from the second Z slice */
833 for (chan = 0; chan < 4; chan++) {
834 colors1[chan] = lp_build_lerp_2d(&bld->texel_bld,
835 s_fpart, t_fpart,
836 neighbors1[0][0][chan],
837 neighbors1[0][1][chan],
838 neighbors1[1][0][chan],
839 neighbors1[1][1][chan]);
840 }
841
842 /* Linearly interpolate the two samples from the two 3D slices */
843 for (chan = 0; chan < 4; chan++) {
844 colors_out[chan] = lp_build_lerp(&bld->texel_bld,
845 r_fpart,
846 colors0[chan], colors1[chan]);
847 }
848 }
849 else {
850 /* 2D tex */
851 for (chan = 0; chan < 4; chan++) {
852 colors_out[chan] = colors0[chan];
853 }
854 }
855 }
856 }
857
858
859 /**
860 * Sample the texture/mipmap using given image filter and mip filter.
861 * data0_ptr and data1_ptr point to the two mipmap levels to sample
862 * from. width0/1_vec, height0/1_vec, depth0/1_vec indicate their sizes.
863 * If we're using nearest miplevel sampling the '1' values will be null/unused.
864 */
865 static void
866 lp_build_sample_mipmap(struct lp_build_sample_context *bld,
867 unsigned unit,
868 unsigned img_filter,
869 unsigned mip_filter,
870 LLVMValueRef s,
871 LLVMValueRef t,
872 LLVMValueRef r,
873 LLVMValueRef ilevel0,
874 LLVMValueRef ilevel1,
875 LLVMValueRef lod_fpart,
876 LLVMValueRef *colors_out)
877 {
878 LLVMBuilderRef builder = bld->gallivm->builder;
879 LLVMValueRef size0 = NULL;
880 LLVMValueRef size1 = NULL;
881 LLVMValueRef row_stride0_vec = NULL;
882 LLVMValueRef row_stride1_vec = NULL;
883 LLVMValueRef img_stride0_vec = NULL;
884 LLVMValueRef img_stride1_vec = NULL;
885 LLVMValueRef data_ptr0 = NULL;
886 LLVMValueRef data_ptr1 = NULL;
887 LLVMValueRef mipoff0 = NULL;
888 LLVMValueRef mipoff1 = NULL;
889 LLVMValueRef colors0[4], colors1[4];
890 unsigned chan;
891
892 /* sample the first mipmap level */
893 lp_build_mipmap_level_sizes(bld, ilevel0,
894 &size0,
895 &row_stride0_vec, &img_stride0_vec);
896 if (bld->num_lods == 1) {
897 data_ptr0 = lp_build_get_mipmap_level(bld, ilevel0);
898 }
899 else {
900 /* This path should work for num_lods 1 too but slightly less efficient */
901 data_ptr0 = bld->base_ptr;
902 mipoff0 = lp_build_get_mip_offsets(bld, ilevel0);
903 }
904 if (img_filter == PIPE_TEX_FILTER_NEAREST) {
905 lp_build_sample_image_nearest(bld, unit,
906 size0,
907 row_stride0_vec, img_stride0_vec,
908 data_ptr0, mipoff0, s, t, r,
909 colors0);
910 }
911 else {
912 assert(img_filter == PIPE_TEX_FILTER_LINEAR);
913 lp_build_sample_image_linear(bld, unit,
914 size0,
915 row_stride0_vec, img_stride0_vec,
916 data_ptr0, mipoff0, s, t, r,
917 colors0);
918 }
919
920 /* Store the first level's colors in the output variables */
921 for (chan = 0; chan < 4; chan++) {
922 LLVMBuildStore(builder, colors0[chan], colors_out[chan]);
923 }
924
925 if (mip_filter == PIPE_TEX_MIPFILTER_LINEAR) {
926 struct lp_build_if_state if_ctx;
927 LLVMValueRef need_lerp;
928 unsigned num_quads = bld->coord_bld.type.length / 4;
929
930 /* need_lerp = lod_fpart > 0 */
931 if (num_quads == 1) {
932 need_lerp = LLVMBuildFCmp(builder, LLVMRealUGT,
933 lod_fpart, bld->perquadf_bld.zero,
934 "need_lerp");
935 }
936 else {
937 /*
938 * We'll do mip filtering if any of the quads need it.
939 * It might be better to split the vectors here and only fetch/filter
940 * quads which need it.
941 */
942 /*
943 * We unfortunately need to clamp lod_fpart here since we can get
944 * negative values which would screw up filtering if not all
945 * lod_fpart values have same sign.
946 */
947 lod_fpart = lp_build_max(&bld->perquadf_bld, lod_fpart,
948 bld->perquadf_bld.zero);
949 need_lerp = lp_build_compare(bld->gallivm, bld->perquadf_bld.type,
950 PIPE_FUNC_GREATER,
951 lod_fpart, bld->perquadf_bld.zero);
952 need_lerp = lp_build_any_true_range(&bld->perquadi_bld, num_quads, need_lerp);
953 }
954
955 lp_build_if(&if_ctx, bld->gallivm, need_lerp);
956 {
957 /* sample the second mipmap level */
958 lp_build_mipmap_level_sizes(bld, ilevel1,
959 &size1,
960 &row_stride1_vec, &img_stride1_vec);
961 if (bld->num_lods == 1) {
962 data_ptr1 = lp_build_get_mipmap_level(bld, ilevel1);
963 }
964 else {
965 data_ptr1 = bld->base_ptr;
966 mipoff1 = lp_build_get_mip_offsets(bld, ilevel1);
967 }
968 if (img_filter == PIPE_TEX_FILTER_NEAREST) {
969 lp_build_sample_image_nearest(bld, unit,
970 size1,
971 row_stride1_vec, img_stride1_vec,
972 data_ptr1, mipoff1, s, t, r,
973 colors1);
974 }
975 else {
976 lp_build_sample_image_linear(bld, unit,
977 size1,
978 row_stride1_vec, img_stride1_vec,
979 data_ptr1, mipoff1, s, t, r,
980 colors1);
981 }
982
983 /* interpolate samples from the two mipmap levels */
984
985 lod_fpart = lp_build_unpack_broadcast_aos_scalars(bld->gallivm,
986 bld->perquadf_bld.type,
987 bld->texel_bld.type,
988 lod_fpart);
989
990 for (chan = 0; chan < 4; chan++) {
991 colors0[chan] = lp_build_lerp(&bld->texel_bld, lod_fpart,
992 colors0[chan], colors1[chan]);
993 LLVMBuildStore(builder, colors0[chan], colors_out[chan]);
994 }
995 }
996 lp_build_endif(&if_ctx);
997 }
998 }
999
1000 /**
1001 * Calculate cube face, lod, mip levels.
1002 */
1003 static void
1004 lp_build_sample_common(struct lp_build_sample_context *bld,
1005 unsigned unit,
1006 LLVMValueRef *s,
1007 LLVMValueRef *t,
1008 LLVMValueRef *r,
1009 const struct lp_derivatives *derivs,
1010 LLVMValueRef lod_bias, /* optional */
1011 LLVMValueRef explicit_lod, /* optional */
1012 LLVMValueRef *lod_ipart,
1013 LLVMValueRef *lod_fpart,
1014 LLVMValueRef *ilevel0,
1015 LLVMValueRef *ilevel1)
1016 {
1017 const unsigned mip_filter = bld->static_state->min_mip_filter;
1018 const unsigned min_filter = bld->static_state->min_img_filter;
1019 const unsigned mag_filter = bld->static_state->mag_img_filter;
1020 LLVMValueRef first_level;
1021 struct lp_derivatives face_derivs;
1022
1023 /*
1024 printf("%s mip %d min %d mag %d\n", __FUNCTION__,
1025 mip_filter, min_filter, mag_filter);
1026 */
1027
1028 /*
1029 * Choose cube face, recompute texcoords and derivatives for the chosen face.
1030 */
1031 if (bld->static_state->target == PIPE_TEXTURE_CUBE) {
1032 LLVMValueRef face, face_s, face_t;
1033 lp_build_cube_lookup(bld, *s, *t, *r, &face, &face_s, &face_t);
1034 *s = face_s; /* vec */
1035 *t = face_t; /* vec */
1036 /* use 'r' to indicate cube face */
1037 *r = face; /* vec */
1038
1039 /* recompute ddx, ddy using the new (s,t) face texcoords */
1040 face_derivs.ddx_ddy[0] = lp_build_packed_ddx_ddy_twocoord(&bld->coord_bld, *s, *t);
1041 face_derivs.ddx_ddy[1] = NULL;
1042 derivs = &face_derivs;
1043 }
1044
1045 /*
1046 * Compute the level of detail (float).
1047 */
1048 if (min_filter != mag_filter ||
1049 mip_filter != PIPE_TEX_MIPFILTER_NONE) {
1050 /* Need to compute lod either to choose mipmap levels or to
1051 * distinguish between minification/magnification with one mipmap level.
1052 */
1053 lp_build_lod_selector(bld, unit, derivs,
1054 lod_bias, explicit_lod,
1055 mip_filter,
1056 lod_ipart, lod_fpart);
1057 } else {
1058 *lod_ipart = bld->perquadi_bld.zero;
1059 }
1060
1061 /*
1062 * Compute integer mipmap level(s) to fetch texels from: ilevel0, ilevel1
1063 */
1064 switch (mip_filter) {
1065 default:
1066 assert(0 && "bad mip_filter value in lp_build_sample_soa()");
1067 /* fall-through */
1068 case PIPE_TEX_MIPFILTER_NONE:
1069 /* always use mip level 0 */
1070 if (bld->static_state->target == PIPE_TEXTURE_CUBE) {
1071 /* XXX this is a work-around for an apparent bug in LLVM 2.7.
1072 * We should be able to set ilevel0 = const(0) but that causes
1073 * bad x86 code to be emitted.
1074 * XXX should probably disable that on other llvm versions.
1075 */
1076 assert(*lod_ipart);
1077 lp_build_nearest_mip_level(bld, unit, *lod_ipart, ilevel0);
1078 }
1079 else {
1080 first_level = bld->dynamic_state->first_level(bld->dynamic_state,
1081 bld->gallivm, unit);
1082 first_level = lp_build_broadcast_scalar(&bld->perquadi_bld, first_level);
1083 *ilevel0 = first_level;
1084 }
1085 break;
1086 case PIPE_TEX_MIPFILTER_NEAREST:
1087 assert(*lod_ipart);
1088 lp_build_nearest_mip_level(bld, unit, *lod_ipart, ilevel0);
1089 break;
1090 case PIPE_TEX_MIPFILTER_LINEAR:
1091 assert(*lod_ipart);
1092 assert(*lod_fpart);
1093 lp_build_linear_mip_levels(bld, unit,
1094 *lod_ipart, lod_fpart,
1095 ilevel0, ilevel1);
1096 break;
1097 }
1098 }
1099
1100 /**
1101 * General texture sampling codegen.
1102 * This function handles texture sampling for all texture targets (1D,
1103 * 2D, 3D, cube) and all filtering modes.
1104 */
1105 static void
1106 lp_build_sample_general(struct lp_build_sample_context *bld,
1107 unsigned unit,
1108 LLVMValueRef s,
1109 LLVMValueRef t,
1110 LLVMValueRef r,
1111 LLVMValueRef lod_ipart,
1112 LLVMValueRef lod_fpart,
1113 LLVMValueRef ilevel0,
1114 LLVMValueRef ilevel1,
1115 LLVMValueRef *colors_out)
1116 {
1117 struct lp_build_context *int_bld = &bld->int_bld;
1118 LLVMBuilderRef builder = bld->gallivm->builder;
1119 const unsigned mip_filter = bld->static_state->min_mip_filter;
1120 const unsigned min_filter = bld->static_state->min_img_filter;
1121 const unsigned mag_filter = bld->static_state->mag_img_filter;
1122 LLVMValueRef texels[4];
1123 unsigned chan;
1124
1125 /*
1126 * Get/interpolate texture colors.
1127 */
1128
1129 for (chan = 0; chan < 4; ++chan) {
1130 texels[chan] = lp_build_alloca(bld->gallivm, bld->texel_bld.vec_type, "");
1131 lp_build_name(texels[chan], "sampler%u_texel_%c_var", unit, "xyzw"[chan]);
1132 }
1133
1134 if (min_filter == mag_filter) {
1135 /* no need to distinguish between minification and magnification */
1136 lp_build_sample_mipmap(bld, unit,
1137 min_filter, mip_filter,
1138 s, t, r,
1139 ilevel0, ilevel1, lod_fpart,
1140 texels);
1141 }
1142 else {
1143 /* Emit conditional to choose min image filter or mag image filter
1144 * depending on the lod being > 0 or <= 0, respectively.
1145 */
1146 struct lp_build_if_state if_ctx;
1147 LLVMValueRef minify;
1148
1149 /*
1150 * XXX this should to all lods into account, if some are min
1151 * some max probably could hack up the coords/weights in the linear
1152 * path with selects to work for nearest.
1153 * If that's just two quads sitting next to each other it seems
1154 * quite ok to do the same filtering method on both though, at
1155 * least unless we have explicit lod (and who uses different
1156 * min/mag filter with that?)
1157 */
1158 if (bld->num_lods > 1)
1159 lod_ipart = LLVMBuildExtractElement(builder, lod_ipart,
1160 lp_build_const_int32(bld->gallivm, 0), "");
1161
1162 /* minify = lod >= 0.0 */
1163 minify = LLVMBuildICmp(builder, LLVMIntSGE,
1164 lod_ipart, int_bld->zero, "");
1165
1166 lp_build_if(&if_ctx, bld->gallivm, minify);
1167 {
1168 /* Use the minification filter */
1169 lp_build_sample_mipmap(bld, unit,
1170 min_filter, mip_filter,
1171 s, t, r,
1172 ilevel0, ilevel1, lod_fpart,
1173 texels);
1174 }
1175 lp_build_else(&if_ctx);
1176 {
1177 /* Use the magnification filter */
1178 lp_build_sample_mipmap(bld, unit,
1179 mag_filter, PIPE_TEX_MIPFILTER_NONE,
1180 s, t, r,
1181 ilevel0, NULL, NULL,
1182 texels);
1183 }
1184 lp_build_endif(&if_ctx);
1185 }
1186
1187 for (chan = 0; chan < 4; ++chan) {
1188 colors_out[chan] = LLVMBuildLoad(builder, texels[chan], "");
1189 lp_build_name(colors_out[chan], "sampler%u_texel_%c", unit, "xyzw"[chan]);
1190 }
1191 }
1192
1193
1194 /**
1195 * Texel fetch function.
1196 * In contrast to general sampling there is no filtering, no coord minification,
1197 * lod (if any) is always explicit uint, coords are uints (in terms of texel units)
1198 * directly to be applied to the selected mip level (after adding texel offsets).
1199 * This function handles texel fetch for all targets where texel fetch is supported
1200 * (no cube maps, but 1d, 2d, 3d are supported, arrays and buffers should be too).
1201 */
1202 static void
1203 lp_build_fetch_texel(struct lp_build_sample_context *bld,
1204 unsigned unit,
1205 const LLVMValueRef *coords,
1206 LLVMValueRef explicit_lod,
1207 const LLVMValueRef *offsets,
1208 LLVMValueRef *colors_out)
1209 {
1210 struct lp_build_context *perquadi_bld = &bld->perquadi_bld;
1211 struct lp_build_context *int_coord_bld = &bld->int_coord_bld;
1212 unsigned dims = bld->dims, chan;
1213 LLVMValueRef size, ilevel;
1214 LLVMValueRef row_stride_vec = NULL, img_stride_vec = NULL;
1215 LLVMValueRef x = coords[0], y = coords[1], z = coords[2];
1216 LLVMValueRef width, height, depth, i, j;
1217 LLVMValueRef offset, out_of_bounds, out1;
1218
1219 /* XXX just like ordinary sampling, we don't handle per-pixel lod (yet). */
1220 if (explicit_lod && bld->static_state->target != PIPE_BUFFER) {
1221 ilevel = lp_build_pack_aos_scalars(bld->gallivm, int_coord_bld->type,
1222 perquadi_bld->type, explicit_lod, 0);
1223 lp_build_nearest_mip_level(bld, unit, ilevel, &ilevel);
1224 }
1225 else {
1226 bld->num_lods = 1;
1227 ilevel = lp_build_const_int32(bld->gallivm, 0);
1228 }
1229 lp_build_mipmap_level_sizes(bld, ilevel,
1230 &size,
1231 &row_stride_vec, &img_stride_vec);
1232 lp_build_extract_image_sizes(bld, &bld->int_size_bld, int_coord_bld->type,
1233 size, &width, &height, &depth);
1234
1235 /* This is a lot like border sampling */
1236 if (offsets[0]) {
1237 /* XXX coords are really unsigned, offsets are signed */
1238 x = lp_build_add(int_coord_bld, x, offsets[0]);
1239 }
1240 out_of_bounds = lp_build_cmp(int_coord_bld, PIPE_FUNC_LESS, x, int_coord_bld->zero);
1241 out1 = lp_build_cmp(int_coord_bld, PIPE_FUNC_GEQUAL, x, width);
1242 out_of_bounds = lp_build_or(int_coord_bld, out_of_bounds, out1);
1243
1244 if (dims >= 2) {
1245 if (offsets[1]) {
1246 y = lp_build_add(int_coord_bld, y, offsets[1]);
1247 }
1248 out1 = lp_build_cmp(int_coord_bld, PIPE_FUNC_LESS, y, int_coord_bld->zero);
1249 out_of_bounds = lp_build_or(int_coord_bld, out_of_bounds, out1);
1250 out1 = lp_build_cmp(int_coord_bld, PIPE_FUNC_GEQUAL, y, height);
1251 out_of_bounds = lp_build_or(int_coord_bld, out_of_bounds, out1);
1252
1253 if (dims >= 3) {
1254 if (offsets[2]) {
1255 z = lp_build_add(int_coord_bld, z, offsets[2]);
1256 }
1257 out1 = lp_build_cmp(int_coord_bld, PIPE_FUNC_LESS, z, int_coord_bld->zero);
1258 out_of_bounds = lp_build_or(int_coord_bld, out_of_bounds, out1);
1259 out1 = lp_build_cmp(int_coord_bld, PIPE_FUNC_GEQUAL, z, depth);
1260 out_of_bounds = lp_build_or(int_coord_bld, out_of_bounds, out1);
1261 }
1262 }
1263
1264 lp_build_sample_offset(int_coord_bld,
1265 bld->format_desc,
1266 x, y, z, row_stride_vec, img_stride_vec,
1267 &offset, &i, &j);
1268
1269 if (bld->static_state->target != PIPE_BUFFER) {
1270 offset = lp_build_add(int_coord_bld, offset,
1271 lp_build_get_mip_offsets(bld, ilevel));
1272 }
1273
1274 offset = lp_build_andnot(int_coord_bld, offset, out_of_bounds);
1275
1276 lp_build_fetch_rgba_soa(bld->gallivm,
1277 bld->format_desc,
1278 bld->texel_type,
1279 bld->base_ptr, offset,
1280 i, j,
1281 colors_out);
1282
1283 if (0) {
1284 /*
1285 * Not needed except for ARB_robust_buffer_access_behavior.
1286 * Could use min/max above instead of out-of-bounds comparisons
1287 * (in fact cast to unsigned and min only is sufficient)
1288 * if we don't care about the result returned for out-of-bounds.
1289 */
1290 for (chan = 0; chan < 4; chan++) {
1291 colors_out[chan] = lp_build_select(&bld->texel_bld, out_of_bounds,
1292 bld->texel_bld.zero, colors_out[chan]);
1293 }
1294 }
1295 }
1296
1297
1298 /**
1299 * Do shadow test/comparison.
1300 * \param p the texcoord Z (aka R, aka P) component
1301 * \param texel the texel to compare against (use the X channel)
1302 * Ideally this should really be done per-sample.
1303 */
1304 static void
1305 lp_build_sample_compare(struct lp_build_sample_context *bld,
1306 LLVMValueRef p,
1307 LLVMValueRef texel[4])
1308 {
1309 struct lp_build_context *texel_bld = &bld->texel_bld;
1310 LLVMBuilderRef builder = bld->gallivm->builder;
1311 LLVMValueRef res;
1312 const unsigned chan = 0;
1313
1314 if (bld->static_state->compare_mode == PIPE_TEX_COMPARE_NONE)
1315 return;
1316
1317 /* debug code */
1318 if (0) {
1319 LLVMValueRef indx = lp_build_const_int32(bld->gallivm, 0);
1320 LLVMValueRef coord = LLVMBuildExtractElement(builder, p, indx, "");
1321 LLVMValueRef tex = LLVMBuildExtractElement(builder, texel[chan], indx, "");
1322 lp_build_printf(bld->gallivm, "shadow compare coord %f to texture %f\n",
1323 coord, tex);
1324 }
1325
1326 /* Clamp p coords to [0,1] */
1327 p = lp_build_clamp(&bld->coord_bld, p,
1328 bld->coord_bld.zero,
1329 bld->coord_bld.one);
1330
1331 /* result = (p FUNC texel) ? 1 : 0 */
1332 res = lp_build_cmp(texel_bld, bld->static_state->compare_func,
1333 p, texel[chan]);
1334 res = lp_build_select(texel_bld, res, texel_bld->one, texel_bld->zero);
1335
1336 /* XXX returning result for default GL_DEPTH_TEXTURE_MODE = GL_LUMINANCE */
1337 texel[0] =
1338 texel[1] =
1339 texel[2] = res;
1340 texel[3] = texel_bld->one;
1341 }
1342
1343
1344 /**
1345 * Just set texels to white instead of actually sampling the texture.
1346 * For debugging.
1347 */
1348 void
1349 lp_build_sample_nop(struct gallivm_state *gallivm,
1350 struct lp_type type,
1351 const LLVMValueRef *coords,
1352 LLVMValueRef texel_out[4])
1353 {
1354 LLVMValueRef one = lp_build_one(gallivm, type);
1355 unsigned chan;
1356
1357 for (chan = 0; chan < 4; chan++) {
1358 texel_out[chan] = one;
1359 }
1360 }
1361
1362
1363 /**
1364 * Build texture sampling code.
1365 * 'texel' will return a vector of four LLVMValueRefs corresponding to
1366 * R, G, B, A.
1367 * \param type vector float type to use for coords, etc.
1368 * \param is_fetch if this is a texel fetch instruction.
1369 * \param derivs partial derivatives of (s,t,r,q) with respect to x and y
1370 */
1371 void
1372 lp_build_sample_soa(struct gallivm_state *gallivm,
1373 const struct lp_sampler_static_state *static_state,
1374 struct lp_sampler_dynamic_state *dynamic_state,
1375 struct lp_type type,
1376 boolean is_fetch,
1377 unsigned unit,
1378 const LLVMValueRef *coords,
1379 const LLVMValueRef *offsets,
1380 const struct lp_derivatives *derivs,
1381 LLVMValueRef lod_bias, /* optional */
1382 LLVMValueRef explicit_lod, /* optional */
1383 LLVMValueRef texel_out[4])
1384 {
1385 unsigned dims = texture_dims(static_state->target);
1386 unsigned num_quads = type.length / 4;
1387 unsigned mip_filter = static_state->min_mip_filter;
1388 struct lp_build_sample_context bld;
1389 LLVMTypeRef i32t = LLVMInt32TypeInContext(gallivm->context);
1390 LLVMBuilderRef builder = gallivm->builder;
1391 LLVMValueRef tex_width, tex_height, tex_depth;
1392 LLVMValueRef s;
1393 LLVMValueRef t;
1394 LLVMValueRef r;
1395
1396 if (0) {
1397 enum pipe_format fmt = static_state->format;
1398 debug_printf("Sample from %s\n", util_format_name(fmt));
1399 }
1400
1401 assert(type.floating);
1402
1403 /* Setup our build context */
1404 memset(&bld, 0, sizeof bld);
1405 bld.gallivm = gallivm;
1406 bld.static_state = static_state;
1407 bld.dynamic_state = dynamic_state;
1408 bld.format_desc = util_format_description(static_state->format);
1409 bld.dims = dims;
1410
1411 bld.vector_width = lp_type_width(type);
1412
1413 bld.float_type = lp_type_float(32);
1414 bld.int_type = lp_type_int(32);
1415 bld.coord_type = type;
1416 bld.int_coord_type = lp_int_type(type);
1417 bld.float_size_in_type = lp_type_float(32);
1418 bld.float_size_in_type.length = dims > 1 ? 4 : 1;
1419 bld.int_size_in_type = lp_int_type(bld.float_size_in_type);
1420 bld.texel_type = type;
1421 bld.perquadf_type = type;
1422 /* we want native vector size to be able to use our intrinsics */
1423 bld.perquadf_type.length = type.length > 4 ? ((type.length + 15) / 16) * 4 : 1;
1424 bld.perquadi_type = lp_int_type(bld.perquadf_type);
1425
1426 /*
1427 * There are other situations where at least the multiple int lods could be
1428 * avoided like min and max lod being equal.
1429 */
1430 if ((is_fetch && explicit_lod && bld.static_state->target != PIPE_BUFFER) ||
1431 (!is_fetch && mip_filter != PIPE_TEX_MIPFILTER_NONE)) {
1432 bld.num_lods = num_quads;
1433 }
1434 else {
1435 bld.num_lods = 1;
1436 }
1437
1438 bld.float_size_type = bld.float_size_in_type;
1439 bld.float_size_type.length = bld.num_lods > 1 ? type.length :
1440 bld.float_size_in_type.length;
1441 bld.int_size_type = lp_int_type(bld.float_size_type);
1442
1443 lp_build_context_init(&bld.float_bld, gallivm, bld.float_type);
1444 lp_build_context_init(&bld.float_vec_bld, gallivm, type);
1445 lp_build_context_init(&bld.int_bld, gallivm, bld.int_type);
1446 lp_build_context_init(&bld.coord_bld, gallivm, bld.coord_type);
1447 lp_build_context_init(&bld.int_coord_bld, gallivm, bld.int_coord_type);
1448 lp_build_context_init(&bld.int_size_in_bld, gallivm, bld.int_size_in_type);
1449 lp_build_context_init(&bld.float_size_in_bld, gallivm, bld.float_size_in_type);
1450 lp_build_context_init(&bld.int_size_bld, gallivm, bld.int_size_type);
1451 lp_build_context_init(&bld.float_size_bld, gallivm, bld.float_size_type);
1452 lp_build_context_init(&bld.texel_bld, gallivm, bld.texel_type);
1453 lp_build_context_init(&bld.perquadf_bld, gallivm, bld.perquadf_type);
1454 lp_build_context_init(&bld.perquadi_bld, gallivm, bld.perquadi_type);
1455
1456 /* Get the dynamic state */
1457 tex_width = dynamic_state->width(dynamic_state, gallivm, unit);
1458 tex_height = dynamic_state->height(dynamic_state, gallivm, unit);
1459 tex_depth = dynamic_state->depth(dynamic_state, gallivm, unit);
1460 bld.row_stride_array = dynamic_state->row_stride(dynamic_state, gallivm, unit);
1461 bld.img_stride_array = dynamic_state->img_stride(dynamic_state, gallivm, unit);
1462 bld.base_ptr = dynamic_state->base_ptr(dynamic_state, gallivm, unit);
1463 bld.mip_offsets = dynamic_state->mip_offsets(dynamic_state, gallivm, unit);
1464 /* Note that mip_offsets is an array[level] of offsets to texture images */
1465
1466 s = coords[0];
1467 t = coords[1];
1468 r = coords[2];
1469
1470 /* width, height, depth as single int vector */
1471 if (dims <= 1) {
1472 bld.int_size = tex_width;
1473 }
1474 else {
1475 bld.int_size = LLVMBuildInsertElement(builder, bld.int_size_in_bld.undef,
1476 tex_width, LLVMConstInt(i32t, 0, 0), "");
1477 if (dims >= 2) {
1478 bld.int_size = LLVMBuildInsertElement(builder, bld.int_size,
1479 tex_height, LLVMConstInt(i32t, 1, 0), "");
1480 if (dims >= 3) {
1481 bld.int_size = LLVMBuildInsertElement(builder, bld.int_size,
1482 tex_depth, LLVMConstInt(i32t, 2, 0), "");
1483 }
1484 }
1485 }
1486
1487 if (0) {
1488 /* For debug: no-op texture sampling */
1489 lp_build_sample_nop(gallivm,
1490 bld.texel_type,
1491 coords,
1492 texel_out);
1493 }
1494 else {
1495 LLVMValueRef lod_ipart = NULL, lod_fpart = NULL;
1496 LLVMValueRef ilevel0 = NULL, ilevel1 = NULL;
1497 boolean use_aos = util_format_fits_8unorm(bld.format_desc) &&
1498 lp_is_simple_wrap_mode(static_state->wrap_s) &&
1499 lp_is_simple_wrap_mode(static_state->wrap_t);
1500
1501 if ((gallivm_debug & GALLIVM_DEBUG_PERF) &&
1502 !use_aos && util_format_fits_8unorm(bld.format_desc)) {
1503 debug_printf("%s: using floating point linear filtering for %s\n",
1504 __FUNCTION__, bld.format_desc->short_name);
1505 debug_printf(" min_img %d mag_img %d mip %d wraps %d wrapt %d\n",
1506 static_state->min_img_filter,
1507 static_state->mag_img_filter,
1508 static_state->min_mip_filter,
1509 static_state->wrap_s,
1510 static_state->wrap_t);
1511 }
1512
1513 if (is_fetch) {
1514 lp_build_fetch_texel(&bld, unit, coords,
1515 explicit_lod, offsets,
1516 texel_out);
1517
1518 if (static_state->target != PIPE_BUFFER) {
1519 apply_sampler_swizzle(&bld, texel_out);
1520 }
1521
1522 return;
1523 }
1524
1525 lp_build_sample_common(&bld, unit,
1526 &s, &t, &r,
1527 derivs, lod_bias, explicit_lod,
1528 &lod_ipart, &lod_fpart,
1529 &ilevel0, &ilevel1);
1530
1531 /*
1532 * we only try 8-wide sampling with soa as it appears to
1533 * be a loss with aos with AVX (but it should work).
1534 * (It should be faster if we'd support avx2)
1535 */
1536 if (num_quads == 1 || !use_aos) {
1537
1538 if (num_quads > 1) {
1539 if (mip_filter == PIPE_TEX_MIPFILTER_NONE) {
1540 LLVMValueRef index0 = lp_build_const_int32(gallivm, 0);
1541 /*
1542 * These parameters are the same for all quads,
1543 * could probably simplify.
1544 */
1545 lod_ipart = LLVMBuildExtractElement(builder, lod_ipart, index0, "");
1546 ilevel0 = LLVMBuildExtractElement(builder, ilevel0, index0, "");
1547 }
1548 }
1549 if (use_aos) {
1550 /* do sampling/filtering with fixed pt arithmetic */
1551 lp_build_sample_aos(&bld, unit,
1552 s, t, r,
1553 lod_ipart, lod_fpart,
1554 ilevel0, ilevel1,
1555 texel_out);
1556 }
1557
1558 else {
1559 lp_build_sample_general(&bld, unit,
1560 s, t, r,
1561 lod_ipart, lod_fpart,
1562 ilevel0, ilevel1,
1563 texel_out);
1564 }
1565 }
1566 else {
1567 unsigned j;
1568 struct lp_build_sample_context bld4;
1569 struct lp_type type4 = type;
1570 unsigned i;
1571 LLVMValueRef texelout4[4];
1572 LLVMValueRef texelouttmp[4][LP_MAX_VECTOR_LENGTH/16];
1573
1574 type4.length = 4;
1575
1576 /* Setup our build context */
1577 memset(&bld4, 0, sizeof bld4);
1578 bld4.gallivm = bld.gallivm;
1579 bld4.static_state = bld.static_state;
1580 bld4.dynamic_state = bld.dynamic_state;
1581 bld4.format_desc = bld.format_desc;
1582 bld4.dims = bld.dims;
1583 bld4.row_stride_array = bld.row_stride_array;
1584 bld4.img_stride_array = bld.img_stride_array;
1585 bld4.base_ptr = bld.base_ptr;
1586 bld4.mip_offsets = bld.mip_offsets;
1587 bld4.int_size = bld.int_size;
1588
1589 bld4.vector_width = lp_type_width(type4);
1590
1591 bld4.float_type = lp_type_float(32);
1592 bld4.int_type = lp_type_int(32);
1593 bld4.coord_type = type4;
1594 bld4.int_coord_type = lp_int_type(type4);
1595 bld4.float_size_in_type = lp_type_float(32);
1596 bld4.float_size_in_type.length = dims > 1 ? 4 : 1;
1597 bld4.int_size_in_type = lp_int_type(bld4.float_size_in_type);
1598 bld4.texel_type = type4;
1599 bld4.perquadf_type = type4;
1600 /* we want native vector size to be able to use our intrinsics */
1601 bld4.perquadf_type.length = 1;
1602 bld4.perquadi_type = lp_int_type(bld4.perquadf_type);
1603
1604 bld4.num_lods = 1;
1605 bld4.int_size_type = bld4.int_size_in_type;
1606 bld4.float_size_type = bld4.float_size_in_type;
1607
1608 lp_build_context_init(&bld4.float_bld, gallivm, bld4.float_type);
1609 lp_build_context_init(&bld4.float_vec_bld, gallivm, type4);
1610 lp_build_context_init(&bld4.int_bld, gallivm, bld4.int_type);
1611 lp_build_context_init(&bld4.coord_bld, gallivm, bld4.coord_type);
1612 lp_build_context_init(&bld4.int_coord_bld, gallivm, bld4.int_coord_type);
1613 lp_build_context_init(&bld4.int_size_in_bld, gallivm, bld4.int_size_in_type);
1614 lp_build_context_init(&bld4.float_size_in_bld, gallivm, bld4.float_size_in_type);
1615 lp_build_context_init(&bld4.int_size_bld, gallivm, bld4.int_size_type);
1616 lp_build_context_init(&bld4.float_size_bld, gallivm, bld4.float_size_type);
1617 lp_build_context_init(&bld4.texel_bld, gallivm, bld4.texel_type);
1618 lp_build_context_init(&bld4.perquadf_bld, gallivm, bld4.perquadf_type);
1619 lp_build_context_init(&bld4.perquadi_bld, gallivm, bld4.perquadi_type);
1620
1621 for (i = 0; i < num_quads; i++) {
1622 LLVMValueRef s4, t4, r4;
1623 LLVMValueRef lod_iparts, lod_fparts = NULL;
1624 LLVMValueRef ilevel0s, ilevel1s = NULL;
1625 LLVMValueRef indexi = lp_build_const_int32(gallivm, i);
1626
1627 s4 = lp_build_extract_range(gallivm, s, 4*i, 4);
1628 t4 = lp_build_extract_range(gallivm, t, 4*i, 4);
1629 r4 = lp_build_extract_range(gallivm, r, 4*i, 4);
1630 lod_iparts = LLVMBuildExtractElement(builder, lod_ipart, indexi, "");
1631 ilevel0s = LLVMBuildExtractElement(builder, ilevel0, indexi, "");
1632 if (mip_filter == PIPE_TEX_MIPFILTER_LINEAR) {
1633 ilevel1s = LLVMBuildExtractElement(builder, ilevel1, indexi, "");
1634 lod_fparts = LLVMBuildExtractElement(builder, lod_fpart, indexi, "");
1635 }
1636
1637 if (use_aos) {
1638 /* do sampling/filtering with fixed pt arithmetic */
1639 lp_build_sample_aos(&bld4, unit,
1640 s4, t4, r4,
1641 lod_iparts, lod_fparts,
1642 ilevel0s, ilevel1s,
1643 texelout4);
1644 }
1645
1646 else {
1647 lp_build_sample_general(&bld4, unit,
1648 s4, t4, r4,
1649 lod_iparts, lod_fparts,
1650 ilevel0s, ilevel1s,
1651 texelout4);
1652 }
1653 for (j = 0; j < 4; j++) {
1654 texelouttmp[j][i] = texelout4[j];
1655 }
1656 }
1657
1658 for (j = 0; j < 4; j++) {
1659 texel_out[j] = lp_build_concat(gallivm, texelouttmp[j], type4, num_quads);
1660 }
1661 }
1662 }
1663
1664 lp_build_sample_compare(&bld, r, texel_out);
1665
1666 apply_sampler_swizzle(&bld, texel_out);
1667 }
1668
1669 void
1670 lp_build_size_query_soa(struct gallivm_state *gallivm,
1671 const struct lp_sampler_static_state *static_state,
1672 struct lp_sampler_dynamic_state *dynamic_state,
1673 struct lp_type int_type,
1674 unsigned unit,
1675 LLVMValueRef explicit_lod,
1676 LLVMValueRef *sizes_out)
1677 {
1678 LLVMValueRef lod;
1679 LLVMValueRef size;
1680 int dims, i;
1681 struct lp_build_context bld_int_vec;
1682
1683 switch (static_state->target) {
1684 case PIPE_TEXTURE_1D:
1685 case PIPE_BUFFER:
1686 dims = 1;
1687 break;
1688 case PIPE_TEXTURE_2D:
1689 case PIPE_TEXTURE_CUBE:
1690 case PIPE_TEXTURE_RECT:
1691 dims = 2;
1692 break;
1693 case PIPE_TEXTURE_3D:
1694 dims = 3;
1695 break;
1696
1697 default:
1698 assert(0);
1699 return;
1700 }
1701
1702 assert(!int_type.floating);
1703
1704 lp_build_context_init(&bld_int_vec, gallivm, lp_type_int_vec(32, 128));
1705
1706 if (explicit_lod) {
1707 LLVMValueRef first_level;
1708 lod = LLVMBuildExtractElement(gallivm->builder, explicit_lod, lp_build_const_int32(gallivm, 0), "");
1709 first_level = dynamic_state->first_level(dynamic_state, gallivm, unit);
1710 lod = lp_build_broadcast_scalar(&bld_int_vec,
1711 LLVMBuildAdd(gallivm->builder, lod, first_level, "lod"));
1712
1713 } else {
1714 lod = bld_int_vec.zero;
1715 }
1716
1717 size = bld_int_vec.undef;
1718
1719 size = LLVMBuildInsertElement(gallivm->builder, size,
1720 dynamic_state->width(dynamic_state, gallivm, unit),
1721 lp_build_const_int32(gallivm, 0), "");
1722
1723 if (dims >= 2) {
1724 size = LLVMBuildInsertElement(gallivm->builder, size,
1725 dynamic_state->height(dynamic_state, gallivm, unit),
1726 lp_build_const_int32(gallivm, 1), "");
1727 }
1728
1729 if (dims >= 3) {
1730 size = LLVMBuildInsertElement(gallivm->builder, size,
1731 dynamic_state->depth(dynamic_state, gallivm, unit),
1732 lp_build_const_int32(gallivm, 2), "");
1733 }
1734
1735 size = lp_build_minify(&bld_int_vec, size, lod);
1736
1737 for (i=0; i < dims; i++) {
1738 sizes_out[i] = lp_build_extract_broadcast(gallivm, bld_int_vec.type, int_type,
1739 size,
1740 lp_build_const_int32(gallivm, i));
1741 }
1742 }