src/gallium/auxiliary/util/u_blitter.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2009 Marek Olšák <maraeo@gmail.com>
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the
   7  * "Software"), to deal in the Software without restriction, including
   8  * without limitation the rights to use, copy, modify, merge, publish,
   9  * distribute, sub license, and/or sell copies of the Software, and to
  10  * permit persons to whom the Software is furnished to do so, subject to
  11  * the following conditions:
  12  *
  13  * The above copyright notice and this permission notice (including the
  14  * next paragraph) shall be included in all copies or substantial portions
  15  * of the Software.
  16  *
  17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  18  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  19  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  20  * IN NO EVENT SHALL TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR
  21  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  22  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  23  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  24  *
  25  **************************************************************************/
  26
  27 /**
  28  * @file
  29  * Blitter utility to facilitate acceleration of the clear, surface_copy,
  30  * and surface_fill functions.
  31  *
  32  * @author Marek Olšák
  33  */
  34
  35 #include "pipe/p_context.h"
  36 #include "pipe/p_defines.h"
  37 #include "util/u_inlines.h"
  38 #include "pipe/p_shader_tokens.h"
  39 #include "pipe/p_state.h"
  40
  41 #include "util/u_format.h"
  42 #include "util/u_memory.h"
  43 #include "util/u_math.h"
  44 #include "util/u_blitter.h"
  45 #include "util/u_draw_quad.h"
  46 #include "util/u_pack_color.h"
  47 #include "util/u_rect.h"
  48 #include "util/u_simple_shaders.h"
  49 #include "util/u_texture.h"
  50
  51 #define INVALID_PTR ((void*)~0)
  52
  53 struct blitter_context_priv
  54 {
  55    struct blitter_context blitter;
  56
  57    struct pipe_context *pipe; /**< pipe context */
  58    struct pipe_buffer *vbuf;  /**< quad */
  59
  60    float vertices[4][2][4];   /**< {pos, color} or {pos, texcoord} */
  61
  62    /* Templates for various state objects. */
  63    struct pipe_sampler_state template_sampler_state;
  64
  65    /* Constant state objects. */
  66    /* Vertex shaders. */
  67    void *vs_col; /**< Vertex shader which passes {pos, color} to the output */
  68    void *vs_tex; /**< Vertex shader which passes {pos, texcoord} to the output.*/
  69
  70    /* Fragment shaders. */
  71    /* FS which outputs a color to multiple color buffers. */
  72    void *fs_col[PIPE_MAX_COLOR_BUFS];
  73
  74    /* FS which outputs a color from a texture,
  75       where the index is PIPE_TEXTURE_* to be sampled. */
  76    void *fs_texfetch_col[PIPE_MAX_TEXTURE_TYPES];
  77
  78    /* FS which outputs a depth from a texture,
  79       where the index is PIPE_TEXTURE_* to be sampled. */
  80    void *fs_texfetch_depth[PIPE_MAX_TEXTURE_TYPES];
  81
  82    /* Blend state. */
  83    void *blend_write_color;   /**< blend state with writemask of RGBA */
  84    void *blend_keep_color;    /**< blend state with writemask of 0 */
  85
  86    /* Depth stencil alpha state. */
  87    void *dsa_write_depth_stencil;
  88    void *dsa_write_depth_keep_stencil;
  89    void *dsa_keep_depth_stencil;
  90
  91    /* Sampler state for clamping to a miplevel. */
  92    void *sampler_state[PIPE_MAX_TEXTURE_LEVELS];
  93
  94    /* Rasterizer state. */
  95    void *rs_state;
  96 };
  97
  98 struct blitter_context *util_blitter_create(struct pipe_context *pipe)
  99 {
 100    struct blitter_context_priv *ctx;
 101    struct pipe_blend_state blend = { 0 };
 102    struct pipe_depth_stencil_alpha_state dsa = { { 0 } };
 103    struct pipe_rasterizer_state rs_state = { 0 };
 104    struct pipe_sampler_state *sampler_state;
 105    unsigned i;
 106
 107    ctx = CALLOC_STRUCT(blitter_context_priv);
 108    if (!ctx)
 109       return NULL;
 110
 111    ctx->pipe = pipe;
 112
 113    /* init state objects for them to be considered invalid */
 114    ctx->blitter.saved_blend_state = INVALID_PTR;
 115    ctx->blitter.saved_dsa_state = INVALID_PTR;
 116    ctx->blitter.saved_rs_state = INVALID_PTR;
 117    ctx->blitter.saved_fs = INVALID_PTR;
 118    ctx->blitter.saved_vs = INVALID_PTR;
 119    ctx->blitter.saved_fb_state.nr_cbufs = ~0;
 120    ctx->blitter.saved_num_textures = ~0;
 121    ctx->blitter.saved_num_sampler_states = ~0;
 122
 123    /* blend state objects */
 124    ctx->blend_keep_color = pipe->create_blend_state(pipe, &blend);
 125
 126    blend.rt[0].colormask = PIPE_MASK_RGBA;
 127    ctx->blend_write_color = pipe->create_blend_state(pipe, &blend);
 128
 129    /* depth stencil alpha state objects */
 130    ctx->dsa_keep_depth_stencil =
 131       pipe->create_depth_stencil_alpha_state(pipe, &dsa);
 132
 133    dsa.depth.enabled = 1;
 134    dsa.depth.writemask = 1;
 135    dsa.depth.func = PIPE_FUNC_ALWAYS;
 136    ctx->dsa_write_depth_keep_stencil =
 137       pipe->create_depth_stencil_alpha_state(pipe, &dsa);
 138
 139    dsa.stencil[0].enabled = 1;
 140    dsa.stencil[0].func = PIPE_FUNC_ALWAYS;
 141    dsa.stencil[0].fail_op = PIPE_STENCIL_OP_REPLACE;
 142    dsa.stencil[0].zpass_op = PIPE_STENCIL_OP_REPLACE;
 143    dsa.stencil[0].zfail_op = PIPE_STENCIL_OP_REPLACE;
 144    dsa.stencil[0].valuemask = 0xff;
 145    dsa.stencil[0].writemask = 0xff;
 146    ctx->dsa_write_depth_stencil =
 147       pipe->create_depth_stencil_alpha_state(pipe, &dsa);
 148    /* The DSA state objects which write depth and stencil are created
 149     * on-demand. */
 150
 151    /* sampler state */
 152    sampler_state = &ctx->template_sampler_state;
 153    sampler_state->wrap_s = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 154    sampler_state->wrap_t = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 155    sampler_state->wrap_r = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 156    /* The sampler state objects which sample from a specified mipmap level
 157     * are created on-demand. */
 158
 159    /* rasterizer state */
 160    memset(&rs_state, 0, sizeof(rs_state));
 161    rs_state.front_winding = PIPE_WINDING_CW;
 162    rs_state.cull_mode = PIPE_WINDING_NONE;
 163    rs_state.gl_rasterization_rules = 1;
 164    rs_state.flatshade = 1;
 165    ctx->rs_state = pipe->create_rasterizer_state(pipe, &rs_state);
 166
 167    /* fragment shaders are created on-demand */
 168
 169    /* vertex shaders */
 170    {
 171       const uint semantic_names[] = { TGSI_SEMANTIC_POSITION,
 172                                       TGSI_SEMANTIC_COLOR };
 173       const uint semantic_indices[] = { 0, 0 };
 174       ctx->vs_col =
 175          util_make_vertex_passthrough_shader(pipe, 2, semantic_names,
 176                                              semantic_indices);
 177    }
 178    {
 179       const uint semantic_names[] = { TGSI_SEMANTIC_POSITION,
 180                                       TGSI_SEMANTIC_GENERIC };
 181       const uint semantic_indices[] = { 0, 0 };
 182       ctx->vs_tex =
 183          util_make_vertex_passthrough_shader(pipe, 2, semantic_names,
 184                                              semantic_indices);
 185    }
 186
 187    /* set invariant vertex coordinates */
 188    for (i = 0; i < 4; i++)
 189       ctx->vertices[i][0][3] = 1; /*v.w*/
 190
 191    /* create the vertex buffer */
 192    ctx->vbuf = pipe_buffer_create(ctx->pipe->screen,
 193                                   32,
 194                                   PIPE_BUFFER_USAGE_VERTEX,
 195                                   sizeof(ctx->vertices));
 196
 197    return &ctx->blitter;
 198 }
 199
 200 void util_blitter_destroy(struct blitter_context *blitter)
 201 {
 202    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 203    struct pipe_context *pipe = ctx->pipe;
 204    int i;
 205
 206    pipe->delete_blend_state(pipe, ctx->blend_write_color);
 207    pipe->delete_blend_state(pipe, ctx->blend_keep_color);
 208    pipe->delete_depth_stencil_alpha_state(pipe, ctx->dsa_keep_depth_stencil);
 209    pipe->delete_depth_stencil_alpha_state(pipe,
 210                                           ctx->dsa_write_depth_keep_stencil);
 211    pipe->delete_depth_stencil_alpha_state(pipe, ctx->dsa_write_depth_stencil);
 212
 213    pipe->delete_rasterizer_state(pipe, ctx->rs_state);
 214    pipe->delete_vs_state(pipe, ctx->vs_col);
 215    pipe->delete_vs_state(pipe, ctx->vs_tex);
 216
 217    for (i = 0; i < PIPE_MAX_TEXTURE_TYPES; i++) {
 218       if (ctx->fs_texfetch_col[i])
 219          pipe->delete_fs_state(pipe, ctx->fs_texfetch_col[i]);
 220       if (ctx->fs_texfetch_depth[i])
 221          pipe->delete_fs_state(pipe, ctx->fs_texfetch_depth[i]);
 222    }
 223
 224    for (i = 0; i < PIPE_MAX_COLOR_BUFS && ctx->fs_col[i]; i++)
 225       if (ctx->fs_col[i])
 226          pipe->delete_fs_state(pipe, ctx->fs_col[i]);
 227
 228    for (i = 0; i < PIPE_MAX_TEXTURE_LEVELS; i++)
 229       if (ctx->sampler_state[i])
 230          pipe->delete_sampler_state(pipe, ctx->sampler_state[i]);
 231
 232    pipe_buffer_reference(&ctx->vbuf, NULL);
 233    FREE(ctx);
 234 }
 235
 236 static void blitter_check_saved_CSOs(struct blitter_context_priv *ctx)
 237 {
 238    /* make sure these CSOs have been saved */
 239    assert(ctx->blitter.saved_blend_state != INVALID_PTR &&
 240           ctx->blitter.saved_dsa_state != INVALID_PTR &&
 241           ctx->blitter.saved_rs_state != INVALID_PTR &&
 242           ctx->blitter.saved_fs != INVALID_PTR &&
 243           ctx->blitter.saved_vs != INVALID_PTR);
 244 }
 245
 246 static void blitter_restore_CSOs(struct blitter_context_priv *ctx)
 247 {
 248    struct pipe_context *pipe = ctx->pipe;
 249
 250    /* restore the state objects which are always required to be saved */
 251    pipe->bind_blend_state(pipe, ctx->blitter.saved_blend_state);
 252    pipe->bind_depth_stencil_alpha_state(pipe, ctx->blitter.saved_dsa_state);
 253    pipe->bind_rasterizer_state(pipe, ctx->blitter.saved_rs_state);
 254    pipe->bind_fs_state(pipe, ctx->blitter.saved_fs);
 255    pipe->bind_vs_state(pipe, ctx->blitter.saved_vs);
 256
 257    ctx->blitter.saved_blend_state = INVALID_PTR;
 258    ctx->blitter.saved_dsa_state = INVALID_PTR;
 259    ctx->blitter.saved_rs_state = INVALID_PTR;
 260    ctx->blitter.saved_fs = INVALID_PTR;
 261    ctx->blitter.saved_vs = INVALID_PTR;
 262
 263    pipe->set_stencil_ref(pipe, &ctx->blitter.saved_stencil_ref);
 264
 265    /* restore the state objects which are required to be saved before copy/fill
 266     */
 267    if (ctx->blitter.saved_fb_state.nr_cbufs != ~0) {
 268       pipe->set_framebuffer_state(pipe, &ctx->blitter.saved_fb_state);
 269       ctx->blitter.saved_fb_state.nr_cbufs = ~0;
 270    }
 271
 272    if (ctx->blitter.saved_num_sampler_states != ~0) {
 273       pipe->bind_fragment_sampler_states(pipe,
 274                                          ctx->blitter.saved_num_sampler_states,
 275                                          ctx->blitter.saved_sampler_states);
 276       ctx->blitter.saved_num_sampler_states = ~0;
 277    }
 278
 279    if (ctx->blitter.saved_num_textures != ~0) {
 280       pipe->set_fragment_sampler_textures(pipe,
 281                                           ctx->blitter.saved_num_textures,
 282                                           ctx->blitter.saved_textures);
 283       ctx->blitter.saved_num_textures = ~0;
 284    }
 285 }
 286
 287 static void blitter_set_rectangle(struct blitter_context_priv *ctx,
 288                                   unsigned x1, unsigned y1,
 289                                   unsigned x2, unsigned y2,
 290                                   float depth)
 291 {
 292    int i;
 293
 294    /* set vertex positions */
 295    ctx->vertices[0][0][0] = x1; /*v0.x*/
 296    ctx->vertices[0][0][1] = y1; /*v0.y*/
 297
 298    ctx->vertices[1][0][0] = x2; /*v1.x*/
 299    ctx->vertices[1][0][1] = y1; /*v1.y*/
 300
 301    ctx->vertices[2][0][0] = x2; /*v2.x*/
 302    ctx->vertices[2][0][1] = y2; /*v2.y*/
 303
 304    ctx->vertices[3][0][0] = x1; /*v3.x*/
 305    ctx->vertices[3][0][1] = y2; /*v3.y*/
 306
 307    for (i = 0; i < 4; i++)
 308       ctx->vertices[i][0][2] = depth; /*z*/
 309 }
 310
 311 static void blitter_set_clear_color(struct blitter_context_priv *ctx,
 312                                     const float *rgba)
 313 {
 314    int i;
 315
 316    for (i = 0; i < 4; i++) {
 317       ctx->vertices[i][1][0] = rgba[0];
 318       ctx->vertices[i][1][1] = rgba[1];
 319       ctx->vertices[i][1][2] = rgba[2];
 320       ctx->vertices[i][1][3] = rgba[3];
 321    }
 322 }
 323
 324 static void blitter_set_texcoords_2d(struct blitter_context_priv *ctx,
 325                                      struct pipe_surface *surf,
 326                                      unsigned x1, unsigned y1,
 327                                      unsigned x2, unsigned y2)
 328 {
 329    int i;
 330    float s1 = x1 / (float)surf->width;
 331    float t1 = y1 / (float)surf->height;
 332    float s2 = x2 / (float)surf->width;
 333    float t2 = y2 / (float)surf->height;
 334
 335    ctx->vertices[0][1][0] = s1; /*t0.s*/
 336    ctx->vertices[0][1][1] = t1; /*t0.t*/
 337
 338    ctx->vertices[1][1][0] = s2; /*t1.s*/
 339    ctx->vertices[1][1][1] = t1; /*t1.t*/
 340
 341    ctx->vertices[2][1][0] = s2; /*t2.s*/
 342    ctx->vertices[2][1][1] = t2; /*t2.t*/
 343
 344    ctx->vertices[3][1][0] = s1; /*t3.s*/
 345    ctx->vertices[3][1][1] = t2; /*t3.t*/
 346
 347    for (i = 0; i < 4; i++) {
 348       ctx->vertices[i][1][2] = 0; /*r*/
 349       ctx->vertices[i][1][3] = 1; /*q*/
 350    }
 351 }
 352
 353 static void blitter_set_texcoords_3d(struct blitter_context_priv *ctx,
 354                                      struct pipe_surface *surf,
 355                                      unsigned x1, unsigned y1,
 356                                      unsigned x2, unsigned y2)
 357 {
 358    int i;
 359    float depth = u_minify(surf->texture->depth0, surf->level);
 360    float r = surf->zslice / depth;
 361
 362    blitter_set_texcoords_2d(ctx, surf, x1, y1, x2, y2);
 363
 364    for (i = 0; i < 4; i++)
 365       ctx->vertices[i][1][2] = r; /*r*/
 366 }
 367
 368 static void blitter_set_texcoords_cube(struct blitter_context_priv *ctx,
 369                                        struct pipe_surface *surf,
 370                                        unsigned x1, unsigned y1,
 371                                        unsigned x2, unsigned y2)
 372 {
 373    int i;
 374    float s1 = x1 / (float)surf->width;
 375    float t1 = y1 / (float)surf->height;
 376    float s2 = x2 / (float)surf->width;
 377    float t2 = y2 / (float)surf->height;
 378    float st[4][2];
 379
 380    st[0][0] = s1;
 381    st[0][1] = t1;
 382    st[1][0] = s2;
 383    st[1][1] = t1;
 384    st[2][0] = s2;
 385    st[2][1] = t2;
 386    st[3][0] = s1;
 387    st[3][1] = t2;
 388
 389    util_map_texcoords2d_onto_cubemap(surf->face,
 390                                      /* pointer, stride in floats */
 391                                      &st[0][0], 2,
 392                                      &ctx->vertices[0][1][0], 8);
 393
 394    for (i = 0; i < 4; i++)
 395       ctx->vertices[i][1][3] = 1; /*q*/
 396 }
 397
 398 static void blitter_draw_quad(struct blitter_context_priv *ctx)
 399 {
 400    struct pipe_context *pipe = ctx->pipe;
 401
 402    /* write vertices and draw them */
 403    pipe_buffer_write(pipe->screen, ctx->vbuf,
 404                      0, sizeof(ctx->vertices), ctx->vertices);
 405
 406    util_draw_vertex_buffer(pipe, ctx->vbuf, 0, PIPE_PRIM_TRIANGLE_FAN,
 407                            4,  /* verts */
 408                            2); /* attribs/vert */
 409 }
 410
 411 static INLINE
 412 void **blitter_get_sampler_state(struct blitter_context_priv *ctx,
 413                                  int miplevel)
 414 {
 415    struct pipe_context *pipe = ctx->pipe;
 416    struct pipe_sampler_state *sampler_state = &ctx->template_sampler_state;
 417
 418    assert(miplevel < PIPE_MAX_TEXTURE_LEVELS);
 419
 420    /* Create the sampler state on-demand. */
 421    if (!ctx->sampler_state[miplevel]) {
 422       sampler_state->lod_bias = miplevel;
 423       sampler_state->min_lod = miplevel;
 424       sampler_state->max_lod = miplevel;
 425
 426       ctx->sampler_state[miplevel] = pipe->create_sampler_state(pipe,
 427                                                                 sampler_state);
 428    }
 429
 430    /* Return void** so that it can be passed to bind_fragment_sampler_states
 431     * directly. */
 432    return &ctx->sampler_state[miplevel];
 433 }
 434
 435 static INLINE
 436 void *blitter_get_fs_col(struct blitter_context_priv *ctx, unsigned num_cbufs)
 437 {
 438    struct pipe_context *pipe = ctx->pipe;
 439    unsigned index = num_cbufs ? num_cbufs - 1 : 0;
 440
 441    assert(num_cbufs <= PIPE_MAX_COLOR_BUFS);
 442
 443    if (!ctx->fs_col[index])
 444       ctx->fs_col[index] =
 445          util_make_fragment_clonecolor_shader(pipe, num_cbufs);
 446
 447    return ctx->fs_col[index];
 448 }
 449
 450 static INLINE
 451 void *blitter_get_fs_texfetch_col(struct blitter_context_priv *ctx,
 452                                   unsigned tex_target)
 453 {
 454    struct pipe_context *pipe = ctx->pipe;
 455
 456    assert(tex_target < PIPE_MAX_TEXTURE_TYPES);
 457
 458    /* Create the fragment shader on-demand. */
 459    if (!ctx->fs_texfetch_col[tex_target]) {
 460       switch (tex_target) {
 461          case PIPE_TEXTURE_1D:
 462             ctx->fs_texfetch_col[PIPE_TEXTURE_1D] =
 463                util_make_fragment_tex_shader(pipe, TGSI_TEXTURE_1D);
 464             break;
 465          case PIPE_TEXTURE_2D:
 466             ctx->fs_texfetch_col[PIPE_TEXTURE_2D] =
 467                util_make_fragment_tex_shader(pipe, TGSI_TEXTURE_2D);
 468             break;
 469          case PIPE_TEXTURE_3D:
 470             ctx->fs_texfetch_col[PIPE_TEXTURE_3D] =
 471                util_make_fragment_tex_shader(pipe, TGSI_TEXTURE_3D);
 472             break;
 473          case PIPE_TEXTURE_CUBE:
 474             ctx->fs_texfetch_col[PIPE_TEXTURE_CUBE] =
 475                util_make_fragment_tex_shader(pipe, TGSI_TEXTURE_CUBE);
 476             break;
 477          default:;
 478       }
 479    }
 480
 481    return ctx->fs_texfetch_col[tex_target];
 482 }
 483
 484 static INLINE
 485 void *blitter_get_fs_texfetch_depth(struct blitter_context_priv *ctx,
 486                                     unsigned tex_target)
 487 {
 488    struct pipe_context *pipe = ctx->pipe;
 489
 490    assert(tex_target < PIPE_MAX_TEXTURE_TYPES);
 491
 492    /* Create the fragment shader on-demand. */
 493    if (!ctx->fs_texfetch_depth[tex_target]) {
 494       switch (tex_target) {
 495          case PIPE_TEXTURE_1D:
 496             ctx->fs_texfetch_depth[PIPE_TEXTURE_1D] =
 497                util_make_fragment_tex_shader_writedepth(pipe, TGSI_TEXTURE_1D);
 498             break;
 499          case PIPE_TEXTURE_2D:
 500             ctx->fs_texfetch_depth[PIPE_TEXTURE_2D] =
 501                util_make_fragment_tex_shader_writedepth(pipe, TGSI_TEXTURE_2D);
 502             break;
 503          case PIPE_TEXTURE_3D:
 504             ctx->fs_texfetch_depth[PIPE_TEXTURE_3D] =
 505                util_make_fragment_tex_shader_writedepth(pipe, TGSI_TEXTURE_3D);
 506             break;
 507          case PIPE_TEXTURE_CUBE:
 508             ctx->fs_texfetch_depth[PIPE_TEXTURE_CUBE] =
 509                util_make_fragment_tex_shader_writedepth(pipe,TGSI_TEXTURE_CUBE);
 510             break;
 511          default:;
 512       }
 513    }
 514
 515    return ctx->fs_texfetch_depth[tex_target];
 516 }
 517
 518 void util_blitter_clear(struct blitter_context *blitter,
 519                         unsigned width, unsigned height,
 520                         unsigned num_cbufs,
 521                         unsigned clear_buffers,
 522                         const float *rgba,
 523                         double depth, unsigned stencil)
 524 {
 525    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 526    struct pipe_context *pipe = ctx->pipe;
 527    struct pipe_stencil_ref sr = { { 0 } };
 528
 529    assert(num_cbufs <= PIPE_MAX_COLOR_BUFS);
 530
 531    blitter_check_saved_CSOs(ctx);
 532
 533    /* bind CSOs */
 534    if (clear_buffers & PIPE_CLEAR_COLOR)
 535       pipe->bind_blend_state(pipe, ctx->blend_write_color);
 536    else
 537       pipe->bind_blend_state(pipe, ctx->blend_keep_color);
 538
 539    if (clear_buffers & PIPE_CLEAR_DEPTHSTENCIL) {
 540       sr.ref_value[0] = stencil & 0xff;
 541       pipe->bind_depth_stencil_alpha_state(pipe, ctx->dsa_write_depth_stencil);
 542       pipe->set_stencil_ref(pipe, &sr);
 543    }
 544    else
 545       pipe->bind_depth_stencil_alpha_state(pipe, ctx->dsa_keep_depth_stencil);
 546
 547    pipe->bind_rasterizer_state(pipe, ctx->rs_state);
 548    pipe->bind_fs_state(pipe, blitter_get_fs_col(ctx, num_cbufs));
 549    pipe->bind_vs_state(pipe, ctx->vs_col);
 550
 551    blitter_set_clear_color(ctx, rgba);
 552    blitter_set_rectangle(ctx, 0, 0, width, height, depth);
 553    blitter_draw_quad(ctx);
 554    blitter_restore_CSOs(ctx);
 555 }
 556
 557 static boolean
 558 is_overlap(unsigned sx1, unsigned sx2, unsigned sy1, unsigned sy2,
 559            unsigned dx1, unsigned dx2, unsigned dy1, unsigned dy2)
 560 {
 561     if (sx1 >= dx2 || sx2 <= dx1 || sy1 >= dy2 || sy2 <= dy1) {
 562         return FALSE;
 563     } else {
 564         return TRUE;
 565     }
 566 }
 567
 568 static void util_blitter_do_copy(struct blitter_context *blitter,
 569                                  struct pipe_surface *dst,
 570                                  unsigned dstx, unsigned dsty,
 571                                  struct pipe_surface *src,
 572                                  unsigned srcx, unsigned srcy,
 573                                  unsigned width, unsigned height,
 574                                  boolean is_depth)
 575 {
 576    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 577    struct pipe_context *pipe = ctx->pipe;
 578    struct pipe_framebuffer_state fb_state;
 579
 580    assert(blitter->saved_fb_state.nr_cbufs != ~0);
 581    assert(blitter->saved_num_textures != ~0);
 582    assert(blitter->saved_num_sampler_states != ~0);
 583    assert(src->texture->target < PIPE_MAX_TEXTURE_TYPES);
 584
 585    /* bind CSOs */
 586    fb_state.width = dst->width;
 587    fb_state.height = dst->height;
 588
 589    if (is_depth) {
 590       pipe->bind_blend_state(pipe, ctx->blend_keep_color);
 591       pipe->bind_depth_stencil_alpha_state(pipe,
 592                                            ctx->dsa_write_depth_keep_stencil);
 593       pipe->bind_fs_state(pipe,
 594          blitter_get_fs_texfetch_depth(ctx, src->texture->target));
 595
 596       fb_state.nr_cbufs = 0;
 597       fb_state.zsbuf = dst;
 598    } else {
 599       pipe->bind_blend_state(pipe, ctx->blend_write_color);
 600       pipe->bind_depth_stencil_alpha_state(pipe, ctx->dsa_keep_depth_stencil);
 601       pipe->bind_fs_state(pipe,
 602          blitter_get_fs_texfetch_col(ctx, src->texture->target));
 603
 604       fb_state.nr_cbufs = 1;
 605       fb_state.cbufs[0] = dst;
 606       fb_state.zsbuf = 0;
 607    }
 608
 609    pipe->bind_rasterizer_state(pipe, ctx->rs_state);
 610    pipe->bind_vs_state(pipe, ctx->vs_tex);
 611    pipe->bind_fragment_sampler_states(pipe, 1,
 612       blitter_get_sampler_state(ctx, src->level));
 613    pipe->set_fragment_sampler_textures(pipe, 1, &src->texture);
 614    pipe->set_framebuffer_state(pipe, &fb_state);
 615
 616    /* set texture coordinates */
 617    switch (src->texture->target) {
 618       case PIPE_TEXTURE_1D:
 619       case PIPE_TEXTURE_2D:
 620          blitter_set_texcoords_2d(ctx, src, srcx, srcy,
 621                                   srcx+width, srcy+height);
 622          break;
 623       case PIPE_TEXTURE_3D:
 624          blitter_set_texcoords_3d(ctx, src, srcx, srcy,
 625                                   srcx+width, srcy+height);
 626          break;
 627       case PIPE_TEXTURE_CUBE:
 628          blitter_set_texcoords_cube(ctx, src, srcx, srcy,
 629                                     srcx+width, srcy+height);
 630          break;
 631       default:
 632          assert(0);
 633    }
 634
 635    blitter_set_rectangle(ctx, dstx, dsty, dstx+width, dsty+height, 0);
 636    blitter_draw_quad(ctx);
 637
 638 }
 639
 640 static void util_blitter_overlap_copy(struct blitter_context *blitter,
 641                                       struct pipe_surface *dst,
 642                                       unsigned dstx, unsigned dsty,
 643                                       struct pipe_surface *src,
 644                                       unsigned srcx, unsigned srcy,
 645                                       unsigned width, unsigned height)
 646 {
 647    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 648    struct pipe_context *pipe = ctx->pipe;
 649    struct pipe_screen *screen = pipe->screen;
 650
 651    struct pipe_texture texTemp;
 652    struct pipe_texture *texture;
 653    struct pipe_surface *tex_surf;
 654
 655    /* check whether the states are properly saved */
 656    blitter_check_saved_CSOs(ctx);
 657
 658    memset(&texTemp, 0, sizeof(texTemp));
 659    texTemp.target = PIPE_TEXTURE_2D;
 660    texTemp.format = dst->texture->format; /* XXX verify supported by driver! */
 661    texTemp.last_level = 0;
 662    texTemp.width0 = width;
 663    texTemp.height0 = height;
 664    texTemp.depth0 = 1;
 665
 666    texture = screen->texture_create(screen, &texTemp);
 667    if (!texture)
 668       return;
 669
 670    tex_surf = screen->get_tex_surface(screen, texture, 0, 0, 0,
 671                                       PIPE_BUFFER_USAGE_GPU_READ |
 672                                       PIPE_BUFFER_USAGE_GPU_WRITE);
 673
 674    /* blit from the src to the temp */
 675    util_blitter_do_copy(blitter, tex_surf, 0, 0,
 676                         src, srcx, srcy,
 677                         width, height,
 678                         FALSE);
 679    util_blitter_do_copy(blitter, dst, dstx, dsty,
 680                         tex_surf, 0, 0,
 681                         width, height,
 682                         FALSE);
 683    pipe_surface_reference(&tex_surf, NULL);
 684    pipe_texture_reference(&texture, NULL);
 685    blitter_restore_CSOs(ctx);
 686 }
 687
 688 void util_blitter_copy(struct blitter_context *blitter,
 689                        struct pipe_surface *dst,
 690                        unsigned dstx, unsigned dsty,
 691                        struct pipe_surface *src,
 692                        unsigned srcx, unsigned srcy,
 693                        unsigned width, unsigned height,
 694                        boolean ignore_stencil)
 695 {
 696    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 697    struct pipe_context *pipe = ctx->pipe;
 698    struct pipe_screen *screen = pipe->screen;
 699    boolean is_stencil, is_depth;
 700    unsigned dst_tex_usage;
 701
 702    /* give up if textures are not set */
 703    assert(dst->texture && src->texture);
 704    if (!dst->texture || !src->texture)
 705       return;
 706
 707    if (dst->texture == src->texture) {
 708       if (is_overlap(srcx, srcx + width, srcy, srcy + height,
 709                              dstx, dstx + width, dsty, dsty + height)) {
 710          util_blitter_overlap_copy(blitter, dst, dstx, dsty, src, srcx, srcy,
 711                                    width, height);
 712          return;
 713       }
 714    }
 715
 716    is_depth = util_format_get_component_bits(src->format, UTIL_FORMAT_COLORSPACE_ZS, 0) != 0;
 717    is_stencil = util_format_get_component_bits(src->format, UTIL_FORMAT_COLORSPACE_ZS, 1) != 0;
 718    dst_tex_usage = is_depth || is_stencil ? PIPE_TEXTURE_USAGE_DEPTH_STENCIL :
 719                                             PIPE_TEXTURE_USAGE_RENDER_TARGET;
 720
 721    /* check if we can sample from and render to the surfaces */
 722    /* (assuming copying a stencil buffer is not possible) */
 723    if ((!ignore_stencil && is_stencil) ||
 724        !screen->is_format_supported(screen, dst->format, dst->texture->target,
 725                                     dst_tex_usage, 0) ||
 726        !screen->is_format_supported(screen, src->format, src->texture->target,
 727                                     PIPE_TEXTURE_USAGE_SAMPLER, 0)) {
 728       util_surface_copy(pipe, FALSE, dst, dstx, dsty, src, srcx, srcy,
 729                         width, height);
 730       return;
 731    }
 732
 733    /* check whether the states are properly saved */
 734    blitter_check_saved_CSOs(ctx);
 735    util_blitter_do_copy(blitter,
 736                         dst, dstx, dsty,
 737                         src, srcx, srcy,
 738                         width, height, is_depth);
 739    blitter_restore_CSOs(ctx);
 740 }
 741
 742 void util_blitter_fill(struct blitter_context *blitter,
 743                        struct pipe_surface *dst,
 744                        unsigned dstx, unsigned dsty,
 745                        unsigned width, unsigned height,
 746                        unsigned value)
 747 {
 748    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 749    struct pipe_context *pipe = ctx->pipe;
 750    struct pipe_screen *screen = pipe->screen;
 751    struct pipe_framebuffer_state fb_state;
 752    float rgba[4];
 753    ubyte ub_rgba[4] = {0};
 754    union util_color color;
 755    int i;
 756
 757    assert(dst->texture);
 758    if (!dst->texture)
 759       return;
 760
 761    /* check if we can render to the surface */
 762    if (util_format_is_depth_or_stencil(dst->format) || /* unlikely, but you never know */
 763        !screen->is_format_supported(screen, dst->format, dst->texture->target,
 764                                     PIPE_TEXTURE_USAGE_RENDER_TARGET, 0)) {
 765       util_surface_fill(pipe, dst, dstx, dsty, width, height, value);
 766       return;
 767    }
 768
 769    /* unpack the color */
 770    color.ui = value;
 771    util_unpack_color_ub(dst->format, &color,
 772                         ub_rgba, ub_rgba+1, ub_rgba+2, ub_rgba+3);
 773    for (i = 0; i < 4; i++)
 774       rgba[i] = ubyte_to_float(ub_rgba[i]);
 775
 776    /* check the saved state */
 777    blitter_check_saved_CSOs(ctx);
 778    assert(blitter->saved_fb_state.nr_cbufs != ~0);
 779
 780    /* bind CSOs */
 781    pipe->bind_blend_state(pipe, ctx->blend_write_color);
 782    pipe->bind_depth_stencil_alpha_state(pipe, ctx->dsa_keep_depth_stencil);
 783    pipe->bind_rasterizer_state(pipe, ctx->rs_state);
 784    pipe->bind_fs_state(pipe, blitter_get_fs_col(ctx, 1));
 785    pipe->bind_vs_state(pipe, ctx->vs_col);
 786
 787    /* set a framebuffer state */
 788    fb_state.width = dst->width;
 789    fb_state.height = dst->height;
 790    fb_state.nr_cbufs = 1;
 791    fb_state.cbufs[0] = dst;
 792    fb_state.zsbuf = 0;
 793    pipe->set_framebuffer_state(pipe, &fb_state);
 794
 795    blitter_set_clear_color(ctx, rgba);
 796    blitter_set_rectangle(ctx, 0, 0, width, height, 0);
 797    blitter_draw_quad(ctx);
 798    blitter_restore_CSOs(ctx);
 799 }