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_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
  56 typedef void DECODER_BUFFER_DESC;
  57
  58 typedef 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
  67 typedef 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) ? ctx->d3d11va.decoder : 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) ? ctx->d3d11va.cfg : 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
 138 unsigned ff_dxva2_get_surface_index(const AVCodecContext *avctx,
 139                                     const AVDXVAContext *,
 140                                     const AVFrame *frame);
 141
 142 int ff_dxva2_commit_buffer(AVCodecContext *, AVDXVAContext *,
 143                            DECODER_BUFFER_DESC *,
 144                            unsigned type, const void *data, unsigned size,
 145                            unsigned mb_count);
 146
 147
 148 int 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
 155 int ff_dxva2_decode_init(AVCodecContext *avctx);
 156
 157 int ff_dxva2_decode_uninit(AVCodecContext *avctx);
 158
 159 int ff_dxva2_common_frame_params(AVCodecContext *avctx,
 160                                  AVBufferRef *hw_frames_ctx);
 161
 162 int ff_dxva2_is_d3d11(const AVCodecContext *avctx);
 163
 164 #endif /* AVCODEC_DXVA2_INTERNAL_H */