mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2026-06-19 19:03:00 +02:00
pass dts and pts through ffm, should fix streaming b frames
Originally committed as revision 13725 to svn://svn.ffmpeg.org/ffmpeg/trunk
This commit is contained in:
+3
-3
@@ -29,8 +29,9 @@
|
||||
#define PACKET_ID 0x666d
|
||||
|
||||
/* each packet contains frames (which can span several packets */
|
||||
#define FRAME_HEADER_SIZE 8
|
||||
#define FRAME_HEADER_SIZE 16
|
||||
#define FLAG_KEY_FRAME 0x01
|
||||
#define FLAG_DTS 0x02
|
||||
|
||||
enum {
|
||||
READ_HEADER,
|
||||
@@ -41,11 +42,10 @@ typedef struct FFMContext {
|
||||
/* only reading mode */
|
||||
offset_t write_index, file_size;
|
||||
int read_state;
|
||||
uint8_t header[FRAME_HEADER_SIZE];
|
||||
uint8_t header[FRAME_HEADER_SIZE+4];
|
||||
|
||||
/* read and write */
|
||||
int first_packet; /* true if first packet, needed to set the discontinuity tag */
|
||||
int first_frame_in_packet; /* true if first frame in packet, needed to know if PTS information is valid */
|
||||
int packet_size;
|
||||
int frame_offset;
|
||||
int64_t pts;
|
||||
|
||||
@@ -100,7 +100,6 @@ static int ffm_read_data(AVFormatContext *s,
|
||||
get_be16(pb); /* PACKET_ID */
|
||||
fill_size = get_be16(pb);
|
||||
ffm->pts = get_be64(pb);
|
||||
ffm->first_frame_in_packet = 1;
|
||||
frame_offset = get_be16(pb);
|
||||
get_buffer(pb, ffm->packet, ffm->packet_size - FFM_HEADER_SIZE);
|
||||
ffm->packet_end = ffm->packet + (ffm->packet_size - FFM_HEADER_SIZE - fill_size);
|
||||
@@ -361,7 +360,7 @@ static int ffm_read_packet(AVFormatContext *s, AVPacket *pkt)
|
||||
|
||||
switch(ffm->read_state) {
|
||||
case READ_HEADER:
|
||||
if (!ffm_is_avail_data(s, FRAME_HEADER_SIZE)) {
|
||||
if (!ffm_is_avail_data(s, FRAME_HEADER_SIZE+4)) {
|
||||
return AVERROR(EAGAIN);
|
||||
}
|
||||
dprintf(s, "pos=%08"PRIx64" spos=%"PRIx64", write_index=%"PRIx64" size=%"PRIx64"\n",
|
||||
@@ -369,6 +368,9 @@ static int ffm_read_packet(AVFormatContext *s, AVPacket *pkt)
|
||||
if (ffm_read_data(s, ffm->header, FRAME_HEADER_SIZE, 1) !=
|
||||
FRAME_HEADER_SIZE)
|
||||
return AVERROR(EAGAIN);
|
||||
if (ffm->header[1] & FLAG_DTS)
|
||||
if (ffm_read_data(s, ffm->header+16, 4, 1) != 4)
|
||||
return AVERROR(EAGAIN);
|
||||
#if 0
|
||||
av_hexdump_log(s, AV_LOG_DEBUG, ffm->header, FRAME_HEADER_SIZE);
|
||||
#endif
|
||||
@@ -400,11 +402,11 @@ static int ffm_read_packet(AVFormatContext *s, AVPacket *pkt)
|
||||
av_free_packet(pkt);
|
||||
return AVERROR(EAGAIN);
|
||||
}
|
||||
if (ffm->first_frame_in_packet)
|
||||
{
|
||||
pkt->pts = ffm->pts;
|
||||
ffm->first_frame_in_packet = 0;
|
||||
}
|
||||
pkt->pts = AV_RB64(ffm->header+8);
|
||||
if (ffm->header[1] & FLAG_DTS)
|
||||
pkt->dts = pkt->pts - AV_RB32(ffm->header+16);
|
||||
else
|
||||
pkt->dts = pkt->pts;
|
||||
pkt->duration = duration;
|
||||
break;
|
||||
}
|
||||
|
||||
@@ -197,6 +197,7 @@ static int ffm_write_packet(AVFormatContext *s, AVPacket *pkt)
|
||||
AVStream *st = s->streams[pkt->stream_index];
|
||||
int64_t pts;
|
||||
uint8_t header[FRAME_HEADER_SIZE];
|
||||
int header_size = FRAME_HEADER_SIZE;
|
||||
|
||||
pts = s->timestamp + pkt->pts;
|
||||
/* packet size & key_frame */
|
||||
@@ -206,7 +207,13 @@ static int ffm_write_packet(AVFormatContext *s, AVPacket *pkt)
|
||||
header[1] |= FLAG_KEY_FRAME;
|
||||
AV_WB24(header+2, pkt->size);
|
||||
AV_WB24(header+5, pkt->duration);
|
||||
ffm_write_data(s, header, FRAME_HEADER_SIZE, pts, 1);
|
||||
AV_WB64(header+8, pts);
|
||||
if (pkt->pts != pkt->dts) {
|
||||
header[1] |= FLAG_DTS;
|
||||
AV_WB32(header+16, pkt->pts - pkt->dts);
|
||||
header_size += 4;
|
||||
}
|
||||
ffm_write_data(s, header, header_size, pts, 1);
|
||||
ffm_write_data(s, pkt->data, pkt->size, pts, 0);
|
||||
|
||||
return 0;
|
||||
|
||||
Reference in New Issue
Block a user