git.sesse.net Git - ffmpeg/blob - libavcodec/dxva2.c

   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 #include <assert.h>
  24 #include <string.h>
  25 #include <initguid.h>
  26
  27 #include "libavutil/common.h"
  28 #include "libavutil/log.h"
  29 #include "libavutil/time.h"
  30
  31 #include "avcodec.h"
  32 #include "decode.h"
  33 #include "dxva2_internal.h"
  34
  35 /* define all the GUIDs used directly here,
  36  to avoid problems with inconsistent dxva2api.h versions in mingw-w64 and different MSVC version */
  37 DEFINE_GUID(ff_DXVA2_ModeMPEG2_VLD,      0xee27417f, 0x5e28,0x4e65,0xbe,0xea,0x1d,0x26,0xb5,0x08,0xad,0xc9);
  38 DEFINE_GUID(ff_DXVA2_ModeMPEG2and1_VLD,  0x86695f12, 0x340e,0x4f04,0x9f,0xd3,0x92,0x53,0xdd,0x32,0x74,0x60);
  39 DEFINE_GUID(ff_DXVA2_ModeH264_E,         0x1b81be68, 0xa0c7,0x11d3,0xb9,0x84,0x00,0xc0,0x4f,0x2e,0x73,0xc5);
  40 DEFINE_GUID(ff_DXVA2_ModeH264_F,         0x1b81be69, 0xa0c7,0x11d3,0xb9,0x84,0x00,0xc0,0x4f,0x2e,0x73,0xc5);
  41 DEFINE_GUID(ff_DXVADDI_Intel_ModeH264_E, 0x604F8E68, 0x4951,0x4C54,0x88,0xFE,0xAB,0xD2,0x5C,0x15,0xB3,0xD6);
  42 DEFINE_GUID(ff_DXVA2_ModeVC1_D,          0x1b81beA3, 0xa0c7,0x11d3,0xb9,0x84,0x00,0xc0,0x4f,0x2e,0x73,0xc5);
  43 DEFINE_GUID(ff_DXVA2_ModeVC1_D2010,      0x1b81beA4, 0xa0c7,0x11d3,0xb9,0x84,0x00,0xc0,0x4f,0x2e,0x73,0xc5);
  44 DEFINE_GUID(ff_DXVA2_ModeHEVC_VLD_Main,  0x5b11d51b, 0x2f4c,0x4452,0xbc,0xc3,0x09,0xf2,0xa1,0x16,0x0c,0xc0);
  45 DEFINE_GUID(ff_DXVA2_ModeHEVC_VLD_Main10,0x107af0e0, 0xef1a,0x4d19,0xab,0xa8,0x67,0xa1,0x63,0x07,0x3d,0x13);
  46 DEFINE_GUID(ff_DXVA2_ModeVP9_VLD_Profile0,0x463707f8,0xa1d0,0x4585,0x87,0x6d,0x83,0xaa,0x6d,0x60,0xb8,0x9e);
  47 DEFINE_GUID(ff_DXVA2_NoEncrypt,          0x1b81beD0, 0xa0c7,0x11d3,0xb9,0x84,0x00,0xc0,0x4f,0x2e,0x73,0xc5);
  48 DEFINE_GUID(ff_GUID_NULL,                0x00000000, 0x0000,0x0000,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00);
  49 DEFINE_GUID(ff_IID_IDirectXVideoDecoderService, 0xfc51a551,0xd5e7,0x11d9,0xaf,0x55,0x00,0x05,0x4e,0x43,0xff,0x02);
  50
  51 typedef struct dxva_mode {
  52     const GUID     *guid;
  53     enum AVCodecID codec;
  54     // List of supported profiles, terminated by a FF_PROFILE_UNKNOWN entry.
  55     // If NULL, don't check profile.
  56     const int      *profiles;
  57 } dxva_mode;
  58
  59 static const int prof_mpeg2_main[]   = {FF_PROFILE_MPEG2_SIMPLE,
  60                                         FF_PROFILE_MPEG2_MAIN,
  61                                         FF_PROFILE_UNKNOWN};
  62 static const int prof_h264_high[]    = {FF_PROFILE_H264_CONSTRAINED_BASELINE,
  63                                         FF_PROFILE_H264_MAIN,
  64                                         FF_PROFILE_H264_HIGH,
  65                                         FF_PROFILE_UNKNOWN};
  66 static const int prof_hevc_main[]    = {FF_PROFILE_HEVC_MAIN,
  67                                         FF_PROFILE_UNKNOWN};
  68 static const int prof_hevc_main10[]  = {FF_PROFILE_HEVC_MAIN_10,
  69                                         FF_PROFILE_UNKNOWN};
  70
  71 static const dxva_mode dxva_modes[] = {
  72     /* MPEG-2 */
  73     { &ff_DXVA2_ModeMPEG2_VLD,       AV_CODEC_ID_MPEG2VIDEO, prof_mpeg2_main },
  74     { &ff_DXVA2_ModeMPEG2and1_VLD,   AV_CODEC_ID_MPEG2VIDEO, prof_mpeg2_main },
  75
  76     /* H.264 */
  77     { &ff_DXVA2_ModeH264_F,          AV_CODEC_ID_H264, prof_h264_high },
  78     { &ff_DXVA2_ModeH264_E,          AV_CODEC_ID_H264, prof_h264_high },
  79     /* Intel specific H.264 mode */
  80     { &ff_DXVADDI_Intel_ModeH264_E,  AV_CODEC_ID_H264, prof_h264_high },
  81
  82     /* VC-1 / WMV3 */
  83     { &ff_DXVA2_ModeVC1_D2010,       AV_CODEC_ID_VC1 },
  84     { &ff_DXVA2_ModeVC1_D2010,       AV_CODEC_ID_WMV3 },
  85     { &ff_DXVA2_ModeVC1_D,           AV_CODEC_ID_VC1 },
  86     { &ff_DXVA2_ModeVC1_D,           AV_CODEC_ID_WMV3 },
  87
  88     /* HEVC/H.265 */
  89     { &ff_DXVA2_ModeHEVC_VLD_Main10, AV_CODEC_ID_HEVC, prof_hevc_main10 },
  90     { &ff_DXVA2_ModeHEVC_VLD_Main,   AV_CODEC_ID_HEVC, prof_hevc_main },
  91
  92     /* VP8/9 */
  93     { &ff_DXVA2_ModeVP9_VLD_Profile0,AV_CODEC_ID_VP9 },
  94
  95     { NULL,                          0 },
  96 };
  97
  98 static int dxva_get_decoder_configuration(AVCodecContext *avctx,
  99                                           const void *cfg_list,
 100                                           unsigned cfg_count)
 101 {
 102     FFDXVASharedContext *sctx = DXVA_SHARED_CONTEXT(avctx);
 103     unsigned i, best_score = 0;
 104     int best_cfg = -1;
 105
 106     for (i = 0; i < cfg_count; i++) {
 107         unsigned score;
 108         UINT ConfigBitstreamRaw;
 109         GUID guidConfigBitstreamEncryption;
 110
 111 #if CONFIG_D3D11VA
 112         if (sctx->pix_fmt == AV_PIX_FMT_D3D11) {
 113             D3D11_VIDEO_DECODER_CONFIG *cfg = &((D3D11_VIDEO_DECODER_CONFIG *)cfg_list)[i];
 114             ConfigBitstreamRaw = cfg->ConfigBitstreamRaw;
 115             guidConfigBitstreamEncryption = cfg->guidConfigBitstreamEncryption;
 116         }
 117 #endif
 118 #if CONFIG_DXVA2
 119         if (sctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) {
 120             DXVA2_ConfigPictureDecode *cfg = &((DXVA2_ConfigPictureDecode *)cfg_list)[i];
 121             ConfigBitstreamRaw = cfg->ConfigBitstreamRaw;
 122             guidConfigBitstreamEncryption = cfg->guidConfigBitstreamEncryption;
 123         }
 124 #endif
 125
 126         if (ConfigBitstreamRaw == 1)
 127             score = 1;
 128         else if (avctx->codec_id == AV_CODEC_ID_H264 && ConfigBitstreamRaw == 2)
 129             score = 2;
 130         else
 131             continue;
 132         if (IsEqualGUID(&guidConfigBitstreamEncryption, &ff_DXVA2_NoEncrypt))
 133             score += 16;
 134         if (score > best_score) {
 135             best_score = score;
 136             best_cfg = i;
 137         }
 138     }
 139
 140     if (!best_score) {
 141         av_log(avctx, AV_LOG_VERBOSE, "No valid decoder configuration available\n");
 142         return AVERROR(EINVAL);
 143     }
 144
 145     return best_cfg;
 146 }
 147
 148 #if CONFIG_D3D11VA
 149 static int d3d11va_validate_output(void *service, GUID guid, const void *surface_format)
 150 {
 151     HRESULT hr;
 152     BOOL is_supported = FALSE;
 153     hr = ID3D11VideoDevice_CheckVideoDecoderFormat((ID3D11VideoDevice *)service,
 154                                                    &guid,
 155                                                    *(DXGI_FORMAT *)surface_format,
 156                                                    &is_supported);
 157     return SUCCEEDED(hr) && is_supported;
 158 }
 159 #endif
 160
 161 #if CONFIG_DXVA2
 162 static int dxva2_validate_output(void *decoder_service, GUID guid, const void *surface_format)
 163 {
 164     HRESULT hr;
 165     int ret = 0;
 166     unsigned j, target_count;
 167     D3DFORMAT *target_list;
 168     hr = IDirectXVideoDecoderService_GetDecoderRenderTargets((IDirectXVideoDecoderService *)decoder_service, &guid, &target_count, &target_list);
 169     if (SUCCEEDED(hr)) {
 170         for (j = 0; j < target_count; j++) {
 171             const D3DFORMAT format = target_list[j];
 172             if (format == *(D3DFORMAT *)surface_format) {
 173                 ret = 1;
 174                 break;
 175             }
 176         }
 177         CoTaskMemFree(target_list);
 178     }
 179     return ret;
 180 }
 181 #endif
 182
 183 static int dxva_check_codec_compatibility(AVCodecContext *avctx, const dxva_mode *mode)
 184 {
 185     if (mode->codec != avctx->codec_id)
 186             return 0;
 187
 188     if (mode->profiles && !(avctx->hwaccel_flags & AV_HWACCEL_FLAG_ALLOW_PROFILE_MISMATCH)) {
 189         int i, found = 0;
 190         for (i = 0; mode->profiles[i] != FF_PROFILE_UNKNOWN; i++) {
 191             if (avctx->profile == mode->profiles[i]) {
 192                 found = 1;
 193                 break;
 194             }
 195         }
 196         if (!found)
 197             return 0;
 198     }
 199
 200     return 1;
 201 }
 202
 203 static void dxva_list_guids_debug(AVCodecContext *avctx, void *service,
 204                                  unsigned guid_count, const GUID *guid_list)
 205 {
 206     FFDXVASharedContext *sctx = DXVA_SHARED_CONTEXT(avctx);
 207     int i;
 208
 209     av_log(avctx, AV_LOG_VERBOSE, "Decoder GUIDs reported as supported:\n");
 210
 211     for (i = 0; i < guid_count; i++) {
 212         const GUID *guid = &guid_list[i];
 213
 214         av_log(avctx, AV_LOG_VERBOSE,
 215              "{%8.8x-%4.4x-%4.4x-%2.2x%2.2x-%2.2x%2.2x%2.2x%2.2x%2.2x%2.2x}",
 216              (unsigned) guid->Data1, guid->Data2, guid->Data3,
 217              guid->Data4[0], guid->Data4[1],
 218              guid->Data4[2], guid->Data4[3],
 219              guid->Data4[4], guid->Data4[5],
 220              guid->Data4[6], guid->Data4[7]);
 221
 222 #if CONFIG_D3D11VA
 223         if (sctx->pix_fmt == AV_PIX_FMT_D3D11) {
 224             DXGI_FORMAT format;
 225             // We don't know the maximum valid DXGI_FORMAT, so use 200 as
 226             // arbitrary upper bound (that could become outdated).
 227             for (format = 0; format < 200; format++) {
 228                 if (d3d11va_validate_output(service, *guid, &format))
 229                     av_log(avctx, AV_LOG_VERBOSE, " %d", (int)format);
 230             }
 231         }
 232 #endif
 233 #if CONFIG_DXVA2
 234         if (sctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) {
 235             const D3DFORMAT formats[] = {MKTAG('N', 'V', '1', '2'),
 236                                          MKTAG('P', '0', '1', '0')};
 237             int i;
 238             for (i = 0; i < FF_ARRAY_ELEMS(formats); i++) {
 239                 if (dxva2_validate_output(service, *guid, &formats[i]))
 240                     av_log(avctx, AV_LOG_VERBOSE, " %d", i);
 241             }
 242         }
 243 #endif
 244         av_log(avctx, AV_LOG_VERBOSE, "\n");
 245     }
 246 }
 247
 248 static int dxva_get_decoder_guid(AVCodecContext *avctx, void *service, void *surface_format,
 249                                  unsigned guid_count, const GUID *guid_list, GUID *decoder_guid)
 250 {
 251     FFDXVASharedContext *sctx = DXVA_SHARED_CONTEXT(avctx);
 252     unsigned i, j;
 253
 254     dxva_list_guids_debug(avctx, service, guid_count, guid_list);
 255
 256     *decoder_guid = ff_GUID_NULL;
 257     for (i = 0; dxva_modes[i].guid; i++) {
 258         const dxva_mode *mode = &dxva_modes[i];
 259         int validate;
 260         if (!dxva_check_codec_compatibility(avctx, mode))
 261             continue;
 262
 263         for (j = 0; j < guid_count; j++) {
 264             if (IsEqualGUID(mode->guid, &guid_list[j]))
 265                 break;
 266         }
 267         if (j == guid_count)
 268             continue;
 269
 270 #if CONFIG_D3D11VA
 271         if (sctx->pix_fmt == AV_PIX_FMT_D3D11)
 272             validate = d3d11va_validate_output(service, *mode->guid, surface_format);
 273 #endif
 274 #if CONFIG_DXVA2
 275         if (sctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD)
 276             validate = dxva2_validate_output(service, *mode->guid, surface_format);
 277 #endif
 278         if (validate) {
 279             *decoder_guid = *mode->guid;
 280             break;
 281         }
 282     }
 283
 284     if (IsEqualGUID(decoder_guid, &ff_GUID_NULL)) {
 285         av_log(avctx, AV_LOG_VERBOSE, "No decoder device for codec found\n");
 286         return AVERROR(EINVAL);
 287     }
 288
 289     if (IsEqualGUID(decoder_guid, &ff_DXVADDI_Intel_ModeH264_E))
 290         sctx->workaround |= FF_DXVA2_WORKAROUND_INTEL_CLEARVIDEO;
 291
 292     return 0;
 293 }
 294
 295 static void bufref_free_interface(void *opaque, uint8_t *data)
 296 {
 297     IUnknown_Release((IUnknown *)opaque);
 298 }
 299
 300 static AVBufferRef *bufref_wrap_interface(IUnknown *iface)
 301 {
 302     return av_buffer_create((uint8_t*)iface, 1, bufref_free_interface, iface, 0);
 303 }
 304
 305 #if CONFIG_DXVA2
 306
 307 static int dxva2_get_decoder_configuration(AVCodecContext *avctx, const GUID *device_guid,
 308                                            const DXVA2_VideoDesc *desc,
 309                                            DXVA2_ConfigPictureDecode *config)
 310 {
 311     FFDXVASharedContext *sctx = DXVA_SHARED_CONTEXT(avctx);
 312     unsigned cfg_count;
 313     DXVA2_ConfigPictureDecode *cfg_list;
 314     HRESULT hr;
 315     int ret;
 316
 317     hr = IDirectXVideoDecoderService_GetDecoderConfigurations(sctx->dxva2_service, device_guid, desc, NULL, &cfg_count, &cfg_list);
 318     if (FAILED(hr)) {
 319         av_log(avctx, AV_LOG_ERROR, "Unable to retrieve decoder configurations\n");
 320         return AVERROR(EINVAL);
 321     }
 322
 323     ret = dxva_get_decoder_configuration(avctx, cfg_list, cfg_count);
 324     if (ret >= 0)
 325         *config = cfg_list[ret];
 326     CoTaskMemFree(cfg_list);
 327     return ret;
 328 }
 329
 330 static int dxva2_create_decoder(AVCodecContext *avctx)
 331 {
 332     FFDXVASharedContext *sctx = DXVA_SHARED_CONTEXT(avctx);
 333     GUID *guid_list;
 334     unsigned guid_count;
 335     GUID device_guid;
 336     D3DFORMAT surface_format = avctx->sw_pix_fmt == AV_PIX_FMT_YUV420P10 ?
 337                                MKTAG('P', '0', '1', '0') : MKTAG('N', 'V', '1', '2');
 338     DXVA2_VideoDesc desc = { 0 };
 339     DXVA2_ConfigPictureDecode config;
 340     HRESULT hr;
 341     int ret;
 342     HANDLE device_handle;
 343     AVHWFramesContext *frames_ctx = (AVHWFramesContext*)avctx->hw_frames_ctx->data;
 344     AVDXVA2FramesContext *frames_hwctx = frames_ctx->hwctx;
 345     AVDXVA2DeviceContext *device_hwctx = frames_ctx->device_ctx->hwctx;
 346
 347     hr = IDirect3DDeviceManager9_OpenDeviceHandle(device_hwctx->devmgr,
 348                                                   &device_handle);
 349     if (FAILED(hr)) {
 350         av_log(avctx, AV_LOG_ERROR, "Failed to open a device handle\n");
 351         goto fail;
 352     }
 353
 354     hr = IDirect3DDeviceManager9_GetVideoService(device_hwctx->devmgr, device_handle,
 355                                                  &ff_IID_IDirectXVideoDecoderService,
 356                                                  (void **)&sctx->dxva2_service);
 357     IDirect3DDeviceManager9_CloseDeviceHandle(device_hwctx->devmgr, device_handle);
 358     if (FAILED(hr)) {
 359         av_log(avctx, AV_LOG_ERROR, "Failed to create IDirectXVideoDecoderService\n");
 360         goto fail;
 361     }
 362
 363     hr = IDirectXVideoDecoderService_GetDecoderDeviceGuids(sctx->dxva2_service, &guid_count, &guid_list);
 364     if (FAILED(hr)) {
 365         av_log(avctx, AV_LOG_ERROR, "Failed to retrieve decoder device GUIDs\n");
 366         goto fail;
 367     }
 368
 369     ret = dxva_get_decoder_guid(avctx, sctx->dxva2_service, &surface_format,
 370                                 guid_count, guid_list, &device_guid);
 371     CoTaskMemFree(guid_list);
 372     if (ret < 0) {
 373         goto fail;
 374     }
 375
 376     desc.SampleWidth  = avctx->coded_width;
 377     desc.SampleHeight = avctx->coded_height;
 378     desc.Format       = surface_format;
 379
 380     ret = dxva2_get_decoder_configuration(avctx, &device_guid, &desc, &config);
 381     if (ret < 0) {
 382         goto fail;
 383     }
 384
 385     hr = IDirectXVideoDecoderService_CreateVideoDecoder(sctx->dxva2_service, &device_guid,
 386                                                         &desc, &config, frames_hwctx->surfaces,
 387                                                         frames_hwctx->nb_surfaces, &sctx->dxva2_decoder);
 388     if (FAILED(hr)) {
 389         av_log(avctx, AV_LOG_ERROR, "Failed to create DXVA2 video decoder\n");
 390         goto fail;
 391     }
 392
 393     sctx->dxva2_config = config;
 394
 395     sctx->decoder_ref = bufref_wrap_interface((IUnknown *)sctx->dxva2_decoder);
 396     if (!sctx->decoder_ref)
 397         return AVERROR(ENOMEM);
 398
 399     return 0;
 400 fail:
 401     return AVERROR(EINVAL);
 402 }
 403
 404 #endif
 405
 406 #if CONFIG_D3D11VA
 407
 408 static int d3d11va_get_decoder_configuration(AVCodecContext *avctx,
 409                                              ID3D11VideoDevice *video_device,
 410                                              const D3D11_VIDEO_DECODER_DESC *desc,
 411                                              D3D11_VIDEO_DECODER_CONFIG *config)
 412 {
 413     unsigned cfg_count = 0;
 414     D3D11_VIDEO_DECODER_CONFIG *cfg_list = NULL;
 415     HRESULT hr;
 416     int i, ret;
 417
 418     hr = ID3D11VideoDevice_GetVideoDecoderConfigCount(video_device, desc, &cfg_count);
 419     if (FAILED(hr)) {
 420         av_log(avctx, AV_LOG_ERROR, "Unable to retrieve decoder configurations\n");
 421         return AVERROR(EINVAL);
 422     }
 423
 424     cfg_list = av_malloc_array(cfg_count, sizeof(D3D11_VIDEO_DECODER_CONFIG));
 425     if (cfg_list == NULL)
 426         return AVERROR(ENOMEM);
 427     for (i = 0; i < cfg_count; i++) {
 428         hr = ID3D11VideoDevice_GetVideoDecoderConfig(video_device, desc, i, &cfg_list[i]);
 429         if (FAILED(hr)) {
 430             av_log(avctx, AV_LOG_ERROR, "Unable to retrieve decoder configurations. (hr=0x%lX)\n", hr);
 431             av_free(cfg_list);
 432             return AVERROR(EINVAL);
 433         }
 434     }
 435
 436     ret = dxva_get_decoder_configuration(avctx, cfg_list, cfg_count);
 437     if (ret >= 0)
 438         *config = cfg_list[ret];
 439     av_free(cfg_list);
 440     return ret;
 441 }
 442
 443 static DXGI_FORMAT d3d11va_map_sw_to_hw_format(enum AVPixelFormat pix_fmt)
 444 {
 445     switch (pix_fmt) {
 446     case AV_PIX_FMT_NV12:       return DXGI_FORMAT_NV12;
 447     case AV_PIX_FMT_P010:       return DXGI_FORMAT_P010;
 448     case AV_PIX_FMT_YUV420P:    return DXGI_FORMAT_420_OPAQUE;
 449     default:                    return DXGI_FORMAT_UNKNOWN;
 450     }
 451 }
 452
 453 static int d3d11va_create_decoder(AVCodecContext *avctx)
 454 {
 455     FFDXVASharedContext *sctx = DXVA_SHARED_CONTEXT(avctx);
 456     GUID *guid_list;
 457     unsigned guid_count, i;
 458     GUID decoder_guid;
 459     D3D11_VIDEO_DECODER_DESC desc = { 0 };
 460     D3D11_VIDEO_DECODER_CONFIG config;
 461     AVHWFramesContext *frames_ctx = (AVHWFramesContext *)avctx->hw_frames_ctx->data;
 462     AVD3D11VADeviceContext *device_hwctx = frames_ctx->device_ctx->hwctx;
 463     AVD3D11VAFramesContext *frames_hwctx = frames_ctx->hwctx;
 464     DXGI_FORMAT surface_format = d3d11va_map_sw_to_hw_format(frames_ctx->sw_format);
 465     D3D11_TEXTURE2D_DESC texdesc;
 466     HRESULT hr;
 467     int ret;
 468
 469     if (!frames_hwctx->texture) {
 470         av_log(avctx, AV_LOG_ERROR, "AVD3D11VAFramesContext.texture not set.\n");
 471         return AVERROR(EINVAL);
 472     }
 473     ID3D11Texture2D_GetDesc(frames_hwctx->texture, &texdesc);
 474
 475     guid_count = ID3D11VideoDevice_GetVideoDecoderProfileCount(device_hwctx->video_device);
 476     guid_list = av_malloc_array(guid_count, sizeof(*guid_list));
 477     if (guid_list == NULL || guid_count == 0) {
 478         av_log(avctx, AV_LOG_ERROR, "Failed to get the decoder GUIDs\n");
 479         av_free(guid_list);
 480         return AVERROR(EINVAL);
 481     }
 482     for (i = 0; i < guid_count; i++) {
 483         hr = ID3D11VideoDevice_GetVideoDecoderProfile(device_hwctx->video_device, i, &guid_list[i]);
 484         if (FAILED(hr)) {
 485             av_log(avctx, AV_LOG_ERROR, "Failed to retrieve decoder GUID %d\n", i);
 486             av_free(guid_list);
 487             return AVERROR(EINVAL);
 488         }
 489     }
 490
 491     ret = dxva_get_decoder_guid(avctx, device_hwctx->video_device, &surface_format,
 492                                 guid_count, guid_list, &decoder_guid);
 493     av_free(guid_list);
 494     if (ret < 0)
 495         return AVERROR(EINVAL);
 496
 497     desc.SampleWidth  = avctx->coded_width;
 498     desc.SampleHeight = avctx->coded_height;
 499     desc.OutputFormat = surface_format;
 500     desc.Guid         = decoder_guid;
 501
 502     ret = d3d11va_get_decoder_configuration(avctx, device_hwctx->video_device, &desc, &config);
 503     if (ret < 0)
 504         return AVERROR(EINVAL);
 505
 506     sctx->d3d11_views = av_mallocz_array(texdesc.ArraySize, sizeof(sctx->d3d11_views[0]));
 507     if (!sctx->d3d11_views)
 508         return AVERROR(ENOMEM);
 509     sctx->nb_d3d11_views = texdesc.ArraySize;
 510
 511     for (i = 0; i < sctx->nb_d3d11_views; i++) {
 512         D3D11_VIDEO_DECODER_OUTPUT_VIEW_DESC viewDesc = {
 513             .DecodeProfile = decoder_guid,
 514             .ViewDimension = D3D11_VDOV_DIMENSION_TEXTURE2D,
 515             .Texture2D = {
 516                 .ArraySlice = i,
 517             }
 518         };
 519         hr = ID3D11VideoDevice_CreateVideoDecoderOutputView(device_hwctx->video_device,
 520                                                             (ID3D11Resource*) frames_hwctx->texture,
 521                                                             &viewDesc,
 522                                                             (ID3D11VideoDecoderOutputView**) &sctx->d3d11_views[i]);
 523         if (FAILED(hr)) {
 524             av_log(avctx, AV_LOG_ERROR, "Could not create the decoder output view %d\n", i);
 525             return AVERROR_UNKNOWN;
 526         }
 527     }
 528
 529     hr = ID3D11VideoDevice_CreateVideoDecoder(device_hwctx->video_device, &desc,
 530                                               &config, &sctx->d3d11_decoder);
 531     if (FAILED(hr)) {
 532         av_log(avctx, AV_LOG_ERROR, "Failed to create D3D11VA video decoder\n");
 533         return AVERROR(EINVAL);
 534     }
 535
 536     sctx->d3d11_config = config;
 537     sctx->d3d11_texture = frames_hwctx->texture;
 538
 539     sctx->decoder_ref = bufref_wrap_interface((IUnknown *)sctx->d3d11_decoder);
 540     if (!sctx->decoder_ref)
 541         return AVERROR(ENOMEM);
 542
 543     return 0;
 544 }
 545
 546 #endif
 547
 548 static void ff_dxva2_lock(AVCodecContext *avctx)
 549 {
 550 #if CONFIG_D3D11VA
 551     if (ff_dxva2_is_d3d11(avctx)) {
 552         FFDXVASharedContext *sctx = DXVA_SHARED_CONTEXT(avctx);
 553         AVDXVAContext *ctx = DXVA_CONTEXT(avctx);
 554         if (D3D11VA_CONTEXT(ctx)->context_mutex != INVALID_HANDLE_VALUE)
 555             WaitForSingleObjectEx(D3D11VA_CONTEXT(ctx)->context_mutex, INFINITE, FALSE);
 556         if (sctx->device_ctx) {
 557             AVD3D11VADeviceContext *hwctx = sctx->device_ctx->hwctx;
 558             hwctx->lock(hwctx->lock_ctx);
 559         }
 560     }
 561 #endif
 562 }
 563
 564 static void ff_dxva2_unlock(AVCodecContext *avctx)
 565 {
 566 #if CONFIG_D3D11VA
 567     if (ff_dxva2_is_d3d11(avctx)) {
 568         FFDXVASharedContext *sctx = DXVA_SHARED_CONTEXT(avctx);
 569         AVDXVAContext *ctx = DXVA_CONTEXT(avctx);
 570         if (D3D11VA_CONTEXT(ctx)->context_mutex != INVALID_HANDLE_VALUE)
 571             ReleaseMutex(D3D11VA_CONTEXT(ctx)->context_mutex);
 572         if (sctx->device_ctx) {
 573             AVD3D11VADeviceContext *hwctx = sctx->device_ctx->hwctx;
 574             hwctx->unlock(hwctx->lock_ctx);
 575         }
 576     }
 577 #endif
 578 }
 579
 580 int ff_dxva2_common_frame_params(AVCodecContext *avctx,
 581                                  AVBufferRef *hw_frames_ctx)
 582 {
 583     AVHWFramesContext *frames_ctx = (AVHWFramesContext *)hw_frames_ctx->data;
 584     AVHWDeviceContext *device_ctx = frames_ctx->device_ctx;
 585     int surface_alignment, num_surfaces;
 586
 587     if (device_ctx->type == AV_HWDEVICE_TYPE_DXVA2) {
 588         frames_ctx->format = AV_PIX_FMT_DXVA2_VLD;
 589     } else if (device_ctx->type == AV_HWDEVICE_TYPE_D3D11VA) {
 590         frames_ctx->format = AV_PIX_FMT_D3D11;
 591     } else {
 592         return AVERROR(EINVAL);
 593     }
 594
 595     /* decoding MPEG-2 requires additional alignment on some Intel GPUs,
 596     but it causes issues for H.264 on certain AMD GPUs..... */
 597     if (avctx->codec_id == AV_CODEC_ID_MPEG2VIDEO)
 598         surface_alignment = 32;
 599     /* the HEVC DXVA2 spec asks for 128 pixel aligned surfaces to ensure
 600     all coding features have enough room to work with */
 601     else if (avctx->codec_id == AV_CODEC_ID_HEVC)
 602         surface_alignment = 128;
 603     else
 604         surface_alignment = 16;
 605
 606     /* 1 base work surface */
 607     num_surfaces = 1;
 608
 609     /* add surfaces based on number of possible refs */
 610     if (avctx->codec_id == AV_CODEC_ID_H264 || avctx->codec_id == AV_CODEC_ID_HEVC)
 611         num_surfaces += 16;
 612     else if (avctx->codec_id == AV_CODEC_ID_VP9)
 613         num_surfaces += 8;
 614     else
 615         num_surfaces += 2;
 616
 617     frames_ctx->sw_format = avctx->sw_pix_fmt == AV_PIX_FMT_YUV420P10 ?
 618                             AV_PIX_FMT_P010 : AV_PIX_FMT_NV12;
 619     frames_ctx->width = FFALIGN(avctx->coded_width, surface_alignment);
 620     frames_ctx->height = FFALIGN(avctx->coded_height, surface_alignment);
 621     frames_ctx->initial_pool_size = num_surfaces;
 622
 623
 624 #if CONFIG_DXVA2
 625     if (frames_ctx->format == AV_PIX_FMT_DXVA2_VLD) {
 626         AVDXVA2FramesContext *frames_hwctx = frames_ctx->hwctx;
 627
 628         frames_hwctx->surface_type = DXVA2_VideoDecoderRenderTarget;
 629     }
 630 #endif
 631
 632 #if CONFIG_D3D11VA
 633     if (frames_ctx->format == AV_PIX_FMT_D3D11) {
 634         AVD3D11VAFramesContext *frames_hwctx = frames_ctx->hwctx;
 635
 636         frames_hwctx->BindFlags |= D3D11_BIND_DECODER;
 637     }
 638 #endif
 639
 640     return 0;
 641 }
 642
 643 int ff_dxva2_decode_init(AVCodecContext *avctx)
 644 {
 645     FFDXVASharedContext *sctx = DXVA_SHARED_CONTEXT(avctx);
 646     AVHWFramesContext *frames_ctx;
 647     enum AVHWDeviceType dev_type = avctx->hwaccel->pix_fmt == AV_PIX_FMT_DXVA2_VLD
 648                             ? AV_HWDEVICE_TYPE_DXVA2 : AV_HWDEVICE_TYPE_D3D11VA;
 649     int ret = 0;
 650
 651     // Old API.
 652     if (avctx->hwaccel_context)
 653         return 0;
 654
 655     // (avctx->pix_fmt is not updated yet at this point)
 656     sctx->pix_fmt = avctx->hwaccel->pix_fmt;
 657
 658     ret = ff_decode_get_hw_frames_ctx(avctx, dev_type);
 659     if (ret < 0)
 660         return ret;
 661
 662     frames_ctx = (AVHWFramesContext*)avctx->hw_frames_ctx->data;
 663     sctx->device_ctx = frames_ctx->device_ctx;
 664
 665     if (frames_ctx->format != sctx->pix_fmt) {
 666         av_log(avctx, AV_LOG_ERROR, "Invalid pixfmt for hwaccel!\n");
 667         ret = AVERROR(EINVAL);
 668         goto fail;
 669     }
 670
 671 #if CONFIG_D3D11VA
 672     if (sctx->pix_fmt == AV_PIX_FMT_D3D11) {
 673         AVD3D11VADeviceContext *device_hwctx = frames_ctx->device_ctx->hwctx;
 674         AVD3D11VAContext *d3d11_ctx = &sctx->ctx.d3d11va;
 675
 676         ff_dxva2_lock(avctx);
 677         ret = d3d11va_create_decoder(avctx);
 678         ff_dxva2_unlock(avctx);
 679         if (ret < 0)
 680             goto fail;
 681
 682         d3d11_ctx->decoder       = sctx->d3d11_decoder;
 683         d3d11_ctx->video_context = device_hwctx->video_context;
 684         d3d11_ctx->cfg           = &sctx->d3d11_config;
 685         d3d11_ctx->surface_count = sctx->nb_d3d11_views;
 686         d3d11_ctx->surface       = sctx->d3d11_views;
 687         d3d11_ctx->workaround    = sctx->workaround;
 688         d3d11_ctx->context_mutex = INVALID_HANDLE_VALUE;
 689     }
 690 #endif
 691
 692 #if CONFIG_DXVA2
 693     if (sctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) {
 694         AVDXVA2FramesContext *frames_hwctx = frames_ctx->hwctx;
 695         struct dxva_context *dxva_ctx = &sctx->ctx.dxva2;
 696
 697         ff_dxva2_lock(avctx);
 698         ret = dxva2_create_decoder(avctx);
 699         ff_dxva2_unlock(avctx);
 700         if (ret < 0)
 701             goto fail;
 702
 703         dxva_ctx->decoder       = sctx->dxva2_decoder;
 704         dxva_ctx->cfg           = &sctx->dxva2_config;
 705         dxva_ctx->surface       = frames_hwctx->surfaces;
 706         dxva_ctx->surface_count = frames_hwctx->nb_surfaces;
 707         dxva_ctx->workaround    = sctx->workaround;
 708     }
 709 #endif
 710
 711     return 0;
 712
 713 fail:
 714     ff_dxva2_decode_uninit(avctx);
 715     return ret;
 716 }
 717
 718 int ff_dxva2_decode_uninit(AVCodecContext *avctx)
 719 {
 720     FFDXVASharedContext *sctx = DXVA_SHARED_CONTEXT(avctx);
 721     int i;
 722
 723     av_buffer_unref(&sctx->decoder_ref);
 724
 725 #if CONFIG_D3D11VA
 726     for (i = 0; i < sctx->nb_d3d11_views; i++) {
 727         if (sctx->d3d11_views[i])
 728             ID3D11VideoDecoderOutputView_Release(sctx->d3d11_views[i]);
 729     }
 730     av_freep(&sctx->d3d11_views);
 731 #endif
 732
 733 #if CONFIG_DXVA2
 734     if (sctx->dxva2_service)
 735         IDirectXVideoDecoderService_Release(sctx->dxva2_service);
 736 #endif
 737
 738     return 0;
 739 }
 740
 741 static void *get_surface(const AVCodecContext *avctx, const AVFrame *frame)
 742 {
 743 #if CONFIG_D3D11VA
 744     if (frame->format == AV_PIX_FMT_D3D11) {
 745         FFDXVASharedContext *sctx = DXVA_SHARED_CONTEXT(avctx);
 746         intptr_t index = (intptr_t)frame->data[1];
 747         if (index < 0 || index >= sctx->nb_d3d11_views ||
 748             sctx->d3d11_texture != (ID3D11Texture2D *)frame->data[0]) {
 749             av_log((void *)avctx, AV_LOG_ERROR, "get_buffer frame is invalid!\n");
 750             return NULL;
 751         }
 752         return sctx->d3d11_views[index];
 753     }
 754 #endif
 755     return frame->data[3];
 756 }
 757
 758 unsigned ff_dxva2_get_surface_index(const AVCodecContext *avctx,
 759                                     const AVDXVAContext *ctx,
 760                                     const AVFrame *frame)
 761 {
 762     void *surface = get_surface(avctx, frame);
 763     unsigned i;
 764
 765 #if CONFIG_D3D11VA
 766     if (avctx->pix_fmt == AV_PIX_FMT_D3D11)
 767         return (intptr_t)frame->data[1];
 768     if (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD) {
 769         D3D11_VIDEO_DECODER_OUTPUT_VIEW_DESC viewDesc;
 770         ID3D11VideoDecoderOutputView_GetDesc((ID3D11VideoDecoderOutputView*) surface, &viewDesc);
 771         return viewDesc.Texture2D.ArraySlice;
 772     }
 773 #endif
 774 #if CONFIG_DXVA2
 775     for (i = 0; i < DXVA_CONTEXT_COUNT(avctx, ctx); i++) {
 776         if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD && ctx->dxva2.surface[i] == surface)
 777             return i;
 778     }
 779 #endif
 780
 781     assert(0);
 782     return 0;
 783 }
 784
 785 int ff_dxva2_commit_buffer(AVCodecContext *avctx,
 786                            AVDXVAContext *ctx,
 787                            DECODER_BUFFER_DESC *dsc,
 788                            unsigned type, const void *data, unsigned size,
 789                            unsigned mb_count)
 790 {
 791     void     *dxva_data;
 792     unsigned dxva_size;
 793     int      result;
 794     HRESULT hr = 0;
 795
 796 #if CONFIG_D3D11VA
 797     if (ff_dxva2_is_d3d11(avctx))
 798         hr = ID3D11VideoContext_GetDecoderBuffer(D3D11VA_CONTEXT(ctx)->video_context,
 799                                                  D3D11VA_CONTEXT(ctx)->decoder,
 800                                                  type,
 801                                                  &dxva_size, &dxva_data);
 802 #endif
 803 #if CONFIG_DXVA2
 804     if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD)
 805         hr = IDirectXVideoDecoder_GetBuffer(DXVA2_CONTEXT(ctx)->decoder, type,
 806                                             &dxva_data, &dxva_size);
 807 #endif
 808     if (FAILED(hr)) {
 809         av_log(avctx, AV_LOG_ERROR, "Failed to get a buffer for %u: 0x%x\n",
 810                type, (unsigned)hr);
 811         return -1;
 812     }
 813     if (size <= dxva_size) {
 814         memcpy(dxva_data, data, size);
 815
 816 #if CONFIG_D3D11VA
 817         if (ff_dxva2_is_d3d11(avctx)) {
 818             D3D11_VIDEO_DECODER_BUFFER_DESC *dsc11 = dsc;
 819             memset(dsc11, 0, sizeof(*dsc11));
 820             dsc11->BufferType           = type;
 821             dsc11->DataSize             = size;
 822             dsc11->NumMBsInBuffer       = mb_count;
 823         }
 824 #endif
 825 #if CONFIG_DXVA2
 826         if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) {
 827             DXVA2_DecodeBufferDesc *dsc2 = dsc;
 828             memset(dsc2, 0, sizeof(*dsc2));
 829             dsc2->CompressedBufferType = type;
 830             dsc2->DataSize             = size;
 831             dsc2->NumMBsInBuffer       = mb_count;
 832         }
 833 #endif
 834
 835         result = 0;
 836     } else {
 837         av_log(avctx, AV_LOG_ERROR, "Buffer for type %u was too small\n", type);
 838         result = -1;
 839     }
 840
 841 #if CONFIG_D3D11VA
 842     if (ff_dxva2_is_d3d11(avctx))
 843         hr = ID3D11VideoContext_ReleaseDecoderBuffer(D3D11VA_CONTEXT(ctx)->video_context, D3D11VA_CONTEXT(ctx)->decoder, type);
 844 #endif
 845 #if CONFIG_DXVA2
 846     if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD)
 847         hr = IDirectXVideoDecoder_ReleaseBuffer(DXVA2_CONTEXT(ctx)->decoder, type);
 848 #endif
 849     if (FAILED(hr)) {
 850         av_log(avctx, AV_LOG_ERROR,
 851                "Failed to release buffer type %u: 0x%x\n",
 852                type, (unsigned)hr);
 853         result = -1;
 854     }
 855     return result;
 856 }
 857
 858 static int frame_add_buf(AVFrame *frame, AVBufferRef *ref)
 859 {
 860     int i;
 861
 862     for (i = 0; i < AV_NUM_DATA_POINTERS; i++) {
 863         if (!frame->buf[i]) {
 864             frame->buf[i] = av_buffer_ref(ref);
 865             return frame->buf[i] ? 0 : AVERROR(ENOMEM);
 866         }
 867     }
 868
 869     // For now we expect that the caller does not use more than
 870     // AV_NUM_DATA_POINTERS-1 buffers if the user uses a custom pool.
 871     return AVERROR(EINVAL);
 872 }
 873
 874 int ff_dxva2_common_end_frame(AVCodecContext *avctx, AVFrame *frame,
 875                               const void *pp, unsigned pp_size,
 876                               const void *qm, unsigned qm_size,
 877                               int (*commit_bs_si)(AVCodecContext *,
 878                                                   DECODER_BUFFER_DESC *bs,
 879                                                   DECODER_BUFFER_DESC *slice))
 880 {
 881     AVDXVAContext *ctx = DXVA_CONTEXT(avctx);
 882     unsigned               buffer_count = 0;
 883 #if CONFIG_D3D11VA
 884     D3D11_VIDEO_DECODER_BUFFER_DESC buffer11[4];
 885 #endif
 886 #if CONFIG_DXVA2
 887     DXVA2_DecodeBufferDesc          buffer2[4];
 888 #endif
 889     DECODER_BUFFER_DESC             *buffer = NULL, *buffer_slice = NULL;
 890     int result, runs = 0;
 891     HRESULT hr;
 892     unsigned type;
 893     FFDXVASharedContext *sctx = DXVA_SHARED_CONTEXT(avctx);
 894
 895     if (sctx->decoder_ref) {
 896         result = frame_add_buf(frame, sctx->decoder_ref);
 897         if (result < 0)
 898             return result;
 899     }
 900
 901     do {
 902         ff_dxva2_lock(avctx);
 903 #if CONFIG_D3D11VA
 904         if (ff_dxva2_is_d3d11(avctx))
 905             hr = ID3D11VideoContext_DecoderBeginFrame(D3D11VA_CONTEXT(ctx)->video_context, D3D11VA_CONTEXT(ctx)->decoder,
 906                                                       get_surface(avctx, frame),
 907                                                       0, NULL);
 908 #endif
 909 #if CONFIG_DXVA2
 910         if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD)
 911             hr = IDirectXVideoDecoder_BeginFrame(DXVA2_CONTEXT(ctx)->decoder,
 912                                                  get_surface(avctx, frame),
 913                                                  NULL);
 914 #endif
 915         if (hr != E_PENDING || ++runs > 50)
 916             break;
 917         ff_dxva2_unlock(avctx);
 918         av_usleep(2000);
 919     } while(1);
 920
 921     if (FAILED(hr)) {
 922         av_log(avctx, AV_LOG_ERROR, "Failed to begin frame: 0x%x\n", (unsigned)hr);
 923         ff_dxva2_unlock(avctx);
 924         return -1;
 925     }
 926
 927 #if CONFIG_D3D11VA
 928     if (ff_dxva2_is_d3d11(avctx)) {
 929         buffer = &buffer11[buffer_count];
 930         type = D3D11_VIDEO_DECODER_BUFFER_PICTURE_PARAMETERS;
 931     }
 932 #endif
 933 #if CONFIG_DXVA2
 934     if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) {
 935         buffer = &buffer2[buffer_count];
 936         type = DXVA2_PictureParametersBufferType;
 937     }
 938 #endif
 939     result = ff_dxva2_commit_buffer(avctx, ctx, buffer,
 940                                     type,
 941                                     pp, pp_size, 0);
 942     if (result) {
 943         av_log(avctx, AV_LOG_ERROR,
 944                "Failed to add picture parameter buffer\n");
 945         goto end;
 946     }
 947     buffer_count++;
 948
 949     if (qm_size > 0) {
 950 #if CONFIG_D3D11VA
 951         if (ff_dxva2_is_d3d11(avctx)) {
 952             buffer = &buffer11[buffer_count];
 953             type = D3D11_VIDEO_DECODER_BUFFER_INVERSE_QUANTIZATION_MATRIX;
 954         }
 955 #endif
 956 #if CONFIG_DXVA2
 957         if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) {
 958             buffer = &buffer2[buffer_count];
 959             type = DXVA2_InverseQuantizationMatrixBufferType;
 960         }
 961 #endif
 962         result = ff_dxva2_commit_buffer(avctx, ctx, buffer,
 963                                         type,
 964                                         qm, qm_size, 0);
 965         if (result) {
 966             av_log(avctx, AV_LOG_ERROR,
 967                    "Failed to add inverse quantization matrix buffer\n");
 968             goto end;
 969         }
 970         buffer_count++;
 971     }
 972
 973 #if CONFIG_D3D11VA
 974     if (ff_dxva2_is_d3d11(avctx)) {
 975         buffer       = &buffer11[buffer_count + 0];
 976         buffer_slice = &buffer11[buffer_count + 1];
 977     }
 978 #endif
 979 #if CONFIG_DXVA2
 980     if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) {
 981         buffer       = &buffer2[buffer_count + 0];
 982         buffer_slice = &buffer2[buffer_count + 1];
 983     }
 984 #endif
 985
 986     result = commit_bs_si(avctx,
 987                           buffer,
 988                           buffer_slice);
 989     if (result) {
 990         av_log(avctx, AV_LOG_ERROR,
 991                "Failed to add bitstream or slice control buffer\n");
 992         goto end;
 993     }
 994     buffer_count += 2;
 995
 996     /* TODO Film Grain when possible */
 997
 998     assert(buffer_count == 1 + (qm_size > 0) + 2);
 999
