nir: Move nir_lower_uniforms_to_ubo to compiler/nir.
[mesa.git] / src / mesa / state_tracker / st_draw_feedback.c
1 /**************************************************************************
2 *
3 * Copyright 2007 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 #include "main/imports.h"
29 #include "main/arrayobj.h"
30 #include "main/image.h"
31 #include "main/macros.h"
32 #include "main/varray.h"
33
34 #include "vbo/vbo.h"
35
36 #include "st_context.h"
37 #include "st_atom.h"
38 #include "st_cb_bitmap.h"
39 #include "st_cb_bufferobjects.h"
40 #include "st_draw.h"
41 #include "st_program.h"
42
43 #include "pipe/p_context.h"
44 #include "pipe/p_defines.h"
45 #include "util/u_inlines.h"
46 #include "util/u_draw.h"
47
48 #include "draw/draw_private.h"
49 #include "draw/draw_context.h"
50
51
52 /**
53 * Set the (private) draw module's post-transformed vertex format when in
54 * GL_SELECT or GL_FEEDBACK mode or for glRasterPos.
55 */
56 static void
57 set_feedback_vertex_format(struct gl_context *ctx)
58 {
59 #if 0
60 struct st_context *st = st_context(ctx);
61 struct vertex_info vinfo;
62 GLuint i;
63
64 memset(&vinfo, 0, sizeof(vinfo));
65
66 if (ctx->RenderMode == GL_SELECT) {
67 assert(ctx->RenderMode == GL_SELECT);
68 vinfo.num_attribs = 1;
69 vinfo.format[0] = FORMAT_4F;
70 vinfo.interp_mode[0] = INTERP_LINEAR;
71 }
72 else {
73 /* GL_FEEDBACK, or glRasterPos */
74 /* emit all attribs (pos, color, texcoord) as GLfloat[4] */
75 vinfo.num_attribs = st->state.vs->cso->state.num_outputs;
76 for (i = 0; i < vinfo.num_attribs; i++) {
77 vinfo.format[i] = FORMAT_4F;
78 vinfo.interp_mode[i] = INTERP_LINEAR;
79 }
80 }
81
82 draw_set_vertex_info(st->draw, &vinfo);
83 #endif
84 }
85
86
87 /**
88 * Called by VBO to draw arrays when in selection or feedback mode and
89 * to implement glRasterPos.
90 * This function mirrors the normal st_draw_vbo().
91 * Look at code refactoring some day.
92 */
93 void
94 st_feedback_draw_vbo(struct gl_context *ctx,
95 const struct _mesa_prim *prims,
96 GLuint nr_prims,
97 const struct _mesa_index_buffer *ib,
98 GLboolean index_bounds_valid,
99 GLuint min_index,
100 GLuint max_index,
101 struct gl_transform_feedback_object *tfb_vertcount,
102 unsigned stream,
103 struct gl_buffer_object *indirect)
104 {
105 struct st_context *st = st_context(ctx);
106 struct pipe_context *pipe = st->pipe;
107 struct draw_context *draw = st_get_draw_context(st);
108 const struct st_vertex_program *vp;
109 struct st_vp_variant *vp_variant;
110 const struct pipe_shader_state *vs;
111 struct pipe_vertex_buffer vbuffers[PIPE_MAX_SHADER_INPUTS];
112 unsigned num_vbuffers = 0;
113 struct pipe_vertex_element velements[PIPE_MAX_ATTRIBS];
114 struct pipe_transfer *vb_transfer[PIPE_MAX_ATTRIBS] = {NULL};
115 struct pipe_transfer *ib_transfer = NULL;
116 GLuint i;
117 const void *mapped_indices = NULL;
118 struct pipe_draw_info info;
119
120 if (!draw)
121 return;
122
123 /* Initialize pipe_draw_info. */
124 info.primitive_restart = false;
125 info.vertices_per_patch = ctx->TessCtrlProgram.patch_vertices;
126 info.indirect = NULL;
127 info.count_from_stream_output = NULL;
128 info.restart_index = 0;
129
130 st_flush_bitmap_cache(st);
131 st_invalidate_readpix_cache(st);
132
133 st_validate_state(st, ST_PIPELINE_RENDER);
134
135 if (!index_bounds_valid)
136 vbo_get_minmax_indices(ctx, prims, ib, &min_index, &max_index, nr_prims);
137
138 /* must get these after state validation! */
139 vp = st->vp;
140 vp_variant = st->vp_variant;
141 vs = &vp_variant->tgsi;
142
143 if (!vp_variant->draw_shader) {
144 vp_variant->draw_shader = draw_create_vertex_shader(draw, vs);
145 }
146
147 /*
148 * Set up the draw module's state.
149 *
150 * We'd like to do this less frequently, but the normal state-update
151 * code sends state updates to the pipe, not to our private draw module.
152 */
153 assert(draw);
154 draw_set_viewport_states(draw, 0, 1, &st->state.viewport[0]);
155 draw_set_clip_state(draw, &st->state.clip);
156 draw_set_rasterizer_state(draw, &st->state.rasterizer, NULL);
157 draw_bind_vertex_shader(draw, vp_variant->draw_shader);
158 set_feedback_vertex_format(ctx);
159
160 /* Must setup these after state validation! */
161 /* Setup arrays */
162 st_setup_arrays(st, vp, vp_variant, velements, vbuffers, &num_vbuffers);
163 /* Setup current values as userspace arrays */
164 st_setup_current_user(st, vp, vp_variant, velements, vbuffers, &num_vbuffers);
165
166 /* Map all buffers and tell draw about their mapping */
167 for (unsigned buf = 0; buf < num_vbuffers; ++buf) {
168 struct pipe_vertex_buffer *vbuffer = &vbuffers[buf];
169
170 if (vbuffer->is_user_buffer) {
171 draw_set_mapped_vertex_buffer(draw, buf, vbuffer->buffer.user, ~0);
172 } else {
173 void *map = pipe_buffer_map(pipe, vbuffer->buffer.resource,
174 PIPE_TRANSFER_READ, &vb_transfer[buf]);
175 draw_set_mapped_vertex_buffer(draw, buf, map,
176 vbuffer->buffer.resource->width0);
177 }
178 }
179
180 draw_set_vertex_buffers(draw, 0, num_vbuffers, vbuffers);
181 draw_set_vertex_elements(draw, vp->num_inputs, velements);
182
183 unsigned start = 0;
184
185 if (ib) {
186 struct gl_buffer_object *bufobj = ib->obj;
187 unsigned index_size = ib->index_size;
188
189 if (index_size == 0)
190 goto out_unref_vertex;
191
192 if (bufobj && bufobj->Name) {
193 struct st_buffer_object *stobj = st_buffer_object(bufobj);
194
195 start = pointer_to_offset(ib->ptr) / index_size;
196 mapped_indices = pipe_buffer_map(pipe, stobj->buffer,
197 PIPE_TRANSFER_READ, &ib_transfer);
198 }
199 else {
200 mapped_indices = ib->ptr;
201 }
202
203 info.index_size = ib->index_size;
204 info.min_index = min_index;
205 info.max_index = max_index;
206 info.has_user_indices = true;
207 info.index.user = mapped_indices;
208
209 draw_set_indexes(draw,
210 (ubyte *) mapped_indices,
211 index_size, ~0);
212
213 if (ctx->Array._PrimitiveRestart) {
214 info.primitive_restart = true;
215 info.restart_index = _mesa_primitive_restart_index(ctx, info.index_size);
216 }
217 } else {
218 info.index_size = 0;
219 info.has_user_indices = false;
220 }
221
222 /* set the constant buffer */
223 draw_set_mapped_constant_buffer(st->draw, PIPE_SHADER_VERTEX, 0,
224 st->state.constants[PIPE_SHADER_VERTEX].ptr,
225 st->state.constants[PIPE_SHADER_VERTEX].size);
226
227
228 /* draw here */
229 for (i = 0; i < nr_prims; i++) {
230 info.count = prims[i].count;
231
232 if (!info.count)
233 continue;
234
235 info.mode = prims[i].mode;
236 info.start = start + prims[i].start;
237 info.start_instance = prims[i].base_instance;
238 info.instance_count = prims[i].num_instances;
239 info.index_bias = prims[i].basevertex;
240 info.drawid = prims[i].draw_id;
241 if (!ib) {
242 info.min_index = info.start;
243 info.max_index = info.start + info.count - 1;
244 }
245
246 draw_vbo(draw, &info);
247 }
248
249
250 /*
251 * unmap vertex/index buffers
252 */
253 if (ib) {
254 draw_set_indexes(draw, NULL, 0, 0);
255 if (ib_transfer)
256 pipe_buffer_unmap(pipe, ib_transfer);
257 }
258
259 out_unref_vertex:
260 for (unsigned buf = 0; buf < num_vbuffers; ++buf) {
261 if (vb_transfer[buf])
262 pipe_buffer_unmap(pipe, vb_transfer[buf]);
263 draw_set_mapped_vertex_buffer(draw, buf, NULL, 0);
264 }
265 draw_set_vertex_buffers(draw, 0, num_vbuffers, NULL);
266 }