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;
 102    struct pipe_depth_stencil_alpha_state dsa;
 103    struct pipe_rasterizer_state rs_state;
 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    memset(&blend, 0, sizeof(blend));
 125    ctx->blend_keep_color = pipe->create_blend_state(pipe, &blend);
 126
 127    blend.rt[0].colormask = PIPE_MASK_RGBA;
 128    ctx->blend_write_color = pipe->create_blend_state(pipe, &blend);
 129
 130    /* depth stencil alpha state objects */
 131    ctx->dsa_keep_depth_stencil =
 132       pipe->create_depth_stencil_alpha_state(pipe, &dsa);
 133
 134    dsa.depth.enabled = 1;
 135    dsa.depth.writemask = 1;
 136    dsa.depth.func = PIPE_FUNC_ALWAYS;
 137    ctx->dsa_write_depth_keep_stencil =
 138       pipe->create_depth_stencil_alpha_state(pipe, &dsa);
 139
 140    dsa.stencil[0].enabled = 1;
 141    dsa.stencil[0].func = PIPE_FUNC_ALWAYS;
 142    dsa.stencil[0].fail_op = PIPE_STENCIL_OP_REPLACE;
 143    dsa.stencil[0].zpass_op = PIPE_STENCIL_OP_REPLACE;
 144    dsa.stencil[0].zfail_op = PIPE_STENCIL_OP_REPLACE;
 145    dsa.stencil[0].valuemask = 0xff;
 146    dsa.stencil[0].writemask = 0xff;
 147    ctx->dsa_write_depth_stencil =
 148       pipe->create_depth_stencil_alpha_state(pipe, &dsa);
 149    /* The DSA state objects which write depth and stencil are created
 150     * on-demand. */
 151
 152    /* sampler state */
 153    sampler_state = &ctx->template_sampler_state;
 154    sampler_state->wrap_s = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 155    sampler_state->wrap_t = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 156    sampler_state->wrap_r = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 157    /* The sampler state objects which sample from a specified mipmap level
 158     * are created on-demand. */
 159
 160    /* rasterizer state */
 161    memset(&rs_state, 0, sizeof(rs_state));
 162    rs_state.front_winding = PIPE_WINDING_CW;
 163    rs_state.cull_mode = PIPE_WINDING_NONE;
 164    rs_state.bypass_vs_clip_and_viewport = 1;
 165    rs_state.gl_rasterization_rules = 1;
 166    rs_state.flatshade = 1;
 167    ctx->rs_state = pipe->create_rasterizer_state(pipe, &rs_state);
 168
 169    /* fragment shaders are created on-demand */
 170
 171    /* vertex shaders */
 172    {
 173       const uint semantic_names[] = { TGSI_SEMANTIC_POSITION,
 174                                       TGSI_SEMANTIC_COLOR };
 175       const uint semantic_indices[] = { 0, 0 };
 176       ctx->vs_col =
 177          util_make_vertex_passthrough_shader(pipe, 2, semantic_names,
 178                                              semantic_indices);
 179    }
 180    {
 181       const uint semantic_names[] = { TGSI_SEMANTIC_POSITION,
 182                                       TGSI_SEMANTIC_GENERIC };
 183       const uint semantic_indices[] = { 0, 0 };
 184       ctx->vs_tex =
 185          util_make_vertex_passthrough_shader(pipe, 2, semantic_names,
 186                                              semantic_indices);
 187    }
 188
 189    /* set invariant vertex coordinates */
 190    for (i = 0; i < 4; i++)
 191       ctx->vertices[i][0][3] = 1; /*v.w*/
 192
 193    /* create the vertex buffer */
 194    ctx->vbuf = pipe_buffer_create(ctx->pipe->screen,
 195                                   32,
 196                                   PIPE_BUFFER_USAGE_VERTEX,
 197                                   sizeof(ctx->vertices));
 198
 199    return &ctx->blitter;
 200 }
 201
 202 void util_blitter_destroy(struct blitter_context *blitter)
 203 {
 204    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 205    struct pipe_context *pipe = ctx->pipe;
 206    int i;
 207
 208    pipe->delete_blend_state(pipe, ctx->blend_write_color);
 209    pipe->delete_blend_state(pipe, ctx->blend_keep_color);
 210    pipe->delete_depth_stencil_alpha_state(pipe, ctx->dsa_keep_depth_stencil);
 211    pipe->delete_depth_stencil_alpha_state(pipe,
 212                                           ctx->dsa_write_depth_keep_stencil);
 213    pipe->delete_depth_stencil_alpha_state(pipe, ctx->dsa_write_depth_stencil);
 214
 215    pipe->delete_rasterizer_state(pipe, ctx->rs_state);
 216    pipe->delete_vs_state(pipe, ctx->vs_col);
 217    pipe->delete_vs_state(pipe, ctx->vs_tex);
 218
 219    for (i = 0; i < PIPE_MAX_TEXTURE_TYPES; i++) {
 220       if (ctx->fs_texfetch_col[i])
 221          pipe->delete_fs_state(pipe, ctx->fs_texfetch_col[i]);
 222       if (ctx->fs_texfetch_depth[i])
 223          pipe->delete_fs_state(pipe, ctx->fs_texfetch_depth[i]);
 224    }
 225
 226    for (i = 0; i < PIPE_MAX_COLOR_BUFS && ctx->fs_col[i]; i++)
 227       if (ctx->fs_col[i])
 228          pipe->delete_fs_state(pipe, ctx->fs_col[i]);
 229
 230    for (i = 0; i < PIPE_MAX_TEXTURE_LEVELS; i++)
 231       if (ctx->sampler_state[i])
 232          pipe->delete_sampler_state(pipe, ctx->sampler_state[i]);
 233
 234    pipe_buffer_reference(&ctx->vbuf, NULL);
 235    FREE(ctx);
 236 }
 237
 238 static void blitter_check_saved_CSOs(struct blitter_context_priv *ctx)
 239 {
 240    /* make sure these CSOs have been saved */
 241    assert(ctx->blitter.saved_blend_state != INVALID_PTR &&
 242           ctx->blitter.saved_dsa_state != INVALID_PTR &&
 243           ctx->blitter.saved_rs_state != INVALID_PTR &&
 244           ctx->blitter.saved_fs != INVALID_PTR &&
 245           ctx->blitter.saved_vs != INVALID_PTR);
 246 }
 247
 248 static void blitter_restore_CSOs(struct blitter_context_priv *ctx)
 249 {
 250    struct pipe_context *pipe = ctx->pipe;
 251
 252    /* restore the state objects which are always required to be saved */
 253    pipe->bind_blend_state(pipe, ctx->blitter.saved_blend_state);
 254    pipe->bind_depth_stencil_alpha_state(pipe, ctx->blitter.saved_dsa_state);
 255    pipe->bind_rasterizer_state(pipe, ctx->blitter.saved_rs_state);
 256    pipe->bind_fs_state(pipe, ctx->blitter.saved_fs);
 257    pipe->bind_vs_state(pipe, ctx->blitter.saved_vs);
 258
 259    ctx->blitter.saved_blend_state = INVALID_PTR;
 260    ctx->blitter.saved_dsa_state = INVALID_PTR;
 261    ctx->blitter.saved_rs_state = INVALID_PTR;
 262    ctx->blitter.saved_fs = INVALID_PTR;
 263    ctx->blitter.saved_vs = INVALID_PTR;
 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
 528    assert(num_cbufs <= PIPE_MAX_COLOR_BUFS);
 529
 530    blitter_check_saved_CSOs(ctx);
 531
 532    /* bind CSOs */
 533    if (clear_buffers & PIPE_CLEAR_COLOR)
 534       pipe->bind_blend_state(pipe, ctx->blend_write_color);
 535    else
 536       pipe->bind_blend_state(pipe, ctx->blend_keep_color);
 537
 538    if (clear_buffers & PIPE_CLEAR_DEPTHSTENCIL) {
 539       struct pipe_stencil_ref sr;
 540       memset (&sr, 0, sizeof(sr));
 541       sr.ref_value[0] = stencil & 0xff;
 542       pipe->bind_depth_stencil_alpha_state(pipe, ctx->dsa_write_depth_stencil);
 543       pipe->set_stencil_ref(pipe, &sr);
 544    }
 545    else
 546       pipe->bind_depth_stencil_alpha_state(pipe, ctx->dsa_keep_depth_stencil);
 547
 548    pipe->bind_rasterizer_state(pipe, ctx->rs_state);
 549    pipe->bind_fs_state(pipe, blitter_get_fs_col(ctx, num_cbufs));
 550    pipe->bind_vs_state(pipe, ctx->vs_col);
 551
 552    blitter_set_clear_color(ctx, rgba);
 553    blitter_set_rectangle(ctx, 0, 0, width, height, depth);
 554    blitter_draw_quad(ctx);
 555    blitter_restore_CSOs(ctx);
 556    /* XXX driver's responsibility to restore stencil refs? */
 557 }
 558
 559 static boolean
 560 is_overlap(unsigned sx1, unsigned sx2, unsigned sy1, unsigned sy2,
 561            unsigned dx1, unsigned dx2, unsigned dy1, unsigned dy2)
 562 {
 563     if (sx1 >= dx2 || sx2 <= dx1 || sy1 >= dy2 || sy2 <= dy1) {
 564         return FALSE;
 565     } else {
 566         return TRUE;
 567     }
 568 }
 569
 570 static void util_blitter_do_copy(struct blitter_context *blitter,
 571                                  struct pipe_surface *dst,
 572                                  unsigned dstx, unsigned dsty,
 573                                  struct pipe_surface *src,
 574                                  unsigned srcx, unsigned srcy,
 575                                  unsigned width, unsigned height,
 576                                  boolean is_depth)
 577 {
 578    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 579    struct pipe_context *pipe = ctx->pipe;
 580    struct pipe_framebuffer_state fb_state;
 581
 582    assert(blitter->saved_fb_state.nr_cbufs != ~0);
 583    assert(blitter->saved_num_textures != ~0);
 584    assert(blitter->saved_num_sampler_states != ~0);
 585    assert(src->texture->target < PIPE_MAX_TEXTURE_TYPES);
 586
 587    /* bind CSOs */
 588    fb_state.width = dst->width;
 589    fb_state.height = dst->height;
 590
 591    if (is_depth) {
 592       pipe->bind_blend_state(pipe, ctx->blend_keep_color);
 593       pipe->bind_depth_stencil_alpha_state(pipe,
 594                                            ctx->dsa_write_depth_keep_stencil);
 595       pipe->bind_fs_state(pipe,
 596          blitter_get_fs_texfetch_depth(ctx, src->texture->target));
 597
 598       fb_state.nr_cbufs = 0;
 599       fb_state.zsbuf = dst;
 600    } else {
 601       pipe->bind_blend_state(pipe, ctx->blend_write_color);
 602       pipe->bind_depth_stencil_alpha_state(pipe, ctx->dsa_keep_depth_stencil);
 603       pipe->bind_fs_state(pipe,
 604          blitter_get_fs_texfetch_col(ctx, src->texture->target));
 605
 606       fb_state.nr_cbufs = 1;
 607       fb_state.cbufs[0] = dst;
 608       fb_state.zsbuf = 0;
 609    }
 610
 611    pipe->bind_rasterizer_state(pipe, ctx->rs_state);
 612    pipe->bind_vs_state(pipe, ctx->vs_tex);
 613    pipe->bind_fragment_sampler_states(pipe, 1,
 614       blitter_get_sampler_state(ctx, src->level));
 615    pipe->set_fragment_sampler_textures(pipe, 1, &src->texture);
 616    pipe->set_framebuffer_state(pipe, &fb_state);
 617
 618    /* set texture coordinates */
 619    switch (src->texture->target) {
 620       case PIPE_TEXTURE_1D:
 621       case PIPE_TEXTURE_2D:
 622          blitter_set_texcoords_2d(ctx, src, srcx, srcy,
 623                                   srcx+width, srcy+height);
 624          break;
 625       case PIPE_TEXTURE_3D:
 626          blitter_set_texcoords_3d(ctx, src, srcx, srcy,
 627                                   srcx+width, srcy+height);
 628          break;
 629       case PIPE_TEXTURE_CUBE:
 630          blitter_set_texcoords_cube(ctx, src, srcx, srcy,
 631                                     srcx+width, srcy+height);
 632          break;
 633       default:
 634          assert(0);
 635    }
 636
 637    blitter_set_rectangle(ctx, dstx, dsty, dstx+width, dsty+height, 0);
 638    blitter_draw_quad(ctx);
 639
 640 }
 641
 642 static void util_blitter_overlap_copy(struct blitter_context *blitter,
 643                                       struct pipe_surface *dst,
 644                                       unsigned dstx, unsigned dsty,
 645                                       struct pipe_surface *src,
 646                                       unsigned srcx, unsigned srcy,
 647                                       unsigned width, unsigned height)
 648 {
 649    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 650    struct pipe_context *pipe = ctx->pipe;
 651    struct pipe_screen *screen = pipe->screen;
 652
 653    struct pipe_texture texTemp;
 654    struct pipe_texture *texture;
 655    struct pipe_surface *tex_surf;
 656
 657    /* check whether the states are properly saved */
 658    blitter_check_saved_CSOs(ctx);
 659
 660    memset(&texTemp, 0, sizeof(texTemp));
 661    texTemp.target = PIPE_TEXTURE_2D;
 662    texTemp.format = dst->texture->format; /* XXX verify supported by driver! */
 663    texTemp.last_level = 0;
 664    texTemp.width0 = width;
 665    texTemp.height0 = height;
 666    texTemp.depth0 = 1;
 667
 668    texture = screen->texture_create(screen, &texTemp);
 669    if (!texture)
 670       return;
 671
 672    tex_surf = screen->get_tex_surface(screen, texture, 0, 0, 0,
 673                                       PIPE_BUFFER_USAGE_GPU_READ |
 674                                       PIPE_BUFFER_USAGE_GPU_WRITE);
 675
 676    /* blit from the src to the temp */
 677    util_blitter_do_copy(blitter, tex_surf, 0, 0,
 678                         src, srcx, srcy,
 679                         width, height,
 680                         FALSE);
 681    util_blitter_do_copy(blitter, dst, dstx, dsty,
 682                         tex_surf, 0, 0,
 683                         width, height,
 684                         FALSE);
 685    pipe_surface_reference(&tex_surf, NULL);
 686    pipe_texture_reference(&texture, NULL);
 687    blitter_restore_CSOs(ctx);
 688 }
 689
 690 void util_blitter_copy(struct blitter_context *blitter,
 691                        struct pipe_surface *dst,
 692                        unsigned dstx, unsigned dsty,
 693                        struct pipe_surface *src,
 694                        unsigned srcx, unsigned srcy,
 695                        unsigned width, unsigned height,
 696                        boolean ignore_stencil)
 697 {
 698    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 699    struct pipe_context *pipe = ctx->pipe;
 700    struct pipe_screen *screen = pipe->screen;
 701    boolean is_stencil, is_depth;
 702    unsigned dst_tex_usage;
 703
 704    /* give up if textures are not set */
 705    assert(dst->texture && src->texture);
 706    if (!dst->texture || !src->texture)
 707       return;
 708
 709    if (dst->texture == src->texture) {
 710       if (is_overlap(srcx, srcx + width, srcy, srcy + height,
 711                              dstx, dstx + width, dsty, dsty + height)) {
 712          util_blitter_overlap_copy(blitter, dst, dstx, dsty, src, srcx, srcy,
 713                                    width, height);
 714          return;
 715       }
 716    }
 717
 718    is_depth = util_format_get_component_bits(src->format, UTIL_FORMAT_COLORSPACE_ZS, 0) != 0;
 719    is_stencil = util_format_get_component_bits(src->format, UTIL_FORMAT_COLORSPACE_ZS, 1) != 0;
 720    dst_tex_usage = is_depth || is_stencil ? PIPE_TEXTURE_USAGE_DEPTH_STENCIL :
 721                                             PIPE_TEXTURE_USAGE_RENDER_TARGET;
 722
 723    /* check if we can sample from and render to the surfaces */
 724    /* (assuming copying a stencil buffer is not possible) */
 725    if ((!ignore_stencil && is_stencil) ||
 726        !screen->is_format_supported(screen, dst->format, dst->texture->target,
 727                                     dst_tex_usage, 0) ||
 728        !screen->is_format_supported(screen, src->format, src->texture->target,
 729                                     PIPE_TEXTURE_USAGE_SAMPLER, 0)) {
 730       util_surface_copy(pipe, FALSE, dst, dstx, dsty, src, srcx, srcy,
 731                         width, height);
 732       return;
 733    }
 734
 735    /* check whether the states are properly saved */
 736    blitter_check_saved_CSOs(ctx);
 737    util_blitter_do_copy(blitter,
 738                         dst, dstx, dsty,
 739                         src, srcx, srcy,
 740                         width, height, is_depth);
 741    blitter_restore_CSOs(ctx);
 742 }
 743
 744 void util_blitter_fill(struct blitter_context *blitter,
 745                        struct pipe_surface *dst,
 746                        unsigned dstx, unsigned dsty,
 747                        unsigned width, unsigned height,
 748                        unsigned value)
 749 {
 750    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 751    struct pipe_context *pipe = ctx->pipe;
 752    struct pipe_screen *screen = pipe->screen;
 753    struct pipe_framebuffer_state fb_state;
 754    float rgba[4];
 755    ubyte ub_rgba[4] = {0};
 756    union util_color color;
 757    int i;
 758
 759    assert(dst->texture);
 760    if (!dst->texture)
 761       return;
 762
 763    /* check if we can render to the surface */
 764    if (util_format_is_depth_or_stencil(dst->format) || /* unlikely, but you never know */
 765        !screen->is_format_supported(screen, dst->format, dst->texture->target,
 766                                     PIPE_TEXTURE_USAGE_RENDER_TARGET, 0)) {
 767       util_surface_fill(pipe, dst, dstx, dsty, width, height, value);
 768       return;
 769    }
 770
 771    /* unpack the color */
 772    color.ui = value;
 773    util_unpack_color_ub(dst->format, &color,
 774                         ub_rgba, ub_rgba+1, ub_rgba+2, ub_rgba+3);
 775    for (i = 0; i < 4; i++)
 776       rgba[i] = ubyte_to_float(ub_rgba[i]);
 777
 778    /* check the saved state */
 779    blitter_check_saved_CSOs(ctx);
 780    assert(blitter->saved_fb_state.nr_cbufs != ~0);
 781
 782    /* bind CSOs */
 783    pipe->bind_blend_state(pipe, ctx->blend_write_color);
 784    pipe->bind_depth_stencil_alpha_state(pipe, ctx->dsa_keep_depth_stencil);
 785    pipe->bind_rasterizer_state(pipe, ctx->rs_state);
 786    pipe->bind_fs_state(pipe, blitter_get_fs_col(ctx, 1));
 787    pipe->bind_vs_state(pipe, ctx->vs_col);
 788
 789    /* set a framebuffer state */
 790    fb_state.width = dst->width;
 791    fb_state.height = dst->height;
 792    fb_state.nr_cbufs = 1;
 793    fb_state.cbufs[0] = dst;
 794    fb_state.zsbuf = 0;
 795    pipe->set_framebuffer_state(pipe, &fb_state);
 796
 797    blitter_set_clear_color(ctx, rgba);
 798    blitter_set_rectangle(ctx, 0, 0, width, height, 0);
 799    blitter_draw_quad(ctx);
 800    blitter_restore_CSOs(ctx);
 801 }