scottk/UltraGrid
1
0
Fork 0
mirror of https://github.com/outbackdingo/UltraGrid.git synced 2026-03-21 18:40:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

source formatted

Browse Source
This commit is contained in:
Castillo, Gerard
2013-09-25 11:23:41 +02:00
parent 51a2e40bb7
commit c0d335c2f7
4 changed files with 610 additions and 565 deletions
Download Patch File Download Diff File Expand all files Collapse all files

264
src/rtp/rtpdec_h264.c
View File

@@ -53,162 +53,164 @@
#include "video_capture/rtsp.h"
#include "rtp/pbuf.h"
struct recieved_data{
uint32_t buffer_len;//[MAX_SUBSTREAMS];
struct recieved_data {
uint32_t buffer_len; //[MAX_SUBSTREAMS];
//uint32_t buffer_num;//[MAX_SUBSTREAMS];
char *frame_buffer;//[MAX_SUBSTREAMS];
char *frame_buffer; //[MAX_SUBSTREAMS];
};
static const uint8_t start_sequence[] = { 0, 0, 0, 1 };
int decode_frame_h264(struct coded_data *cdata, void *rx_data);
int
decode_frame_h264(struct coded_data *cdata, void *rx_data);
int decode_frame_h264(struct coded_data *cdata, void *rx_data) {
rtp_packet *pckt = NULL;
struct coded_data *orig = cdata;
int
decode_frame_h264(struct coded_data *cdata, void *rx_data) {
rtp_packet *pckt = NULL;
struct coded_data *orig = cdata;
uint8_t nal;
uint8_t type;
uint8_t nal;
uint8_t type;
int pass;
int total_length = 0;
int pass;
int total_length = 0;
char *dst = NULL;
int src_len;
char *dst = NULL;
int src_len;
struct recieved_data *buffers = (struct recieved_data *) rx_data;
struct recieved_data *buffers = (struct recieved_data *) rx_data;
for (pass = 0; pass < 2; pass++) {
for (pass = 0; pass < 2; pass++) {
if (pass > 0) {
cdata = orig;
buffers->buffer_len = total_length;
dst = buffers->frame_buffer + total_length;
}
if (pass > 0) {
cdata = orig;
buffers->buffer_len = total_length;
dst = buffers->frame_buffer + total_length;
}
while (cdata != NULL) {
pckt = cdata->data;
while (cdata != NULL) {
pckt = cdata->data;
if (pckt->pt != PT_H264) {
error_msg("Wrong Payload type: %u\n", pckt->pt);
return FALSE;
}
if (pckt->pt != PT_H264) {
error_msg("Wrong Payload type: %u\n", pckt->pt);
return FALSE;
}
nal = (uint8_t) pckt->data[0];
type = nal & 0x1f;
nal = (uint8_t) pckt->data[0];
type = nal & 0x1f;
if (type >= 1 && type <= 23) {
type = 1;
}
if (type >= 1 && type <= 23) {
type = 1;
}
const uint8_t *src = NULL;
const uint8_t *src = NULL;
switch (type) {
case 0:
case 1:
if (pass == 0) {
debug_msg("NAL type 1\n");
total_length += sizeof(start_sequence) + pckt->data_len;
} else {
dst -= pckt->data_len + sizeof(start_sequence);
memcpy(dst, start_sequence, sizeof(start_sequence));
memcpy(dst + sizeof(start_sequence), pckt->data, pckt->data_len);
unsigned char *dst2 = (unsigned char *)dst;
}
break;
case 24:
src = (const uint8_t *) pckt->data;
src_len = pckt->data_len;
switch (type) {
case 0:
case 1:
if (pass == 0) {
debug_msg("NAL type 1\n");
total_length += sizeof(start_sequence) + pckt->data_len;
} else {
dst -= pckt->data_len + sizeof(start_sequence);
memcpy(dst, start_sequence, sizeof(start_sequence));
memcpy(dst + sizeof(start_sequence), pckt->data, pckt->data_len);
unsigned char *dst2 = (unsigned char *)dst;
}
break;
case 24:
src = (const uint8_t *) pckt->data;
src_len = pckt->data_len;
src++;
src_len--;
src++;
src_len--;
while (src_len > 2) {
//TODO: Not properly tested
uint16_t nal_size;
memcpy(&nal_size, src, sizeof(uint16_t));
while (src_len > 2) {
//TODO: Not properly tested
uint16_t nal_size;
memcpy(&nal_size, src, sizeof(uint16_t));
src += 2;
src_len -= 2;
src += 2;
src_len -= 2;
if (nal_size <= src_len) {
if (pass == 0) {
total_length += sizeof(start_sequence) + nal_size;
} else {
dst -= nal_size + sizeof(start_sequence);
memcpy(dst, start_sequence, sizeof(start_sequence));
memcpy(dst + sizeof(start_sequence), src, nal_size);
}
} else {
error_msg("NAL size exceeds length: %u %d\n", nal_size, src_len);
return FALSE;
}
src += nal_size;
src_len -= nal_size;
if (nal_size <= src_len) {
if (pass == 0) {
total_length += sizeof(start_sequence) + nal_size;
} else {
dst -= nal_size + sizeof(start_sequence);
memcpy(dst, start_sequence, sizeof(start_sequence));
memcpy(dst + sizeof(start_sequence), src, nal_size);
}
} else {
error_msg("NAL size exceeds length: %u %d\n", nal_size, src_len);
return FALSE;
}
src += nal_size;
src_len -= nal_size;
if (src_len < 0) {
error_msg("Consumed more bytes than we got! (%d)\n", src_len);
return FALSE;
}
}
break;
if (src_len < 0) {
error_msg("Consumed more bytes than we got! (%d)\n", src_len);
return FALSE;
}
}
break;
case 25:
case 26:
case 27:
case 29:
error_msg("Unhandled NAL type\n");
return FALSE;
case 28:
src = (const uint8_t *) pckt->data;
src_len = pckt->data_len;
case 25:
case 26:
case 27:
case 29:
error_msg("Unhandled NAL type\n");
return FALSE;
case 28:
src = (const uint8_t *) pckt->data;
src_len = pckt->data_len;
src++;
src_len--;
src++;
src_len--;
if (src_len > 1) {
uint8_t fu_header = *src;
uint8_t start_bit = fu_header >> 7;
//uint8_t end_bit = (fu_header & 0x40) >> 6;
uint8_t nal_type = fu_header & 0x1f;
uint8_t reconstructed_nal;
if (src_len > 1) {
uint8_t fu_header = *src;
uint8_t start_bit = fu_header >> 7;
//uint8_t end_bit = (fu_header & 0x40) >> 6;
uint8_t nal_type = fu_header & 0x1f;
uint8_t reconstructed_nal;
// Reconstruct this packet's true nal; only the data follows.
/* The original nal forbidden bit and NRI are stored in this
* packet's nal. */
reconstructed_nal = nal & 0xe0;
reconstructed_nal |= nal_type;
// Reconstruct this packet's true nal; only the data follows.
/* The original nal forbidden bit and NRI are stored in this
* packet's nal. */
reconstructed_nal = nal & 0xe0;
reconstructed_nal |= nal_type;
// skip the fu_header
src++;
src_len--;
// skip the fu_header
src++;
src_len--;
if (pass == 0) {
if (start_bit) {
total_length += sizeof(start_sequence) + sizeof(reconstructed_nal) + src_len;
} else {
total_length += src_len;
}
} else {
if (start_bit) {
dst -= sizeof(start_sequence) + sizeof(reconstructed_nal) + src_len;
memcpy(dst, start_sequence, sizeof(start_sequence));
memcpy(dst + sizeof(start_sequence), &reconstructed_nal, sizeof(reconstructed_nal));
memcpy(dst + sizeof(start_sequence) + sizeof(reconstructed_nal), src, src_len);
} else {
dst -= src_len;
memcpy(dst, src, src_len);
}
}
} else {
error_msg("Too short data for FU-A H264 RTP packet\n");
return FALSE;
}
break;
default:
error_msg("Unknown NAL type\n");
return FALSE;
}
cdata = cdata->nxt;
}
}
return TRUE;
if (pass == 0) {
if (start_bit) {
total_length += sizeof(start_sequence) + sizeof(reconstructed_nal) + src_len;
} else {
total_length += src_len;
}
} else {
if (start_bit) {
dst -= sizeof(start_sequence) + sizeof(reconstructed_nal) + src_len;
memcpy(dst, start_sequence, sizeof(start_sequence));
memcpy(dst + sizeof(start_sequence), &reconstructed_nal, sizeof(reconstructed_nal));
memcpy(dst + sizeof(start_sequence) + sizeof(reconstructed_nal), src, src_len);
} else {
dst -= src_len;
memcpy(dst, src, src_len);
}
}
} else {
error_msg("Too short data for FU-A H264 RTP packet\n");
return FALSE;
}
break;
default:
error_msg("Unknown NAL type\n");
return FALSE;
}
cdata = cdata->nxt;
}
}
return TRUE;
}

3
src/rtp/rtpdec_h264.h
View File

@@ -44,6 +44,7 @@
#ifndef _RTP_DEC_H264_H
#define _RTP_DEC_H264_H
int decode_frame_h264(struct coded_data *cdata, void *rx_data);
int
decode_frame_h264(struct coded_data *cdata, void *rx_data);
#endif

895
src/video_capture/rtsp.c
View File

File diff suppressed because it is too large Load Diff

13
src/video_capture/rtsp.h
View File

@@ -40,11 +40,14 @@ extern "C" {
#endif
#define VIDCAP_RTSP_ID 0x45b3d828 //md5 hash of VIDCAP_RTSP_ID string == a208d26f519a2664a48781c845b3d828
struct vidcap_type *vidcap_rtsp_probe(void);
void *vidcap_rtsp_init(const struct vidcap_params *params);
void vidcap_rtsp_done(void *state);
struct video_frame *vidcap_rtsp_grab(void *state, struct audio_frame **audio);
struct vidcap_type *
vidcap_rtsp_probe(void);
void *
vidcap_rtsp_init(const struct vidcap_params *params);
void
vidcap_rtsp_done(void *state);
struct video_frame *
vidcap_rtsp_grab(void *state, struct audio_frame **audio);
#endif