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, line);
198static int mf_decompress(H264_CONTEXT* WINPR_RESTRICT h264,
const BYTE* WINPR_RESTRICT pSrcData,
201 BYTE* pbBuffer =
nullptr;
202 DWORD cbMaxLength = 0;
203 DWORD cbCurrentLength = 0;
204 DWORD outputStatus = 0;
205 IMFSample* inputSample =
nullptr;
206 IMFMediaBuffer* inputBuffer =
nullptr;
207 IMFMediaBuffer* outputBuffer =
nullptr;
208 MFT_OUTPUT_DATA_BUFFER outputDataBuffer = WINPR_C_ARRAY_INIT;
212 H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*)h264->pSystemData;
213 UINT32* iStride = h264->iStride;
214 BYTE** pYUVData = h264->pYUVData;
215 HRESULT hr = sys->MFCreateMemoryBuffer(SrcSize, &inputBuffer);
219 WLog_Print(h264->log, WLOG_ERROR,
"MFCreateMemoryBuffer failure: 0x%08" PRIX32
"", hr);
223 hr = inputBuffer->lpVtbl->Lock(inputBuffer, &pbBuffer, &cbMaxLength, &cbCurrentLength);
227 WLog_Print(h264->log, WLOG_ERROR,
"Lock failure: 0x%08" PRIX32
"", hr);
231 CopyMemory(pbBuffer, pSrcData, SrcSize);
232 hr = inputBuffer->lpVtbl->SetCurrentLength(inputBuffer, SrcSize);
236 WLog_Print(h264->log, WLOG_ERROR,
"SetCurrentLength failure: 0x%08" PRIX32
"", hr);
240 hr = inputBuffer->lpVtbl->Unlock(inputBuffer);
244 WLog_Print(h264->log, WLOG_ERROR,
"Unlock failure: 0x%08" PRIX32
"", hr);
248 hr = sys->MFCreateSample(&inputSample);
252 WLog_Print(h264->log, WLOG_ERROR,
"MFCreateSample failure: 0x%08" PRIX32
"", hr);
256 hr = inputSample->lpVtbl->AddBuffer(inputSample, inputBuffer);
260 WLog_Print(h264->log, WLOG_ERROR,
"AddBuffer failure: 0x%08" PRIX32
"", hr);
264 inputBuffer->lpVtbl->Release(inputBuffer);
265 hr = sys->transform->lpVtbl->ProcessInput(sys->transform, 0, inputSample, 0);
269 WLog_Print(h264->log, WLOG_ERROR,
"ProcessInput failure: 0x%08" PRIX32
"", hr);
273 hr = mf_create_output_sample(h264, sys);
277 WLog_Print(h264->log, WLOG_ERROR,
"mf_create_output_sample failure: 0x%08" PRIX32
"", hr);
281 outputDataBuffer.dwStreamID = 0;
282 outputDataBuffer.dwStatus = 0;
283 outputDataBuffer.pEvents =
nullptr;
284 outputDataBuffer.pSample = sys->outputSample;
285 hr = sys->transform->lpVtbl->ProcessOutput(sys->transform, 0, 1, &outputDataBuffer,
288 if (hr == MF_E_TRANSFORM_STREAM_CHANGE)
290 UINT64 frameSize = 0;
294 sys->outputType->lpVtbl->Release(sys->outputType);
295 sys->outputType =
nullptr;
298 hr = mf_find_output_type(sys, &sMFVideoFormat_IYUV, &sys->outputType);
302 WLog_Print(h264->log, WLOG_ERROR,
"mf_find_output_type failure: 0x%08" PRIX32
"", hr);
306 hr = sys->transform->lpVtbl->SetOutputType(sys->transform, 0, sys->outputType, 0);
310 WLog_Print(h264->log, WLOG_ERROR,
"SetOutputType failure: 0x%08" PRIX32
"", hr);
314 hr = mf_create_output_sample(h264, sys);
318 WLog_Print(h264->log, WLOG_ERROR,
"mf_create_output_sample failure: 0x%08" PRIX32
"",
323 hr = sys->outputType->lpVtbl->GetUINT64(sys->outputType, &sMF_MT_FRAME_SIZE, &frameSize);
327 WLog_Print(h264->log, WLOG_ERROR,
328 "GetUINT64(MF_MT_FRAME_SIZE) failure: 0x%08" PRIX32
"", hr);
332 sys->frameWidth = (UINT32)(frameSize >> 32);
333 sys->frameHeight = (UINT32)frameSize;
334 hr = sys->outputType->lpVtbl->GetUINT32(sys->outputType, &sMF_MT_DEFAULT_STRIDE,
339 WLog_Print(h264->log, WLOG_ERROR,
340 "GetUINT32(MF_MT_DEFAULT_STRIDE) failure: 0x%08" PRIX32
"", hr);
344 const INT32 istride = (INT32)sys->stride;
347 sys->bottomUp = TRUE;
348 sys->stride = istride * -1;
351 if (!avc420_ensure_buffer(h264, sys->stride, sys->frameWidth, sys->frameHeight))
354 else if (hr == MF_E_TRANSFORM_NEED_MORE_INPUT)
359 WLog_Print(h264->log, WLOG_ERROR,
"ProcessOutput failure: 0x%08" PRIX32
"", hr);
365 BYTE* buffer =
nullptr;
366 DWORD bufferCount = 0;
367 DWORD cbMaxLength = 0;
368 DWORD cbCurrentLength = 0;
369 hr = sys->outputSample->lpVtbl->GetBufferCount(sys->outputSample, &bufferCount);
373 WLog_Print(h264->log, WLOG_ERROR,
"GetBufferCount failure: 0x%08" PRIX32
"", hr);
377 hr = sys->outputSample->lpVtbl->GetBufferByIndex(sys->outputSample, 0, &outputBuffer);
381 WLog_Print(h264->log, WLOG_ERROR,
"GetBufferByIndex failure: 0x%08" PRIX32
"", hr);
385 hr = outputBuffer->lpVtbl->Lock(outputBuffer, &buffer, &cbMaxLength, &cbCurrentLength);
389 WLog_Print(h264->log, WLOG_ERROR,
"Lock failure: 0x%08" PRIX32
"", hr);
397 for (UINT32 x = 0; x < sys->frameHeight; x++)
399 const srcOffset = getSrcOffset(sys, x, FALSE);
400 const dstOffset = iStride[0] * x;
401 CopyMemory(&pYUVData[0][dstOffset], &buffer[srcOffset], sys->stride);
403 for (UINT32 x = 0; x < sys->frameHeight / 2; x++)
405 const srcOffset = getSrcOffset(sys, x, TRUE);
406 const dstUOffset = iStride[1] * x;
407 const dstVOffset = iStride[2] * x;
408 CopyMemory(&pYUVData[1][dstUOffset], &buffer[srcOffset], sys->stride / 2u);
409 CopyMemory(&pYUVData[2][dstVOffset], &buffer[srcOffset], sys->stride / 2u);
412 hr = outputBuffer->lpVtbl->Unlock(outputBuffer);
416 WLog_Print(h264->log, WLOG_ERROR,
"Unlock failure: 0x%08" PRIX32
"", hr);
420 outputBuffer->lpVtbl->Release(outputBuffer);
421 h264->YUVWidth = sys->frameWidth;
422 h264->YUVHeight = sys->frameHeight;
425 inputSample->lpVtbl->Release(inputSample);
428 (void)fprintf(stderr,
"mf_decompress error\n");
432static int mf_compress(H264_CONTEXT* WINPR_RESTRICT h264,
const BYTE** WINPR_RESTRICT ppSrcYuv,
433 const UINT32* WINPR_RESTRICT pStride, BYTE** WINPR_RESTRICT ppDstData,
434 UINT32* WINPR_RESTRICT pDstSize)
436 H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*)h264->pSystemData;
440static BOOL mf_plat_loaded(H264_CONTEXT_MF* sys)
442 return sys->MFStartup && sys->MFShutdown && sys->MFCreateSample && sys->MFCreateMemoryBuffer &&
443 sys->MFCreateMediaType;
446static void mf_uninit(H264_CONTEXT* h264)
448 H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*)h264->pSystemData;
454 sys->transform->lpVtbl->Release(sys->transform);
455 sys->transform =
nullptr;
460 sys->codecApi->lpVtbl->Release(sys->codecApi);
461 sys->codecApi =
nullptr;
466 sys->inputType->lpVtbl->Release(sys->inputType);
467 sys->inputType =
nullptr;
472 sys->outputType->lpVtbl->Release(sys->outputType);
473 sys->outputType =
nullptr;
476 if (sys->outputSample)
478 sys->outputSample->lpVtbl->Release(sys->outputSample);
479 sys->outputSample =
nullptr;
484 if (mf_plat_loaded(sys))
487 FreeLibrary(sys->mfplat);
488 sys->mfplat =
nullptr;
490 if (mf_plat_loaded(sys))
494 for (
size_t x = 0; x <
sizeof(h264->pYUVData) /
sizeof(h264->pYUVData[0]); x++)
495 winpr_aligned_free(h264->pYUVData[x]);
497 memset(h264->pYUVData, 0,
sizeof(h264->pYUVData));
498 memset(h264->iStride, 0,
sizeof(h264->iStride));
501 h264->pSystemData =
nullptr;
505static BOOL mf_init(H264_CONTEXT* h264)
508 H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*)calloc(1,
sizeof(H264_CONTEXT_MF));
513 h264->pSystemData = (
void*)sys;
516 sys->mfplat = LoadLibraryA(
"mfplat.dll");
521 sys->MFStartup = GetProcAddressAs(sys->mfplat,
"MFStartup", pfnMFStartup);
522 sys->MFShutdown = GetProcAddressAs(sys->mfplat,
"MFShutdown", pfnMFShutdown);
523 sys->MFCreateSample = GetProcAddressAs(sys->mfplat,
"MFCreateSample", pfnMFCreateSample);
524 sys->MFCreateMemoryBuffer =
525 GetProcAddressAs(sys->mfplat,
"MFCreateMemoryBuffer", pfnMFCreateMemoryBuffer);
526 sys->MFCreateMediaType =
527 GetProcAddressAs(sys->mfplat,
"MFCreateMediaType", pfnMFCreateMediaType);
529 if (!mf_plat_loaded(sys))
532 CoInitializeEx(
nullptr, COINIT_APARTMENTTHREADED);
534 if (h264->Compressor)
539 VARIANT var = WINPR_C_ARRAY_INIT;
540 hr = sys->MFStartup(MF_VERSION, 0);
544 WLog_Print(h264->log, WLOG_ERROR,
"MFStartup failure: 0x%08" PRIX32
"", hr);
548 hr = CoCreateInstance(&sCLSID_CMSH264DecoderMFT,
nullptr, CLSCTX_INPROC_SERVER,
549 &sIID_IMFTransform, (
void**)&sys->transform);
553 WLog_Print(h264->log, WLOG_ERROR,
554 "CoCreateInstance(CLSID_CMSH264DecoderMFT) failure: 0x%08" PRIX32
"", hr);
558 hr = sys->transform->lpVtbl->QueryInterface(sys->transform, &sIID_ICodecAPI,
559 (
void**)&sys->codecApi);
563 WLog_Print(h264->log, WLOG_ERROR,
564 "QueryInterface(IID_ICodecAPI) failure: 0x%08" PRIX32
"", hr);
568 var.n1.n2.vt = VT_UI4;
569 var.n1.n2.n3.ulVal = 1;
570 hr = sys->codecApi->lpVtbl->SetValue(sys->codecApi, &sCODECAPI_AVLowLatencyMode, &var);
574 WLog_Print(h264->log, WLOG_ERROR,
575 "SetValue(CODECAPI_AVLowLatencyMode) failure: 0x%08" PRIX32
"", hr);
579 hr = sys->MFCreateMediaType(&sys->inputType);
583 WLog_Print(h264->log, WLOG_ERROR,
"MFCreateMediaType failure: 0x%08" PRIX32
"", hr);
587 hr = sys->inputType->lpVtbl->SetGUID(sys->inputType, &sMF_MT_MAJOR_TYPE,
588 &sMFMediaType_Video);
592 WLog_Print(h264->log, WLOG_ERROR,
"SetGUID(MF_MT_MAJOR_TYPE) failure: 0x%08" PRIX32
"",
597 hr = sys->inputType->lpVtbl->SetGUID(sys->inputType, &sMF_MT_SUBTYPE, &sMFVideoFormat_H264);
601 WLog_Print(h264->log, WLOG_ERROR,
"SetGUID(MF_MT_SUBTYPE) failure: 0x%08" PRIX32
"",
606 hr = sys->transform->lpVtbl->SetInputType(sys->transform, 0, sys->inputType, 0);
610 WLog_Print(h264->log, WLOG_ERROR,
"SetInputType failure: 0x%08" PRIX32
"", hr);
614 hr = mf_find_output_type(sys, &sMFVideoFormat_IYUV, &sys->outputType);
618 WLog_Print(h264->log, WLOG_ERROR,
"mf_find_output_type failure: 0x%08" PRIX32
"", hr);
622 hr = sys->transform->lpVtbl->SetOutputType(sys->transform, 0, sys->outputType, 0);
626 WLog_Print(h264->log, WLOG_ERROR,
"SetOutputType failure: 0x%08" PRIX32
"", hr);
630 hr = mf_create_output_sample(h264, sys);
634 WLog_Print(h264->log, WLOG_ERROR,
"mf_create_output_sample failure: 0x%08" PRIX32
"",
642 WLog_Print(h264->log, WLOG_ERROR,
"mf_init failure");
647const H264_CONTEXT_SUBSYSTEM g_Subsystem_MF = {
"MediaFoundation", mf_init, mf_uninit,
648 mf_decompress, mf_compress };