|
- /*
- * 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"
- #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;
-
- #define D3D11VA_CONTEXT(ctx) (&ctx->d3d11va)
- #define DXVA2_CONTEXT(ctx) (&ctx->dxva2)
-
- #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_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)
- #define DXVA_CONTEXT_VALID(avctx, ctx) (DXVA_CONTEXT_DECODER(avctx, ctx) && \
- DXVA_CONTEXT_CFG(avctx, ctx) && \
- (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD || 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));
-
- #endif /* AVCODEC_DXVA2_INTERNAL_H */
|