1 /**************************************************************************
3 * Copyright 2009 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 TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR
22 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
26 **************************************************************************/
30 #include "draw_private.h"
31 #include "draw_context.h"
33 #include "tgsi/tgsi_parse.h"
34 #include "tgsi/tgsi_exec.h"
36 #include "pipe/p_shader_tokens.h"
38 #include "util/u_math.h"
39 #include "util/u_memory.h"
41 #define MAX_PRIM_VERTICES 6
42 /* fixme: move it from here */
43 #define MAX_PRIMITIVES 64
46 draw_gs_init( struct draw_context
*draw
)
48 draw
->gs
.machine
= tgsi_exec_machine_create();
49 if (!draw
->gs
.machine
)
52 draw
->gs
.machine
->Primitives
= align_malloc(
53 MAX_PRIMITIVES
* sizeof(struct tgsi_exec_vector
), 16);
54 if (!draw
->gs
.machine
->Primitives
)
56 memset(draw
->gs
.machine
->Primitives
, 0,
57 MAX_PRIMITIVES
* sizeof(struct tgsi_exec_vector
));
62 void draw_gs_destroy( struct draw_context
*draw
)
64 if (!draw
->gs
.machine
)
67 align_free(draw
->gs
.machine
->Primitives
);
69 tgsi_exec_machine_destroy(draw
->gs
.machine
);
73 draw_gs_set_constants(struct draw_context
*draw
,
75 const void *constants
,
81 struct draw_geometry_shader
*
82 draw_create_geometry_shader(struct draw_context
*draw
,
83 const struct pipe_shader_state
*state
)
85 struct draw_geometry_shader
*gs
;
88 gs
= CALLOC_STRUCT(draw_geometry_shader
);
94 gs
->state
.tokens
= tgsi_dup_tokens(state
->tokens
);
95 if (!gs
->state
.tokens
) {
100 tgsi_scan_shader(state
->tokens
, &gs
->info
);
102 /* setup the defaults */
103 gs
->input_primitive
= PIPE_PRIM_TRIANGLES
;
104 gs
->output_primitive
= PIPE_PRIM_TRIANGLE_STRIP
;
105 gs
->max_output_vertices
= 32;
107 for (i
= 0; i
< gs
->info
.num_properties
; ++i
) {
108 if (gs
->info
.properties
[i
].name
==
109 TGSI_PROPERTY_GS_INPUT_PRIM
)
110 gs
->input_primitive
= gs
->info
.properties
[i
].data
[0];
111 else if (gs
->info
.properties
[i
].name
==
112 TGSI_PROPERTY_GS_OUTPUT_PRIM
)
113 gs
->output_primitive
= gs
->info
.properties
[i
].data
[0];
114 else if (gs
->info
.properties
[i
].name
==
115 TGSI_PROPERTY_GS_MAX_VERTICES
)
116 gs
->max_output_vertices
= gs
->info
.properties
[i
].data
[0];
119 gs
->machine
= draw
->gs
.machine
;
124 for (i
= 0; i
< gs
->info
.num_outputs
; i
++) {
125 if (gs
->info
.output_semantic_name
[i
] == TGSI_SEMANTIC_POSITION
&&
126 gs
->info
.output_semantic_index
[i
] == 0)
127 gs
->position_output
= i
;
134 void draw_bind_geometry_shader(struct draw_context
*draw
,
135 struct draw_geometry_shader
*dgs
)
137 draw_do_flush(draw
, DRAW_FLUSH_STATE_CHANGE
);
140 draw
->gs
.geometry_shader
= dgs
;
141 draw
->gs
.num_gs_outputs
= dgs
->info
.num_outputs
;
142 draw
->gs
.position_output
= dgs
->position_output
;
143 draw_geometry_shader_prepare(dgs
, draw
);
146 draw
->gs
.geometry_shader
= NULL
;
147 draw
->gs
.num_gs_outputs
= 0;
151 void draw_delete_geometry_shader(struct draw_context
*draw
,
152 struct draw_geometry_shader
*dgs
)
157 static INLINE
int num_vertices_for_prim(int prim
)
160 case PIPE_PRIM_POINTS
:
162 case PIPE_PRIM_LINES
:
164 case PIPE_PRIM_LINE_LOOP
:
166 case PIPE_PRIM_LINE_STRIP
:
168 case PIPE_PRIM_TRIANGLES
:
170 case PIPE_PRIM_TRIANGLE_STRIP
:
172 case PIPE_PRIM_TRIANGLE_FAN
:
174 case PIPE_PRIM_LINES_ADJACENCY
:
175 case PIPE_PRIM_LINE_STRIP_ADJACENCY
:
177 case PIPE_PRIM_TRIANGLES_ADJACENCY
:
178 case PIPE_PRIM_TRIANGLE_STRIP_ADJACENCY
:
181 assert(!"Bad geometry shader input");
186 static void draw_fetch_geometry_input(struct draw_geometry_shader
*shader
,
189 const float (*input_ptr
)[4],
190 unsigned input_vertex_stride
,
191 unsigned inputs_from_vs
)
193 struct tgsi_exec_machine
*machine
= shader
->machine
;
194 unsigned slot
, vs_slot
, k
, j
;
195 unsigned num_vertices
= num_vertices_for_prim(shader
->input_primitive
);
198 for (slot
= 0, vs_slot
= 0; slot
< shader
->info
.num_inputs
; slot
++) {
199 /*debug_printf("Slot = %d (semantic = %d)\n", slot,
200 shader->info.input_semantic_name[slot]);*/
201 if (shader
->info
.input_semantic_name
[slot
] ==
202 TGSI_SEMANTIC_PRIMID
) {
203 for (j
= 0; j
< num_primitives
; ++j
) {
204 machine
->Inputs
[idx
].xyzw
[0].f
[j
] = (float)start_primitive
+ j
;
205 machine
->Inputs
[idx
].xyzw
[1].f
[j
] = (float)start_primitive
+ j
;
206 machine
->Inputs
[idx
].xyzw
[2].f
[j
] = (float)start_primitive
+ j
;
207 machine
->Inputs
[idx
].xyzw
[3].f
[j
] = (float)start_primitive
+ j
;
211 for (j
= 0; j
< num_primitives
; ++j
) {
213 const float (*prim_ptr
)[4];
214 /*debug_printf(" %d) Prim (num_verts = %d)\n", start_primitive + j,
216 prim_ptr
= (const float (*)[4])(
217 (const char *)input_ptr
+
218 (j
* num_vertices
* input_vertex_stride
));
220 for (k
= 0; k
< num_vertices
; ++k
, ++vidx
) {
221 const float (*input
)[4];
222 input
= (const float (*)[4])(
223 (const char *)prim_ptr
+ (k
* input_vertex_stride
));
224 vidx
= k
* TGSI_EXEC_MAX_INPUT_ATTRIBS
+ slot
;
225 /*debug_printf("\t%d)(%d) Input vert:\n", vidx, k);*/
227 assert(!util_is_inf_or_nan(input
[vs_slot
][0]));
228 assert(!util_is_inf_or_nan(input
[vs_slot
][1]));
229 assert(!util_is_inf_or_nan(input
[vs_slot
][2]));
230 assert(!util_is_inf_or_nan(input
[vs_slot
][3]));
232 machine
->Inputs
[vidx
].xyzw
[0].f
[j
] = input
[vs_slot
][0];
233 machine
->Inputs
[vidx
].xyzw
[1].f
[j
] = input
[vs_slot
][1];
234 machine
->Inputs
[vidx
].xyzw
[2].f
[j
] = input
[vs_slot
][2];
235 machine
->Inputs
[vidx
].xyzw
[3].f
[j
] = input
[vs_slot
][3];
237 debug_printf("\t\t%d %f %f %f %f\n", slot
,
238 machine
->Inputs
[vidx
].xyzw
[0].f
[j
],
239 machine
->Inputs
[vidx
].xyzw
[1].f
[j
],
240 machine
->Inputs
[vidx
].xyzw
[2].f
[j
],
241 machine
->Inputs
[vidx
].xyzw
[3].f
[j
]);
252 draw_geometry_fetch_outputs(struct draw_geometry_shader
*shader
,
255 unsigned vertex_size
)
257 struct tgsi_exec_machine
*machine
= shader
->machine
;
258 unsigned prim_idx
, j
, slot
;
260 /* Unswizzle all output results.
262 /* FIXME: handle all the primitives produced by the gs, not just
264 unsigned prim_count =
265 mach->Temps[TEMP_PRIMITIVE_I].xyzw[TEMP_PRIMITIVE_C].u[0];*/
266 for (prim_idx
= 0; prim_idx
< num_primitives
; ++prim_idx
) {
267 unsigned num_verts_per_prim
= machine
->Primitives
[0];
268 for (j
= 0; j
< num_verts_per_prim
; j
++) {
269 int idx
= (prim_idx
* num_verts_per_prim
+ j
) *
270 shader
->info
.num_outputs
;
272 debug_printf("%d) Output vert:\n", idx
);
274 for (slot
= 0; slot
< shader
->info
.num_outputs
; slot
++) {
275 output
[slot
][0] = machine
->Outputs
[idx
+ slot
].xyzw
[0].f
[prim_idx
];
276 output
[slot
][1] = machine
->Outputs
[idx
+ slot
].xyzw
[1].f
[prim_idx
];
277 output
[slot
][2] = machine
->Outputs
[idx
+ slot
].xyzw
[2].f
[prim_idx
];
278 output
[slot
][3] = machine
->Outputs
[idx
+ slot
].xyzw
[3].f
[prim_idx
];
280 debug_printf("\t%d: %f %f %f %f\n", slot
,
286 debug_assert(!util_is_inf_or_nan(output
[slot
][0]));
288 output
= (float (*)[4])((char *)output
+ vertex_size
);
293 void draw_geometry_shader_run(struct draw_geometry_shader
*shader
,
294 const float (*input
)[4],
296 const void *constants
[PIPE_MAX_CONSTANT_BUFFERS
],
298 unsigned input_stride
,
299 unsigned vertex_size
)
301 struct tgsi_exec_machine
*machine
= shader
->machine
;
303 unsigned num_vertices
= num_vertices_for_prim(shader
->input_primitive
);
304 unsigned num_primitives
= count
/num_vertices
;
305 unsigned inputs_from_vs
= 0;
307 for (i
= 0; i
< PIPE_MAX_CONSTANT_BUFFERS
; i
++) {
308 machine
->Consts
[i
] = constants
[i
];
311 for (i
= 0; i
< shader
->info
.num_inputs
; ++i
) {
312 if (shader
->info
.input_semantic_name
[i
] != TGSI_SEMANTIC_PRIMID
)
316 for (i
= 0; i
< num_primitives
; ++i
) {
317 unsigned int max_primitives
= 1;
319 draw_fetch_geometry_input(shader
, i
, max_primitives
, input
,
320 input_stride
, inputs_from_vs
);
322 tgsi_set_exec_mask(machine
,
328 /* run interpreter */
329 tgsi_exec_machine_run(machine
);
331 draw_geometry_fetch_outputs(shader
, max_primitives
,
332 output
, vertex_size
);
336 void draw_geometry_shader_delete(struct draw_geometry_shader
*shader
)
338 FREE((void*) shader
->state
.tokens
);
342 void draw_geometry_shader_prepare(struct draw_geometry_shader
*shader
,
343 struct draw_context
*draw
)
345 if (shader
->machine
->Tokens
!= shader
->state
.tokens
) {
346 tgsi_exec_machine_bind_shader(shader
->machine
,
347 shader
->state
.tokens
,
348 draw
->gs
.num_samplers
,