xorg_xv.c revision 2947d1420270476730711892909c3683bb6c5bff 
1#include "xorg_tracker.h"
2
3#include <xf86xv.h>
4#include <X11/extensions/Xv.h>
5#include <fourcc.h>
6
7#include "xorg_exa.h"
8#include "xorg_renderer.h"
9#include "xorg_exa_tgsi.h"
10
11#include "cso_cache/cso_context.h"
12
13#include "pipe/p_screen.h"
14#include "pipe/p_inlines.h"
15
16/*XXX get these from pipe's texture limits */
17#define IMAGE_MAX_WIDTH		2048
18#define IMAGE_MAX_HEIGHT	2048
19
20#define RES_720P_X 1280
21#define RES_720P_Y 720
22
23
24/* The ITU-R BT.601 conversion matrix for SDTV. */
25/* original, matrix, but we transpose it to
26 * make the shader easier
27static const float bt_601[] = {
28    1.0, 0.0, 1.4075,   ,
29    1.0, -0.3455, -0.7169, 0,
30    1.0, 1.7790, 0., 0,
31};*/
32static const float bt_601[] = {
33    1.0, 1.0, 1.0,        0.5,
34    0.0, -0.3455, 1.7790, 0,
35    1.4075, -0.7169, 0.,  0,
36};
37
38/* The ITU-R BT.709 conversion matrix for HDTV. */
39/* original, but we transpose to make the conversion
40 * in the shader easier
41static const float bt_709[] = {
42    1.0, 0.0, 1.581, 0,
43    1.0, -0.1881, -0.47, 0,
44    1.0, 1.8629, 0., 0,
45};*/
46static const float bt_709[] = {
47    1.0,   1.0,     1.0,     0.5,
48    0.0,  -0.1881,  1.8629,  0,
49    1.581,-0.47   , 0.0,     0,
50};
51
52#define MAKE_ATOM(a) MakeAtom(a, sizeof(a) - 1, TRUE)
53
54static Atom xvBrightness, xvContrast;
55
56#define NUM_TEXTURED_ATTRIBUTES 2
57static XF86AttributeRec TexturedAttributes[NUM_TEXTURED_ATTRIBUTES] = {
58   {XvSettable | XvGettable, -128, 127, "XV_BRIGHTNESS"},
59   {XvSettable | XvGettable, 0, 255, "XV_CONTRAST"}
60};
61
62#define NUM_FORMATS 3
63static XF86VideoFormatRec Formats[NUM_FORMATS] = {
64   {15, TrueColor}, {16, TrueColor}, {24, TrueColor}
65};
66
67static XF86VideoEncodingRec DummyEncoding[1] = {
68   {
69      0,
70      "XV_IMAGE",
71      IMAGE_MAX_WIDTH, IMAGE_MAX_HEIGHT,
72      {1, 1}
73   }
74};
75
76#define NUM_IMAGES 2
77static XF86ImageRec Images[NUM_IMAGES] = {
78   XVIMAGE_UYVY,
79   XVIMAGE_YUY2,
80};
81
82struct xorg_xv_port_priv {
83   struct xorg_renderer *r;
84
85   RegionRec clip;
86
87   int brightness;
88   int contrast;
89
90   int current_set;
91   /* juggle two sets of seperate Y, U and V
92    * textures */
93   struct pipe_texture *yuv[2][3];
94};
95
96
97static void
98stop_video(ScrnInfoPtr pScrn, pointer data, Bool shutdown)
99{
100   struct xorg_xv_port_priv *priv = (struct xorg_xv_port_priv *)data;
101
102   REGION_EMPTY(pScrn->pScreen, &priv->clip);
103}
104
105static int
106set_port_attribute(ScrnInfoPtr pScrn,
107                   Atom attribute, INT32 value, pointer data)
108{
109   struct xorg_xv_port_priv *priv = (struct xorg_xv_port_priv *)data;
110
111   if (attribute == xvBrightness) {
112      if ((value < -128) || (value > 127))
113         return BadValue;
114      priv->brightness = value;
115   } else if (attribute == xvContrast) {
116      if ((value < 0) || (value > 255))
117         return BadValue;
118      priv->contrast = value;
119   } else
120      return BadMatch;
121
122   return Success;
123}
124
125static int
126get_port_attribute(ScrnInfoPtr pScrn,
127                   Atom attribute, INT32 * value, pointer data)
128{
129   struct xorg_xv_port_priv *priv = (struct xorg_xv_port_priv *)data;
130
131   if (attribute == xvBrightness)
132      *value = priv->brightness;
133   else if (attribute == xvContrast)
134      *value = priv->contrast;
135   else
136      return BadMatch;
137
138   return Success;
139}
140
141static void
142query_best_size(ScrnInfoPtr pScrn,
143                Bool motion,
144                short vid_w, short vid_h,
145                short drw_w, short drw_h,
146                unsigned int *p_w, unsigned int *p_h, pointer data)
147{
148   if (vid_w > (drw_w << 1))
149      drw_w = vid_w >> 1;
150   if (vid_h > (drw_h << 1))
151      drw_h = vid_h >> 1;
152
153   *p_w = drw_w;
154   *p_h = drw_h;
155}
156
157static INLINE struct pipe_texture *
158create_component_texture(struct pipe_context *pipe,
159                         int width, int height)
160{
161   struct pipe_screen *screen = pipe->screen;
162   struct pipe_texture *tex = 0;
163   struct pipe_texture templ;
164
165   memset(&templ, 0, sizeof(templ));
166   templ.target = PIPE_TEXTURE_2D;
167   templ.format = PIPE_FORMAT_L8_UNORM;
168   templ.last_level = 0;
169   templ.width[0] = width;
170   templ.height[0] = height;
171   templ.depth[0] = 1;
172   pf_get_block(PIPE_FORMAT_L8_UNORM, &templ.block);
173   templ.tex_usage = PIPE_TEXTURE_USAGE_SAMPLER;
174
175   tex = screen->texture_create(screen, &templ);
176
177   return tex;
178}
179
180static int
181check_yuv_textures(struct xorg_xv_port_priv *priv,  int width, int height)
182{
183   struct pipe_texture **dst = priv->yuv[priv->current_set];
184   if (!dst[0] ||
185       dst[0]->width[0] != width ||
186       dst[0]->height[0] != height) {
187      pipe_texture_reference(&dst[0], NULL);
188   }
189   if (!dst[1] ||
190       dst[1]->width[0] != width ||
191       dst[1]->height[0] != height) {
192      pipe_texture_reference(&dst[1], NULL);
193   }
194   if (!dst[2] ||
195       dst[2]->width[0] != width ||
196       dst[2]->height[0] != height) {
197      pipe_texture_reference(&dst[2], NULL);
198   }
199
200   if (!dst[0])
201      dst[0] = create_component_texture(priv->r->pipe, width, height);
202
203   if (!dst[1])
204      dst[1] = create_component_texture(priv->r->pipe, width, height);
205
206   if (!dst[2])
207      dst[2] = create_component_texture(priv->r->pipe, width, height);
208
209   if (!dst[0] || !dst[1] || !dst[2])
210      return BadAlloc;
211
212   return Success;
213}
214
215static void
216copy_packed_data(ScrnInfoPtr pScrn,
217                 struct xorg_xv_port_priv *port,
218                 int id,
219                 unsigned char *buf,
220                 int srcPitch,
221                 int left,
222                 int top,
223                 int w, int h)
224{
225   unsigned char *src;
226   int i, j;
227   struct pipe_texture **dst = port->yuv[port->current_set];
228   struct pipe_transfer *ytrans, *utrans, *vtrans;
229   struct pipe_screen *screen = port->r->pipe->screen;
230   char *ymap, *vmap, *umap;
231   unsigned char y1, y2, u, v;
232   int yidx, uidx, vidx;
233   int y_array_size = w * h;
234
235   src = buf + (top * srcPitch) + (left << 1);
236
237   ytrans = screen->get_tex_transfer(screen, dst[0],
238                                     0, 0, 0,
239                                     PIPE_TRANSFER_WRITE,
240                                     left, top, w, h);
241   utrans = screen->get_tex_transfer(screen, dst[1],
242                                     0, 0, 0,
243                                     PIPE_TRANSFER_WRITE,
244                                     left, top, w, h);
245   vtrans = screen->get_tex_transfer(screen, dst[2],
246                                     0, 0, 0,
247                                     PIPE_TRANSFER_WRITE,
248                                     left, top, w, h);
249
250   ymap = (char*)screen->transfer_map(screen, ytrans);
251   umap = (char*)screen->transfer_map(screen, utrans);
252   vmap = (char*)screen->transfer_map(screen, vtrans);
253
254   yidx = uidx = vidx = 0;
255
256   switch (id) {
257   case FOURCC_YV12: {
258      for (i = 0; i < w; ++i) {
259         for (j = 0; i < h; ++j) {
260            /*XXX use src? */
261            y1  = buf[j*w + i];
262            u   = buf[(j/2) * (w/2) + i/2 + y_array_size];
263            v   = buf[(j/2) * (w/2) + i/2 + y_array_size + y_array_size/4];
264            ymap[yidx++] = y1;
265            umap[uidx++] = u;
266            vmap[vidx++] = v;
267         }
268      }
269   }
270      break;
271   case FOURCC_UYVY:
272      for (i = 0; i < y_array_size; i +=2 ) {
273         /* extracting two pixels */
274         u  = buf[0];
275         y1 = buf[1];
276         v  = buf[2];
277         y2 = buf[3];
278         buf += 4;
279
280         ymap[yidx++] = y1;
281         ymap[yidx++] = y2;
282         umap[uidx++] = u;
283         umap[uidx++] = u;
284         vmap[vidx++] = v;
285         vmap[vidx++] = v;
286      }
287      break;
288   case FOURCC_YUY2:
289      for (i = 0; i < y_array_size; i +=2 ) {
290         /* extracting two pixels */
291         y1 = buf[0];
292         u  = buf[1];
293         y2 = buf[2];
294         v  = buf[3];
295
296         buf += 4;
297
298         ymap[yidx++] = y1;
299         ymap[yidx++] = y2;
300         umap[uidx++] = u;
301         umap[uidx++] = u;
302         vmap[vidx++] = v;
303         vmap[vidx++] = v;
304      }
305      break;
306   default:
307      debug_assert(!"Unsupported yuv format!");
308      break;
309   }
310
311   screen->transfer_unmap(screen, ytrans);
312   screen->transfer_unmap(screen, utrans);
313   screen->transfer_unmap(screen, vtrans);
314   screen->tex_transfer_destroy(ytrans);
315   screen->tex_transfer_destroy(utrans);
316   screen->tex_transfer_destroy(vtrans);
317}
318
319
320static void
321setup_vs_video_constants(struct xorg_renderer *r, struct exa_pixmap_priv *dst)
322{
323   int width = dst->tex->width[0];
324   int height = dst->tex->height[0];
325   const int param_bytes = 8 * sizeof(float);
326   float vs_consts[8] = {
327      2.f/width, 2.f/height, 1, 1,
328      -1, -1, 0, 0
329   };
330
331   renderer_set_constants(r, PIPE_SHADER_VERTEX,
332                          vs_consts, param_bytes);
333}
334
335static void
336setup_fs_video_constants(struct xorg_renderer *r, boolean hdtv)
337{
338   const int param_bytes = 12 * sizeof(float);
339   const float *video_constants = (hdtv) ? bt_709 : bt_601;
340
341   renderer_set_constants(r, PIPE_SHADER_FRAGMENT,
342                          video_constants, param_bytes);
343}
344
345static void
346draw_yuv(struct xorg_xv_port_priv *port, int src_x, int src_y,
347         int dst_x, int dst_y,
348         int w, int h)
349{
350   int pos[4] = {src_x, src_y,
351                 dst_x, dst_y};
352   struct pipe_texture **textures = port->yuv[port->current_set];
353
354   renderer_draw_textures(port->r,
355                          pos, w, h,
356                          textures,
357                          3, /*bound samplers/textures */
358                          NULL, NULL /* no transformations */);
359}
360
361static void
362bind_blend_state(struct xorg_xv_port_priv *port)
363{
364   struct pipe_blend_state blend;
365
366   memset(&blend, 0, sizeof(struct pipe_blend_state));
367   blend.blend_enable = 1;
368   blend.colormask |= PIPE_MASK_RGBA;
369
370   /* porter&duff src */
371   blend.rgb_src_factor   = PIPE_BLENDFACTOR_ONE;
372   blend.alpha_src_factor = PIPE_BLENDFACTOR_ONE;
373   blend.rgb_dst_factor   = PIPE_BLENDFACTOR_ZERO;
374   blend.alpha_dst_factor = PIPE_BLENDFACTOR_ZERO;
375
376   cso_set_blend(port->r->cso, &blend);
377}
378
379
380static void
381bind_shaders(struct xorg_xv_port_priv *port)
382{
383   unsigned vs_traits = 0, fs_traits = 0;
384   struct xorg_shader shader;
385
386   vs_traits |= VS_YUV;
387   fs_traits |= FS_YUV;
388
389   shader = xorg_shaders_get(port->r->shaders, vs_traits, fs_traits);
390   cso_set_vertex_shader_handle(port->r->cso, shader.vs);
391   cso_set_fragment_shader_handle(port->r->cso, shader.fs);
392}
393
394static INLINE void
395conditional_flush(struct pipe_context *pipe, struct pipe_texture **tex,
396                  int num)
397{
398   int i;
399   for (i = 0; i < num; ++i) {
400      if (tex[i] && pipe->is_texture_referenced(pipe, tex[i], 0, 0) &
401          PIPE_REFERENCED_FOR_WRITE) {
402         pipe->flush(pipe, PIPE_FLUSH_RENDER_CACHE, NULL);
403         return;
404      }
405   }
406}
407
408static void
409bind_samplers(struct xorg_xv_port_priv *port)
410{
411   struct pipe_sampler_state *samplers[PIPE_MAX_SAMPLERS];
412   struct pipe_sampler_state sampler;
413   struct pipe_texture **dst = port->yuv[port->current_set];
414
415   memset(&sampler, 0, sizeof(struct pipe_sampler_state));
416
417   conditional_flush(port->r->pipe, dst, 3);
418
419   sampler.wrap_s = PIPE_TEX_WRAP_CLAMP;
420   sampler.wrap_t = PIPE_TEX_WRAP_CLAMP;
421   sampler.min_img_filter = PIPE_TEX_FILTER_LINEAR;
422   sampler.mag_img_filter = PIPE_TEX_FILTER_LINEAR;
423   sampler.min_mip_filter = PIPE_TEX_MIPFILTER_NEAREST;
424   sampler.normalized_coords = 1;
425
426   samplers[0] = &sampler;
427   samplers[1] = &sampler;
428   samplers[2] = &sampler;
429
430
431   cso_set_samplers(port->r->cso, 3,
432                    (const struct pipe_sampler_state **)samplers);
433   cso_set_sampler_textures(port->r->cso, 3,
434                            dst);
435}
436
437static int
438display_video(ScrnInfoPtr pScrn, struct xorg_xv_port_priv *pPriv, int id,
439              RegionPtr dstRegion,
440              int src_x, int src_y, int src_w, int src_h,
441              int dstX, int dstY,
442              short width, short height,
443              PixmapPtr pPixmap)
444{
445   modesettingPtr ms = modesettingPTR(pScrn);
446   BoxPtr pbox;
447   int nbox;
448   int dxo, dyo;
449   Bool hdtv;
450   int x, y, w, h;
451   struct exa_pixmap_priv *dst = exaGetPixmapDriverPrivate(pPixmap);
452
453   if (!dst || !dst->tex)
454      XORG_FALLBACK("Xv destination %s", !dst ? "!dst" : "!dst->tex");
455
456   hdtv = ((src_w >= RES_720P_X) && (src_h >= RES_720P_Y));
457
458   REGION_TRANSLATE(pScrn->pScreen, dstRegion, -pPixmap->screen_x,
459                    -pPixmap->screen_y);
460
461   dxo = dstRegion->extents.x1;
462   dyo = dstRegion->extents.y1;
463
464   pbox = REGION_RECTS(dstRegion);
465   nbox = REGION_NUM_RECTS(dstRegion);
466
467   renderer_bind_framebuffer(pPriv->r, dst);
468   renderer_bind_viewport(pPriv->r, dst);
469   bind_blend_state(pPriv);
470   renderer_bind_rasterizer(pPriv->r);
471   bind_shaders(pPriv);
472   bind_samplers(pPriv);
473   setup_vs_video_constants(pPriv->r, dst);
474   setup_fs_video_constants(pPriv->r, hdtv);
475
476   while (nbox--) {
477      int box_x1 = pbox->x1;
478      int box_y1 = pbox->y1;
479      int box_x2 = pbox->x2;
480      int box_y2 = pbox->y2;
481
482      x = box_x1;
483      y = box_y1;
484      w = box_x2 - box_x1;
485      h = box_y2 - box_y1;
486
487      draw_yuv(pPriv, x, y, x, y, w, h);
488
489      pbox++;
490   }
491   DamageDamageRegion(&pPixmap->drawable, dstRegion);
492
493   return TRUE;
494}
495
496static int
497put_image(ScrnInfoPtr pScrn,
498          short src_x, short src_y,
499          short drw_x, short drw_y,
500          short src_w, short src_h,
501          short drw_w, short drw_h,
502          int id, unsigned char *buf,
503          short width, short height,
504          Bool sync, RegionPtr clipBoxes, pointer data,
505          DrawablePtr pDraw)
506{
507   struct xorg_xv_port_priv *pPriv = (struct xorg_xv_port_priv *) data;
508   ScreenPtr pScreen = screenInfo.screens[pScrn->scrnIndex];
509   PixmapPtr pPixmap;
510   INT32 x1, x2, y1, y2;
511   int srcPitch;
512   BoxRec dstBox;
513   int ret;
514
515   /* Clip */
516   x1 = src_x;
517   x2 = src_x + src_w;
518   y1 = src_y;
519   y2 = src_y + src_h;
520
521   dstBox.x1 = drw_x;
522   dstBox.x2 = drw_x + drw_w;
523   dstBox.y1 = drw_y;
524   dstBox.y2 = drw_y + drw_h;
525
526   if (!xf86XVClipVideoHelper(&dstBox, &x1, &x2, &y1, &y2, clipBoxes,
527			      width, height))
528      return Success;
529
530   switch (id) {
531   case FOURCC_UYVY:
532   case FOURCC_YUY2:
533   default:
534      srcPitch = width << 1;
535      break;
536   }
537
538   ret = check_yuv_textures(pPriv, width, height);
539
540   if (ret)
541      return ret;
542
543   copy_packed_data(pScrn, pPriv, id, buf, srcPitch,
544                    src_x, src_y, width, height);
545
546   if (pDraw->type == DRAWABLE_WINDOW) {
547      pPixmap = (*pScreen->GetWindowPixmap)((WindowPtr)pDraw);
548   } else {
549      pPixmap = (PixmapPtr)pDraw;
550   }
551
552   display_video(pScrn, pPriv, id, clipBoxes,
553                 src_x, src_y, src_w, src_h,
554                 drw_x, drw_y,
555                 drw_w, drw_h, pPixmap);
556
557   pPriv->current_set = (pPriv->current_set + 1) & 1;
558   return Success;
559}
560
561static int
562query_image_attributes(ScrnInfoPtr pScrn,
563                       int id,
564                       unsigned short *w, unsigned short *h,
565                       int *pitches, int *offsets)
566{
567   int size;
568
569   if (*w > IMAGE_MAX_WIDTH)
570      *w = IMAGE_MAX_WIDTH;
571   if (*h > IMAGE_MAX_HEIGHT)
572      *h = IMAGE_MAX_HEIGHT;
573
574   *w = (*w + 1) & ~1;
575   if (offsets)
576      offsets[0] = 0;
577
578   switch (id) {
579   case FOURCC_UYVY:
580   case FOURCC_YUY2:
581   default:
582      size = *w << 1;
583      if (pitches)
584	 pitches[0] = size;
585      size *= *h;
586      break;
587   }
588
589   return size;
590}
591
592static struct xorg_xv_port_priv *
593port_priv_create(struct xorg_renderer *r)
594{
595   struct xorg_xv_port_priv *priv = NULL;
596
597   priv = calloc(1, sizeof(struct xorg_xv_port_priv));
598
599   if (!priv)
600      return NULL;
601
602   priv->r = r;
603
604   REGION_NULL(pScreen, &priv->clip);
605
606   debug_assert(priv && priv->r);
607
608   return priv;
609}
610
611static XF86VideoAdaptorPtr
612xorg_setup_textured_adapter(ScreenPtr pScreen)
613{
614   ScrnInfoPtr pScrn = xf86Screens[pScreen->myNum];
615   modesettingPtr ms = modesettingPTR(pScrn);
616   XF86VideoAdaptorPtr adapt;
617   XF86AttributePtr attrs;
618   DevUnion *dev_unions;
619   int nports = 16, i;
620   int nattributes;
621
622   nattributes = NUM_TEXTURED_ATTRIBUTES;
623
624   debug_assert(ms->exa);
625   debug_assert(ms->exa->renderer);
626
627   adapt = calloc(1, sizeof(XF86VideoAdaptorRec));
628   dev_unions = calloc(nports, sizeof(DevUnion));
629   attrs = calloc(nattributes, sizeof(XF86AttributeRec));
630   if (adapt == NULL || dev_unions == NULL || attrs == NULL) {
631      free(adapt);
632      free(dev_unions);
633      free(attrs);
634      return NULL;
635   }
636
637   adapt->type = XvWindowMask | XvInputMask | XvImageMask;
638   adapt->flags = 0;
639   adapt->name = "Gallium3D Textured Video";
640   adapt->nEncodings = 1;
641   adapt->pEncodings = DummyEncoding;
642   adapt->nFormats = NUM_FORMATS;
643   adapt->pFormats = Formats;
644   adapt->nPorts = 0;
645   adapt->pPortPrivates = dev_unions;
646   adapt->nAttributes = nattributes;
647   adapt->pAttributes = attrs;
648   memcpy(attrs, TexturedAttributes, nattributes * sizeof(XF86AttributeRec));
649   adapt->nImages = NUM_IMAGES;
650   adapt->pImages = Images;
651   adapt->PutVideo = NULL;
652   adapt->PutStill = NULL;
653   adapt->GetVideo = NULL;
654   adapt->GetStill = NULL;
655   adapt->StopVideo = stop_video;
656   adapt->SetPortAttribute = set_port_attribute;
657   adapt->GetPortAttribute = get_port_attribute;
658   adapt->QueryBestSize = query_best_size;
659   adapt->PutImage = put_image;
660   adapt->QueryImageAttributes = query_image_attributes;
661
662   for (i = 0; i < nports; i++) {
663      struct xorg_xv_port_priv *priv =
664         port_priv_create(ms->exa->renderer);
665
666      adapt->pPortPrivates[i].ptr = (pointer) (priv);
667      adapt->nPorts++;
668   }
669
670   return adapt;
671}
672
673void
674xorg_init_video(ScreenPtr pScreen)
675{
676   ScrnInfoPtr pScrn = xf86Screens[pScreen->myNum];
677   /*modesettingPtr ms = modesettingPTR(pScrn);*/
678   XF86VideoAdaptorPtr *adaptors, *new_adaptors = NULL;
679   XF86VideoAdaptorPtr textured_adapter;
680   int num_adaptors;
681
682   num_adaptors = xf86XVListGenericAdaptors(pScrn, &adaptors);
683   new_adaptors = malloc((num_adaptors + 1) * sizeof(XF86VideoAdaptorPtr *));
684   if (new_adaptors == NULL)
685      return;
686
687   memcpy(new_adaptors, adaptors, num_adaptors * sizeof(XF86VideoAdaptorPtr));
688   adaptors = new_adaptors;
689
690   /* Add the adaptors supported by our hardware.  First, set up the atoms
691    * that will be used by both output adaptors.
692    */
693   xvBrightness = MAKE_ATOM("XV_BRIGHTNESS");
694   xvContrast = MAKE_ATOM("XV_CONTRAST");
695
696   textured_adapter = xorg_setup_textured_adapter(pScreen);
697
698   debug_assert(textured_adapter);
699
700   if (textured_adapter) {
701      adaptors[num_adaptors++] = textured_adapter;
702   }
703
704   if (num_adaptors) {
705      xf86XVScreenInit(pScreen, adaptors, num_adaptors);
706   } else {
707      xf86DrvMsg(pScrn->scrnIndex, X_WARNING,
708                 "Disabling Xv because no adaptors could be initialized.\n");
709   }
710
711   free(adaptors);
712}
713