121 lines
		
	
	
		
			5.4 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			121 lines
		
	
	
		
			5.4 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_DXVA_INTERNAL_H
 | 
						|
#define AVCODEC_DXVA_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"
 | 
						|
#endif
 | 
						|
#if CONFIG_D3D11VA
 | 
						|
#include "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 "avcodec.h"
 | 
						|
 | 
						|
typedef void DECODER_BUFFER_DESC;
 | 
						|
 | 
						|
typedef union {
 | 
						|
#if CONFIG_D3D11VA
 | 
						|
    struct AVD3D11VAContext  d3d11va;
 | 
						|
#endif
 | 
						|
#if CONFIG_DXVA2
 | 
						|
    struct dxva_context      dxva2;
 | 
						|
#endif
 | 
						|
} AVDXVAContext;
 | 
						|
 | 
						|
#if CONFIG_D3D11VA
 | 
						|
#define D3D11VA_CONTEXT(ctx) (&ctx->d3d11va)
 | 
						|
#endif
 | 
						|
#if CONFIG_DXVA2
 | 
						|
#define DXVA2_CONTEXT(ctx)   (&ctx->dxva2)
 | 
						|
#endif
 | 
						|
 | 
						|
#if CONFIG_D3D11VA && CONFIG_DXVA2
 | 
						|
#define DXVA_CONTEXT_WORKAROUND(avctx, ctx)     (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.workaround : ctx->dxva2.workaround)
 | 
						|
#define DXVA_CONTEXT_COUNT(avctx, ctx)          (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.surface_count : ctx->dxva2.surface_count)
 | 
						|
#define DXVA_CONTEXT_SURFACE(avctx, ctx, i)     (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.surface[i] : ctx->dxva2.surface[i])
 | 
						|
#define DXVA_CONTEXT_DECODER(avctx, ctx)        (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.decoder : ctx->dxva2.decoder)
 | 
						|
#define DXVA_CONTEXT_REPORT_ID(avctx, ctx)      (*(avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? &ctx->d3d11va.report_id : &ctx->dxva2.report_id))
 | 
						|
#define DXVA_CONTEXT_CFG(avctx, ctx)            (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.cfg : ctx->dxva2.cfg)
 | 
						|
#define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx)  (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.cfg->ConfigBitstreamRaw : ctx->dxva2.cfg->ConfigBitstreamRaw)
 | 
						|
#define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.cfg->ConfigIntraResidUnsigned : ctx->dxva2.cfg->ConfigIntraResidUnsigned)
 | 
						|
#define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.cfg->ConfigResidDiffAccelerator : ctx->dxva2.cfg->ConfigResidDiffAccelerator)
 | 
						|
#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_SURFACE(avctx, ctx, i)     (ctx->dxva2.surface[i])
 | 
						|
#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)
 | 
						|
#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_SURFACE(avctx, ctx, i)     (ctx->d3d11va.surface[i])
 | 
						|
#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)
 | 
						|
#endif
 | 
						|
 | 
						|
void *ff_dxva2_get_surface(const AVFrame *frame);
 | 
						|
 | 
						|
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));
 | 
						|
 | 
						|
#endif /* AVCODEC_DXVA_INTERNAL_H */
 |