vl_compositor.c revision 2cbf532ae13513c3cbc7dad8ae41beeff931529e
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
43typedef float csc_matrix[16];
44
45static void *
46create_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
73static void *
74create_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
116static void *
117create_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
161static void *
162create_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
186static bool
187init_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
218static 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
228static bool
229init_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
295static 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
305static bool
306create_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
321static bool
322init_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
361static void
362cleanup_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
371static INLINE struct pipe_video_rect
372default_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
379static INLINE struct vertex2f
380calc_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
386static INLINE struct vertex2f
387calc_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
393static INLINE void
394calc_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
405static void
406gen_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
431static void
432gen_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
474static void
475draw_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
502void
503vl_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
511void
512vl_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
522void
523vl_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
537void
538vl_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
549void
550vl_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
568void
569vl_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
597void
598vl_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
624void
625vl_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
649void
650vl_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                     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   if (dst_area) {
667      c->viewport.scale[0] = dst_area->w;
668      c->viewport.scale[1] = dst_area->h;
669      c->viewport.translate[0] = dst_area->x;
670      c->viewport.translate[1] = dst_area->y;
671   } else {
672      c->viewport.scale[0] = dst_surface->width;
673      c->viewport.scale[1] = dst_surface->height;
674      c->viewport.translate[0] = 0;
675      c->viewport.translate[1] = 0;
676   }
677
678   if (dst_clip) {
679      scissor.minx = dst_clip->x;
680      scissor.miny = dst_clip->y;
681      scissor.maxx = dst_clip->x + dst_clip->w;
682      scissor.maxy = dst_clip->y + dst_clip->h;
683   } else {
684      scissor.minx = 0;
685      scissor.miny = 0;
686      scissor.maxx = dst_surface->width;
687      scissor.maxy = dst_surface->height;
688   }
689
690   gen_vertex_data(c);
691
692   if (c->dirty_tl.x < c->dirty_br.x || c->dirty_tl.y < c->dirty_br.y) {
693      util_clear_render_target(c->pipe, dst_surface, c->clear_color, 0, 0, dst_surface->width, dst_surface->height);
694      c->dirty_tl.x = c->dirty_tl.y = 1.0f;
695      c->dirty_br.x = c->dirty_br.y = 0.0f;
696   }
697
698   c->pipe->set_scissor_state(c->pipe, &scissor);
699   c->pipe->set_framebuffer_state(c->pipe, &c->fb_state);
700   c->pipe->set_viewport_state(c->pipe, &c->viewport);
701   c->pipe->bind_vs_state(c->pipe, c->vs);
702   c->pipe->set_vertex_buffers(c->pipe, 1, &c->vertex_buf);
703   c->pipe->bind_vertex_elements_state(c->pipe, c->vertex_elems_state);
704   c->pipe->set_constant_buffer(c->pipe, PIPE_SHADER_FRAGMENT, 0, c->csc_matrix);
705   c->pipe->bind_blend_state(c->pipe, c->blend);
706   c->pipe->bind_rasterizer_state(c->pipe, c->rast);
707
708   draw_layers(c);
709
710   c->pipe->flush(c->pipe, fence);
711}
712
713bool
714vl_compositor_init(struct vl_compositor *c, struct pipe_context *pipe)
715{
716   csc_matrix csc_matrix;
717
718   c->pipe = pipe;
719
720   if (!init_pipe_state(c))
721      return false;
722
723   if (!init_shaders(c)) {
724      cleanup_pipe_state(c);
725      return false;
726   }
727   if (!init_buffers(c)) {
728      cleanup_shaders(c);
729      cleanup_pipe_state(c);
730      return false;
731   }
732
733   vl_compositor_clear_layers(c);
734
735   vl_csc_get_matrix(VL_CSC_COLOR_STANDARD_IDENTITY, NULL, true, csc_matrix);
736   vl_compositor_set_csc_matrix(c, csc_matrix);
737
738   c->clear_color[0] = c->clear_color[1] = 0.0f;
739   c->clear_color[2] = c->clear_color[3] = 0.0f;
740   vl_compositor_reset_dirty_area(c);
741
742   return true;
743}
744