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