1000 #if CONFIG_D3D11VA
1001     if (ff_dxva2_is_d3d11(avctx))
1002         hr = ID3D11VideoContext_SubmitDecoderBuffers(D3D11VA_CONTEXT(ctx)->video_context,
1003                                                      D3D11VA_CONTEXT(ctx)->decoder,
1004                                                      buffer_count, buffer11);
1005 #endif
1006 #if CONFIG_DXVA2
1007     if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) {
1008         DXVA2_DecodeExecuteParams exec = {
1009             .NumCompBuffers     = buffer_count,
1010             .pCompressedBuffers = buffer2,
1011             .pExtensionData     = NULL,
1012         };
1013         hr = IDirectXVideoDecoder_Execute(DXVA2_CONTEXT(ctx)->decoder, &exec);
1014     }
1015 #endif
1016     if (FAILED(hr)) {
1017         av_log(avctx, AV_LOG_ERROR, "Failed to execute: 0x%x\n", (unsigned)hr);
1018         result = -1;
1019     }
1020
1021 end:
1022 #if CONFIG_D3D11VA
1023     if (ff_dxva2_is_d3d11(avctx))
1024         hr = ID3D11VideoContext_DecoderEndFrame(D3D11VA_CONTEXT(ctx)->video_context, D3D11VA_CONTEXT(ctx)->decoder);
1025 #endif
1026 #if CONFIG_DXVA2
1027     if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD)
1028         hr = IDirectXVideoDecoder_EndFrame(DXVA2_CONTEXT(ctx)->decoder, NULL);
1029 #endif
1030     ff_dxva2_unlock(avctx);
1031     if (FAILED(hr)) {
1032         av_log(avctx, AV_LOG_ERROR, "Failed to end frame: 0x%x\n", (unsigned)hr);
1033         result = -1;
1034     }
1035
1036     return result;
1037 }
1038
1039 int ff_dxva2_is_d3d11(const AVCodecContext *avctx)
1040 {
1041     if (CONFIG_D3D11VA)
1042         return avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD ||
1043                avctx->pix_fmt == AV_PIX_FMT_D3D11;
1044     else
1045         return 0;
1046 }