FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
dxva2_vp9.c
Go to the documentation of this file.
1 /*
2  * DXVA2 VP9 HW acceleration.
3  *
4  * copyright (c) 2015 Hendrik Leppkes
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
23 #include "libavutil/avassert.h"
24 #include "libavutil/pixdesc.h"
25 
26 #include "vp9shared.h"
27 
28 // The headers above may include w32threads.h, which uses the original
29 // _WIN32_WINNT define, while dxva2_internal.h redefines it to target a
30 // potentially newer version.
31 #include "dxva2_internal.h"
32 
34  DXVA_PicParams_VP9 pp;
35  DXVA_Slice_VPx_Short slice;
37  unsigned bitstream_size;
38 };
39 
40 static void fill_picture_entry(DXVA_PicEntry_VPx *pic,
41  unsigned index, unsigned flag)
42 {
43  av_assert0((index & 0x7f) == index && (flag & 0x01) == flag);
44  pic->bPicEntry = index | (flag << 7);
45 }
46 
48  DXVA_PicParams_VP9 *pp)
49 {
50  int i;
51  const AVPixFmtDescriptor * pixdesc = av_pix_fmt_desc_get(avctx->sw_pix_fmt);
52 
53  if (!pixdesc)
54  return -1;
55 
56  memset(pp, 0, sizeof(*pp));
57 
58  fill_picture_entry(&pp->CurrPic, ff_dxva2_get_surface_index(avctx, ctx, h->frames[CUR_FRAME].tf.f), 0);
59 
60  pp->profile = h->h.profile;
61  pp->wFormatAndPictureInfoFlags = ((h->h.keyframe == 0) << 0) |
62  ((h->h.invisible == 0) << 1) |
63  (h->h.errorres << 2) |
64  (pixdesc->log2_chroma_w << 3) | /* subsampling_x */
65  (pixdesc->log2_chroma_h << 4) | /* subsampling_y */
66  (0 << 5) | /* extra_plane */
67  (h->h.refreshctx << 6) |
68  (h->h.parallelmode << 7) |
69  (h->h.intraonly << 8) |
70  (h->h.framectxid << 9) |
71  (h->h.resetctx << 11) |
72  ((h->h.keyframe ? 0 : h->h.highprecisionmvs) << 13) |
73  (0 << 14); /* ReservedFormatInfo2Bits */
74 
75  pp->width = avctx->width;
76  pp->height = avctx->height;
77  pp->BitDepthMinus8Luma = pixdesc->comp[0].depth - 8;
78  pp->BitDepthMinus8Chroma = pixdesc->comp[1].depth - 8;
79  /* swap 0/1 to match the reference */
80  pp->interp_filter = h->h.filtermode ^ (h->h.filtermode <= 1);
81  pp->Reserved8Bits = 0;
82 
83  for (i = 0; i < 8; i++) {
84  if (h->refs[i].f->buf[0]) {
85  fill_picture_entry(&pp->ref_frame_map[i], ff_dxva2_get_surface_index(avctx, ctx, h->refs[i].f), 0);
86  pp->ref_frame_coded_width[i] = h->refs[i].f->width;
87  pp->ref_frame_coded_height[i] = h->refs[i].f->height;
88  } else
89  pp->ref_frame_map[i].bPicEntry = 0xFF;
90  }
91 
92  for (i = 0; i < 3; i++) {
93  uint8_t refidx = h->h.refidx[i];
94  if (h->refs[refidx].f->buf[0])
95  fill_picture_entry(&pp->frame_refs[i], ff_dxva2_get_surface_index(avctx, ctx, h->refs[refidx].f), 0);
96  else
97  pp->frame_refs[i].bPicEntry = 0xFF;
98 
99  pp->ref_frame_sign_bias[i + 1] = h->h.signbias[i];
100  }
101 
102  pp->filter_level = h->h.filter.level;
103  pp->sharpness_level = h->h.filter.sharpness;
104 
105  pp->wControlInfoFlags = (h->h.lf_delta.enabled << 0) |
106  (h->h.lf_delta.updated << 1) |
107  (h->h.use_last_frame_mvs << 2) |
108  (0 << 3); /* ReservedControlInfo5Bits */
109 
110  for (i = 0; i < 4; i++)
111  pp->ref_deltas[i] = h->h.lf_delta.ref[i];
112 
113  for (i = 0; i < 2; i++)
114  pp->mode_deltas[i] = h->h.lf_delta.mode[i];
115 
116  pp->base_qindex = h->h.yac_qi;
117  pp->y_dc_delta_q = h->h.ydc_qdelta;
118  pp->uv_dc_delta_q = h->h.uvdc_qdelta;
119  pp->uv_ac_delta_q = h->h.uvac_qdelta;
120 
121  /* segmentation data */
122  pp->stVP9Segments.wSegmentInfoFlags = (h->h.segmentation.enabled << 0) |
123  (h->h.segmentation.update_map << 1) |
124  (h->h.segmentation.temporal << 2) |
125  (h->h.segmentation.absolute_vals << 3) |
126  (0 << 4); /* ReservedSegmentFlags4Bits */
127 
128  for (i = 0; i < 7; i++)
129  pp->stVP9Segments.tree_probs[i] = h->h.segmentation.prob[i];
130 
132  for (i = 0; i < 3; i++)
133  pp->stVP9Segments.pred_probs[i] = h->h.segmentation.pred_prob[i];
134  else
135  memset(pp->stVP9Segments.pred_probs, 255, sizeof(pp->stVP9Segments.pred_probs));
136 
137  for (i = 0; i < 8; i++) {
138  pp->stVP9Segments.feature_mask[i] = (h->h.segmentation.feat[i].q_enabled << 0) |
139  (h->h.segmentation.feat[i].lf_enabled << 1) |
140  (h->h.segmentation.feat[i].ref_enabled << 2) |
141  (h->h.segmentation.feat[i].skip_enabled << 3);
142 
143  pp->stVP9Segments.feature_data[i][0] = h->h.segmentation.feat[i].q_val;
144  pp->stVP9Segments.feature_data[i][1] = h->h.segmentation.feat[i].lf_val;
145  pp->stVP9Segments.feature_data[i][2] = h->h.segmentation.feat[i].ref_val;
146  pp->stVP9Segments.feature_data[i][3] = 0; /* no data for skip */
147  }
148 
149  pp->log2_tile_cols = h->h.tiling.log2_tile_cols;
150  pp->log2_tile_rows = h->h.tiling.log2_tile_rows;
151 
152  pp->uncompressed_header_size_byte_aligned = h->h.uncompressed_header_size;
153  pp->first_partition_size = h->h.compressed_header_size;
154 
155  pp->StatusReportFeedbackNumber = 1 + DXVA_CONTEXT_REPORT_ID(avctx, ctx)++;
156  return 0;
157 }
158 
159 static void fill_slice_short(DXVA_Slice_VPx_Short *slice,
160  unsigned position, unsigned size)
161 {
162  memset(slice, 0, sizeof(*slice));
163  slice->BSNALunitDataLocation = position;
164  slice->SliceBytesInBuffer = size;
165  slice->wBadSliceChopping = 0;
166 }
167 
171 {
172  const VP9SharedContext *h = avctx->priv_data;
175  void *dxva_data_ptr;
176  uint8_t *dxva_data;
177  unsigned dxva_size;
178  unsigned padding;
179  unsigned type;
180 
181 #if CONFIG_D3D11VA
182  if (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD) {
183  type = D3D11_VIDEO_DECODER_BUFFER_BITSTREAM;
184  if (FAILED(ID3D11VideoContext_GetDecoderBuffer(D3D11VA_CONTEXT(ctx)->video_context,
185  D3D11VA_CONTEXT(ctx)->decoder,
186  type,
187  &dxva_size, &dxva_data_ptr)))
188  return -1;
189  }
190 #endif
191 #if CONFIG_DXVA2
192  if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) {
193  type = DXVA2_BitStreamDateBufferType;
194  if (FAILED(IDirectXVideoDecoder_GetBuffer(DXVA2_CONTEXT(ctx)->decoder,
195  type,
196  &dxva_data_ptr, &dxva_size)))
197  return -1;
198  }
199 #endif
200 
201  dxva_data = dxva_data_ptr;
202 
203  if (ctx_pic->slice.SliceBytesInBuffer > dxva_size) {
204  av_log(avctx, AV_LOG_ERROR, "Failed to build bitstream");
205  return -1;
206  }
207 
208  memcpy(dxva_data, ctx_pic->bitstream, ctx_pic->slice.SliceBytesInBuffer);
209 
210  padding = FFMIN(128 - ((ctx_pic->slice.SliceBytesInBuffer) & 127), dxva_size - ctx_pic->slice.SliceBytesInBuffer);
211  if (padding > 0) {
212  memset(dxva_data + ctx_pic->slice.SliceBytesInBuffer, 0, padding);
213  ctx_pic->slice.SliceBytesInBuffer += padding;
214  }
215 
216 #if CONFIG_D3D11VA
217  if (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD)
218  if (FAILED(ID3D11VideoContext_ReleaseDecoderBuffer(D3D11VA_CONTEXT(ctx)->video_context, D3D11VA_CONTEXT(ctx)->decoder, type)))
219  return -1;
220 #endif
221 #if CONFIG_DXVA2
222  if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD)
223  if (FAILED(IDirectXVideoDecoder_ReleaseBuffer(DXVA2_CONTEXT(ctx)->decoder, type)))
224  return -1;
225 #endif
226 
227 #if CONFIG_D3D11VA
228  if (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD) {
229  D3D11_VIDEO_DECODER_BUFFER_DESC *dsc11 = bs;
230  memset(dsc11, 0, sizeof(*dsc11));
231  dsc11->BufferType = type;
232  dsc11->DataSize = ctx_pic->slice.SliceBytesInBuffer;
233  dsc11->NumMBsInBuffer = 0;
234 
235  type = D3D11_VIDEO_DECODER_BUFFER_SLICE_CONTROL;
236  }
237 #endif
238 #if CONFIG_DXVA2
239  if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) {
240  DXVA2_DecodeBufferDesc *dsc2 = bs;
241  memset(dsc2, 0, sizeof(*dsc2));
242  dsc2->CompressedBufferType = type;
243  dsc2->DataSize = ctx_pic->slice.SliceBytesInBuffer;
244  dsc2->NumMBsInBuffer = 0;
245 
246  type = DXVA2_SliceControlBufferType;
247  }
248 #endif
249 
250  return ff_dxva2_commit_buffer(avctx, ctx, sc,
251  type,
252  &ctx_pic->slice, sizeof(ctx_pic->slice), 0);
253 }
254 
255 
257  av_unused const uint8_t *buffer,
258  av_unused uint32_t size)
259 {
260  const VP9SharedContext *h = avctx->priv_data;
263 
264  if (!DXVA_CONTEXT_VALID(avctx, ctx))
265  return -1;
266  av_assert0(ctx_pic);
267 
268  /* Fill up DXVA_PicParams_VP9 */
269  if (fill_picture_parameters(avctx, ctx, h, &ctx_pic->pp) < 0)
270  return -1;
271 
272  ctx_pic->bitstream_size = 0;
273  ctx_pic->bitstream = NULL;
274  return 0;
275 }
276 
278  const uint8_t *buffer,
279  uint32_t size)
280 {
281  const VP9SharedContext *h = avctx->priv_data;
283  unsigned position;
284 
285  if (!ctx_pic->bitstream)
286  ctx_pic->bitstream = buffer;
287  ctx_pic->bitstream_size += size;
288 
289  position = buffer - ctx_pic->bitstream;
290  fill_slice_short(&ctx_pic->slice, position, size);
291 
292  return 0;
293 }
294 
296 {
297  VP9SharedContext *h = avctx->priv_data;
299  int ret;
300 
301  if (ctx_pic->bitstream_size <= 0)
302  return -1;
303 
304  ret = ff_dxva2_common_end_frame(avctx, h->frames[CUR_FRAME].tf.f,
305  &ctx_pic->pp, sizeof(ctx_pic->pp),
306  NULL, 0,
308  return ret;
309 }
310 
311 #if CONFIG_VP9_DXVA2_HWACCEL
312 AVHWAccel ff_vp9_dxva2_hwaccel = {
313  .name = "vp9_dxva2",
314  .type = AVMEDIA_TYPE_VIDEO,
315  .id = AV_CODEC_ID_VP9,
316  .pix_fmt = AV_PIX_FMT_DXVA2_VLD,
317  .start_frame = dxva2_vp9_start_frame,
318  .decode_slice = dxva2_vp9_decode_slice,
319  .end_frame = dxva2_vp9_end_frame,
320  .frame_priv_data_size = sizeof(struct vp9_dxva2_picture_context),
321 };
322 #endif
323 
324 #if CONFIG_VP9_D3D11VA_HWACCEL
325 AVHWAccel ff_vp9_d3d11va_hwaccel = {
326  .name = "vp9_d3d11va",
327  .type = AVMEDIA_TYPE_VIDEO,
328  .id = AV_CODEC_ID_VP9,
329  .pix_fmt = AV_PIX_FMT_D3D11VA_VLD,
330  .start_frame = dxva2_vp9_start_frame,
331  .decode_slice = dxva2_vp9_decode_slice,
332  .end_frame = dxva2_vp9_end_frame,
333  .frame_priv_data_size = sizeof(struct vp9_dxva2_picture_context),
334 };
335 #endif
ThreadFrame tf
Definition: vp9shared.h:60
#define NULL
Definition: coverity.c:32
uint8_t parallelmode
Definition: vp9shared.h:108
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2333
uint8_t update_map
Definition: vp9shared.h:133
AVFrame * f
Definition: thread.h:36
AVBufferRef * buf[AV_NUM_DATA_POINTERS]
AVBuffer references backing the data for this frame.
Definition: frame.h:370
VP9BitstreamHeader h
Definition: vp9shared.h:160
static int commit_bitstream_and_slice_buffer(AVCodecContext *avctx, DECODER_BUFFER_DESC *bs, DECODER_BUFFER_DESC *sc)
Definition: dxva2_vp9.c:168
DXVA_PicParams_VP9 pp
Definition: dxva2_vp9.c:34
int flag
Definition: cpu.c:34
uint8_t prob[7]
Definition: vp9shared.h:134
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1960
uint8_t framectxid
Definition: vp9shared.h:109
uint8_t log2_chroma_w
Amount to shift the luma width right to find the chroma width.
Definition: pixdesc.h:92
unsigned log2_tile_rows
Definition: vp9shared.h:151
int uncompressed_header_size
Definition: vp9shared.h:155
enum FilterMode filtermode
Definition: vp9shared.h:105
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:37
static void fill_slice_short(DXVA_Slice_VPx_Short *slice, unsigned position, unsigned size)
Definition: dxva2_vp9.c:159
AVComponentDescriptor comp[4]
Parameters that describe how pixels are packed.
Definition: pixdesc.h:117
uint8_t
void * hwaccel_context
Hardware accelerator context.
Definition: avcodec.h:3064
uint8_t absolute_vals
Definition: vp9shared.h:132
VP9Frame frames[3]
Definition: vp9shared.h:166
int ff_dxva2_common_end_frame(AVCodecContext *avctx, AVFrame *frame, const void *pp, unsigned pp_size, const void *qm, unsigned qm_size, int(*commit_bs_si)(AVCodecContext *, DECODER_BUFFER_DESC *bs, DECODER_BUFFER_DESC *slice))
Definition: dxva2.c:135
ptrdiff_t size
Definition: opengl_enc.c:101
unsigned log2_tile_cols
Definition: vp9shared.h:151
uint8_t refidx[3]
Definition: vp9shared.h:111
struct VP9BitstreamHeader::@154::@156 feat[MAX_SEGMENT]
#define av_log(a,...)
int width
width and height of the video frame
Definition: frame.h:239
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
uint8_t log2_chroma_h
Amount to shift the luma height right to find the chroma height.
Definition: pixdesc.h:101
const uint8_t * bitstream
Definition: dxva2_vp9.c:36
uint8_t signbias[3]
Definition: vp9shared.h:112
uint8_t refreshctx
Definition: vp9shared.h:107
void * hwaccel_picture_private
Definition: vp9shared.h:67
simple assert() macros that are a bit more flexible than ISO C assert().
struct VP9BitstreamHeader::@152 filter
const char * name
Name of the hardware accelerated codec.
Definition: avcodec.h:3807
#define FFMIN(a, b)
Definition: common.h:96
uint8_t keyframe
Definition: vp9shared.h:98
static const chunk_decoder decoder[8]
Definition: dfa.c:328
int width
picture width / height.
Definition: avcodec.h:1919
AVFormatContext * ctx
Definition: movenc.c:48
struct VP9BitstreamHeader::@155 tiling
static void fill_picture_entry(DXVA_PicEntry_VPx *pic, unsigned index, unsigned flag)
Definition: dxva2_vp9.c:40
unsigned ff_dxva2_get_surface_index(const AVCodecContext *avctx, const AVDXVAContext *ctx, const AVFrame *frame)
Definition: dxva2.c:37
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:81
main external API structure.
Definition: avcodec.h:1732
int8_t mode[2]
Definition: vp9shared.h:122
#define CUR_FRAME
Definition: vp9shared.h:163
int ff_dxva2_commit_buffer(AVCodecContext *avctx, AVDXVAContext *ctx, DECODER_BUFFER_DESC *dsc, unsigned type, const void *data, unsigned size, unsigned mb_count)
Definition: dxva2.c:62
GLint GLenum type
Definition: opengl_enc.c:105
int index
Definition: gxfenc.c:89
static int fill_picture_parameters(const AVCodecContext *avctx, AVDXVAContext *ctx, const VP9SharedContext *h, DXVA_PicParams_VP9 *pp)
Definition: dxva2_vp9.c:47
void DECODER_BUFFER_DESC
HW decoding through DXVA2, Picture.data[3] contains a LPDIRECT3DSURFACE9 pointer. ...
Definition: pixfmt.h:148
#define FAILED(hr)
Definition: windows2linux.h:48
if(ret< 0)
Definition: vf_mcdeint.c:282
static int dxva2_vp9_end_frame(AVCodecContext *avctx)
Definition: dxva2_vp9.c:295
struct VP9BitstreamHeader::@153 lf_delta
DXVA_Slice_VPx_Short slice
Definition: dxva2_vp9.c:35
void * priv_data
Definition: avcodec.h:1774
static int dxva2_vp9_decode_slice(AVCodecContext *avctx, const uint8_t *buffer, uint32_t size)
Definition: dxva2_vp9.c:277
static int dxva2_vp9_start_frame(AVCodecContext *avctx, av_unused const uint8_t *buffer, av_unused uint32_t size)
Definition: dxva2_vp9.c:256
uint8_t invisible
Definition: vp9shared.h:99
uint8_t use_last_frame_mvs
Definition: vp9shared.h:110
int height
Definition: frame.h:239
ThreadFrame refs[8]
Definition: vp9shared.h:162
uint8_t pred_prob[3]
Definition: vp9shared.h:135
HW decoding through Direct3D11, Picture.data[3] contains a ID3D11VideoDecoderOutputView pointer...
Definition: pixfmt.h:243
struct VP9BitstreamHeader::@154 segmentation
int depth
Number of bits in the component.
Definition: pixdesc.h:58
uint8_t highprecisionmvs
Definition: vp9shared.h:104
enum AVPixelFormat sw_pix_fmt
Nominal unaccelerated pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:3436
for(j=16;j >0;--j)
GLuint buffer
Definition: opengl_enc.c:102
#define av_unused
Definition: attributes.h:125
int compressed_header_size
Definition: vp9shared.h:156