d114feac79cd47574862e5b07c2ec61ff75fa412
[vlc.git] / 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 #include "directx_va.h"
40
41 #include "avcodec.h"
42 #include "../../packetizer/h264_nal.h"
43 #include "../../packetizer/hevc_nal.h"
44
45 static const int PROF_MPEG2_SIMPLE[] = { FF_PROFILE_MPEG2_SIMPLE, 0 };
46 static const int PROF_MPEG2_MAIN[]   = { FF_PROFILE_MPEG2_SIMPLE,
47                                          FF_PROFILE_MPEG2_MAIN, 0 };
48 static const int PROF_H264_HIGH[]    = { FF_PROFILE_H264_CONSTRAINED_BASELINE,
49                                          FF_PROFILE_H264_MAIN,
50                                          FF_PROFILE_H264_HIGH, 0 };
51 static const int PROF_HEVC_MAIN[]    = { FF_PROFILE_HEVC_MAIN, 0 };
52 static const int PROF_HEVC_MAIN10[]  = { FF_PROFILE_HEVC_MAIN,
53                                          FF_PROFILE_HEVC_MAIN_10, 0 };
54
55 #include <d3d9.h>
56 #include <dxva2api.h>
57
58 #include <initguid.h> /* must be last included to not redefine existing GUIDs */
59
60 /* dxva2api.h GUIDs: http://msdn.microsoft.com/en-us/library/windows/desktop/ms697067(v=vs100).aspx
61  * assume that they are declared in dxva2api.h */
62 #define MS_GUID(name, l, w1, w2, b1, b2, b3, b4, b5, b6, b7, b8)
63
64 #ifdef __MINGW32__
65 # include <_mingw.h>
66
67 # if !defined(__MINGW64_VERSION_MAJOR)
68 #  undef MS_GUID
69 #  define MS_GUID DEFINE_GUID /* dxva2api.h fails to declare those, redefine as static */
70 #  define DXVA2_E_NEW_VIDEO_DEVICE MAKE_HRESULT(1, 4, 4097)
71 # else
72 #  include <dxva.h>
73 # endif
74
75 #endif /* __MINGW32__ */
76
77 /* Codec capabilities GUID, sorted by codec */
78 MS_GUID    (DXVA2_ModeMPEG2_MoComp,                 0xe6a9f44b, 0x61b0, 0x4563, 0x9e, 0xa4, 0x63, 0xd2, 0xa3, 0xc6, 0xfe, 0x66);
79 MS_GUID    (DXVA2_ModeMPEG2_IDCT,                   0xbf22ad00, 0x03ea, 0x4690, 0x80, 0x77, 0x47, 0x33, 0x46, 0x20, 0x9b, 0x7e);
80 MS_GUID    (DXVA2_ModeMPEG2_VLD,                    0xee27417f, 0x5e28, 0x4e65, 0xbe, 0xea, 0x1d, 0x26, 0xb5, 0x08, 0xad, 0xc9);
81 DEFINE_GUID(DXVA_ModeMPEG1_A,                       0x1b81be09, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
82 DEFINE_GUID(DXVA_ModeMPEG2_A,                       0x1b81be0A, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
83 DEFINE_GUID(DXVA_ModeMPEG2_B,                       0x1b81be0B, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
84 DEFINE_GUID(DXVA_ModeMPEG2_C,                       0x1b81be0C, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
85 DEFINE_GUID(DXVA_ModeMPEG2_D,                       0x1b81be0D, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
86 DEFINE_GUID(DXVA2_ModeMPEG2and1_VLD,                0x86695f12, 0x340e, 0x4f04, 0x9f, 0xd3, 0x92, 0x53, 0xdd, 0x32, 0x74, 0x60);
87 DEFINE_GUID(DXVA2_ModeMPEG1_VLD,                    0x6f3ec719, 0x3735, 0x42cc, 0x80, 0x63, 0x65, 0xcc, 0x3c, 0xb3, 0x66, 0x16);
88
89 MS_GUID    (DXVA2_ModeH264_A,                       0x1b81be64, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
90 MS_GUID    (DXVA2_ModeH264_B,                       0x1b81be65, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
91 MS_GUID    (DXVA2_ModeH264_C,                       0x1b81be66, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
92 MS_GUID    (DXVA2_ModeH264_D,                       0x1b81be67, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
93 MS_GUID    (DXVA2_ModeH264_E,                       0x1b81be68, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
94 MS_GUID    (DXVA2_ModeH264_F,                       0x1b81be69, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
95 DEFINE_GUID(DXVA_ModeH264_VLD_Multiview,            0x9901CCD3, 0xca12, 0x4b7e, 0x86, 0x7a, 0xe2, 0x22, 0x3d, 0x92, 0x55, 0xc3); // MVC
96 DEFINE_GUID(DXVA_ModeH264_VLD_WithFMOASO_NoFGT,     0xd5f04ff9, 0x3418, 0x45d8, 0x95, 0x61, 0x32, 0xa7, 0x6a, 0xae, 0x2d, 0xdd);
97 DEFINE_GUID(DXVADDI_Intel_ModeH264_A,               0x604F8E64, 0x4951, 0x4c54, 0x88, 0xFE, 0xAB, 0xD2, 0x5C, 0x15, 0xB3, 0xD6);
98 DEFINE_GUID(DXVADDI_Intel_ModeH264_C,               0x604F8E66, 0x4951, 0x4c54, 0x88, 0xFE, 0xAB, 0xD2, 0x5C, 0x15, 0xB3, 0xD6);
99 DEFINE_GUID(DXVA_Intel_H264_NoFGT_ClearVideo,       0x604F8E68, 0x4951, 0x4c54, 0x88, 0xFE, 0xAB, 0xD2, 0x5C, 0x15, 0xB3, 0xD6);
100 DEFINE_GUID(DXVA_ModeH264_VLD_NoFGT_Flash,          0x4245F676, 0x2BBC, 0x4166, 0xa0, 0xBB, 0x54, 0xE7, 0xB8, 0x49, 0xC3, 0x80);
101
102 MS_GUID    (DXVA2_ModeWMV8_A,                       0x1b81be80, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
103 MS_GUID    (DXVA2_ModeWMV8_B,                       0x1b81be81, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
104
105 MS_GUID    (DXVA2_ModeWMV9_A,                       0x1b81be90, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
106 MS_GUID    (DXVA2_ModeWMV9_B,                       0x1b81be91, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
107 MS_GUID    (DXVA2_ModeWMV9_C,                       0x1b81be94, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
108
109 MS_GUID    (DXVA2_ModeVC1_A,                        0x1b81beA0, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
110 MS_GUID    (DXVA2_ModeVC1_B,                        0x1b81beA1, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
111 MS_GUID    (DXVA2_ModeVC1_C,                        0x1b81beA2, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
112 MS_GUID    (DXVA2_ModeVC1_D,                        0x1b81beA3, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
113 DEFINE_GUID(DXVA2_ModeVC1_D2010,                    0x1b81beA4, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5); // August 2010 update
114 DEFINE_GUID(DXVA_Intel_VC1_ClearVideo,              0xBCC5DB6D, 0xA2B6, 0x4AF0, 0xAC, 0xE4, 0xAD, 0xB1, 0xF7, 0x87, 0xBC, 0x89);
115 DEFINE_GUID(DXVA_Intel_VC1_ClearVideo_2,            0xE07EC519, 0xE651, 0x4CD6, 0xAC, 0x84, 0x13, 0x70, 0xCC, 0xEE, 0xC8, 0x51);
116
117 DEFINE_GUID(DXVA_nVidia_MPEG4_ASP,                  0x9947EC6F, 0x689B, 0x11DC, 0xA3, 0x20, 0x00, 0x19, 0xDB, 0xBC, 0x41, 0x84);
118 DEFINE_GUID(DXVA_ModeMPEG4pt2_VLD_Simple,           0xefd64d74, 0xc9e8, 0x41d7, 0xa5, 0xe9, 0xe9, 0xb0, 0xe3, 0x9f, 0xa3, 0x19);
119 DEFINE_GUID(DXVA_ModeMPEG4pt2_VLD_AdvSimple_NoGMC,  0xed418a9f, 0x010d, 0x4eda, 0x9a, 0xe3, 0x9a, 0x65, 0x35, 0x8d, 0x8d, 0x2e);
120 DEFINE_GUID(DXVA_ModeMPEG4pt2_VLD_AdvSimple_GMC,    0xab998b5b, 0x4258, 0x44a9, 0x9f, 0xeb, 0x94, 0xe5, 0x97, 0xa6, 0xba, 0xae);
121 DEFINE_GUID(DXVA_ModeMPEG4pt2_VLD_AdvSimple_Avivo,  0x7C74ADC6, 0xe2ba, 0x4ade, 0x86, 0xde, 0x30, 0xbe, 0xab, 0xb4, 0x0c, 0xc1);
122
123 DEFINE_GUID(DXVA_ModeHEVC_VLD_Main,                 0x5b11d51b, 0x2f4c, 0x4452,0xbc,0xc3,0x09,0xf2,0xa1,0x16,0x0c,0xc0);
124 DEFINE_GUID(DXVA_ModeHEVC_VLD_Main10,               0x107af0e0, 0xef1a, 0x4d19,0xab,0xa8,0x67,0xa1,0x63,0x07,0x3d,0x13);
125
126 DEFINE_GUID(DXVA_ModeH264_VLD_Stereo_Progressive_NoFGT,     0xd79be8da, 0x0cf1, 0x4c81,0xb8,0x2a,0x69,0xa4,0xe2,0x36,0xf4,0x3d);
127 DEFINE_GUID(DXVA_ModeH264_VLD_Stereo_NoFGT,                 0xf9aaccbb, 0xc2b6, 0x4cfc,0x87,0x79,0x57,0x07,0xb1,0x76,0x05,0x52);
128 DEFINE_GUID(DXVA_ModeH264_VLD_Multiview_NoFGT,              0x705b9d82, 0x76cf, 0x49d6,0xb7,0xe6,0xac,0x88,0x72,0xdb,0x01,0x3c);
129
130 DEFINE_GUID(DXVA_ModeH264_VLD_SVC_Scalable_Baseline,                    0xc30700c4, 0xe384, 0x43e0, 0xb9, 0x82, 0x2d, 0x89, 0xee, 0x7f, 0x77, 0xc4);
131 DEFINE_GUID(DXVA_ModeH264_VLD_SVC_Restricted_Scalable_Baseline,         0x9b8175d4, 0xd670, 0x4cf2, 0xa9, 0xf0, 0xfa, 0x56, 0xdf, 0x71, 0xa1, 0xae);
132 DEFINE_GUID(DXVA_ModeH264_VLD_SVC_Scalable_High,                        0x728012c9, 0x66a8, 0x422f, 0x97, 0xe9, 0xb5, 0xe3, 0x9b, 0x51, 0xc0, 0x53);
133 DEFINE_GUID(DXVA_ModeH264_VLD_SVC_Restricted_Scalable_High_Progressive, 0x8efa5926, 0xbd9e, 0x4b04, 0x8b, 0x72, 0x8f, 0x97, 0x7d, 0xc4, 0x4c, 0x36);
134
135 DEFINE_GUID(DXVA_ModeH261_A,                        0x1b81be01, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
136 DEFINE_GUID(DXVA_ModeH261_B,                        0x1b81be02, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
137
138 DEFINE_GUID(DXVA_ModeH263_A,                        0x1b81be03, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
139 DEFINE_GUID(DXVA_ModeH263_B,                        0x1b81be04, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
140 DEFINE_GUID(DXVA_ModeH263_C,                        0x1b81be05, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
141 DEFINE_GUID(DXVA_ModeH263_D,                        0x1b81be06, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
142 DEFINE_GUID(DXVA_ModeH263_E,                        0x1b81be07, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
143 DEFINE_GUID(DXVA_ModeH263_F,                        0x1b81be08, 0xa0c7, 0x11d3, 0xb9, 0x84, 0x00, 0xc0, 0x4f, 0x2e, 0x73, 0xc5);
144
145 DEFINE_GUID(DXVA_ModeVP8_VLD,                       0x90b899ea, 0x3a62, 0x4705, 0x88, 0xb3, 0x8d, 0xf0, 0x4b, 0x27, 0x44, 0xe7);
146 DEFINE_GUID(DXVA_ModeVP9_VLD_Profile0,              0x463707f8, 0xa1d0, 0x4585, 0x87, 0x6d, 0x83, 0xaa, 0x6d, 0x60, 0xb8, 0x9e);
147
148 typedef struct {
149     const char   *name;
150     const GUID   *guid;
151     int          codec;
152     const int    *p_profiles; // NULL or ends with 0
153 } directx_va_mode_t;
154
155 /* XXX Prefered modes must come first */
156 static const directx_va_mode_t DXVA_MODES[] = {
157     /* MPEG-1/2 */
158     { "MPEG-1 decoder, restricted profile A",                                         &DXVA_ModeMPEG1_A,                      0, NULL },
159     { "MPEG-2 decoder, restricted profile A",                                         &DXVA_ModeMPEG2_A,                      0, NULL },
160     { "MPEG-2 decoder, restricted profile B",                                         &DXVA_ModeMPEG2_B,                      0, NULL },
161     { "MPEG-2 decoder, restricted profile C",                                         &DXVA_ModeMPEG2_C,                      0, NULL },
162     { "MPEG-2 decoder, restricted profile D",                                         &DXVA_ModeMPEG2_D,                      0, NULL },
163
164     { "MPEG-2 variable-length decoder",                                               &DXVA2_ModeMPEG2_VLD,                   AV_CODEC_ID_MPEG2VIDEO, PROF_MPEG2_SIMPLE },
165     { "MPEG-2 & MPEG-1 variable-length decoder",                                      &DXVA2_ModeMPEG2and1_VLD,               AV_CODEC_ID_MPEG2VIDEO, PROF_MPEG2_MAIN },
166     { "MPEG-2 & MPEG-1 variable-length decoder",                                      &DXVA2_ModeMPEG2and1_VLD,               AV_CODEC_ID_MPEG1VIDEO, NULL },
167     { "MPEG-2 motion compensation",                                                   &DXVA2_ModeMPEG2_MoComp,                0, NULL },
168     { "MPEG-2 inverse discrete cosine transform",                                     &DXVA2_ModeMPEG2_IDCT,                  0, NULL },
169
170     /* MPEG-1 http://download.microsoft.com/download/B/1/7/B172A3C8-56F2-4210-80F1-A97BEA9182ED/DXVA_MPEG1_VLD.pdf */
171     { "MPEG-1 variable-length decoder, no D pictures",                                &DXVA2_ModeMPEG1_VLD,                   0, NULL },
172
173     /* H.264 http://www.microsoft.com/downloads/details.aspx?displaylang=en&FamilyID=3d1c290b-310b-4ea2-bf76-714063a6d7a6 */
174     { "H.264 variable-length decoder, film grain technology",                         &DXVA2_ModeH264_F,                      AV_CODEC_ID_H264, PROF_H264_HIGH },
175     { "H.264 variable-length decoder, no film grain technology (Intel ClearVideo)",   &DXVA_Intel_H264_NoFGT_ClearVideo,      AV_CODEC_ID_H264, PROF_H264_HIGH },
176     { "H.264 variable-length decoder, no film grain technology",                      &DXVA2_ModeH264_E,                      AV_CODEC_ID_H264, PROF_H264_HIGH },
177     { "H.264 variable-length decoder, no film grain technology, FMO/ASO",             &DXVA_ModeH264_VLD_WithFMOASO_NoFGT,    AV_CODEC_ID_H264, PROF_H264_HIGH },
178     { "H.264 variable-length decoder, no film grain technology, Flash",               &DXVA_ModeH264_VLD_NoFGT_Flash,         AV_CODEC_ID_H264, PROF_H264_HIGH },
179
180     { "H.264 inverse discrete cosine transform, film grain technology",               &DXVA2_ModeH264_D,                      0, NULL },
181     { "H.264 inverse discrete cosine transform, no film grain technology",            &DXVA2_ModeH264_C,                      0, NULL },
182     { "H.264 inverse discrete cosine transform, no film grain technology (Intel)",    &DXVADDI_Intel_ModeH264_C,              0, NULL },
183
184     { "H.264 motion compensation, film grain technology",                             &DXVA2_ModeH264_B,                      0, NULL },
185     { "H.264 motion compensation, no film grain technology",                          &DXVA2_ModeH264_A,                      0, NULL },
186     { "H.264 motion compensation, no film grain technology (Intel)",                  &DXVADDI_Intel_ModeH264_A,              0, NULL },
187
188     /* http://download.microsoft.com/download/2/D/0/2D02E72E-7890-430F-BA91-4A363F72F8C8/DXVA_H264_MVC.pdf */
189     { "H.264 stereo high profile, mbs flag set",                                      &DXVA_ModeH264_VLD_Stereo_Progressive_NoFGT, 0, NULL },
190     { "H.264 stereo high profile",                                                    &DXVA_ModeH264_VLD_Stereo_NoFGT,             0, NULL },
191     { "H.264 multiview high profile",                                                 &DXVA_ModeH264_VLD_Multiview_NoFGT,          0, NULL },
192
193     /* SVC http://download.microsoft.com/download/C/8/A/C8AD9F1B-57D1-4C10-85A0-09E3EAC50322/DXVA_SVC_2012_06.pdf */
194     { "H.264 scalable video coding, Scalable Baseline Profile",                       &DXVA_ModeH264_VLD_SVC_Scalable_Baseline,            0, NULL },
195     { "H.264 scalable video coding, Scalable Constrained Baseline Profile",           &DXVA_ModeH264_VLD_SVC_Restricted_Scalable_Baseline, 0, NULL },
196     { "H.264 scalable video coding, Scalable High Profile",                           &DXVA_ModeH264_VLD_SVC_Scalable_High,                0, NULL },
197     { "H.264 scalable video coding, Scalable Constrained High Profile",               &DXVA_ModeH264_VLD_SVC_Restricted_Scalable_High_Progressive, 0, NULL },
198
199     /* WMV */
200     { "Windows Media Video 8 motion compensation",                                    &DXVA2_ModeWMV8_B,                      0, NULL },
201     { "Windows Media Video 8 post processing",                                        &DXVA2_ModeWMV8_A,                      0, NULL },
202
203     { "Windows Media Video 9 IDCT",                                                   &DXVA2_ModeWMV9_C,                      0, NULL },
204     { "Windows Media Video 9 motion compensation",                                    &DXVA2_ModeWMV9_B,                      0, NULL },
205     { "Windows Media Video 9 post processing",                                        &DXVA2_ModeWMV9_A,                      0, NULL },
206
207     /* VC-1 */
208     { "VC-1 variable-length decoder",                                                 &DXVA2_ModeVC1_D,                       AV_CODEC_ID_VC1, NULL },
209     { "VC-1 variable-length decoder",                                                 &DXVA2_ModeVC1_D,                       AV_CODEC_ID_WMV3, NULL },
210     { "VC-1 variable-length decoder",                                                 &DXVA2_ModeVC1_D2010,                   AV_CODEC_ID_VC1, NULL },
211     { "VC-1 variable-length decoder",                                                 &DXVA2_ModeVC1_D2010,                   AV_CODEC_ID_WMV3, NULL },
212     { "VC-1 variable-length decoder 2 (Intel)",                                       &DXVA_Intel_VC1_ClearVideo_2,           0, NULL },
213     { "VC-1 variable-length decoder (Intel)",                                         &DXVA_Intel_VC1_ClearVideo,             0, NULL },
214
215     { "VC-1 inverse discrete cosine transform",                                       &DXVA2_ModeVC1_C,                       0, NULL },
216     { "VC-1 motion compensation",                                                     &DXVA2_ModeVC1_B,                       0, NULL },
217     { "VC-1 post processing",                                                         &DXVA2_ModeVC1_A,                       0, NULL },
218
219     /* Xvid/Divx: TODO */
220     { "MPEG-4 Part 2 nVidia bitstream decoder",                                       &DXVA_nVidia_MPEG4_ASP,                 0, NULL },
221     { "MPEG-4 Part 2 variable-length decoder, Simple Profile",                        &DXVA_ModeMPEG4pt2_VLD_Simple,          0, NULL },
222     { "MPEG-4 Part 2 variable-length decoder, Simple&Advanced Profile, no GMC",       &DXVA_ModeMPEG4pt2_VLD_AdvSimple_NoGMC, 0, NULL },
223     { "MPEG-4 Part 2 variable-length decoder, Simple&Advanced Profile, GMC",          &DXVA_ModeMPEG4pt2_VLD_AdvSimple_GMC,   0, NULL },
224     { "MPEG-4 Part 2 variable-length decoder, Simple&Advanced Profile, Avivo",        &DXVA_ModeMPEG4pt2_VLD_AdvSimple_Avivo, 0, NULL },
225
226     /* HEVC */
227     { "HEVC Main profile",                                                            &DXVA_ModeHEVC_VLD_Main,                AV_CODEC_ID_HEVC, PROF_HEVC_MAIN },
228     { "HEVC Main 10 profile",                                                         &DXVA_ModeHEVC_VLD_Main10,              AV_CODEC_ID_HEVC, PROF_HEVC_MAIN10 },
229
230     /* H.261 */
231     { "H.261 decoder, restricted profile A",                                          &DXVA_ModeH261_A,                       0, NULL },
232     { "H.261 decoder, restricted profile B",                                          &DXVA_ModeH261_B,                       0, NULL },
233
234     /* H.263 */
235     { "H.263 decoder, restricted profile A",                                          &DXVA_ModeH263_A,                       0, NULL },
236     { "H.263 decoder, restricted profile B",                                          &DXVA_ModeH263_B,                       0, NULL },
237     { "H.263 decoder, restricted profile C",                                          &DXVA_ModeH263_C,                       0, NULL },
238     { "H.263 decoder, restricted profile D",                                          &DXVA_ModeH263_D,                       0, NULL },
239     { "H.263 decoder, restricted profile E",                                          &DXVA_ModeH263_E,                       0, NULL },
240     { "H.263 decoder, restricted profile F",                                          &DXVA_ModeH263_F,                       0, NULL },
241
242     /* VPx */
243     { "VP8",                                                                          &DXVA_ModeVP8_VLD,                      0, NULL },
244 #if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT( 57, 17, 100 ) && LIBAVCODEC_VERSION_MICRO >= 100
245     { "VP9 profile 0",                                                                &DXVA_ModeVP9_VLD_Profile0,             AV_CODEC_ID_VP9, NULL },
246 #else
247     { "VP9 profile 0",                                                                &DXVA_ModeVP9_VLD_Profile0,             0, NULL },
248 #endif
249
250     { NULL, NULL, 0, NULL }
251 };
252
253 static int FindVideoServiceConversion(vlc_va_t *, directx_sys_t *, const es_format_t *fmt);
254 static void DestroyVideoDecoder(vlc_va_t *, directx_sys_t *);
255 static void DestroyVideoService(vlc_va_t *, directx_sys_t *);
256 static void DestroyDeviceManager(vlc_va_t *, directx_sys_t *);
257 static void DestroyDevice(vlc_va_t *, directx_sys_t *);
258
259 char *directx_va_GetDecoderName(const GUID *guid)
260 {
261     for (unsigned i = 0; DXVA_MODES[i].name; i++) {
262         if (IsEqualGUID(DXVA_MODES[i].guid, guid))
263             return strdup(DXVA_MODES[i].name);
264     }
265
266     char *psz_name = malloc(36);
267     if (likely(psz_name))
268         asprintf(&psz_name, "Unknown decoder " GUID_FMT, GUID_PRINT(*guid));
269     return psz_name;
270 }
271
272 /* */
273 int directx_va_Setup(vlc_va_t *va, directx_sys_t *dx_sys, AVCodecContext *avctx)
274 {
275     int surface_alignment = 16;
276     int surface_count = 2;
277
278     if (dx_sys->width == avctx->coded_width && dx_sys->height == avctx->coded_height
279      && dx_sys->decoder != NULL)
280         goto ok;
281
282     /* */
283     DestroyVideoDecoder(va, dx_sys);
284
285     avctx->hwaccel_context = NULL;
286     if (avctx->coded_width <= 0 || avctx->coded_height <= 0)
287         return VLC_EGENERIC;
288
289     /* */
290     msg_Dbg(va, "directx_va_Setup id %d %dx%d", dx_sys->codec_id, avctx->coded_width, avctx->coded_height);
291
292     switch ( dx_sys->codec_id )
293     {
294     case AV_CODEC_ID_MPEG2VIDEO:
295         /* decoding MPEG-2 requires additional alignment on some Intel GPUs,
296            but it causes issues for H.264 on certain AMD GPUs..... */
297         surface_alignment = 32;
298         surface_count += 2;
299         break;
300     case AV_CODEC_ID_HEVC:
301         /* the HEVC DXVA2 spec asks for 128 pixel aligned surfaces to ensure
302            all coding features have enough room to work with */
303         surface_alignment = 128;
304         surface_count += 16;
305         break;
306     case AV_CODEC_ID_H264:
307         surface_count += 16;
308         break;
309     default:
310         surface_count += 2;
311     }
312
313     if ( avctx->active_thread_type & FF_THREAD_FRAME )
314         surface_count += dx_sys->thread_count;
315
316     if (surface_count > MAX_SURFACE_COUNT)
317         return VLC_EGENERIC;
318
319     dx_sys->surface_count = surface_count;
320
321 #define ALIGN(x, y) (((x) + ((y) - 1)) & ~((y) - 1))
322     dx_sys->width  = avctx->coded_width;
323     dx_sys->height = avctx->coded_height;
324     dx_sys->surface_width  = ALIGN(dx_sys->width, surface_alignment);
325     dx_sys->surface_height = ALIGN(dx_sys->height, surface_alignment);
326
327     /* FIXME transmit a video_format_t by VaSetup directly */
328     video_format_t fmt;
329     memset(&fmt, 0, sizeof(fmt));
330     fmt.i_width = dx_sys->width;
331     fmt.i_height = dx_sys->height;
332     fmt.i_frame_rate = avctx->framerate.num;
333     fmt.i_frame_rate_base = avctx->framerate.den;
334
335     if (dx_sys->pf_create_decoder_surfaces(va, dx_sys->codec_id, &fmt))
336         return VLC_EGENERIC;
337
338     if (avctx->coded_width != dx_sys->surface_width ||
339         avctx->coded_height != dx_sys->surface_height)
340         msg_Warn( va, "surface dimensions (%dx%d) differ from avcodec dimensions (%dx%d)",
341                   dx_sys->surface_width, dx_sys->surface_height,
342                   avctx->coded_width, avctx->coded_height);
343
344     dx_sys->pf_setup_avcodec_ctx(va);
345
346     for (int i = 0; i < dx_sys->surface_count; i++) {
347         vlc_va_surface_t *surface = &dx_sys->surface[i];
348         surface->refcount = 0;
349         surface->order = 0;
350         surface->p_lock = &dx_sys->surface_lock;
351         surface->p_pic = dx_sys->pf_alloc_surface_pic(va, &fmt, i);
352         if (unlikely(surface->p_pic == NULL))
353             return VLC_EGENERIC;
354     }
355
356 ok:
357     return VLC_SUCCESS;
358 }
359
360 void DestroyVideoDecoder(vlc_va_t *va, directx_sys_t *dx_sys)
361 {
362     dx_sys->pf_destroy_surfaces(va);
363
364     for (int i = 0; i < dx_sys->surface_count; i++)
365         IUnknown_Release( dx_sys->hw_surface[i] );
366
367     for (int i = 0; i < dx_sys->surface_count; i++)
368         if (dx_sys->surface[i].p_pic)
369             picture_Release(dx_sys->surface[i].p_pic);
370
371     if (dx_sys->decoder)
372         IUnknown_Release( dx_sys->decoder );
373
374     dx_sys->decoder = NULL;
375     dx_sys->surface_count = 0;
376 }
377
378 /* FIXME it is nearly common with VAAPI */
379 int directx_va_Get(vlc_va_t *va, directx_sys_t *dx_sys, picture_t *pic, uint8_t **data)
380 {
381     /* Check the device */
382     if (dx_sys->pf_check_device(va)!=VLC_SUCCESS)
383         return VLC_EGENERIC;
384
385     vlc_mutex_lock( &dx_sys->surface_lock );
386
387     /* Grab the oldest unused surface, in case none are, use the oldest used one
388      * XXX using the used one is a workaround in case a problem happens with libavcodec */
389     int i, old = -1, old_used = -1;
390
391     for (i = 0; i < dx_sys->surface_count; i++) {
392         vlc_va_surface_t *surface = &dx_sys->surface[i];
393         if (((old == -1 || surface->order < dx_sys->surface[old].order)) && !surface->refcount)
394             old = i;
395         if (old_used == -1 || surface->order < dx_sys->surface[old_used].order)
396             old_used = i;
397     }
398     if (old >= 0)
399         i = old;
400     else if (old_used >= 0)
401     {
402         msg_Warn(va, "couldn't find a free decoding buffer, using index %d", old_used);
403         i = old_used;
404     }
405
406     vlc_va_surface_t *surface = &dx_sys->surface[i];
407
408     surface->refcount = 1;
409     surface->order = ++dx_sys->surface_order;
410     *data = (void *)dx_sys->hw_surface[i];
411     pic->context = surface;
412
413     vlc_mutex_unlock( &dx_sys->surface_lock );
414
415     return VLC_SUCCESS;
416 }
417
418 void directx_va_Release(void *opaque, uint8_t *data)
419 {
420     picture_t *pic = opaque;
421     vlc_va_surface_t *surface = pic->context;
422     vlc_mutex_lock( surface->p_lock );
423
424     surface->refcount--;
425     pic->context = NULL;
426     picture_Release(pic);
427
428     vlc_mutex_unlock( surface->p_lock );
429 }
430
431 void directx_va_Close(vlc_va_t *va, directx_sys_t *dx_sys)
432 {
433     DestroyVideoDecoder(va, dx_sys);
434     DestroyVideoService(va, dx_sys);
435     DestroyDeviceManager(va, dx_sys);
436     DestroyDevice(va, dx_sys);
437
438     if (dx_sys->hdecoder_dll)
439         FreeLibrary(dx_sys->hdecoder_dll);
440
441     vlc_mutex_destroy( &dx_sys->surface_lock );
442 }
443
444 int directx_va_Open(vlc_va_t *va, directx_sys_t *dx_sys,
445                     AVCodecContext *ctx, const es_format_t *fmt, bool b_dll)
446 {
447     // TODO va->sys = sys;
448     dx_sys->codec_id = ctx->codec_id;
449
450     vlc_mutex_init( &dx_sys->surface_lock );
451
452     if (b_dll) {
453         /* Load dll*/
454         dx_sys->hdecoder_dll = LoadLibrary(dx_sys->psz_decoder_dll);
455         if (!dx_sys->hdecoder_dll) {
456             msg_Warn(va, "cannot load DirectX decoder DLL");
457             goto error;
458         }
459         msg_Dbg(va, "DLLs loaded");
460     }
461
462     /* */
463     if (dx_sys->pf_create_device(va)) {
464         msg_Err(va, "Failed to create DirectX device");
465         goto error;
466     }
467     msg_Dbg(va, "CreateDevice succeed");
468
469     if (dx_sys->pf_create_device_manager(va)) {
470         msg_Err(va, "D3dCreateDeviceManager failed");
471         goto error;
472     }
473
474     if (dx_sys->pf_create_video_service(va)) {
475         msg_Err(va, "DxCreateVideoService failed");
476         goto error;
477     }
478
479     /* */
480     if (FindVideoServiceConversion(va, dx_sys, fmt)) {
481         msg_Err(va, "FindVideoServiceConversion failed");
482         goto error;
483     }
484
485     dx_sys->thread_count = ctx->thread_count;
486
487     return VLC_SUCCESS;
488
489 error:
490     return VLC_EGENERIC;
491 }
492
493 static bool profile_supported(const directx_va_mode_t *mode, const es_format_t *fmt)
494 {
495     bool is_supported = mode->p_profiles == NULL || !mode->p_profiles[0];
496     if (!is_supported)
497     {
498         int profile = fmt->i_profile;
499         if (mode->codec == AV_CODEC_ID_H264)
500         {
501             uint8_t h264_profile;
502             if ( h264_get_profile_level(fmt, &h264_profile, NULL, NULL) )
503                 profile = h264_profile;
504         }
505         if (mode->codec == AV_CODEC_ID_HEVC)
506         {
507             uint8_t hevc_profile;
508             if (hevc_get_profile_level(fmt, &hevc_profile, NULL, NULL) )
509                 profile = hevc_profile;
510         }
511
512         if (profile <= 0)
513             is_supported = true;
514         else for (const int *p_profile = &mode->p_profiles[0]; *p_profile; ++p_profile)
515         {
516             if (*p_profile == profile)
517             {
518                 is_supported = true;
519                 break;
520             }
521         }
522     }
523     return is_supported;
524 }
525
526 void DestroyVideoService(vlc_va_t *va, directx_sys_t *dx_sys)
527 {
528     dx_sys->pf_destroy_video_service(va);
529     if (dx_sys->d3ddec)
530         IUnknown_Release(dx_sys->d3ddec);
531 }
532
533 /**
534  * Find the best suited decoder mode GUID and render format.
535  */
536 static int FindVideoServiceConversion(vlc_va_t *va, directx_sys_t *dx_sys, const es_format_t *fmt)
537 {
538     input_list_t p_list = { 0 };
539     int err = dx_sys->pf_get_input_list(va, &p_list);
540     if (err != VLC_SUCCESS)
541         return err;
542     if (p_list.count == 0) {
543         msg_Warn( va, "No input format found for HWAccel" );
544         return VLC_EGENERIC;
545     }
546
547     err = VLC_EGENERIC;
548     /* Retreive supported modes from the decoder service */
549     for (unsigned i = 0; i < p_list.count; i++) {
550         const GUID *g = &p_list.list[i];
551         char *psz_decoder_name = directx_va_GetDecoderName(g);
552         msg_Dbg(va, "- '%s' is supported", psz_decoder_name);
553         free(psz_decoder_name);
554     }
555
556     /* Try all supported mode by our priority */
557     const directx_va_mode_t *mode = DXVA_MODES;
558     for (; mode->name; ++mode) {
559         if (!mode->codec || mode->codec != dx_sys->codec_id)
560             continue;
561
562         /* */
563         bool is_supported = false;
564         for (const GUID *g = &p_list.list[0]; !is_supported && g < &p_list.list[p_list.count]; g++) {
565             is_supported = IsEqualGUID(mode->guid, g);
566         }
567         if ( is_supported )
568         {
569             is_supported = profile_supported( mode, fmt );
570             if (!is_supported)
571                 msg_Warn( va, "Unsupported profile %d for %s ",
572                           fmt->i_profile, directx_va_GetDecoderName(mode->guid) );
573         }
574         if (!is_supported)
575             continue;
576
577         /* */
578         msg_Dbg(va, "Trying to use '%s' as input", mode->name);
579         if (dx_sys->pf_setup_output(va, mode->guid, &fmt->video)==VLC_SUCCESS)
580         {
581             dx_sys->input = *mode->guid;
582             err = VLC_SUCCESS;
583             break;
584         }
585     }
586
587     p_list.pf_release(&p_list);
588     return err;
589 }
590
591 void DestroyDeviceManager(vlc_va_t *va, directx_sys_t *dx_sys)
592 {
593     dx_sys->pf_destroy_device_manager(va);
594 }
595
596 void DestroyDevice(vlc_va_t *va, directx_sys_t *dx_sys)
597 {
598     dx_sys->pf_destroy_device(va);
599     if (dx_sys->d3ddev)
600         IUnknown_Release( dx_sys->d3ddev );
601 }