src/gallium/auxiliary/vl/vl_compositor.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2009 Younes Manton.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the
  15  * next paragraph) shall be included in all copies or substantial portions
  16  * of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  21  * IN NO EVENT SHALL TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR
  22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  **************************************************************************/
  27
  28 #include <assert.h>
  29
  30 #include <pipe/p_compiler.h>
  31 #include <pipe/p_context.h>
  32
  33 #include <util/u_memory.h>
  34 #include <util/u_draw.h>
  35 #include <util/u_surface.h>
  36
  37 #include <tgsi/tgsi_ureg.h>
  38
  39 #include "vl_csc.h"
  40 #include "vl_types.h"
  41 #include "vl_compositor.h"
  42
  43 typedef float csc_matrix[16];
  44
  45 static void *
  46 create_vert_shader(struct vl_compositor *c)
  47 {
  48    struct ureg_program *shader;
  49    struct ureg_src vpos, vtex;
  50    struct ureg_dst o_vpos, o_vtex;
  51
  52    shader = ureg_create(TGSI_PROCESSOR_VERTEX);
  53    if (!shader)
  54       return false;
  55
  56    vpos = ureg_DECL_vs_input(shader, 0);
  57    vtex = ureg_DECL_vs_input(shader, 1);
  58    o_vpos = ureg_DECL_output(shader, TGSI_SEMANTIC_POSITION, 0);
  59    o_vtex = ureg_DECL_output(shader, TGSI_SEMANTIC_GENERIC, 1);
  60
  61    /*
  62     * o_vpos = vpos
  63     * o_vtex = vtex
  64     */
  65    ureg_MOV(shader, o_vpos, vpos);
  66    ureg_MOV(shader, o_vtex, vtex);
  67
  68    ureg_END(shader);
  69
  70    return ureg_create_shader_and_destroy(shader, c->pipe);
  71 }
  72
  73 static void *
  74 create_frag_shader_video_buffer(struct vl_compositor *c)
  75 {
  76    struct ureg_program *shader;
  77    struct ureg_src tc;
  78    struct ureg_src csc[3];
  79    struct ureg_src sampler[3];
  80    struct ureg_dst texel;
  81    struct ureg_dst fragment;
  82    unsigned i;
  83
  84    shader = ureg_create(TGSI_PROCESSOR_FRAGMENT);
  85    if (!shader)
  86       return false;
  87
  88    tc = ureg_DECL_fs_input(shader, TGSI_SEMANTIC_GENERIC, 1, TGSI_INTERPOLATE_LINEAR);
  89    for (i = 0; i < 3; ++i) {
  90       csc[i] = ureg_DECL_constant(shader, i);
  91       sampler[i] = ureg_DECL_sampler(shader, i);
  92    }
  93    texel = ureg_DECL_temporary(shader);
  94    fragment = ureg_DECL_output(shader, TGSI_SEMANTIC_COLOR, 0);
  95
  96    /*
  97     * texel.xyz = tex(tc, sampler[i])
  98     * fragment = csc * texel
  99     */
 100    for (i = 0; i < 3; ++i)
 101       ureg_TEX(shader, ureg_writemask(texel, TGSI_WRITEMASK_X << i), TGSI_TEXTURE_2D, tc, sampler[i]);
 102
 103    ureg_MOV(shader, ureg_writemask(texel, TGSI_WRITEMASK_W), ureg_imm1f(shader, 1.0f));
 104
 105    for (i = 0; i < 3; ++i)
 106       ureg_DP4(shader, ureg_writemask(fragment, TGSI_WRITEMASK_X << i), csc[i], ureg_src(texel));
 107
 108    ureg_MOV(shader, ureg_writemask(fragment, TGSI_WRITEMASK_W), ureg_imm1f(shader, 1.0f));
 109
 110    ureg_release_temporary(shader, texel);
 111    ureg_END(shader);
 112
 113    return ureg_create_shader_and_destroy(shader, c->pipe);
 114 }
 115
 116 static void *
 117 create_frag_shader_palette(struct vl_compositor *c)
 118 {
 119    struct ureg_program *shader;
 120    struct ureg_src csc[3];
 121    struct ureg_src tc;
 122    struct ureg_src sampler;
 123    struct ureg_src palette;
 124    struct ureg_dst texel;
 125    struct ureg_dst fragment;
 126    unsigned i;
 127
 128    shader = ureg_create(TGSI_PROCESSOR_FRAGMENT);
 129    if (!shader)
 130       return false;
 131
 132    for (i = 0; i < 3; ++i)
 133       csc[i] = ureg_DECL_constant(shader, i);
 134
 135    tc = ureg_DECL_fs_input(shader, TGSI_SEMANTIC_GENERIC, 1, TGSI_INTERPOLATE_LINEAR);
 136    sampler = ureg_DECL_sampler(shader, 0);
 137    palette = ureg_DECL_sampler(shader, 1);
 138    texel = ureg_DECL_temporary(shader);
 139    fragment = ureg_DECL_output(shader, TGSI_SEMANTIC_COLOR, 0);
 140
 141    /*
 142     * texel = tex(tc, sampler)
 143     * fragment.xyz = tex(texel, palette) * csc
 144     * fragment.a = texel.a
 145     */
 146    ureg_TEX(shader, texel, TGSI_TEXTURE_2D, tc, sampler);
 147    ureg_MUL(shader, ureg_writemask(texel, TGSI_WRITEMASK_X), ureg_src(texel), ureg_imm1f(shader, 15.0f / 16.0f));
 148    ureg_MOV(shader, ureg_writemask(fragment, TGSI_WRITEMASK_W), ureg_src(texel));
 149
 150    ureg_TEX(shader, texel, TGSI_TEXTURE_1D, ureg_src(texel), palette);
 151
 152    for (i = 0; i < 3; ++i)
 153       ureg_DP4(shader, ureg_writemask(fragment, TGSI_WRITEMASK_X << i), csc[i], ureg_src(texel));
 154
 155    ureg_release_temporary(shader, texel);
 156    ureg_END(shader);
 157
 158    return ureg_create_shader_and_destroy(shader, c->pipe);
 159 }
 160
 161 static void *
 162 create_frag_shader_rgba(struct vl_compositor *c)
 163 {
 164    struct ureg_program *shader;
 165    struct ureg_src tc;
 166    struct ureg_src sampler;
 167    struct ureg_dst fragment;
 168
 169    shader = ureg_create(TGSI_PROCESSOR_FRAGMENT);
 170    if (!shader)
 171       return false;
 172
 173    tc = ureg_DECL_fs_input(shader, TGSI_SEMANTIC_GENERIC, 1, TGSI_INTERPOLATE_LINEAR);
 174    sampler = ureg_DECL_sampler(shader, 0);
 175    fragment = ureg_DECL_output(shader, TGSI_SEMANTIC_COLOR, 0);
 176
 177    /*
 178     * fragment = tex(tc, sampler)
 179     */
 180    ureg_TEX(shader, fragment, TGSI_TEXTURE_2D, tc, sampler);
 181    ureg_END(shader);
 182
 183    return ureg_create_shader_and_destroy(shader, c->pipe);
 184 }
 185
 186 static bool
 187 init_shaders(struct vl_compositor *c)
 188 {
 189    assert(c);
 190
 191    c->vs = create_vert_shader(c);
 192    if (!c->vs) {
 193       debug_printf("Unable to create vertex shader.\n");
 194       return false;
 195    }
 196
 197    c->fs_video_buffer = create_frag_shader_video_buffer(c);
 198    if (!c->fs_video_buffer) {
 199       debug_printf("Unable to create YCbCr-to-RGB fragment shader.\n");
 200       return false;
 201    }
 202
 203    c->fs_palette = create_frag_shader_palette(c);
 204    if (!c->fs_palette) {
 205       debug_printf("Unable to create Palette-to-RGB fragment shader.\n");
 206       return false;
 207    }
 208
 209    c->fs_rgba = create_frag_shader_rgba(c);
 210    if (!c->fs_rgba) {
 211       debug_printf("Unable to create RGB-to-RGB fragment shader.\n");
 212       return false;
 213    }
 214
 215    return true;
 216 }
 217
 218 static void cleanup_shaders(struct vl_compositor *c)
 219 {
 220    assert(c);
 221
 222    c->pipe->delete_vs_state(c->pipe, c->vs);
 223    c->pipe->delete_fs_state(c->pipe, c->fs_video_buffer);
 224    c->pipe->delete_fs_state(c->pipe, c->fs_palette);
 225    c->pipe->delete_fs_state(c->pipe, c->fs_rgba);
 226 }
 227
 228 static bool
 229 init_pipe_state(struct vl_compositor *c)
 230 {
 231    struct pipe_rasterizer_state rast;
 232    struct pipe_sampler_state sampler;
 233    struct pipe_blend_state blend;
 234
 235    assert(c);
 236
 237    c->fb_state.nr_cbufs = 1;
 238    c->fb_state.zsbuf = NULL;
 239
 240    c->viewport.scale[2] = 1;
 241    c->viewport.scale[3] = 1;
 242    c->viewport.translate[2] = 0;
 243    c->viewport.translate[3] = 0;
 244
 245    memset(&sampler, 0, sizeof(sampler));
 246    sampler.wrap_s = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 247    sampler.wrap_t = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 248    sampler.wrap_r = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 249    sampler.min_img_filter = PIPE_TEX_FILTER_LINEAR;
 250    sampler.min_mip_filter = PIPE_TEX_MIPFILTER_NONE;
 251    sampler.mag_img_filter = PIPE_TEX_FILTER_LINEAR;
 252    sampler.compare_mode = PIPE_TEX_COMPARE_NONE;
 253    sampler.compare_func = PIPE_FUNC_ALWAYS;
 254    sampler.normalized_coords = 1;
 255
 256    c->sampler_linear = c->pipe->create_sampler_state(c->pipe, &sampler);
 257
 258    sampler.min_img_filter = PIPE_TEX_FILTER_NEAREST;
 259    sampler.mag_img_filter = PIPE_TEX_FILTER_NEAREST;
 260    c->sampler_nearest = c->pipe->create_sampler_state(c->pipe, &sampler);
 261
 262    memset(&blend, 0, sizeof blend);
 263    blend.independent_blend_enable = 0;
 264    blend.rt[0].blend_enable = 1;
 265    blend.rt[0].rgb_func = PIPE_BLEND_ADD;
 266    blend.rt[0].rgb_src_factor = PIPE_BLENDFACTOR_SRC_ALPHA;
 267    blend.rt[0].rgb_dst_factor = PIPE_BLENDFACTOR_INV_SRC_ALPHA;
 268    blend.rt[0].alpha_func = PIPE_BLEND_ADD;
 269    blend.rt[0].alpha_src_factor = PIPE_BLENDFACTOR_ONE;
 270    blend.rt[0].alpha_dst_factor = PIPE_BLENDFACTOR_ONE;
 271    blend.logicop_enable = 0;
 272    blend.logicop_func = PIPE_LOGICOP_CLEAR;
 273    blend.rt[0].colormask = PIPE_MASK_RGBA;
 274    blend.dither = 0;
 275    c->blend = c->pipe->create_blend_state(c->pipe, &blend);
 276
 277    memset(&rast, 0, sizeof rast);
 278    rast.flatshade = 1;
 279    rast.front_ccw = 1;
 280    rast.cull_face = PIPE_FACE_NONE;
 281    rast.fill_back = PIPE_POLYGON_MODE_FILL;
 282    rast.fill_front = PIPE_POLYGON_MODE_FILL;
 283    rast.scissor = 1;
 284    rast.line_width = 1;
 285    rast.point_size_per_vertex = 1;
 286    rast.offset_units = 1;
 287    rast.offset_scale = 1;
 288    rast.gl_rasterization_rules = 1;
 289
 290    c->rast = c->pipe->create_rasterizer_state(c->pipe, &rast);
 291
 292    return true;
 293 }
 294
 295 static void cleanup_pipe_state(struct vl_compositor *c)
 296 {
 297    assert(c);
 298
 299    c->pipe->delete_sampler_state(c->pipe, c->sampler_linear);
 300    c->pipe->delete_sampler_state(c->pipe, c->sampler_nearest);
 301    c->pipe->delete_blend_state(c->pipe, c->blend);
 302    c->pipe->delete_rasterizer_state(c->pipe, c->rast);
 303 }
 304
 305 static bool
 306 create_vertex_buffer(struct vl_compositor *c)
 307 {
 308    assert(c);
 309
 310    pipe_resource_reference(&c->vertex_buf.buffer, NULL);
 311    c->vertex_buf.buffer = pipe_buffer_create
 312    (
 313       c->pipe->screen,
 314       PIPE_BIND_VERTEX_BUFFER,
 315       PIPE_USAGE_STREAM,
 316       sizeof(struct vertex4f) * VL_COMPOSITOR_MAX_LAYERS * 4
 317    );
 318    return c->vertex_buf.buffer != NULL;
 319 }
 320
 321 static bool
 322 init_buffers(struct vl_compositor *c)
 323 {
 324    struct pipe_vertex_element vertex_elems[2];
 325
 326    assert(c);
 327
 328    /*
 329     * Create our vertex buffer and vertex buffer elements
 330     */
 331    c->vertex_buf.stride = sizeof(struct vertex4f);
 332    c->vertex_buf.buffer_offset = 0;
 333    create_vertex_buffer(c);
 334
 335    vertex_elems[0].src_offset = 0;
 336    vertex_elems[0].instance_divisor = 0;
 337    vertex_elems[0].vertex_buffer_index = 0;
 338    vertex_elems[0].src_format = PIPE_FORMAT_R32G32_FLOAT;
 339    vertex_elems[1].src_offset = sizeof(struct vertex2f);
 340    vertex_elems[1].instance_divisor = 0;
 341    vertex_elems[1].vertex_buffer_index = 0;
 342    vertex_elems[1].src_format = PIPE_FORMAT_R32G32_FLOAT;
 343    c->vertex_elems_state = c->pipe->create_vertex_elements_state(c->pipe, 2, vertex_elems);
 344
 345    /*
 346     * Create our fragment shader's constant buffer
 347     * Const buffer contains the color conversion matrix and bias vectors
 348     */
 349    /* XXX: Create with IMMUTABLE/STATIC... although it does change every once in a long while... */
 350    c->csc_matrix = pipe_buffer_create
 351    (
 352       c->pipe->screen,
 353       PIPE_BIND_CONSTANT_BUFFER,
 354       PIPE_USAGE_STATIC,
 355       sizeof(csc_matrix)
 356    );
 357
 358    return true;
 359 }
 360
 361 static void
 362 cleanup_buffers(struct vl_compositor *c)
 363 {
 364    assert(c);
 365
 366    c->pipe->delete_vertex_elements_state(c->pipe, c->vertex_elems_state);
 367    pipe_resource_reference(&c->vertex_buf.buffer, NULL);
 368    pipe_resource_reference(&c->csc_matrix, NULL);
 369 }
 370
 371 static INLINE struct pipe_video_rect
 372 default_rect(struct vl_compositor_layer *layer)
 373 {
 374    struct pipe_resource *res = layer->sampler_views[0]->texture;
 375    struct pipe_video_rect rect = { 0, 0, res->width0, res->height0 };
 376    return rect;
 377 }
 378
 379 static INLINE struct vertex2f
 380 calc_topleft(struct vertex2f size, struct pipe_video_rect rect)
 381 {
 382    struct vertex2f res = { rect.x / size.x, rect.y / size.y };
 383    return res;
 384 }
 385
 386 static INLINE struct vertex2f
 387 calc_bottomright(struct vertex2f size, struct pipe_video_rect rect)
 388 {
 389    struct vertex2f res = { (rect.x + rect.w) / size.x, (rect.y + rect.h) / size.y };
 390    return res;
 391 }
 392
 393 static INLINE void
 394 calc_src_and_dst(struct vl_compositor_layer *layer, unsigned width, unsigned height,
 395                  struct pipe_video_rect src, struct pipe_video_rect dst)
 396 {
 397    struct vertex2f size =  { width, height };
 398
 399    layer->src.tl = calc_topleft(size, src);
 400    layer->src.br = calc_bottomright(size, src);
 401    layer->dst.tl = calc_topleft(size, dst);
 402    layer->dst.br = calc_bottomright(size, dst);
 403 }
 404
 405 static void
 406 gen_rect_verts(struct vertex4f *vb, struct vl_compositor_layer *layer)
 407 {
 408    assert(vb && layer);
 409
 410    vb[0].x = layer->dst.tl.x;
 411    vb[0].y = layer->dst.tl.y;
 412    vb[0].z = layer->src.tl.x;
 413    vb[0].w = layer->src.tl.y;
 414
 415    vb[1].x = layer->dst.br.x;
 416    vb[1].y = layer->dst.tl.y;
 417    vb[1].z = layer->src.br.x;
 418    vb[1].w = layer->src.tl.y;
 419
 420    vb[2].x = layer->dst.br.x;
 421    vb[2].y = layer->dst.br.y;
 422    vb[2].z = layer->src.br.x;
 423    vb[2].w = layer->src.br.y;
 424
 425    vb[3].x = layer->dst.tl.x;
 426    vb[3].y = layer->dst.br.y;
 427    vb[3].z = layer->src.tl.x;
 428    vb[3].w = layer->src.br.y;
 429 }
 430
 431 static void
 432 gen_vertex_data(struct vl_compositor *c)
 433 {
 434    struct vertex4f *vb;
 435    struct pipe_transfer *buf_transfer;
 436    unsigned i;
 437
 438    assert(c);
 439
 440    vb = pipe_buffer_map(c->pipe, c->vertex_buf.buffer,
 441                         PIPE_TRANSFER_WRITE | PIPE_TRANSFER_DISCARD | PIPE_TRANSFER_DONTBLOCK,
 442                         &buf_transfer);
 443
 444    if (!vb) {
 445       // If buffer is still locked from last draw create a new one
 446       create_vertex_buffer(c);
 447       vb = pipe_buffer_map(c->pipe, c->vertex_buf.buffer,
 448                            PIPE_TRANSFER_WRITE | PIPE_TRANSFER_DISCARD,
 449                            &buf_transfer);
 450    }
 451
 452    for (i = 0; i < VL_COMPOSITOR_MAX_LAYERS; i++) {
 453       if (c->used_layers & (1 << i)) {
 454          struct vl_compositor_layer *layer = &c->layers[i];
 455          gen_rect_verts(vb, layer);
 456          vb += 4;
 457
 458          if (layer->clearing &&
 459              c->dirty_tl.x >= layer->dst.tl.x &&
 460              c->dirty_tl.y >= layer->dst.tl.y &&
 461              c->dirty_br.x <= layer->dst.br.x &&
 462              c->dirty_br.y <= layer->dst.br.y) {
 463
 464             // We clear the dirty area anyway, no need for clear_render_target
 465             c->dirty_tl.x = c->dirty_tl.y = 1.0f;
 466             c->dirty_br.x = c->dirty_br.y = 0.0f;
 467          }
 468       }
 469    }
 470
 471    pipe_buffer_unmap(c->pipe, buf_transfer);
 472 }
 473
 474 static void
 475 draw_layers(struct vl_compositor *c)
 476 {
 477    unsigned vb_index, i;
 478
 479    assert(c);
 480
 481    for (i = 0, vb_index = 0; i < VL_COMPOSITOR_MAX_LAYERS; ++i) {
 482       if (c->used_layers & (1 << i)) {
 483          struct vl_compositor_layer *layer = &c->layers[i];
 484          struct pipe_sampler_view **samplers = &layer->sampler_views[0];
 485          unsigned num_sampler_views = !samplers[1] ? 1 : !samplers[2] ? 2 : 3;
 486
 487          c->pipe->bind_fs_state(c->pipe, layer->fs);
 488          c->pipe->bind_fragment_sampler_states(c->pipe, num_sampler_views, layer->samplers);
 489          c->pipe->set_fragment_sampler_views(c->pipe, num_sampler_views, samplers);
 490          util_draw_arrays(c->pipe, PIPE_PRIM_QUADS, vb_index * 4, 4);
 491          vb_index++;
 492
 493          // Remember the currently drawn area as dirty for the next draw command
 494          c->dirty_tl.x = MIN2(layer->dst.tl.x, c->dirty_tl.x);
 495          c->dirty_tl.y = MIN2(layer->dst.tl.y, c->dirty_tl.y);
 496          c->dirty_br.x = MAX2(layer->dst.br.x, c->dirty_br.x);
 497          c->dirty_br.y = MAX2(layer->dst.br.y, c->dirty_br.y);
 498       }
 499    }
 500 }
 501
 502 void
 503 vl_compositor_reset_dirty_area(struct vl_compositor *c)
 504 {
 505    assert(c);
 506
 507    c->dirty_tl.x = c->dirty_tl.y = 0.0f;
 508    c->dirty_br.x = c->dirty_br.y = 1.0f;
 509 }
 510
 511 void
 512 vl_compositor_set_clear_color(struct vl_compositor *c, float color[4])
 513 {
 514    unsigned i;
 515
 516    assert(c);
 517
 518    for (i = 0; i < 4; ++i)
 519       c->clear_color[i] = color[i];
 520 }
 521
 522 void
 523 vl_compositor_clear_layers(struct vl_compositor *c)
 524 {
 525    unsigned i, j;
 526
 527    assert(c);
 528
 529    c->used_layers = 0;
 530    for ( i = 0; i < VL_COMPOSITOR_MAX_LAYERS; ++i) {
 531       c->layers[i].fs = NULL;
 532       for ( j = 0; j < 3; j++)
 533          pipe_sampler_view_reference(&c->layers[i].sampler_views[j], NULL);
 534    }
 535 }
 536
 537 void
 538 vl_compositor_cleanup(struct vl_compositor *c)
 539 {
 540    assert(c);
 541
 542    vl_compositor_clear_layers(c);
 543
 544    cleanup_buffers(c);
 545    cleanup_shaders(c);
 546    cleanup_pipe_state(c);
 547 }
 548
 549 void
 550 vl_compositor_set_csc_matrix(struct vl_compositor *c, const float matrix[16])
 551 {
 552    struct pipe_transfer *buf_transfer;
 553
 554    assert(c);
 555
 556    memcpy
 557    (
 558       pipe_buffer_map(c->pipe, c->csc_matrix,
 559                       PIPE_TRANSFER_WRITE | PIPE_TRANSFER_DISCARD,
 560                       &buf_transfer),
 561                 matrix,
 562                 sizeof(csc_matrix)
 563    );
 564
 565    pipe_buffer_unmap(c->pipe, buf_transfer);
 566 }
 567
 568 void
 569 vl_compositor_set_buffer_layer(struct vl_compositor *c,
 570                                unsigned layer,
 571                                struct pipe_video_buffer *buffer,
 572                                struct pipe_video_rect *src_rect,
 573                                struct pipe_video_rect *dst_rect)
 574 {
 575    struct pipe_sampler_view **sampler_views;
 576    unsigned i;
 577
 578    assert(c && buffer);
 579
 580    assert(layer < VL_COMPOSITOR_MAX_LAYERS);
 581
 582    c->used_layers |= 1 << layer;
 583    c->layers[layer].clearing = true;
 584    c->layers[layer].fs = c->fs_video_buffer;
 585
 586    sampler_views = buffer->get_sampler_view_components(buffer);
 587    for (i = 0; i < 3; ++i) {
 588       c->layers[layer].samplers[i] = c->sampler_linear;
 589       pipe_sampler_view_reference(&c->layers[layer].sampler_views[i], sampler_views[i]);
 590    }
 591
 592    calc_src_and_dst(&c->layers[layer], buffer->width, buffer->height,
 593                     src_rect ? *src_rect : default_rect(&c->layers[layer]),
 594                     dst_rect ? *dst_rect : default_rect(&c->layers[layer]));
 595 }
 596
 597 void
 598 vl_compositor_set_palette_layer(struct vl_compositor *c,
 599                                 unsigned layer,
 600                                 struct pipe_sampler_view *indexes,
 601                                 struct pipe_sampler_view *palette,
 602                                 struct pipe_video_rect *src_rect,
 603                                 struct pipe_video_rect *dst_rect)
 604 {
 605    assert(c && indexes && palette);
 606
 607    assert(layer < VL_COMPOSITOR_MAX_LAYERS);
 608
 609    c->used_layers |= 1 << layer;
 610    c->layers[layer].clearing = false;
 611    c->layers[layer].fs = c->fs_palette;
 612    c->layers[layer].samplers[0] = c->sampler_linear;
 613    c->layers[layer].samplers[1] = c->sampler_nearest;
 614    c->layers[layer].samplers[2] = NULL;
 615    pipe_sampler_view_reference(&c->layers[layer].sampler_views[0], indexes);
 616    pipe_sampler_view_reference(&c->layers[layer].sampler_views[1], palette);
 617    pipe_sampler_view_reference(&c->layers[layer].sampler_views[2], NULL);
 618    calc_src_and_dst(&c->layers[layer], indexes->texture->width0, indexes->texture->height0,
 619                     src_rect ? *src_rect : default_rect(&c->layers[layer]),
 620                     dst_rect ? *dst_rect : default_rect(&c->layers[layer]));
 621
 622 }
 623
 624 void
 625 vl_compositor_set_rgba_layer(struct vl_compositor *c,
 626                              unsigned layer,
 627                              struct pipe_sampler_view *rgba,
 628                              struct pipe_video_rect *src_rect,
 629                              struct pipe_video_rect *dst_rect)
 630 {
 631    assert(c && rgba);
 632
 633    assert(layer < VL_COMPOSITOR_MAX_LAYERS);
 634
 635    c->used_layers |= 1 << layer;
 636    c->layers[layer].clearing = rgba->swizzle_a == PIPE_SWIZZLE_ONE;
 637    c->layers[layer].fs = c->fs_rgba;
 638    c->layers[layer].samplers[0] = c->sampler_linear;
 639    c->layers[layer].samplers[1] = NULL;
 640    c->layers[layer].samplers[2] = NULL;
 641    pipe_sampler_view_reference(&c->layers[layer].sampler_views[0], rgba);
 642    pipe_sampler_view_reference(&c->layers[layer].sampler_views[1], NULL);
 643    pipe_sampler_view_reference(&c->layers[layer].sampler_views[2], NULL);
 644    calc_src_and_dst(&c->layers[layer], rgba->texture->width0, rgba->texture->height0,
 645                     src_rect ? *src_rect : default_rect(&c->layers[layer]),
 646                     dst_rect ? *dst_rect : default_rect(&c->layers[layer]));
 647 }
 648
 649 void
 650 vl_compositor_render(struct vl_compositor *c,
 651                      enum pipe_mpeg12_picture_type picture_type,
 652                      struct pipe_surface           *dst_surface,
 653                      struct pipe_video_rect        *dst_area,
 654                      struct pipe_video_rect        *dst_clip)
 655 {
 656    struct pipe_scissor_state scissor;
 657
 658    assert(c);
 659    assert(dst_surface);
 660
 661    c->fb_state.width = dst_surface->width;
 662    c->fb_state.height = dst_surface->height;
 663    c->fb_state.cbufs[0] = dst_surface;
 664
 665    if (dst_area) {
 666       c->viewport.scale[0] = dst_area->w;
 667       c->viewport.scale[1] = dst_area->h;
 668       c->viewport.translate[0] = dst_area->x;
 669       c->viewport.translate[1] = dst_area->y;
 670    } else {
 671       c->viewport.scale[0] = dst_surface->width;
 672       c->viewport.scale[1] = dst_surface->height;
 673       c->viewport.translate[0] = 0;
 674       c->viewport.translate[1] = 0;
 675    }
 676
 677    if (dst_clip) {
 678       scissor.minx = dst_clip->x;
 679       scissor.miny = dst_clip->y;
 680       scissor.maxx = dst_clip->x + dst_clip->w;
 681       scissor.maxy = dst_clip->y + dst_clip->h;
 682    } else {
 683       scissor.minx = 0;
 684       scissor.miny = 0;
 685       scissor.maxx = dst_surface->width;
 686       scissor.maxy = dst_surface->height;
 687    }
 688
 689    gen_vertex_data(c);
 690
 691    if (c->dirty_tl.x < c->dirty_br.x || c->dirty_tl.y < c->dirty_br.y) {
 692       util_clear_render_target(c->pipe, dst_surface, c->clear_color, 0, 0, dst_surface->width, dst_surface->height);
 693       c->dirty_tl.x = c->dirty_tl.y = 1.0f;
 694       c->dirty_br.x = c->dirty_br.y = 0.0f;
 695    }
 696
 697    c->pipe->set_scissor_state(c->pipe, &scissor);
 698    c->pipe->set_framebuffer_state(c->pipe, &c->fb_state);
 699    c->pipe->set_viewport_state(c->pipe, &c->viewport);
 700    c->pipe->bind_vs_state(c->pipe, c->vs);
 701    c->pipe->set_vertex_buffers(c->pipe, 1, &c->vertex_buf);
 702    c->pipe->bind_vertex_elements_state(c->pipe, c->vertex_elems_state);
 703    c->pipe->set_constant_buffer(c->pipe, PIPE_SHADER_FRAGMENT, 0, c->csc_matrix);
 704    c->pipe->bind_blend_state(c->pipe, c->blend);
 705    c->pipe->bind_rasterizer_state(c->pipe, c->rast);
 706
 707    draw_layers(c);
 708 }
 709
 710 bool
 711 vl_compositor_init(struct vl_compositor *c, struct pipe_context *pipe)
 712 {
 713    csc_matrix csc_matrix;
 714
 715    c->pipe = pipe;
 716
 717    if (!init_pipe_state(c))
 718       return false;
 719
 720    if (!init_shaders(c)) {
 721       cleanup_pipe_state(c);
 722       return false;
 723    }
 724    if (!init_buffers(c)) {
 725       cleanup_shaders(c);
 726       cleanup_pipe_state(c);
 727       return false;
 728    }
 729
 730    vl_compositor_clear_layers(c);
 731
 732    vl_csc_get_matrix(VL_CSC_COLOR_STANDARD_IDENTITY, NULL, true, csc_matrix);
 733    vl_compositor_set_csc_matrix(c, csc_matrix);
 734
 735    c->clear_color[0] = c->clear_color[1] = 0.0f;
 736    c->clear_color[2] = c->clear_color[3] = 0.0f;
 737    vl_compositor_reset_dirty_area(c);
 738
 739    return true;
 740 }