dxva: preparations for new hwaccel API
[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 FFmpeg.
7  *
8  * FFmpeg 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  * FFmpeg 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 FFmpeg; 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_DXVA2_INTERNAL_H
24 #define AVCODEC_DXVA2_INTERNAL_H
25
26 #define COBJMACROS
27
28 #include "config.h"
29
30 /* define the proper COM entries before forcing desktop APIs */
31 #include <objbase.h>
32
33 #if CONFIG_DXVA2
34 #include "dxva2.h"
35 #endif
36 #if CONFIG_D3D11VA
37 #include "d3d11va.h"
38 #endif
39 #if HAVE_DXVA_H
40 /* When targeting WINAPI_FAMILY_PHONE_APP or WINAPI_FAMILY_APP, dxva.h
41  * defines nothing. Force the struct definitions to be visible. */
42 #undef WINAPI_FAMILY
43 #define WINAPI_FAMILY WINAPI_FAMILY_DESKTOP_APP
44 #undef _CRT_BUILD_DESKTOP_APP
45 #define _CRT_BUILD_DESKTOP_APP 0
46 #include <dxva.h>
47 #endif
48
49 #include "avcodec.h"
50
51 typedef void DECODER_BUFFER_DESC;
52
53 typedef union {
54 #if CONFIG_D3D11VA
55     struct AVD3D11VAContext  d3d11va;
56 #endif
57 #if CONFIG_DXVA2
58     struct dxva_context      dxva2;
59 #endif
60 } AVDXVAContext;
61
62 #define DXVA_CONTEXT(avctx) ((AVDXVAContext *)(avctx)->hwaccel_context)
63
64 #define D3D11VA_CONTEXT(ctx) (&ctx->d3d11va)
65 #define DXVA2_CONTEXT(ctx)   (&ctx->dxva2)
66
67 #if CONFIG_D3D11VA && CONFIG_DXVA2
68 #define DXVA_CONTEXT_WORKAROUND(avctx, ctx)     (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.workaround : ctx->dxva2.workaround)
69 #define DXVA_CONTEXT_COUNT(avctx, ctx)          (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.surface_count : ctx->dxva2.surface_count)
70 #define DXVA_CONTEXT_DECODER(avctx, ctx)        (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.decoder : ctx->dxva2.decoder)
71 #define DXVA_CONTEXT_REPORT_ID(avctx, ctx)      (*(ff_dxva2_is_d3d11(avctx) ? &ctx->d3d11va.report_id : &ctx->dxva2.report_id))
72 #define DXVA_CONTEXT_CFG(avctx, ctx)            (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.cfg : ctx->dxva2.cfg)
73 #define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx)  (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.cfg->ConfigBitstreamRaw : ctx->dxva2.cfg->ConfigBitstreamRaw)
74 #define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.cfg->ConfigIntraResidUnsigned : ctx->dxva2.cfg->ConfigIntraResidUnsigned)
75 #define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.cfg->ConfigResidDiffAccelerator : ctx->dxva2.cfg->ConfigResidDiffAccelerator)
76 #define DXVA_CONTEXT_VALID(avctx, ctx)          (DXVA_CONTEXT_DECODER(avctx, ctx) && \
77                                                  DXVA_CONTEXT_CFG(avctx, ctx)     && \
78                                                  (ff_dxva2_is_d3d11(avctx) || ctx->dxva2.surface_count))
79 #elif CONFIG_DXVA2
80 #define DXVA_CONTEXT_WORKAROUND(avctx, ctx)     (ctx->dxva2.workaround)
81 #define DXVA_CONTEXT_COUNT(avctx, ctx)          (ctx->dxva2.surface_count)
82 #define DXVA_CONTEXT_DECODER(avctx, ctx)        (ctx->dxva2.decoder)
83 #define DXVA_CONTEXT_REPORT_ID(avctx, ctx)      (*(&ctx->dxva2.report_id))
84 #define DXVA_CONTEXT_CFG(avctx, ctx)            (ctx->dxva2.cfg)
85 #define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx)  (ctx->dxva2.cfg->ConfigBitstreamRaw)
86 #define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (ctx->dxva2.cfg->ConfigIntraResidUnsigned)
87 #define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (ctx->dxva2.cfg->ConfigResidDiffAccelerator)
88 #define DXVA_CONTEXT_VALID(avctx, ctx)          (ctx->dxva2.decoder && ctx->dxva2.cfg && ctx->dxva2.surface_count)
89 #elif CONFIG_D3D11VA
90 #define DXVA_CONTEXT_WORKAROUND(avctx, ctx)     (ctx->d3d11va.workaround)
91 #define DXVA_CONTEXT_COUNT(avctx, ctx)          (ctx->d3d11va.surface_count)
92 #define DXVA_CONTEXT_DECODER(avctx, ctx)        (ctx->d3d11va.decoder)
93 #define DXVA_CONTEXT_REPORT_ID(avctx, ctx)      (*(&ctx->d3d11va.report_id))
94 #define DXVA_CONTEXT_CFG(avctx, ctx)            (ctx->d3d11va.cfg)
95 #define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx)  (ctx->d3d11va.cfg->ConfigBitstreamRaw)
96 #define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (ctx->d3d11va.cfg->ConfigIntraResidUnsigned)
97 #define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (ctx->d3d11va.cfg->ConfigResidDiffAccelerator)
98 #define DXVA_CONTEXT_VALID(avctx, ctx)          (ctx->d3d11va.decoder && ctx->d3d11va.cfg)
99 #endif
100
101 unsigned ff_dxva2_get_surface_index(const AVCodecContext *avctx,
102                                     const AVDXVAContext *,
103                                     const AVFrame *frame);
104
105 int ff_dxva2_commit_buffer(AVCodecContext *, AVDXVAContext *,
106                            DECODER_BUFFER_DESC *,
107                            unsigned type, const void *data, unsigned size,
108                            unsigned mb_count);
109
110
111 int ff_dxva2_common_end_frame(AVCodecContext *, AVFrame *,
112                               const void *pp, unsigned pp_size,
113                               const void *qm, unsigned qm_size,
114                               int (*commit_bs_si)(AVCodecContext *,
115                                                   DECODER_BUFFER_DESC *bs,
116                                                   DECODER_BUFFER_DESC *slice));
117
118 int ff_dxva2_is_d3d11(const AVCodecContext *avctx);
119
120 #endif /* AVCODEC_DXVA2_INTERNAL_H */