FreeRDP
Loading...
Searching...
No Matches
h264_mf.c
1
20#include <winpr/winpr.h>
21#include <winpr/library.h>
22
23#include <freerdp/log.h>
24#include <freerdp/codec/h264.h>
25
26#include <ks.h>
27#include <codecapi.h>
28
29#include <mfapi.h>
30#include <mferror.h>
31#include <wmcodecdsp.h>
32#include <mftransform.h>
33
34#include "h264.h"
35
36#define TAG FREERDP_TAG("codec")
37
38static const GUID sCLSID_CMSH264DecoderMFT = {
39 0x62CE7E72, 0x4C71, 0x4d20, { 0xB1, 0x5D, 0x45, 0x28, 0x31, 0xA8, 0x7D, 0x9D }
40};
41static const GUID sIID_IMFTransform = {
42 0xbf94c121, 0x5b05, 0x4e6f, { 0x80, 0x00, 0xba, 0x59, 0x89, 0x61, 0x41, 0x4d }
43};
44static const GUID sMF_MT_MAJOR_TYPE = {
45 0x48eba18e, 0xf8c9, 0x4687, { 0xbf, 0x11, 0x0a, 0x74, 0xc9, 0xf9, 0x6a, 0x8f }
46};
47static const GUID sMF_MT_FRAME_SIZE = {
48 0x1652c33d, 0xd6b2, 0x4012, { 0xb8, 0x34, 0x72, 0x03, 0x08, 0x49, 0xa3, 0x7d }
49};
50static const GUID sMF_MT_DEFAULT_STRIDE = {
51 0x644b4e48, 0x1e02, 0x4516, { 0xb0, 0xeb, 0xc0, 0x1c, 0xa9, 0xd4, 0x9a, 0xc6 }
52};
53static const GUID sMF_MT_SUBTYPE = {
54 0xf7e34c9a, 0x42e8, 0x4714, { 0xb7, 0x4b, 0xcb, 0x29, 0xd7, 0x2c, 0x35, 0xe5 }
55};
56static const GUID sMFMediaType_Video = {
57 0x73646976, 0x0000, 0x0010, { 0x80, 0x00, 0x00, 0xAA, 0x00, 0x38, 0x9B, 0x71 }
58};
59static const GUID sMFVideoFormat_H264 = {
60 0x34363248, 0x0000, 0x0010, { 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71 }
61};
62static const GUID sMFVideoFormat_IYUV = {
63 0x56555949, 0x0000, 0x0010, { 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71 }
64};
65static const GUID sIID_ICodecAPI = {
66 0x901db4c7, 0x31ce, 0x41a2, { 0x85, 0xdc, 0x8f, 0xa0, 0xbf, 0x41, 0xb8, 0xda }
67};
68static const GUID sCODECAPI_AVLowLatencyMode = {
69 0x9c27891a, 0xed7a, 0x40e1, { 0x88, 0xe8, 0xb2, 0x27, 0x27, 0xa0, 0x24, 0xee }
70};
71
72typedef HRESULT(__stdcall* pfnMFStartup)(ULONG Version, DWORD dwFlags);
73typedef HRESULT(__stdcall* pfnMFShutdown)(void);
74typedef HRESULT(__stdcall* pfnMFCreateSample)(IMFSample** ppIMFSample);
75typedef HRESULT(__stdcall* pfnMFCreateMemoryBuffer)(DWORD cbMaxLength, IMFMediaBuffer** ppBuffer);
76typedef HRESULT(__stdcall* pfnMFCreateMediaType)(IMFMediaType** ppMFType);
77
78typedef struct
79{
80 ICodecAPI* codecApi;
81 IMFTransform* transform;
82 IMFMediaType* inputType;
83 IMFMediaType* outputType;
84 IMFSample* sample;
85 UINT32 frameWidth;
86 UINT32 frameHeight;
87 IMFSample* outputSample;
88 IMFMediaBuffer* outputBuffer;
89 HMODULE mfplat;
90 pfnMFStartup MFStartup;
91 pfnMFShutdown MFShutdown;
92 pfnMFCreateSample MFCreateSample;
93 pfnMFCreateMemoryBuffer MFCreateMemoryBuffer;
94 pfnMFCreateMediaType MFCreateMediaType;
95} H264_CONTEXT_MF;
96
97static HRESULT mf_find_output_type(H264_CONTEXT_MF* sys, const GUID* guid,
98 IMFMediaType** ppMediaType)
99{
100 DWORD idx = 0;
101 GUID mediaGuid;
102 HRESULT hr = S_OK;
103 IMFMediaType* pMediaType = nullptr;
104
105 while (1)
106 {
107 hr = sys->transform->lpVtbl->GetOutputAvailableType(sys->transform, 0, idx, &pMediaType);
108
109 if (FAILED(hr))
110 break;
111
112 pMediaType->lpVtbl->GetGUID(pMediaType, &sMF_MT_SUBTYPE, &mediaGuid);
113
114 if (IsEqualGUID(&mediaGuid, guid))
115 {
116 *ppMediaType = pMediaType;
117 return S_OK;
118 }
119
120 pMediaType->lpVtbl->Release(pMediaType);
121 idx++;
122 }
123
124 return hr;
125}
126
127static HRESULT mf_create_output_sample(H264_CONTEXT* h264, H264_CONTEXT_MF* sys)
128{
129 HRESULT hr = S_OK;
130 MFT_OUTPUT_STREAM_INFO streamInfo;
131
132 if (sys->outputSample)
133 {
134 sys->outputSample->lpVtbl->Release(sys->outputSample);
135 sys->outputSample = nullptr;
136 }
137
138 hr = sys->MFCreateSample(&sys->outputSample);
139
140 if (FAILED(hr))
141 {
142 WLog_Print(h264->log, WLOG_ERROR, "MFCreateSample failure: 0x%08" PRIX32 "", hr);
143 goto error;
144 }
145
146 hr = sys->transform->lpVtbl->GetOutputStreamInfo(sys->transform, 0, &streamInfo);
147
148 if (FAILED(hr))
149 {
150 WLog_Print(h264->log, WLOG_ERROR, "GetOutputStreamInfo failure: 0x%08" PRIX32 "", hr);
151 goto error;
152 }
153
154 hr = sys->MFCreateMemoryBuffer(streamInfo.cbSize, &sys->outputBuffer);
155
156 if (FAILED(hr))
157 {
158 WLog_Print(h264->log, WLOG_ERROR, "MFCreateMemoryBuffer failure: 0x%08" PRIX32 "", hr);
159 goto error;
160 }
161
162 sys->outputSample->lpVtbl->AddBuffer(sys->outputSample, sys->outputBuffer);
163
164 if (FAILED(hr))
165 {
166 WLog_Print(h264->log, WLOG_ERROR, "AddBuffer failure: 0x%08" PRIX32 "", hr);
167 goto error;
168 }
169
170 sys->outputBuffer->lpVtbl->Release(sys->outputBuffer);
171error:
172 return hr;
173}
174
175static int mf_decompress(H264_CONTEXT* WINPR_RESTRICT h264, const BYTE* WINPR_RESTRICT pSrcData,
176 UINT32 SrcSize)
177{
178 HRESULT hr;
179 BYTE* pbBuffer = nullptr;
180 DWORD cbMaxLength = 0;
181 DWORD cbCurrentLength = 0;
182 DWORD outputStatus = 0;
183 IMFSample* inputSample = nullptr;
184 IMFMediaBuffer* inputBuffer = nullptr;
185 IMFMediaBuffer* outputBuffer = nullptr;
186 MFT_OUTPUT_DATA_BUFFER outputDataBuffer;
187 H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*)h264->pSystemData;
188 UINT32* iStride = h264->iStride;
189 BYTE** pYUVData = h264->pYUVData;
190 hr = sys->MFCreateMemoryBuffer(SrcSize, &inputBuffer);
191
192 if (FAILED(hr))
193 {
194 WLog_Print(h264->log, WLOG_ERROR, "MFCreateMemoryBuffer failure: 0x%08" PRIX32 "", hr);
195 goto error;
196 }
197
198 hr = inputBuffer->lpVtbl->Lock(inputBuffer, &pbBuffer, &cbMaxLength, &cbCurrentLength);
199
200 if (FAILED(hr))
201 {
202 WLog_Print(h264->log, WLOG_ERROR, "Lock failure: 0x%08" PRIX32 "", hr);
203 goto error;
204 }
205
206 CopyMemory(pbBuffer, pSrcData, SrcSize);
207 hr = inputBuffer->lpVtbl->SetCurrentLength(inputBuffer, SrcSize);
208
209 if (FAILED(hr))
210 {
211 WLog_Print(h264->log, WLOG_ERROR, "SetCurrentLength failure: 0x%08" PRIX32 "", hr);
212 goto error;
213 }
214
215 hr = inputBuffer->lpVtbl->Unlock(inputBuffer);
216
217 if (FAILED(hr))
218 {
219 WLog_Print(h264->log, WLOG_ERROR, "Unlock failure: 0x%08" PRIX32 "", hr);
220 goto error;
221 }
222
223 hr = sys->MFCreateSample(&inputSample);
224
225 if (FAILED(hr))
226 {
227 WLog_Print(h264->log, WLOG_ERROR, "MFCreateSample failure: 0x%08" PRIX32 "", hr);
228 goto error;
229 }
230
231 inputSample->lpVtbl->AddBuffer(inputSample, inputBuffer);
232
233 if (FAILED(hr))
234 {
235 WLog_Print(h264->log, WLOG_ERROR, "AddBuffer failure: 0x%08" PRIX32 "", hr);
236 goto error;
237 }
238
239 inputBuffer->lpVtbl->Release(inputBuffer);
240 hr = sys->transform->lpVtbl->ProcessInput(sys->transform, 0, inputSample, 0);
241
242 if (FAILED(hr))
243 {
244 WLog_Print(h264->log, WLOG_ERROR, "ProcessInput failure: 0x%08" PRIX32 "", hr);
245 goto error;
246 }
247
248 hr = mf_create_output_sample(h264, sys);
249
250 if (FAILED(hr))
251 {
252 WLog_Print(h264->log, WLOG_ERROR, "mf_create_output_sample failure: 0x%08" PRIX32 "", hr);
253 goto error;
254 }
255
256 outputDataBuffer.dwStreamID = 0;
257 outputDataBuffer.dwStatus = 0;
258 outputDataBuffer.pEvents = nullptr;
259 outputDataBuffer.pSample = sys->outputSample;
260 hr = sys->transform->lpVtbl->ProcessOutput(sys->transform, 0, 1, &outputDataBuffer,
261 &outputStatus);
262
263 if (hr == MF_E_TRANSFORM_STREAM_CHANGE)
264 {
265 UINT32 stride = 0;
266 UINT64 frameSize = 0;
267
268 if (sys->outputType)
269 {
270 sys->outputType->lpVtbl->Release(sys->outputType);
271 sys->outputType = nullptr;
272 }
273
274 hr = mf_find_output_type(sys, &sMFVideoFormat_IYUV, &sys->outputType);
275
276 if (FAILED(hr))
277 {
278 WLog_Print(h264->log, WLOG_ERROR, "mf_find_output_type failure: 0x%08" PRIX32 "", hr);
279 goto error;
280 }
281
282 hr = sys->transform->lpVtbl->SetOutputType(sys->transform, 0, sys->outputType, 0);
283
284 if (FAILED(hr))
285 {
286 WLog_Print(h264->log, WLOG_ERROR, "SetOutputType failure: 0x%08" PRIX32 "", hr);
287 goto error;
288 }
289
290 hr = mf_create_output_sample(h264, sys);
291
292 if (FAILED(hr))
293 {
294 WLog_Print(h264->log, WLOG_ERROR, "mf_create_output_sample failure: 0x%08" PRIX32 "",
295 hr);
296 goto error;
297 }
298
299 hr = sys->outputType->lpVtbl->GetUINT64(sys->outputType, &sMF_MT_FRAME_SIZE, &frameSize);
300
301 if (FAILED(hr))
302 {
303 WLog_Print(h264->log, WLOG_ERROR,
304 "GetUINT64(MF_MT_FRAME_SIZE) failure: 0x%08" PRIX32 "", hr);
305 goto error;
306 }
307
308 sys->frameWidth = (UINT32)(frameSize >> 32);
309 sys->frameHeight = (UINT32)frameSize;
310 hr = sys->outputType->lpVtbl->GetUINT32(sys->outputType, &sMF_MT_DEFAULT_STRIDE, &stride);
311
312 if (FAILED(hr))
313 {
314 WLog_Print(h264->log, WLOG_ERROR,
315 "GetUINT32(MF_MT_DEFAULT_STRIDE) failure: 0x%08" PRIX32 "", hr);
316 goto error;
317 }
318
319 if (!avc420_ensure_buffer(h264, stride, sys->frameWidth, sys->frameHeight))
320 goto error;
321 }
322 else if (hr == MF_E_TRANSFORM_NEED_MORE_INPUT)
323 {
324 }
325 else if (FAILED(hr))
326 {
327 WLog_Print(h264->log, WLOG_ERROR, "ProcessOutput failure: 0x%08" PRIX32 "", hr);
328 goto error;
329 }
330 else
331 {
332 int offset = 0;
333 BYTE* buffer = nullptr;
334 DWORD bufferCount = 0;
335 DWORD cbMaxLength = 0;
336 DWORD cbCurrentLength = 0;
337 hr = sys->outputSample->lpVtbl->GetBufferCount(sys->outputSample, &bufferCount);
338
339 if (FAILED(hr))
340 {
341 WLog_Print(h264->log, WLOG_ERROR, "GetBufferCount failure: 0x%08" PRIX32 "", hr);
342 goto error;
343 }
344
345 hr = sys->outputSample->lpVtbl->GetBufferByIndex(sys->outputSample, 0, &outputBuffer);
346
347 if (FAILED(hr))
348 {
349 WLog_Print(h264->log, WLOG_ERROR, "GetBufferByIndex failure: 0x%08" PRIX32 "", hr);
350 goto error;
351 }
352
353 hr = outputBuffer->lpVtbl->Lock(outputBuffer, &buffer, &cbMaxLength, &cbCurrentLength);
354
355 if (FAILED(hr))
356 {
357 WLog_Print(h264->log, WLOG_ERROR, "Lock failure: 0x%08" PRIX32 "", hr);
358 goto error;
359 }
360
361 CopyMemory(pYUVData[0], &buffer[offset], iStride[0] * sys->frameHeight);
362 offset += iStride[0] * sys->frameHeight;
363 CopyMemory(pYUVData[1], &buffer[offset], iStride[1] * (sys->frameHeight / 2));
364 offset += iStride[1] * (sys->frameHeight / 2);
365 CopyMemory(pYUVData[2], &buffer[offset], iStride[2] * (sys->frameHeight / 2));
366 offset += iStride[2] * (sys->frameHeight / 2);
367 hr = outputBuffer->lpVtbl->Unlock(outputBuffer);
368
369 if (FAILED(hr))
370 {
371 WLog_Print(h264->log, WLOG_ERROR, "Unlock failure: 0x%08" PRIX32 "", hr);
372 goto error;
373 }
374
375 outputBuffer->lpVtbl->Release(outputBuffer);
376 }
377
378 inputSample->lpVtbl->Release(inputSample);
379 return 1;
380error:
381 (void)fprintf(stderr, "mf_decompress error\n");
382 return -1;
383}
384
385static int mf_compress(H264_CONTEXT* WINPR_RESTRICT h264, const BYTE** WINPR_RESTRICT ppSrcYuv,
386 const UINT32* WINPR_RESTRICT pStride, BYTE** WINPR_RESTRICT ppDstData,
387 UINT32* WINPR_RESTRICT pDstSize)
388{
389 H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*)h264->pSystemData;
390 return 1;
391}
392
393static BOOL mf_plat_loaded(H264_CONTEXT_MF* sys)
394{
395 return sys->MFStartup && sys->MFShutdown && sys->MFCreateSample && sys->MFCreateMemoryBuffer &&
396 sys->MFCreateMediaType;
397}
398
399static void mf_uninit(H264_CONTEXT* h264)
400{
401 H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*)h264->pSystemData;
402
403 if (sys)
404 {
405 if (sys->transform)
406 {
407 sys->transform->lpVtbl->Release(sys->transform);
408 sys->transform = nullptr;
409 }
410
411 if (sys->codecApi)
412 {
413 sys->codecApi->lpVtbl->Release(sys->codecApi);
414 sys->codecApi = nullptr;
415 }
416
417 if (sys->inputType)
418 {
419 sys->inputType->lpVtbl->Release(sys->inputType);
420 sys->inputType = nullptr;
421 }
422
423 if (sys->outputType)
424 {
425 sys->outputType->lpVtbl->Release(sys->outputType);
426 sys->outputType = nullptr;
427 }
428
429 if (sys->outputSample)
430 {
431 sys->outputSample->lpVtbl->Release(sys->outputSample);
432 sys->outputSample = nullptr;
433 }
434
435 if (sys->mfplat)
436 {
437 if (mf_plat_loaded(sys))
438 sys->MFShutdown();
439
440 FreeLibrary(sys->mfplat);
441 sys->mfplat = nullptr;
442
443 if (mf_plat_loaded(sys))
444 CoUninitialize();
445 }
446
447 for (size_t x = 0; x < sizeof(h264->pYUVData) / sizeof(h264->pYUVData[0]); x++)
448 winpr_aligned_free(h264->pYUVData[x]);
449
450 memset(h264->pYUVData, 0, sizeof(h264->pYUVData));
451 memset(h264->iStride, 0, sizeof(h264->iStride));
452
453 free(sys);
454 h264->pSystemData = nullptr;
455 }
456}
457
458static BOOL mf_init(H264_CONTEXT* h264)
459{
460 HRESULT hr;
461 H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*)calloc(1, sizeof(H264_CONTEXT_MF));
462
463 if (!sys)
464 goto error;
465
466 h264->pSystemData = (void*)sys;
467 /* http://decklink-sdk-delphi.googlecode.com/svn/trunk/Blackmagic%20DeckLink%20SDK%209.7/Win/Samples/Streaming/StreamingPreview/DecoderMF.cpp
468 */
469 sys->mfplat = LoadLibraryA("mfplat.dll");
470
471 if (!sys->mfplat)
472 goto error;
473
474 sys->MFStartup = GetProcAddressAs(sys->mfplat, "MFStartup", pfnMFStartup);
475 sys->MFShutdown = GetProcAddressAs(sys->mfplat, "MFShutdown", pfnMFShutdown);
476 sys->MFCreateSample = GetProcAddressAs(sys->mfplat, "MFCreateSample", pfnMFCreateSample);
477 sys->MFCreateMemoryBuffer =
478 GetProcAddressAs(sys->mfplat, "MFCreateMemoryBuffer", pfnMFCreateMemoryBuffer);
479 sys->MFCreateMediaType =
480 GetProcAddressAs(sys->mfplat, "MFCreateMediaType", pfnMFCreateMediaType);
481
482 if (!mf_plat_loaded(sys))
483 goto error;
484
485 CoInitializeEx(nullptr, COINIT_APARTMENTTHREADED);
486
487 if (h264->Compressor)
488 {
489 }
490 else
491 {
492 VARIANT var = WINPR_C_ARRAY_INIT;
493 hr = sys->MFStartup(MF_VERSION, 0);
494
495 if (FAILED(hr))
496 {
497 WLog_Print(h264->log, WLOG_ERROR, "MFStartup failure: 0x%08" PRIX32 "", hr);
498 goto error;
499 }
500
501 hr = CoCreateInstance(&sCLSID_CMSH264DecoderMFT, nullptr, CLSCTX_INPROC_SERVER,
502 &sIID_IMFTransform, (void**)&sys->transform);
503
504 if (FAILED(hr))
505 {
506 WLog_Print(h264->log, WLOG_ERROR,
507 "CoCreateInstance(CLSID_CMSH264DecoderMFT) failure: 0x%08" PRIX32 "", hr);
508 goto error;
509 }
510
511 hr = sys->transform->lpVtbl->QueryInterface(sys->transform, &sIID_ICodecAPI,
512 (void**)&sys->codecApi);
513
514 if (FAILED(hr))
515 {
516 WLog_Print(h264->log, WLOG_ERROR,
517 "QueryInterface(IID_ICodecAPI) failure: 0x%08" PRIX32 "", hr);
518 goto error;
519 }
520
521 var.n1.n2.vt = VT_UI4;
522 var.n1.n2.n3.ulVal = 1;
523 hr = sys->codecApi->lpVtbl->SetValue(sys->codecApi, &sCODECAPI_AVLowLatencyMode, &var);
524
525 if (FAILED(hr))
526 {
527 WLog_Print(h264->log, WLOG_ERROR,
528 "SetValue(CODECAPI_AVLowLatencyMode) failure: 0x%08" PRIX32 "", hr);
529 goto error;
530 }
531
532 hr = sys->MFCreateMediaType(&sys->inputType);
533
534 if (FAILED(hr))
535 {
536 WLog_Print(h264->log, WLOG_ERROR, "MFCreateMediaType failure: 0x%08" PRIX32 "", hr);
537 goto error;
538 }
539
540 hr = sys->inputType->lpVtbl->SetGUID(sys->inputType, &sMF_MT_MAJOR_TYPE,
541 &sMFMediaType_Video);
542
543 if (FAILED(hr))
544 {
545 WLog_Print(h264->log, WLOG_ERROR, "SetGUID(MF_MT_MAJOR_TYPE) failure: 0x%08" PRIX32 "",
546 hr);
547 goto error;
548 }
549
550 hr = sys->inputType->lpVtbl->SetGUID(sys->inputType, &sMF_MT_SUBTYPE, &sMFVideoFormat_H264);
551
552 if (FAILED(hr))
553 {
554 WLog_Print(h264->log, WLOG_ERROR, "SetGUID(MF_MT_SUBTYPE) failure: 0x%08" PRIX32 "",
555 hr);
556 goto error;
557 }
558
559 hr = sys->transform->lpVtbl->SetInputType(sys->transform, 0, sys->inputType, 0);
560
561 if (FAILED(hr))
562 {
563 WLog_Print(h264->log, WLOG_ERROR, "SetInputType failure: 0x%08" PRIX32 "", hr);
564 goto error;
565 }
566
567 hr = mf_find_output_type(sys, &sMFVideoFormat_IYUV, &sys->outputType);
568
569 if (FAILED(hr))
570 {
571 WLog_Print(h264->log, WLOG_ERROR, "mf_find_output_type failure: 0x%08" PRIX32 "", hr);
572 goto error;
573 }
574
575 hr = sys->transform->lpVtbl->SetOutputType(sys->transform, 0, sys->outputType, 0);
576
577 if (FAILED(hr))
578 {
579 WLog_Print(h264->log, WLOG_ERROR, "SetOutputType failure: 0x%08" PRIX32 "", hr);
580 goto error;
581 }
582
583 hr = mf_create_output_sample(h264, sys);
584
585 if (FAILED(hr))
586 {
587 WLog_Print(h264->log, WLOG_ERROR, "mf_create_output_sample failure: 0x%08" PRIX32 "",
588 hr);
589 goto error;
590 }
591 }
592
593 return TRUE;
594error:
595 WLog_Print(h264->log, WLOG_ERROR, "mf_init failure");
596 mf_uninit(h264);
597 return FALSE;
598}
599
600const H264_CONTEXT_SUBSYSTEM g_Subsystem_MF = { "MediaFoundation", mf_init, mf_uninit,
601 mf_decompress, mf_compress };