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#include "libavutil/hwcontext_dxva2.h"
36#endif
37#if CONFIG_D3D11VA
38#include "d3d11va.h"
39#include "libavutil/hwcontext_d3d11va.h"
40#endif
41#if HAVE_DXVA_H
42/* When targeting WINAPI_FAMILY_PHONE_APP or WINAPI_FAMILY_APP, dxva.h
43 * defines nothing. Force the struct definitions to be visible. */
44#undef WINAPI_FAMILY
45#define WINAPI_FAMILY WINAPI_FAMILY_DESKTOP_APP
46#undef _CRT_BUILD_DESKTOP_APP
47#define _CRT_BUILD_DESKTOP_APP 0
48#include <dxva.h>
49#endif
50
51#include "libavutil/hwcontext.h"
52
53#include "avcodec.h"
54#include "internal.h"
55
56typedef void DECODER_BUFFER_DESC;
57
58typedef union {
59#if CONFIG_D3D11VA
60    struct AVD3D11VAContext  d3d11va;
61#endif
62#if CONFIG_DXVA2
63    struct dxva_context      dxva2;
64#endif
65} AVDXVAContext;
66
67typedef struct FFDXVASharedContext {
68    AVBufferRef *decoder_ref;
69
70    // FF_DXVA2_WORKAROUND_* flags
71    uint64_t workaround;
72
73    // E.g. AV_PIX_FMT_D3D11 (same as AVCodecContext.pix_fmt, except during init)
74    enum AVPixelFormat pix_fmt;
75
76    AVHWDeviceContext *device_ctx;
77
78#if CONFIG_D3D11VA
79    ID3D11VideoDecoder             *d3d11_decoder;
80    D3D11_VIDEO_DECODER_CONFIG      d3d11_config;
81    ID3D11VideoDecoderOutputView  **d3d11_views;
82    int                          nb_d3d11_views;
83    ID3D11Texture2D                *d3d11_texture;
84#endif
85
86#if CONFIG_DXVA2
87    IDirectXVideoDecoder           *dxva2_decoder;
88    IDirectXVideoDecoderService    *dxva2_service;
89    DXVA2_ConfigPictureDecode       dxva2_config;
90#endif
91
92    // Legacy (but used by code outside of setup)
93    // In generic mode, DXVA_CONTEXT() will return a pointer to this.
94    AVDXVAContext ctx;
95} FFDXVASharedContext;
96
97#define DXVA_SHARED_CONTEXT(avctx) ((FFDXVASharedContext *)((avctx)->internal->hwaccel_priv_data))
98
99#define DXVA_CONTEXT(avctx) (AVDXVAContext *)((avctx)->hwaccel_context ? (avctx)->hwaccel_context : (&(DXVA_SHARED_CONTEXT(avctx)->ctx)))
100
101#define D3D11VA_CONTEXT(ctx) (&ctx->d3d11va)
102#define DXVA2_CONTEXT(ctx)   (&ctx->dxva2)
103
104#if CONFIG_D3D11VA && CONFIG_DXVA2
105#define DXVA_CONTEXT_WORKAROUND(avctx, ctx)     (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.workaround : ctx->dxva2.workaround)
106#define DXVA_CONTEXT_COUNT(avctx, ctx)          (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.surface_count : ctx->dxva2.surface_count)
107#define DXVA_CONTEXT_DECODER(avctx, ctx)        (ff_dxva2_is_d3d11(avctx) ? (void *)ctx->d3d11va.decoder : (void *)ctx->dxva2.decoder)
108#define DXVA_CONTEXT_REPORT_ID(avctx, ctx)      (*(ff_dxva2_is_d3d11(avctx) ? &ctx->d3d11va.report_id : &ctx->dxva2.report_id))
109#define DXVA_CONTEXT_CFG(avctx, ctx)            (ff_dxva2_is_d3d11(avctx) ? (void *)ctx->d3d11va.cfg : (void *)ctx->dxva2.cfg)
110#define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx)  (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.cfg->ConfigBitstreamRaw : ctx->dxva2.cfg->ConfigBitstreamRaw)
111#define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.cfg->ConfigIntraResidUnsigned : ctx->dxva2.cfg->ConfigIntraResidUnsigned)
112#define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (ff_dxva2_is_d3d11(avctx) ? ctx->d3d11va.cfg->ConfigResidDiffAccelerator : ctx->dxva2.cfg->ConfigResidDiffAccelerator)
113#define DXVA_CONTEXT_VALID(avctx, ctx)          (DXVA_CONTEXT_DECODER(avctx, ctx) && \
114                                                 DXVA_CONTEXT_CFG(avctx, ctx)     && \
115                                                 (ff_dxva2_is_d3d11(avctx) || ctx->dxva2.surface_count))
116#elif CONFIG_DXVA2
117#define DXVA_CONTEXT_WORKAROUND(avctx, ctx)     (ctx->dxva2.workaround)
118#define DXVA_CONTEXT_COUNT(avctx, ctx)          (ctx->dxva2.surface_count)
119#define DXVA_CONTEXT_DECODER(avctx, ctx)        (ctx->dxva2.decoder)
120#define DXVA_CONTEXT_REPORT_ID(avctx, ctx)      (*(&ctx->dxva2.report_id))
121#define DXVA_CONTEXT_CFG(avctx, ctx)            (ctx->dxva2.cfg)
122#define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx)  (ctx->dxva2.cfg->ConfigBitstreamRaw)
123#define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (ctx->dxva2.cfg->ConfigIntraResidUnsigned)
124#define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (ctx->dxva2.cfg->ConfigResidDiffAccelerator)
125#define DXVA_CONTEXT_VALID(avctx, ctx)          (ctx->dxva2.decoder && ctx->dxva2.cfg && ctx->dxva2.surface_count)
126#elif CONFIG_D3D11VA
127#define DXVA_CONTEXT_WORKAROUND(avctx, ctx)     (ctx->d3d11va.workaround)
128#define DXVA_CONTEXT_COUNT(avctx, ctx)          (ctx->d3d11va.surface_count)
129#define DXVA_CONTEXT_DECODER(avctx, ctx)        (ctx->d3d11va.decoder)
130#define DXVA_CONTEXT_REPORT_ID(avctx, ctx)      (*(&ctx->d3d11va.report_id))
131#define DXVA_CONTEXT_CFG(avctx, ctx)            (ctx->d3d11va.cfg)
132#define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx)  (ctx->d3d11va.cfg->ConfigBitstreamRaw)
133#define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (ctx->d3d11va.cfg->ConfigIntraResidUnsigned)
134#define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (ctx->d3d11va.cfg->ConfigResidDiffAccelerator)
135#define DXVA_CONTEXT_VALID(avctx, ctx)          (ctx->d3d11va.decoder && ctx->d3d11va.cfg)
136#endif
137
138unsigned ff_dxva2_get_surface_index(const AVCodecContext *avctx,
139                                    const AVDXVAContext *,
140                                    const AVFrame *frame);
141
142int ff_dxva2_commit_buffer(AVCodecContext *, AVDXVAContext *,
143                           DECODER_BUFFER_DESC *,
144                           unsigned type, const void *data, unsigned size,
145                           unsigned mb_count);
146
147
148int ff_dxva2_common_end_frame(AVCodecContext *, AVFrame *,
149                              const void *pp, unsigned pp_size,
150                              const void *qm, unsigned qm_size,
151                              int (*commit_bs_si)(AVCodecContext *,
152                                                  DECODER_BUFFER_DESC *bs,
153                                                  DECODER_BUFFER_DESC *slice));
154
155int ff_dxva2_decode_init(AVCodecContext *avctx);
156
157int ff_dxva2_decode_uninit(AVCodecContext *avctx);
158
159int ff_dxva2_common_frame_params(AVCodecContext *avctx,
160                                 AVBufferRef *hw_frames_ctx);
161
162int ff_dxva2_is_d3d11(const AVCodecContext *avctx);
163
164#endif /* AVCODEC_DXVA2_INTERNAL_H */
165