modules/codec/avcodec/directx_va.c

   1 /*****************************************************************************
   2  * directx_va.c: DirectX Generic Video Acceleration helpers
   3  *****************************************************************************
   4  * Copyright (C) 2009 Geoffroy Couprie
   5  * Copyright (C) 2009 Laurent Aimar
   6  * Copyright (C) 2015 Steve Lhomme
   7  * $Id$
   8  *
   9  * Authors: Geoffroy Couprie <geal@videolan.org>
  10  *          Laurent Aimar <fenrir _AT_ videolan _DOT_ org>
  11  *          Steve Lhomme <robux4@gmail.com>
  12  *
  13  * This program is free software; you can redistribute it and/or modify it
  14  * under the terms of the GNU Lesser General Public License as published by
  15  * the Free Software Foundation; either version 2.1 of the License, or
  16  * (at your option) any later version.
  17  *
  18  * This program is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  21  * GNU Lesser General Public License for more details.
  22  *
  23  * You should have received a copy of the GNU Lesser General Public License
  24  * along with this program; if not, write to the Free Software Foundation,
  25  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  26  *****************************************************************************/
  27 #ifdef HAVE_CONFIG_H
  28 # include "config.h"
  29 #endif
  30
  31 #include <assert.h>
  32
  33 #include <vlc_common.h>
  34 #include <vlc_codecs.h>
  35 #include <vlc_codec.h>
  36
  37 #define COBJMACROS
  38
  39 #define D3D_DecoderType     IUnknown
  40 #define D3D_DecoderDevice   IUnknown
  41 #define D3D_DecoderSurface  IUnknown
  42 typedef struct
  43 {
  44     void *dummy;
  45 } picture_sys_t;
  46 #include "directx_va.h"
  47
  48 #include "avcodec.h"
  49 #include "../../packetizer/h264_nal.h"
  50 #include "../../packetizer/hevc_nal.h"
  51
  52 static const int PROF_MPEG2_SIMPLE[] = { FF_PROFILE_MPEG2_SIMPLE, 0 };
  53 static const int PROF_MPEG2_MAIN[]   = { FF_PROFILE_MPEG2_SIMPLE,
  54                                          FF_PROFILE_MPEG2_MAIN, 0 };
  55 static const int PROF_H264_HIGH[]    = { FF_PROFILE_H264_BASELINE,
  56                                          FF_PROFILE_H264_CONSTRAINED_BASELINE,
  57                                          FF_PROFILE_H264_MAIN,
  58                                          FF_PROFILE_H264_HIGH, 0 };
  59 static const int PROF_HEVC_MAIN[]    = { FF_PROFILE_HEVC_MAIN, 0 };
  60 static const int PROF_HEVC_MAIN10[]  = { FF_PROFILE_HEVC_MAIN,
  61                                          FF_PROFILE_HEVC_MAIN_10, 0 };
  62
  63 static const int PROF_VP9_MAIN[]    = { FF_PROFILE_VP9_0, 0 };
  64 static const int PROF_VP9_10[]      = { FF_PROFILE_VP9_2, 0 };
  65
  66 #include <winapifamily.h>
  67 #if defined(WINAPI_FAMILY)
  68 # undef WINAPI_FAMILY
  69 #endif
  70 #define WINAPI_FAMILY WINAPI_PARTITION_DESKTOP
  71
  72 #include <d3d9.h>
  73 #include <dxva2api.h>
  74
  75 #include <initguid.h> /* must be last included to not redefine existing GUIDs */
  76
  77 /* dxva2api.h GUIDs: http://msdn.microsoft.com/en-us/library/windows/desktop/ms697067(v=vs100).aspx
  78  * assume that they are declared in dxva2api.h */
  79 #define MS_GUID(name, l, w1, w2, b1, b2, b3, b4, b5, b6, b7, b8)
  80
  81 #ifdef __MINGW32__
  82 # include <_mingw.h>
  83
  84 # if !defined(__MINGW64_VERSION_MAJOR)
  85 #  undef MS_GUID
  86 #  define MS_GUID DEFINE_GUID /* dxva2api.h fails to declare those, redefine as static */
  87 #  define DXVA2_E_NEW_VIDEO_DEVICE MAKE_HRESULT(1, 4, 4097)
  88 # else
  89 #  include <dxva.h>
  90 # endif
  91
  92 #endif /* __MINGW32__ */
  93
  94 /* Codec capabilities GUID, sorted by codec */
  95 MS_GUID    (DXVA2_ModeMPEG2_MoComp,                 0xe6a9f44b, 0x61b0, 0x4563, 0x9e, 0xa4, 0x63, 0xd2, 0xa3, 0xc6, 0xfe, 0x66);
  96 MS_GUID    (DXVA2_ModeMPEG2_IDCT,                   0xbf22ad00, 0x03ea, 0x4690, 0x80, 0x77, 0x47, 0x33, 0x46, 0x20, 0x9b, 0x7e);
  97 MS_GUID    (DXVA2_ModeMPEG2_VLD,                    0xee27417f, 0x5e28, 0x4e65, 0xbe, 0xea, 0x1d, 0x26, 0xb5, 0x08, 0xad, 0xc9);
  98 DEFINE_GUID(DXVA_ModeMPEG1_A,                       0x1b81be09, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
  99 DEFINE_GUID(DXVA_ModeMPEG2_A,                       0x1b81be0A, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 100 DEFINE_GUID(DXVA_ModeMPEG2_B,                       0x1b81be0B, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 101 DEFINE_GUID(DXVA_ModeMPEG2_C,                       0x1b81be0C, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 102 DEFINE_GUID(DXVA_ModeMPEG2_D,                       0x1b81be0D, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 103 DEFINE_GUID(DXVA2_ModeMPEG2and1_VLD,                0x86695f12, 0x340e, 0x4f04, 0x9f, 0xd3, 0x92, 0x53, 0xdd, 0x32, 0x74, 0x60);
 104 DEFINE_GUID(DXVA2_ModeMPEG1_VLD,                    0x6f3ec719, 0x3735, 0x42cc, 0x80, 0x63, 0x65, 0xcc, 0x3c, 0xb3, 0x66, 0x16);
 105
 106 MS_GUID    (DXVA2_ModeH264_A,                       0x1b81be64, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 107 MS_GUID    (DXVA2_ModeH264_B,                       0x1b81be65, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 108 MS_GUID    (DXVA2_ModeH264_C,                       0x1b81be66, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 109 MS_GUID    (DXVA2_ModeH264_D,                       0x1b81be67, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 110 MS_GUID    (DXVA2_ModeH264_E,                       0x1b81be68, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 111 MS_GUID    (DXVA2_ModeH264_F,                       0x1b81be69, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 112 DEFINE_GUID(DXVA_ModeH264_VLD_Multiview,            0x9901CCD3, 0xca12, 0x4b7e, 0x86, 0x7a, 0xe2, 0x22, 0x3d, 0x92, 0x55, 0xc3); // MVC
 113 DEFINE_GUID(DXVA_ModeH264_VLD_WithFMOASO_NoFGT,     0xd5f04ff9, 0x3418, 0x45d8, 0x95, 0x61, 0x32, 0xa7, 0x6a, 0xae, 0x2d, 0xdd);
 114 DEFINE_GUID(DXVADDI_Intel_ModeH264_A,               0x604F8E64, 0x4951, 0x4c54, 0x88, 0xFE, 0xAB, 0xD2, 0x5C, 0x15, 0xB3, 0xD6);
 115 DEFINE_GUID(DXVADDI_Intel_ModeH264_C,               0x604F8E66, 0x4951, 0x4c54, 0x88, 0xFE, 0xAB, 0xD2, 0x5C, 0x15, 0xB3, 0xD6);
 116 DEFINE_GUID(DXVA_Intel_H264_NoFGT_ClearVideo,       0x604F8E68, 0x4951, 0x4c54, 0x88, 0xFE, 0xAB, 0xD2, 0x5C, 0x15, 0xB3, 0xD6);
 117 DEFINE_GUID(DXVA_ModeH264_VLD_NoFGT_Flash,          0x4245F676, 0x2BBC, 0x4166, 0xa0, 0xBB, 0x54, 0xE7, 0xB8, 0x49, 0xC3, 0x80);
 118
 119 MS_GUID    (DXVA2_ModeWMV8_A,                       0x1b81be80, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 120 MS_GUID    (DXVA2_ModeWMV8_B,                       0x1b81be81, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 121
 122 MS_GUID    (DXVA2_ModeWMV9_A,                       0x1b81be90, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 123 MS_GUID    (DXVA2_ModeWMV9_B,                       0x1b81be91, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 124 MS_GUID    (DXVA2_ModeWMV9_C,                       0x1b81be94, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 125
 126 MS_GUID    (DXVA2_ModeVC1_A,                        0x1b81beA0, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 127 MS_GUID    (DXVA2_ModeVC1_B,                        0x1b81beA1, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 128 MS_GUID    (DXVA2_ModeVC1_C,                        0x1b81beA2, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 129 MS_GUID    (DXVA2_ModeVC1_D,                        0x1b81beA3, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 130 DEFINE_GUID(DXVA2_ModeVC1_D2010,                    0x1b81beA4, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5); // August 2010 update
 131 DEFINE_GUID(DXVA_Intel_VC1_ClearVideo,              0xBCC5DB6D, 0xA2B6, 0x4AF0, 0xAC, 0xE4, 0xAD, 0xB1, 0xF7, 0x87, 0xBC, 0x89);
 132 DEFINE_GUID(DXVA_Intel_VC1_ClearVideo_2,            0xE07EC519, 0xE651, 0x4CD6, 0xAC, 0x84, 0x13, 0x70, 0xCC, 0xEE, 0xC8, 0x51);
 133
 134 DEFINE_GUID(DXVA_nVidia_MPEG4_ASP,                  0x9947EC6F, 0x689B, 0x11DC, 0xA3, 0x20, 0x00, 0x19, 0xDB, 0xBC, 0x41, 0x84);
 135 DEFINE_GUID(DXVA_ModeMPEG4pt2_VLD_Simple,           0xefd64d74, 0xc9e8, 0x41d7, 0xa5, 0xe9, 0xe9, 0xb0, 0xe3, 0x9f, 0xa3, 0x19);
 136 DEFINE_GUID(DXVA_ModeMPEG4pt2_VLD_AdvSimple_NoGMC,  0xed418a9f, 0x010d, 0x4eda, 0x9a, 0xe3, 0x9a, 0x65, 0x35, 0x8d, 0x8d, 0x2e);
 137 DEFINE_GUID(DXVA_ModeMPEG4pt2_VLD_AdvSimple_GMC,    0xab998b5b, 0x4258, 0x44a9, 0x9f, 0xeb, 0x94, 0xe5, 0x97, 0xa6, 0xba, 0xae);
 138 DEFINE_GUID(DXVA_ModeMPEG4pt2_VLD_AdvSimple_Avivo,  0x7C74ADC6, 0xe2ba, 0x4ade, 0x86, 0xde, 0x30, 0xbe, 0xab, 0xb4, 0x0c, 0xc1);
 139
 140 DEFINE_GUID(DXVA_ModeHEVC_VLD_Main,                 0x5b11d51b, 0x2f4c, 0x4452,0xbc,0xc3,0x09,0xf2,0xa1,0x16,0x0c,0xc0);
 141 DEFINE_GUID(DXVA_ModeHEVC_VLD_Main10,               0x107af0e0, 0xef1a, 0x4d19,0xab,0xa8,0x67,0xa1,0x63,0x07,0x3d,0x13);
 142
 143 DEFINE_GUID(DXVA_ModeH264_VLD_Stereo_Progressive_NoFGT,     0xd79be8da, 0x0cf1, 0x4c81,0xb8,0x2a,0x69,0xa4,0xe2,0x36,0xf4,0x3d);
 144 DEFINE_GUID(DXVA_ModeH264_VLD_Stereo_NoFGT,                 0xf9aaccbb, 0xc2b6, 0x4cfc,0x87,0x79,0x57,0x07,0xb1,0x76,0x05,0x52);
 145 DEFINE_GUID(DXVA_ModeH264_VLD_Multiview_NoFGT,              0x705b9d82, 0x76cf, 0x49d6,0xb7,0xe6,0xac,0x88,0x72,0xdb,0x01,0x3c);
 146
 147 DEFINE_GUID(DXVA_ModeH264_VLD_SVC_Scalable_Baseline,                    0xc30700c4, 0xe384, 0x43e0, 0xb9, 0x82, 0x2d, 0x89, 0xee, 0x7f, 0x77, 0xc4);
 148 DEFINE_GUID(DXVA_ModeH264_VLD_SVC_Restricted_Scalable_Baseline,         0x9b8175d4, 0xd670, 0x4cf2, 0xa9, 0xf0, 0xfa, 0x56, 0xdf, 0x71, 0xa1, 0xae);
 149 DEFINE_GUID(DXVA_ModeH264_VLD_SVC_Scalable_High,                        0x728012c9, 0x66a8, 0x422f, 0x97, 0xe9, 0xb5, 0xe3, 0x9b, 0x51, 0xc0, 0x53);
 150 DEFINE_GUID(DXVA_ModeH264_VLD_SVC_Restricted_Scalable_High_Progressive, 0x8efa5926, 0xbd9e, 0x4b04, 0x8b, 0x72, 0x8f, 0x97, 0x7d, 0xc4, 0x4c, 0x36);
 151
 152 DEFINE_GUID(DXVA_ModeH261_A,                        0x1b81be01, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 153 DEFINE_GUID(DXVA_ModeH261_B,                        0x1b81be02, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 154
 155 DEFINE_GUID(DXVA_ModeH263_A,                        0x1b81be03, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 156 DEFINE_GUID(DXVA_ModeH263_B,                        0x1b81be04, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 157 DEFINE_GUID(DXVA_ModeH263_C,                        0x1b81be05, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 158 DEFINE_GUID(DXVA_ModeH263_D,                        0x1b81be06, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 159 DEFINE_GUID(DXVA_ModeH263_E,                        0x1b81be07, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 160 DEFINE_GUID(DXVA_ModeH263_F,                        0x1b81be08, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
 161
 162 DEFINE_GUID(DXVA_ModeVP8_VLD,                       0x90b899ea, 0x3a62, 0x4705, 0x88, 0xb3, 0x8d, 0xf0, 0x4b, 0x27, 0x44, 0xe7);
 163 DEFINE_GUID(DXVA_ModeVP9_VLD_Profile0,              0x463707f8, 0xa1d0, 0x4585, 0x87, 0x6d, 0x83, 0xaa, 0x6d, 0x60, 0xb8, 0x9e);
 164 DEFINE_GUID(DXVA_ModeVP9_VLD_10bit_Profile2,        0xa4c749ef, 0x6ecf, 0x48aa, 0x84, 0x48, 0x50, 0xa7, 0xa1, 0x16, 0x5f, 0xf7);
 165 DEFINE_GUID(DXVA_ModeVP9_VLD_Intel,                 0x76988a52, 0xdf13, 0x419a, 0x8e, 0x64, 0xff, 0xcf, 0x4a, 0x33, 0x6c, 0xf5);
 166
 167 typedef struct {
 168     const char   *name;
 169     const GUID   *guid;
 170     enum AVCodecID codec;
 171     const int    *p_profiles; // NULL or ends with 0
 172 } directx_va_mode_t;
 173
 174 /* XXX Prefered modes must come first */
 175 static const directx_va_mode_t DXVA_MODES[] = {
 176     /* MPEG-1/2 */
 177     { "MPEG-1 decoder, restricted profile A",                                         &DXVA_ModeMPEG1_A,                      0, NULL },
 178     { "MPEG-2 decoder, restricted profile A",                                         &DXVA_ModeMPEG2_A,                      0, NULL },
 179     { "MPEG-2 decoder, restricted profile B",                                         &DXVA_ModeMPEG2_B,                      0, NULL },
 180     { "MPEG-2 decoder, restricted profile C",                                         &DXVA_ModeMPEG2_C,                      0, NULL },
 181     { "MPEG-2 decoder, restricted profile D",                                         &DXVA_ModeMPEG2_D,                      0, NULL },
 182
 183     { "MPEG-2 variable-length decoder",                                               &DXVA2_ModeMPEG2_VLD,                   AV_CODEC_ID_MPEG2VIDEO, PROF_MPEG2_SIMPLE },
 184     { "MPEG-2 & MPEG-1 variable-length decoder",                                      &DXVA2_ModeMPEG2and1_VLD,               AV_CODEC_ID_MPEG2VIDEO, PROF_MPEG2_MAIN },
 185     { "MPEG-2 & MPEG-1 variable-length decoder",                                      &DXVA2_ModeMPEG2and1_VLD,               AV_CODEC_ID_MPEG1VIDEO, NULL },
 186     { "MPEG-2 motion compensation",                                                   &DXVA2_ModeMPEG2_MoComp,                0, NULL },
 187     { "MPEG-2 inverse discrete cosine transform",                                     &DXVA2_ModeMPEG2_IDCT,                  0, NULL },
 188
 189     /* MPEG-1 http://download.microsoft.com/download/B/1/7/B172A3C8-56F2-4210-80F1-A97BEA9182ED/DXVA_MPEG1_VLD.pdf */
 190     { "MPEG-1 variable-length decoder, no D pictures",                                &DXVA2_ModeMPEG1_VLD,                   0, NULL },
 191
 192     /* H.264 http://www.microsoft.com/downloads/details.aspx?displaylang=en&FamilyID=3d1c290b-310b-4ea2-bf76-714063a6d7a6 */
 193     { "H.264 variable-length decoder, film grain technology",                         &DXVA2_ModeH264_F,                      AV_CODEC_ID_H264, PROF_H264_HIGH },
 194     { "H.264 variable-length decoder, no film grain technology",                      &DXVA2_ModeH264_E,                      AV_CODEC_ID_H264, PROF_H264_HIGH },
 195     { "H.264 variable-length decoder, no film grain technology (Intel ClearVideo)",   &DXVA_Intel_H264_NoFGT_ClearVideo,      AV_CODEC_ID_H264, PROF_H264_HIGH },
 196     { "H.264 variable-length decoder, no film grain technology, FMO/ASO",             &DXVA_ModeH264_VLD_WithFMOASO_NoFGT,    AV_CODEC_ID_H264, PROF_H264_HIGH },
 197     { "H.264 variable-length decoder, no film grain technology, Flash",               &DXVA_ModeH264_VLD_NoFGT_Flash,         AV_CODEC_ID_H264, PROF_H264_HIGH },
 198
 199     { "H.264 inverse discrete cosine transform, film grain technology",               &DXVA2_ModeH264_D,                      0, NULL },
 200     { "H.264 inverse discrete cosine transform, no film grain technology",            &DXVA2_ModeH264_C,                      0, NULL },
 201     { "H.264 inverse discrete cosine transform, no film grain technology (Intel)",    &DXVADDI_Intel_ModeH264_C,              0, NULL },
 202
 203     { "H.264 motion compensation, film grain technology",                             &DXVA2_ModeH264_B,                      0, NULL },
 204     { "H.264 motion compensation, no film grain technology",                          &DXVA2_ModeH264_A,                      0, NULL },
 205     { "H.264 motion compensation, no film grain technology (Intel)",                  &DXVADDI_Intel_ModeH264_A,              0, NULL },
 206
 207     /* http://download.microsoft.com/download/2/D/0/2D02E72E-7890-430F-BA91-4A363F72F8C8/DXVA_H264_MVC.pdf */
 208     { "H.264 stereo high profile, mbs flag set",                                      &DXVA_ModeH264_VLD_Stereo_Progressive_NoFGT, 0, NULL },
 209     { "H.264 stereo high profile",                                                    &DXVA_ModeH264_VLD_Stereo_NoFGT,             0, NULL },
 210     { "H.264 multiview high profile",                                                 &DXVA_ModeH264_VLD_Multiview_NoFGT,          0, NULL },
 211
 212     /* SVC http://download.microsoft.com/download/C/8/A/C8AD9F1B-57D1-4C10-85A0-09E3EAC50322/DXVA_SVC_2012_06.pdf */
 213     { "H.264 scalable video coding, Scalable Baseline Profile",                       &DXVA_ModeH264_VLD_SVC_Scalable_Baseline,            0, NULL },
 214     { "H.264 scalable video coding, Scalable Constrained Baseline Profile",           &DXVA_ModeH264_VLD_SVC_Restricted_Scalable_Baseline, 0, NULL },
 215     { "H.264 scalable video coding, Scalable High Profile",                           &DXVA_ModeH264_VLD_SVC_Scalable_High,                0, NULL },
 216     { "H.264 scalable video coding, Scalable Constrained High Profile",               &DXVA_ModeH264_VLD_SVC_Restricted_Scalable_High_Progressive, 0, NULL },
 217
 218     /* WMV */
 219     { "Windows Media Video 8 motion compensation",                                    &DXVA2_ModeWMV8_B,                      0, NULL },
 220     { "Windows Media Video 8 post processing",                                        &DXVA2_ModeWMV8_A,                      0, NULL },
 221
 222     { "Windows Media Video 9 IDCT",                                                   &DXVA2_ModeWMV9_C,                      0, NULL },
 223     { "Windows Media Video 9 motion compensation",                                    &DXVA2_ModeWMV9_B,                      0, NULL },
 224     { "Windows Media Video 9 post processing",                                        &DXVA2_ModeWMV9_A,                      0, NULL },
 225
 226     /* VC-1 */
 227     { "VC-1 variable-length decoder",                                                 &DXVA2_ModeVC1_D,                       AV_CODEC_ID_VC1, NULL },
 228     { "VC-1 variable-length decoder",                                                 &DXVA2_ModeVC1_D,                       AV_CODEC_ID_WMV3, NULL },
 229     { "VC-1 variable-length decoder",                                                 &DXVA2_ModeVC1_D2010,                   AV_CODEC_ID_VC1, NULL },
 230     { "VC-1 variable-length decoder",                                                 &DXVA2_ModeVC1_D2010,                   AV_CODEC_ID_WMV3, NULL },
 231     { "VC-1 variable-length decoder 2 (Intel)",                                       &DXVA_Intel_VC1_ClearVideo_2,           0, NULL },
 232     { "VC-1 variable-length decoder (Intel)",                                         &DXVA_Intel_VC1_ClearVideo,             0, NULL },
 233
 234     { "VC-1 inverse discrete cosine transform",                                       &DXVA2_ModeVC1_C,                       0, NULL },
 235     { "VC-1 motion compensation",                                                     &DXVA2_ModeVC1_B,                       0, NULL },
 236     { "VC-1 post processing",                                                         &DXVA2_ModeVC1_A,                       0, NULL },
 237
 238     /* Xvid/Divx: TODO */
 239     { "MPEG-4 Part 2 nVidia bitstream decoder",                                       &DXVA_nVidia_MPEG4_ASP,                 0, NULL },
 240     { "MPEG-4 Part 2 variable-length decoder, Simple Profile",                        &DXVA_ModeMPEG4pt2_VLD_Simple,          0, NULL },
 241     { "MPEG-4 Part 2 variable-length decoder, Simple&Advanced Profile, no GMC",       &DXVA_ModeMPEG4pt2_VLD_AdvSimple_NoGMC, 0, NULL },
 242     { "MPEG-4 Part 2 variable-length decoder, Simple&Advanced Profile, GMC",          &DXVA_ModeMPEG4pt2_VLD_AdvSimple_GMC,   0, NULL },
 243     { "MPEG-4 Part 2 variable-length decoder, Simple&Advanced Profile, Avivo",        &DXVA_ModeMPEG4pt2_VLD_AdvSimple_Avivo, 0, NULL },
 244
 245     /* HEVC */
 246     { "HEVC Main profile",                                                            &DXVA_ModeHEVC_VLD_Main,                AV_CODEC_ID_HEVC, PROF_HEVC_MAIN },
 247     { "HEVC Main 10 profile",                                                         &DXVA_ModeHEVC_VLD_Main10,              AV_CODEC_ID_HEVC, PROF_HEVC_MAIN10 },
 248
 249     /* H.261 */
 250     { "H.261 decoder, restricted profile A",                                          &DXVA_ModeH261_A,                       0, NULL },
 251     { "H.261 decoder, restricted profile B",                                          &DXVA_ModeH261_B,                       0, NULL },
 252
 253     /* H.263 */
 254     { "H.263 decoder, restricted profile A",                                          &DXVA_ModeH263_A,                       0, NULL },
 255     { "H.263 decoder, restricted profile B",                                          &DXVA_ModeH263_B,                       0, NULL },
 256     { "H.263 decoder, restricted profile C",                                          &DXVA_ModeH263_C,                       0, NULL },
 257     { "H.263 decoder, restricted profile D",                                          &DXVA_ModeH263_D,                       0, NULL },
 258     { "H.263 decoder, restricted profile E",                                          &DXVA_ModeH263_E,                       0, NULL },
 259     { "H.263 decoder, restricted profile F",                                          &DXVA_ModeH263_F,                       0, NULL },
 260
 261     /* VPx */
 262     { "VP8",                                                                          &DXVA_ModeVP8_VLD,                      0, NULL },
 263 #if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT( 57, 17, 100 ) && LIBAVCODEC_VERSION_MICRO >= 100
 264     { "VP9 profile 0",                                                                &DXVA_ModeVP9_VLD_Profile0,             AV_CODEC_ID_VP9, PROF_VP9_MAIN },
 265     { "VP9 profile 2",                                                                &DXVA_ModeVP9_VLD_10bit_Profile2,       AV_CODEC_ID_VP9, PROF_VP9_10 },
 266 #else
 267     { "VP9 profile 0",                                                                &DXVA_ModeVP9_VLD_Profile0,             0, NULL },
 268     { "VP9 profile 2",                                                                &DXVA_ModeVP9_VLD_10bit_Profile2,       0, NULL },
 269 #endif
 270     { "VP9 profile Intel",                                                            &DXVA_ModeVP9_VLD_Intel,                0, NULL },
 271
 272     { NULL, NULL, 0, NULL }
 273 };
 274
 275 static int FindVideoServiceConversion(vlc_va_t *, directx_sys_t *, const es_format_t *, const AVCodecContext *);
 276
 277 char *directx_va_GetDecoderName(const GUID *guid)
 278 {
 279     for (unsigned i = 0; DXVA_MODES[i].name; i++) {
 280         if (IsEqualGUID(DXVA_MODES[i].guid, guid))
 281             return strdup(DXVA_MODES[i].name);
 282     }
 283
 284     char *psz_name;
 285     if (asprintf(&psz_name, "Unknown decoder " GUID_FMT, GUID_PRINT(*guid)) < 0)
 286         return NULL;
 287     return psz_name;
 288 }
 289
 290 /* */
 291 int directx_va_Setup(vlc_va_t *va, directx_sys_t *dx_sys, const AVCodecContext *avctx,
 292                      const es_format_t *fmt, int flag_xbox)
 293 {
 294     /* */
 295     if (FindVideoServiceConversion(va, dx_sys, fmt, avctx)) {
 296         msg_Err(va, "FindVideoServiceConversion failed");
 297         return VLC_EGENERIC;
 298     }
 299
 300     int surface_alignment = 16;
 301     unsigned surface_count = 2;
 302
 303     switch ( avctx->codec_id )
 304     {
 305     case AV_CODEC_ID_MPEG2VIDEO:
 306         /* decoding MPEG-2 requires additional alignment on some Intel GPUs,
 307            but it causes issues for H.264 on certain AMD GPUs..... */
 308         surface_alignment = 32;
 309         surface_count += 2 + 2; /* 2 for deinterlacing which can hold up to 2
 310                                  * pictures from the decoder for smoothing */
 311         break;
 312     case AV_CODEC_ID_HEVC:
 313         /* the HEVC DXVA2 spec asks for 128 pixel aligned surfaces to ensure
 314            all coding features have enough room to work with */
 315            /* On the Xbox 1/S, the decoder cannot do 4K aligned to 128 but is OK with 64 */
 316         if (flag_xbox)
 317             surface_alignment = 16;
 318         else
 319             surface_alignment = 128;
 320         surface_count += 16;
 321         break;
 322     case AV_CODEC_ID_H264:
 323         surface_count += 16;
 324         break;
 325     case AV_CODEC_ID_VP9:
 326         surface_count += 4;
 327         break;
 328     default:
 329         surface_count += 2;
 330     }
 331
 332     if ( avctx->active_thread_type & FF_THREAD_FRAME )
 333         surface_count += avctx->thread_count;
 334
 335     int err = va_pool_SetupDecoder(va, &dx_sys->va_pool, avctx, surface_count, surface_alignment);
 336     if (err != VLC_SUCCESS)
 337         return err;
 338     if (dx_sys->can_extern_pool)
 339         return VLC_SUCCESS;
 340     return va_pool_SetupSurfaces(va, &dx_sys->va_pool, surface_count);
 341 }
 342
 343 void directx_va_Close(vlc_va_t *va, directx_sys_t *dx_sys)
 344 {
 345     va_pool_Close(va, &dx_sys->va_pool);
 346 }
 347
 348 int directx_va_Open(vlc_va_t *va, directx_sys_t *dx_sys)
 349 {
 350     if (va_pool_Open(va, &dx_sys->va_pool) != VLC_SUCCESS)
 351         goto error;
 352
 353     return VLC_SUCCESS;
 354
 355 error:
 356     return VLC_EGENERIC;
 357 }
 358
 359 static bool profile_supported(const directx_va_mode_t *mode, const es_format_t *fmt,
 360                               const AVCodecContext *avctx)
 361 {
 362     bool is_supported = mode->p_profiles == NULL;
 363     if (!is_supported)
 364     {
 365         int profile = fmt->i_profile >= 0 ? fmt->i_profile : avctx->profile;
 366         if (mode->codec == AV_CODEC_ID_H264)
 367         {
 368             uint8_t h264_profile;
 369             if ( h264_get_profile_level(fmt, &h264_profile, NULL, NULL) )
 370                 profile = h264_profile;
 371         }
 372         if (mode->codec == AV_CODEC_ID_HEVC)
 373         {
 374             uint8_t hevc_profile;
 375             if (hevc_get_profile_level(fmt, &hevc_profile, NULL, NULL) )
 376                 profile = hevc_profile;
 377         }
 378
 379         if (profile <= 0)
 380             is_supported = true;
 381         else for (const int *p_profile = &mode->p_profiles[0]; *p_profile; ++p_profile)
 382         {
 383             if (*p_profile == profile)
 384             {
 385                 is_supported = true;
 386                 break;
 387             }
 388         }
 389     }
 390     return is_supported;
 391 }
 392
 393 /**
 394  * Find the best suited decoder mode GUID and render format.
 395  */
 396 static int FindVideoServiceConversion(vlc_va_t *va, directx_sys_t *dx_sys,
 397                                       const es_format_t *fmt, const AVCodecContext *avctx)
 398 {
 399     input_list_t p_list = { 0 };
 400     int err = dx_sys->pf_get_input_list(va, &p_list);
 401     if (err != VLC_SUCCESS)
 402         return err;
 403     if (p_list.count == 0) {
 404         msg_Warn( va, "No input format found for HWAccel" );
 405         return VLC_EGENERIC;
 406     }
 407
 408     err = VLC_EGENERIC;
 409     /* Retreive supported modes from the decoder service */
 410     for (unsigned i = 0; i < p_list.count; i++) {
 411         const GUID *g = &p_list.list[i];
 412         char *psz_decoder_name = directx_va_GetDecoderName(g);
 413         msg_Dbg(va, "- '%s' is supported", psz_decoder_name);
 414         free(psz_decoder_name);
 415     }
 416
 417     /* Try all supported mode by our priority */
 418     const directx_va_mode_t *mode = DXVA_MODES;
 419     for (; mode->name; ++mode) {
 420         if (!mode->codec || mode->codec != avctx->codec_id)
 421             continue;
 422
 423         /* */
 424         bool is_supported = false;
 425         for (const GUID *g = &p_list.list[0]; !is_supported && g < &p_list.list[p_list.count]; g++) {
 426             is_supported = IsEqualGUID(mode->guid, g);
 427         }
 428         if ( is_supported )
 429         {
 430             is_supported = profile_supported( mode, fmt, avctx );
 431             if (!is_supported)
 432             {
 433                 char *psz_name = directx_va_GetDecoderName(mode->guid);
 434                 msg_Warn( va, "Unsupported profile %d for %s ",
 435                           fmt->i_profile, psz_name );
 436                 free( psz_name );
 437             }
 438         }
 439         if (!is_supported)
 440             continue;
 441
 442         /* */
 443         msg_Dbg(va, "Trying to use '%s' as input", mode->name);
 444         if (dx_sys->pf_setup_output(va, mode->guid, &fmt->video)==VLC_SUCCESS)
 445         {
 446             dx_sys->input = *mode->guid;
 447             err = VLC_SUCCESS;
 448             break;
 449         }
 450     }
 451
 452     p_list.pf_release(&p_list);
 453     return err;
 454 }