moonlight-stream/moonlight-embedded
1
0
Fork 0
mirror of https://github.com/moonlight-stream/moonlight-embedded.git synced 2025-07-01 23:35:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix for video corruption

Browse Source
This commit is contained in:
Iwan Timmer 2018-01-14 14:06:36 +00:00
parent 5d708b98c9
commit 6411150383
4 changed files with 48 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libgamestream/CMakeLists.txt
View File

@ -1,4 +1,4 @@
set(SO_VERSION 2) set(SO_VERSION 3)
find_package(LibUUID REQUIRED) find_package(LibUUID REQUIRED)
find_package(Threads REQUIRED) find_package(Threads REQUIRED)

14
libgamestream/sps.c
View File

@ -32,19 +32,20 @@ void gs_sps_init(int width, int height) {
initial_height = height; initial_height = height;
} }
PLENTRY gs_sps_fix(PLENTRY *head, int flags) { void gs_sps_fix(PDECODE_UNIT decodeUnit, int flags) {
PLENTRY entry = *head; PLENTRY entry = decodeUnit->bufferList;
const char naluHeader[] = {0x00, 0x00, 0x00, 0x01}; const char naluHeader[] = {0x00, 0x00, 0x00, 0x01};
if (replay_sps == 1) { if (replay_sps == 1) {
PLENTRY replay_entry = (PLENTRY) malloc(sizeof(*replay_entry) + 128); PLENTRY replay_entry = (PLENTRY) malloc(sizeof(*replay_entry) + 128);
if (replay_entry == NULL) if (replay_entry == NULL)
return NULL; return;
replay_entry->data = (char *) (entry + 1); replay_entry->data = (char *) (entry + 1);
memcpy(replay_entry->data, naluHeader, sizeof(naluHeader)); memcpy(replay_entry->data, naluHeader, sizeof(naluHeader));
h264_stream->sps->profile_idc = H264_PROFILE_HIGH; h264_stream->sps->profile_idc = H264_PROFILE_HIGH;
replay_entry->length = write_nal_unit(h264_stream, replay_entry->data+4, 124) + 4; replay_entry->length = write_nal_unit(h264_stream, replay_entry->data+4, 124) + 4;
decodeUnit->fullLength += replay_entry->length;
replay_entry->next = entry; replay_entry->next = entry;
entry = replay_entry; entry = replay_entry;
@ -97,20 +98,21 @@ PLENTRY gs_sps_fix(PLENTRY *head, int flags) {
PLENTRY sps_entry = (PLENTRY) malloc(sizeof(*sps_entry) + 128); PLENTRY sps_entry = (PLENTRY) malloc(sizeof(*sps_entry) + 128);
if (sps_entry == NULL) if (sps_entry == NULL)
return NULL; return;
PLENTRY next = entry->next; PLENTRY next = entry->next;
decodeUnit->fullLength -= entry->length;
free(entry); free(entry);
sps_entry->data = (char*) (sps_entry + 1); sps_entry->data = (char*) (sps_entry + 1);
memcpy(sps_entry->data, naluHeader, sizeof(naluHeader)); memcpy(sps_entry->data, naluHeader, sizeof(naluHeader));
sps_entry->length = write_nal_unit(h264_stream, sps_entry->data+4, 124) + 4; sps_entry->length = write_nal_unit(h264_stream, sps_entry->data+4, 124) + 4;
sps_entry->next = next; sps_entry->next = next;
entry = sps_entry; entry = sps_entry;
decodeUnit->fullLength += entry->length;
} else if ((entry->data[4] & 0x1F) == NAL_UNIT_TYPE_PPS) { } else if ((entry->data[4] & 0x1F) == NAL_UNIT_TYPE_PPS) {
if ((flags & GS_SPS_BASELINE_HACK) == GS_SPS_BASELINE_HACK && !replay_sps) if ((flags & GS_SPS_BASELINE_HACK) == GS_SPS_BASELINE_HACK && !replay_sps)
replay_sps = 1; replay_sps = 1;
} }
*head = entry; decodeUnit->bufferList = entry;
return entry;
} }

2
libgamestream/sps.h
View File

@ -23,4 +23,4 @@
#define GS_SPS_BASELINE_HACK 0x02 #define GS_SPS_BASELINE_HACK 0x02
void gs_sps_init(int width, int height); void gs_sps_init(int width, int height);
PLENTRY gs_sps_fix(PLENTRY *entry, int flags); void gs_sps_fix(PDECODE_UNIT decodeUnit, int flags);

81
src/video/pi.c
View File

@ -47,9 +47,6 @@ static ILCLIENT_T *client;
static COMPONENT_T *video_decode = NULL, *video_scheduler = NULL, *video_render = NULL; static COMPONENT_T *video_decode = NULL, *video_scheduler = NULL, *video_render = NULL;
static OMX_BUFFERHEADERTYPE *buf;
static unsigned char *dest;
static int port_settings_changed; static int port_settings_changed;
static int first_packet; static int first_packet;
@ -177,6 +174,7 @@ static int decoder_renderer_setup(int videoFormat, int width, int height, int re
static void decoder_renderer_cleanup() { static void decoder_renderer_cleanup() {
int status = 0; int status = 0;
OMX_BUFFERHEADERTYPE *buf;
if((buf = ilclient_get_input_buffer(video_decode, 130, 1)) == NULL){ if((buf = ilclient_get_input_buffer(video_decode, 130, 1)) == NULL){
fprintf(stderr, "Can't get video buffer\n"); fprintf(stderr, "Can't get video buffer\n");
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
@ -209,51 +207,48 @@ static void decoder_renderer_cleanup() {
} }
static int decoder_renderer_submit_decode_unit(PDECODE_UNIT decodeUnit) { static int decoder_renderer_submit_decode_unit(PDECODE_UNIT decodeUnit) {
if((buf = ilclient_get_input_buffer(video_decode, 130, 1)) == NULL){ OMX_BUFFERHEADERTYPE *buf = NULL;
fprintf(stderr, "Can't get video buffer\n");
exit(EXIT_FAILURE);
}
// feed data and wait until we get port settings changed gs_sps_fix(decodeUnit, GS_SPS_BITSTREAM_FIXUP);
dest = buf->pBuffer; PLENTRY entry = decodeUnit->bufferList;
buf->nFilledLen = 0;
buf->nOffset = 0;
buf->nFlags = OMX_BUFFERFLAG_ENDOFFRAME | OMX_BUFFERFLAG_EOS;
if(first_packet) {
buf->nFlags = OMX_BUFFERFLAG_STARTTIME;
first_packet = 0;
}
PLENTRY entry = gs_sps_fix(&decodeUnit->bufferList, GS_SPS_BITSTREAM_FIXUP);
decodeUnit->bufferList = entry;
while (entry != NULL) { while (entry != NULL) {
memcpy(dest, entry->data, entry->length); if (buf == NULL) {
buf->nFilledLen += entry->length; if ((buf = ilclient_get_input_buffer(video_decode, 130, 1)) == NULL) {
dest += entry->length; fprintf(stderr, "Can't get video buffer\n");
entry = entry->next; exit(EXIT_FAILURE);
} }
buf->nFilledLen = 0;
if(port_settings_changed == 0 && buf->nOffset = 0;
((buf->nFilledLen > 0 && ilclient_remove_event(video_decode, OMX_EventPortSettingsChanged, 131, 0, 0, 1) == 0) || buf->nFlags = OMX_BUFFERFLAG_ENDOFFRAME | OMX_BUFFERFLAG_EOS;
(buf->nFilledLen == 0 && ilclient_wait_for_event(video_decode, OMX_EventPortSettingsChanged, 131, 0, 0, 1, if(first_packet) {
ILCLIENT_EVENT_ERROR | ILCLIENT_PARAMETER_CHANGED, 10000) == 0))) { buf->nFlags = OMX_BUFFERFLAG_STARTTIME;
port_settings_changed = 1; first_packet = 0;
}
if(ilclient_setup_tunnel(tunnel, 0, 0) != 0){
fprintf(stderr, "Can't setup video\n");
exit(EXIT_FAILURE);
} }
ilclient_change_component_state(video_render, OMX_StateExecuting); memcpy(buf->pBuffer + buf->nFilledLen, entry->data, entry->length);
} buf->nFilledLen += entry->length;
if(OMX_EmptyThisBuffer(ILC_GET_HANDLE(video_decode), buf) != OMX_ErrorNone){ if (entry->bufferType != BUFFER_TYPE_PICDATA || entry->next == NULL || entry->next->bufferType != BUFFER_TYPE_PICDATA) {
fprintf(stderr, "Can't empty video buffer\n"); if(port_settings_changed == 0 && ((buf->nFilledLen > 0 && ilclient_remove_event(video_decode, OMX_EventPortSettingsChanged, 131, 0, 0, 1) == 0) || (buf->nFilledLen == 0 && ilclient_wait_for_event(video_decode, OMX_EventPortSettingsChanged, 131, 0, 0, 1, ILCLIENT_EVENT_ERROR | ILCLIENT_PARAMETER_CHANGED, 10000) == 0))) {
exit(EXIT_FAILURE); port_settings_changed = 1;
if(ilclient_setup_tunnel(tunnel, 0, 0) != 0) {
fprintf(stderr, "Can't setup video\n");
exit(EXIT_FAILURE);
}
ilclient_change_component_state(video_render, OMX_StateExecuting);
}
if(OMX_EmptyThisBuffer(ILC_GET_HANDLE(video_decode), buf) != OMX_ErrorNone){
fprintf(stderr, "Can't empty video buffer\n");
exit(EXIT_FAILURE);
}
buf = NULL;
}
entry = entry->next;
} }
return DR_OK; return DR_OK;