1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
media / base / win / color_space_util_win.cc [blame]
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "media/base/win/color_space_util_win.h"
#include <initguid.h>
#include <mfapi.h>
#include "base/logging.h"
#include "media/base/video_color_space.h"
namespace media {
namespace {
gfx::ColorSpace::PrimaryID MFPrimaryToColorSpace(uint32_t mf_primary) {
switch (mf_primary) {
case MFVideoPrimaries_BT709:
return gfx::ColorSpace::PrimaryID::BT709;
case MFVideoPrimaries_BT470_2_SysM:
return gfx::ColorSpace::PrimaryID::BT470M;
case MFVideoPrimaries_BT470_2_SysBG:
return gfx::ColorSpace::PrimaryID::BT470BG;
case MFVideoPrimaries_SMPTE170M:
case MFVideoPrimaries_SMPTE_C:
return gfx::ColorSpace::PrimaryID::SMPTE170M;
case MFVideoPrimaries_SMPTE240M:
return gfx::ColorSpace::PrimaryID::SMPTE240M;
case MFVideoPrimaries_EBU3213:
return gfx::ColorSpace::PrimaryID::EBU_3213_E;
case MFVideoPrimaries_BT2020:
return gfx::ColorSpace::PrimaryID::BT2020;
case MFVideoPrimaries_XYZ:
return gfx::ColorSpace::PrimaryID::SMPTEST428_1;
case MFVideoPrimaries_DCI_P3:
return gfx::ColorSpace::PrimaryID::SMPTEST431_2;
default:
return gfx::ColorSpace::PrimaryID::INVALID;
}
}
MFVideoPrimaries ColorSpaceToMFPrimary(gfx::ColorSpace::PrimaryID color_space) {
switch (color_space) {
case gfx::ColorSpace::PrimaryID::BT709:
return MFVideoPrimaries_BT709;
case gfx::ColorSpace::PrimaryID::BT470M:
return MFVideoPrimaries_BT470_2_SysM;
case gfx::ColorSpace::PrimaryID::BT470BG:
return MFVideoPrimaries_BT470_2_SysBG;
case gfx::ColorSpace::PrimaryID::SMPTE170M:
return MFVideoPrimaries_SMPTE170M;
case gfx::ColorSpace::PrimaryID::SMPTE240M:
return MFVideoPrimaries_SMPTE240M;
case gfx::ColorSpace::PrimaryID::EBU_3213_E:
return MFVideoPrimaries_EBU3213;
case gfx::ColorSpace::PrimaryID::BT2020:
return MFVideoPrimaries_BT2020;
case gfx::ColorSpace::PrimaryID::SMPTEST428_1:
return MFVideoPrimaries_XYZ;
case gfx::ColorSpace::PrimaryID::SMPTEST431_2:
return MFVideoPrimaries_DCI_P3;
default:
return MFVideoPrimaries_Unknown;
}
}
gfx::ColorSpace::TransferID MFTransferToColorSpace(uint32_t mf_transfer) {
switch (mf_transfer) {
case MFVideoTransFunc_18:
return gfx::ColorSpace::TransferID::GAMMA18;
case MFVideoTransFunc_22:
return gfx::ColorSpace::TransferID::GAMMA22;
case MFVideoTransFunc_709:
return gfx::ColorSpace::TransferID::BT709;
case MFVideoTransFunc_240M:
return gfx::ColorSpace::TransferID::SMPTE240M;
case MFVideoTransFunc_sRGB:
return gfx::ColorSpace::TransferID::SRGB;
case MFVideoTransFunc_28:
return gfx::ColorSpace::TransferID::GAMMA28;
case MFVideoTransFunc_2020:
return gfx::ColorSpace::TransferID::BT2020_10;
case MFVideoTransFunc_2084:
return gfx::ColorSpace::TransferID::PQ;
case MFVideoTransFunc_HLG:
return gfx::ColorSpace::TransferID::HLG;
default:
return gfx::ColorSpace::TransferID::INVALID;
}
}
MFVideoTransferFunction ColorSpaceToMFTransfer(
gfx::ColorSpace::TransferID color_space) {
switch (color_space) {
case gfx::ColorSpace::TransferID::GAMMA18:
return MFVideoTransFunc_18;
case gfx::ColorSpace::TransferID::GAMMA22:
return MFVideoTransFunc_22;
case gfx::ColorSpace::TransferID::BT709:
return MFVideoTransFunc_709;
case gfx::ColorSpace::TransferID::SMPTE240M:
return MFVideoTransFunc_240M;
case gfx::ColorSpace::TransferID::SRGB:
return MFVideoTransFunc_sRGB;
case gfx::ColorSpace::TransferID::GAMMA28:
return MFVideoTransFunc_28;
case gfx::ColorSpace::TransferID::BT2020_10:
case gfx::ColorSpace::TransferID::BT2020_12:
return MFVideoTransFunc_2020;
case gfx::ColorSpace::TransferID::PQ:
return MFVideoTransFunc_2084;
case gfx::ColorSpace::TransferID::HLG:
return MFVideoTransFunc_HLG;
default:
return MFVideoTransFunc_Unknown;
}
}
gfx::ColorSpace::MatrixID MFMatrixToColorSpace(uint32_t mf_matrix) {
switch (mf_matrix) {
case MFVideoTransferMatrix_BT709:
return gfx::ColorSpace::MatrixID::BT709;
case MFVideoTransferMatrix_BT601:
return gfx::ColorSpace::MatrixID::SMPTE170M;
case MFVideoTransferMatrix_SMPTE240M:
return gfx::ColorSpace::MatrixID::SMPTE240M;
case MFVideoTransferMatrix_BT2020_10:
case MFVideoTransferMatrix_BT2020_12:
return gfx::ColorSpace::MatrixID::BT2020_NCL;
default:
return gfx::ColorSpace::MatrixID::INVALID;
}
}
MFVideoTransferMatrix ColorSpaceToMFMatrix(
gfx::ColorSpace::MatrixID color_space) {
switch (color_space) {
case gfx::ColorSpace::MatrixID::BT709:
return MFVideoTransferMatrix_BT709;
case gfx::ColorSpace::MatrixID::SMPTE170M:
return MFVideoTransferMatrix_BT601;
case gfx::ColorSpace::MatrixID::SMPTE240M:
return MFVideoTransferMatrix_SMPTE240M;
case gfx::ColorSpace::MatrixID::BT2020_NCL:
return MFVideoTransferMatrix_BT2020_10;
default:
return MFVideoTransferMatrix_Unknown;
}
}
gfx::ColorSpace::RangeID MFRangeToColorSpace(uint32_t mf_range) {
switch (mf_range) {
case MFNominalRange_0_255:
return gfx::ColorSpace::RangeID::FULL;
case MFNominalRange_16_235:
return gfx::ColorSpace::RangeID::LIMITED;
case MFNominalRange_48_208:
return gfx::ColorSpace::RangeID::LIMITED;
case MFNominalRange_64_127:
return gfx::ColorSpace::RangeID::LIMITED;
default:
return gfx::ColorSpace::RangeID::INVALID;
}
}
MFNominalRange ColorSpaceToMFRange(gfx::ColorSpace::RangeID color_space) {
switch (color_space) {
case gfx::ColorSpace::RangeID::FULL:
return MFNominalRange_0_255;
case gfx::ColorSpace::RangeID::DERIVED:
case gfx::ColorSpace::RangeID::LIMITED:
return MFNominalRange_16_235;
default:
return MFNominalRange_Unknown;
}
}
} // anonymous namespace
gfx::ColorSpace GetMediaTypeColorSpace(IMFMediaType* media_type) {
uint32_t mf_primary, mf_transfer, mf_matrix, mf_range;
HRESULT hr = media_type->GetUINT32(MF_MT_VIDEO_PRIMARIES, &mf_primary);
if (FAILED(hr))
mf_primary = MFVideoPrimaries_Unknown;
hr = media_type->GetUINT32(MF_MT_TRANSFER_FUNCTION, &mf_transfer);
if (FAILED(hr))
mf_transfer = MFVideoTransFunc_Unknown;
hr = media_type->GetUINT32(MF_MT_YUV_MATRIX, &mf_matrix);
if (FAILED(hr))
mf_matrix = MFVideoTransferMatrix_Unknown;
hr = media_type->GetUINT32(MF_MT_VIDEO_NOMINAL_RANGE, &mf_range);
if (FAILED(hr))
mf_range = MFNominalRange_Unknown;
gfx::ColorSpace::PrimaryID primary = MFPrimaryToColorSpace(mf_primary);
gfx::ColorSpace::TransferID transfer = MFTransferToColorSpace(mf_transfer);
gfx::ColorSpace::MatrixID matrix = MFMatrixToColorSpace(mf_matrix);
gfx::ColorSpace::RangeID range = MFRangeToColorSpace(mf_range);
// Convert to VideoColorSpace and back to fill in any gaps.
VideoColorSpace guesser = VideoColorSpace::FromGfxColorSpace(
gfx::ColorSpace(primary, transfer, matrix, range));
return guesser.GuessGfxColorSpace();
}
void GetMediaTypeColorValues(const gfx::ColorSpace& color_space,
MFVideoPrimaries* out_primaries,
MFVideoTransferFunction* out_transfer,
MFVideoTransferMatrix* out_matrix,
MFNominalRange* out_range) {
*out_primaries = ColorSpaceToMFPrimary(color_space.GetPrimaryID());
*out_transfer = ColorSpaceToMFTransfer(color_space.GetTransferID());
*out_matrix = ColorSpaceToMFMatrix(color_space.GetMatrixID());
*out_range = ColorSpaceToMFRange(color_space.GetRangeID());
}
} // namespace media