165 lines
6.9 KiB
C
165 lines
6.9 KiB
C
|
/*
|
||
|
|
* DXVA2 HW acceleration
|
||
|
|
*
|
||
|
|
* copyright (c) 2010 Laurent Aimar
|
||
|
|
*
|
||
|
|
* 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
|
||
|
|
*/
|
||
|
|
|
||
|
|
#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) ? (void *)ctx->d3d11va.decoder : (void *)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) ? (void *)ctx->d3d11va.cfg : (void *)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_common_frame_params(AVCodecContext *avctx,
|
||
|
|
AVBufferRef *hw_frames_ctx);
|
||
|
|
|
||
|
|
int ff_dxva2_is_d3d11(const AVCodecContext *avctx);
|
||
|
|
|
||
|
|
#endif /* AVCODEC_DXVA2_INTERNAL_H */
|