20#include <winpr/winpr.h>
21#include <winpr/library.h>
23#include <freerdp/log.h>
24#include <freerdp/codec/h264.h>
31#include <wmcodecdsp.h>
32#include <mftransform.h>
36#define TAG FREERDP_TAG("codec")
38static const GUID sCLSID_CMSH264DecoderMFT = {
39 0x62CE7E72, 0x4C71, 0x4d20, { 0xB1, 0x5D, 0x45, 0x28, 0x31, 0xA8, 0x7D, 0x9D }
41static const GUID sIID_IMFTransform = {
42 0xbf94c121, 0x5b05, 0x4e6f, { 0x80, 0x00, 0xba, 0x59, 0x89, 0x61, 0x41, 0x4d }
44static const GUID sMF_MT_MAJOR_TYPE = {
45 0x48eba18e, 0xf8c9, 0x4687, { 0xbf, 0x11, 0x0a, 0x74, 0xc9, 0xf9, 0x6a, 0x8f }
47static const GUID sMF_MT_FRAME_SIZE = {
48 0x1652c33d, 0xd6b2, 0x4012, { 0xb8, 0x34, 0x72, 0x03, 0x08, 0x49, 0xa3, 0x7d }
50static const GUID sMF_MT_DEFAULT_STRIDE = {
51 0x644b4e48, 0x1e02, 0x4516, { 0xb0, 0xeb, 0xc0, 0x1c, 0xa9, 0xd4, 0x9a, 0xc6 }
53static const GUID sMF_MT_SUBTYPE = {
54 0xf7e34c9a, 0x42e8, 0x4714, { 0xb7, 0x4b, 0xcb, 0x29, 0xd7, 0x2c, 0x35, 0xe5 }
56static const GUID sMFMediaType_Video = {
57 0x73646976, 0x0000, 0x0010, { 0x80, 0x00, 0x00, 0xAA, 0x00, 0x38, 0x9B, 0x71 }
59static const GUID sMFVideoFormat_H264 = {
60 0x34363248, 0x0000, 0x0010, { 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71 }
62static const GUID sMFVideoFormat_IYUV = {
63 0x56555949, 0x0000, 0x0010, { 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71 }
65static const GUID sIID_ICodecAPI = {
66 0x901db4c7, 0x31ce, 0x41a2, { 0x85, 0xdc, 0x8f, 0xa0, 0xbf, 0x41, 0xb8, 0xda }
68static const GUID sCODECAPI_AVLowLatencyMode = {
69 0x9c27891a, 0xed7a, 0x40e1, { 0x88, 0xe8, 0xb2, 0x27, 0x27, 0xa0, 0x24, 0xee }
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);
81 IMFTransform* transform;
82 IMFMediaType* inputType;
83 IMFMediaType* outputType;
87 IMFSample* outputSample;
88 IMFMediaBuffer* outputBuffer;
90 pfnMFStartup MFStartup;
91 pfnMFShutdown MFShutdown;
92 pfnMFCreateSample MFCreateSample;
93 pfnMFCreateMemoryBuffer MFCreateMemoryBuffer;
94 pfnMFCreateMediaType MFCreateMediaType;
99static HRESULT mf_find_output_type(H264_CONTEXT_MF* sys,
const GUID* guid,
100 IMFMediaType** ppMediaType)
105 IMFMediaType* pMediaType =
nullptr;
109 hr = sys->transform->lpVtbl->GetOutputAvailableType(sys->transform, 0, idx, &pMediaType);
114 pMediaType->lpVtbl->GetGUID(pMediaType, &sMF_MT_SUBTYPE, &mediaGuid);
116 if (IsEqualGUID(&mediaGuid, guid))
118 *ppMediaType = pMediaType;
122 pMediaType->lpVtbl->Release(pMediaType);
129static HRESULT mf_create_output_sample(H264_CONTEXT* h264, H264_CONTEXT_MF* sys)
132 MFT_OUTPUT_STREAM_INFO streamInfo;
134 if (sys->outputSample)
136 sys->outputSample->lpVtbl->Release(sys->outputSample);
137 sys->outputSample =
nullptr;
140 hr = sys->MFCreateSample(&sys->outputSample);
144 WLog_Print(h264->log, WLOG_ERROR,
"MFCreateSample failure: 0x%08" PRIX32
"", hr);
148 hr = sys->transform->lpVtbl->GetOutputStreamInfo(sys->transform, 0, &streamInfo);
152 WLog_Print(h264->log, WLOG_ERROR,
"GetOutputStreamInfo failure: 0x%08" PRIX32
"", hr);
156 hr = sys->MFCreateMemoryBuffer(streamInfo.cbSize, &sys->outputBuffer);
160 WLog_Print(h264->log, WLOG_ERROR,
"MFCreateMemoryBuffer failure: 0x%08" PRIX32
"", hr);
164 hr = sys->outputSample->lpVtbl->AddBuffer(sys->outputSample, sys->outputBuffer);
168 WLog_Print(h264->log, WLOG_ERROR,
"AddBuffer failure: 0x%08" PRIX32
"", hr);
172 sys->outputBuffer->lpVtbl->Release(sys->outputBuffer);
178static UINT32 getSrcOffset(H264_CONTEXT_MF* sys, UINT32 line, BOOL isUV)
182 if (sys->frameHeight < 1)
184 if (line >= sys->frameHeight)
187 UINT64 offset = (sys->frameHeight - line - 1u);
191 offset *= sys->stride;
194 return WINPR_ASSERTING_INT_CAST(UINT32, offset);
197static int mf_decompress(H264_CONTEXT* WINPR_RESTRICT h264,
const BYTE* WINPR_RESTRICT pSrcData,
200 BYTE* pbBuffer =
nullptr;
201 DWORD cbMaxLength = 0;
202 DWORD cbCurrentLength = 0;
203 DWORD outputStatus = 0;
204 IMFSample* inputSample =
nullptr;
205 IMFMediaBuffer* inputBuffer =
nullptr;
206 IMFMediaBuffer* outputBuffer =
nullptr;
207 MFT_OUTPUT_DATA_BUFFER outputDataBuffer = WINPR_C_ARRAY_INIT;
211 H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*)h264->pSystemData;
212 UINT32* iStride = h264->iStride;
213 BYTE** pYUVData = h264->pYUVData;
214 HRESULT hr = sys->MFCreateMemoryBuffer(SrcSize, &inputBuffer);
218 WLog_Print(h264->log, WLOG_ERROR,
"MFCreateMemoryBuffer failure: 0x%08" PRIX32
"", hr);
222 hr = inputBuffer->lpVtbl->Lock(inputBuffer, &pbBuffer, &cbMaxLength, &cbCurrentLength);
226 WLog_Print(h264->log, WLOG_ERROR,
"Lock failure: 0x%08" PRIX32
"", hr);
230 CopyMemory(pbBuffer, pSrcData, SrcSize);
231 hr = inputBuffer->lpVtbl->SetCurrentLength(inputBuffer, SrcSize);
235 WLog_Print(h264->log, WLOG_ERROR,
"SetCurrentLength failure: 0x%08" PRIX32
"", hr);
239 hr = inputBuffer->lpVtbl->Unlock(inputBuffer);
243 WLog_Print(h264->log, WLOG_ERROR,
"Unlock failure: 0x%08" PRIX32
"", hr);
247 hr = sys->MFCreateSample(&inputSample);
251 WLog_Print(h264->log, WLOG_ERROR,
"MFCreateSample failure: 0x%08" PRIX32
"", hr);
255 hr = inputSample->lpVtbl->AddBuffer(inputSample, inputBuffer);
259 WLog_Print(h264->log, WLOG_ERROR,
"AddBuffer failure: 0x%08" PRIX32
"", hr);
263 inputBuffer->lpVtbl->Release(inputBuffer);
264 hr = sys->transform->lpVtbl->ProcessInput(sys->transform, 0, inputSample, 0);
268 WLog_Print(h264->log, WLOG_ERROR,
"ProcessInput failure: 0x%08" PRIX32
"", hr);
272 hr = mf_create_output_sample(h264, sys);
276 WLog_Print(h264->log, WLOG_ERROR,
"mf_create_output_sample failure: 0x%08" PRIX32
"", hr);
280 outputDataBuffer.dwStreamID = 0;
281 outputDataBuffer.dwStatus = 0;
282 outputDataBuffer.pEvents =
nullptr;
283 outputDataBuffer.pSample = sys->outputSample;
284 hr = sys->transform->lpVtbl->ProcessOutput(sys->transform, 0, 1, &outputDataBuffer,
287 if (hr == MF_E_TRANSFORM_STREAM_CHANGE)
289 UINT64 frameSize = 0;
293 sys->outputType->lpVtbl->Release(sys->outputType);
294 sys->outputType =
nullptr;
297 hr = mf_find_output_type(sys, &sMFVideoFormat_IYUV, &sys->outputType);
301 WLog_Print(h264->log, WLOG_ERROR,
"mf_find_output_type failure: 0x%08" PRIX32
"", hr);
305 hr = sys->transform->lpVtbl->SetOutputType(sys->transform, 0, sys->outputType, 0);
309 WLog_Print(h264->log, WLOG_ERROR,
"SetOutputType failure: 0x%08" PRIX32
"", hr);
313 hr = mf_create_output_sample(h264, sys);
317 WLog_Print(h264->log, WLOG_ERROR,
"mf_create_output_sample failure: 0x%08" PRIX32
"",
322 hr = sys->outputType->lpVtbl->GetUINT64(sys->outputType, &sMF_MT_FRAME_SIZE, &frameSize);
326 WLog_Print(h264->log, WLOG_ERROR,
327 "GetUINT64(MF_MT_FRAME_SIZE) failure: 0x%08" PRIX32
"", hr);
331 sys->frameWidth = (UINT32)(frameSize >> 32);
332 sys->frameHeight = (UINT32)frameSize;
333 hr = sys->outputType->lpVtbl->GetUINT32(sys->outputType, &sMF_MT_DEFAULT_STRIDE,
338 WLog_Print(h264->log, WLOG_ERROR,
339 "GetUINT32(MF_MT_DEFAULT_STRIDE) failure: 0x%08" PRIX32
"", hr);
343 const INT32 istride = (INT32)sys->stride;
346 sys->bottomUp = TRUE;
347 sys->stride = istride * -1;
350 if (!avc420_ensure_buffer(h264, sys->stride, sys->frameWidth, sys->frameHeight))
353 else if (hr == MF_E_TRANSFORM_NEED_MORE_INPUT)
358 WLog_Print(h264->log, WLOG_ERROR,
"ProcessOutput failure: 0x%08" PRIX32
"", hr);
364 BYTE* buffer =
nullptr;
365 DWORD bufferCount = 0;
366 DWORD cbMaxLength = 0;
367 DWORD cbCurrentLength = 0;
368 hr = sys->outputSample->lpVtbl->GetBufferCount(sys->outputSample, &bufferCount);
372 WLog_Print(h264->log, WLOG_ERROR,
"GetBufferCount failure: 0x%08" PRIX32
"", hr);
376 hr = sys->outputSample->lpVtbl->GetBufferByIndex(sys->outputSample, 0, &outputBuffer);
380 WLog_Print(h264->log, WLOG_ERROR,
"GetBufferByIndex failure: 0x%08" PRIX32
"", hr);
384 hr = outputBuffer->lpVtbl->Lock(outputBuffer, &buffer, &cbMaxLength, &cbCurrentLength);
388 WLog_Print(h264->log, WLOG_ERROR,
"Lock failure: 0x%08" PRIX32
"", hr);
396 for (UINT32 x = 0; x < sys->frameHeight; x++)
398 const UINT32 srcOffset = getSrcOffset(sys, x, FALSE);
399 const UINT32 dstOffset = (UINT32)iStride[0] * x;
400 CopyMemory(&pYUVData[0][dstOffset], &buffer[srcOffset], sys->stride);
402 for (UINT32 x = 0; x < sys->frameHeight / 2; x++)
404 const UINT32 srcOffset = getSrcOffset(sys, x, TRUE);
405 const UINT32 dstUOffset = (UINT32)iStride[1] * x;
406 const UINT32 dstVOffset = (UINT32)iStride[2] * x;
407 CopyMemory(&pYUVData[1][dstUOffset], &buffer[srcOffset], sys->stride / 2u);
408 CopyMemory(&pYUVData[2][dstVOffset], &buffer[srcOffset], sys->stride / 2u);
411 hr = outputBuffer->lpVtbl->Unlock(outputBuffer);
415 WLog_Print(h264->log, WLOG_ERROR,
"Unlock failure: 0x%08" PRIX32
"", hr);
419 outputBuffer->lpVtbl->Release(outputBuffer);
420 h264->YUVWidth = sys->frameWidth;
421 h264->YUVHeight = sys->frameHeight;
424 inputSample->lpVtbl->Release(inputSample);
427 (void)fprintf(stderr,
"mf_decompress error\n");
431static int mf_compress(H264_CONTEXT* WINPR_RESTRICT h264,
const BYTE** WINPR_RESTRICT ppSrcYuv,
432 const UINT32* WINPR_RESTRICT pStride, BYTE** WINPR_RESTRICT ppDstData,
433 UINT32* WINPR_RESTRICT pDstSize)
435 H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*)h264->pSystemData;
439static BOOL mf_plat_loaded(H264_CONTEXT_MF* sys)
441 return sys->MFStartup && sys->MFShutdown && sys->MFCreateSample && sys->MFCreateMemoryBuffer &&
442 sys->MFCreateMediaType;
445static void mf_uninit(H264_CONTEXT* h264)
447 H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*)h264->pSystemData;
453 sys->transform->lpVtbl->Release(sys->transform);
454 sys->transform =
nullptr;
459 sys->codecApi->lpVtbl->Release(sys->codecApi);
460 sys->codecApi =
nullptr;
465 sys->inputType->lpVtbl->Release(sys->inputType);
466 sys->inputType =
nullptr;
471 sys->outputType->lpVtbl->Release(sys->outputType);
472 sys->outputType =
nullptr;
475 if (sys->outputSample)
477 sys->outputSample->lpVtbl->Release(sys->outputSample);
478 sys->outputSample =
nullptr;
483 if (mf_plat_loaded(sys))
486 FreeLibrary(sys->mfplat);
487 sys->mfplat =
nullptr;
489 if (mf_plat_loaded(sys))
493 for (
size_t x = 0; x <
sizeof(h264->pYUVData) /
sizeof(h264->pYUVData[0]); x++)
494 winpr_aligned_free(h264->pYUVData[x]);
496 memset(h264->pYUVData, 0,
sizeof(h264->pYUVData));
497 memset(h264->iStride, 0,
sizeof(h264->iStride));
500 h264->pSystemData =
nullptr;
504static BOOL mf_init(H264_CONTEXT* h264)
507 H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*)calloc(1,
sizeof(H264_CONTEXT_MF));
512 h264->pSystemData = (
void*)sys;
515 sys->mfplat = LoadLibraryA(
"mfplat.dll");
520 sys->MFStartup = GetProcAddressAs(sys->mfplat,
"MFStartup", pfnMFStartup);
521 sys->MFShutdown = GetProcAddressAs(sys->mfplat,
"MFShutdown", pfnMFShutdown);
522 sys->MFCreateSample = GetProcAddressAs(sys->mfplat,
"MFCreateSample", pfnMFCreateSample);
523 sys->MFCreateMemoryBuffer =
524 GetProcAddressAs(sys->mfplat,
"MFCreateMemoryBuffer", pfnMFCreateMemoryBuffer);
525 sys->MFCreateMediaType =
526 GetProcAddressAs(sys->mfplat,
"MFCreateMediaType", pfnMFCreateMediaType);
528 if (!mf_plat_loaded(sys))
531 CoInitializeEx(
nullptr, COINIT_APARTMENTTHREADED);
533 if (h264->Compressor)
538 VARIANT var = WINPR_C_ARRAY_INIT;
539 hr = sys->MFStartup(MF_VERSION, 0);
543 WLog_Print(h264->log, WLOG_ERROR,
"MFStartup failure: 0x%08" PRIX32
"", hr);
547 hr = CoCreateInstance(&sCLSID_CMSH264DecoderMFT,
nullptr, CLSCTX_INPROC_SERVER,
548 &sIID_IMFTransform, (
void**)&sys->transform);
552 WLog_Print(h264->log, WLOG_ERROR,
553 "CoCreateInstance(CLSID_CMSH264DecoderMFT) failure: 0x%08" PRIX32
"", hr);
557 hr = sys->transform->lpVtbl->QueryInterface(sys->transform, &sIID_ICodecAPI,
558 (
void**)&sys->codecApi);
562 WLog_Print(h264->log, WLOG_ERROR,
563 "QueryInterface(IID_ICodecAPI) failure: 0x%08" PRIX32
"", hr);
567 var.n1.n2.vt = VT_UI4;
568 var.n1.n2.n3.ulVal = 1;
569 hr = sys->codecApi->lpVtbl->SetValue(sys->codecApi, &sCODECAPI_AVLowLatencyMode, &var);
573 WLog_Print(h264->log, WLOG_ERROR,
574 "SetValue(CODECAPI_AVLowLatencyMode) failure: 0x%08" PRIX32
"", hr);
578 hr = sys->MFCreateMediaType(&sys->inputType);
582 WLog_Print(h264->log, WLOG_ERROR,
"MFCreateMediaType failure: 0x%08" PRIX32
"", hr);
586 hr = sys->inputType->lpVtbl->SetGUID(sys->inputType, &sMF_MT_MAJOR_TYPE,
587 &sMFMediaType_Video);
591 WLog_Print(h264->log, WLOG_ERROR,
"SetGUID(MF_MT_MAJOR_TYPE) failure: 0x%08" PRIX32
"",
596 hr = sys->inputType->lpVtbl->SetGUID(sys->inputType, &sMF_MT_SUBTYPE, &sMFVideoFormat_H264);
600 WLog_Print(h264->log, WLOG_ERROR,
"SetGUID(MF_MT_SUBTYPE) failure: 0x%08" PRIX32
"",
605 hr = sys->transform->lpVtbl->SetInputType(sys->transform, 0, sys->inputType, 0);
609 WLog_Print(h264->log, WLOG_ERROR,
"SetInputType failure: 0x%08" PRIX32
"", hr);
613 hr = mf_find_output_type(sys, &sMFVideoFormat_IYUV, &sys->outputType);
617 WLog_Print(h264->log, WLOG_ERROR,
"mf_find_output_type failure: 0x%08" PRIX32
"", hr);
621 hr = sys->transform->lpVtbl->SetOutputType(sys->transform, 0, sys->outputType, 0);
625 WLog_Print(h264->log, WLOG_ERROR,
"SetOutputType failure: 0x%08" PRIX32
"", hr);
629 hr = mf_create_output_sample(h264, sys);
633 WLog_Print(h264->log, WLOG_ERROR,
"mf_create_output_sample failure: 0x%08" PRIX32
"",
641 WLog_Print(h264->log, WLOG_ERROR,
"mf_init failure");
646const H264_CONTEXT_SUBSYSTEM g_Subsystem_MF = {
"MediaFoundation", mf_init, mf_uninit,
647 mf_decompress, mf_compress };