mirror of
https://github.com/morgan9e/FreeRDP
synced 2026-04-15 00:44:19 +09:00
Split h264 subsystems in different files
This commit is contained in:
@@ -165,23 +165,30 @@ if(WITH_JPEG)
|
||||
endif()
|
||||
|
||||
if(WITH_X264)
|
||||
set(CODEC_SRCS ${CODEC_SRCS} codec/h264_x264.c)
|
||||
freerdp_definition_add(-DWITH_X264)
|
||||
freerdp_include_directory_add(${X264_INCLUDE_DIR})
|
||||
freerdp_library_add(${X264_LIBRARIES})
|
||||
endif()
|
||||
|
||||
if(WITH_OPENH264)
|
||||
set(CODEC_SRCS ${CODEC_SRCS} codec/h264_openh264.c)
|
||||
freerdp_definition_add(-DWITH_OPENH264)
|
||||
freerdp_include_directory_add(${OPENH264_INCLUDE_DIR})
|
||||
freerdp_library_add(${OPENH264_LIBRARIES})
|
||||
endif()
|
||||
|
||||
if(WITH_FFMPEG)
|
||||
set(CODEC_SRCS ${CODEC_SRCS} codec/h264_ffmpeg.c)
|
||||
freerdp_definition_add(-DWITH_LIBAVCODEC)
|
||||
freerdp_include_directory_add(${FFMPEG_INCLUDE_DIRS})
|
||||
freerdp_library_add(${FFMPEG_LIBRARIES})
|
||||
endif()
|
||||
|
||||
if(WIN32 AND WITH_MEDIA_FOUNDATION)
|
||||
set(CODEC_SRCS ${CODEC_SRCS} codec/h264_mf.c)
|
||||
endif()
|
||||
|
||||
freerdp_module_add(${CODEC_SRCS})
|
||||
|
||||
if(BUILD_TESTING)
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
221
libfreerdp/codec/h264_ffmpeg.c
Normal file
221
libfreerdp/codec/h264_ffmpeg.c
Normal file
@@ -0,0 +1,221 @@
|
||||
/**
|
||||
* FreeRDP: A Remote Desktop Protocol Implementation
|
||||
* H.264 Bitmap Compression
|
||||
*
|
||||
* Copyright 2015 Marc-André Moreau <marcandre.moreau@gmail.com>
|
||||
* Copyright 2014 Mike McDonald <Mike.McDonald@software.dell.com>
|
||||
* Copyright 2014 erbth <t.erbesdobler@team103.com>
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include <winpr/wlog.h>
|
||||
#include <freerdp/log.h>
|
||||
#include <freerdp/codec/h264.h>
|
||||
#include <libavcodec/avcodec.h>
|
||||
#include <libavutil/avutil.h>
|
||||
|
||||
#define TAG FREERDP_TAG("codec")
|
||||
|
||||
/* Fallback support for older libavcodec versions */
|
||||
#if LIBAVCODEC_VERSION_INT < AV_VERSION_INT(54, 59, 100)
|
||||
#define AV_CODEC_ID_H264 CODEC_ID_H264
|
||||
#endif
|
||||
|
||||
struct _H264_CONTEXT_LIBAVCODEC
|
||||
{
|
||||
AVCodec* codec;
|
||||
AVCodecContext* codecContext;
|
||||
AVCodecParserContext* codecParser;
|
||||
AVFrame* videoFrame;
|
||||
};
|
||||
typedef struct _H264_CONTEXT_LIBAVCODEC H264_CONTEXT_LIBAVCODEC;
|
||||
|
||||
static int libavcodec_decompress(H264_CONTEXT* h264, const BYTE* pSrcData,
|
||||
UINT32 SrcSize)
|
||||
{
|
||||
int status;
|
||||
int gotFrame = 0;
|
||||
AVPacket packet;
|
||||
H264_CONTEXT_LIBAVCODEC* sys = (H264_CONTEXT_LIBAVCODEC*) h264->pSystemData;
|
||||
BYTE** pYUVData = h264->pYUVData;
|
||||
UINT32* iStride = h264->iStride;
|
||||
av_init_packet(&packet);
|
||||
packet.data = (BYTE*)pSrcData;
|
||||
packet.size = SrcSize;
|
||||
/* avcodec_decode_video2 is deprecated with libavcodec 57.48.101 */
|
||||
#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT(57, 48, 101)
|
||||
status = avcodec_send_packet(sys->codecContext, &packet);
|
||||
|
||||
if (status < 0)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to decode video frame (status=%d)", status);
|
||||
return -1;
|
||||
}
|
||||
|
||||
do
|
||||
{
|
||||
status = avcodec_receive_frame(sys->codecContext, sys->videoFrame);
|
||||
}
|
||||
while (status == AVERROR(EAGAIN));
|
||||
|
||||
gotFrame = (status == 0);
|
||||
#else
|
||||
status = avcodec_decode_video2(sys->codecContext, sys->videoFrame, &gotFrame,
|
||||
&packet);
|
||||
#endif
|
||||
|
||||
if (status < 0)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to decode video frame (status=%d)", status);
|
||||
return -1;
|
||||
}
|
||||
|
||||
#if 0
|
||||
WLog_INFO(TAG,
|
||||
"libavcodec_decompress: frame decoded (status=%d, gotFrame=%d, width=%d, height=%d, Y=[%p,%d], U=[%p,%d], V=[%p,%d])",
|
||||
status, gotFrame, sys->videoFrame->width, sys->videoFrame->height,
|
||||
(void*) sys->videoFrame->data[0], sys->videoFrame->linesize[0],
|
||||
(void*) sys->videoFrame->data[1], sys->videoFrame->linesize[1],
|
||||
(void*) sys->videoFrame->data[2], sys->videoFrame->linesize[2]);
|
||||
#endif
|
||||
|
||||
if (gotFrame)
|
||||
{
|
||||
pYUVData[0] = sys->videoFrame->data[0];
|
||||
pYUVData[1] = sys->videoFrame->data[1];
|
||||
pYUVData[2] = sys->videoFrame->data[2];
|
||||
iStride[0] = sys->videoFrame->linesize[0];
|
||||
iStride[1] = sys->videoFrame->linesize[1];
|
||||
iStride[2] = sys->videoFrame->linesize[2];
|
||||
h264->width = sys->videoFrame->width;
|
||||
h264->height = sys->videoFrame->height;
|
||||
}
|
||||
else
|
||||
return -2;
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
static int libavcodec_compress(H264_CONTEXT* h264, BYTE** ppDstData, UINT32* pDstSize)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
|
||||
static void libavcodec_uninit(H264_CONTEXT* h264)
|
||||
{
|
||||
H264_CONTEXT_LIBAVCODEC* sys = (H264_CONTEXT_LIBAVCODEC*) h264->pSystemData;
|
||||
|
||||
if (!sys)
|
||||
return;
|
||||
|
||||
if (sys->videoFrame)
|
||||
{
|
||||
#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT(55, 18, 102)
|
||||
av_frame_free(&sys->videoFrame);
|
||||
#else
|
||||
av_free(sys->videoFrame);
|
||||
#endif
|
||||
}
|
||||
|
||||
if (sys->codecParser)
|
||||
{
|
||||
av_parser_close(sys->codecParser);
|
||||
}
|
||||
|
||||
if (sys->codecContext)
|
||||
{
|
||||
avcodec_close(sys->codecContext);
|
||||
#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT(55, 69, 100)
|
||||
avcodec_free_context(&sys->codecContext);
|
||||
#else
|
||||
av_free(sys->codecContext);
|
||||
#endif
|
||||
}
|
||||
|
||||
free(sys);
|
||||
h264->pSystemData = NULL;
|
||||
}
|
||||
|
||||
static BOOL libavcodec_init(H264_CONTEXT* h264)
|
||||
{
|
||||
H264_CONTEXT_LIBAVCODEC* sys;
|
||||
sys = (H264_CONTEXT_LIBAVCODEC*) calloc(1, sizeof(H264_CONTEXT_LIBAVCODEC));
|
||||
|
||||
if (!sys)
|
||||
{
|
||||
goto EXCEPTION;
|
||||
}
|
||||
|
||||
h264->pSystemData = (void*) sys;
|
||||
avcodec_register_all();
|
||||
sys->codec = avcodec_find_decoder(AV_CODEC_ID_H264);
|
||||
|
||||
if (!sys->codec)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to find libav H.264 codec");
|
||||
goto EXCEPTION;
|
||||
}
|
||||
|
||||
sys->codecContext = avcodec_alloc_context3(sys->codec);
|
||||
|
||||
if (!sys->codecContext)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to allocate libav codec context");
|
||||
goto EXCEPTION;
|
||||
}
|
||||
|
||||
if (sys->codec->capabilities & CODEC_CAP_TRUNCATED)
|
||||
{
|
||||
sys->codecContext->flags |= CODEC_FLAG_TRUNCATED;
|
||||
}
|
||||
|
||||
if (avcodec_open2(sys->codecContext, sys->codec, NULL) < 0)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to open libav codec");
|
||||
goto EXCEPTION;
|
||||
}
|
||||
|
||||
sys->codecParser = av_parser_init(AV_CODEC_ID_H264);
|
||||
|
||||
if (!sys->codecParser)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to initialize libav parser");
|
||||
goto EXCEPTION;
|
||||
}
|
||||
|
||||
#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT(55, 18, 102)
|
||||
sys->videoFrame = av_frame_alloc();
|
||||
#else
|
||||
sys->videoFrame = avcodec_alloc_frame();
|
||||
#endif
|
||||
|
||||
if (!sys->videoFrame)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to allocate libav frame");
|
||||
goto EXCEPTION;
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
EXCEPTION:
|
||||
libavcodec_uninit(h264);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
H264_CONTEXT_SUBSYSTEM g_Subsystem_libavcodec =
|
||||
{
|
||||
"libavcodec",
|
||||
libavcodec_init,
|
||||
libavcodec_uninit,
|
||||
libavcodec_decompress,
|
||||
libavcodec_compress
|
||||
};
|
||||
641
libfreerdp/codec/h264_mf.c
Normal file
641
libfreerdp/codec/h264_mf.c
Normal file
@@ -0,0 +1,641 @@
|
||||
/**
|
||||
* FreeRDP: A Remote Desktop Protocol Implementation
|
||||
* H.264 Bitmap Compression
|
||||
*
|
||||
* Copyright 2014 Mike McDonald <Mike.McDonald@software.dell.com>
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include <freerdp/codec/h264.h>
|
||||
|
||||
#include <ks.h>
|
||||
#include <codecapi.h>
|
||||
|
||||
#include <mfapi.h>
|
||||
#include <mferror.h>
|
||||
#include <wmcodecdsp.h>
|
||||
#include <mftransform.h>
|
||||
|
||||
#undef DEFINE_GUID
|
||||
#define INITGUID
|
||||
#include <initguid.h>
|
||||
|
||||
DEFINE_GUID(CLSID_CMSH264DecoderMFT, 0x62CE7E72, 0x4C71, 0x4d20, 0xB1, 0x5D,
|
||||
0x45, 0x28, 0x31, 0xA8, 0x7D, 0x9D);
|
||||
DEFINE_GUID(CLSID_VideoProcessorMFT, 0x88753b26, 0x5b24, 0x49bd, 0xb2, 0xe7,
|
||||
0x0c, 0x44, 0x5c, 0x78, 0xc9, 0x82);
|
||||
DEFINE_GUID(IID_IMFTransform, 0xbf94c121, 0x5b05, 0x4e6f, 0x80, 0x00, 0xba,
|
||||
0x59, 0x89, 0x61, 0x41, 0x4d);
|
||||
DEFINE_GUID(MF_MT_MAJOR_TYPE, 0x48eba18e, 0xf8c9, 0x4687, 0xbf, 0x11, 0x0a,
|
||||
0x74, 0xc9, 0xf9, 0x6a, 0x8f);
|
||||
DEFINE_GUID(MF_MT_FRAME_SIZE, 0x1652c33d, 0xd6b2, 0x4012, 0xb8, 0x34, 0x72,
|
||||
0x03, 0x08, 0x49, 0xa3, 0x7d);
|
||||
DEFINE_GUID(MF_MT_DEFAULT_STRIDE, 0x644b4e48, 0x1e02, 0x4516, 0xb0, 0xeb, 0xc0,
|
||||
0x1c, 0xa9, 0xd4, 0x9a, 0xc6);
|
||||
DEFINE_GUID(MF_MT_SUBTYPE, 0xf7e34c9a, 0x42e8, 0x4714, 0xb7, 0x4b, 0xcb, 0x29,
|
||||
0xd7, 0x2c, 0x35, 0xe5);
|
||||
DEFINE_GUID(MF_XVP_DISABLE_FRC, 0x2c0afa19, 0x7a97, 0x4d5a, 0x9e, 0xe8, 0x16,
|
||||
0xd4, 0xfc, 0x51, 0x8d, 0x8c);
|
||||
DEFINE_GUID(MFMediaType_Video, 0x73646976, 0x0000, 0x0010, 0x80, 0x00, 0x00,
|
||||
0xAA, 0x00, 0x38, 0x9B, 0x71);
|
||||
DEFINE_GUID(MFVideoFormat_RGB32, 22, 0x0000, 0x0010, 0x80, 0x00, 0x00, 0xAA,
|
||||
0x00, 0x38, 0x9B, 0x71);
|
||||
DEFINE_GUID(MFVideoFormat_ARGB32, 21, 0x0000, 0x0010, 0x80, 0x00, 0x00, 0xAA,
|
||||
0x00, 0x38, 0x9B, 0x71);
|
||||
DEFINE_GUID(MFVideoFormat_H264, 0x34363248, 0x0000, 0x0010, 0x80, 0x00, 0x00,
|
||||
0xaa, 0x00, 0x38, 0x9b, 0x71);
|
||||
DEFINE_GUID(MFVideoFormat_IYUV, 0x56555949, 0x0000, 0x0010, 0x80, 0x00, 0x00,
|
||||
0xaa, 0x00, 0x38, 0x9b, 0x71);
|
||||
DEFINE_GUID(IID_ICodecAPI, 0x901db4c7, 0x31ce, 0x41a2, 0x85, 0xdc, 0x8f, 0xa0,
|
||||
0xbf, 0x41, 0xb8, 0xda);
|
||||
DEFINE_GUID(CODECAPI_AVLowLatencyMode, 0x9c27891a, 0xed7a, 0x40e1, 0x88, 0xe8,
|
||||
0xb2, 0x27, 0x27, 0xa0, 0x24, 0xee);
|
||||
DEFINE_GUID(CODECAPI_AVDecVideoMaxCodedWidth, 0x5ae557b8, 0x77af, 0x41f5, 0x9f,
|
||||
0xa6, 0x4d, 0xb2, 0xfe, 0x1d, 0x4b, 0xca);
|
||||
DEFINE_GUID(CODECAPI_AVDecVideoMaxCodedHeight, 0x7262a16a, 0xd2dc, 0x4e75, 0x9b,
|
||||
0xa8, 0x65, 0xc0, 0xc6, 0xd3, 0x2b, 0x13);
|
||||
|
||||
#ifndef __IMFDXGIDeviceManager_FWD_DEFINED__
|
||||
#define __IMFDXGIDeviceManager_FWD_DEFINED__
|
||||
typedef interface IMFDXGIDeviceManager IMFDXGIDeviceManager;
|
||||
#endif /* __IMFDXGIDeviceManager_FWD_DEFINED__ */
|
||||
|
||||
#ifndef __IMFDXGIDeviceManager_INTERFACE_DEFINED__
|
||||
#define __IMFDXGIDeviceManager_INTERFACE_DEFINED__
|
||||
|
||||
typedef struct IMFDXGIDeviceManagerVtbl
|
||||
{
|
||||
HRESULT(STDMETHODCALLTYPE* QueryInterface)(IMFDXGIDeviceManager* This,
|
||||
REFIID riid, void** ppvObject);
|
||||
ULONG(STDMETHODCALLTYPE* AddRef)(IMFDXGIDeviceManager* This);
|
||||
ULONG(STDMETHODCALLTYPE* Release)(IMFDXGIDeviceManager* This);
|
||||
HRESULT(STDMETHODCALLTYPE* CloseDeviceHandle)(IMFDXGIDeviceManager* This,
|
||||
HANDLE hDevice);
|
||||
HRESULT(STDMETHODCALLTYPE* GetVideoService)(IMFDXGIDeviceManager* This,
|
||||
HANDLE hDevice, REFIID riid, void** ppService);
|
||||
HRESULT(STDMETHODCALLTYPE* LockDevice)(IMFDXGIDeviceManager* This,
|
||||
HANDLE hDevice, REFIID riid, void** ppUnkDevice, BOOL fBlock);
|
||||
HRESULT(STDMETHODCALLTYPE* OpenDeviceHandle)(IMFDXGIDeviceManager* This,
|
||||
HANDLE* phDevice);
|
||||
HRESULT(STDMETHODCALLTYPE* ResetDevice)(IMFDXGIDeviceManager* This,
|
||||
IUnknown* pUnkDevice, UINT resetToken);
|
||||
HRESULT(STDMETHODCALLTYPE* TestDevice)(IMFDXGIDeviceManager* This,
|
||||
HANDLE hDevice);
|
||||
HRESULT(STDMETHODCALLTYPE* UnlockDevice)(IMFDXGIDeviceManager* This,
|
||||
HANDLE hDevice, BOOL fSaveState);
|
||||
}
|
||||
IMFDXGIDeviceManagerVtbl;
|
||||
|
||||
interface IMFDXGIDeviceManager
|
||||
{
|
||||
CONST_VTBL struct IMFDXGIDeviceManagerVtbl* lpVtbl;
|
||||
};
|
||||
|
||||
#endif /* __IMFDXGIDeviceManager_INTERFACE_DEFINED__ */
|
||||
|
||||
typedef HRESULT(__stdcall* pfnMFStartup)(ULONG Version, DWORD dwFlags);
|
||||
typedef HRESULT(__stdcall* pfnMFShutdown)(void);
|
||||
typedef HRESULT(__stdcall* pfnMFCreateSample)(IMFSample** ppIMFSample);
|
||||
typedef HRESULT(__stdcall* pfnMFCreateMemoryBuffer)(DWORD cbMaxLength,
|
||||
IMFMediaBuffer** ppBuffer);
|
||||
typedef HRESULT(__stdcall* pfnMFCreateMediaType)(IMFMediaType** ppMFType);
|
||||
typedef HRESULT(__stdcall* pfnMFCreateDXGIDeviceManager)(UINT* pResetToken,
|
||||
IMFDXGIDeviceManager** ppDXVAManager);
|
||||
|
||||
struct _H264_CONTEXT_MF
|
||||
{
|
||||
ICodecAPI* codecApi;
|
||||
IMFTransform* transform;
|
||||
IMFMediaType* inputType;
|
||||
IMFMediaType* outputType;
|
||||
IMFSample* sample;
|
||||
UINT32 frameWidth;
|
||||
UINT32 frameHeight;
|
||||
IMFSample* outputSample;
|
||||
IMFMediaBuffer* outputBuffer;
|
||||
HMODULE mfplat;
|
||||
pfnMFStartup MFStartup;
|
||||
pfnMFShutdown MFShutdown;
|
||||
pfnMFCreateSample MFCreateSample;
|
||||
pfnMFCreateMemoryBuffer MFCreateMemoryBuffer;
|
||||
pfnMFCreateMediaType MFCreateMediaType;
|
||||
pfnMFCreateDXGIDeviceManager MFCreateDXGIDeviceManager;
|
||||
};
|
||||
typedef struct _H264_CONTEXT_MF H264_CONTEXT_MF;
|
||||
|
||||
static HRESULT mf_find_output_type(H264_CONTEXT_MF* sys, const GUID* guid,
|
||||
IMFMediaType** ppMediaType)
|
||||
{
|
||||
DWORD idx = 0;
|
||||
GUID mediaGuid;
|
||||
HRESULT hr = S_OK;
|
||||
IMFMediaType* pMediaType = NULL;
|
||||
|
||||
while (1)
|
||||
{
|
||||
hr = sys->transform->lpVtbl->GetOutputAvailableType(sys->transform, 0, idx,
|
||||
&pMediaType);
|
||||
|
||||
if (FAILED(hr))
|
||||
break;
|
||||
|
||||
pMediaType->lpVtbl->GetGUID(pMediaType, &MF_MT_SUBTYPE, &mediaGuid);
|
||||
|
||||
if (IsEqualGUID(&mediaGuid, guid))
|
||||
{
|
||||
*ppMediaType = pMediaType;
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
pMediaType->lpVtbl->Release(pMediaType);
|
||||
idx++;
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
static HRESULT mf_create_output_sample(H264_CONTEXT_MF* sys)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
MFT_OUTPUT_STREAM_INFO streamInfo;
|
||||
|
||||
if (sys->outputSample)
|
||||
{
|
||||
sys->outputSample->lpVtbl->Release(sys->outputSample);
|
||||
sys->outputSample = NULL;
|
||||
}
|
||||
|
||||
hr = sys->MFCreateSample(&sys->outputSample);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "MFCreateSample failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = sys->transform->lpVtbl->GetOutputStreamInfo(sys->transform, 0,
|
||||
&streamInfo);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "GetOutputStreamInfo failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = sys->MFCreateMemoryBuffer(streamInfo.cbSize, &sys->outputBuffer);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "MFCreateMemoryBuffer failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
sys->outputSample->lpVtbl->AddBuffer(sys->outputSample, sys->outputBuffer);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "AddBuffer failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
sys->outputBuffer->lpVtbl->Release(sys->outputBuffer);
|
||||
error:
|
||||
return hr;
|
||||
}
|
||||
|
||||
static int mf_decompress(H264_CONTEXT* h264, const BYTE* pSrcData, UINT32 SrcSize)
|
||||
{
|
||||
HRESULT hr;
|
||||
BYTE* pbBuffer = NULL;
|
||||
DWORD cbMaxLength = 0;
|
||||
DWORD cbCurrentLength = 0;
|
||||
DWORD outputStatus = 0;
|
||||
IMFSample* inputSample = NULL;
|
||||
IMFMediaBuffer* inputBuffer = NULL;
|
||||
IMFMediaBuffer* outputBuffer = NULL;
|
||||
MFT_OUTPUT_DATA_BUFFER outputDataBuffer;
|
||||
H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*) h264->pSystemData;
|
||||
INT32* iStride = h264->iStride;
|
||||
BYTE** pYUVData = h264->pYUVData;
|
||||
hr = sys->MFCreateMemoryBuffer(SrcSize, &inputBuffer);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "MFCreateMemoryBuffer failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = inputBuffer->lpVtbl->Lock(inputBuffer, &pbBuffer, &cbMaxLength,
|
||||
&cbCurrentLength);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "Lock failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
CopyMemory(pbBuffer, pSrcData, SrcSize);
|
||||
hr = inputBuffer->lpVtbl->SetCurrentLength(inputBuffer, SrcSize);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "SetCurrentLength failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = inputBuffer->lpVtbl->Unlock(inputBuffer);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "Unlock failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = sys->MFCreateSample(&inputSample);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "MFCreateSample failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
inputSample->lpVtbl->AddBuffer(inputSample, inputBuffer);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "AddBuffer failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
inputBuffer->lpVtbl->Release(inputBuffer);
|
||||
hr = sys->transform->lpVtbl->ProcessInput(sys->transform, 0, inputSample, 0);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "ProcessInput failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = mf_create_output_sample(sys);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "mf_create_output_sample failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
outputDataBuffer.dwStreamID = 0;
|
||||
outputDataBuffer.dwStatus = 0;
|
||||
outputDataBuffer.pEvents = NULL;
|
||||
outputDataBuffer.pSample = sys->outputSample;
|
||||
hr = sys->transform->lpVtbl->ProcessOutput(sys->transform, 0, 1,
|
||||
&outputDataBuffer, &outputStatus);
|
||||
|
||||
if (hr == MF_E_TRANSFORM_STREAM_CHANGE)
|
||||
{
|
||||
UINT32 stride = 0;
|
||||
UINT64 frameSize = 0;
|
||||
|
||||
if (sys->outputType)
|
||||
{
|
||||
sys->outputType->lpVtbl->Release(sys->outputType);
|
||||
sys->outputType = NULL;
|
||||
}
|
||||
|
||||
hr = mf_find_output_type(sys, &MFVideoFormat_IYUV, &sys->outputType);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "mf_find_output_type failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = sys->transform->lpVtbl->SetOutputType(sys->transform, 0, sys->outputType,
|
||||
0);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "SetOutputType failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = mf_create_output_sample(sys);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "mf_create_output_sample failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = sys->outputType->lpVtbl->GetUINT64(sys->outputType, &MF_MT_FRAME_SIZE,
|
||||
&frameSize);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "GetUINT64(MF_MT_FRAME_SIZE) failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
sys->frameWidth = (UINT32)(frameSize >> 32);
|
||||
sys->frameHeight = (UINT32) frameSize;
|
||||
hr = sys->outputType->lpVtbl->GetUINT32(sys->outputType, &MF_MT_DEFAULT_STRIDE,
|
||||
&stride);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "GetUINT32(MF_MT_DEFAULT_STRIDE) failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
if (!avc420_ensure_buffer(h264, stride, sys->frameWidth, sys->frameHeight))
|
||||
goto error;
|
||||
}
|
||||
else if (hr == MF_E_TRANSFORM_NEED_MORE_INPUT)
|
||||
{
|
||||
}
|
||||
else if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "ProcessOutput failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
else
|
||||
{
|
||||
int offset = 0;
|
||||
BYTE* buffer = NULL;
|
||||
DWORD bufferCount = 0;
|
||||
DWORD cbMaxLength = 0;
|
||||
DWORD cbCurrentLength = 0;
|
||||
hr = sys->outputSample->lpVtbl->GetBufferCount(sys->outputSample, &bufferCount);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "GetBufferCount failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = sys->outputSample->lpVtbl->GetBufferByIndex(sys->outputSample, 0,
|
||||
&outputBuffer);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "GetBufferByIndex failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = outputBuffer->lpVtbl->Lock(outputBuffer, &buffer, &cbMaxLength,
|
||||
&cbCurrentLength);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "Lock failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
CopyMemory(pYUVData[0], &buffer[offset], iStride[0] * sys->frameHeight);
|
||||
offset += iStride[0] * sys->frameHeight;
|
||||
CopyMemory(pYUVData[1], &buffer[offset], iStride[1] * (sys->frameHeight / 2));
|
||||
offset += iStride[1] * (sys->frameHeight / 2);
|
||||
CopyMemory(pYUVData[2], &buffer[offset], iStride[2] * (sys->frameHeight / 2));
|
||||
offset += iStride[2] * (sys->frameHeight / 2);
|
||||
hr = outputBuffer->lpVtbl->Unlock(outputBuffer);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "Unlock failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
outputBuffer->lpVtbl->Release(outputBuffer);
|
||||
}
|
||||
|
||||
inputSample->lpVtbl->Release(inputSample);
|
||||
return 1;
|
||||
error:
|
||||
fprintf(stderr, "mf_decompress error\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
static int mf_compress(H264_CONTEXT* h264, BYTE** ppDstData, UINT32* pDstSize)
|
||||
{
|
||||
H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*) h264->pSystemData;
|
||||
return 1;
|
||||
}
|
||||
|
||||
static void mf_uninit(H264_CONTEXT* h264)
|
||||
{
|
||||
UINT32 x;
|
||||
H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*) h264->pSystemData;
|
||||
|
||||
if (sys)
|
||||
{
|
||||
if (sys->transform)
|
||||
{
|
||||
sys->transform->lpVtbl->Release(sys->transform);
|
||||
sys->transform = NULL;
|
||||
}
|
||||
|
||||
if (sys->codecApi)
|
||||
{
|
||||
sys->codecApi->lpVtbl->Release(sys->codecApi);
|
||||
sys->codecApi = NULL;
|
||||
}
|
||||
|
||||
if (sys->inputType)
|
||||
{
|
||||
sys->inputType->lpVtbl->Release(sys->inputType);
|
||||
sys->inputType = NULL;
|
||||
}
|
||||
|
||||
if (sys->outputType)
|
||||
{
|
||||
sys->outputType->lpVtbl->Release(sys->outputType);
|
||||
sys->outputType = NULL;
|
||||
}
|
||||
|
||||
if (sys->outputSample)
|
||||
{
|
||||
sys->outputSample->lpVtbl->Release(sys->outputSample);
|
||||
sys->outputSample = NULL;
|
||||
}
|
||||
|
||||
if (sys->mfplat)
|
||||
{
|
||||
FreeLibrary(sys->mfplat);
|
||||
sys->mfplat = NULL;
|
||||
}
|
||||
|
||||
for (x = 0; x < sizeof(h264->pYUVData) / sizeof(h264->pYUVData[0]); x++)
|
||||
_aligned_free(h264->pYUVData[x]);
|
||||
|
||||
memset(h264->pYUVData, 0, sizeof(h264->pYUVData));
|
||||
memset(h264->iStride, 0, sizeof(h264->iStride));
|
||||
sys->MFShutdown();
|
||||
CoUninitialize();
|
||||
free(sys);
|
||||
h264->pSystemData = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
static BOOL mf_init(H264_CONTEXT* h264)
|
||||
{
|
||||
HRESULT hr;
|
||||
H264_CONTEXT_MF* sys = (H264_CONTEXT_MF*) calloc(1, sizeof(H264_CONTEXT_MF));
|
||||
|
||||
if (!sys)
|
||||
goto error;
|
||||
|
||||
h264->pSystemData = (void*) sys;
|
||||
/* http://decklink-sdk-delphi.googlecode.com/svn/trunk/Blackmagic%20DeckLink%20SDK%209.7/Win/Samples/Streaming/StreamingPreview/DecoderMF.cpp */
|
||||
sys->mfplat = LoadLibraryA("mfplat.dll");
|
||||
|
||||
if (!sys->mfplat)
|
||||
goto error;
|
||||
|
||||
sys->MFStartup = (pfnMFStartup) GetProcAddress(sys->mfplat, "MFStartup");
|
||||
sys->MFShutdown = (pfnMFShutdown) GetProcAddress(sys->mfplat, "MFShutdown");
|
||||
sys->MFCreateSample = (pfnMFCreateSample) GetProcAddress(sys->mfplat,
|
||||
"MFCreateSample");
|
||||
sys->MFCreateMemoryBuffer = (pfnMFCreateMemoryBuffer) GetProcAddress(
|
||||
sys->mfplat, "MFCreateMemoryBuffer");
|
||||
sys->MFCreateMediaType = (pfnMFCreateMediaType) GetProcAddress(sys->mfplat,
|
||||
"MFCreateMediaType");
|
||||
sys->MFCreateDXGIDeviceManager = (pfnMFCreateDXGIDeviceManager) GetProcAddress(
|
||||
sys->mfplat, "MFCreateDXGIDeviceManager");
|
||||
|
||||
if (!sys->MFStartup || !sys->MFShutdown || !sys->MFCreateSample
|
||||
|| !sys->MFCreateMemoryBuffer ||
|
||||
!sys->MFCreateMediaType || !sys->MFCreateDXGIDeviceManager)
|
||||
goto error;
|
||||
|
||||
CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
|
||||
|
||||
if (h264->Compressor)
|
||||
{
|
||||
}
|
||||
else
|
||||
{
|
||||
VARIANT var = { 0 };
|
||||
hr = sys->MFStartup(MF_VERSION, 0);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "MFStartup failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = CoCreateInstance(&CLSID_CMSH264DecoderMFT, NULL, CLSCTX_INPROC_SERVER,
|
||||
&IID_IMFTransform, (void**) &sys->transform);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "CoCreateInstance(CLSID_CMSH264DecoderMFT) failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = sys->transform->lpVtbl->QueryInterface(sys->transform, &IID_ICodecAPI,
|
||||
(void**) &sys->codecApi);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "QueryInterface(IID_ICodecAPI) failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
var.vt = VT_UI4;
|
||||
var.ulVal = 1;
|
||||
hr = sys->codecApi->lpVtbl->SetValue(sys->codecApi, &CODECAPI_AVLowLatencyMode,
|
||||
&var);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "SetValue(CODECAPI_AVLowLatencyMode) failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = sys->MFCreateMediaType(&sys->inputType);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "MFCreateMediaType failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = sys->inputType->lpVtbl->SetGUID(sys->inputType, &MF_MT_MAJOR_TYPE,
|
||||
&MFMediaType_Video);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "SetGUID(MF_MT_MAJOR_TYPE) failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = sys->inputType->lpVtbl->SetGUID(sys->inputType, &MF_MT_SUBTYPE,
|
||||
&MFVideoFormat_H264);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "SetGUID(MF_MT_SUBTYPE) failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = sys->transform->lpVtbl->SetInputType(sys->transform, 0, sys->inputType, 0);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "SetInputType failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = mf_find_output_type(sys, &MFVideoFormat_IYUV, &sys->outputType);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "mf_find_output_type failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = sys->transform->lpVtbl->SetOutputType(sys->transform, 0, sys->outputType,
|
||||
0);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "SetOutputType failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
|
||||
hr = mf_create_output_sample(sys);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "mf_create_output_sample failure: 0x%08"PRIX32"", hr);
|
||||
goto error;
|
||||
}
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
error:
|
||||
WLog_ERR(TAG, "mf_init failure");
|
||||
mf_uninit(h264);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
H264_CONTEXT_SUBSYSTEM g_Subsystem_MF =
|
||||
{
|
||||
"MediaFoundation",
|
||||
mf_init,
|
||||
mf_uninit,
|
||||
mf_decompress,
|
||||
mf_compress
|
||||
};
|
||||
|
||||
464
libfreerdp/codec/h264_openh264.c
Normal file
464
libfreerdp/codec/h264_openh264.c
Normal file
@@ -0,0 +1,464 @@
|
||||
/**
|
||||
* FreeRDP: A Remote Desktop Protocol Implementation
|
||||
* H.264 Bitmap Compression
|
||||
*
|
||||
* Copyright 2014 Mike McDonald <Mike.McDonald@software.dell.com>
|
||||
* Copyright 2015 Vic Lee <llyzs.vic@gmail.com>
|
||||
* Copyright 2014 Armin Novak <armin.novak@gmail.com>
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include <freerdp/codec/h264.h>
|
||||
|
||||
#include "wels/codec_def.h"
|
||||
#include "wels/codec_api.h"
|
||||
#include "wels/codec_ver.h"
|
||||
|
||||
#if (OPENH264_MAJOR == 1) && (OPENH264_MINOR < 3) || (OPENH264_MAJOR < 1)
|
||||
#error "Unsupported OpenH264 version "OPENH264_MAJOR"."OPENH264_MINOR"."OPENH264_REVISION" detected!"
|
||||
#elif (OPENH264_MAJOR > 1) || (OPENH264_MINOR > 6)
|
||||
#warning "Untested OpenH264 version "OPENH264_MAJOR"."OPENH264_MINOR"."OPENH264_REVISION" detected!"
|
||||
#endif
|
||||
|
||||
struct _H264_CONTEXT_OPENH264
|
||||
{
|
||||
ISVCDecoder* pDecoder;
|
||||
ISVCEncoder* pEncoder;
|
||||
SEncParamExt EncParamExt;
|
||||
};
|
||||
typedef struct _H264_CONTEXT_OPENH264 H264_CONTEXT_OPENH264;
|
||||
|
||||
static BOOL g_openh264_trace_enabled = FALSE;
|
||||
|
||||
static void openh264_trace_callback(H264_CONTEXT* h264, int level,
|
||||
const char* message)
|
||||
{
|
||||
WLog_INFO(TAG, "%d - %s", level, message);
|
||||
}
|
||||
|
||||
static int openh264_decompress(H264_CONTEXT* h264, const BYTE* pSrcData,
|
||||
UINT32 SrcSize)
|
||||
{
|
||||
DECODING_STATE state;
|
||||
SBufferInfo sBufferInfo;
|
||||
SSysMEMBuffer* pSystemBuffer;
|
||||
H264_CONTEXT_OPENH264* sys = (H264_CONTEXT_OPENH264*) h264->pSystemData;
|
||||
UINT32* iStride = h264->iStride;
|
||||
BYTE** pYUVData = h264->pYUVData;
|
||||
|
||||
if (!sys->pDecoder)
|
||||
return -2001;
|
||||
|
||||
/*
|
||||
* Decompress the image. The RDP host only seems to send I420 format.
|
||||
*/
|
||||
pYUVData[0] = NULL;
|
||||
pYUVData[1] = NULL;
|
||||
pYUVData[2] = NULL;
|
||||
ZeroMemory(&sBufferInfo, sizeof(sBufferInfo));
|
||||
state = (*sys->pDecoder)->DecodeFrame2(sys->pDecoder, pSrcData, SrcSize,
|
||||
pYUVData, &sBufferInfo);
|
||||
|
||||
if (sBufferInfo.iBufferStatus != 1)
|
||||
{
|
||||
if (state == dsNoParamSets)
|
||||
{
|
||||
/* this happens on the first frame due to missing parameter sets */
|
||||
state = (*sys->pDecoder)->DecodeFrame2(sys->pDecoder, NULL, 0, pYUVData,
|
||||
&sBufferInfo);
|
||||
}
|
||||
else if (state == dsErrorFree)
|
||||
{
|
||||
/* call DecodeFrame2 again to decode without delay */
|
||||
state = (*sys->pDecoder)->DecodeFrame2(sys->pDecoder, NULL, 0, pYUVData,
|
||||
&sBufferInfo);
|
||||
}
|
||||
else
|
||||
{
|
||||
WLog_WARN(TAG, "DecodeFrame2 state: 0x%04X iBufferStatus: %d", state,
|
||||
sBufferInfo.iBufferStatus);
|
||||
return -2002;
|
||||
}
|
||||
}
|
||||
|
||||
pSystemBuffer = &sBufferInfo.UsrData.sSystemBuffer;
|
||||
iStride[0] = pSystemBuffer->iStride[0];
|
||||
iStride[1] = pSystemBuffer->iStride[1];
|
||||
iStride[2] = pSystemBuffer->iStride[1];
|
||||
|
||||
if (sBufferInfo.iBufferStatus != 1)
|
||||
{
|
||||
WLog_WARN(TAG, "DecodeFrame2 iBufferStatus: %d", sBufferInfo.iBufferStatus);
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (state != dsErrorFree)
|
||||
{
|
||||
WLog_WARN(TAG, "DecodeFrame2 state: 0x%02X", state);
|
||||
return -2003;
|
||||
}
|
||||
|
||||
#if 0
|
||||
WLog_INFO(TAG,
|
||||
"h264_decompress: state=%u, pYUVData=[%p,%p,%p], bufferStatus=%d, width=%d, height=%d, format=%d, stride=[%d,%d]",
|
||||
state, (void*) pYUVData[0], (void*) pYUVData[1], (void*) pYUVData[2], sBufferInfo.iBufferStatus,
|
||||
pSystemBuffer->iWidth, pSystemBuffer->iHeight, pSystemBuffer->iFormat,
|
||||
pSystemBuffer->iStride[0], pSystemBuffer->iStride[1]);
|
||||
#endif
|
||||
|
||||
if (pSystemBuffer->iFormat != videoFormatI420)
|
||||
return -2004;
|
||||
|
||||
if (!pYUVData[0] || !pYUVData[1] || !pYUVData[2])
|
||||
return -2005;
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
static int openh264_compress(H264_CONTEXT* h264, BYTE** ppDstData,
|
||||
UINT32* pDstSize)
|
||||
{
|
||||
int i, j;
|
||||
int status;
|
||||
SFrameBSInfo info;
|
||||
SSourcePicture pic;
|
||||
SBitrateInfo bitrate;
|
||||
H264_CONTEXT_OPENH264* sys;
|
||||
BYTE** pYUVData = h264->pYUVData;
|
||||
UINT32* iStride = h264->iStride;
|
||||
sys = &((H264_CONTEXT_OPENH264*) h264->pSystemData)[0];
|
||||
|
||||
if (!sys->pEncoder)
|
||||
return -1;
|
||||
|
||||
if (!pYUVData[0] || !pYUVData[1] || !pYUVData[2])
|
||||
return -1;
|
||||
|
||||
if ((sys->EncParamExt.iPicWidth != h264->width)
|
||||
|| (sys->EncParamExt.iPicHeight != h264->height))
|
||||
{
|
||||
status = (*sys->pEncoder)->GetDefaultParams(sys->pEncoder, &sys->EncParamExt);
|
||||
|
||||
if (status < 0)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to get OpenH264 default parameters (status=%d)", status);
|
||||
return status;
|
||||
}
|
||||
|
||||
sys->EncParamExt.iUsageType = SCREEN_CONTENT_REAL_TIME;
|
||||
sys->EncParamExt.iPicWidth = h264->width;
|
||||
sys->EncParamExt.iPicHeight = h264->height;
|
||||
sys->EncParamExt.fMaxFrameRate = h264->FrameRate;
|
||||
sys->EncParamExt.iMaxBitrate = UNSPECIFIED_BIT_RATE;
|
||||
sys->EncParamExt.bEnableDenoise = 0;
|
||||
sys->EncParamExt.bEnableLongTermReference = 0;
|
||||
sys->EncParamExt.bEnableFrameSkip = 0;
|
||||
sys->EncParamExt.iSpatialLayerNum = 1;
|
||||
sys->EncParamExt.iMultipleThreadIdc = h264->NumberOfThreads;
|
||||
sys->EncParamExt.sSpatialLayers[0].fFrameRate = h264->FrameRate;
|
||||
sys->EncParamExt.sSpatialLayers[0].iVideoWidth = sys->EncParamExt.iPicWidth;
|
||||
sys->EncParamExt.sSpatialLayers[0].iVideoHeight = sys->EncParamExt.iPicHeight;
|
||||
sys->EncParamExt.sSpatialLayers[0].iMaxSpatialBitrate =
|
||||
sys->EncParamExt.iMaxBitrate;
|
||||
|
||||
switch (h264->RateControlMode)
|
||||
{
|
||||
case H264_RATECONTROL_VBR:
|
||||
sys->EncParamExt.iRCMode = RC_BITRATE_MODE;
|
||||
sys->EncParamExt.iTargetBitrate = h264->BitRate;
|
||||
sys->EncParamExt.sSpatialLayers[0].iSpatialBitrate =
|
||||
sys->EncParamExt.iTargetBitrate;
|
||||
break;
|
||||
|
||||
case H264_RATECONTROL_CQP:
|
||||
sys->EncParamExt.iRCMode = RC_OFF_MODE;
|
||||
sys->EncParamExt.sSpatialLayers[0].iDLayerQp = h264->QP;
|
||||
break;
|
||||
}
|
||||
|
||||
if (sys->EncParamExt.iMultipleThreadIdc > 1)
|
||||
{
|
||||
#if (OPENH264_MAJOR == 1) && (OPENH264_MINOR <= 5)
|
||||
sys->EncParamExt.sSpatialLayers[0].sSliceCfg.uiSliceMode = SM_AUTO_SLICE;
|
||||
#else
|
||||
sys->EncParamExt.sSpatialLayers[0].sSliceArgument.uiSliceMode = SM_FIXEDSLCNUM_SLICE;
|
||||
#endif
|
||||
}
|
||||
|
||||
status = (*sys->pEncoder)->InitializeExt(sys->pEncoder, &sys->EncParamExt);
|
||||
|
||||
if (status < 0)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to initialize OpenH264 encoder (status=%d)", status);
|
||||
return status;
|
||||
}
|
||||
|
||||
status = (*sys->pEncoder)->GetOption(sys->pEncoder,
|
||||
ENCODER_OPTION_SVC_ENCODE_PARAM_EXT,
|
||||
&sys->EncParamExt);
|
||||
|
||||
if (status < 0)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to get initial OpenH264 encoder parameters (status=%d)",
|
||||
status);
|
||||
return status;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
switch (h264->RateControlMode)
|
||||
{
|
||||
case H264_RATECONTROL_VBR:
|
||||
if (sys->EncParamExt.iTargetBitrate != h264->BitRate)
|
||||
{
|
||||
sys->EncParamExt.iTargetBitrate = h264->BitRate;
|
||||
bitrate.iLayer = SPATIAL_LAYER_ALL;
|
||||
bitrate.iBitrate = h264->BitRate;
|
||||
status = (*sys->pEncoder)->SetOption(sys->pEncoder, ENCODER_OPTION_BITRATE,
|
||||
&bitrate);
|
||||
|
||||
if (status < 0)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to set encoder bitrate (status=%d)", status);
|
||||
return status;
|
||||
}
|
||||
}
|
||||
|
||||
if (sys->EncParamExt.fMaxFrameRate != h264->FrameRate)
|
||||
{
|
||||
sys->EncParamExt.fMaxFrameRate = h264->FrameRate;
|
||||
status = (*sys->pEncoder)->SetOption(sys->pEncoder, ENCODER_OPTION_FRAME_RATE,
|
||||
&sys->EncParamExt.fMaxFrameRate);
|
||||
|
||||
if (status < 0)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to set encoder framerate (status=%d)", status);
|
||||
return status;
|
||||
}
|
||||
}
|
||||
|
||||
break;
|
||||
|
||||
case H264_RATECONTROL_CQP:
|
||||
if (sys->EncParamExt.sSpatialLayers[0].iDLayerQp != h264->QP)
|
||||
{
|
||||
sys->EncParamExt.sSpatialLayers[0].iDLayerQp = h264->QP;
|
||||
status = (*sys->pEncoder)->SetOption(sys->pEncoder,
|
||||
ENCODER_OPTION_SVC_ENCODE_PARAM_EXT,
|
||||
&sys->EncParamExt);
|
||||
|
||||
if (status < 0)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to set encoder parameters (status=%d)", status);
|
||||
return status;
|
||||
}
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
memset(&info, 0, sizeof(SFrameBSInfo));
|
||||
memset(&pic, 0, sizeof(SSourcePicture));
|
||||
pic.iPicWidth = h264->width;
|
||||
pic.iPicHeight = h264->height;
|
||||
pic.iColorFormat = videoFormatI420;
|
||||
pic.iStride[0] = iStride[0];
|
||||
pic.iStride[1] = iStride[1];
|
||||
pic.iStride[2] = iStride[2];
|
||||
pic.pData[0] = pYUVData[0];
|
||||
pic.pData[1] = pYUVData[1];
|
||||
pic.pData[2] = pYUVData[2];
|
||||
status = (*sys->pEncoder)->EncodeFrame(sys->pEncoder, &pic, &info);
|
||||
|
||||
if (status < 0)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to encode frame (status=%d)", status);
|
||||
return status;
|
||||
}
|
||||
|
||||
*ppDstData = info.sLayerInfo[0].pBsBuf;
|
||||
*pDstSize = 0;
|
||||
|
||||
for (i = 0; i < info.iLayerNum; i++)
|
||||
{
|
||||
for (j = 0; j < info.sLayerInfo[i].iNalCount; j++)
|
||||
{
|
||||
*pDstSize += info.sLayerInfo[i].pNalLengthInByte[j];
|
||||
}
|
||||
}
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
static void openh264_uninit(H264_CONTEXT* h264)
|
||||
{
|
||||
UINT32 x;
|
||||
H264_CONTEXT_OPENH264* sysContexts = (H264_CONTEXT_OPENH264*) h264->pSystemData;
|
||||
|
||||
if (sysContexts)
|
||||
{
|
||||
for (x = 0; x < h264->numSystemData; x++)
|
||||
{
|
||||
H264_CONTEXT_OPENH264* sys = &sysContexts[x];
|
||||
|
||||
if (sys->pDecoder)
|
||||
{
|
||||
(*sys->pDecoder)->Uninitialize(sys->pDecoder);
|
||||
WelsDestroyDecoder(sys->pDecoder);
|
||||
sys->pDecoder = NULL;
|
||||
}
|
||||
|
||||
if (sys->pEncoder)
|
||||
{
|
||||
(*sys->pEncoder)->Uninitialize(sys->pEncoder);
|
||||
WelsDestroySVCEncoder(sys->pEncoder);
|
||||
sys->pEncoder = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
free(h264->pSystemData);
|
||||
h264->pSystemData = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
static BOOL openh264_init(H264_CONTEXT* h264)
|
||||
{
|
||||
UINT32 x;
|
||||
long status;
|
||||
SDecodingParam sDecParam;
|
||||
H264_CONTEXT_OPENH264* sysContexts;
|
||||
static int traceLevel = WELS_LOG_DEBUG;
|
||||
#if (OPENH264_MAJOR == 1) && (OPENH264_MINOR <= 5)
|
||||
static EVideoFormatType videoFormat = videoFormatI420;
|
||||
#endif
|
||||
static WelsTraceCallback traceCallback = (WelsTraceCallback)
|
||||
openh264_trace_callback;
|
||||
h264->numSystemData = 1;
|
||||
sysContexts = (H264_CONTEXT_OPENH264*) calloc(h264->numSystemData,
|
||||
sizeof(H264_CONTEXT_OPENH264));
|
||||
|
||||
if (!sysContexts)
|
||||
goto EXCEPTION;
|
||||
|
||||
h264->pSystemData = (void*) sysContexts;
|
||||
|
||||
for (x = 0; x < h264->numSystemData; x++)
|
||||
{
|
||||
H264_CONTEXT_OPENH264* sys = &sysContexts[x];
|
||||
|
||||
if (h264->Compressor)
|
||||
{
|
||||
WelsCreateSVCEncoder(&sys->pEncoder);
|
||||
|
||||
if (!sys->pEncoder)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to create OpenH264 encoder");
|
||||
goto EXCEPTION;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
WelsCreateDecoder(&sys->pDecoder);
|
||||
|
||||
if (!sys->pDecoder)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to create OpenH264 decoder");
|
||||
goto EXCEPTION;
|
||||
}
|
||||
|
||||
ZeroMemory(&sDecParam, sizeof(sDecParam));
|
||||
#if (OPENH264_MAJOR == 1) && (OPENH264_MINOR <= 5)
|
||||
sDecParam.eOutputColorFormat = videoFormatI420;
|
||||
#endif
|
||||
sDecParam.eEcActiveIdc = ERROR_CON_FRAME_COPY;
|
||||
sDecParam.sVideoProperty.eVideoBsType = VIDEO_BITSTREAM_AVC;
|
||||
status = (*sys->pDecoder)->Initialize(sys->pDecoder, &sDecParam);
|
||||
|
||||
if (status != 0)
|
||||
{
|
||||
WLog_ERR(TAG, "Failed to initialize OpenH264 decoder (status=%ld)",
|
||||
status);
|
||||
goto EXCEPTION;
|
||||
}
|
||||
|
||||
#if (OPENH264_MAJOR == 1) && (OPENH264_MINOR <= 5)
|
||||
status = (*sys->pDecoder)->SetOption(
|
||||
sys->pDecoder, DECODER_OPTION_DATAFORMAT,
|
||||
&videoFormat);
|
||||
#endif
|
||||
|
||||
if (status != 0)
|
||||
{
|
||||
WLog_ERR(TAG,
|
||||
"Failed to set data format option on OpenH264 decoder (status=%ld)",
|
||||
status);
|
||||
goto EXCEPTION;
|
||||
}
|
||||
|
||||
if (g_openh264_trace_enabled)
|
||||
{
|
||||
status = (*sys->pDecoder)->SetOption(
|
||||
sys->pDecoder, DECODER_OPTION_TRACE_LEVEL,
|
||||
&traceLevel);
|
||||
|
||||
if (status != 0)
|
||||
{
|
||||
WLog_ERR(TAG,
|
||||
"Failed to set trace level option on OpenH264 decoder (status=%ld)",
|
||||
status);
|
||||
goto EXCEPTION;
|
||||
}
|
||||
|
||||
status = (*sys->pDecoder)->SetOption(
|
||||
sys->pDecoder, DECODER_OPTION_TRACE_CALLBACK,
|
||||
&traceCallback);
|
||||
|
||||
if (status != 0)
|
||||
{
|
||||
WLog_ERR(TAG,
|
||||
"Failed to set trace callback option on OpenH264 decoder (status=%ld)",
|
||||
status);
|
||||
goto EXCEPTION;
|
||||
}
|
||||
|
||||
status = (*sys->pDecoder)->SetOption(
|
||||
sys->pDecoder,
|
||||
DECODER_OPTION_TRACE_CALLBACK_CONTEXT,
|
||||
&h264);
|
||||
|
||||
if (status != 0)
|
||||
{
|
||||
WLog_ERR(TAG,
|
||||
"Failed to set trace callback context option on OpenH264 decoder (status=%ld)",
|
||||
status);
|
||||
goto EXCEPTION;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
EXCEPTION:
|
||||
openh264_uninit(h264);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
H264_CONTEXT_SUBSYSTEM g_Subsystem_OpenH264 =
|
||||
{
|
||||
"OpenH264",
|
||||
openh264_init,
|
||||
openh264_uninit,
|
||||
openh264_decompress,
|
||||
openh264_compress
|
||||
};
|
||||
121
libfreerdp/codec/h264_x264.c
Normal file
121
libfreerdp/codec/h264_x264.c
Normal file
@@ -0,0 +1,121 @@
|
||||
/**
|
||||
* FreeRDP: A Remote Desktop Protocol Implementation
|
||||
* H.264 Bitmap Compression
|
||||
*
|
||||
* Copyright 2015 Marc-André Moreau <marcandre.moreau@gmail.com>
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#define NAL_UNKNOWN X264_NAL_UNKNOWN
|
||||
#define NAL_SLICE X264_NAL_SLICE
|
||||
#define NAL_SLICE_DPA X264_NAL_SLICE_DPA
|
||||
#define NAL_SLICE_DPB X264_NAL_SLICE_DPB
|
||||
#define NAL_SLICE_DPC X264_NAL_SLICE_DPC
|
||||
#define NAL_SLICE_IDR X264_NAL_SLICE_IDR
|
||||
#define NAL_SEI X264_NAL_SEI
|
||||
#define NAL_SPS X264_NAL_SPS
|
||||
#define NAL_PPS X264_NAL_PPS
|
||||
#define NAL_AUD X264_NAL_AUD
|
||||
#define NAL_FILLER X264_NAL_FILLER
|
||||
|
||||
#define NAL_PRIORITY_DISPOSABLE X264_NAL_PRIORITY_DISPOSABLE
|
||||
#define NAL_PRIORITY_LOW X264_NAL_PRIORITY_LOW
|
||||
#define NAL_PRIORITY_HIGH X264_NAL_PRIORITY_HIGH
|
||||
#define NAL_PRIORITY_HIGHEST X264_NAL_PRIORITY_HIGHEST
|
||||
|
||||
#include <stdint.h>
|
||||
#include <x264.h>
|
||||
#include <freerdp/codec/h264.h>
|
||||
|
||||
struct _H264_CONTEXT_X264
|
||||
{
|
||||
void* dummy;
|
||||
};
|
||||
typedef struct _H264_CONTEXT_X264 H264_CONTEXT_X264;
|
||||
|
||||
static int x264_decompress(H264_CONTEXT* h264, BYTE* pSrcData, UINT32 SrcSize)
|
||||
{
|
||||
//H264_CONTEXT_X264* sys = (H264_CONTEXT_X264*) h264->pSystemData;
|
||||
return -1;
|
||||
}
|
||||
|
||||
static int x264_compress(H264_CONTEXT* h264, BYTE** ppDstData, UINT32* pDstSize)
|
||||
{
|
||||
//H264_CONTEXT_X264* sys = (H264_CONTEXT_X264*) h264->pSystemData;
|
||||
return -1;
|
||||
}
|
||||
|
||||
static void x264_uninit(H264_CONTEXT* h264)
|
||||
{
|
||||
H264_CONTEXT_X264* sys = (H264_CONTEXT_X264*) h264->pSystemData;
|
||||
|
||||
if (sys)
|
||||
{
|
||||
free(sys);
|
||||
h264->pSystemData = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
static BOOL x264_init(H264_CONTEXT* h264)
|
||||
{
|
||||
H264_CONTEXT_X264* sys;
|
||||
h264->numSystemData = 1;
|
||||
sys = (H264_CONTEXT_X264*) calloc(h264->numSystemData,
|
||||
sizeof(H264_CONTEXT_X264));
|
||||
|
||||
if (!sys)
|
||||
{
|
||||
goto EXCEPTION;
|
||||
}
|
||||
|
||||
h264->pSystemData = (void*) sys;
|
||||
|
||||
if (h264->Compressor)
|
||||
{
|
||||
}
|
||||
else
|
||||
{
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
EXCEPTION:
|
||||
x264_uninit(h264);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
H264_CONTEXT_SUBSYSTEM g_Subsystem_x264 =
|
||||
{
|
||||
"x264",
|
||||
x264_init,
|
||||
x264_uninit,
|
||||
x264_decompress,
|
||||
x264_compress
|
||||
};
|
||||
|
||||
#undef NAL_UNKNOWN
|
||||
#undef NAL_SLICE
|
||||
#undef NAL_SLICE_DPA
|
||||
#undef NAL_SLICE_DPB
|
||||
#undef NAL_SLICE_DPC
|
||||
#undef NAL_SLICE_IDR
|
||||
#undef NAL_SEI
|
||||
#undef NAL_SPS
|
||||
#undef NAL_PPS
|
||||
#undef NAL_AUD
|
||||
#undef NAL_FILLER
|
||||
|
||||
#undef NAL_PRIORITY_DISPOSABLE
|
||||
#undef NAL_PRIORITY_LOW
|
||||
#undef NAL_PRIORITY_HIGH
|
||||
#undef NAL_PRIORITY_HIGHEST
|
||||
Reference in New Issue
Block a user