1 /**************************************************************************
3 * Copyright 2010 VMware, Inc.
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:
14 * The above copyright notice and this permission notice (including the
15 * next paragraph) shall be included in all copies or substantial portions
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.
26 **************************************************************************/
28 #include "util/u_math.h"
29 #include "util/u_memory.h"
30 #include "util/u_prim.h"
31 #include "draw/draw_context.h"
32 #include "draw/draw_gs.h"
33 #include "draw/draw_vbuf.h"
34 #include "draw/draw_vertex.h"
35 #include "draw/draw_pt.h"
36 #include "draw/draw_prim_assembler.h"
37 #include "draw/draw_vs.h"
38 #include "draw/draw_llvm.h"
39 #include "gallivm/lp_bld_init.h"
42 struct llvm_middle_end
{
43 struct draw_pt_middle_end base
;
44 struct draw_context
*draw
;
47 struct pt_so_emit
*so_emit
;
48 struct pt_fetch
*fetch
;
49 struct pt_post_vs
*post_vs
;
52 unsigned vertex_data_offset
;
57 struct draw_llvm
*llvm
;
58 struct draw_llvm_variant
*current_variant
;
63 static inline struct llvm_middle_end
*
64 llvm_middle_end(struct draw_pt_middle_end
*middle
)
66 return (struct llvm_middle_end
*) middle
;
71 llvm_middle_end_prepare_gs(struct llvm_middle_end
*fpme
)
73 struct draw_context
*draw
= fpme
->draw
;
74 struct draw_geometry_shader
*gs
= draw
->gs
.geometry_shader
;
75 struct draw_gs_llvm_variant_key
*key
;
76 struct draw_gs_llvm_variant
*variant
= NULL
;
77 struct draw_gs_llvm_variant_list_item
*li
;
78 struct llvm_geometry_shader
*shader
= llvm_geometry_shader(gs
);
79 char store
[DRAW_GS_LLVM_MAX_VARIANT_KEY_SIZE
];
82 key
= draw_gs_llvm_make_variant_key(fpme
->llvm
, store
);
84 /* Search shader's list of variants for the key */
85 li
= first_elem(&shader
->variants
);
86 while (!at_end(&shader
->variants
, li
)) {
87 if (memcmp(&li
->base
->key
, key
, shader
->variant_key_size
) == 0) {
95 /* found the variant, move to head of global list (for LRU) */
96 move_to_head(&fpme
->llvm
->gs_variants_list
,
97 &variant
->list_item_global
);
100 /* Need to create new variant */
102 /* First check if we've created too many variants. If so, free
103 * 25% of the LRU to avoid using too much memory.
105 if (fpme
->llvm
->nr_gs_variants
>= DRAW_MAX_SHADER_VARIANTS
) {
107 * XXX: should we flush here ?
109 for (i
= 0; i
< DRAW_MAX_SHADER_VARIANTS
/ 4; i
++) {
110 struct draw_gs_llvm_variant_list_item
*item
;
111 if (is_empty_list(&fpme
->llvm
->gs_variants_list
)) {
114 item
= last_elem(&fpme
->llvm
->gs_variants_list
);
117 draw_gs_llvm_destroy_variant(item
->base
);
121 variant
= draw_gs_llvm_create_variant(fpme
->llvm
, gs
->info
.num_outputs
, key
);
124 insert_at_head(&shader
->variants
, &variant
->list_item_local
);
125 insert_at_head(&fpme
->llvm
->gs_variants_list
,
126 &variant
->list_item_global
);
127 fpme
->llvm
->nr_gs_variants
++;
128 shader
->variants_cached
++;
132 gs
->current_variant
= variant
;
136 * Prepare/validate middle part of the vertex pipeline.
137 * NOTE: if you change this function, also look at the non-LLVM
138 * function fetch_pipeline_prepare() for similar changes.
141 llvm_middle_end_prepare( struct draw_pt_middle_end
*middle
,
144 unsigned *max_vertices
)
146 struct llvm_middle_end
*fpme
= llvm_middle_end(middle
);
147 struct draw_context
*draw
= fpme
->draw
;
148 struct draw_vertex_shader
*vs
= draw
->vs
.vertex_shader
;
149 struct draw_geometry_shader
*gs
= draw
->gs
.geometry_shader
;
150 const unsigned out_prim
= gs
? gs
->output_primitive
:
151 u_assembled_prim(in_prim
);
152 unsigned point_clip
= draw
->rasterizer
->fill_front
== PIPE_POLYGON_MODE_POINT
||
153 out_prim
== PIPE_PRIM_POINTS
;
156 fpme
->input_prim
= in_prim
;
159 draw_pt_post_vs_prepare( fpme
->post_vs
,
163 point_clip
? draw
->guard_band_points_xy
:
165 draw
->bypass_viewport
,
166 draw
->rasterizer
->clip_halfz
,
167 (draw
->vs
.edgeflag_output
? TRUE
: FALSE
) );
169 draw_pt_so_emit_prepare( fpme
->so_emit
, gs
== NULL
);
171 if (!(opt
& PT_PIPELINE
)) {
172 draw_pt_emit_prepare( fpme
->emit
, out_prim
,
175 *max_vertices
= MAX2( *max_vertices
, 4096 );
178 /* limit max fetches by limiting max_vertices */
179 *max_vertices
= 4096;
182 /* Get the number of float[4] attributes per vertex.
183 * Note: this must be done after draw_pt_emit_prepare() since that
184 * can effect the vertex size.
186 nr
= MAX2(vs
->info
.num_inputs
, draw_total_vs_outputs(draw
));
188 /* Always leave room for the vertex header whether we need it or
189 * not. It's hard to get rid of it in particular because of the
190 * viewport code in draw_pt_post_vs.c.
192 fpme
->vertex_size
= sizeof(struct vertex_header
) + nr
* 4 * sizeof(float);
194 /* return even number */
195 *max_vertices
= *max_vertices
& ~1;
197 /* Find/create the vertex shader variant */
199 struct draw_llvm_variant_key
*key
;
200 struct draw_llvm_variant
*variant
= NULL
;
201 struct draw_llvm_variant_list_item
*li
;
202 struct llvm_vertex_shader
*shader
= llvm_vertex_shader(vs
);
203 char store
[DRAW_LLVM_MAX_VARIANT_KEY_SIZE
];
206 key
= draw_llvm_make_variant_key(fpme
->llvm
, store
);
208 /* Search shader's list of variants for the key */
209 li
= first_elem(&shader
->variants
);
210 while (!at_end(&shader
->variants
, li
)) {
211 if (memcmp(&li
->base
->key
, key
, shader
->variant_key_size
) == 0) {
219 /* found the variant, move to head of global list (for LRU) */
220 move_to_head(&fpme
->llvm
->vs_variants_list
,
221 &variant
->list_item_global
);
224 /* Need to create new variant */
226 /* First check if we've created too many variants. If so, free
227 * 25% of the LRU to avoid using too much memory.
229 if (fpme
->llvm
->nr_variants
>= DRAW_MAX_SHADER_VARIANTS
) {
231 * XXX: should we flush here ?
233 for (i
= 0; i
< DRAW_MAX_SHADER_VARIANTS
/ 4; i
++) {
234 struct draw_llvm_variant_list_item
*item
;
235 if (is_empty_list(&fpme
->llvm
->vs_variants_list
)) {
238 item
= last_elem(&fpme
->llvm
->vs_variants_list
);
241 draw_llvm_destroy_variant(item
->base
);
245 variant
= draw_llvm_create_variant(fpme
->llvm
, nr
, key
);
248 insert_at_head(&shader
->variants
, &variant
->list_item_local
);
249 insert_at_head(&fpme
->llvm
->vs_variants_list
,
250 &variant
->list_item_global
);
251 fpme
->llvm
->nr_variants
++;
252 shader
->variants_cached
++;
256 fpme
->current_variant
= variant
;
260 llvm_middle_end_prepare_gs(fpme
);
266 * Bind/update constant buffer pointers, clip planes and viewport dims.
267 * These are "light weight" parameters which aren't baked into the
268 * generated code. Updating these items is much cheaper than revalidating
269 * and rebuilding the generated pipeline code.
272 llvm_middle_end_bind_parameters(struct draw_pt_middle_end
*middle
)
274 static const float fake_const_buf
[4];
275 struct llvm_middle_end
*fpme
= llvm_middle_end(middle
);
276 struct draw_context
*draw
= fpme
->draw
;
277 struct draw_llvm
*llvm
= fpme
->llvm
;
280 for (i
= 0; i
< ARRAY_SIZE(llvm
->jit_context
.vs_constants
); ++i
) {
282 draw
->pt
.user
.vs_constants_size
[i
] / (sizeof(float) * 4);
283 llvm
->jit_context
.vs_constants
[i
] = draw
->pt
.user
.vs_constants
[i
];
284 llvm
->jit_context
.num_vs_constants
[i
] = num_consts
;
285 if (num_consts
== 0) {
286 llvm
->jit_context
.vs_constants
[i
] = fake_const_buf
;
289 for (i
= 0; i
< ARRAY_SIZE(llvm
->gs_jit_context
.constants
); ++i
) {
291 draw
->pt
.user
.gs_constants_size
[i
] / (sizeof(float) * 4);
292 llvm
->gs_jit_context
.constants
[i
] = draw
->pt
.user
.gs_constants
[i
];
293 llvm
->gs_jit_context
.num_constants
[i
] = num_consts
;
294 if (num_consts
== 0) {
295 llvm
->gs_jit_context
.constants
[i
] = fake_const_buf
;
299 llvm
->jit_context
.planes
=
300 (float (*)[DRAW_TOTAL_CLIP_PLANES
][4]) draw
->pt
.user
.planes
[0];
301 llvm
->gs_jit_context
.planes
=
302 (float (*)[DRAW_TOTAL_CLIP_PLANES
][4]) draw
->pt
.user
.planes
[0];
304 llvm
->jit_context
.viewports
= draw
->viewports
;
305 llvm
->gs_jit_context
.viewports
= draw
->viewports
;
310 pipeline(struct llvm_middle_end
*llvm
,
311 const struct draw_vertex_info
*vert_info
,
312 const struct draw_prim_info
*prim_info
)
314 if (prim_info
->linear
)
315 draw_pipeline_run_linear( llvm
->draw
,
319 draw_pipeline_run( llvm
->draw
,
326 emit(struct pt_emit
*emit
,
327 const struct draw_vertex_info
*vert_info
,
328 const struct draw_prim_info
*prim_info
)
330 if (prim_info
->linear
) {
331 draw_pt_emit_linear(emit
, vert_info
, prim_info
);
334 draw_pt_emit(emit
, vert_info
, prim_info
);
340 llvm_pipeline_generic(struct draw_pt_middle_end
*middle
,
341 const struct draw_fetch_info
*fetch_info
,
342 const struct draw_prim_info
*in_prim_info
)
344 struct llvm_middle_end
*fpme
= llvm_middle_end(middle
);
345 struct draw_context
*draw
= fpme
->draw
;
346 struct draw_geometry_shader
*gshader
= draw
->gs
.geometry_shader
;
347 struct draw_prim_info gs_prim_info
;
348 struct draw_vertex_info llvm_vert_info
;
349 struct draw_vertex_info gs_vert_info
;
350 struct draw_vertex_info
*vert_info
;
351 struct draw_prim_info ia_prim_info
;
352 struct draw_vertex_info ia_vert_info
;
353 const struct draw_prim_info
*prim_info
= in_prim_info
;
354 boolean free_prim_info
= FALSE
;
355 unsigned opt
= fpme
->opt
;
357 unsigned start_or_maxelt
, vid_base
;
358 const unsigned *elts
;
360 llvm_vert_info
.count
= fetch_info
->count
;
361 llvm_vert_info
.vertex_size
= fpme
->vertex_size
;
362 llvm_vert_info
.stride
= fpme
->vertex_size
;
363 llvm_vert_info
.verts
= (struct vertex_header
*)
364 MALLOC(fpme
->vertex_size
*
365 align(fetch_info
->count
, lp_native_vector_width
/ 32));
366 if (!llvm_vert_info
.verts
) {
371 if (draw
->collect_statistics
) {
372 draw
->statistics
.ia_vertices
+= prim_info
->count
;
373 draw
->statistics
.ia_primitives
+=
374 u_decomposed_prims_for_vertices(prim_info
->prim
, prim_info
->count
);
375 draw
->statistics
.vs_invocations
+= fetch_info
->count
;
378 if (fetch_info
->linear
) {
379 start_or_maxelt
= fetch_info
->start
;
380 vid_base
= draw
->start_index
;
384 start_or_maxelt
= draw
->pt
.user
.eltMax
;
385 vid_base
= draw
->pt
.user
.eltBias
;
386 elts
= fetch_info
->elts
;
388 clipped
= fpme
->current_variant
->jit_func(&fpme
->llvm
->jit_context
,
389 llvm_vert_info
.verts
,
390 draw
->pt
.user
.vbuffer
,
394 draw
->pt
.vertex_buffer
,
397 draw
->start_instance
,
400 /* Finished with fetch and vs:
403 vert_info
= &llvm_vert_info
;
405 if ((opt
& PT_SHADE
) && gshader
) {
406 struct draw_vertex_shader
*vshader
= draw
->vs
.vertex_shader
;
407 draw_geometry_shader_run(gshader
,
408 draw
->pt
.user
.gs_constants
,
409 draw
->pt
.user
.gs_constants_size
,
416 FREE(vert_info
->verts
);
417 vert_info
= &gs_vert_info
;
418 prim_info
= &gs_prim_info
;
420 if (draw_prim_assembler_is_required(draw
, prim_info
, vert_info
)) {
421 draw_prim_assembler_run(draw
, prim_info
, vert_info
,
422 &ia_prim_info
, &ia_vert_info
);
424 if (ia_vert_info
.count
) {
425 FREE(vert_info
->verts
);
426 vert_info
= &ia_vert_info
;
427 prim_info
= &ia_prim_info
;
428 free_prim_info
= TRUE
;
432 if (prim_info
->count
== 0) {
433 debug_printf("GS/IA didn't emit any vertices!\n");
435 FREE(vert_info
->verts
);
436 if (free_prim_info
) {
437 FREE(prim_info
->primitive_lengths
);
442 /* stream output needs to be done before clipping */
443 draw_pt_so_emit( fpme
->so_emit
, vert_info
, prim_info
);
445 draw_stats_clipper_primitives(draw
, prim_info
);
448 * if there's no position, need to stop now, or the latter stages
449 * will try to access non-existent position output.
451 if (draw_current_shader_position_output(draw
) != -1) {
452 if ((opt
& PT_SHADE
) && (gshader
||
453 draw
->vs
.vertex_shader
->info
.writes_viewport_index
)) {
454 clipped
= draw_pt_post_vs_run( fpme
->post_vs
, vert_info
, prim_info
);
456 /* "clipped" also includes non-one edgeflag */
461 /* Do we need to run the pipeline? Now will come here if clipped
463 if (opt
& PT_PIPELINE
) {
464 pipeline( fpme
, vert_info
, prim_info
);
467 emit( fpme
->emit
, vert_info
, prim_info
);
470 FREE(vert_info
->verts
);
471 if (free_prim_info
) {
472 FREE(prim_info
->primitive_lengths
);
477 static inline unsigned
478 prim_type(unsigned prim
, unsigned flags
)
480 if (flags
& DRAW_LINE_LOOP_AS_STRIP
)
481 return PIPE_PRIM_LINE_STRIP
;
488 llvm_middle_end_run(struct draw_pt_middle_end
*middle
,
489 const unsigned *fetch_elts
,
490 unsigned fetch_count
,
491 const ushort
*draw_elts
,
495 struct llvm_middle_end
*fpme
= llvm_middle_end(middle
);
496 struct draw_fetch_info fetch_info
;
497 struct draw_prim_info prim_info
;
499 fetch_info
.linear
= FALSE
;
500 fetch_info
.start
= 0;
501 fetch_info
.elts
= fetch_elts
;
502 fetch_info
.count
= fetch_count
;
504 prim_info
.linear
= FALSE
;
506 prim_info
.count
= draw_count
;
507 prim_info
.elts
= draw_elts
;
508 prim_info
.prim
= prim_type(fpme
->input_prim
, prim_flags
);
509 prim_info
.flags
= prim_flags
;
510 prim_info
.primitive_count
= 1;
511 prim_info
.primitive_lengths
= &draw_count
;
513 llvm_pipeline_generic( middle
, &fetch_info
, &prim_info
);
518 llvm_middle_end_linear_run(struct draw_pt_middle_end
*middle
,
523 struct llvm_middle_end
*fpme
= llvm_middle_end(middle
);
524 struct draw_fetch_info fetch_info
;
525 struct draw_prim_info prim_info
;
527 fetch_info
.linear
= TRUE
;
528 fetch_info
.start
= start
;
529 fetch_info
.count
= count
;
530 fetch_info
.elts
= NULL
;
532 prim_info
.linear
= TRUE
;
534 prim_info
.count
= count
;
535 prim_info
.elts
= NULL
;
536 prim_info
.prim
= prim_type(fpme
->input_prim
, prim_flags
);
537 prim_info
.flags
= prim_flags
;
538 prim_info
.primitive_count
= 1;
539 prim_info
.primitive_lengths
= &count
;
541 llvm_pipeline_generic( middle
, &fetch_info
, &prim_info
);
546 llvm_middle_end_linear_run_elts(struct draw_pt_middle_end
*middle
,
549 const ushort
*draw_elts
,
553 struct llvm_middle_end
*fpme
= llvm_middle_end(middle
);
554 struct draw_fetch_info fetch_info
;
555 struct draw_prim_info prim_info
;
557 fetch_info
.linear
= TRUE
;
558 fetch_info
.start
= start
;
559 fetch_info
.count
= count
;
560 fetch_info
.elts
= NULL
;
562 prim_info
.linear
= FALSE
;
564 prim_info
.count
= draw_count
;
565 prim_info
.elts
= draw_elts
;
566 prim_info
.prim
= prim_type(fpme
->input_prim
, prim_flags
);
567 prim_info
.flags
= prim_flags
;
568 prim_info
.primitive_count
= 1;
569 prim_info
.primitive_lengths
= &draw_count
;
571 llvm_pipeline_generic( middle
, &fetch_info
, &prim_info
);
578 llvm_middle_end_finish(struct draw_pt_middle_end
*middle
)
585 llvm_middle_end_destroy(struct draw_pt_middle_end
*middle
)
587 struct llvm_middle_end
*fpme
= llvm_middle_end(middle
);
590 draw_pt_fetch_destroy( fpme
->fetch
);
593 draw_pt_emit_destroy( fpme
->emit
);
596 draw_pt_so_emit_destroy( fpme
->so_emit
);
599 draw_pt_post_vs_destroy( fpme
->post_vs
);
605 struct draw_pt_middle_end
*
606 draw_pt_fetch_pipeline_or_emit_llvm(struct draw_context
*draw
)
608 struct llvm_middle_end
*fpme
= 0;
613 fpme
= CALLOC_STRUCT( llvm_middle_end
);
617 fpme
->base
.prepare
= llvm_middle_end_prepare
;
618 fpme
->base
.bind_parameters
= llvm_middle_end_bind_parameters
;
619 fpme
->base
.run
= llvm_middle_end_run
;
620 fpme
->base
.run_linear
= llvm_middle_end_linear_run
;
621 fpme
->base
.run_linear_elts
= llvm_middle_end_linear_run_elts
;
622 fpme
->base
.finish
= llvm_middle_end_finish
;
623 fpme
->base
.destroy
= llvm_middle_end_destroy
;
627 fpme
->fetch
= draw_pt_fetch_create( draw
);
631 fpme
->post_vs
= draw_pt_post_vs_create( draw
);
635 fpme
->emit
= draw_pt_emit_create( draw
);
639 fpme
->so_emit
= draw_pt_so_emit_create( draw
);
643 fpme
->llvm
= draw
->llvm
;
647 fpme
->current_variant
= NULL
;
653 llvm_middle_end_destroy( &fpme
->base
);