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, resource_copy_region,
  30  * and resource_fill_region 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_sampler.h"
  48 #include "util/u_simple_shaders.h"
  49 #include "util/u_surface.h"
  50 #include "util/u_texture.h"
  51
  52 #define INVALID_PTR ((void*)~0)
  53
  54 struct blitter_context_priv
  55 {
  56    struct blitter_context blitter;
  57
  58    struct pipe_context *pipe; /**< pipe context */
  59    struct pipe_resource *vbuf;  /**< quad */
  60
  61    float vertices[4][2][4];   /**< {pos, color} or {pos, texcoord} */
  62
  63    /* Templates for various state objects. */
  64    struct pipe_sampler_state template_sampler_state;
  65
  66    /* Constant state objects. */
  67    /* Vertex shaders. */
  68    void *vs_col; /**< Vertex shader which passes {pos, color} to the output */
  69    void *vs_tex; /**< Vertex shader which passes {pos, texcoord} to the output.*/
  70
  71    /* Fragment shaders. */
  72    /* The shader at index i outputs color to color buffers 0,1,...,i-1. */
  73    void *fs_col[PIPE_MAX_COLOR_BUFS+1];
  74
  75    /* FS which outputs a color from a texture,
  76       where the index is PIPE_TEXTURE_* to be sampled. */
  77    void *fs_texfetch_col[PIPE_MAX_TEXTURE_TYPES];
  78
  79    /* FS which outputs a depth from a texture,
  80       where the index is PIPE_TEXTURE_* to be sampled. */
  81    void *fs_texfetch_depth[PIPE_MAX_TEXTURE_TYPES];
  82
  83    /* Blend state. */
  84    void *blend_write_color;   /**< blend state with writemask of RGBA */
  85    void *blend_keep_color;    /**< blend state with writemask of 0 */
  86
  87    /* Depth stencil alpha state. */
  88    void *dsa_write_depth_stencil;
  89    void *dsa_write_depth_keep_stencil;
  90    void *dsa_keep_depth_stencil;
  91
  92    void *velem_state;
  93
  94    /* Sampler state for clamping to a miplevel. */
  95    void *sampler_state[PIPE_MAX_TEXTURE_LEVELS];
  96
  97    /* Rasterizer state. */
  98    void *rs_state;
  99
 100    struct pipe_sampler_view *sampler_view;
 101
 102    /* Viewport state. */
 103    struct pipe_viewport_state viewport;
 104
 105    /* Clip state. */
 106    struct pipe_clip_state clip;
 107 };
 108
 109 struct blitter_context *util_blitter_create(struct pipe_context *pipe)
 110 {
 111    struct blitter_context_priv *ctx;
 112    struct pipe_blend_state blend;
 113    struct pipe_depth_stencil_alpha_state dsa;
 114    struct pipe_rasterizer_state rs_state;
 115    struct pipe_sampler_state *sampler_state;
 116    struct pipe_vertex_element velem[2];
 117    unsigned i;
 118
 119    ctx = CALLOC_STRUCT(blitter_context_priv);
 120    if (!ctx)
 121       return NULL;
 122
 123    ctx->pipe = pipe;
 124
 125    /* init state objects for them to be considered invalid */
 126    ctx->blitter.saved_blend_state = INVALID_PTR;
 127    ctx->blitter.saved_dsa_state = INVALID_PTR;
 128    ctx->blitter.saved_rs_state = INVALID_PTR;
 129    ctx->blitter.saved_fs = INVALID_PTR;
 130    ctx->blitter.saved_vs = INVALID_PTR;
 131    ctx->blitter.saved_velem_state = INVALID_PTR;
 132    ctx->blitter.saved_fb_state.nr_cbufs = ~0;
 133    ctx->blitter.saved_num_sampler_views = ~0;
 134    ctx->blitter.saved_num_sampler_states = ~0;
 135    ctx->blitter.saved_num_vertex_buffers = ~0;
 136
 137    /* blend state objects */
 138    memset(&blend, 0, sizeof(blend));
 139    ctx->blend_keep_color = pipe->create_blend_state(pipe, &blend);
 140
 141    blend.rt[0].colormask = PIPE_MASK_RGBA;
 142    ctx->blend_write_color = pipe->create_blend_state(pipe, &blend);
 143
 144    /* depth stencil alpha state objects */
 145    memset(&dsa, 0, sizeof(dsa));
 146    ctx->dsa_keep_depth_stencil =
 147       pipe->create_depth_stencil_alpha_state(pipe, &dsa);
 148
 149    dsa.depth.enabled = 1;
 150    dsa.depth.writemask = 1;
 151    dsa.depth.func = PIPE_FUNC_ALWAYS;
 152    ctx->dsa_write_depth_keep_stencil =
 153       pipe->create_depth_stencil_alpha_state(pipe, &dsa);
 154
 155    dsa.stencil[0].enabled = 1;
 156    dsa.stencil[0].func = PIPE_FUNC_ALWAYS;
 157    dsa.stencil[0].fail_op = PIPE_STENCIL_OP_REPLACE;
 158    dsa.stencil[0].zpass_op = PIPE_STENCIL_OP_REPLACE;
 159    dsa.stencil[0].zfail_op = PIPE_STENCIL_OP_REPLACE;
 160    dsa.stencil[0].valuemask = 0xff;
 161    dsa.stencil[0].writemask = 0xff;
 162    ctx->dsa_write_depth_stencil =
 163       pipe->create_depth_stencil_alpha_state(pipe, &dsa);
 164    /* The DSA state objects which write depth and stencil are created
 165     * on-demand. */
 166
 167    /* sampler state */
 168    sampler_state = &ctx->template_sampler_state;
 169    sampler_state->wrap_s = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 170    sampler_state->wrap_t = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 171    sampler_state->wrap_r = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 172    sampler_state->normalized_coords = TRUE;
 173    /* The sampler state objects which sample from a specified mipmap level
 174     * are created on-demand. */
 175
 176    /* rasterizer state */
 177    memset(&rs_state, 0, sizeof(rs_state));
 178    rs_state.cull_face = PIPE_FACE_NONE;
 179    rs_state.gl_rasterization_rules = 1;
 180    rs_state.flatshade = 1;
 181    ctx->rs_state = pipe->create_rasterizer_state(pipe, &rs_state);
 182
 183    /* vertex elements state */
 184    memset(&velem[0], 0, sizeof(velem[0]) * 2);
 185    for (i = 0; i < 2; i++) {
 186       velem[i].src_offset = i * 4 * sizeof(float);
 187       velem[i].instance_divisor = 0;
 188       velem[i].vertex_buffer_index = 0;
 189       velem[i].src_format = PIPE_FORMAT_R32G32B32A32_FLOAT;
 190    }
 191    ctx->velem_state = pipe->create_vertex_elements_state(pipe, 2, &velem[0]);
 192
 193    /* fragment shaders are created on-demand */
 194
 195    /* vertex shaders */
 196    {
 197       const uint semantic_names[] = { TGSI_SEMANTIC_POSITION,
 198                                       TGSI_SEMANTIC_COLOR };
 199       const uint semantic_indices[] = { 0, 0 };
 200       ctx->vs_col =
 201          util_make_vertex_passthrough_shader(pipe, 2, semantic_names,
 202                                              semantic_indices);
 203    }
 204    {
 205       const uint semantic_names[] = { TGSI_SEMANTIC_POSITION,
 206                                       TGSI_SEMANTIC_GENERIC };
 207       const uint semantic_indices[] = { 0, 0 };
 208       ctx->vs_tex =
 209          util_make_vertex_passthrough_shader(pipe, 2, semantic_names,
 210                                              semantic_indices);
 211    }
 212
 213    /* set invariant vertex coordinates */
 214    for (i = 0; i < 4; i++)
 215       ctx->vertices[i][0][3] = 1; /*v.w*/
 216
 217    /* create the vertex buffer */
 218    ctx->vbuf = pipe_buffer_create(ctx->pipe->screen,
 219                                   PIPE_BIND_VERTEX_BUFFER,
 220                                   sizeof(ctx->vertices));
 221
 222    return &ctx->blitter;
 223 }
 224
 225 void util_blitter_destroy(struct blitter_context *blitter)
 226 {
 227    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 228    struct pipe_context *pipe = ctx->pipe;
 229    int i;
 230
 231    pipe->delete_blend_state(pipe, ctx->blend_write_color);
 232    pipe->delete_blend_state(pipe, ctx->blend_keep_color);
 233    pipe->delete_depth_stencil_alpha_state(pipe, ctx->dsa_keep_depth_stencil);
 234    pipe->delete_depth_stencil_alpha_state(pipe,
 235                                           ctx->dsa_write_depth_keep_stencil);
 236    pipe->delete_depth_stencil_alpha_state(pipe, ctx->dsa_write_depth_stencil);
 237
 238    pipe->delete_rasterizer_state(pipe, ctx->rs_state);
 239    pipe->delete_vs_state(pipe, ctx->vs_col);
 240    pipe->delete_vs_state(pipe, ctx->vs_tex);
 241    pipe->delete_vertex_elements_state(pipe, ctx->velem_state);
 242
 243    for (i = 0; i < PIPE_MAX_TEXTURE_TYPES; i++) {
 244       if (ctx->fs_texfetch_col[i])
 245          pipe->delete_fs_state(pipe, ctx->fs_texfetch_col[i]);
 246       if (ctx->fs_texfetch_depth[i])
 247          pipe->delete_fs_state(pipe, ctx->fs_texfetch_depth[i]);
 248    }
 249
 250    for (i = 0; i <= PIPE_MAX_COLOR_BUFS && ctx->fs_col[i]; i++)
 251       if (ctx->fs_col[i])
 252          pipe->delete_fs_state(pipe, ctx->fs_col[i]);
 253
 254    for (i = 0; i < PIPE_MAX_TEXTURE_LEVELS; i++)
 255       if (ctx->sampler_state[i])
 256          pipe->delete_sampler_state(pipe, ctx->sampler_state[i]);
 257
 258    if (ctx->sampler_view) {
 259       pipe_sampler_view_reference(&ctx->sampler_view, NULL);
 260    }
 261
 262    pipe_resource_reference(&ctx->vbuf, NULL);
 263    FREE(ctx);
 264 }
 265
 266 static void blitter_check_saved_CSOs(struct blitter_context_priv *ctx)
 267 {
 268    /* make sure these CSOs have been saved */
 269    assert(ctx->blitter.saved_blend_state != INVALID_PTR &&
 270           ctx->blitter.saved_dsa_state != INVALID_PTR &&
 271           ctx->blitter.saved_rs_state != INVALID_PTR &&
 272           ctx->blitter.saved_fs != INVALID_PTR &&
 273           ctx->blitter.saved_vs != INVALID_PTR &&
 274           ctx->blitter.saved_velem_state != INVALID_PTR);
 275 }
 276
 277 static void blitter_restore_CSOs(struct blitter_context_priv *ctx)
 278 {
 279    struct pipe_context *pipe = ctx->pipe;
 280
 281    /* restore the state objects which are always required to be saved */
 282    pipe->bind_blend_state(pipe, ctx->blitter.saved_blend_state);
 283    pipe->bind_depth_stencil_alpha_state(pipe, ctx->blitter.saved_dsa_state);
 284    pipe->bind_rasterizer_state(pipe, ctx->blitter.saved_rs_state);
 285    pipe->bind_fs_state(pipe, ctx->blitter.saved_fs);
 286    pipe->bind_vs_state(pipe, ctx->blitter.saved_vs);
 287    pipe->bind_vertex_elements_state(pipe, ctx->blitter.saved_velem_state);
 288
 289    ctx->blitter.saved_blend_state = INVALID_PTR;
 290    ctx->blitter.saved_dsa_state = INVALID_PTR;
 291    ctx->blitter.saved_rs_state = INVALID_PTR;
 292    ctx->blitter.saved_fs = INVALID_PTR;
 293    ctx->blitter.saved_vs = INVALID_PTR;
 294    ctx->blitter.saved_velem_state = INVALID_PTR;
 295
 296    pipe->set_stencil_ref(pipe, &ctx->blitter.saved_stencil_ref);
 297
 298    pipe->set_viewport_state(pipe, &ctx->blitter.saved_viewport);
 299    pipe->set_clip_state(pipe, &ctx->blitter.saved_clip);
 300
 301    /* restore the state objects which are required to be saved before copy/fill
 302     */
 303    if (ctx->blitter.saved_fb_state.nr_cbufs != ~0) {
 304       pipe->set_framebuffer_state(pipe, &ctx->blitter.saved_fb_state);
 305       ctx->blitter.saved_fb_state.nr_cbufs = ~0;
 306    }
 307
 308    if (ctx->blitter.saved_num_sampler_states != ~0) {
 309       pipe->bind_fragment_sampler_states(pipe,
 310                                          ctx->blitter.saved_num_sampler_states,
 311                                          ctx->blitter.saved_sampler_states);
 312       ctx->blitter.saved_num_sampler_states = ~0;
 313    }
 314
 315    if (ctx->blitter.saved_num_sampler_views != ~0) {
 316       pipe->set_fragment_sampler_views(pipe,
 317                                        ctx->blitter.saved_num_sampler_views,
 318                                        ctx->blitter.saved_sampler_views);
 319       ctx->blitter.saved_num_sampler_views = ~0;
 320    }
 321
 322    if (ctx->blitter.saved_num_vertex_buffers != ~0) {
 323       pipe->set_vertex_buffers(pipe,
 324                                ctx->blitter.saved_num_vertex_buffers,
 325                                ctx->blitter.saved_vertex_buffers);
 326       ctx->blitter.saved_num_vertex_buffers = ~0;
 327    }
 328 }
 329
 330 static void blitter_set_rectangle(struct blitter_context_priv *ctx,
 331                                   unsigned x1, unsigned y1,
 332                                   unsigned x2, unsigned y2,
 333                                   unsigned width, unsigned height,
 334                                   float depth)
 335 {
 336    int i;
 337
 338    /* set vertex positions */
 339    ctx->vertices[0][0][0] = (float)x1 / width * 2.0f - 1.0f; /*v0.x*/
 340    ctx->vertices[0][0][1] = (float)y1 / height * 2.0f - 1.0f; /*v0.y*/
 341
 342    ctx->vertices[1][0][0] = (float)x2 / width * 2.0f - 1.0f; /*v1.x*/
 343    ctx->vertices[1][0][1] = (float)y1 / height * 2.0f - 1.0f; /*v1.y*/
 344
 345    ctx->vertices[2][0][0] = (float)x2 / width * 2.0f - 1.0f; /*v2.x*/
 346    ctx->vertices[2][0][1] = (float)y2 / height * 2.0f - 1.0f; /*v2.y*/
 347
 348    ctx->vertices[3][0][0] = (float)x1 / width * 2.0f - 1.0f; /*v3.x*/
 349    ctx->vertices[3][0][1] = (float)y2 / height * 2.0f - 1.0f; /*v3.y*/
 350
 351    for (i = 0; i < 4; i++)
 352       ctx->vertices[i][0][2] = depth; /*z*/
 353
 354    /* viewport */
 355    ctx->viewport.scale[0] = 0.5f * width;
 356    ctx->viewport.scale[1] = 0.5f * height;
 357    ctx->viewport.scale[2] = 1.0f;
 358    ctx->viewport.scale[3] = 1.0f;
 359    ctx->viewport.translate[0] = 0.5f * width;
 360    ctx->viewport.translate[1] = 0.5f * height;
 361    ctx->viewport.translate[2] = 0.0f;
 362    ctx->viewport.translate[3] = 0.0f;
 363    ctx->pipe->set_viewport_state(ctx->pipe, &ctx->viewport);
 364
 365    /* clip */
 366    ctx->pipe->set_clip_state(ctx->pipe, &ctx->clip);
 367 }
 368
 369 static void blitter_set_clear_color(struct blitter_context_priv *ctx,
 370                                     const float *rgba)
 371 {
 372    int i;
 373
 374    if (rgba) {
 375       for (i = 0; i < 4; i++) {
 376          ctx->vertices[i][1][0] = rgba[0];
 377          ctx->vertices[i][1][1] = rgba[1];
 378          ctx->vertices[i][1][2] = rgba[2];
 379          ctx->vertices[i][1][3] = rgba[3];
 380       }
 381    } else {
 382       for (i = 0; i < 4; i++) {
 383          ctx->vertices[i][1][0] = 0;
 384          ctx->vertices[i][1][1] = 0;
 385          ctx->vertices[i][1][2] = 0;
 386          ctx->vertices[i][1][3] = 0;
 387       }
 388    }
 389 }
 390
 391 static void blitter_set_texcoords_2d(struct blitter_context_priv *ctx,
 392                                      struct pipe_resource *src,
 393                                      struct pipe_subresource subsrc,
 394                                      unsigned x1, unsigned y1,
 395                                      unsigned x2, unsigned y2)
 396 {
 397    int i;
 398    float s1 = x1 / (float)u_minify(src->width0,  subsrc.level);
 399    float t1 = y1 / (float)u_minify(src->height0, subsrc.level);
 400    float s2 = x2 / (float)u_minify(src->width0,  subsrc.level);
 401    float t2 = y2 / (float)u_minify(src->height0, subsrc.level);
 402
 403    ctx->vertices[0][1][0] = s1; /*t0.s*/
 404    ctx->vertices[0][1][1] = t1; /*t0.t*/
 405
 406    ctx->vertices[1][1][0] = s2; /*t1.s*/
 407    ctx->vertices[1][1][1] = t1; /*t1.t*/
 408
 409    ctx->vertices[2][1][0] = s2; /*t2.s*/
 410    ctx->vertices[2][1][1] = t2; /*t2.t*/
 411
 412    ctx->vertices[3][1][0] = s1; /*t3.s*/
 413    ctx->vertices[3][1][1] = t2; /*t3.t*/
 414
 415    for (i = 0; i < 4; i++) {
 416       ctx->vertices[i][1][2] = 0; /*r*/
 417       ctx->vertices[i][1][3] = 1; /*q*/
 418    }
 419 }
 420
 421 static void blitter_set_texcoords_3d(struct blitter_context_priv *ctx,
 422                                      struct pipe_resource *src,
 423                                      struct pipe_subresource subsrc,
 424                                      unsigned zslice,
 425                                      unsigned x1, unsigned y1,
 426                                      unsigned x2, unsigned y2)
 427 {
 428    int i;
 429    float r = zslice / (float)u_minify(src->depth0, subsrc.level);
 430
 431    blitter_set_texcoords_2d(ctx, src, subsrc, x1, y1, x2, y2);
 432
 433    for (i = 0; i < 4; i++)
 434       ctx->vertices[i][1][2] = r; /*r*/
 435 }
 436
 437 static void blitter_set_texcoords_cube(struct blitter_context_priv *ctx,
 438                                        struct pipe_resource *src,
 439                                        struct pipe_subresource subsrc,
 440                                        unsigned x1, unsigned y1,
 441                                        unsigned x2, unsigned y2)
 442 {
 443    int i;
 444    float s1 = x1 / (float)u_minify(src->width0,  subsrc.level);
 445    float t1 = y1 / (float)u_minify(src->height0, subsrc.level);
 446    float s2 = x2 / (float)u_minify(src->width0,  subsrc.level);
 447    float t2 = y2 / (float)u_minify(src->height0, subsrc.level);
 448    float st[4][2];
 449
 450    st[0][0] = s1;
 451    st[0][1] = t1;
 452    st[1][0] = s2;
 453    st[1][1] = t1;
 454    st[2][0] = s2;
 455    st[2][1] = t2;
 456    st[3][0] = s1;
 457    st[3][1] = t2;
 458
 459    util_map_texcoords2d_onto_cubemap(subsrc.face,
 460                                      /* pointer, stride in floats */
 461                                      &st[0][0], 2,
 462                                      &ctx->vertices[0][1][0], 8);
 463
 464    for (i = 0; i < 4; i++)
 465       ctx->vertices[i][1][3] = 1; /*q*/
 466 }
 467
 468 static void blitter_draw_quad(struct blitter_context_priv *ctx)
 469 {
 470    struct pipe_context *pipe = ctx->pipe;
 471
 472    /* write vertices and draw them */
 473    pipe_buffer_write(pipe, ctx->vbuf,
 474                      0, sizeof(ctx->vertices), ctx->vertices);
 475
 476    util_draw_vertex_buffer(pipe, ctx->vbuf, 0, PIPE_PRIM_TRIANGLE_FAN,
 477                            4,  /* verts */
 478                            2); /* attribs/vert */
 479 }
 480
 481 static INLINE
 482 void **blitter_get_sampler_state(struct blitter_context_priv *ctx,
 483                                  int miplevel)
 484 {
 485    struct pipe_context *pipe = ctx->pipe;
 486    struct pipe_sampler_state *sampler_state = &ctx->template_sampler_state;
 487
 488    assert(miplevel < PIPE_MAX_TEXTURE_LEVELS);
 489
 490    /* Create the sampler state on-demand. */
 491    if (!ctx->sampler_state[miplevel]) {
 492       sampler_state->lod_bias = miplevel;
 493       sampler_state->min_lod = miplevel;
 494       sampler_state->max_lod = miplevel;
 495
 496       ctx->sampler_state[miplevel] = pipe->create_sampler_state(pipe,
 497                                                                 sampler_state);
 498    }
 499
 500    /* Return void** so that it can be passed to bind_fragment_sampler_states
 501     * directly. */
 502    return &ctx->sampler_state[miplevel];
 503 }
 504
 505 static INLINE
 506 void *blitter_get_fs_col(struct blitter_context_priv *ctx, unsigned num_cbufs)
 507 {
 508    struct pipe_context *pipe = ctx->pipe;
 509
 510    assert(num_cbufs <= PIPE_MAX_COLOR_BUFS);
 511
 512    if (!ctx->fs_col[num_cbufs])
 513       ctx->fs_col[num_cbufs] =
 514          util_make_fragment_clonecolor_shader(pipe, num_cbufs);
 515
 516    return ctx->fs_col[num_cbufs];
 517 }
 518
 519 /** Convert PIPE_TEXTURE_x to TGSI_TEXTURE_x */
 520 static unsigned
 521 pipe_tex_to_tgsi_tex(unsigned pipe_tex_target)
 522 {
 523    switch (pipe_tex_target) {
 524    case PIPE_TEXTURE_1D:
 525       return TGSI_TEXTURE_1D;
 526    case PIPE_TEXTURE_2D:
 527       return TGSI_TEXTURE_2D;
 528    case PIPE_TEXTURE_3D:
 529       return TGSI_TEXTURE_3D;
 530    case PIPE_TEXTURE_CUBE:
 531       return TGSI_TEXTURE_CUBE;
 532    default:
 533       assert(0 && "unexpected texture target");
 534       return TGSI_TEXTURE_UNKNOWN;
 535    }
 536 }
 537
 538
 539 static INLINE
 540 void *blitter_get_fs_texfetch_col(struct blitter_context_priv *ctx,
 541                                   unsigned tex_target)
 542 {
 543    struct pipe_context *pipe = ctx->pipe;
 544
 545    assert(tex_target < PIPE_MAX_TEXTURE_TYPES);
 546
 547    /* Create the fragment shader on-demand. */
 548    if (!ctx->fs_texfetch_col[tex_target]) {
 549       unsigned tgsi_tex = pipe_tex_to_tgsi_tex(tex_target);
 550
 551       ctx->fs_texfetch_col[tex_target] =
 552         util_make_fragment_tex_shader(pipe, tgsi_tex, TGSI_INTERPOLATE_LINEAR);
 553    }
 554
 555    return ctx->fs_texfetch_col[tex_target];
 556 }
 557
 558 static INLINE
 559 void *blitter_get_fs_texfetch_depth(struct blitter_context_priv *ctx,
 560                                     unsigned tex_target)
 561 {
 562    struct pipe_context *pipe = ctx->pipe;
 563
 564    assert(tex_target < PIPE_MAX_TEXTURE_TYPES);
 565
 566    /* Create the fragment shader on-demand. */
 567    if (!ctx->fs_texfetch_depth[tex_target]) {
 568       unsigned tgsi_tex = pipe_tex_to_tgsi_tex(tex_target);
 569
 570       ctx->fs_texfetch_depth[tex_target] =
 571          util_make_fragment_tex_shader_writedepth(pipe, tgsi_tex,
 572                                                   TGSI_INTERPOLATE_LINEAR);
 573    }
 574
 575    return ctx->fs_texfetch_depth[tex_target];
 576 }
 577
 578 void util_blitter_clear(struct blitter_context *blitter,
 579                         unsigned width, unsigned height,
 580                         unsigned num_cbufs,
 581                         unsigned clear_buffers,
 582                         const float *rgba,
 583                         double depth, unsigned stencil)
 584 {
 585    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 586    struct pipe_context *pipe = ctx->pipe;
 587    struct pipe_stencil_ref sr = { { 0 } };
 588
 589    assert(num_cbufs <= PIPE_MAX_COLOR_BUFS);
 590
 591    blitter_check_saved_CSOs(ctx);
 592
 593    /* bind CSOs */
 594    if (clear_buffers & PIPE_CLEAR_COLOR)
 595       pipe->bind_blend_state(pipe, ctx->blend_write_color);
 596    else
 597       pipe->bind_blend_state(pipe, ctx->blend_keep_color);
 598
 599    if (clear_buffers & PIPE_CLEAR_DEPTHSTENCIL) {
 600       sr.ref_value[0] = stencil & 0xff;
 601       pipe->bind_depth_stencil_alpha_state(pipe, ctx->dsa_write_depth_stencil);
 602       pipe->set_stencil_ref(pipe, &sr);
 603    }
 604    else
 605       pipe->bind_depth_stencil_alpha_state(pipe, ctx->dsa_keep_depth_stencil);
 606
 607    pipe->bind_rasterizer_state(pipe, ctx->rs_state);
 608    pipe->bind_vertex_elements_state(pipe, ctx->velem_state);
 609    pipe->bind_fs_state(pipe, blitter_get_fs_col(ctx, num_cbufs));
 610    pipe->bind_vs_state(pipe, ctx->vs_col);
 611
 612    blitter_set_clear_color(ctx, rgba);
 613    blitter_set_rectangle(ctx, 0, 0, width, height, width, height, depth);
 614    blitter_draw_quad(ctx);
 615    blitter_restore_CSOs(ctx);
 616 }
 617
 618 static
 619 boolean is_overlap(unsigned sx1, unsigned sx2, unsigned sy1, unsigned sy2,
 620                    unsigned dx1, unsigned dx2, unsigned dy1, unsigned dy2)
 621 {
 622    return sx1 < dx2 && sx2 > dx1 && sy1 < dy2 && sy2 > dy1;
 623 }
 624
 625 void util_blitter_copy_region(struct blitter_context *blitter,
 626                               struct pipe_resource *dst,
 627                               struct pipe_subresource subdst,
 628                               unsigned dstx, unsigned dsty, unsigned dstz,
 629                               struct pipe_resource *src,
 630                               struct pipe_subresource subsrc,
 631                               unsigned srcx, unsigned srcy, unsigned srcz,
 632                               unsigned width, unsigned height,
 633                               boolean ignore_stencil)
 634 {
 635    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 636    struct pipe_context *pipe = ctx->pipe;
 637    struct pipe_screen *screen = pipe->screen;
 638    struct pipe_surface *dstsurf;
 639    struct pipe_framebuffer_state fb_state;
 640    struct pipe_sampler_view viewTempl, *view;
 641    unsigned bind;
 642    boolean is_stencil, is_depth;
 643
 644    /* Give up if textures are not set. */
 645    assert(dst && src);
 646    if (!dst || !src)
 647       return;
 648
 649    /* Sanity checks. */
 650    if (dst == src) {
 651       assert(!is_overlap(srcx, srcx + width, srcy, srcy + height,
 652                          dstx, dstx + width, dsty, dsty + height));
 653    } else {
 654       assert(dst->format == src->format);
 655    }
 656    assert(src->target < PIPE_MAX_TEXTURE_TYPES);
 657
 658    /* Is this a ZS format? */
 659    is_depth = util_format_get_component_bits(src->format, UTIL_FORMAT_COLORSPACE_ZS, 0) != 0;
 660    is_stencil = util_format_get_component_bits(src->format, UTIL_FORMAT_COLORSPACE_ZS, 1) != 0;
 661
 662    if (is_depth || is_stencil)
 663       bind = PIPE_BIND_DEPTH_STENCIL;
 664    else
 665       bind = PIPE_BIND_RENDER_TARGET;
 666
 667    /* Check if we can sample from and render to the surfaces. */
 668    /* (assuming copying a stencil buffer is not possible) */
 669     if ((!ignore_stencil && is_stencil) ||
 670        !screen->is_format_supported(screen, dst->format, dst->target,
 671                                     dst->nr_samples, bind, 0) ||
 672        !screen->is_format_supported(screen, src->format, src->target,
 673                                     src->nr_samples, PIPE_BIND_SAMPLER_VIEW, 0)) {
 674       util_resource_copy_region(pipe, dst, subdst, dstx, dsty, dstz,
 675                                 src, subsrc, srcx, srcy, srcz, width, height);
 676       return;
 677    }
 678
 679    /* Get surfaces. */
 680    dstsurf = screen->get_tex_surface(screen, dst,
 681                                      subdst.face, subdst.level, dstz,
 682                                      bind);
 683
 684    /* Check whether the states are properly saved. */
 685    blitter_check_saved_CSOs(ctx);
 686    assert(blitter->saved_fb_state.nr_cbufs != ~0);
 687    assert(blitter->saved_num_sampler_views != ~0);
 688    assert(blitter->saved_num_sampler_states != ~0);
 689
 690    /* Initialize framebuffer state. */
 691    fb_state.width = dstsurf->width;
 692    fb_state.height = dstsurf->height;
 693
 694    if (is_depth) {
 695       pipe->bind_blend_state(pipe, ctx->blend_keep_color);
 696       pipe->bind_depth_stencil_alpha_state(pipe,
 697                                            ctx->dsa_write_depth_keep_stencil);
 698       pipe->bind_fs_state(pipe,
 699                           blitter_get_fs_texfetch_depth(ctx, src->target));
 700
 701       fb_state.nr_cbufs = 0;
 702       fb_state.zsbuf = dstsurf;
 703    } else {
 704       pipe->bind_blend_state(pipe, ctx->blend_write_color);
 705       pipe->bind_depth_stencil_alpha_state(pipe, ctx->dsa_keep_depth_stencil);
 706       pipe->bind_fs_state(pipe,
 707                           blitter_get_fs_texfetch_col(ctx, src->target));
 708
 709       fb_state.nr_cbufs = 1;
 710       fb_state.cbufs[0] = dstsurf;
 711       fb_state.zsbuf = 0;
 712    }
 713
 714    /* Initialize sampler view. */
 715    u_sampler_view_default_template(&viewTempl, src, src->format);
 716    view = pipe->create_sampler_view(pipe, src, &viewTempl);
 717
 718    if (ctx->sampler_view) {
 719       pipe_sampler_view_reference(&ctx->sampler_view, NULL);
 720    }
 721    ctx->sampler_view = view;
 722
 723    /* Set rasterizer state, shaders, and textures. */
 724    pipe->bind_rasterizer_state(pipe, ctx->rs_state);
 725    pipe->bind_vs_state(pipe, ctx->vs_tex);
 726    pipe->bind_fragment_sampler_states(pipe, 1,
 727                                       blitter_get_sampler_state(ctx, subsrc.level));
 728    pipe->bind_vertex_elements_state(pipe, ctx->velem_state);
 729    pipe->set_fragment_sampler_views(pipe, 1, &view);
 730    pipe->set_framebuffer_state(pipe, &fb_state);
 731
 732    /* Set texture coordinates. */
 733    switch (src->target) {
 734       case PIPE_TEXTURE_1D:
 735       case PIPE_TEXTURE_2D:
 736          blitter_set_texcoords_2d(ctx, src, subsrc,
 737                                   srcx, srcy, srcx+width, srcy+height);
 738          break;
 739       case PIPE_TEXTURE_3D:
 740          blitter_set_texcoords_3d(ctx, src, subsrc, srcz,
 741                                   srcx, srcy, srcx+width, srcy+height);
 742          break;
 743       case PIPE_TEXTURE_CUBE:
 744          blitter_set_texcoords_cube(ctx, src, subsrc,
 745                                     srcx, srcy, srcx+width, srcy+height);
 746          break;
 747       default:
 748          assert(0);
 749          return;
 750    }
 751
 752    blitter_set_rectangle(ctx, dstx, dsty, dstx+width, dsty+height,
 753                          dstsurf->width, dstsurf->height, 0);
 754    blitter_draw_quad(ctx);
 755    blitter_restore_CSOs(ctx);
 756
 757    pipe_surface_reference(&dstsurf, NULL);
 758 }
 759
 760 /* Fill a region of a surface with a constant value. */
 761 void util_blitter_fill_region(struct blitter_context *blitter,
 762                               struct pipe_resource *dst,
 763                               struct pipe_subresource subdst,
 764                               unsigned dstx, unsigned dsty, unsigned dstz,
 765                               unsigned width, unsigned height,
 766                               unsigned value)
 767 {
 768    struct blitter_context_priv *ctx = (struct blitter_context_priv*)blitter;
 769    struct pipe_context *pipe = ctx->pipe;
 770    struct pipe_screen *screen = pipe->screen;
 771    struct pipe_surface *dstsurf;
 772    struct pipe_framebuffer_state fb_state;
 773    float rgba[4];
 774    ubyte ub_rgba[4] = {0};
 775    union util_color color;
 776    int i;
 777
 778    assert(dst);
 779    if (!dst)
 780       return;
 781
 782    /* check if we can render to the surface */
 783    if (util_format_is_depth_or_stencil(dst->format) || /* unlikely, but you never know */
 784        !screen->is_format_supported(screen, dst->format, dst->target,
 785                                     dst->nr_samples,
 786                                     PIPE_BIND_RENDER_TARGET, 0)) {
 787       util_resource_fill_region(pipe, dst, subdst, dstx, dsty, dstz,
 788                                 width, height, value);
 789       return;
 790    }
 791
 792    dstsurf = screen->get_tex_surface(screen, dst, subdst.face, subdst.level,
 793                                      dstz, PIPE_BIND_RENDER_TARGET);
 794
 795    /* unpack the color */
 796    color.ui = value;
 797    util_unpack_color_ub(dst->format, &color,
 798                         ub_rgba, ub_rgba+1, ub_rgba+2, ub_rgba+3);
 799    for (i = 0; i < 4; i++)
 800       rgba[i] = ubyte_to_float(ub_rgba[i]);
 801
 802    /* check the saved state */
 803    blitter_check_saved_CSOs(ctx);
 804    assert(blitter->saved_fb_state.nr_cbufs != ~0);
 805
 806    /* bind CSOs */
 807    pipe->bind_blend_state(pipe, ctx->blend_write_color);
 808    pipe->bind_depth_stencil_alpha_state(pipe, ctx->dsa_keep_depth_stencil);
 809    pipe->bind_rasterizer_state(pipe, ctx->rs_state);
 810    pipe->bind_fs_state(pipe, blitter_get_fs_col(ctx, 1));
 811    pipe->bind_vs_state(pipe, ctx->vs_col);
 812    pipe->bind_vertex_elements_state(pipe, ctx->velem_state);
 813
 814    /* set a framebuffer state */
 815    fb_state.width = dstsurf->width;
 816    fb_state.height = dstsurf->height;
 817    fb_state.nr_cbufs = 1;
 818    fb_state.cbufs[0] = dstsurf;
 819    fb_state.zsbuf = 0;
 820    pipe->set_framebuffer_state(pipe, &fb_state);
 821
 822    blitter_set_clear_color(ctx, rgba);
 823    blitter_set_rectangle(ctx, 0, 0, width, height, dstsurf->width, dstsurf->height, 0);
 824    blitter_draw_quad(ctx);
 825    blitter_restore_CSOs(ctx);
 826
 827    pipe_surface_reference(&dstsurf, NULL);
 828 }