D3D11va: add a Direct3D11 video decoder similar to DXVA2
[ffmpeg.git] / libavcodec / dxva2_internal.h
1 /*
2  * DXVA2 HW acceleration
3  *
4  * copyright (c) 2010 Laurent Aimar
5  *
6  * This file is part of Libav.
7  *
8  * Libav is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * Libav is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with Libav; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22
23 #ifndef AVCODEC_DXVA_INTERNAL_H
24 #define AVCODEC_DXVA_INTERNAL_H
25
26 #define COBJMACROS
27
28 #include "config.h"
29
30 #include "dxva2.h"
31 #include "d3d11va.h"
32 #if HAVE_DXVA_H
33 #include <dxva.h>
34 #endif
35
36 #include "avcodec.h"
37 #include "mpegvideo.h"
38
39 typedef void DECODER_BUFFER_DESC;
40
41 typedef union {
42     struct AVD3D11VAContext  d3d11va;
43     struct dxva_context      dxva2;
44 } AVDXVAContext;
45
46 #define D3D11VA_CONTEXT(ctx) (&ctx->d3d11va)
47 #define DXVA2_CONTEXT(ctx)   (&ctx->dxva2)
48
49 #define DXVA_CONTEXT_WORKAROUND(avctx, ctx)     (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.workaround : ctx->dxva2.workaround)
50 #define DXVA_CONTEXT_COUNT(avctx, ctx)          (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.surface_count : ctx->dxva2.surface_count)
51 #define DXVA_CONTEXT_SURFACE(avctx, ctx, i)     (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.surface[i] : ctx->dxva2.surface[i])
52 #define DXVA_CONTEXT_DECODER(avctx, ctx)        (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.decoder : ctx->dxva2.decoder)
53 #define DXVA_CONTEXT_REPORT_ID(avctx, ctx)      (*(avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? &ctx->d3d11va.report_id : &ctx->dxva2.report_id))
54 #define DXVA_CONTEXT_CFG(avctx, ctx)            (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.cfg : ctx->dxva2.cfg)
55 #define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx)  (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.cfg->ConfigBitstreamRaw : ctx->dxva2.cfg->ConfigBitstreamRaw)
56 #define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.cfg->ConfigIntraResidUnsigned : ctx->dxva2.cfg->ConfigIntraResidUnsigned)
57 #define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.cfg->ConfigResidDiffAccelerator : ctx->dxva2.cfg->ConfigResidDiffAccelerator)
58
59 void *ff_dxva2_get_surface(const AVFrame *frame);
60
61 unsigned ff_dxva2_get_surface_index(const AVCodecContext *avctx,
62                                     const AVDXVAContext *,
63                                     const AVFrame *frame);
64
65 int ff_dxva2_commit_buffer(AVCodecContext *, AVDXVAContext *,
66                            DECODER_BUFFER_DESC *,
67                            unsigned type, const void *data, unsigned size,
68                            unsigned mb_count);
69
70
71 int ff_dxva2_common_end_frame(AVCodecContext *, AVFrame *,
72                               const void *pp, unsigned pp_size,
73                               const void *qm, unsigned qm_size,
74                               int (*commit_bs_si)(AVCodecContext *,
75                                                   DECODER_BUFFER_DESC *bs,
76                                                   DECODER_BUFFER_DESC *slice));
77
78 #endif /* AVCODEC_DXVA_INTERNAL_H */