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#ifndef DRAW_GS_H
29#define DRAW_GS_H
30
31#include "draw_context.h"
32#include "tgsi/tgsi_exec.h"
33#include "draw_private.h"
34
35#define MAX_TGSI_PRIMITIVES 4
36
37struct draw_context;
38
39#ifdef HAVE_LLVM
40struct draw_gs_jit_context;
41struct draw_gs_llvm_variant;
42
43/**
44 * Structure holding the inputs to the geometry shader. It uses SOA layout.
45 * The dimensions are as follows:
46 * - maximum number of vertices for a geometry shader input primitive
47 *   (6 for triangle_adjacency)
48 * - maximum number of attributes for each vertex
49 * - four channels per each attribute (x,y,z,w)
50 * - number of input primitives equal to the SOA vector length
51 */
52struct draw_gs_inputs {
53   float data[6][PIPE_MAX_SHADER_INPUTS][TGSI_NUM_CHANNELS][TGSI_NUM_CHANNELS];
54};
55#endif
56
57/**
58 * Private version of the compiled geometry shader
59 */
60struct draw_vertex_stream {
61   unsigned *primitive_lengths;
62   unsigned emitted_vertices;
63   unsigned emitted_primitives;
64   float (*tmp_output)[4];
65};
66
67struct draw_geometry_shader {
68   struct draw_context *draw;
69
70   struct tgsi_exec_machine *machine;
71
72   /* This member will disappear shortly:*/
73   struct pipe_shader_state state;
74
75   struct tgsi_shader_info info;
76   unsigned position_output;
77   unsigned viewport_index_output;
78   unsigned ccdistance_output[PIPE_MAX_CLIP_OR_CULL_DISTANCE_ELEMENT_COUNT];
79
80   unsigned max_output_vertices;
81   unsigned primitive_boundary;
82   unsigned input_primitive;
83   unsigned output_primitive;
84   unsigned vertex_size;
85
86   struct draw_vertex_stream stream[TGSI_MAX_VERTEX_STREAMS];
87   unsigned num_vertex_streams;
88
89   unsigned in_prim_idx;
90   unsigned input_vertex_stride;
91   unsigned fetched_prim_count;
92   const float (*input)[4];
93   const struct tgsi_shader_info *input_info;
94   unsigned vector_length;
95   unsigned max_out_prims;
96
97   unsigned num_invocations;
98   unsigned invocation_id;
99#ifdef HAVE_LLVM
100   struct draw_gs_inputs *gs_input;
101   struct draw_gs_jit_context *jit_context;
102   struct draw_gs_llvm_variant *current_variant;
103   struct vertex_header *gs_output;
104
105   int **llvm_prim_lengths;
106   int *llvm_emitted_primitives;
107   int *llvm_emitted_vertices;
108   int *llvm_prim_ids;
109#endif
110
111   void (*fetch_inputs)(struct draw_geometry_shader *shader,
112                        unsigned *indices,
113                        unsigned num_vertices,
114                        unsigned prim_idx);
115   void (*fetch_outputs)(struct draw_geometry_shader *shader,
116                         unsigned vertex_stream,
117                         unsigned num_primitives,
118                         float (**p_output)[4]);
119
120   void     (*prepare)(struct draw_geometry_shader *shader,
121                       const void *constants[PIPE_MAX_CONSTANT_BUFFERS],
122                       const unsigned constants_size[PIPE_MAX_CONSTANT_BUFFERS]);
123   void (*run)(struct draw_geometry_shader *shader,
124               unsigned input_primitives, unsigned *out_prims);
125};
126
127void draw_geometry_shader_new_instance(struct draw_geometry_shader *gs);
128
129/*
130 * Returns the number of vertices emitted.
131 * The vertex shader can emit any number of vertices as long as it's
132 * smaller than the GS_MAX_OUTPUT_VERTICES shader property.
133 */
134int draw_geometry_shader_run(struct draw_geometry_shader *shader,
135                             const void *constants[PIPE_MAX_CONSTANT_BUFFERS],
136                             const unsigned constants_size[PIPE_MAX_CONSTANT_BUFFERS],
137                             const struct draw_vertex_info *input_verts,
138                             const struct draw_prim_info *input_prim,
139                             const struct tgsi_shader_info *input_info,
140                             struct draw_vertex_info *output_verts,
141                             struct draw_prim_info *output_prims );
142
143void draw_geometry_shader_prepare(struct draw_geometry_shader *shader,
144                                  struct draw_context *draw);
145
146int draw_gs_max_output_vertices(struct draw_geometry_shader *shader,
147                                unsigned pipe_prim);
148
149#ifdef HAVE_LLVM
150void draw_gs_set_current_variant(struct draw_geometry_shader *shader,
151                                 struct draw_gs_llvm_variant *variant);
152#endif
153
154#endif
155