Libav
vdpau_vc1.c
Go to the documentation of this file.
1 /*
2  * VC-1 decode acceleration through VDPAU
3  *
4  * Copyright (c) 2008 NVIDIA
5  * Copyright (c) 2013 RĂ©mi Denis-Courmont
6  *
7  * This file is part of Libav.
8  *
9  * Libav is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * Libav is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with Libav; if not, write to the Free Software Foundation,
21  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22  */
23 
24 #include <vdpau/vdpau.h>
25 
26 #include "avcodec.h"
27 #include "vc1.h"
28 #include "vdpau.h"
29 #include "vdpau_internal.h"
30 
32  const uint8_t *buffer, uint32_t size)
33 {
34  VC1Context * const v = avctx->priv_data;
35  MpegEncContext * const s = &v->s;
36  Picture *pic = s->current_picture_ptr;
37  struct vdpau_picture_context *pic_ctx = pic->hwaccel_picture_private;
38  VdpPictureInfoVC1 *info = &pic_ctx->info.vc1;
39  VdpVideoSurface ref;
40 
41  /* fill LvPictureInfoVC1 struct */
42  info->forward_reference = VDP_INVALID_HANDLE;
43  info->backward_reference = VDP_INVALID_HANDLE;
44 
45  switch (s->pict_type) {
46  case AV_PICTURE_TYPE_B:
48  assert(ref != VDP_INVALID_HANDLE);
49  info->backward_reference = ref;
50  /* fall-through */
51  case AV_PICTURE_TYPE_P:
53  assert(ref != VDP_INVALID_HANDLE);
54  info->forward_reference = ref;
55  }
56 
57  info->slice_count = 0;
58  if (v->bi_type)
59  info->picture_type = 4;
60  else
61  info->picture_type = s->pict_type - 1 + s->pict_type / 3;
62 
63  info->frame_coding_mode = v->fcm ? (v->fcm + 1) : 0;
64  info->postprocflag = v->postprocflag;
65  info->pulldown = v->broadcast;
66  info->interlace = v->interlace;
67  info->tfcntrflag = v->tfcntrflag;
68  info->finterpflag = v->finterpflag;
69  info->psf = v->psf;
70  info->dquant = v->dquant;
71  info->panscan_flag = v->panscanflag;
72  info->refdist_flag = v->refdist_flag;
73  info->quantizer = v->quantizer_mode;
74  info->extended_mv = v->extended_mv;
75  info->extended_dmv = v->extended_dmv;
76  info->overlap = v->overlap;
77  info->vstransform = v->vstransform;
78  info->loopfilter = v->s.loop_filter;
79  info->fastuvmc = v->fastuvmc;
80  info->range_mapy_flag = v->range_mapy_flag;
81  info->range_mapy = v->range_mapy;
82  info->range_mapuv_flag = v->range_mapuv_flag;
83  info->range_mapuv = v->range_mapuv;
84  /* Specific to simple/main profile only */
85  info->multires = v->multires;
86  info->syncmarker = v->resync_marker;
87  info->rangered = v->rangered | (v->rangeredfrm << 1);
88  info->maxbframes = v->s.max_b_frames;
89  info->deblockEnable = v->postprocflag & 1;
90  info->pquant = v->pq;
91 
92  return ff_vdpau_common_start_frame(pic, buffer, size);
93 }
94 
96  const uint8_t *buffer, uint32_t size)
97 {
98  VC1Context * const v = avctx->priv_data;
99  MpegEncContext * const s = &v->s;
100  Picture *pic = s->current_picture_ptr;
101  struct vdpau_picture_context *pic_ctx = pic->hwaccel_picture_private;
102  int val;
103 
104  val = ff_vdpau_add_buffer(pic, buffer, size);
105  if (val < 0)
106  return val;
107 
108  pic_ctx->info.vc1.slice_count++;
109  return 0;
110 }
111 
112 #if CONFIG_WMV3_VDPAU_HWACCEL
113 AVHWAccel ff_wmv3_vdpau_hwaccel = {
114  .name = "wm3_vdpau",
115  .type = AVMEDIA_TYPE_VIDEO,
116  .id = AV_CODEC_ID_WMV3,
117  .pix_fmt = AV_PIX_FMT_VDPAU,
118  .start_frame = vdpau_vc1_start_frame,
119  .end_frame = ff_vdpau_mpeg_end_frame,
120  .decode_slice = vdpau_vc1_decode_slice,
121  .priv_data_size = sizeof(struct vdpau_picture_context),
122 };
123 #endif
124 
126  .name = "vc1_vdpau",
127  .type = AVMEDIA_TYPE_VIDEO,
128  .id = AV_CODEC_ID_VC1,
129  .pix_fmt = AV_PIX_FMT_VDPAU,
130  .start_frame = vdpau_vc1_start_frame,
131  .end_frame = ff_vdpau_mpeg_end_frame,
132  .decode_slice = vdpau_vc1_decode_slice,
133  .priv_data_size = sizeof(struct vdpau_picture_context),
134 };