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 D3D11VA_CONTEXT(ctx) (&ctx->d3d11va)
  63 #define DXVA2_CONTEXT(ctx)   (&ctx->dxva2)
  64
  65 #if CONFIG_D3D11VA && CONFIG_DXVA2
  66 #define DXVA_CONTEXT_WORKAROUND(avctx, ctx)     (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.workaround : ctx->dxva2.workaround)
  67 #define DXVA_CONTEXT_COUNT(avctx, ctx)          (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.surface_count : ctx->dxva2.surface_count)
  68 #define DXVA_CONTEXT_DECODER(avctx, ctx)        (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.decoder : ctx->dxva2.decoder)
  69 #define DXVA_CONTEXT_REPORT_ID(avctx, ctx)      (*(avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? &ctx->d3d11va.report_id : &ctx->dxva2.report_id))
  70 #define DXVA_CONTEXT_CFG(avctx, ctx)            (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.cfg : ctx->dxva2.cfg)
  71 #define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx)  (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.cfg->ConfigBitstreamRaw : ctx->dxva2.cfg->ConfigBitstreamRaw)
  72 #define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.cfg->ConfigIntraResidUnsigned : ctx->dxva2.cfg->ConfigIntraResidUnsigned)
  73 #define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ? ctx->d3d11va.cfg->ConfigResidDiffAccelerator : ctx->dxva2.cfg->ConfigResidDiffAccelerator)
  74 #define DXVA_CONTEXT_VALID(avctx, ctx)          (DXVA_CONTEXT_DECODER(avctx, ctx) && \
  75                                                  DXVA_CONTEXT_CFG(avctx, ctx)     && \
  76                                                  (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD || ctx->dxva2.surface_count))
  77 #elif CONFIG_DXVA2
  78 #define DXVA_CONTEXT_WORKAROUND(avctx, ctx)     (ctx->dxva2.workaround)
  79 #define DXVA_CONTEXT_COUNT(avctx, ctx)          (ctx->dxva2.surface_count)
  80 #define DXVA_CONTEXT_DECODER(avctx, ctx)        (ctx->dxva2.decoder)
  81 #define DXVA_CONTEXT_REPORT_ID(avctx, ctx)      (*(&ctx->dxva2.report_id))
  82 #define DXVA_CONTEXT_CFG(avctx, ctx)            (ctx->dxva2.cfg)
  83 #define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx)  (ctx->dxva2.cfg->ConfigBitstreamRaw)
  84 #define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (ctx->dxva2.cfg->ConfigIntraResidUnsigned)
  85 #define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (ctx->dxva2.cfg->ConfigResidDiffAccelerator)
  86 #define DXVA_CONTEXT_VALID(avctx, ctx)          (ctx->dxva2.decoder && ctx->dxva2.cfg && ctx->dxva2.surface_count)
  87 #elif CONFIG_D3D11VA
  88 #define DXVA_CONTEXT_WORKAROUND(avctx, ctx)     (ctx->d3d11va.workaround)
  89 #define DXVA_CONTEXT_COUNT(avctx, ctx)          (ctx->d3d11va.surface_count)
  90 #define DXVA_CONTEXT_DECODER(avctx, ctx)        (ctx->d3d11va.decoder)
  91 #define DXVA_CONTEXT_REPORT_ID(avctx, ctx)      (*(&ctx->d3d11va.report_id))
  92 #define DXVA_CONTEXT_CFG(avctx, ctx)            (ctx->d3d11va.cfg)
  93 #define DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx)  (ctx->d3d11va.cfg->ConfigBitstreamRaw)
  94 #define DXVA_CONTEXT_CFG_INTRARESID(avctx, ctx) (ctx->d3d11va.cfg->ConfigIntraResidUnsigned)
  95 #define DXVA_CONTEXT_CFG_RESIDACCEL(avctx, ctx) (ctx->d3d11va.cfg->ConfigResidDiffAccelerator)
  96 #define DXVA_CONTEXT_VALID(avctx, ctx)          (ctx->d3d11va.decoder && ctx->d3d11va.cfg)
  97 #endif
  98
  99 unsigned ff_dxva2_get_surface_index(const AVCodecContext *avctx,
 100                                     const AVDXVAContext *,
 101                                     const AVFrame *frame);
 102
 103 int ff_dxva2_commit_buffer(AVCodecContext *, AVDXVAContext *,
 104                            DECODER_BUFFER_DESC *,
 105                            unsigned type, const void *data, unsigned size,
 106                            unsigned mb_count);
 107
 108
 109 int ff_dxva2_common_end_frame(AVCodecContext *, AVFrame *,
 110                               const void *pp, unsigned pp_size,
 111                               const void *qm, unsigned qm_size,
 112                               int (*commit_bs_si)(AVCodecContext *,
 113                                                   DECODER_BUFFER_DESC *bs,
 114                                                   DECODER_BUFFER_DESC *slice));
 115
 116 #endif /* AVCODEC_DXVA2_INTERNAL_H */