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