/* * DXVA2 VP9 HW acceleration. * * copyright (c) 2015 Hendrik Leppkes * * This file is part of FFmpeg. * * FFmpeg is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * FFmpeg is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with FFmpeg; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #include "libavutil/avassert.h" #include "libavutil/pixdesc.h" #include "dxva2_internal.h" #include "vp9shared.h" struct vp9_dxva2_picture_context { DXVA_PicParams_VP9 pp; DXVA_Slice_VPx_Short slice; const uint8_t *bitstream; unsigned bitstream_size; }; static void fill_picture_entry(DXVA_PicEntry_VPx *pic, unsigned index, unsigned flag) { av_assert0((index & 0x7f) == index && (flag & 0x01) == flag); pic->bPicEntry = index | (flag << 7); } static int fill_picture_parameters(const AVCodecContext *avctx, AVDXVAContext *ctx, const VP9SharedContext *h, DXVA_PicParams_VP9 *pp) { int i; const AVPixFmtDescriptor * pixdesc = av_pix_fmt_desc_get(avctx->sw_pix_fmt); if (!pixdesc) return -1; memset(pp, 0, sizeof(*pp)); fill_picture_entry(&pp->CurrPic, ff_dxva2_get_surface_index(avctx, ctx, h->frames[CUR_FRAME].tf.f), 0); pp->profile = h->h.profile; pp->wFormatAndPictureInfoFlags = ((h->h.keyframe == 0) << 0) | ((h->h.invisible == 0) << 1) | (h->h.errorres << 2) | (pixdesc->log2_chroma_w << 3) | /* subsampling_x */ (pixdesc->log2_chroma_h << 4) | /* subsampling_y */ (0 << 5) | /* extra_plane */ (h->h.refreshctx << 6) | (h->h.parallelmode << 7) | (h->h.intraonly << 8) | (h->h.framectxid << 9) | (h->h.resetctx << 11) | ((h->h.keyframe ? 0 : h->h.highprecisionmvs) << 13) | (0 << 14); /* ReservedFormatInfo2Bits */ pp->width = avctx->width; pp->height = avctx->height; pp->BitDepthMinus8Luma = pixdesc->comp[0].depth - 8; pp->BitDepthMinus8Chroma = pixdesc->comp[1].depth - 8; /* swap 0/1 to match the reference */ pp->interp_filter = h->h.filtermode ^ (h->h.filtermode <= 1); pp->Reserved8Bits = 0; for (i = 0; i < 8; i++) { if (h->refs[i].f->buf[0]) { fill_picture_entry(&pp->ref_frame_map[i], ff_dxva2_get_surface_index(avctx, ctx, h->refs[i].f), 0); pp->ref_frame_coded_width[i] = h->refs[i].f->width; pp->ref_frame_coded_height[i] = h->refs[i].f->height; } else pp->ref_frame_map[i].bPicEntry = 0xFF; } for (i = 0; i < 3; i++) { uint8_t refidx = h->h.refidx[i]; if (h->refs[refidx].f->buf[0]) fill_picture_entry(&pp->frame_refs[i], ff_dxva2_get_surface_index(avctx, ctx, h->refs[refidx].f), 0); else pp->frame_refs[i].bPicEntry = 0xFF; pp->ref_frame_sign_bias[i + 1] = h->h.signbias[i]; } pp->filter_level = h->h.filter.level; pp->sharpness_level = h->h.filter.sharpness; pp->wControlInfoFlags = (h->h.lf_delta.enabled << 0) | (h->h.lf_delta.updated << 1) | (h->h.use_last_frame_mvs << 2) | (0 << 3); /* ReservedControlInfo5Bits */ for (i = 0; i < 4; i++) pp->ref_deltas[i] = h->h.lf_delta.ref[i]; for (i = 0; i < 2; i++) pp->mode_deltas[i] = h->h.lf_delta.mode[i]; pp->base_qindex = h->h.yac_qi; pp->y_dc_delta_q = h->h.ydc_qdelta; pp->uv_dc_delta_q = h->h.uvdc_qdelta; pp->uv_ac_delta_q = h->h.uvac_qdelta; /* segmentation data */ pp->stVP9Segments.wSegmentInfoFlags = (h->h.segmentation.enabled << 0) | (h->h.segmentation.update_map << 1) | (h->h.segmentation.temporal << 2) | (h->h.segmentation.absolute_vals << 3) | (0 << 4); /* ReservedSegmentFlags4Bits */ for (i = 0; i < 7; i++) pp->stVP9Segments.tree_probs[i] = h->h.segmentation.prob[i]; if (h->h.segmentation.temporal) for (i = 0; i < 3; i++) pp->stVP9Segments.pred_probs[i] = h->h.segmentation.pred_prob[i]; else memset(pp->stVP9Segments.pred_probs, 255, sizeof(pp->stVP9Segments.pred_probs)); for (i = 0; i < 8; i++) { pp->stVP9Segments.feature_mask[i] = (h->h.segmentation.feat[i].q_enabled << 0) | (h->h.segmentation.feat[i].lf_enabled << 1) | (h->h.segmentation.feat[i].ref_enabled << 2) | (h->h.segmentation.feat[i].skip_enabled << 3); pp->stVP9Segments.feature_data[i][0] = h->h.segmentation.feat[i].q_val; pp->stVP9Segments.feature_data[i][1] = h->h.segmentation.feat[i].lf_val; pp->stVP9Segments.feature_data[i][2] = h->h.segmentation.feat[i].ref_val; pp->stVP9Segments.feature_data[i][3] = 0; /* no data for skip */ } pp->log2_tile_cols = h->h.tiling.log2_tile_cols; pp->log2_tile_rows = h->h.tiling.log2_tile_rows; pp->uncompressed_header_size_byte_aligned = h->h.uncompressed_header_size; pp->first_partition_size = h->h.compressed_header_size; pp->StatusReportFeedbackNumber = 1 + DXVA_CONTEXT_REPORT_ID(avctx, ctx)++; return 0; } static void fill_slice_short(DXVA_Slice_VPx_Short *slice, unsigned position, unsigned size) { memset(slice, 0, sizeof(*slice)); slice->BSNALunitDataLocation = position; slice->SliceBytesInBuffer = size; slice->wBadSliceChopping = 0; } static int commit_bitstream_and_slice_buffer(AVCodecContext *avctx, DECODER_BUFFER_DESC *bs, DECODER_BUFFER_DESC *sc) { const VP9SharedContext *h = avctx->priv_data; AVDXVAContext *ctx = DXVA_CONTEXT(avctx); struct vp9_dxva2_picture_context *ctx_pic = h->frames[CUR_FRAME].hwaccel_picture_private; void *dxva_data_ptr; uint8_t *dxva_data; unsigned dxva_size; unsigned padding; unsigned type; #if CONFIG_D3D11VA if (ff_dxva2_is_d3d11(avctx)) { type = D3D11_VIDEO_DECODER_BUFFER_BITSTREAM; if (FAILED(ID3D11VideoContext_GetDecoderBuffer(D3D11VA_CONTEXT(ctx)->video_context, D3D11VA_CONTEXT(ctx)->decoder, type, &dxva_size, &dxva_data_ptr))) return -1; } #endif #if CONFIG_DXVA2 if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) { type = DXVA2_BitStreamDateBufferType; if (FAILED(IDirectXVideoDecoder_GetBuffer(DXVA2_CONTEXT(ctx)->decoder, type, &dxva_data_ptr, &dxva_size))) return -1; } #endif dxva_data = dxva_data_ptr; if (ctx_pic->slice.SliceBytesInBuffer > dxva_size) { av_log(avctx, AV_LOG_ERROR, "Failed to build bitstream"); return -1; } memcpy(dxva_data, ctx_pic->bitstream, ctx_pic->slice.SliceBytesInBuffer); padding = FFMIN(128 - ((ctx_pic->slice.SliceBytesInBuffer) & 127), dxva_size - ctx_pic->slice.SliceBytesInBuffer); if (padding > 0) { memset(dxva_data + ctx_pic->slice.SliceBytesInBuffer, 0, padding); ctx_pic->slice.SliceBytesInBuffer += padding; } #if CONFIG_D3D11VA if (ff_dxva2_is_d3d11(avctx)) if (FAILED(ID3D11VideoContext_ReleaseDecoderBuffer(D3D11VA_CONTEXT(ctx)->video_context, D3D11VA_CONTEXT(ctx)->decoder, type))) return -1; #endif #if CONFIG_DXVA2 if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) if (FAILED(IDirectXVideoDecoder_ReleaseBuffer(DXVA2_CONTEXT(ctx)->decoder, type))) return -1; #endif #if CONFIG_D3D11VA if (ff_dxva2_is_d3d11(avctx)) { D3D11_VIDEO_DECODER_BUFFER_DESC *dsc11 = bs; memset(dsc11, 0, sizeof(*dsc11)); dsc11->BufferType = type; dsc11->DataSize = ctx_pic->slice.SliceBytesInBuffer; dsc11->NumMBsInBuffer = 0; type = D3D11_VIDEO_DECODER_BUFFER_SLICE_CONTROL; } #endif #if CONFIG_DXVA2 if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) { DXVA2_DecodeBufferDesc *dsc2 = bs; memset(dsc2, 0, sizeof(*dsc2)); dsc2->CompressedBufferType = type; dsc2->DataSize = ctx_pic->slice.SliceBytesInBuffer; dsc2->NumMBsInBuffer = 0; type = DXVA2_SliceControlBufferType; } #endif return ff_dxva2_commit_buffer(avctx, ctx, sc, type, &ctx_pic->slice, sizeof(ctx_pic->slice), 0); } static int dxva2_vp9_start_frame(AVCodecContext *avctx, av_unused const uint8_t *buffer, av_unused uint32_t size) { const VP9SharedContext *h = avctx->priv_data; AVDXVAContext *ctx = DXVA_CONTEXT(avctx); struct vp9_dxva2_picture_context *ctx_pic = h->frames[CUR_FRAME].hwaccel_picture_private; if (!DXVA_CONTEXT_VALID(avctx, ctx)) return -1; av_assert0(ctx_pic); /* Fill up DXVA_PicParams_VP9 */ if (fill_picture_parameters(avctx, ctx, h, &ctx_pic->pp) < 0) return -1; ctx_pic->bitstream_size = 0; ctx_pic->bitstream = NULL; return 0; } static int dxva2_vp9_decode_slice(AVCodecContext *avctx, const uint8_t *buffer, uint32_t size) { const VP9SharedContext *h = avctx->priv_data; struct vp9_dxva2_picture_context *ctx_pic = h->frames[CUR_FRAME].hwaccel_picture_private; unsigned position; if (!ctx_pic->bitstream) ctx_pic->bitstream = buffer; ctx_pic->bitstream_size += size; position = buffer - ctx_pic->bitstream; fill_slice_short(&ctx_pic->slice, position, size); return 0; } static int dxva2_vp9_end_frame(AVCodecContext *avctx) { VP9SharedContext *h = avctx->priv_data; struct vp9_dxva2_picture_context *ctx_pic = h->frames[CUR_FRAME].hwaccel_picture_private; int ret; if (ctx_pic->bitstream_size <= 0) return -1; ret = ff_dxva2_common_end_frame(avctx, h->frames[CUR_FRAME].tf.f, &ctx_pic->pp, sizeof(ctx_pic->pp), NULL, 0, commit_bitstream_and_slice_buffer); return ret; } #if CONFIG_VP9_DXVA2_HWACCEL const AVHWAccel ff_vp9_dxva2_hwaccel = { .name = "vp9_dxva2", .type = AVMEDIA_TYPE_VIDEO, .id = AV_CODEC_ID_VP9, .pix_fmt = AV_PIX_FMT_DXVA2_VLD, .init = ff_dxva2_decode_init, .uninit = ff_dxva2_decode_uninit, .start_frame = dxva2_vp9_start_frame, .decode_slice = dxva2_vp9_decode_slice, .end_frame = dxva2_vp9_end_frame, .frame_params = ff_dxva2_common_frame_params, .frame_priv_data_size = sizeof(struct vp9_dxva2_picture_context), .priv_data_size = sizeof(FFDXVASharedContext), }; #endif #if CONFIG_VP9_D3D11VA_HWACCEL const AVHWAccel ff_vp9_d3d11va_hwaccel = { .name = "vp9_d3d11va", .type = AVMEDIA_TYPE_VIDEO, .id = AV_CODEC_ID_VP9, .pix_fmt = AV_PIX_FMT_D3D11VA_VLD, .init = ff_dxva2_decode_init, .uninit = ff_dxva2_decode_uninit, .start_frame = dxva2_vp9_start_frame, .decode_slice = dxva2_vp9_decode_slice, .end_frame = dxva2_vp9_end_frame, .frame_params = ff_dxva2_common_frame_params, .frame_priv_data_size = sizeof(struct vp9_dxva2_picture_context), .priv_data_size = sizeof(FFDXVASharedContext), }; #endif #if CONFIG_VP9_D3D11VA2_HWACCEL const AVHWAccel ff_vp9_d3d11va2_hwaccel = { .name = "vp9_d3d11va2", .type = AVMEDIA_TYPE_VIDEO, .id = AV_CODEC_ID_VP9, .pix_fmt = AV_PIX_FMT_D3D11, .init = ff_dxva2_decode_init, .uninit = ff_dxva2_decode_uninit, .start_frame = dxva2_vp9_start_frame, .decode_slice = dxva2_vp9_decode_slice, .end_frame = dxva2_vp9_end_frame, .frame_params = ff_dxva2_common_frame_params, .frame_priv_data_size = sizeof(struct vp9_dxva2_picture_context), .priv_data_size = sizeof(FFDXVASharedContext), }; #endif