FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
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 FFmpeg.
8  *
9  * FFmpeg 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  * FFmpeg 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 FFmpeg; 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:
47  if (s->next_picture_ptr) {
49  assert(ref != VDP_INVALID_HANDLE);
50  info->backward_reference = ref;
51  }
52  /* fall-through */
53  case AV_PICTURE_TYPE_P:
54  if (s->last_picture_ptr) {
56  assert(ref != VDP_INVALID_HANDLE);
57  info->forward_reference = ref;
58  }
59  }
60 
61  info->slice_count = 0;
62  if (v->bi_type)
63  info->picture_type = 4;
64  else
65  info->picture_type = s->pict_type - 1 + s->pict_type / 3;
66 
67  info->frame_coding_mode = v->fcm ? (v->fcm + 1) : 0;
68  info->postprocflag = v->postprocflag;
69  info->pulldown = v->broadcast;
70  info->interlace = v->interlace;
71  info->tfcntrflag = v->tfcntrflag;
72  info->finterpflag = v->finterpflag;
73  info->psf = v->psf;
74  info->dquant = v->dquant;
75  info->panscan_flag = v->panscanflag;
76  info->refdist_flag = v->refdist_flag;
77  info->quantizer = v->quantizer_mode;
78  info->extended_mv = v->extended_mv;
79  info->extended_dmv = v->extended_dmv;
80  info->overlap = v->overlap;
81  info->vstransform = v->vstransform;
82  info->loopfilter = v->s.loop_filter;
83  info->fastuvmc = v->fastuvmc;
84  info->range_mapy_flag = v->range_mapy_flag;
85  info->range_mapy = v->range_mapy;
86  info->range_mapuv_flag = v->range_mapuv_flag;
87  info->range_mapuv = v->range_mapuv;
88  /* Specific to simple/main profile only */
89  info->multires = v->multires;
90  info->syncmarker = v->resync_marker;
91  info->rangered = v->rangered | (v->rangeredfrm << 1);
92  info->maxbframes = v->s.max_b_frames;
93  info->deblockEnable = v->postprocflag & 1;
94  info->pquant = v->pq;
95 
96  return ff_vdpau_common_start_frame(pic_ctx, buffer, size);
97 }
98 
100  const uint8_t *buffer, uint32_t size)
101 {
102  VC1Context * const v = avctx->priv_data;
103  MpegEncContext * const s = &v->s;
104  Picture *pic = s->current_picture_ptr;
105  struct vdpau_picture_context *pic_ctx = pic->hwaccel_picture_private;
106  int val;
107 
108  val = ff_vdpau_add_buffer(pic_ctx, buffer, size);
109  if (val < 0)
110  return val;
111 
112  pic_ctx->info.vc1.slice_count++;
113  return 0;
114 }
115 
116 static int vdpau_vc1_init(AVCodecContext *avctx)
117 {
118  VdpDecoderProfile profile;
119 
120  switch (avctx->profile) {
122  profile = VDP_DECODER_PROFILE_VC1_SIMPLE;
123  break;
124  case FF_PROFILE_VC1_MAIN:
125  profile = VDP_DECODER_PROFILE_VC1_MAIN;
126  break;
128  profile = VDP_DECODER_PROFILE_VC1_ADVANCED;
129  break;
130  default:
131  return AVERROR(ENOTSUP);
132  }
133 
134  return ff_vdpau_common_init(avctx, profile, avctx->level);
135 }
136 
137 #if CONFIG_WMV3_VDPAU_HWACCEL
138 AVHWAccel ff_wmv3_vdpau_hwaccel = {
139  .name = "wm3_vdpau",
140  .type = AVMEDIA_TYPE_VIDEO,
141  .id = AV_CODEC_ID_WMV3,
142  .pix_fmt = AV_PIX_FMT_VDPAU,
143  .start_frame = vdpau_vc1_start_frame,
144  .end_frame = ff_vdpau_mpeg_end_frame,
145  .decode_slice = vdpau_vc1_decode_slice,
146  .frame_priv_data_size = sizeof(struct vdpau_picture_context),
147  .init = vdpau_vc1_init,
148  .uninit = ff_vdpau_common_uninit,
149  .priv_data_size = sizeof(VDPAUContext),
150 };
151 #endif
152 
154  .name = "vc1_vdpau",
155  .type = AVMEDIA_TYPE_VIDEO,
156  .id = AV_CODEC_ID_VC1,
157  .pix_fmt = AV_PIX_FMT_VDPAU,
158  .start_frame = vdpau_vc1_start_frame,
159  .end_frame = ff_vdpau_mpeg_end_frame,
160  .decode_slice = vdpau_vc1_decode_slice,
161  .frame_priv_data_size = sizeof(struct vdpau_picture_context),
162  .init = vdpau_vc1_init,
163  .uninit = ff_vdpau_common_uninit,
164  .priv_data_size = sizeof(VDPAUContext),
165 };