/*
- * MPEG1/2 mux/demux
- * Copyright (c) 2000, 2001, 2002 Fabrice Bellard.
+ * MPEG1/2 demuxer
+ * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
*
- * This library is free software; you can redistribute it and/or
+ * This file is part of Libav.
+ *
+ * Libav is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
- * version 2 of the License, or (at your option) any later version.
+ * version 2.1 of the License, or (at your option) any later version.
*
- * This library is distributed in the hope that it will be useful,
+ * Libav is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
- * License along with this library; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ * License along with Libav; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
+
#include "avformat.h"
+#include "internal.h"
+#include "mpeg.h"
-#define MAX_PAYLOAD_SIZE 4096
-#define NB_STREAMS 2
-
-typedef struct {
- UINT8 buffer[MAX_PAYLOAD_SIZE];
- int buffer_ptr;
- UINT8 id;
- int max_buffer_size; /* in bytes */
- int packet_number;
- INT64 start_pts;
-} StreamInfo;
-
-typedef struct {
- int packet_size; /* required packet size */
- int packet_data_max_size; /* maximum data size inside a packet */
- int packet_number;
- int pack_header_freq; /* frequency (in packets^-1) at which we send pack headers */
- int system_header_freq;
- int mux_rate; /* bitrate in units of 50 bytes/s */
- /* stream info */
- int audio_bound;
- int video_bound;
- int is_mpeg2;
- int is_vcd;
-} MpegMuxContext;
-
-#define PACK_START_CODE ((unsigned int)0x000001ba)
-#define SYSTEM_HEADER_START_CODE ((unsigned int)0x000001bb)
-#define SEQUENCE_END_CODE ((unsigned int)0x000001b7)
-#define PACKET_START_CODE_MASK ((unsigned int)0xffffff00)
-#define PACKET_START_CODE_PREFIX ((unsigned int)0x00000100)
-#define ISO_11172_END_CODE ((unsigned int)0x000001b9)
-
-/* mpeg2 */
-#define PROGRAM_STREAM_MAP 0x1bc
-#define PRIVATE_STREAM_1 0x1bd
-#define PADDING_STREAM 0x1be
-#define PRIVATE_STREAM_2 0x1bf
-
-
-#define AUDIO_ID 0xc0
-#define VIDEO_ID 0xe0
-
-extern AVOutputFormat mpeg1system_mux;
-extern AVOutputFormat mpeg1vcd_mux;
-extern AVOutputFormat mpeg2vob_mux;
-
-static int put_pack_header(AVFormatContext *ctx,
- UINT8 *buf, INT64 timestamp)
-{
- MpegMuxContext *s = ctx->priv_data;
- PutBitContext pb;
-
- init_put_bits(&pb, buf, 128, NULL, NULL);
-
- put_bits(&pb, 32, PACK_START_CODE);
- if (s->is_mpeg2) {
- put_bits(&pb, 2, 0x2);
- } else {
- put_bits(&pb, 4, 0x2);
- }
- put_bits(&pb, 3, (UINT32)((timestamp >> 30) & 0x07));
- put_bits(&pb, 1, 1);
- put_bits(&pb, 15, (UINT32)((timestamp >> 15) & 0x7fff));
- put_bits(&pb, 1, 1);
- put_bits(&pb, 15, (UINT32)((timestamp) & 0x7fff));
- put_bits(&pb, 1, 1);
- if (s->is_mpeg2) {
- /* clock extension */
- put_bits(&pb, 9, 0);
- put_bits(&pb, 1, 1);
- }
- put_bits(&pb, 1, 1);
- put_bits(&pb, 22, s->mux_rate);
- put_bits(&pb, 1, 1);
- if (s->is_mpeg2) {
- put_bits(&pb, 5, 0x1f); /* reserved */
- put_bits(&pb, 3, 0); /* stuffing length */
- }
- flush_put_bits(&pb);
- return pbBufPtr(&pb) - pb.buf;
-}
+#undef NDEBUG
+#include <assert.h>
-static int put_system_header(AVFormatContext *ctx, UINT8 *buf)
-{
- MpegMuxContext *s = ctx->priv_data;
- int size, rate_bound, i, private_stream_coded, id;
- PutBitContext pb;
-
- init_put_bits(&pb, buf, 128, NULL, NULL);
-
- put_bits(&pb, 32, SYSTEM_HEADER_START_CODE);
- put_bits(&pb, 16, 0);
- put_bits(&pb, 1, 1);
-
- rate_bound = s->mux_rate; /* maximum bit rate of the multiplexed stream */
- put_bits(&pb, 22, rate_bound);
- put_bits(&pb, 1, 1); /* marker */
- put_bits(&pb, 6, s->audio_bound);
-
- put_bits(&pb, 1, 1); /* variable bitrate */
- put_bits(&pb, 1, 1); /* non constrainted bit stream */
-
- put_bits(&pb, 1, 0); /* audio locked */
- put_bits(&pb, 1, 0); /* video locked */
- put_bits(&pb, 1, 1); /* marker */
-
- put_bits(&pb, 5, s->video_bound);
- put_bits(&pb, 8, 0xff); /* reserved byte */
-
- /* audio stream info */
- private_stream_coded = 0;
- for(i=0;i<ctx->nb_streams;i++) {
- StreamInfo *stream = ctx->streams[i]->priv_data;
- id = stream->id;
- if (id < 0xc0) {
- /* special case for private streams (AC3 use that) */
- if (private_stream_coded)
- continue;
- private_stream_coded = 1;
- id = 0xbd;
- }
- put_bits(&pb, 8, id); /* stream ID */
- put_bits(&pb, 2, 3);
- if (id < 0xe0) {
- /* audio */
- put_bits(&pb, 1, 0);
- put_bits(&pb, 13, stream->max_buffer_size / 128);
- } else {
- /* video */
- put_bits(&pb, 1, 1);
- put_bits(&pb, 13, stream->max_buffer_size / 1024);
- }
- }
- flush_put_bits(&pb);
- size = pbBufPtr(&pb) - pb.buf;
- /* patch packet size */
- buf[4] = (size - 6) >> 8;
- buf[5] = (size - 6) & 0xff;
+/*********************************************/
+/* demux code */
- return size;
-}
+#define MAX_SYNC_SIZE 100000
-static int mpeg_mux_init(AVFormatContext *ctx)
+static int check_pes(uint8_t *p, uint8_t *end)
{
- MpegMuxContext *s = ctx->priv_data;
- int bitrate, i, mpa_id, mpv_id, ac3_id;
- AVStream *st;
- StreamInfo *stream;
-
- s->packet_number = 0;
- s->is_vcd = (ctx->oformat == &mpeg1vcd_mux);
- s->is_mpeg2 = (ctx->oformat == &mpeg2vob_mux);
-
- if (s->is_vcd)
- s->packet_size = 2324; /* VCD packet size */
+ int pes1;
+ int pes2 = (p[3] & 0xC0) == 0x80 &&
+ (p[4] & 0xC0) != 0x40 &&
+ ((p[4] & 0xC0) == 0x00 ||
+ (p[4] & 0xC0) >> 2 == (p[6] & 0xF0));
+
+ for (p += 3; p < end && *p == 0xFF; p++) ;
+ if ((*p & 0xC0) == 0x40)
+ p += 2;
+
+ if ((*p & 0xF0) == 0x20)
+ pes1 = p[0] & p[2] & p[4] & 1;
+ else if ((*p & 0xF0) == 0x30)
+ pes1 = p[0] & p[2] & p[4] & p[5] & p[7] & p[9] & 1;
else
- s->packet_size = 2048;
-
- /* startcode(4) + length(2) + flags(1) */
- s->packet_data_max_size = s->packet_size - 7;
- s->audio_bound = 0;
- s->video_bound = 0;
- mpa_id = AUDIO_ID;
- ac3_id = 0x80;
- mpv_id = VIDEO_ID;
- for(i=0;i<ctx->nb_streams;i++) {
- st = ctx->streams[i];
- stream = av_mallocz(sizeof(StreamInfo));
- if (!stream)
- goto fail;
- st->priv_data = stream;
-
- switch(st->codec.codec_type) {
- case CODEC_TYPE_AUDIO:
- if (st->codec.codec_id == CODEC_ID_AC3)
- stream->id = ac3_id++;
- else
- stream->id = mpa_id++;
- stream->max_buffer_size = 4 * 1024;
- s->audio_bound++;
- break;
- case CODEC_TYPE_VIDEO:
- stream->id = mpv_id++;
- stream->max_buffer_size = 46 * 1024;
- s->video_bound++;
- break;
- default:
- av_abort();
- }
- }
+ pes1 = *p == 0x0F;
- /* we increase slightly the bitrate to take into account the
- headers. XXX: compute it exactly */
- bitrate = 2000;
- for(i=0;i<ctx->nb_streams;i++) {
- st = ctx->streams[i];
- bitrate += st->codec.bit_rate;
- }
- s->mux_rate = (bitrate + (8 * 50) - 1) / (8 * 50);
-
- if (s->is_vcd || s->is_mpeg2)
- /* every packet */
- s->pack_header_freq = 1;
- else
- /* every 2 seconds */
- s->pack_header_freq = 2 * bitrate / s->packet_size / 8;
-
- if (s->is_mpeg2)
- /* every 200 packets. Need to look at the spec. */
- s->system_header_freq = s->pack_header_freq * 40;
- else if (s->is_vcd)
- /* every 40 packets, this is my invention */
- s->system_header_freq = s->pack_header_freq * 40;
- else
- s->system_header_freq = s->pack_header_freq * 5;
-
- for(i=0;i<ctx->nb_streams;i++) {
- stream = ctx->streams[i]->priv_data;
- stream->buffer_ptr = 0;
- stream->packet_number = 0;
- stream->start_pts = -1;
- }
- return 0;
- fail:
- for(i=0;i<ctx->nb_streams;i++) {
- av_free(ctx->streams[i]->priv_data);
- }
- return -ENOMEM;
+ return pes1 || pes2;
}
-/* flush the packet on stream stream_index */
-static void flush_packet(AVFormatContext *ctx, int stream_index, int last_pkt)
+static int check_pack_header(const uint8_t *buf)
{
- MpegMuxContext *s = ctx->priv_data;
- StreamInfo *stream = ctx->streams[stream_index]->priv_data;
- UINT8 *buf_ptr;
- int size, payload_size, startcode, id, len, stuffing_size, i, header_len;
- INT64 timestamp;
- UINT8 buffer[128];
- int last = last_pkt ? 4 : 0;
-
- id = stream->id;
- timestamp = stream->start_pts;
-
-#if 0
- printf("packet ID=%2x PTS=%0.3f\n",
- id, timestamp / 90000.0);
-#endif
-
- buf_ptr = buffer;
- if (((s->packet_number % s->pack_header_freq) == 0)) {
- /* output pack and systems header if needed */
- size = put_pack_header(ctx, buf_ptr, timestamp);
- buf_ptr += size;
- if ((s->packet_number % s->system_header_freq) == 0) {
- size = put_system_header(ctx, buf_ptr);
- buf_ptr += size;
- }
- }
- size = buf_ptr - buffer;
- put_buffer(&ctx->pb, buffer, size);
-
- /* packet header */
- if (s->is_mpeg2) {
- header_len = 8;
- } else {
- header_len = 5;
- }
- payload_size = s->packet_size - (size + 6 + header_len + last);
- if (id < 0xc0) {
- startcode = PRIVATE_STREAM_1;
- payload_size -= 4;
- } else {
- startcode = 0x100 + id;
- }
- stuffing_size = payload_size - stream->buffer_ptr;
- if (stuffing_size < 0)
- stuffing_size = 0;
-
- put_be32(&ctx->pb, startcode);
+ return (buf[1] & 0xC0) == 0x40 || (buf[1] & 0xF0) == 0x20;
+}
- put_be16(&ctx->pb, payload_size + header_len);
- /* stuffing */
- for(i=0;i<stuffing_size;i++)
- put_byte(&ctx->pb, 0xff);
+static int mpegps_probe(AVProbeData *p)
+{
+ uint32_t code = -1;
+ int i;
+ int sys = 0, pspack = 0, priv1 = 0, vid = 0;
+ int audio = 0, invalid = 0, score = 0;
- if (s->is_mpeg2) {
- put_byte(&ctx->pb, 0x80); /* mpeg2 id */
- put_byte(&ctx->pb, 0x80); /* flags */
- put_byte(&ctx->pb, 0x05); /* header len (only pts is included) */
- }
- put_byte(&ctx->pb,
- (0x02 << 4) |
- (((timestamp >> 30) & 0x07) << 1) |
- 1);
- put_be16(&ctx->pb, (UINT16)((((timestamp >> 15) & 0x7fff) << 1) | 1));
- put_be16(&ctx->pb, (UINT16)((((timestamp) & 0x7fff) << 1) | 1));
-
- if (startcode == PRIVATE_STREAM_1) {
- put_byte(&ctx->pb, id);
- if (id >= 0x80 && id <= 0xbf) {
- /* XXX: need to check AC3 spec */
- put_byte(&ctx->pb, 1);
- put_byte(&ctx->pb, 0);
- put_byte(&ctx->pb, 2);
+ for (i = 0; i < p->buf_size; i++) {
+ code = (code << 8) + p->buf[i];
+ if ((code & 0xffffff00) == 0x100) {
+ int len = p->buf[i + 1] << 8 | p->buf[i + 2];
+ int pes = check_pes(p->buf + i, p->buf + p->buf_size);
+ int pack = check_pack_header(p->buf + i);
+
+ if (code == SYSTEM_HEADER_START_CODE)
+ sys++;
+ else if (code == PACK_START_CODE && pack)
+ pspack++;
+ else if ((code & 0xf0) == VIDEO_ID && pes)
+ vid++;
+ // skip pes payload to avoid start code emulation for private
+ // and audio streams
+ else if ((code & 0xe0) == AUDIO_ID && pes) {
+ audio++;
+ i += len;
+ } else if (code == PRIVATE_STREAM_1 && pes) {
+ priv1++;
+ i += len;
+ } else if ((code & 0xf0) == VIDEO_ID && !pes)
+ invalid++;
+ else if ((code & 0xe0) == AUDIO_ID && !pes)
+ invalid++;
+ else if (code == PRIVATE_STREAM_1 && !pes)
+ invalid++;
}
}
- if (last_pkt) {
- put_be32(&ctx->pb, ISO_11172_END_CODE);
- }
- /* output data */
- put_buffer(&ctx->pb, stream->buffer, payload_size - stuffing_size);
- put_flush_packet(&ctx->pb);
-
- /* preserve remaining data */
- len = stream->buffer_ptr - payload_size;
- if (len < 0)
- len = 0;
- memmove(stream->buffer, stream->buffer + stream->buffer_ptr - len, len);
- stream->buffer_ptr = len;
-
- s->packet_number++;
- stream->packet_number++;
- stream->start_pts = -1;
+ if (vid + audio > invalid) /* invalid VDR files nd short PES streams */
+ score = AVPROBE_SCORE_EXTENSION / 2;
+
+ if (sys > invalid && sys * 9 <= pspack * 10)
+ return pspack > 2 ? AVPROBE_SCORE_EXTENSION + 2
+ : AVPROBE_SCORE_EXTENSION / 2; // 1 more than .mpg
+ if (pspack > invalid && (priv1 + vid + audio) * 10 >= pspack * 9)
+ return pspack > 2 ? AVPROBE_SCORE_EXTENSION + 2
+ : AVPROBE_SCORE_EXTENSION / 2; // 1 more than .mpg
+ if ((!!vid ^ !!audio) && (audio > 4 || vid > 1) && !sys &&
+ !pspack && p->buf_size > 2048 && vid + audio > invalid) /* PES stream */
+ return (audio > 12 || vid > 3) ? AVPROBE_SCORE_EXTENSION + 2
+ : AVPROBE_SCORE_EXTENSION / 2;
+
+ // 02-Penguin.flac has sys:0 priv1:0 pspack:0 vid:0 audio:1
+ // mp3_misidentified_2.mp3 has sys:0 priv1:0 pspack:0 vid:0 audio:6
+ return score;
}
-static int mpeg_mux_write_packet(AVFormatContext *ctx, int stream_index,
- UINT8 *buf, int size, int pts)
-{
- MpegMuxContext *s = ctx->priv_data;
- AVStream *st = ctx->streams[stream_index];
- StreamInfo *stream = st->priv_data;
- int len;
-
- while (size > 0) {
- /* set pts */
- if (stream->start_pts == -1) {
- stream->start_pts = pts;
- }
- len = s->packet_data_max_size - stream->buffer_ptr;
- if (len > size)
- len = size;
- memcpy(stream->buffer + stream->buffer_ptr, buf, len);
- stream->buffer_ptr += len;
- buf += len;
- size -= len;
- while (stream->buffer_ptr >= s->packet_data_max_size) {
- /* output the packet */
- if (stream->start_pts == -1)
- stream->start_pts = pts;
- flush_packet(ctx, stream_index, 0);
- }
- }
- return 0;
-}
+typedef struct MpegDemuxContext {
+ int32_t header_state;
+ unsigned char psm_es_type[256];
+ int sofdec;
+} MpegDemuxContext;
-static int mpeg_mux_end(AVFormatContext *ctx)
+static int mpegps_read_header(AVFormatContext *s)
{
- StreamInfo *stream;
- int i;
-
- /* flush each packet */
- for(i=0;i<ctx->nb_streams;i++) {
- stream = ctx->streams[i]->priv_data;
- if (stream->buffer_ptr > 0) {
- if (i == (ctx->nb_streams - 1))
- flush_packet(ctx, i, 1);
- else
- flush_packet(ctx, i, 0);
- }
- }
+ MpegDemuxContext *m = s->priv_data;
+ const char *sofdec = "Sofdec";
+ int v, i = 0;
- /* write the end header */
- //put_be32(&ctx->pb, ISO_11172_END_CODE);
- //put_flush_packet(&ctx->pb);
- return 0;
-}
+ m->header_state = 0xff;
+ s->ctx_flags |= AVFMTCTX_NOHEADER;
-/*********************************************/
-/* demux code */
+ m->sofdec = -1;
+ do {
+ v = avio_r8(s->pb);
+ m->header_state = m->header_state << 8 | v;
+ m->sofdec++;
+ } while (v == sofdec[i] && i++ < 6);
-#define MAX_SYNC_SIZE 100000
+ m->sofdec = (m->sofdec == 6) ? 1 : 0;
-static int mpegps_probe(AVProbeData *p)
-{
- int code, c, i;
- code = 0xff;
-
- /* we search the first start code. If it is a packet start code,
- then we decide it is mpeg ps. We do not send highest value to
- give a chance to mpegts */
- for(i=0;i<p->buf_size;i++) {
- c = p->buf[i];
- code = (code << 8) | c;
- if ((code & 0xffffff00) == 0x100) {
- if (code == PACK_START_CODE ||
- code == SYSTEM_HEADER_START_CODE ||
- (code >= 0x1e0 && code <= 0x1ef) ||
- (code >= 0x1c0 && code <= 0x1df) ||
- code == PRIVATE_STREAM_2 ||
- code == PROGRAM_STREAM_MAP ||
- code == PRIVATE_STREAM_1 ||
- code == PADDING_STREAM)
- return AVPROBE_SCORE_MAX - 1;
- else
- return 0;
- }
- }
+ /* no need to do more */
return 0;
}
+static int64_t get_pts(AVIOContext *pb, int c)
+{
+ uint8_t buf[5];
-typedef struct MpegDemuxContext {
- int header_state;
-} MpegDemuxContext;
+ buf[0] = c < 0 ? avio_r8(pb) : c;
+ avio_read(pb, buf + 1, 4);
+
+ return ff_parse_pes_pts(buf);
+}
-static int find_start_code(ByteIOContext *pb, int *size_ptr,
- UINT32 *header_state)
+static int find_next_start_code(AVIOContext *pb, int *size_ptr,
+ int32_t *header_state)
{
unsigned int state, v;
int val, n;
state = *header_state;
- n = *size_ptr;
+ n = *size_ptr;
while (n > 0) {
- if (url_feof(pb))
+ if (pb->eof_reached)
break;
- v = get_byte(pb);
+ v = avio_r8(pb);
n--;
if (state == 0x000001) {
state = ((state << 8) | v) & 0xffffff;
- val = state;
+ val = state;
goto found;
}
state = ((state << 8) | v) & 0xffffff;
}
val = -1;
- found:
+
+found:
*header_state = state;
- *size_ptr = n;
+ *size_ptr = n;
return val;
}
-static int mpegps_read_header(AVFormatContext *s,
- AVFormatParameters *ap)
-{
- MpegDemuxContext *m = s->priv_data;
- m->header_state = 0xff;
- /* no need to do more */
- return 0;
-}
-
-static INT64 get_pts(ByteIOContext *pb, int c)
+/**
+ * Extract stream types from a program stream map
+ * According to ISO/IEC 13818-1 ('MPEG-2 Systems') table 2-35
+ *
+ * @return number of bytes occupied by PSM in the bitstream
+ */
+static long mpegps_psm_parse(MpegDemuxContext *m, AVIOContext *pb)
{
- INT64 pts;
- int val;
-
- if (c < 0)
- c = get_byte(pb);
- pts = (INT64)((c >> 1) & 0x07) << 30;
- val = get_be16(pb);
- pts |= (INT64)(val >> 1) << 15;
- val = get_be16(pb);
- pts |= (INT64)(val >> 1);
- return pts;
+ int psm_length, ps_info_length, es_map_length;
+
+ psm_length = avio_rb16(pb);
+ avio_r8(pb);
+ avio_r8(pb);
+ ps_info_length = avio_rb16(pb);
+
+ /* skip program_stream_info */
+ avio_skip(pb, ps_info_length);
+ es_map_length = avio_rb16(pb);
+
+ /* at least one es available? */
+ while (es_map_length >= 4) {
+ unsigned char type = avio_r8(pb);
+ unsigned char es_id = avio_r8(pb);
+ uint16_t es_info_length = avio_rb16(pb);
+
+ /* remember mapping from stream id to stream type */
+ m->psm_es_type[es_id] = type;
+ /* skip program_stream_info */
+ avio_skip(pb, es_info_length);
+ es_map_length -= 4 + es_info_length;
+ }
+ avio_rb32(pb); /* crc32 */
+ return 2 + psm_length;
}
-static int mpegps_read_packet(AVFormatContext *s,
- AVPacket *pkt)
+/* read the next PES header. Return its position in ppos
+ * (if not NULL), and its start code, pts and dts.
+ */
+static int mpegps_read_pes_header(AVFormatContext *s,
+ int64_t *ppos, int *pstart_code,
+ int64_t *ppts, int64_t *pdts)
{
MpegDemuxContext *m = s->priv_data;
- AVStream *st;
- int len, size, startcode, i, c, flags, header_len, type, codec_id;
- INT64 pts, dts;
-
+ int len, size, startcode, c, flags, header_len;
+ int pes_ext, ext2_len, id_ext, skip;
+ int64_t pts, dts;
+ int64_t last_sync = avio_tell(s->pb);
+
+error_redo:
+ avio_seek(s->pb, last_sync, SEEK_SET);
+redo:
/* next start code (should be immediately after) */
- redo:
m->header_state = 0xff;
- size = MAX_SYNC_SIZE;
- startcode = find_start_code(&s->pb, &size, &m->header_state);
- //printf("startcode=%x pos=0x%Lx\n", startcode, url_ftell(&s->pb));
- if (startcode < 0)
- return -EIO;
+ size = MAX_SYNC_SIZE;
+ startcode = find_next_start_code(s->pb, &size, &m->header_state);
+ last_sync = avio_tell(s->pb);
+ if (startcode < 0) {
+ if (s->pb->eof_reached)
+ return AVERROR_EOF;
+ // FIXME we should remember header_state
+ return AVERROR(EAGAIN);
+ }
+
if (startcode == PACK_START_CODE)
goto redo;
if (startcode == SYSTEM_HEADER_START_CODE)
goto redo;
- if (startcode == PADDING_STREAM ||
- startcode == PRIVATE_STREAM_2) {
- /* skip them */
- len = get_be16(&s->pb);
- url_fskip(&s->pb, len);
+ if (startcode == PADDING_STREAM) {
+ avio_skip(s->pb, avio_rb16(s->pb));
goto redo;
}
+ if (startcode == PRIVATE_STREAM_2) {
+ len = avio_rb16(s->pb);
+ if (!m->sofdec) {
+ while (len-- >= 6) {
+ if (avio_r8(s->pb) == 'S') {
+ uint8_t buf[5];
+ avio_read(s->pb, buf, sizeof(buf));
+ m->sofdec = !memcmp(buf, "ofdec", 5);
+ len -= sizeof(buf);
+ break;
+ }
+ }
+ m->sofdec -= !m->sofdec;
+ }
+ avio_skip(s->pb, len);
+ goto redo;
+ }
+ if (startcode == PROGRAM_STREAM_MAP) {
+ mpegps_psm_parse(m, s->pb);
+ goto redo;
+ }
+
/* find matching stream */
if (!((startcode >= 0x1c0 && startcode <= 0x1df) ||
(startcode >= 0x1e0 && startcode <= 0x1ef) ||
- (startcode == 0x1bd)))
+ (startcode == 0x1bd) || (startcode == 0x1fd)))
goto redo;
-
- len = get_be16(&s->pb);
- pts = AV_NOPTS_VALUE;
+ if (ppos) {
+ *ppos = avio_tell(s->pb) - 4;
+ }
+ len = avio_rb16(s->pb);
+ pts =
dts = AV_NOPTS_VALUE;
/* stuffing */
- for(;;) {
- c = get_byte(&s->pb);
+ for (;;) {
+ if (len < 1)
+ goto error_redo;
+ c = avio_r8(s->pb);
len--;
/* XXX: for mpeg1, should test only bit 7 */
- if (c != 0xff)
+ if (c != 0xff)
break;
}
if ((c & 0xc0) == 0x40) {
/* buffer scale & size */
- get_byte(&s->pb);
- c = get_byte(&s->pb);
+ avio_r8(s->pb);
+ c = avio_r8(s->pb);
len -= 2;
}
- if ((c & 0xf0) == 0x20) {
- pts = get_pts(&s->pb, c);
+ if ((c & 0xe0) == 0x20) {
+ dts =
+ pts = get_pts(s->pb, c);
len -= 4;
- } else if ((c & 0xf0) == 0x30) {
- pts = get_pts(&s->pb, c);
- dts = get_pts(&s->pb, -1);
- len -= 9;
+ if (c & 0x10) {
+ dts = get_pts(s->pb, -1);
+ len -= 5;
+ }
} else if ((c & 0xc0) == 0x80) {
/* mpeg 2 PES */
- if ((c & 0x30) != 0) {
- fprintf(stderr, "Encrypted multiplex not handled\n");
- return -EIO;
- }
- flags = get_byte(&s->pb);
- header_len = get_byte(&s->pb);
- len -= 2;
+ flags = avio_r8(s->pb);
+ header_len = avio_r8(s->pb);
+ len -= 2;
if (header_len > len)
- goto redo;
- if ((flags & 0xc0) == 0x80) {
- pts = get_pts(&s->pb, -1);
+ goto error_redo;
+ len -= header_len;
+ if (flags & 0x80) {
+ dts = pts = get_pts(s->pb, -1);
header_len -= 5;
- len -= 5;
- } if ((flags & 0xc0) == 0xc0) {
- pts = get_pts(&s->pb, -1);
- dts = get_pts(&s->pb, -1);
- header_len -= 10;
- len -= 10;
+ if (flags & 0x40) {
+ dts = get_pts(s->pb, -1);
+ header_len -= 5;
+ }
}
- len -= header_len;
- while (header_len > 0) {
- get_byte(&s->pb);
+ if (flags & 0x3f && header_len == 0) {
+ flags &= 0xC0;
+ av_log(s, AV_LOG_WARNING, "Further flags set but no bytes left\n");
+ }
+ if (flags & 0x01) { /* PES extension */
+ pes_ext = avio_r8(s->pb);
header_len--;
+ /* Skip PES private data, program packet sequence counter
+ * and P-STD buffer */
+ skip = (pes_ext >> 4) & 0xb;
+ skip += skip & 0x9;
+ if (pes_ext & 0x40 || skip > header_len) {
+ av_log(s, AV_LOG_WARNING, "pes_ext %X is invalid\n", pes_ext);
+ pes_ext = skip = 0;
+ }
+ avio_skip(s->pb, skip);
+ header_len -= skip;
+
+ if (pes_ext & 0x01) { /* PES extension 2 */
+ ext2_len = avio_r8(s->pb);
+ header_len--;
+ if ((ext2_len & 0x7f) > 0) {
+ id_ext = avio_r8(s->pb);
+ if ((id_ext & 0x80) == 0)
+ startcode = ((startcode & 0xff) << 8) | id_ext;
+ header_len--;
+ }
+ }
}
- }
- if (startcode == 0x1bd) {
- startcode = get_byte(&s->pb);
+ if (header_len < 0)
+ goto error_redo;
+ avio_skip(s->pb, header_len);
+ } else if (c != 0xf)
+ goto redo;
+
+ if (startcode == PRIVATE_STREAM_1 && !m->psm_es_type[startcode & 0xff]) {
+ startcode = avio_r8(s->pb);
len--;
- if (startcode >= 0x80 && startcode <= 0xbf) {
+ if (startcode >= 0x80 && startcode <= 0xcf) {
/* audio: skip header */
- get_byte(&s->pb);
- get_byte(&s->pb);
- get_byte(&s->pb);
+ avio_r8(s->pb);
+ avio_r8(s->pb);
+ avio_r8(s->pb);
len -= 3;
+ if (startcode >= 0xb0 && startcode <= 0xbf) {
+ /* MLP/TrueHD audio has a 4-byte header */
+ avio_r8(s->pb);
+ len--;
+ }
+ }
+ }
+ if (len < 0)
+ goto error_redo;
+ if (dts != AV_NOPTS_VALUE && ppos) {
+ int i;
+ for (i = 0; i < s->nb_streams; i++) {
+ if (startcode == s->streams[i]->id &&
+ s->pb->seekable /* index useless on streams anyway */) {
+ ff_reduce_index(s, i);
+ av_add_index_entry(s->streams[i], *ppos, dts, 0, 0,
+ AVINDEX_KEYFRAME /* FIXME keyframe? */);
+ }
}
}
+ *pstart_code = startcode;
+ *ppts = pts;
+ *pdts = dts;
+ return len;
+}
+
+static int mpegps_read_packet(AVFormatContext *s,
+ AVPacket *pkt)
+{
+ MpegDemuxContext *m = s->priv_data;
+ AVStream *st;
+ int len, startcode, i, es_type, ret;
+ enum AVCodecID codec_id = AV_CODEC_ID_NONE;
+ enum AVMediaType type;
+ int64_t pts, dts, dummy_pos; // dummy_pos is needed for the index building to work
+ uint8_t av_uninit(dvdaudio_substream_type);
+
+redo:
+ len = mpegps_read_pes_header(s, &dummy_pos, &startcode, &pts, &dts);
+ if (len < 0)
+ return len;
+
+ if (startcode == 0x1bd) {
+ dvdaudio_substream_type = avio_r8(s->pb);
+ avio_skip(s->pb, 3);
+ len -= 4;
+ }
+
/* now find stream */
- for(i=0;i<s->nb_streams;i++) {
+ for (i = 0; i < s->nb_streams; i++) {
st = s->streams[i];
if (st->id == startcode)
goto found;
}
- if (startcode >= 0x1e0 && startcode <= 0x1ef) {
- type = CODEC_TYPE_VIDEO;
- codec_id = CODEC_ID_MPEG1VIDEO;
+
+ es_type = m->psm_es_type[startcode & 0xff];
+ if (es_type > 0 && es_type != STREAM_TYPE_PRIVATE_DATA) {
+ if (es_type == STREAM_TYPE_VIDEO_MPEG1) {
+ codec_id = AV_CODEC_ID_MPEG2VIDEO;
+ type = AVMEDIA_TYPE_VIDEO;
+ } else if (es_type == STREAM_TYPE_VIDEO_MPEG2) {
+ codec_id = AV_CODEC_ID_MPEG2VIDEO;
+ type = AVMEDIA_TYPE_VIDEO;
+ } else if (es_type == STREAM_TYPE_AUDIO_MPEG1 ||
+ es_type == STREAM_TYPE_AUDIO_MPEG2) {
+ codec_id = AV_CODEC_ID_MP3;
+ type = AVMEDIA_TYPE_AUDIO;
+ } else if (es_type == STREAM_TYPE_AUDIO_AAC) {
+ codec_id = AV_CODEC_ID_AAC;
+ type = AVMEDIA_TYPE_AUDIO;
+ } else if (es_type == STREAM_TYPE_VIDEO_MPEG4) {
+ codec_id = AV_CODEC_ID_MPEG4;
+ type = AVMEDIA_TYPE_VIDEO;
+ } else if (es_type == STREAM_TYPE_VIDEO_H264) {
+ codec_id = AV_CODEC_ID_H264;
+ type = AVMEDIA_TYPE_VIDEO;
+ } else if (es_type == STREAM_TYPE_AUDIO_AC3) {
+ codec_id = AV_CODEC_ID_AC3;
+ type = AVMEDIA_TYPE_AUDIO;
+ } else {
+ goto skip;
+ }
+ } else if (startcode >= 0x1e0 && startcode <= 0x1ef) {
+ static const unsigned char avs_seqh[4] = { 0, 0, 1, 0xb0 };
+ unsigned char buf[8];
+
+ avio_read(s->pb, buf, 8);
+ avio_seek(s->pb, -8, SEEK_CUR);
+ if (!memcmp(buf, avs_seqh, 4) && (buf[6] != 0 || buf[7] != 1))
+ codec_id = AV_CODEC_ID_CAVS;
+ else
+ codec_id = AV_CODEC_ID_PROBE;
+ type = AVMEDIA_TYPE_VIDEO;
} else if (startcode >= 0x1c0 && startcode <= 0x1df) {
- type = CODEC_TYPE_AUDIO;
- codec_id = CODEC_ID_MP2;
- } else if (startcode >= 0x80 && startcode <= 0x9f) {
- type = CODEC_TYPE_AUDIO;
- codec_id = CODEC_ID_AC3;
+ type = AVMEDIA_TYPE_AUDIO;
+ codec_id = m->sofdec > 0 ? AV_CODEC_ID_ADPCM_ADX : AV_CODEC_ID_MP2;
+ } else if (startcode >= 0x80 && startcode <= 0x87) {
+ type = AVMEDIA_TYPE_AUDIO;
+ codec_id = AV_CODEC_ID_AC3;
+ } else if ((startcode >= 0x88 && startcode <= 0x8f) ||
+ (startcode >= 0x98 && startcode <= 0x9f)) {
+ /* 0x90 - 0x97 is reserved for SDDS in DVD specs */
+ type = AVMEDIA_TYPE_AUDIO;
+ codec_id = AV_CODEC_ID_DTS;
+ } else if (startcode >= 0xa0 && startcode <= 0xaf) {
+ type = AVMEDIA_TYPE_AUDIO;
+ codec_id = AV_CODEC_ID_PCM_DVD;
+ } else if (startcode >= 0xb0 && startcode <= 0xbf) {
+ type = AVMEDIA_TYPE_AUDIO;
+ codec_id = AV_CODEC_ID_TRUEHD;
+ } else if (startcode >= 0xc0 && startcode <= 0xcf) {
+ /* Used for both AC-3 and E-AC-3 in EVOB files */
+ type = AVMEDIA_TYPE_AUDIO;
+ codec_id = AV_CODEC_ID_AC3;
+ } else if (startcode >= 0x20 && startcode <= 0x3f) {
+ type = AVMEDIA_TYPE_SUBTITLE;
+ codec_id = AV_CODEC_ID_DVD_SUBTITLE;
+ } else if (startcode >= 0xfd55 && startcode <= 0xfd5f) {
+ type = AVMEDIA_TYPE_VIDEO;
+ codec_id = AV_CODEC_ID_VC1;
+ } else if (startcode == 0x1bd) {
+ // check dvd audio substream type
+ type = AVMEDIA_TYPE_AUDIO;
+ switch (dvdaudio_substream_type & 0xe0) {
+ case 0xa0:
+ codec_id = AV_CODEC_ID_PCM_DVD;
+ break;
+ case 0x80:
+ if ((dvdaudio_substream_type & 0xf8) == 0x88)
+ codec_id = AV_CODEC_ID_DTS;
+ else
+ codec_id = AV_CODEC_ID_AC3;
+ break;
+ default:
+ av_log(s, AV_LOG_ERROR, "Unknown 0x1bd sub-stream\n");
+ goto skip;
+ }
} else {
- skip:
+skip:
/* skip packet */
- url_fskip(&s->pb, len);
+ avio_skip(s->pb, len);
goto redo;
}
/* no stream found: add a new stream */
- st = av_new_stream(s, startcode);
- if (!st)
+ st = avformat_new_stream(s, NULL);
+ if (!st)
goto skip;
- st->codec.codec_type = type;
- st->codec.codec_id = codec_id;
- found:
- av_new_packet(pkt, len);
- //printf("\nRead Packet ID: %x PTS: %f Size: %d", startcode,
- // (float)pts/90000, len);
- get_buffer(&s->pb, pkt->data, pkt->size);
- pkt->pts = pts;
+ st->id = startcode;
+ st->codec->codec_type = type;
+ st->codec->codec_id = codec_id;
+ st->need_parsing = AVSTREAM_PARSE_FULL;
+
+found:
+ if (st->discard >= AVDISCARD_ALL)
+ goto skip;
+ ret = av_get_packet(s->pb, pkt, len);
+
+ pkt->pts = pts;
+ pkt->dts = dts;
+ pkt->pos = dummy_pos;
pkt->stream_index = st->index;
- return 0;
+ av_dlog(s, "%d: pts=%0.3f dts=%0.3f size=%d\n",
+ pkt->stream_index, pkt->pts / 90000.0, pkt->dts / 90000.0,
+ pkt->size);
+
+ return (ret < 0) ? ret : 0;
}
-static int mpegps_read_close(AVFormatContext *s)
+static int64_t mpegps_read_dts(AVFormatContext *s, int stream_index,
+ int64_t *ppos, int64_t pos_limit)
{
- return 0;
+ int len, startcode;
+ int64_t pos, pts, dts;
+
+ pos = *ppos;
+ if (avio_seek(s->pb, pos, SEEK_SET) < 0)
+ return AV_NOPTS_VALUE;
+
+ for (;;) {
+ len = mpegps_read_pes_header(s, &pos, &startcode, &pts, &dts);
+ if (len < 0) {
+ av_dlog(s, "none (ret=%d)\n", len);
+ return AV_NOPTS_VALUE;
+ }
+ if (startcode == s->streams[stream_index]->id &&
+ dts != AV_NOPTS_VALUE) {
+ break;
+ }
+ avio_skip(s->pb, len);
+ }
+ av_dlog(s, "pos=0x%"PRIx64" dts=0x%"PRIx64" %0.3f\n",
+ pos, dts, dts / 90000.0);
+ *ppos = pos;
+ return dts;
}
-static AVOutputFormat mpeg1system_mux = {
- "mpeg",
- "MPEG1 System format",
- "video/x-mpeg",
- "mpg,mpeg",
- sizeof(MpegMuxContext),
- CODEC_ID_MP2,
- CODEC_ID_MPEG1VIDEO,
- mpeg_mux_init,
- mpeg_mux_write_packet,
- mpeg_mux_end,
+AVInputFormat ff_mpegps_demuxer = {
+ .name = "mpeg",
+ .long_name = NULL_IF_CONFIG_SMALL("MPEG-PS (MPEG-2 Program Stream)"),
+ .priv_data_size = sizeof(MpegDemuxContext),
+ .read_probe = mpegps_probe,
+ .read_header = mpegps_read_header,
+ .read_packet = mpegps_read_packet,
+ .read_timestamp = mpegps_read_dts,
+ .flags = AVFMT_SHOW_IDS | AVFMT_TS_DISCONT,
};
-
-static AVOutputFormat mpeg1vcd_mux = {
- "vcd",
- "MPEG1 System format (VCD)",
- "video/x-mpeg",
- NULL,
- sizeof(MpegMuxContext),
- CODEC_ID_MP2,
- CODEC_ID_MPEG1VIDEO,
- mpeg_mux_init,
- mpeg_mux_write_packet,
- mpeg_mux_end,
-};
-
-static AVOutputFormat mpeg2vob_mux = {
- "vob",
- "MPEG2 PS format (VOB)",
- "video/x-mpeg",
- "vob",
- sizeof(MpegMuxContext),
- CODEC_ID_MP2,
- CODEC_ID_MPEG1VIDEO,
- mpeg_mux_init,
- mpeg_mux_write_packet,
- mpeg_mux_end,
-};
-
-static AVInputFormat mpegps_demux = {
- "mpeg",
- "MPEG PS format",
- sizeof(MpegDemuxContext),
- mpegps_probe,
- mpegps_read_header,
- mpegps_read_packet,
- mpegps_read_close,
- .flags = AVFMT_NOHEADER,
-};
-
-int mpegps_init(void)
-{
- av_register_output_format(&mpeg1system_mux);
- av_register_output_format(&mpeg1vcd_mux);
- av_register_output_format(&mpeg2vob_mux);
- av_register_input_format(&mpegps_demux);
- return 0;
-}