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
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
|
/* Copyright (C) 2013 Mamadou DIOP
* Copyright (C) 2013 Doubango Telecom <http://www.doubango.org>
*
* This file is part of Open Source Doubango Framework.
*
* DOUBANGO is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* DOUBANGO is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with DOUBANGO.
*/
#ifndef PLUGIN_WIN_MF_UTILS_H
#define PLUGIN_WIN_MF_UTILS_H
#include "../plugin_win_mf_config.h"
#include <new>
#include <mfapi.h>
#include <mfidl.h>
#include <Mferror.h>
#include <shlwapi.h>
#undef SafeRelease
#define SafeRelease(ppT) \
{ \
if (*ppT) \
{ \
(*ppT)->Release(); \
*ppT = NULL; \
} \
}
#undef CHECK_HR
// In CHECK_HR(x) When (x) is a function it will be executed twice when used in "TSK_DEBUG_ERROR(x)" and "If(x)"
#define CHECK_HR(x) { HRESULT __hr__ = (x); if (FAILED(__hr__)) { TSK_DEBUG_ERROR("Operation Failed (%08x)", __hr__); goto bail; } }
typedef struct VideoSubTypeGuidPair {
enum tmedia_chroma_e chroma;
const GUID& fourcc;
}
VideoSubTypeGuidPair;
class MFUtils
{
public:
static HRESULT Startup();
static HRESULT Shutdown();
static BOOL IsD3D9Supported();
static BOOL IsLowLatencyH264Supported();
static BOOL IsLowLatencyH264SupportsMaxSliceSize();
static HRESULT IsAsyncMFT(
IMFTransform *pMFT, // The MFT to check
BOOL* pbIsAsync // Whether the MFT is Async
);
static HRESULT UnlockAsyncMFT(
IMFTransform *pMFT // The MFT to unlock
);
static HRESULT CreatePCMAudioType(
UINT32 sampleRate, // Samples per second
UINT32 bitsPerSample, // Bits per sample
UINT32 cChannels, // Number of channels
IMFMediaType **ppType // Receives a pointer to the media type.
);
static HRESULT CreateVideoType(
const GUID* subType, // video subType
IMFMediaType **ppType, // Receives a pointer to the media type.
UINT32 unWidth = 0, // Video width (0 to ignore)
UINT32 unHeight = 0 // Video height (0 to ignore)
);
static HRESULT ConvertVideoTypeToUncompressedType(
IMFMediaType *pType, // Pointer to an encoded video type.
const GUID& subtype, // Uncompressed subtype (eg, RGB-32, AYUV)
IMFMediaType **ppType // Receives a matching uncompressed video type.
);
static HRESULT CreateMediaSample(
DWORD cbData, // Maximum buffer size
IMFSample **ppSample // Receives the sample
);
static HRESULT ValidateVideoFormat(
IMFMediaType *pmt
);
static HRESULT IsVideoProcessorSupported(BOOL *pbSupported);
static HRESULT GetBestVideoProcessor(
const GUID& inputFormat, // The input MediaFormat (e.g. MFVideoFormat_I420)
const GUID& outputFormat, // The output MediaFormat (e.g. MFVideoFormat_NV12)
IMFTransform **ppProcessor // Receives the video processor
);
static HRESULT GetBestCodec(
BOOL bEncoder, // Whether we request an encoder or not (TRUE=encoder, FALSE=decoder)
const GUID& mediaType, // The MediaType
const GUID& inputFormat, // The input MediaFormat (e.g. MFVideoFormat_NV12)
const GUID& outputFormat, // The output MediaFormat (e.g. MFVideoFormat_H264)
IMFTransform **ppMFT // Receives the decoder/encoder transform
);
static HRESULT BindOutputNode(
IMFTopologyNode *pNode // The Node
);
static HRESULT AddOutputNode(
IMFTopology *pTopology, // Topology.
IMFActivate *pActivate, // Media sink activation object.
DWORD dwId, // Identifier of the stream sink.
IMFTopologyNode **ppNode // Receives the node pointer.
);
static HRESULT AddTransformNode(
IMFTopology *pTopology, // Topology.
IMFTransform *pMFT, // MFT.
DWORD dwId, // Identifier of the stream sink.
IMFTopologyNode **ppNode // Receives the node pointer.
);
static HRESULT AddSourceNode(
IMFTopology *pTopology, // Topology.
IMFMediaSource *pSource, // Media source.
IMFPresentationDescriptor *pPD, // Presentation descriptor.
IMFStreamDescriptor *pSD, // Stream descriptor.
IMFTopologyNode **ppNode // Receives the node pointer.
);
static HRESULT CreateTopology(
IMFMediaSource *pSource, // Media source
IMFTransform *pTransform, // Transform filter (e.g. encoder or decoder) to insert between the source and Sink. NULL is valid.
IMFActivate *pSinkActivateMain, // Main sink (e.g. sample grabber or EVR).
IMFActivate *pSinkActivatePreview, // Preview sink. Optional. Could be NULL.
IMFMediaType *pIputTypeMain, // Main sink input MediaType
IMFTopology **ppTopo // Receives the newly created topology
);
static HRESULT ResolveTopology(
IMFTopology *pInputTopo, // A pointer to the IMFTopology interface of the partial topology to be resolved.
IMFTopology **ppOutputTopo, // Receives a pointer to the IMFTopology interface of the completed topology. The caller must release the interface.
IMFTopology *pCurrentTopo = NULL // A pointer to the IMFTopology interface of the previous full topology. The topology loader can re-use objects from this topology in the new topology. This parameter can be NULL.
);
static HRESULT FindNodeObject(
IMFTopology *pInputTopo, // The Topology containing the node to find
TOPOID qwTopoNodeID, //The identifier for the node
void** ppObject // Receives the Object
);
static HRESULT CreateMediaSinkActivate(
IMFStreamDescriptor *pSourceSD, // Pointer to the stream descriptor.
HWND hVideoWindow, // Handle to the video clipping window.
IMFActivate **ppActivate
);
static HRESULT SetMediaType(
IMFMediaSource *pSource, // Media source.
IMFMediaType* pMediaType // Media Type.
);
static HRESULT SetVideoWindow(
IMFTopology *pTopology, // Topology.
IMFMediaSource *pSource, // Media source.
HWND hVideoWnd // Window for video playback.
);
static HRESULT RunSession(
IMFMediaSession *pSession, // Session to run
IMFTopology *pTopology // The toppology
);
static HRESULT ShutdownSession(
IMFMediaSession *pSession, // The Session
IMFMediaSource *pSource = NULL // Source to shutdown (optional)
);
static HRESULT PauseSession(
IMFMediaSession *pSession, // The session
IMFMediaSource *pSource = NULL// Source to pause (optional)
);
static INT GetSupportedSubTypeIndex(
IMFMediaSource *pSource, // The source
const GUID& mediaType, // The MediaType
const VideoSubTypeGuidPair* subTypes, UINT subTypesCount // List of preferred subtypes (in ascending order)
);
static HRESULT IsSupported(
IMFPresentationDescriptor *pPD,
DWORD cStreamIndex,
UINT32 nWidth,
UINT32 nHeight,
UINT32 nFps,
const GUID& guidFormat,
BOOL* pbSupportedSize,
BOOL* pbSupportedFps,
BOOL* pbSupportedFormat
);
static HRESULT IsSupported(
IMFPresentationDescriptor *pPD,
DWORD cStreamIndex,
IMFMediaType* pMediaType,
BOOL* pbSupportedSize,
BOOL* pbSupportedFps,
BOOL* pbSupportedFormat
);
static HRESULT IsSupportedByInput(
IMFPresentationDescriptor *pPD,
DWORD cStreamIndex,
IMFTopologyNode *pNode,
BOOL* pbSupportedSize,
BOOL* pbSupportedFps,
BOOL* pbSupportedFormat
);
static HRESULT ConnectConverters(
IMFTopologyNode *pNode,
DWORD dwOutputIndex,
IMFTopologyNode *pNodeConvFrameRate,
IMFTopologyNode *pNodeConvColor,
IMFTopologyNode *pNodeConvSize
);
static HRESULT GetBestFormat(
IMFMediaSource *pSource,
const GUID *pSubType,
UINT32 nWidth,
UINT32 nHeight,
UINT32 nFps,
UINT32 *pnWidth,
UINT32 *pnHeight,
UINT32 *pnFps,
const VideoSubTypeGuidPair **pSubTypeGuidPair
);
static HWND GetConsoleHwnd(void);
template <class Q>
static HRESULT GetTopoNodeObject(IMFTopologyNode *pNode, Q **ppObject) {
IUnknown *pUnk = NULL; // zero output
HRESULT hr = pNode->GetObject(&pUnk);
if (SUCCEEDED(hr)) {
pUnk->QueryInterface(IID_PPV_ARGS(ppObject));
pUnk->Release();
}
return hr;
}
private:
static BOOL g_bStarted;
static DWORD g_dwMajorVersion;
static DWORD g_dwMinorVersion;
static BOOL g_bLowLatencyH264Checked;
static BOOL g_bLowLatencyH264Supported;
static BOOL g_bLowLatencyH264SupportsMaxSliceSize;
static BOOL g_bD3D9Checked;
static BOOL g_bD3D9Supported;
public:
static const TOPOID g_ullTopoIdSinkMain;
static const TOPOID g_ullTopoIdSinkPreview;
static const TOPOID g_ullTopoIdSource;
static const TOPOID g_ullTopoIdVideoProcessor;
};
#endif /* PLUGIN_WIN_MF_UTILS_H */
|
