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
));
63 void draw_gs_set_constants( struct draw_context
*draw
,
64 const float (*constants
)[4],
70 struct draw_geometry_shader
*
71 draw_create_geometry_shader(struct draw_context
*draw
,
72 const struct pipe_shader_state
*state
)
74 struct draw_geometry_shader
*gs
;
77 gs
= CALLOC_STRUCT(draw_geometry_shader
);
83 gs
->state
.tokens
= tgsi_dup_tokens(state
->tokens
);
84 if (!gs
->state
.tokens
) {
89 tgsi_scan_shader(state
->tokens
, &gs
->info
);
91 /* setup the defaults */
92 gs
->input_primitive
= PIPE_PRIM_TRIANGLES
;
93 gs
->output_primitive
= PIPE_PRIM_TRIANGLE_STRIP
;
94 gs
->max_output_vertices
= 32;
96 for (i
= 0; i
< gs
->info
.num_properties
; ++i
) {
97 if (gs
->info
.properties
[i
].name
==
98 TGSI_PROPERTY_GS_INPUT_PRIM
)
99 gs
->input_primitive
= gs
->info
.properties
[i
].data
[0];
100 else if (gs
->info
.properties
[i
].name
==
101 TGSI_PROPERTY_GS_OUTPUT_PRIM
)
102 gs
->output_primitive
= gs
->info
.properties
[i
].data
[0];
103 else if (gs
->info
.properties
[i
].name
==
104 TGSI_PROPERTY_GS_MAX_VERTICES
)
105 gs
->max_output_vertices
= gs
->info
.properties
[i
].data
[0];
108 gs
->machine
= draw
->gs
.machine
;
113 for (i
= 0; i
< gs
->info
.num_outputs
; i
++) {
114 if (gs
->info
.output_semantic_name
[i
] == TGSI_SEMANTIC_POSITION
&&
115 gs
->info
.output_semantic_index
[i
] == 0)
116 gs
->position_output
= i
;
123 void draw_bind_geometry_shader(struct draw_context
*draw
,
124 struct draw_geometry_shader
*dgs
)
126 draw_do_flush(draw
, DRAW_FLUSH_STATE_CHANGE
);
129 draw
->gs
.geometry_shader
= dgs
;
130 draw
->gs
.num_gs_outputs
= dgs
->info
.num_outputs
;
131 draw
->gs
.position_output
= dgs
->position_output
;
132 draw_geometry_shader_prepare(dgs
, draw
);
135 draw
->gs
.geometry_shader
= NULL
;
136 draw
->gs
.num_gs_outputs
= 0;
140 void draw_delete_geometry_shader(struct draw_context
*draw
,
141 struct draw_geometry_shader
*dgs
)
146 static INLINE
int num_vertices_for_prim(int prim
)
149 case PIPE_PRIM_POINTS
:
151 case PIPE_PRIM_LINES
:
153 case PIPE_PRIM_LINE_LOOP
:
155 case PIPE_PRIM_LINE_STRIP
:
157 case PIPE_PRIM_TRIANGLES
:
159 case PIPE_PRIM_TRIANGLE_STRIP
:
161 case PIPE_PRIM_TRIANGLE_FAN
:
163 case PIPE_PRIM_LINES_ADJACENCY
:
164 case PIPE_PRIM_LINE_STRIP_ADJACENCY
:
166 case PIPE_PRIM_TRIANGLES_ADJACENCY
:
167 case PIPE_PRIM_TRIANGLE_STRIP_ADJACENCY
:
170 assert(!"Bad geometry shader input");
175 static void draw_fetch_geometry_input(struct draw_geometry_shader
*shader
,
178 const float (*input_ptr
)[4],
179 unsigned input_vertex_stride
,
180 unsigned inputs_from_vs
)
182 struct tgsi_exec_machine
*machine
= shader
->machine
;
183 unsigned slot
, vs_slot
, k
, j
;
184 unsigned num_vertices
= num_vertices_for_prim(shader
->input_primitive
);
187 for (slot
= 0, vs_slot
= 0; slot
< shader
->info
.num_inputs
; slot
++) {
188 /*debug_printf("Slot = %d (semantic = %d)\n", slot,
189 shader->info.input_semantic_name[slot]);*/
190 if (shader
->info
.input_semantic_name
[slot
] ==
191 TGSI_SEMANTIC_PRIMID
) {
192 for (j
= 0; j
< num_primitives
; ++j
) {
193 machine
->Inputs
[idx
].xyzw
[0].f
[j
] = (float)start_primitive
+ j
;
194 machine
->Inputs
[idx
].xyzw
[1].f
[j
] = (float)start_primitive
+ j
;
195 machine
->Inputs
[idx
].xyzw
[2].f
[j
] = (float)start_primitive
+ j
;
196 machine
->Inputs
[idx
].xyzw
[3].f
[j
] = (float)start_primitive
+ j
;
200 for (j
= 0; j
< num_primitives
; ++j
) {
202 const float (*prim_ptr
)[4];
203 /*debug_printf(" %d) Prim (num_verts = %d)\n", start_primitive + j,
205 prim_ptr
= (const float (*)[4])(
206 (const char *)input_ptr
+
207 (j
* num_vertices
* input_vertex_stride
));
209 for (k
= 0; k
< num_vertices
; ++k
, ++vidx
) {
210 const float (*input
)[4];
211 input
= (const float (*)[4])(
212 (const char *)prim_ptr
+ (k
* input_vertex_stride
));
213 vidx
= k
* TGSI_EXEC_MAX_INPUT_ATTRIBS
+ slot
;
214 /*debug_printf("\t%d)(%d) Input vert:\n", vidx, k);*/
216 assert(!util_is_inf_or_nan(input
[vs_slot
][0]));
217 assert(!util_is_inf_or_nan(input
[vs_slot
][1]));
218 assert(!util_is_inf_or_nan(input
[vs_slot
][2]));
219 assert(!util_is_inf_or_nan(input
[vs_slot
][3]));
221 machine
->Inputs
[vidx
].xyzw
[0].f
[j
] = input
[vs_slot
][0];
222 machine
->Inputs
[vidx
].xyzw
[1].f
[j
] = input
[vs_slot
][1];
223 machine
->Inputs
[vidx
].xyzw
[2].f
[j
] = input
[vs_slot
][2];
224 machine
->Inputs
[vidx
].xyzw
[3].f
[j
] = input
[vs_slot
][3];
226 debug_printf("\t\t%d %f %f %f %f\n", slot
,
227 machine
->Inputs
[vidx
].xyzw
[0].f
[j
],
228 machine
->Inputs
[vidx
].xyzw
[1].f
[j
],
229 machine
->Inputs
[vidx
].xyzw
[2].f
[j
],
230 machine
->Inputs
[vidx
].xyzw
[3].f
[j
]);
241 draw_geometry_fetch_outputs(struct draw_geometry_shader
*shader
,
244 unsigned vertex_size
)
246 struct tgsi_exec_machine
*machine
= shader
->machine
;
247 unsigned prim_idx
, j
, slot
;
249 /* Unswizzle all output results.
251 /* FIXME: handle all the primitives produced by the gs, not just
253 unsigned prim_count =
254 mach->Temps[TEMP_PRIMITIVE_I].xyzw[TEMP_PRIMITIVE_C].u[0];*/
255 for (prim_idx
= 0; prim_idx
< num_primitives
; ++prim_idx
) {
256 unsigned num_verts_per_prim
= machine
->Primitives
[0];
257 for (j
= 0; j
< num_verts_per_prim
; j
++) {
258 int idx
= (prim_idx
* num_verts_per_prim
+ j
) *
259 shader
->info
.num_outputs
;
261 debug_printf("%d) Output vert:\n", idx
);
263 for (slot
= 0; slot
< shader
->info
.num_outputs
; slot
++) {
264 output
[slot
][0] = machine
->Outputs
[idx
+ slot
].xyzw
[0].f
[prim_idx
];
265 output
[slot
][1] = machine
->Outputs
[idx
+ slot
].xyzw
[1].f
[prim_idx
];
266 output
[slot
][2] = machine
->Outputs
[idx
+ slot
].xyzw
[2].f
[prim_idx
];
267 output
[slot
][3] = machine
->Outputs
[idx
+ slot
].xyzw
[3].f
[prim_idx
];
269 debug_printf("\t%d: %f %f %f %f\n", slot
,
275 debug_assert(!util_is_inf_or_nan(output
[slot
][0]));
277 output
= (float (*)[4])((char *)output
+ vertex_size
);
282 void draw_geometry_shader_run(struct draw_geometry_shader
*shader
,
283 const float (*input
)[4],
285 const float (*constants
)[4],
287 unsigned input_stride
,
288 unsigned vertex_size
)
290 struct tgsi_exec_machine
*machine
= shader
->machine
;
292 unsigned num_vertices
= num_vertices_for_prim(shader
->input_primitive
);
293 unsigned num_primitives
= count
/num_vertices
;
294 unsigned inputs_from_vs
= 0;
296 machine
->Consts
= constants
;
298 for (i
= 0; i
< shader
->info
.num_inputs
; ++i
) {
299 if (shader
->info
.input_semantic_name
[i
] != TGSI_SEMANTIC_PRIMID
)
303 for (i
= 0; i
< num_primitives
; ++i
) {
304 unsigned int max_primitives
= 1;
306 draw_fetch_geometry_input(shader
, i
, max_primitives
, input
,
307 input_stride
, inputs_from_vs
);
309 tgsi_set_exec_mask(machine
,
315 /* run interpreter */
316 tgsi_exec_machine_run(machine
);
318 draw_geometry_fetch_outputs(shader
, max_primitives
,
319 output
, vertex_size
);
323 void draw_geometry_shader_delete(struct draw_geometry_shader
*shader
)
325 FREE((void*) shader
->state
.tokens
);
329 void draw_geometry_shader_prepare(struct draw_geometry_shader
*shader
,
330 struct draw_context
*draw
)
332 if (shader
->machine
->Tokens
!= shader
->state
.tokens
) {
333 tgsi_exec_machine_bind_shader(shader
->machine
,
334 shader
->state
.tokens
,
335 draw
->gs
.num_samplers
,