src/gallium/auxiliary/draw/draw_pt_fetch_shade_pipeline.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 "util/u_math.h"
  29 #include "util/u_memory.h"
  30 #include "util/u_prim.h"
  31 #include "draw/draw_context.h"
  32 #include "draw/draw_vbuf.h"
  33 #include "draw/draw_vertex.h"
  34 #include "draw/draw_pt.h"
  35 #include "draw/draw_vs.h"
  36 #include "draw/draw_gs.h"
  37
  38
  39 struct fetch_pipeline_middle_end {
  40    struct draw_pt_middle_end base;
  41    struct draw_context *draw;
  42
  43    struct pt_emit *emit;
  44    struct pt_so_emit *so_emit;
  45    struct pt_fetch *fetch;
  46    struct pt_post_vs *post_vs;
  47
  48    unsigned vertex_data_offset;
  49    unsigned vertex_size;
  50    unsigned input_prim;
  51    unsigned opt;
  52 };
  53
  54 static void fetch_pipeline_prepare( struct draw_pt_middle_end *middle,
  55                                     unsigned prim,
  56                                     unsigned opt,
  57                                     unsigned *max_vertices )
  58 {
  59    struct fetch_pipeline_middle_end *fpme = (struct fetch_pipeline_middle_end *)middle;
  60    struct draw_context *draw = fpme->draw;
  61    struct draw_vertex_shader *vs = draw->vs.vertex_shader;
  62    unsigned i;
  63    unsigned instance_id_index = ~0;
  64
  65    unsigned gs_out_prim = (draw->gs.geometry_shader ?
  66                            draw->gs.geometry_shader->output_primitive :
  67                            prim);
  68
  69    /* Add one to num_outputs because the pipeline occasionally tags on
  70     * an additional texcoord, eg for AA lines.
  71     */
  72    unsigned nr = MAX2( vs->info.num_inputs,
  73                        vs->info.num_outputs + 1 );
  74
  75    /* Scan for instanceID system value.
  76     */
  77    for (i = 0; i < vs->info.num_inputs; i++) {
  78       if (vs->info.input_semantic_name[i] == TGSI_SEMANTIC_INSTANCEID) {
  79          instance_id_index = i;
  80          break;
  81       }
  82    }
  83
  84    fpme->input_prim = prim;
  85    fpme->opt = opt;
  86
  87    /* Always leave room for the vertex header whether we need it or
  88     * not.  It's hard to get rid of it in particular because of the
  89     * viewport code in draw_pt_post_vs.c.
  90     */
  91    fpme->vertex_size = sizeof(struct vertex_header) + nr * 4 * sizeof(float);
  92
  93
  94
  95    draw_pt_fetch_prepare( fpme->fetch,
  96                           vs->info.num_inputs,
  97                           fpme->vertex_size,
  98                           instance_id_index );
  99    /* XXX: it's not really gl rasterization rules we care about here,
 100     * but gl vs dx9 clip spaces.
 101     */
 102    draw_pt_post_vs_prepare( fpme->post_vs,
 103                             (boolean)draw->bypass_clipping,
 104                             (boolean)draw->identity_viewport,
 105                             (boolean)draw->rasterizer->gl_rasterization_rules,
 106                             (draw->vs.edgeflag_output ? TRUE : FALSE) );
 107
 108    draw_pt_so_emit_prepare( fpme->so_emit );
 109
 110    if (!(opt & PT_PIPELINE)) {
 111       draw_pt_emit_prepare( fpme->emit,
 112                             gs_out_prim,
 113                             max_vertices );
 114
 115       *max_vertices = MAX2( *max_vertices,
 116                             DRAW_PIPE_MAX_VERTICES );
 117    }
 118    else {
 119       *max_vertices = DRAW_PIPE_MAX_VERTICES;
 120    }
 121
 122    /* return even number */
 123    *max_vertices = *max_vertices & ~1;
 124
 125    /* No need to prepare the shader.
 126     */
 127    vs->prepare(vs, draw);
 128 }
 129
 130
 131 static void fetch( struct pt_fetch *fetch,
 132                    const struct draw_fetch_info *fetch_info,
 133                    char *output)
 134 {
 135    if (fetch_info->linear) {
 136       draw_pt_fetch_run_linear( fetch,
 137                                 fetch_info->start,
 138                                 fetch_info->count,
 139                                 output );
 140    }
 141    else {
 142       draw_pt_fetch_run( fetch,
 143                          fetch_info->elts,
 144                          fetch_info->count,
 145                          output );
 146    }
 147 }
 148
 149
 150 static void pipeline(struct fetch_pipeline_middle_end *fpme,
 151                      const struct draw_vertex_info *vert_info,
 152                      const struct draw_prim_info *prim_info)
 153 {
 154    if (prim_info->linear)
 155       draw_pipeline_run_linear( fpme->draw,
 156                                 vert_info,
 157                                 prim_info);
 158    else
 159       draw_pipeline_run( fpme->draw,
 160                          vert_info,
 161                          prim_info );
 162 }
 163
 164 static void emit(struct pt_emit *emit,
 165                  const struct draw_vertex_info *vert_info,
 166                  const struct draw_prim_info *prim_info)
 167 {
 168    if (prim_info->linear) {
 169       draw_pt_emit_linear(emit, vert_info, prim_info);
 170    }
 171    else {
 172       draw_pt_emit(emit, vert_info, prim_info);
 173    }
 174 }
 175
 176
 177 static void draw_vertex_shader_run(struct draw_vertex_shader *vshader,
 178                                    const void *constants[PIPE_MAX_CONSTANT_BUFFERS],
 179                                    const struct draw_vertex_info *input_verts,
 180                                    struct draw_vertex_info *output_verts )
 181 {
 182    output_verts->vertex_size = input_verts->vertex_size;
 183    output_verts->stride = input_verts->vertex_size;
 184    output_verts->count = input_verts->count;
 185    output_verts->verts =
 186       (struct vertex_header *)MALLOC(output_verts->vertex_size *
 187                                      align(output_verts->count, 4));
 188
 189    vshader->run_linear(vshader,
 190                        (const float (*)[4])input_verts->verts->data,
 191                        (      float (*)[4])output_verts->verts->data,
 192                        constants,
 193                        input_verts->count,
 194                        input_verts->vertex_size,
 195                        input_verts->vertex_size);
 196 }
 197
 198 static void fetch_pipeline_generic( struct draw_pt_middle_end *middle,
 199                                     const struct draw_fetch_info *fetch_info,
 200                                     const struct draw_prim_info *prim_info )
 201 {
 202    struct fetch_pipeline_middle_end *fpme = (struct fetch_pipeline_middle_end *)middle;
 203    struct draw_context *draw = fpme->draw;
 204    struct draw_vertex_shader *vshader = draw->vs.vertex_shader;
 205    struct draw_geometry_shader *gshader = draw->gs.geometry_shader;
 206    struct draw_prim_info gs_prim_info;
 207    struct draw_vertex_info fetched_vert_info;
 208    struct draw_vertex_info vs_vert_info;
 209    struct draw_vertex_info gs_vert_info;
 210    struct draw_vertex_info *vert_info;
 211    unsigned opt = fpme->opt;
 212
 213    fetched_vert_info.count = fetch_info->count;
 214    fetched_vert_info.vertex_size = fpme->vertex_size;
 215    fetched_vert_info.stride = fpme->vertex_size;
 216    fetched_vert_info.verts =
 217       (struct vertex_header *)MALLOC(fpme->vertex_size *
 218                                      align(fetch_info->count,  4));
 219    if (!fetched_vert_info.verts) {
 220       assert(0);
 221       return;
 222    }
 223
 224    /* Fetch into our vertex buffer.
 225     */
 226    fetch( fpme->fetch, fetch_info, (char *)fetched_vert_info.verts );
 227
 228    /* Finished with fetch:
 229     */
 230    fetch_info = NULL;
 231    vert_info = &fetched_vert_info;
 232
 233    /* Run the shader, note that this overwrites the data[] parts of
 234     * the pipeline verts.
 235     */
 236    if (fpme->opt & PT_SHADE) {
 237       draw_vertex_shader_run(vshader,
 238                              draw->pt.user.vs_constants,
 239                              vert_info,
 240                              &vs_vert_info);
 241
 242       FREE(vert_info->verts);
 243       vert_info = &vs_vert_info;
 244    }
 245
 246    if ((fpme->opt & PT_SHADE) && gshader) {
 247       draw_geometry_shader_run(gshader,
 248                                draw->pt.user.gs_constants,
 249                                vert_info,
 250                                prim_info,
 251                                &gs_vert_info,
 252                                &gs_prim_info);
 253
 254       FREE(vert_info->verts);
 255       vert_info = &gs_vert_info;
 256       prim_info = &gs_prim_info;
 257    }
 258
 259
 260    /* Stream output needs to be done before clipping.
 261     *
 262     * XXX: Stream output surely needs to respect the prim_info->elt
 263     *      lists.
 264     */
 265    draw_pt_so_emit( fpme->so_emit,
 266                     vert_info,
 267                     prim_info );
 268
 269    if (draw_pt_post_vs_run( fpme->post_vs,
 270                             vert_info ))
 271    {
 272       opt |= PT_PIPELINE;
 273    }
 274
 275    /* Do we need to run the pipeline?
 276     */
 277    if (opt & PT_PIPELINE) {
 278       pipeline( fpme,
 279                 vert_info,
 280                 prim_info );
 281    }
 282    else {
 283       emit( fpme->emit,
 284             vert_info,
 285             prim_info );
 286    }
 287    FREE(vert_info->verts);
 288 }
 289
 290 static void fetch_pipeline_run( struct draw_pt_middle_end *middle,
 291                                 const unsigned *fetch_elts,
 292                                 unsigned fetch_count,
 293                                 const ushort *draw_elts,
 294                                 unsigned draw_count )
 295 {
 296    struct fetch_pipeline_middle_end *fpme = (struct fetch_pipeline_middle_end *)middle;
 297    struct draw_fetch_info fetch_info;
 298    struct draw_prim_info prim_info;
 299
 300    fetch_info.linear = FALSE;
 301    fetch_info.start = 0;
 302    fetch_info.elts = fetch_elts;
 303    fetch_info.count = fetch_count;
 304
 305    prim_info.linear = FALSE;
 306    prim_info.start = 0;
 307    prim_info.count = draw_count;
 308    prim_info.elts = draw_elts;
 309    prim_info.prim = fpme->input_prim;
 310    prim_info.primitive_count = 1;
 311    prim_info.primitive_lengths = &draw_count;
 312
 313    fetch_pipeline_generic( middle, &fetch_info, &prim_info );
 314 }
 315
 316
 317 static void fetch_pipeline_linear_run( struct draw_pt_middle_end *middle,
 318                                        unsigned start,
 319                                        unsigned count)
 320 {
 321    struct fetch_pipeline_middle_end *fpme = (struct fetch_pipeline_middle_end *)middle;
 322    struct draw_fetch_info fetch_info;
 323    struct draw_prim_info prim_info;
 324
 325    fetch_info.linear = TRUE;
 326    fetch_info.start = start;
 327    fetch_info.count = count;
 328    fetch_info.elts = NULL;
 329
 330    prim_info.linear = TRUE;
 331    prim_info.start = 0;
 332    prim_info.count = count;
 333    prim_info.elts = NULL;
 334    prim_info.prim = fpme->input_prim;
 335    prim_info.primitive_count = 1;
 336    prim_info.primitive_lengths = &count;
 337
 338    fetch_pipeline_generic( middle, &fetch_info, &prim_info );
 339 }
 340
 341
 342
 343 static boolean fetch_pipeline_linear_run_elts( struct draw_pt_middle_end *middle,
 344                                                unsigned start,
 345                                                unsigned count,
 346                                                const ushort *draw_elts,
 347                                                unsigned draw_count )
 348 {
 349    struct fetch_pipeline_middle_end *fpme = (struct fetch_pipeline_middle_end *)middle;
 350    struct draw_fetch_info fetch_info;
 351    struct draw_prim_info prim_info;
 352
 353    fetch_info.linear = TRUE;
 354    fetch_info.start = start;
 355    fetch_info.count = count;
 356    fetch_info.elts = NULL;
 357
 358    prim_info.linear = FALSE;
 359    prim_info.start = 0;
 360    prim_info.count = draw_count;
 361    prim_info.elts = draw_elts;
 362    prim_info.prim = fpme->input_prim;
 363    prim_info.primitive_count = 1;
 364    prim_info.primitive_lengths = &draw_count;
 365
 366    fetch_pipeline_generic( middle, &fetch_info, &prim_info );
 367
 368    return TRUE;
 369 }
 370
 371
 372
 373 static void fetch_pipeline_finish( struct draw_pt_middle_end *middle )
 374 {
 375    /* nothing to do */
 376 }
 377
 378 static void fetch_pipeline_destroy( struct draw_pt_middle_end *middle )
 379 {
 380    struct fetch_pipeline_middle_end *fpme = (struct fetch_pipeline_middle_end *)middle;
 381
 382    if (fpme->fetch)
 383       draw_pt_fetch_destroy( fpme->fetch );
 384
 385    if (fpme->emit)
 386       draw_pt_emit_destroy( fpme->emit );
 387
 388    if (fpme->so_emit)
 389       draw_pt_so_emit_destroy( fpme->so_emit );
 390
 391    if (fpme->post_vs)
 392       draw_pt_post_vs_destroy( fpme->post_vs );
 393
 394    FREE(middle);
 395 }
 396
 397
 398 struct draw_pt_middle_end *draw_pt_fetch_pipeline_or_emit( struct draw_context *draw )
 399 {
 400    struct fetch_pipeline_middle_end *fpme = CALLOC_STRUCT( fetch_pipeline_middle_end );
 401    if (!fpme)
 402       goto fail;
 403
 404    fpme->base.prepare        = fetch_pipeline_prepare;
 405    fpme->base.run            = fetch_pipeline_run;
 406    fpme->base.run_linear     = fetch_pipeline_linear_run;
 407    fpme->base.run_linear_elts = fetch_pipeline_linear_run_elts;
 408    fpme->base.finish         = fetch_pipeline_finish;
 409    fpme->base.destroy        = fetch_pipeline_destroy;
 410
 411    fpme->draw = draw;
 412
 413    fpme->fetch = draw_pt_fetch_create( draw );
 414    if (!fpme->fetch)
 415       goto fail;
 416
 417    fpme->post_vs = draw_pt_post_vs_create( draw );
 418    if (!fpme->post_vs)
 419       goto fail;
 420
 421    fpme->emit = draw_pt_emit_create( draw );
 422    if (!fpme->emit)
 423       goto fail;
 424
 425    fpme->so_emit = draw_pt_so_emit_create( draw );
 426    if (!fpme->so_emit)
 427       goto fail;
 428
 429    return &fpme->base;
 430
 431  fail:
 432    if (fpme)
 433       fetch_pipeline_destroy( &fpme->base );
 434
 435    return NULL;
 436 }