mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-23 12:43:46 +02:00
f9e7a2f95a
This also adds support to avconv (which is trivial due to the new hwaccel API being generic enough). The new decoder setup code in dxva2.c is significantly based on work by Steve Lhomme <robux4@gmail.com>, but with heavy changes/rewrites. Signed-off-by: Diego Biurrun <diego@biurrun.de>
162 lines
6.8 KiB
C
162 lines
6.8 KiB
C
/*
|
|
* DXVA2 HW acceleration
|
|
*
|
|
* copyright (c) 2010 Laurent Aimar
|
|
*
|
|
* This file is part of Libav.
|
|
*
|
|
* Libav 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.
|
|
*
|
|
* Libav 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 Libav; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#ifndef AVCODEC_DXVA2_INTERNAL_H
|
|
#define AVCODEC_DXVA2_INTERNAL_H
|
|
|
|
#define COBJMACROS
|
|
|
|
#include "config.h"
|
|
|
|
/* define the proper COM entries before forcing desktop APIs */
|
|
#include <objbase.h>
|
|
|
|
#if CONFIG_DXVA2
|
|
#include "dxva2.h"
|
|
#include "libavutil/hwcontext_dxva2.h"
|
|
#endif
|
|
#if CONFIG_D3D11VA
|
|
#include "d3d11va.h"
|
|
#include "libavutil/hwcontext_d3d11va.h"
|
|
#endif
|
|
#if HAVE_DXVA_H
|
|
/* When targeting WINAPI_FAMILY_PHONE_APP or WINAPI_FAMILY_APP, dxva.h
|
|
* defines nothing. Force the struct definitions to be visible. */
|
|
#undef WINAPI_FAMILY
|
|
#define WINAPI_FAMILY WINAPI_FAMILY_DESKTOP_APP
|
|
#undef _CRT_BUILD_DESKTOP_APP
|
|
#define _CRT_BUILD_DESKTOP_APP 0
|
|
#include <dxva.h>
|
|
#endif
|
|
|
|
#include "libavutil/hwcontext.h"
|
|
|
|
#include "avcodec.h"
|
|
#include "internal.h"
|
|
|
|
typedef void DECODER_BUFFER_DESC;
|
|
|
|
typedef union {
|
|
#if CONFIG_D3D11VA
|
|
struct AVD3D11VAContext d3d11va;
|
|
#endif
|
|
#if CONFIG_DXVA2
|
|
struct dxva_context dxva2;
|
|
#endif
|
|
} AVDXVAContext;
|
|
|
|
typedef struct FFDXVASharedContext {
|
|
AVBufferRef *decoder_ref;
|
|
|
|
// FF_DXVA2_WORKAROUND_* flags
|
|
uint64_t workaround;
|
|
|
|
// E.g. AV_PIX_FMT_D3D11 (same as AVCodecContext.pix_fmt, except during init)
|
|
enum AVPixelFormat pix_fmt;
|
|
|
|
AVHWDeviceContext *device_ctx;
|
|
|
|
#if CONFIG_D3D11VA
|
|
ID3D11VideoDecoder *d3d11_decoder;
|
|
D3D11_VIDEO_DECODER_CONFIG d3d11_config;
|
|
ID3D11VideoDecoderOutputView **d3d11_views;
|
|
int nb_d3d11_views;
|
|
ID3D11Texture2D *d3d11_texture;
|
|
#endif
|
|
|
|
#if CONFIG_DXVA2
|
|
IDirectXVideoDecoder *dxva2_decoder;
|
|
IDirectXVideoDecoderService *dxva2_service;
|
|
DXVA2_ConfigPictureDecode dxva2_config;
|
|
#endif
|
|
|
|
// Legacy (but used by code outside of setup)
|
|
// In generic mode, DXVA_CONTEXT() will return a pointer to this.
|
|
AVDXVAContext ctx;
|
|
} FFDXVASharedContext;
|
|
|
|
#define DXVA_SHARED_CONTEXT(avctx) ((FFDXVASharedContext *)((avctx)->internal->hwaccel_priv_data))
|
|
|
|
#define DXVA_CONTEXT(avctx) (AVDXVAContext *)((avctx)->hwaccel_context ? (avctx)->hwaccel_context : (&(DXVA_SHARED_CONTEXT(avctx)->ctx)))
|
|
|
|
#define D3D11VA_CONTEXT(ctx) (&ctx->d3d11va)
|
|
#define DXVA2_CONTEXT(ctx) (&ctx->dxva2)
|
|
|
|
#if CONFIG_D3D11VA && CONFIG_DXVA2
|
|
#define DXVA_CONTEXT_WORKAROUND(avctx, ctx) (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.workaround : ctx->dxva2.workaround)
|
|
#define DXVA_CONTEXT_COUNT(avctx, ctx) (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.surface_count : ctx->dxva2.surface_count)
|
|
#define DXVA_CONTEXT_DECODER(avctx, ctx) (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.decoder : ctx->dxva2.decoder)
|
|
#define DXVA_CONTEXT_REPORT_ID(avctx, ctx) (*(ff_dxva2_is_d3d11(avctx) ? &ctx->d3d11va.report_id : &ctx->dxva2.report_id))
|
|
#define DXVA_CONTEXT_CFG(avctx, ctx) (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.cfg : ctx->dxva2.cfg)
|
|
#define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx) (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.cfg->ConfigBitstreamRaw : ctx->dxva2.cfg->ConfigBitstreamRaw)
|
|
#define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.cfg->ConfigIntraResidUnsigned : ctx->dxva2.cfg->ConfigIntraResidUnsigned)
|
|
#define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.cfg->ConfigResidDiffAccelerator : ctx->dxva2.cfg->ConfigResidDiffAccelerator)
|
|
#define DXVA_CONTEXT_VALID(avctx, ctx) (DXVA_CONTEXT_DECODER(avctx, ctx) && \
|
|
DXVA_CONTEXT_CFG(avctx, ctx) && \
|
|
(ff_dxva2_is_d3d11(avctx) || ctx->dxva2.surface_count))
|
|
#elif CONFIG_DXVA2
|
|
#define DXVA_CONTEXT_WORKAROUND(avctx, ctx) (ctx->dxva2.workaround)
|
|
#define DXVA_CONTEXT_COUNT(avctx, ctx) (ctx->dxva2.surface_count)
|
|
#define DXVA_CONTEXT_DECODER(avctx, ctx) (ctx->dxva2.decoder)
|
|
#define DXVA_CONTEXT_REPORT_ID(avctx, ctx) (*(&ctx->dxva2.report_id))
|
|
#define DXVA_CONTEXT_CFG(avctx, ctx) (ctx->dxva2.cfg)
|
|
#define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx) (ctx->dxva2.cfg->ConfigBitstreamRaw)
|
|
#define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (ctx->dxva2.cfg->ConfigIntraResidUnsigned)
|
|
#define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (ctx->dxva2.cfg->ConfigResidDiffAccelerator)
|
|
#define DXVA_CONTEXT_VALID(avctx, ctx) (ctx->dxva2.decoder && ctx->dxva2.cfg && ctx->dxva2.surface_count)
|
|
#elif CONFIG_D3D11VA
|
|
#define DXVA_CONTEXT_WORKAROUND(avctx, ctx) (ctx->d3d11va.workaround)
|
|
#define DXVA_CONTEXT_COUNT(avctx, ctx) (ctx->d3d11va.surface_count)
|
|
#define DXVA_CONTEXT_DECODER(avctx, ctx) (ctx->d3d11va.decoder)
|
|
#define DXVA_CONTEXT_REPORT_ID(avctx, ctx) (*(&ctx->d3d11va.report_id))
|
|
#define DXVA_CONTEXT_CFG(avctx, ctx) (ctx->d3d11va.cfg)
|
|
#define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx) (ctx->d3d11va.cfg->ConfigBitstreamRaw)
|
|
#define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (ctx->d3d11va.cfg->ConfigIntraResidUnsigned)
|
|
#define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (ctx->d3d11va.cfg->ConfigResidDiffAccelerator)
|
|
#define DXVA_CONTEXT_VALID(avctx, ctx) (ctx->d3d11va.decoder && ctx->d3d11va.cfg)
|
|
#endif
|
|
|
|
unsigned ff_dxva2_get_surface_index(const AVCodecContext *avctx,
|
|
const AVDXVAContext *,
|
|
const AVFrame *frame);
|
|
|
|
int ff_dxva2_commit_buffer(AVCodecContext *, AVDXVAContext *,
|
|
DECODER_BUFFER_DESC *,
|
|
unsigned type, const void *data, unsigned size,
|
|
unsigned mb_count);
|
|
|
|
|
|
int ff_dxva2_common_end_frame(AVCodecContext *, AVFrame *,
|
|
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));
|
|
|
|
int ff_dxva2_decode_init(AVCodecContext *avctx);
|
|
|
|
int ff_dxva2_decode_uninit(AVCodecContext *avctx);
|
|
|
|
int ff_dxva2_is_d3d11(const AVCodecContext *avctx);
|
|
|
|
#endif /* AVCODEC_DXVA2_INTERNAL_H */
|