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