src/mesa/state_tracker/st_draw_feedback.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2007 Tungsten Graphics, Inc., Cedar Park, Texas.
   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 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.
  25  *
  26  **************************************************************************/
  27
  28 #include "main/imports.h"
  29 #include "main/image.h"
  30 #include "main/macros.h"
  31 #include "main/mfeatures.h"
  32
  33 #include "vbo/vbo.h"
  34
  35 #include "st_context.h"
  36 #include "st_atom.h"
  37 #include "st_cb_bufferobjects.h"
  38 #include "st_draw.h"
  39 #include "st_program.h"
  40
  41 #include "pipe/p_context.h"
  42 #include "pipe/p_defines.h"
  43 #include "util/u_inlines.h"
  44
  45 #include "draw/draw_private.h"
  46 #include "draw/draw_context.h"
  47
  48
  49 #if FEATURE_feedback || FEATURE_rastpos
  50
  51 /**
  52  * Set the (private) draw module's post-transformed vertex format when in
  53  * GL_SELECT or GL_FEEDBACK mode or for glRasterPos.
  54  */
  55 static void
  56 set_feedback_vertex_format(struct gl_context *ctx)
  57 {
  58 #if 0
  59    struct st_context *st = st_context(ctx);
  60    struct vertex_info vinfo;
  61    GLuint i;
  62
  63    memset(&vinfo, 0, sizeof(vinfo));
  64
  65    if (ctx->RenderMode == GL_SELECT) {
  66       assert(ctx->RenderMode == GL_SELECT);
  67       vinfo.num_attribs = 1;
  68       vinfo.format[0] = FORMAT_4F;
  69       vinfo.interp_mode[0] = INTERP_LINEAR;
  70    }
  71    else {
  72       /* GL_FEEDBACK, or glRasterPos */
  73       /* emit all attribs (pos, color, texcoord) as GLfloat[4] */
  74       vinfo.num_attribs = st->state.vs->cso->state.num_outputs;
  75       for (i = 0; i < vinfo.num_attribs; i++) {
  76          vinfo.format[i] = FORMAT_4F;
  77          vinfo.interp_mode[i] = INTERP_LINEAR;
  78       }
  79    }
  80
  81    draw_set_vertex_info(st->draw, &vinfo);
  82 #endif
  83 }
  84
  85
  86 /**
  87  * Called by VBO to draw arrays when in selection or feedback mode and
  88  * to implement glRasterPos.
  89  * This is very much like the normal draw_vbo() function above.
  90  * Look at code refactoring some day.
  91  */
  92 void
  93 st_feedback_draw_vbo(struct gl_context *ctx,
  94                      const struct gl_client_array **arrays,
  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 {
 103    struct st_context *st = st_context(ctx);
 104    struct pipe_context *pipe = st->pipe;
 105    struct draw_context *draw = st->draw;
 106    const struct st_vertex_program *vp;
 107    const struct pipe_shader_state *vs;
 108    struct pipe_vertex_buffer vbuffers[PIPE_MAX_SHADER_INPUTS];
 109    struct pipe_vertex_element velements[PIPE_MAX_ATTRIBS];
 110    struct pipe_index_buffer ibuffer;
 111    struct pipe_transfer *vb_transfer[PIPE_MAX_ATTRIBS];
 112    struct pipe_transfer *ib_transfer = NULL;
 113    GLuint attr, i;
 114    const GLubyte *low_addr = NULL;
 115    const void *mapped_indices = NULL;
 116
 117    assert(draw);
 118
 119    st_validate_state(st);
 120
 121    if (!index_bounds_valid)
 122       vbo_get_minmax_index(ctx, prims, ib, &min_index, &max_index);
 123
 124    /* must get these after state validation! */
 125    vp = st->vp;
 126    vs = &st->vp_variant->tgsi;
 127
 128    if (!st->vp_variant->draw_shader) {
 129       st->vp_variant->draw_shader = draw_create_vertex_shader(draw, vs);
 130    }
 131
 132    /*
 133     * Set up the draw module's state.
 134     *
 135     * We'd like to do this less frequently, but the normal state-update
 136     * code sends state updates to the pipe, not to our private draw module.
 137     */
 138    assert(draw);
 139    draw_set_viewport_state(draw, &st->state.viewport);
 140    draw_set_clip_state(draw, &st->state.clip);
 141    draw_set_rasterizer_state(draw, &st->state.rasterizer, NULL);
 142    draw_bind_vertex_shader(draw, st->vp_variant->draw_shader);
 143    set_feedback_vertex_format(ctx);
 144
 145    /* Find the lowest address of the arrays we're drawing */
 146    if (vp->num_inputs) {
 147       low_addr = arrays[vp->index_to_input[0]]->Ptr;
 148
 149       for (attr = 1; attr < vp->num_inputs; attr++) {
 150          const GLubyte *start = arrays[vp->index_to_input[attr]]->Ptr;
 151          low_addr = MIN2(low_addr, start);
 152       }
 153    }
 154
 155    /* loop over TGSI shader inputs to determine vertex buffer
 156     * and attribute info
 157     */
 158    for (attr = 0; attr < vp->num_inputs; attr++) {
 159       const GLuint mesaAttr = vp->index_to_input[attr];
 160       struct gl_buffer_object *bufobj = arrays[mesaAttr]->BufferObj;
 161       void *map;
 162
 163       if (bufobj && bufobj->Name) {
 164          /* Attribute data is in a VBO.
 165           * Recall that for VBOs, the gl_client_array->Ptr field is
 166           * really an offset from the start of the VBO, not a pointer.
 167           */
 168          struct st_buffer_object *stobj = st_buffer_object(bufobj);
 169          assert(stobj->buffer);
 170
 171          vbuffers[attr].buffer = NULL;
 172          pipe_resource_reference(&vbuffers[attr].buffer, stobj->buffer);
 173          vbuffers[attr].buffer_offset = pointer_to_offset(low_addr);
 174          velements[attr].src_offset = arrays[mesaAttr]->Ptr - low_addr;
 175       }
 176       else {
 177          /* attribute data is in user-space memory, not a VBO */
 178          uint bytes = (arrays[mesaAttr]->Size
 179                        * _mesa_sizeof_type(arrays[mesaAttr]->Type)
 180                        * (max_index + 1));
 181
 182          /* wrap user data */
 183          vbuffers[attr].buffer
 184             = pipe_user_buffer_create(pipe->screen, (void *) arrays[mesaAttr]->Ptr,
 185                                       bytes,
 186                                       PIPE_BIND_VERTEX_BUFFER);
 187          vbuffers[attr].buffer_offset = 0;
 188          velements[attr].src_offset = 0;
 189       }
 190
 191       /* common-case setup */
 192       vbuffers[attr].stride = arrays[mesaAttr]->StrideB; /* in bytes */
 193       velements[attr].instance_divisor = 0;
 194       velements[attr].vertex_buffer_index = attr;
 195       velements[attr].src_format =
 196          st_pipe_vertex_format(arrays[mesaAttr]->Type,
 197                                arrays[mesaAttr]->Size,
 198                                arrays[mesaAttr]->Format,
 199                                arrays[mesaAttr]->Normalized);
 200       assert(velements[attr].src_format);
 201
 202       /* tell draw about this attribute */
 203 #if 0
 204       draw_set_vertex_buffer(draw, attr, &vbuffer[attr]);
 205 #endif
 206
 207       /* map the attrib buffer */
 208       map = pipe_buffer_map(pipe, vbuffers[attr].buffer,
 209                             PIPE_TRANSFER_READ,
 210                             &vb_transfer[attr]);
 211       draw_set_mapped_vertex_buffer(draw, attr, map);
 212    }
 213
 214    draw_set_vertex_buffers(draw, vp->num_inputs, vbuffers);
 215    draw_set_vertex_elements(draw, vp->num_inputs, velements);
 216
 217    memset(&ibuffer, 0, sizeof(ibuffer));
 218    if (ib) {
 219       struct gl_buffer_object *bufobj = ib->obj;
 220
 221       switch (ib->type) {
 222       case GL_UNSIGNED_INT:
 223          ibuffer.index_size = 4;
 224          break;
 225       case GL_UNSIGNED_SHORT:
 226          ibuffer.index_size = 2;
 227          break;
 228       case GL_UNSIGNED_BYTE:
 229          ibuffer.index_size = 1;
 230          break;
 231       default:
 232          assert(0);
 233          goto out_unref_vertex;
 234       }
 235
 236       if (bufobj && bufobj->Name) {
 237          struct st_buffer_object *stobj = st_buffer_object(bufobj);
 238
 239          pipe_resource_reference(&ibuffer.buffer, stobj->buffer);
 240          ibuffer.offset = pointer_to_offset(ib->ptr);
 241
 242          mapped_indices = pipe_buffer_map(pipe, stobj->buffer,
 243                                           PIPE_TRANSFER_READ, &ib_transfer);
 244       }
 245       else {
 246          /* skip setting ibuffer.buffer as the draw module does not use it */
 247          mapped_indices = ib->ptr;
 248       }
 249
 250       draw_set_index_buffer(draw, &ibuffer);
 251       draw_set_mapped_index_buffer(draw, mapped_indices);
 252    }
 253
 254    /* set the constant buffer */
 255    draw_set_mapped_constant_buffer(st->draw, PIPE_SHADER_VERTEX, 0,
 256                                    st->state.constants[PIPE_SHADER_VERTEX].ptr,
 257                                    st->state.constants[PIPE_SHADER_VERTEX].size);
 258
 259
 260    /* draw here */
 261    for (i = 0; i < nr_prims; i++) {
 262       draw_arrays(draw, prims[i].mode, prims[i].start, prims[i].count);
 263    }
 264
 265
 266    /*
 267     * unmap vertex/index buffers
 268     */
 269    if (ib) {
 270       draw_set_mapped_index_buffer(draw, NULL);
 271       draw_set_index_buffer(draw, NULL);
 272
 273       if (ib_transfer)
 274          pipe_buffer_unmap(pipe, ib_transfer);
 275       pipe_resource_reference(&ibuffer.buffer, NULL);
 276    }
 277
 278  out_unref_vertex:
 279    for (attr = 0; attr < vp->num_inputs; attr++) {
 280       pipe_buffer_unmap(pipe, vb_transfer[attr]);
 281       draw_set_mapped_vertex_buffer(draw, attr, NULL);
 282       pipe_resource_reference(&vbuffers[attr].buffer, NULL);
 283    }
 284    draw_set_vertex_buffers(draw, 0, NULL);
 285 }
 286
 287 #endif /* FEATURE_feedback || FEATURE_rastpos */
 288